From a1fa5ee4e080f275492f2f4e0e2166b8b7e5784e Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Thu, 2 May 2024 11:20:49 -0700 Subject: [PATCH] log main_model_group when router.acompletion has an unexpected error (#249) --- skyvern/forge/sdk/api/llm/api_handler_factory.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/skyvern/forge/sdk/api/llm/api_handler_factory.py b/skyvern/forge/sdk/api/llm/api_handler_factory.py index 77092e93..130e60fa 100644 --- 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: