ti.sil/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-12-12 02:45:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add warning log for initial AI prediction parse failure and error log for fallback failure in utils.py

Browse source
This commit is contained in:
mrT23 2024-08-14 08:08:55 +03:00
parent 8414e109c5
commit f1981092d3
No known key found for this signature in database
GPG key ID: D350490E39D5F5AD
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
pr_agent/algo/utils.py
View file

@ -561,8 +561,10 @@ def load_yaml(response_text: str, keys_fix_yaml: List[str] = [], first_key="", l
try: try:
data = yaml.safe_load(response_text) data = yaml.safe_load(response_text)
except Exception as e: except Exception as e:
get_logger().error(f"Failed to parse AI prediction: {e}") get_logger().warning(f"Initial failure to parse AI prediction: {e}")
data = try_fix_yaml(response_text, keys_fix_yaml=keys_fix_yaml, first_key=first_key, last_key=last_key) data = try_fix_yaml(response_text, keys_fix_yaml=keys_fix_yaml, first_key=first_key, last_key=last_key)
if not data:
get_logger().error(f"Failed to parse AI prediction after fallbacks", artifact={'response_text': response_text})
return data return data