phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(utils.py): set call_type at the top of the function

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-31 18:23:44 -08:00
parent 9dc972de70
commit 93a52a2d35
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/utils.py
View file

@ -2094,13 +2094,13 @@ def client(original_function):
logging_obj = kwargs.get("litellm_logging_obj", None)
# only set litellm_call_id if its not in kwargs
call_type = original_function.__name__
if "litellm_call_id" not in kwargs:
kwargs["litellm_call_id"] = str(uuid.uuid4())
try:
model = args[0] if len(args) > 0 else kwargs["model"]
except:
model = None
call_type = original_function.__name__
if (
call_type != CallTypes.image_generation.value
and call_type != CallTypes.text_completion.value