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

feat: Enhance YAML parsing with additional fallbacks and key customization in load_yaml and try_fix_yaml functions

Browse Source
This commit is contained in:
mrT23
2023-12-21 08:24:07 +02:00
parent 553dad0bee
commit 2b23700aec
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pr_agent/algo/utils.py
View File

@ -326,7 +326,7 @@ def load_yaml(response_text: str, keys_fix_yaml: List[str] = []) -> dict:
return data 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') response_text_lines = response_text.split('\n')
keys = ['relevant line:', 'suggestion content:', 'relevant file:', 'existing code:', 'improved code:'] keys = ['relevant line:', 'suggestion content:', 'relevant file:', 'existing code:', 'improved code:']