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

bump: version 0.13.3.dev1 → 0.13.3.dev2

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-06 06:44:15 -08:00
parent 9d673e3f8c
commit 4dd1913da1
3 changed files with 10 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/utils.py
View file

@ -828,7 +828,12 @@ def client(original_function):
model = args[0] if len(args) > 0 else kwargs["model"]
call_type = original_function.__name__
if call_type == CallTypes.completion.value:
messages = args[1] if len(args) > 1 else kwargs["messages"]
if len(args) > 1:
messages = args[1]
elif kwargs.get("messages", None):
messages = kwargs["messages"]
elif kwargs.get("prompt", None):
messages = kwargs["prompt"]
elif call_type == CallTypes.embedding.value:
messages = args[1] if len(args) > 1 else kwargs["input"]
stream = True if "stream" in kwargs and kwargs["stream"] == True else False