diff options
| author | Kujtim Hoxha <[email protected]> | 2025-04-09 17:45:41 +0200 |
|---|---|---|
| committer | Kujtim Hoxha <[email protected]> | 2025-04-09 17:45:41 +0200 |
| commit | 939ae03f42e61d0944da80381219e6bbdfc2d850 (patch) | |
| tree | 3c45cdad120f4e799e92f0a83fad97f8a025cf4f /internal/llm/agent | |
| parent | fde04bbf85ea641a33a282b354d63f227f9945fb (diff) | |
| download | opencode-939ae03f42e61d0944da80381219e6bbdfc2d850.tar.gz opencode-939ae03f42e61d0944da80381219e6bbdfc2d850.zip | |
add bedrock support
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 |