From 66b1c1bb42bbb554c3c5a2da59964e4c8799254c 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:11:20 +0000 Subject: [PATCH] Update litellm/tests/test_model_fallback.py --- litellm/tests/test_model_fallback.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/litellm/tests/test_model_fallback.py b/litellm/tests/test_model_fallback.py index b389e9f6a..a0636cfeb 100644 --- a/litellm/tests/test_model_fallback.py +++ b/litellm/tests/test_model_fallback.py @@ -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()}")