From aa87bc60f6fa448674bd4cd3d4d32a10663fc245 Mon Sep 17 00:00:00 2001 From: mrT23 Date: Sat, 17 Aug 2024 09:20:30 +0300 Subject: [PATCH] Rename 'add_callbacks' to 'add_litellm_callbacks' for clarity in litellm_ai_handler --- pr_agent/algo/ai_handlers/litellm_ai_handler.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pr_agent/algo/ai_handlers/litellm_ai_handler.py b/pr_agent/algo/ai_handlers/litellm_ai_handler.py index 1aa93b36..b4e3d085 100644 --- a/pr_agent/algo/ai_handlers/litellm_ai_handler.py +++ b/pr_agent/algo/ai_handlers/litellm_ai_handler.py @@ -89,7 +89,7 @@ class LiteLLMAIHandler(BaseAiHandler): response_log['main_pr_language'] = 'unknown' return response_log - def add_callbacks(selfs, kwargs): + def add_litellm_callbacks(selfs, kwargs) -> dict: pr_metadata = [] def capture_logs(message): @@ -160,7 +160,7 @@ class LiteLLMAIHandler(BaseAiHandler): } if get_settings().litellm.get("enable_callbacks", False): - kwargs = self.add_callbacks(kwargs) + kwargs = self.add_litellm_callbacks(kwargs) seed = get_settings().config.get("seed", -1) if temperature > 0 and seed >= 0: