Merge pull request #795 from Codium-ai/tr/fixes4

Enhance AI handler logging and add main PR language attribute to AI h…

pr_agent/algo/ai_handlers/litellm_ai_handler.py

@@ -73,6 +73,18 @@ class LiteLLMAIHandler(BaseAiHandler):
                 region_name=get_settings().aws.bedrock_region,
             )
 
+    def prepare_logs(self, response, system, user, resp, finish_reason):
+        response_log = response.dict().copy()
+        response_log['system'] = system
+        response_log['user'] = user
+        response_log['output'] = resp
+        response_log['finish_reason'] = finish_reason
+        if hasattr(self, 'main_pr_language'):
+            response_log['main_pr_language'] = self.main_pr_language
+        else:
+            response_log['main_pr_language'] = 'unknown'
+        return response_log
+
     @property
     def deployment_id(self):
         """
@@ -125,10 +137,13 @@ class LiteLLMAIHandler(BaseAiHandler):
         else:
             resp = response["choices"][0]['message']['content']
             finish_reason = response["choices"][0]["finish_reason"]
-            # usage = response.get("usage")
             get_logger().debug(f"\nAI response:\n{resp}")
-            get_logger().debug("Full_response", artifact=response)
 
+            # log the full response for debugging
+            response_log = self.prepare_logs(response, system, user, resp, finish_reason)
+            get_logger().debug("Full_response", artifact=response_log)
+
+            # for CLI debugging
             if get_settings().config.verbosity_level >= 2:
                 get_logger().info(f"\nAI response:\n{resp}")
 

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