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

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
No known key found for this signature in database
GPG key ID: D350490E39D5F5AD
1 changed files with 2 additions and 2 deletions
Show stats 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: