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

Merge pull request #462 from Codium-ai/tr/more_protections

Browse Source 
Enhancements in YAML Parsing and Error Handling
This commit is contained in:
mrT23
2023-11-19 07:40:06 -08:00
committed by GitHub
parent 5ca1748b93 3504a64269
commit dff4d1befc
4 changed files with 42 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
pr_agent/tools/pr_reviewer.py
View File

@ -278,14 +278,7 @@ class PRReviewer:
if get_settings().pr_reviewer.num_code_suggestions == 0:
return
review_text = self.prediction.strip()
review_text = review_text.removeprefix('```yaml').rstrip('`')
try:
data = yaml.load(review_text, Loader=SafeLoader)
except Exception as e:
get_logger().error(f"Failed to parse AI prediction: {e}")
data = try_fix_yaml(review_text)
data = load_yaml(self.prediction.strip())
comments: List[str] = []
for suggestion in data.get('PR Feedback', {}).get('Code feedback', []):
relevant_file = suggestion.get('relevant file', '').strip()