From 8e6819c987337d217f49562e25cf26db381a1f02 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Tue, 29 Apr 2025 00:06:37 +0800 Subject: [PATCH] fix anthropic llm caller routing (#2241) --- skyvern/forge/sdk/api/llm/api_handler_factory.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/skyvern/forge/sdk/api/llm/api_handler_factory.py b/skyvern/forge/sdk/api/llm/api_handler_factory.py index 678edc79..021450cc 100644 --- a/skyvern/forge/sdk/api/llm/api_handler_factory.py +++ b/skyvern/forge/sdk/api/llm/api_handler_factory.py @@ -676,7 +676,7 @@ class LLMCaller: timeout: float = settings.LLM_CONFIG_TIMEOUT, **active_parameters: dict[str, Any], ) -> ModelResponse | CustomStreamWrapper | AnthropicMessage: - if self.llm_key and self.llm_key.startswith("ANTHROPIC"): + if self.llm_key and "ANTHROPIC" in self.llm_key: return await self._call_anthropic(messages, tools, timeout, **active_parameters) return await litellm.acompletion(