adding step_exception to provide more signal of why the step failed (#730)

2024-08-28 11:06:04 +03:00
parent 6f4ff4339f
commit a6a545224a
4 changed files with 14 additions and 3 deletions
--- a/skyvern/forge/sdk/api/llm/exceptions.py
+++ b/skyvern/forge/sdk/api/llm/exceptions.py
@@ -40,6 +40,11 @@ class LLMProviderError(BaseLLMError):
        super().__init__(f"Error while using LLMProvider {llm_key}")


+class LLMProviderErrorRetryableTask(LLMProviderError):
+    def __init__(self, llm_key: str) -> None:
+        super().__init__(f"Retryable error while using LLMProvider {llm_key}")
+
+
 class NoProviderEnabledError(BaseLLMError):
    def __init__(self) -> None:
        super().__init__(