From f4b06640d2354d0632e8953251d8fadaa982fbe6 Mon Sep 17 00:00:00 2001 From: mrT23 Date: Wed, 14 Aug 2024 08:14:51 +0300 Subject: [PATCH] Add info log for successful AI prediction parse in utils.py --- pr_agent/algo/utils.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/pr_agent/algo/utils.py b/pr_agent/algo/utils.py index a8347cd4..513f3853 100644 --- a/pr_agent/algo/utils.py +++ b/pr_agent/algo/utils.py @@ -565,6 +565,9 @@ def load_yaml(response_text: str, keys_fix_yaml: List[str] = [], first_key="", l 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}) + else: + get_logger().info(f"Successfully parsed AI prediction after fallbacks", + artifact={'response_text': response_text}) return data