diff --git a/pkg/ai/iai.go b/pkg/ai/iai.go index e9a5618dca..8c94e17569 100644 --- 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 index 79f46d5aea..f70e7b5e68 100644 --- 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 }