From 5580548e7de4f038f5f7b28cb2188ebf62e87076 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Fri, 30 May 2025 16:16:07 -0700 Subject: [PATCH] support anthropic claude 4 bedrock (#2539) --- skyvern/forge/sdk/api/llm/config_registry.py | 40 ++++++++++---------- 1 file changed, 20 insertions(+), 20 deletions(-) diff --git a/skyvern/forge/sdk/api/llm/config_registry.py b/skyvern/forge/sdk/api/llm/config_registry.py index 0e37102c..731b7b8b 100644 --- a/skyvern/forge/sdk/api/llm/config_registry.py +++ b/skyvern/forge/sdk/api/llm/config_registry.py @@ -357,26 +357,6 @@ if settings.ENABLE_BEDROCK: add_assistant_prefix=True, ), ) - LLMConfigRegistry.register_config( - "BEDROCK_ANTHROPIC_CLAUDE4_OPUS", - LLMConfig( - "bedrock/anthropic.claude-opus-4-20250514-v1:0", - ["AWS_REGION"], - supports_vision=True, - add_assistant_prefix=True, - max_completion_tokens=8192, - ), - ) - LLMConfigRegistry.register_config( - "BEDROCK_ANTHROPIC_CLAUDE4_SONNET", - LLMConfig( - "bedrock/anthropic.claude-sonnet-4-20250514-v1:0", - ["AWS_REGION"], - supports_vision=True, - add_assistant_prefix=True, - max_completion_tokens=8192, - ), - ) LLMConfigRegistry.register_config( "BEDROCK_ANTHROPIC_CLAUDE3.7_SONNET_INFERENCE_PROFILE", LLMConfig( @@ -387,6 +367,26 @@ if settings.ENABLE_BEDROCK: max_completion_tokens=64000, ), ) + LLMConfigRegistry.register_config( + "BEDROCK_ANTHROPIC_CLAUDE4_SONNET_INFERENCE_PROFILE", + LLMConfig( + "bedrock/us.anthropic.claude-sonnet-4-20250514-v1:0", + ["AWS_REGION"], + supports_vision=True, + add_assistant_prefix=True, + max_completion_tokens=64000, + ), + ) + LLMConfigRegistry.register_config( + "BEDROCK_ANTHROPIC_CLAUDE4_OPUS_INFERENCE_PROFILE", + LLMConfig( + "bedrock/us.anthropic.claude-opus-4-20250514-v1:0", + ["AWS_REGION"], + supports_vision=True, + add_assistant_prefix=True, + max_completion_tokens=32000, + ), + ) if settings.ENABLE_AZURE: