diff --git a/pr_agent/algo/pr_processing.py b/pr_agent/algo/pr_processing.py index dda0d9a2..20010694 100644 --- a/pr_agent/algo/pr_processing.py +++ b/pr_agent/algo/pr_processing.py @@ -329,9 +329,9 @@ async def retry_with_fallback_models(f: Callable, model_type: ModelType = ModelT ) get_settings().set("openai.deployment_id", deployment_id) return await f(model) - except: + except Exception as e: get_logger().warning( - f"Failed to generate prediction with {model}" + f"Failed to generate prediction with {model}: {e}", ) if i == len(all_models) - 1: # If it's the last iteration raise Exception(f"Failed to generate prediction with any model of {all_models}")