From 37bcc3252ff1bc02e609fb49bd75b6764b0d9ed7 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Mon, 30 Mar 2026 16:35:05 -0700 Subject: [PATCH] test_completion_gpt4_turbo --- tests/local_testing/test_completion.py | 57 -------------------------- 1 file changed, 57 deletions(-) diff --git a/tests/local_testing/test_completion.py b/tests/local_testing/test_completion.py index 080c80e97f..98759a9843 100644 --- a/tests/local_testing/test_completion.py +++ b/tests/local_testing/test_completion.py @@ -868,63 +868,6 @@ def test_completion_mistral_api_modified_input(): pytest.fail(f"Error occurred: {e}") -# def test_completion_oobabooga(): -# try: -# response = completion( -# model="oobabooga/vicuna-1.3b", messages=messages, api_base="http://127.0.0.1:5000" -# ) -# # Add any assertions here to check the response -# print(response) -# except Exception as e: -# pytest.fail(f"Error occurred: {e}") - -# test_completion_oobabooga() -# aleph alpha -# def test_completion_aleph_alpha(): -# try: -# response = completion( -# model="luminous-base", messages=messages, logger_fn=logger_fn -# ) -# # Add any assertions here to check the response -# print(response) -# except Exception as e: -# pytest.fail(f"Error occurred: {e}") -# test_completion_aleph_alpha() - - -# def test_completion_aleph_alpha_control_models(): -# try: -# response = completion( -# model="luminous-base-control", messages=messages, logger_fn=logger_fn -# ) -# # Add any assertions here to check the response -# print(response) -# except Exception as e: -# pytest.fail(f"Error occurred: {e}") -# test_completion_aleph_alpha_control_models() - -import openai - - -def test_completion_gpt4_turbo(): - litellm.set_verbose = True - try: - response = completion( - model="gpt-4-1106-preview", - messages=messages, - max_completion_tokens=10, - ) - print(response) - except openai.RateLimitError: - print("got a rate liimt error") - pass - except Exception as e: - pytest.fail(f"Error occurred: {e}") - - -# test_completion_gpt4_turbo() - - def test_completion_gpt4_turbo_0125(): try: response = completion(