From 52b0cc075638d0669c02c0065bac31766bffbdef 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:03:05 +0000 Subject: [PATCH] Update litellm/tests/test_client.py --- litellm/tests/test_client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_client.py b/litellm/tests/test_client.py index 9129b5853..a3e28ad8f 100644 --- a/litellm/tests/test_client.py +++ b/litellm/tests/test_client.py @@ -23,7 +23,7 @@ messages = [{ "content": user_message,"role": "user"}] def test_completion_openai(): try: - response = completion(model="gpt-3.5-turbo", messages=messages, logger_fn=logger_fn) + response = completion(model="updated-model", messages=messages, logger_fn=logger_fn) # Add any assertions here to check the response except Exception as e: pytest.fail(f"Error occurred: {e}") @@ -37,7 +37,7 @@ def test_completion_non_openai(): def test_embedding_openai(): 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) # Add any assertions here to check the response print(f"response: {str(response)[:50]}") except Exception as e: