From 441e098e2ae4af4d7c71abf6efaf499a9db2e14d Mon Sep 17 00:00:00 2001 From: mrT23 Date: Sun, 25 Aug 2024 11:26:48 +0300 Subject: [PATCH] fix: correct YAML formatting in response text processing in utils.py --- pr_agent/algo/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pr_agent/algo/utils.py b/pr_agent/algo/utils.py index 5572a38c..38a5e5dd 100644 --- a/pr_agent/algo/utils.py +++ b/pr_agent/algo/utils.py @@ -585,9 +585,9 @@ def try_fix_yaml(response_text: str, response_text_lines_copy = response_text_lines.copy() for i in range(0, len(response_text_lines_copy)): for key in keys_yaml: - if key in response_text_lines_copy[i] and not '|-' in response_text_lines_copy[i]: + if key in response_text_lines_copy[i] and not '|' in response_text_lines_copy[i]: response_text_lines_copy[i] = response_text_lines_copy[i].replace(f'{key}', - f'{key} |-\n ') + f'{key} |\n ') try: data = yaml.safe_load('\n'.join(response_text_lines_copy)) get_logger().info(f"Successfully parsed AI prediction after adding |-\n")