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

pr_agent/tools/pr_add_docs.py

@@ -26,6 +26,8 @@ class PRAddDocs:
         )
 
         self.ai_handler = ai_handler()
+        self.ai_handler.main_pr_language = self.main_language
+
         self.patches_diff = None
         self.prediction = None
         self.cli_mode = cli_mode

pr_agent/tools/pr_code_suggestions.py

@@ -46,6 +46,7 @@ class PRCodeSuggestions:
             num_code_suggestions = get_settings().pr_code_suggestions.num_code_suggestions
 
         self.ai_handler = ai_handler()
+        self.ai_handler.main_pr_language = self.main_language
         self.patches_diff = None
         self.prediction = None
         self.cli_mode = cli_mode

pr_agent/tools/pr_description.py

@@ -41,6 +41,7 @@ class PRDescription:
 
         # Initialize the AI handler
         self.ai_handler = ai_handler()
+        self.ai_handler.main_pr_language = self.main_pr_language
 
     
         # Initialize the variables dictionary

pr_agent/tools/pr_generate_labels.py

@@ -35,7 +35,8 @@ class PRGenerateLabels:
 
         # Initialize the AI handler
         self.ai_handler = ai_handler()
-    
+        self.ai_handler.main_pr_language = self.main_pr_language
+
         # Initialize the variables dictionary
         self.vars = {
             "title": self.git_provider.pr.title,

pr_agent/tools/pr_information_from_user.py

@@ -21,6 +21,8 @@ class PRInformationFromUser:
             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,
             "branch": self.git_provider.get_pr_branch(),

pr_agent/tools/pr_line_questions.py

@@ -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,

pr_agent/tools/pr_questions.py

@@ -22,6 +22,8 @@ class PRQuestions:
             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.question_str = question_str
         self.vars = {
             "title": self.git_provider.pr.title,

pr_agent/tools/pr_reviewer.py

@@ -46,6 +46,8 @@ class PRReviewer:
         if self.is_answer and not self.git_provider.is_supported("get_issue_comments"):
             raise Exception(f"Answer mode is not supported for {get_settings().config.git_provider} for now")
         self.ai_handler = ai_handler()
+        self.ai_handler.main_pr_language = self.main_language
+
         self.patches_diff = None
         self.prediction = None
 

pr_agent/tools/pr_update_changelog.py

@@ -26,7 +26,10 @@ class PRUpdateChangelog:
         )
         self.commit_changelog = get_settings().pr_update_changelog.push_changelog_changes
         self._get_changlog_file()  # self.changelog_file_str
+
         self.ai_handler = ai_handler()
+        self.ai_handler.main_pr_language = self.main_language
+
         self.patches_diff = None
         self.prediction = None
         self.cli_mode = cli_mode