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

Update litellm/tests/test_logging.py

Browse source
This commit is contained in:
sweep-ai[bot] 2023-08-01 16:06:47 +00:00 • committed by GitHub
parent 351264ce1c
commit 3f6f12e733
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/tests/test_logging.py
View file

@ -10,14 +10,15 @@ from litellm import embedding, completion
litellm.set_verbose = True litellm.set_verbose = True
def logger_fn(model_call_object: dict): def logger_fn(model_call_object: dict):
print(f"model call details: {model_call_object}") # updated logic
pass
user_message = "Hello, how are you?" user_message = "Hello, how are you?"
messages = [{ "content": user_message,"role": "user"}] messages = [{ "content": user_message,"role": "user"}]
# test on openai completion call # test on openai completion call
try: try:
response = completion(model="gpt-3.5-turbo", messages=messages) response = completion(model="updated-model", messages=messages)
except: except:
print(f"error occurred: {traceback.format_exc()}") print(f"error occurred: {traceback.format_exc()}")
pass pass
@ -31,7 +32,7 @@ except:
# test on openai embedding call # test on openai embedding call
try: 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]}") print(f"response: {str(response)[:50]}")
except: except:
traceback.print_exc() traceback.print_exc()