ti.sil/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-12-12 10:55:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: handle missing score in code suggestions and adjust indentation in pr_code_suggestions.py

Browse source
This commit is contained in:
mrT23 2024-09-01 08:47:23 +03:00
parent 5432469ef6
commit 9e063bf48a
No known key found for this signature in database
GPG key ID: D350490E39D5F5AD
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
pr_agent/tools/pr_code_suggestions.py
View file

@ -104,8 +104,8 @@ class PRCodeSuggestions:
if not data: if not data:
data = {"code_suggestions": []} data = {"code_suggestions": []}
if (data is None or 'code_suggestions' not in data or not data['code_suggestions'] if (data is None or 'code_suggestions' not in data or not data['code_suggestions']
and get_settings().config.publish_output): and get_settings().config.publish_output):
get_logger().warning('No code suggestions found for the PR.') get_logger().warning('No code suggestions found for the PR.')
pr_body = "## PR Code Suggestions ✨\n\nNo code suggestions found for the PR." pr_body = "## PR Code Suggestions ✨\n\nNo code suggestions found for the PR."
get_logger().debug(f"PR output", artifact=pr_body) get_logger().debug(f"PR output", artifact=pr_body)
@ -528,7 +528,7 @@ class PRCodeSuggestions:
for i, prediction in enumerate(predictions["code_suggestions"]): for i, prediction in enumerate(predictions["code_suggestions"]):
try: try:
if get_settings().pr_code_suggestions.self_reflect_on_suggestions: if get_settings().pr_code_suggestions.self_reflect_on_suggestions:
score = int(prediction["score"]) score = int(prediction.get("score", 1))
if score >= score_threshold: if score >= score_threshold:
data["code_suggestions"].append(prediction) data["code_suggestions"].append(prediction)
else: else: