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

feat: add support for PR changes diagram in PR description

Browse Source
This commit is contained in:
mrT23
2025-05-25 09:38:23 +03:00
parent 511f1ba6ae
commit 7273c9c0c1
3 changed files with 22 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pr_agent/tools/pr_description.py
View File

@ -73,6 +73,7 @@ class PRDescription:
"related_tickets": "",
"include_file_summary_changes": len(self.git_provider.get_diff_files()) <= self.COLLAPSIBLE_FILE_LIST_THRESHOLD,
'duplicate_prompt_examples': get_settings().config.get('duplicate_prompt_examples', False),
'enable_pr_diagram': get_settings().pr_description.get('enable_pr_diagram', False),
}
self.user_description = self.git_provider.get_user_description()
@ -257,6 +258,7 @@ class PRDescription:
tasks.append(task)
# Wait for all tasks to complete
results = await asyncio.gather(*tasks)
file_description_str_list = []
for i, result in enumerate(results):
prediction_files = result.strip().removeprefix('```yaml').strip('`').strip()
@ -456,6 +458,10 @@ class PRDescription:
self.data['labels'] = self.data.pop('labels')
if 'description' in self.data:
self.data['description'] = self.data.pop('description')
if 'changes_diagram' in self.data:
changes_diagram = self.data.pop('changes_diagram')
if changes_diagram.strip():
self.data['changes_diagram'] = changes_diagram
if 'pr_files' in self.data:
self.data['pr_files'] = self.data.pop('pr_files')