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

stable

Browse Source
This commit is contained in:
mrT23
2023-07-24 12:41:00 +03:00
parent 3819d52eb0
commit 57ff46ecc1
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pr_agent/tools/pr_description.py
View File

@ -20,11 +20,17 @@ class PRDescription:
Args: Args:
pr_url (str): The URL of the pull request. pr_url (str): The URL of the pull request.
""" """
# Initialize the git provider and main PR language
self.git_provider = get_git_provider()(pr_url) self.git_provider = get_git_provider()(pr_url)
self.main_pr_language = get_main_pr_language( self.main_pr_language = get_main_pr_language(
self.git_provider.get_languages(), self.git_provider.get_files() self.git_provider.get_languages(), self.git_provider.get_files()
) )
# Initialize the AI handler
self.ai_handler = AiHandler() self.ai_handler = AiHandler()
# Initialize the variables dictionary
self.vars = { self.vars = {
"title": self.git_provider.pr.title, "title": self.git_provider.pr.title,
"branch": self.git_provider.get_pr_branch(), "branch": self.git_provider.get_pr_branch(),
@ -32,12 +38,16 @@ class PRDescription:
"language": self.main_pr_language, "language": self.main_pr_language,
"diff": "", # empty diff for initial calculation "diff": "", # empty diff for initial calculation
} }
# Initialize the token handler
self.token_handler = TokenHandler( self.token_handler = TokenHandler(
self.git_provider.pr, self.git_provider.pr,
self.vars, self.vars,
settings.pr_description_prompt.system, settings.pr_description_prompt.system,
settings.pr_description_prompt.user, settings.pr_description_prompt.user,
) )
# Initialize patches_diff and prediction attributes
self.patches_diff = None self.patches_diff = None
self.prediction = None self.prediction = None