diff options
Diffstat (limited to 'internal/llm/agent')
| -rw-r--r-- | internal/llm/agent/agent.go | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/internal/llm/agent/agent.go b/internal/llm/agent/agent.go index baf78be65..78062d060 100644 --- a/internal/llm/agent/agent.go +++ b/internal/llm/agent/agent.go @@ -380,6 +380,29 @@ func getAgentProviders(ctx context.Context, model models.Model) (provider.Provid return nil, nil, err } + case models.ProviderBedrock: + var err error + agentProvider, err = provider.NewBedrockProvider( + provider.WithBedrockSystemMessage( + prompt.CoderAnthropicSystemPrompt(), + ), + provider.WithBedrockMaxTokens(maxTokens), + provider.WithBedrockModel(model), + ) + if err != nil { + return nil, nil, err + } + titleGenerator, err = provider.NewBedrockProvider( + provider.WithBedrockSystemMessage( + prompt.TitlePrompt(), + ), + provider.WithBedrockMaxTokens(maxTokens), + provider.WithBedrockModel(model), + ) + if err != nil { + return nil, nil, err + } + } return agentProvider, titleGenerator, nil |