mirror of
https://github.com/charmbracelet/crush.git
synced 2025-08-02 05:20:46 +03:00
Merge pull request #352 from charmbracelet/fix_providers
chore: fix custom headers and system prefix
This commit is contained in:
@@ -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 {
|
||||
|
||||
@@ -70,8 +70,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))
|
||||
|
||||
Reference in New Issue
Block a user