xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-02 11:50:37 +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

15
pr_agent/algo/ai_handlers/litellm_ai_handler.py
View File

@ -125,10 +125,21 @@ 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, including the system and user prompts
response_log = response.dict()
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'
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}")