xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-05 05:10:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

fallback to try_fix_yaml

Browse Source
This commit is contained in:
tmokmss
2023-11-29 17:57:54 +09:00
parent f8f57419c4
commit 5e642c10fa
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
pr_agent/algo/utils.py
View File

@ -290,7 +290,6 @@ def _fix_key_value(key: str, value: str):
def load_yaml(response_text: str) -> dict: def load_yaml(response_text: str) -> dict:
response_text = response_text.removeprefix('```yaml').rstrip('`') response_text = response_text.removeprefix('```yaml').rstrip('`')
response_text = response_text.strip().rstrip().removeprefix('{').removesuffix('}')
try: try:
data = yaml.safe_load(response_text) data = yaml.safe_load(response_text)
except Exception as e: except Exception as e:
@ -326,7 +325,15 @@ def try_fix_yaml(response_text: str) -> dict:
break break
except: except:
pass pass
return data
# thrid fallback - try to remove leading and trailing curly brackets
response_text_copy = response_text.strip().rstrip().removeprefix('{').removesuffix('}')
try:
data = yaml.safe_load(response_text_copy,)
get_logger().info(f"Successfully parsed AI prediction after removing curly brackets")
return data
except:
pass
def set_custom_labels(variables): def set_custom_labels(variables):