phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

handled merge conflicts

Browse source
This commit is contained in:
Krrish Dholakia 2023-07-31 18:33:29 -07:00
parent 272c0ee7f2
commit f753aac5aa
6 changed files with 6 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/tests/test_model_fallback.py
View file

@ -17,20 +17,10 @@ model_fallback_list = ["replicate/llama-2-70b-chat:2c1608e18606fad2812020dc54193
user_message = "Hello, how are you?"
messages = [{ "content": user_message,"role": "user"}]
# for _ in range(10):
for model in model_fallback_list:
try:
response = embedding(model="text-embedding-ada-002", input=[user_message])
response = completion(model=model, messages=messages)
print(response)
<<<<<<< HEAD
if response != None:
break
except Exception as e:
=======
# if response != None:
# break
except:
>>>>>>> bd42ec8 (clean up code files)
print(f"error occurred: {traceback.format_exc()}")
raise e