From 939ae03f42e61d0944da80381219e6bbdfc2d850 Mon Sep 17 00:00:00 2001 From: Kujtim Hoxha Date: Wed, 9 Apr 2025 17:45:41 +0200 Subject: add bedrock support --- internal/llm/agent/agent.go | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) (limited to 'internal/llm/agent') 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 -- cgit v1.2.3