raise exception when non dict response (#4057)

2025-11-21 15:19:06 +08:00
parent d277882b72
commit 6358b8b1d7
2 changed files with 9 additions and 3 deletions
--- a/skyvern/forge/sdk/api/llm/exceptions.py
+++ b/skyvern/forge/sdk/api/llm/exceptions.py
@@ -24,6 +24,11 @@ class InvalidLLMResponseFormat(BaseLLMError):
        super().__init__(f"LLM response content is not a valid JSON: {response}")


+class InvalidLLMResponseType(BaseLLMError):
+    def __init__(self, response_type: str) -> None:
+        super().__init__(f"LLM response content is expected to be a dict, but got {response_type}")
+
+
 class DuplicateCustomLLMProviderError(BaseLLMError):
    def __init__(self, llm_key: str) -> None:
        super().__init__(f"Custom LLMProvider {llm_key} is already registered")