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

Merge pull request #1020 from Codium-ai/tr/fallbacks

Browse Source 
Tr/fallbacks
This commit is contained in:
Tal
2024-07-04 12:12:39 +03:00
committed by GitHub
parent f7a6e93b6c e30c70d2ca
commit 1f5c3a4c0f
4 changed files with 88 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pr_agent/tools/pr_code_suggestions.py
View File

@ -329,7 +329,8 @@ class PRCodeSuggestions:
def _prepare_pr_code_suggestions(self, predictions: str) -> Dict:
data = load_yaml(predictions.strip(),
keys_fix_yaml=["relevant_file", "suggestion_content", "existing_code", "improved_code"])
keys_fix_yaml=["relevant_file", "suggestion_content", "existing_code", "improved_code"],
first_key="code_suggestions",last_key="label")
if isinstance(data, list):
data = {'code_suggestions': data}