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

Update litellm/tests/test_model_fallback.py

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

6
litellm/tests/test_model_fallback.py
View file

@ -19,8 +19,8 @@ messages = [{ "content": user_message,"role": "user"}]
for model in model_fallback_list:
try:
response = embedding(model="text-embedding-ada-002", input=[user_message])
response = completion(model=model, messages=messages)
response = embedding(model="updated-model", input=[user_message])
response = completion(model="updated-model", messages=messages)
print(response)
except Exception as e:
print(f"error occurred: {traceback.format_exc()}")
print(f"error occurred: {traceback.format_exc()}")