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

Enhance AI handler logging and add main PR language attribute to AI handler in various tools

Browse Source
This commit is contained in:
mrT23
2024-03-16 13:47:44 +02:00
parent 1593d8932b
commit 74345284bd
10 changed files with 32 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
pr_agent/tools/pr_line_questions.py
View File

@ -22,8 +22,11 @@ class PR_LineQuestions:
def __init__(self, pr_url: str, args=None, ai_handler: partial[BaseAiHandler,] = LiteLLMAIHandler):
self.question_str = self.parse_args(args)
self.git_provider = get_git_provider()(pr_url)
self.main_pr_language = get_main_pr_language(
self.git_provider.get_languages(), self.git_provider.get_files()
)
self.ai_handler = ai_handler()
self.ai_handler.main_pr_language = self.main_pr_language
self.vars = {
"title": self.git_provider.pr.title,