ti.sil/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-12-12 10:55:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add logging to pr_agent.py

Browse source
This commit is contained in:
Brian Pham 2023-12-13 08:16:45 +08:00
parent be8d6af87f
commit ebb2ed891b
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
pr_agent/agent/pr_agent.py
View file

@ -4,6 +4,7 @@ from pr_agent.algo.ai_handlers.base_ai_handler import BaseAiHandler
from pr_agent.algo.utils import update_settings_from_args from pr_agent.algo.utils import update_settings_from_args
from pr_agent.config_loader import get_settings from pr_agent.config_loader import get_settings
from pr_agent.git_providers.utils import apply_repo_settings from pr_agent.git_providers.utils import apply_repo_settings
from pr_agent.log import get_logger
from pr_agent.tools.pr_add_docs import PRAddDocs from pr_agent.tools.pr_add_docs import PRAddDocs
from pr_agent.tools.pr_code_suggestions import PRCodeSuggestions from pr_agent.tools.pr_code_suggestions import PRCodeSuggestions
from pr_agent.tools.pr_config import PRConfig from pr_agent.tools.pr_config import PRConfig
@ -42,7 +43,7 @@ class PRAgent:
self.ai_handler = ai_handler self.ai_handler = ai_handler
pass pass
def has_ai_handler_param(cls): def has_ai_handler_param(cls: object):
constructor = getattr(cls, "__init__", None) constructor = getattr(cls, "__init__", None)
if constructor is not None: if constructor is not None:
parameters = inspect.signature(constructor).parameters parameters = inspect.signature(constructor).parameters
@ -73,9 +74,10 @@ class PRAgent:
if notify: if notify:
notify() notify()
if(not self.has_ai_handler_param(command2class[action])): get_logger().info(f"Class: {command2class[action]}")
if(not self.has_ai_handler_param(cls=command2class[action])):
await command2class[action](pr_url, args=args).run() await command2class[action](pr_url, args=args).run()
else else:
await command2class[action](pr_url, ai_handler=self.ai_handler, args=args).run() await command2class[action](pr_url, ai_handler=self.ai_handler, args=args).run()
else: else:
return False return False