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

Merge pull request #759 from Codium-ai/tr/fix_prompts

Browse Source 
fixed bug in inline_code_suggestions in review
This commit is contained in:
Tal
2024-03-09 21:53:26 -08:00
committed by GitHub
parent def78ea723 b4e9c55d8d
commit d1a2f596c7
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pr_agent/tools/pr_reviewer.py
View File

@ -175,7 +175,9 @@ class PRReviewer:
Prepare the PR review by processing the AI prediction and generating a markdown-formatted text that summarizes Prepare the PR review by processing the AI prediction and generating a markdown-formatted text that summarizes
the feedback. the feedback.
""" """
data = load_yaml(self.prediction.strip()) data = load_yaml(self.prediction.strip(),
keys_fix_yaml=["estimated_effort_to_review_[1-5]:", "security_concerns:", "possible_issues:",
"relevant_file:", "relevant_line:", "suggestion:"])
if 'code_feedback' in data: if 'code_feedback' in data:
code_feedback = data['code_feedback'] code_feedback = data['code_feedback']
@ -246,7 +248,7 @@ class PRReviewer:
keys_fix_yaml=["estimated_effort_to_review_[1-5]:", "security_concerns:", "possible_issues:", keys_fix_yaml=["estimated_effort_to_review_[1-5]:", "security_concerns:", "possible_issues:",
"relevant_file:", "relevant_line:", "suggestion:"]) "relevant_file:", "relevant_line:", "suggestion:"])
comments: List[str] = [] comments: List[str] = []
for suggestion in data.get('PR Feedback', {}).get('Code feedback', []): for suggestion in data.get('code_feedback', []):
relevant_file = suggestion.get('relevant_file', '').strip() relevant_file = suggestion.get('relevant_file', '').strip()
relevant_line_in_file = suggestion.get('relevant_line', '').strip() relevant_line_in_file = suggestion.get('relevant_line', '').strip()
content = suggestion.get('suggestion', '') content = suggestion.get('suggestion', '')