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

Refactor pr_code_suggestions logic and update prompts for clarity and consistency

Browse Source
This commit is contained in:
mrT23
2024-05-13 18:13:37 +03:00
parent 95c7b3f55c
commit b0aac4ec5d
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
pr_agent/tools/pr_code_suggestions.py
View File

@ -347,22 +347,22 @@ class PRCodeSuggestions:
prediction = await self._get_prediction(model, patches_diff)
prediction_list.append(prediction)
data = {"code_suggestions": []}
for predictions in enumerate(prediction_list):
if "code_suggestions" in enumerate(predictions):
score_threshold = max(1,get_settings().pr_code_suggestions.suggestions_score_threshold)
for i, prediction in predictions["code_suggestions"]:
for j, predictions in enumerate(prediction_list): # each call adds an element to the list
if "code_suggestions" in predictions:
score_threshold = max(1, get_settings().pr_code_suggestions.suggestions_score_threshold)
for i, prediction in enumerate(predictions["code_suggestions"]):
try:
if get_settings().pr_code_suggestions.self_reflect_on_suggestions:
score = int(prediction["score"])
if score >= score_threshold:
data["code_suggestions"].append(prediction)
else:
get_logger().info(f"Removing suggestions {i}, because score is {score}, and score_threshold is {score_threshold}",
get_logger().info(
f"Removing suggestions {i} from call {j}, because score is {score}, and score_threshold is {score_threshold}",
artifact=prediction)
else:
get_logger().error(f"Error getting PR diff, no code suggestions found in call {i + 1}")
data["code_suggestions"].append(prediction)
except Exception as e:
get_logger().error(f"Error getting PR diff, error: {e}")
self.data = data