mirror of
https://github.com/qodo-ai/pr-agent.git
synced 2025-12-12 02:45:18 +00:00
chore: add error log on model prediction failure
This commit is contained in:
parent
fea8bc5150
commit
be93651db8
1 changed files with 2 additions and 2 deletions
|
|
@ -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}")
|
||||
|
|
|
|||
Loading…
Reference in a new issue