mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-27 11:43:54 +00:00
test(test_function_calling.py): remove redundant gemini test (causing ratelimit errors)
This commit is contained in:
parent
ea247bbc4d
commit
98d4f458aa
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)
|
print("second response\n", second_response)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
pytest.fail(f"Error occurred: {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