test(test_function_calling.py): remove redundant gemini test (causing ratelimit errors)

This commit is contained in:
Krrish Dholakia 2024-08-21 21:48:14 -07:00
parent fd3a6f0870
commit 5a31005b85

View file

@ -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