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

Merge pull request #1067 from Codium-ai/tr/custom_model

Browse Source 
docs: update usage guide and README; fix minor formatting issues in u…
This commit is contained in:
Tal
2024-07-28 09:34:05 +03:00
committed by GitHub
parent 866476080c c2c69f2950
commit af84409c1d
5 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pr_agent/algo/utils.py
View File

@ -557,7 +557,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 = response_text.removeprefix('```yaml').rstrip('`')
response_text = response_text.strip('\n').removeprefix('```yaml').rstrip('`')
try:
data = yaml.safe_load(response_text)
except Exception as e: