From 81525cd25afc459ac15ac1e70110794388ae9da4 Mon Sep 17 00:00:00 2001 From: mrT23 Date: Thu, 14 Aug 2025 18:49:06 +0300 Subject: [PATCH] fix: correct prefix handling in load_yaml function for improved YAML parsing --- 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 9b924554..bf1c7588 100644 --- a/pr_agent/algo/utils.py +++ b/pr_agent/algo/utils.py @@ -739,7 +739,7 @@ def _fix_key_value(key: str, value: str): def load_yaml(response_text: str, keys_fix_yaml: List[str] = [], first_key="", last_key="") -> dict: response_text_original = copy.deepcopy(response_text) - response_text = response_text.strip('\n').removeprefix('```yaml').rstrip().removesuffix('```') + response_text = response_text.strip('\n').removeprefix('yaml').removeprefix('```yaml').rstrip().removesuffix('```') try: data = yaml.safe_load(response_text) except Exception as e: