From 5a31005b85d07be4a5920173446085adaaeac791 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 21 Aug 2024 21:48:14 -0700 Subject: [PATCH] test(test_function_calling.py): remove redundant gemini test (causing ratelimit errors) --- litellm/tests/test_function_calling.py | 24 ------------------------ 1 file changed, 24 deletions(-) diff --git a/litellm/tests/test_function_calling.py b/litellm/tests/test_function_calling.py index 0895dfb1a88..28a57c29458 100644 --- a/litellm/tests/test_function_calling.py +++ b/litellm/tests/test_function_calling.py @@ -339,27 +339,3 @@ def test_groq_parallel_function_call(): print("second response\n", second_response) except Exception as e: pytest.fail(f"Error occurred: {e}") - - -@pytest.mark.parametrize("model", ["gemini/gemini-1.5-pro"]) -def test_simple_function_call_function_param(model): - try: - litellm.set_verbose = True - messages = [{"role": "user", "content": "What is the weather like in Boston?"}] - response = completion( - model=model, - messages=messages, - tools=[ - { - "type": "function", - "function": { - "name": "plot", - "description": "Generate plots", - }, - } - ], - tool_choice="auto", - ) - print(f"response: {response}") - except Exception as e: - raise e