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

protections

Browse Source
This commit is contained in:
mrT23
2024-03-05 18:34:18 +02:00
parent 359a15c84e
commit d77db93f80
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
pr_agent/tools/pr_code_suggestions.py
View File

@ -87,9 +87,14 @@ class PRCodeSuggestions:
else: else:
data = await retry_with_fallback_models(self._prepare_prediction_extended, ModelType.TURBO) data = await retry_with_fallback_models(self._prepare_prediction_extended, ModelType.TURBO)
if (not data) or (not 'code_suggestions' in data) or (not data['code_suggestions']):
if (not data) or (not 'code_suggestions' in data): get_logger().error('No code suggestions found for PR.')
get_logger().info('No code suggestions found for PR.') pr_body = "## PR Code Suggestions\n\nNo code suggestions found for PR."
get_logger().debug(f"PR output", artifact=pr_body)
if self.progress_response:
self.git_provider.edit_comment(self.progress_response, body=pr_body)
else:
self.git_provider.publish_comment(pr_body)
return return
if (not self.is_extended and get_settings().pr_code_suggestions.rank_suggestions) or \ if (not self.is_extended and get_settings().pr_code_suggestions.rank_suggestions) or \
@ -179,6 +184,7 @@ class PRCodeSuggestions:
suggestion_list = [] suggestion_list = []
one_sentence_summary_list = [] one_sentence_summary_list = []
for i, suggestion in enumerate(data['code_suggestions']): for i, suggestion in enumerate(data['code_suggestions']):
continue
if get_settings().pr_code_suggestions.summarize: if get_settings().pr_code_suggestions.summarize:
if not suggestion or 'one_sentence_summary' not in suggestion or 'label' not in suggestion or 'relevant_file' not in suggestion: if not suggestion or 'one_sentence_summary' not in suggestion or 'label' not in suggestion or 'relevant_file' not in suggestion:
get_logger().debug(f"Skipping suggestion {i + 1}, because it is invalid: {suggestion}") get_logger().debug(f"Skipping suggestion {i + 1}, because it is invalid: {suggestion}")
@ -317,7 +323,7 @@ class PRCodeSuggestions:
suggestion_list.append(suggestion) suggestion_list.append(suggestion)
data_sorted = [[]] * len(suggestion_list) data_sorted = [[]] * len(suggestion_list)
if len(suggestion_list ) == 1: if len(suggestion_list) == 1:
return suggestion_list return suggestion_list
try: try: