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

Merge pull request #568 from Codium-ai/tr/user_description

Browse Source 
Enhancement: Improved Extraction and Placement of User Descriptions in PRs
This commit is contained in:
Tal
2024-01-04 08:35:34 -08:00
committed by GitHub
parent e6093cd768 6d18a0c843
commit 782f1ca1bd
2 changed files with 35 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pr_agent/tools/pr_description.py
View File

@ -185,7 +185,12 @@ class PRDescription:
# Load the AI prediction data into a dictionary
self.data = load_yaml(self.prediction.strip())
if get_settings().pr_description.add_original_user_description and self.user_description:
self.data["User Description"] = self.user_description
# re-order keys
if 'User Description' in self.data:
self.data['User Description'] = self.data.pop('User Description')
if 'title' in self.data:
self.data['title'] = self.data.pop('title')
if 'type' in self.data:
@ -197,8 +202,7 @@ class PRDescription:
if 'pr_files' in self.data:
self.data['pr_files'] = self.data.pop('pr_files')
if get_settings().pr_description.add_original_user_description and self.user_description:
self.data["User Description"] = self.user_description
def _prepare_labels(self) -> List[str]: