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

feat: add TODO comments to PR review output

Browse Source
This commit is contained in:
joosomi
2025-05-24 01:52:52 +09:00
parent 4e80f3999c
commit 788c0c12e6
5 changed files with 73 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
pr_agent/algo/utils.py
View File

@ -131,6 +131,7 @@ def convert_to_markdown_v2(output_data: dict,
"Focused PR": "✨",
"Relevant ticket": "🎫",
"Security concerns": "🔒",
"Todo sections": "✅",
"Insights from user's answers": "📝",
"Code feedback": "🤖",
"Estimated effort to review [1-5]": "⏱️",
@ -153,7 +154,7 @@ def convert_to_markdown_v2(output_data: dict,
for key, value in output_data['review'].items():
if value is None or value == '' or value == {} or value == []:
if key.lower() not in ['can_be_split', 'key_issues_to_review']:
if key.lower() not in ['can_be_split', 'key_issues_to_review', 'todo_sections']:
continue
key_nice = key.replace('_', ' ').capitalize()
emoji = emojis.get(key_nice, "")
@ -209,6 +210,50 @@ def convert_to_markdown_v2(output_data: dict,
markdown_text += f"### {emoji} Security concerns\n\n"
value = emphasize_header(value.strip(), only_markdown=True)
markdown_text += f"{value}\n\n"
elif 'todo sections' in key_nice.lower():
if gfm_supported:
markdown_text += f"<tr><td>"
if is_value_no(value):
markdown_text += f"{emoji}&nbsp;<strong>No ToDo sections</strong>"
else:
markdown_text += f"{emoji}&nbsp;<strong>ToDo sections</strong><br><br>\n\n"
if isinstance(value, list):
markdown_text += "<ul>\n"
for todo_item in value:
relevant_file = todo_item.get('relevant_file', '')
line_number = todo_item.get('line_number', '')
content = todo_item.get('content', '')
reference_link = None
if git_provider and relevant_file and line_number:
reference_link = git_provider.get_line_link(relevant_file, line_number, line_number)
file_line = f"{relevant_file} [{line_number}]"
if reference_link:
file_line = f"<a href='{reference_link}'>{file_line}</a>"
if content.strip():
markdown_text += f"<li>{file_line}: {content}</li>\n"
else:
markdown_text += f"<li>{file_line}</li>\n"
markdown_text += "</ul>\n"
markdown_text += f"</td></tr>\n"
else:
if is_value_no(value):
markdown_text += f"### {emoji} No ToDo sections\n\n"
else:
markdown_text += f"### {emoji} ToDo sections\n\n"
if isinstance(value, list):
for todo_item in value:
relevant_file = todo_item.get('relevant_file', '')
line_number = todo_item.get('line_number', '')
content = todo_item.get('content', '')
if git_provider and relevant_file and line_number:
reference_link = git_provider.get_line_link(relevant_file, line_number, line_number)
file_line = f"[{relevant_file} [{line_number}]]({reference_link})"
else:
file_line = f"{relevant_file} [{line_number}]"
if content.strip():
markdown_text += f"- {file_line}: {content}\n"
else:
markdown_text += f"- {file_line}\n"
elif 'can be split' in key_nice.lower():
if gfm_supported:
markdown_text += f"<tr><td>"