Commit messages in pr-description

This commit is contained in:
mrT23
2023-08-01 15:15:59 +03:00
parent 6605f9c444
commit 8c0370a166
5 changed files with 31 additions and 2 deletions

View File

@ -121,3 +121,6 @@ class BitbucketProvider:
def _get_pr_file_content(self, remote_link: str):
return ""
def get_commit_messages(self):
return "" # not implemented yet

View File

@ -343,4 +343,19 @@ class GithubProvider(GitProvider):
return [label.name for label in self.pr.labels]
except Exception as e:
logging.exception(f"Failed to get labels, error: {e}")
return []
return []
def get_commit_messages(self) -> str:
"""
Retrieves the commit messages of a pull request.
Returns:
str: A string containing the commit messages of the pull request.
"""
try:
commit_list = list(self.pr.get_commits())
commit_messages = [commit.commit.message for commit in commit_list]
commit_messages_str = "\n".join([f"{i + 1}. {message}" for i, message in enumerate(commit_messages)])
except:
commit_messages_str = ""
return commit_messages_str

View File

@ -298,3 +298,6 @@ class GitLabProvider(GitProvider):
def get_labels(self):
return self.mr.labels
def get_commit_messages(self):
return "" # not implemented yet

View File

@ -36,8 +36,14 @@ Don't repeat the prompt in the answer, and avoid outputting the 'type' and 'desc
user="""PR Info:
Branch: '{{branch}}'
{%- if language %}
Main language: {{language}}
{%- endif %}
{%- if commit_messages_str %}
Commit messages:
{{commit_messages_str}}
{%- endif %}
The PR Git Diff:

View File

@ -29,7 +29,8 @@ class PRDescription:
self.main_pr_language = get_main_pr_language(
self.git_provider.get_languages(), self.git_provider.get_files()
)
commit_messages_str = self.git_provider.get_commit_messages()
# Initialize the AI handler
self.ai_handler = AiHandler()
@ -41,6 +42,7 @@ class PRDescription:
"language": self.main_pr_language,
"diff": "", # empty diff for initial calculation
"extra_instructions": settings.pr_description.extra_instructions,
"commit_messages_str": commit_messages_str,
}
# Initialize the token handler