From 0f73f5f9064365b3a850c35759d6fcf2499649e1 Mon Sep 17 00:00:00 2001 From: mrT23 Date: Thu, 13 Jul 2023 17:53:17 +0300 Subject: [PATCH] set as title --- pr_agent/git_providers/github_provider.py | 4 ++++ pr_agent/tools/pr_description.py | 20 +++++++++++--------- 2 files changed, 15 insertions(+), 9 deletions(-) diff --git a/pr_agent/git_providers/github_provider.py b/pr_agent/git_providers/github_provider.py index 49b0ccbe..958a56e8 100644 --- a/pr_agent/git_providers/github_provider.py +++ b/pr_agent/git_providers/github_provider.py @@ -39,6 +39,10 @@ class GithubProvider: diff_files.append(FilePatchInfo(original_file_content_str, new_file_content_str, file.patch, file.filename)) 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): response = self.pr.create_issue_comment(pr_comment) if hasattr(response, "user") and hasattr(response.user, "login"): diff --git a/pr_agent/tools/pr_description.py b/pr_agent/tools/pr_description.py index 8ae01791..334678e4 100644 --- a/pr_agent/tools/pr_description.py +++ b/pr_agent/tools/pr_description.py @@ -43,10 +43,10 @@ class PRDescription: logging.info('Getting AI prediction...') self.prediction = await self._get_prediction() logging.info('Preparing answer...') - pr_comment = self._prepare_pr_answer() + pr_title, pr_body = self._prepare_pr_answer() if settings.config.publish_review: 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() return "" @@ -64,18 +64,20 @@ class PRDescription: system=system_prompt, user=user_prompt) return response - def _prepare_pr_answer(self) -> str: + def _prepare_pr_answer(self): data = json.loads(self.prediction) - markdown_text = "" + pr_body = "" # for key, value in data.items(): # markdown_text += f"## {key}\n\n" # markdown_text += f"{value}\n\n" + title = data['PR Title'] + del data['PR Title'] for key, value in data.items(): - markdown_text += f"{key}:\n" + pr_body += f"{key}:\n" if 'walkthrough' in key.lower(): - markdown_text += f"{value}\n" + pr_body += f"{value}\n" else: - markdown_text += f"**{value}**\n\n___\n" + pr_body += f"**{value}**\n\n___\n" if settings.config.verbosity_level >= 2: - logging.info(f"markdown_text:\n{markdown_text}") - return markdown_text + logging.info(f"title:\n{title}\n{pr_body}") + return title, pr_body