From 2b23700aec5b7cb6a9f9cfa8d3aeca9034f1f4bd Mon Sep 17 00:00:00 2001 From: mrT23 Date: Thu, 21 Dec 2023 08:24:07 +0200 Subject: [PATCH] feat: Enhance YAML parsing with additional fallbacks and key customization in load_yaml and try_fix_yaml functions --- 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 1bf37518..2b030bf2 100644 --- a/pr_agent/algo/utils.py +++ b/pr_agent/algo/utils.py @@ -326,7 +326,7 @@ def load_yaml(response_text: str, keys_fix_yaml: List[str] = []) -> dict: return data -def try_fix_yaml(response_text: str, keys_fix_yaml: List[str]) -> dict: +def try_fix_yaml(response_text: str, keys_fix_yaml: List[str] = []) -> dict: response_text_lines = response_text.split('\n') keys = ['relevant line:', 'suggestion content:', 'relevant file:', 'existing code:', 'improved code:']