phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

bug fixes to client

Browse source
This commit is contained in:
Krrish Dholakia 2023-08-28 15:09:14 -07:00
parent e8eb92c108
commit 2462791f65
4 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

BIN
litellm/__pycache__/__init__.cpython-311.pyc
View file

Binary file not shown.

BIN
litellm/__pycache__/main.cpython-311.pyc
View file

Binary file not shown.

BIN
litellm/__pycache__/utils.cpython-311.pyc
View file

Binary file not shown.

9
litellm/utils.py
View file

@ -452,17 +452,18 @@ def client(original_function):
# CRASH REPORTING TELEMETRY # CRASH REPORTING TELEMETRY
crash_reporting(*args, **kwargs) crash_reporting(*args, **kwargs)
# INIT LOGGER - for user-specified integrations # INIT LOGGER - for user-specified integrations
model = args[0] if len(args) > 1 else kwargs["model"] print(f"len args: {len(args)}")
model = args[0] if len(args) > 0 else kwargs["model"]
call_type = original_function.__name__ call_type = original_function.__name__
if call_type == CallTypes.completion.value: if call_type == CallTypes.completion.value:
messages = args[1] if len(args) > 2 else kwargs["messages"] messages = args[1] if len(args) > 1 else kwargs["messages"]
elif call_type == CallTypes.embedding.value: elif call_type == CallTypes.embedding.value:
messages = args[1] if len(args) > 2 else kwargs["input"] messages = args[1] if len(args) > 1 else kwargs["input"]
stream = True if "stream" in kwargs and kwargs["stream"] == True else False stream = True if "stream" in kwargs and kwargs["stream"] == True else False
logging_obj = Logging(model=model, messages=messages, stream=stream, litellm_call_id=kwargs["litellm_call_id"], call_type=call_type) logging_obj = Logging(model=model, messages=messages, stream=stream, litellm_call_id=kwargs["litellm_call_id"], call_type=call_type)
return logging_obj return logging_obj
except: # DO NOT BLOCK running the function because of this except: # DO NOT BLOCK running the function because of this
print_verbose(f"[Non-Blocking] {traceback.format_exc()}") print_verbose(f"[Non-Blocking] {traceback.format_exc()}; args - {args}; kwargs - {kwargs}")
pass pass
def crash_reporting(*args, **kwargs): def crash_reporting(*args, **kwargs):