From c59e9f77a66bdbf6195c867a276324cbb020ee80 Mon Sep 17 00:00:00 2001 From: "Hussam.lawen" Date: Mon, 18 Dec 2023 17:06:02 +0200 Subject: [PATCH] fix --- pr_agent/algo/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pr_agent/algo/utils.py b/pr_agent/algo/utils.py index 303b3da5..0ec1397b 100644 --- a/pr_agent/algo/utils.py +++ b/pr_agent/algo/utils.py @@ -328,7 +328,7 @@ def load_yaml(response_text: str) -> dict: def try_fix_yaml(response_text: str) -> dict: response_text_lines = response_text.split('\n') - keys = ['relevant line:', 'suggestion content:', 'relevant file:'] + keys = ['relevant line:', 'suggestion content:', 'relevant file:', 'existing code:', 'improved code:'] # first fallback - try to convert 'relevant line: ...' to relevant line: |-\n ...' response_text_lines_copy = response_text_lines.copy() for i in range(0, len(response_text_lines_copy)):