From feb91413eccd17754ce5bf9611f172f31be2ae04 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 12 Oct 2023 21:29:35 -0700 Subject: [PATCH] (test) openrouter test use gpt-3.5 instead of gpt-4-32k --- litellm/tests/test_completion.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index fa2850965a..7037503838 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -427,11 +427,13 @@ def test_completion_openrouter1(): print(response) except Exception as e: pytest.fail(f"Error occurred: {e}") +# test_completion_openrouter1() def test_completion_openrouter2(): try: + print("testing openrouter/gpt-3.5-turbo") response = completion( - model="openrouter/openai/gpt-4-32k", + model="openrouter/gpt-3.5-turbo", messages=messages, max_tokens=5, ) @@ -439,6 +441,7 @@ def test_completion_openrouter2(): print(response) except Exception as e: pytest.fail(f"Error occurred: {e}") +# test_completion_openrouter2() def test_completion_openrouter3(): try: @@ -451,8 +454,7 @@ def test_completion_openrouter3(): print(response) except Exception as e: pytest.fail(f"Error occurred: {e}") - -# test_completion_openrouter() +# test_completion_openrouter3() def test_completion_hf_model_no_provider(): try: