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

set as title

Browse Source
This commit is contained in:
mrT23
2023-07-13 17:53:17 +03:00
parent 914cc6639a
commit 0f73f5f906
2 changed files with 15 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pr_agent/git_providers/github_provider.py
View File

@ -39,6 +39,10 @@ class GithubProvider:
diff_files.append(FilePatchInfo(original_file_content_str, new_file_content_str, file.patch, file.filename)) diff_files.append(FilePatchInfo(original_file_content_str, new_file_content_str, file.patch, file.filename))
return diff_files return diff_files
def publish_description(self, pr_title: str, pr_body: str):
self.pr.edit(title=pr_title, body=pr_body)
# self.pr.create_issue_comment(pr_comment)
def publish_comment(self, pr_comment: str, is_temporary: bool = False): def publish_comment(self, pr_comment: str, is_temporary: bool = False):
response = self.pr.create_issue_comment(pr_comment) response = self.pr.create_issue_comment(pr_comment)
if hasattr(response, "user") and hasattr(response.user, "login"): if hasattr(response, "user") and hasattr(response.user, "login"):

20
pr_agent/tools/pr_description.py
View File

@ -43,10 +43,10 @@ class PRDescription:
logging.info('Getting AI prediction...') logging.info('Getting AI prediction...')
self.prediction = await self._get_prediction() self.prediction = await self._get_prediction()
logging.info('Preparing answer...') logging.info('Preparing answer...')
pr_comment = self._prepare_pr_answer() pr_title, pr_body = self._prepare_pr_answer()
if settings.config.publish_review: if settings.config.publish_review:
logging.info('Pushing answer...') logging.info('Pushing answer...')
self.git_provider.publish_comment(pr_comment) self.git_provider.publish_description(pr_title, pr_body)
self.git_provider.remove_initial_comment() self.git_provider.remove_initial_comment()
return "" return ""
@ -64,18 +64,20 @@ class PRDescription:
system=system_prompt, user=user_prompt) system=system_prompt, user=user_prompt)
return response return response
def _prepare_pr_answer(self) -> str: def _prepare_pr_answer(self):
data = json.loads(self.prediction) data = json.loads(self.prediction)
markdown_text = "" pr_body = ""
# for key, value in data.items(): # for key, value in data.items():
# markdown_text += f"## {key}\n\n" # markdown_text += f"## {key}\n\n"
# markdown_text += f"{value}\n\n" # markdown_text += f"{value}\n\n"
title = data['PR Title']
del data['PR Title']
for key, value in data.items(): for key, value in data.items():
markdown_text += f"{key}:\n" pr_body += f"{key}:\n"
if 'walkthrough' in key.lower(): if 'walkthrough' in key.lower():
markdown_text += f"{value}\n" pr_body += f"{value}\n"
else: else:
markdown_text += f"**{value}**\n\n___\n" pr_body += f"**{value}**\n\n___\n"
if settings.config.verbosity_level >= 2: if settings.config.verbosity_level >= 2:
logging.info(f"markdown_text:\n{markdown_text}") logging.info(f"title:\n{title}\n{pr_body}")
return markdown_text return title, pr_body