chore: fix custom headers and system prefix

This commit is contained in:
Kujtim Hoxha
2025-07-30 12:34:02 +02:00
parent 229c0c79b4
commit 11a90e1cf6
2 changed files with 14 additions and 12 deletions

View File

@@ -145,16 +145,17 @@ func (c *Config) configureProviders(env env.Env, resolver VariableResolver, know
maps.Copy(headers, config.ExtraHeaders)
}
prepared := ProviderConfig{
ID: string(p.ID),
Name: p.Name,
BaseURL: p.APIEndpoint,
APIKey: p.APIKey,
Type: p.Type,
Disable: config.Disable,
ExtraHeaders: headers,
ExtraBody: config.ExtraBody,
ExtraParams: make(map[string]string),
Models: p.Models,
ID: string(p.ID),
Name: p.Name,
BaseURL: p.APIEndpoint,
APIKey: p.APIKey,
Type: p.Type,
Disable: config.Disable,
SystemPromptPrefix: config.SystemPromptPrefix,
ExtraHeaders: headers,
ExtraBody: config.ExtraBody,
ExtraParams: make(map[string]string),
Models: p.Models,
}
switch p.ID {

View File

@@ -67,8 +67,8 @@ func createAnthropicClient(opts providerClientOptions, useBedrock bool) anthropi
if useBedrock {
anthropicClientOptions = append(anthropicClientOptions, bedrock.WithLoadDefaultConfig(context.Background()))
}
for _, header := range opts.extraHeaders {
anthropicClientOptions = append(anthropicClientOptions, option.WithHeaderAdd(header, opts.extraHeaders[header]))
for key, header := range opts.extraHeaders {
anthropicClientOptions = append(anthropicClientOptions, option.WithHeaderAdd(key, header))
}
for key, value := range opts.extraBody {
anthropicClientOptions = append(anthropicClientOptions, option.WithJSONSet(key, value))
@@ -223,6 +223,7 @@ func (a *anthropicClient) preparedMessages(messages []anthropic.MessageParam, to
}
systemBlocks := []anthropic.TextBlockParam{}
slog.Info("Testing", "prefix", a.providerOptions.systemPromptPrefix)
// Add custom system prompt prefix if configured
if a.providerOptions.systemPromptPrefix != "" {