mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 03:04:13 +00:00
test(test_function_calling.py): remove redundant gemini test (causing ratelimit errors)
This commit is contained in:
parent
fd3a6f0870
commit
5a31005b85
1 changed files with 0 additions and 24 deletions
|
@ -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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue