fix anthropic llm caller routing (#2241)

2025-04-29 00:06:37 +08:00
parent 9e7906a083
commit 8e6819c987
1 changed files with 1 additions and 1 deletions
--- 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(