fix timeout for svg and css LLM call (#2662)

Co-authored-by: lawyzheng <lawyzheng1106@gmail.com>
2025-06-10 00:23:15 -07:00
parent 4d397e73b7
commit 09ca00e948
2 changed files with 26 additions and 7 deletions
--- a/skyvern/forge/sdk/api/llm/api_handler_factory.py
+++ b/skyvern/forge/sdk/api/llm/api_handler_factory.py
@@ -145,7 +145,9 @@ class LLMAPIHandlerFactory:
                ai_suggestion=ai_suggestion,
            )
            try:
-                response = await router.acompletion(model=main_model_group, messages=messages, **parameters)
+                response = await router.acompletion(
+                    model=main_model_group, messages=messages, timeout=settings.LLM_CONFIG_TIMEOUT, **parameters
+                )
            except litellm.exceptions.APIError as e:
                raise LLMProviderErrorRetryableTask(local_llm_key) from e
            except litellm.exceptions.ContextWindowExceededError as e: