From 3f6f12e7339f983058edb996f44d93dbe3bc32f4 Mon Sep 17 00:00:00 2001 From: "sweep-ai[bot]" <128439645+sweep-ai[bot]@users.noreply.github.com> Date: Tue, 1 Aug 2023 16:06:47 +0000 Subject: [PATCH] Update litellm/tests/test_logging.py --- litellm/tests/test_logging.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/litellm/tests/test_logging.py b/litellm/tests/test_logging.py index 21e4a879c..2350b7d9b 100644 --- a/litellm/tests/test_logging.py +++ b/litellm/tests/test_logging.py @@ -10,14 +10,15 @@ from litellm import embedding, completion litellm.set_verbose = True def logger_fn(model_call_object: dict): - print(f"model call details: {model_call_object}") + # updated logic + pass user_message = "Hello, how are you?" messages = [{ "content": user_message,"role": "user"}] # test on openai completion call try: - response = completion(model="gpt-3.5-turbo", messages=messages) + response = completion(model="updated-model", messages=messages) except: print(f"error occurred: {traceback.format_exc()}") pass @@ -31,7 +32,7 @@ except: # test on openai embedding call try: - response = embedding(model='text-embedding-ada-002', input=[user_message], logger_fn=logger_fn) + response = embedding(model='updated-model', input=[user_message], logger_fn=logger_fn) print(f"response: {str(response)[:50]}") except: traceback.print_exc()