From f1981092d3886879d7e257f62a838bfe2029e8f4 Mon Sep 17 00:00:00 2001 From: mrT23 Date: Wed, 14 Aug 2024 08:08:55 +0300 Subject: [PATCH] Add warning log for initial AI prediction parse failure and error log for fallback failure in utils.py --- pr_agent/algo/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pr_agent/algo/utils.py b/pr_agent/algo/utils.py index 6c90731f..a8347cd4 100644 --- a/pr_agent/algo/utils.py +++ b/pr_agent/algo/utils.py @@ -561,8 +561,10 @@ def load_yaml(response_text: str, keys_fix_yaml: List[str] = [], first_key="", l try: data = yaml.safe_load(response_text) 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) + if not data: + get_logger().error(f"Failed to parse AI prediction after fallbacks", artifact={'response_text': response_text}) return data