log main_model_group when router.acompletion has an unexpected error (#249)

2024-05-02 11:20:49 -07:00
parent b6a85cf3a5
commit a1fa5ee4e0
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
@@ -99,7 +99,7 @@ class LLMAPIHandlerFactory:
            except openai.OpenAIError as e:
                raise LLMProviderError(llm_key) from e
            except Exception as e:
-                LOG.exception("LLM request failed unexpectedly", llm_key=llm_key)
+                LOG.exception("LLM request failed unexpectedly", llm_key=llm_key, model=main_model_group)
                raise LLMProviderError(llm_key) from e

            if step: