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(