feat: enable to source customHeaders from the config file

Signed-off-by: Guido Muscioni <[email protected]>
k8sgpt-ai · Dec 3, 2024 · 8ec2db0 · 8ec2db0
1 parent c3f4486
commit 8ec2db0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pkg/ai/iai.go b/pkg/ai/iai.go
@@ -119,7 +119,7 @@ type AIProvider struct {
 	TopK           int32         `mapstructure:"topk" yaml:"topk,omitempty"`
 	MaxTokens      int           `mapstructure:"maxtokens" yaml:"maxtokens,omitempty"`
 	OrganizationId string        `mapstructure:"organizationid" yaml:"organizationid,omitempty"`
-	CustomHeaders  []http.Header `mapstructure:"customHeaders"`
+	CustomHeaders  []http.Header `mapstructure:"customHeaders" yaml:"customHeaders,omitempty"`
 }
 
 func (p *AIProvider) GetBaseURL() string {

diff --git a/pkg/analysis/analysis.go b/pkg/analysis/analysis.go
@@ -155,7 +155,7 @@ func NewAnalysis(
 
 	aiClient := ai.NewClient(aiProvider.Name)
 	customHeaders := util.NewHeaders(httpHeaders)
-	aiProvider.CustomHeaders = customHeaders
+	aiProvider.CustomHeaders = append(aiProvider.CustomHeaders, customHeaders...)
 	if err := aiClient.Configure(&aiProvider); err != nil {
 		return nil, err
 	}