xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-04 04:40:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

Rename 'add_callbacks' to 'add_litellm_callbacks' for clarity in litellm_ai_handler

Browse Source
This commit is contained in:
mrT23
2024-08-17 09:20:30 +03:00
parent c76aabc71e
commit aa87bc60f6
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -89,7 +89,7 @@ class LiteLLMAIHandler(BaseAiHandler):
response_log['main_pr_language'] = 'unknown' response_log['main_pr_language'] = 'unknown'
return response_log return response_log
def add_callbacks(selfs, kwargs): def add_litellm_callbacks(selfs, kwargs) -> dict:
pr_metadata = [] pr_metadata = []
def capture_logs(message): def capture_logs(message):
@ -160,7 +160,7 @@ class LiteLLMAIHandler(BaseAiHandler):
} }
if get_settings().litellm.get("enable_callbacks", False): 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) seed = get_settings().config.get("seed", -1)
if temperature > 0 and seed >= 0: if temperature > 0 and seed >= 0: