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

adding ability to turn off litedebugger but still use key manager

Browse source
This commit is contained in:
Krrish Dholakia 2023-08-26 11:03:40 -07:00
parent e223fbbb9e
commit 9008973826
8 changed files with 8 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/utils.py
View file

@ -334,16 +334,17 @@ def client(original_function):
global callback_list, add_breadcrumb, user_logger_fn
if litellm.email is not None or os.getenv("LITELLM_EMAIL", None) is not None or litellm.token is not None or os.getenv("LITELLM_TOKEN", None): # add to input, success and failure callbacks if user is using hosted product
get_all_keys()
if "lite_debugger" not in callback_list:
if "lite_debugger" not in callback_list and litellm.logging:
litellm.input_callback.append("lite_debugger")
litellm.success_callback.append("lite_debugger")
litellm.failure_callback.append("lite_debugger")
elif litellm.use_client:
# create a litellm token for users
litellm.token = get_or_generate_uuid()
litellm.input_callback.append("lite_debugger")
litellm.success_callback.append("lite_debugger")
litellm.failure_callback.append("lite_debugger")
if litellm.logging:
litellm.input_callback.append("lite_debugger")
litellm.success_callback.append("lite_debugger")
litellm.failure_callback.append("lite_debugger")
if (
len(litellm.input_callback) > 0
or len(litellm.success_callback) > 0