diff --git a/litellm/__pycache__/main.cpython-311.pyc b/litellm/__pycache__/main.cpython-311.pyc index b753e814fd..c82c00eec6 100644 Binary files a/litellm/__pycache__/main.cpython-311.pyc and b/litellm/__pycache__/main.cpython-311.pyc differ diff --git a/litellm/__pycache__/utils.cpython-311.pyc b/litellm/__pycache__/utils.cpython-311.pyc index 5b4f9b84e1..46a47c0c6d 100644 Binary files a/litellm/__pycache__/utils.cpython-311.pyc and b/litellm/__pycache__/utils.cpython-311.pyc differ diff --git a/litellm/tests/test_litedebugger_integration.py b/litellm/tests/test_litedebugger_integration.py index 05e93d49df..e6235aeb43 100644 --- a/litellm/tests/test_litedebugger_integration.py +++ b/litellm/tests/test_litedebugger_integration.py @@ -30,7 +30,7 @@ sys.path.insert( 0, os.path.abspath("../..") ) # Adds the parent directory to the system path import litellm -from litellm import completion_with_split_tests +from litellm import completion from openai.error import AuthenticationError litellm.set_verbose = True @@ -69,8 +69,8 @@ messages = [{"content": user_message, "role": "user"}] # # Redirect stdout # old_stdout = sys.stdout # sys.stdout = new_stdout = io.StringIO() - -# response = completion_with_split_tests(models=split_per_model, messages=messages, use_client=True, id="6d383c99-488d-481d-aa1b-1f94935cec44") +# litellm.token = "a67abbaf-35b8-4649-8647-68c5fe8d37fb" # generate one here - https://www.uuidgenerator.net/version4 +# response = completion(model="gpt-3.5-turbo", messages=messages, use_client=True) # # Restore stdout # sys.stdout = old_stdout @@ -83,6 +83,7 @@ messages = [{"content": user_message, "role": "user"}] # if "LiteDebugger: Success/Failure Call Logging" not in output: # raise Exception("LiteLLMDebugger: success/failure call not logged!") # except Exception as e: +# print(output) # pytest.fail(f"Error occurred: {e}") # # Test 3: On streaming completion call - setting client to true diff --git a/litellm/utils.py b/litellm/utils.py index 004ab2eb22..d7500e2e28 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -1051,6 +1051,10 @@ def set_callbacks(callback_list, function_id=None): print_verbose(f"instantiating lite_debugger") if function_id: liteDebuggerClient = LiteDebugger(email=function_id) + elif litellm.token: + liteDebuggerClient = LiteDebugger(email=litellm.token) + elif litellm.email: + liteDebuggerClient = LiteDebugger(email=litellm.email) except Exception as e: raise e