phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 19:54:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test(test_function_calling.py): handle for when model returns a text response

Browse source
This commit is contained in:
Krrish Dholakia 2024-04-17 18:32:34 -07:00
parent caa46ca905
commit 10d65f0676
1 changed files with 41 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_function_calling.py
View file

@ -266,9 +266,12 @@ def test_groq_parallel_function_call():
) )
print("Response\n", response) print("Response\n", response)
response_message = response.choices[0].message response_message = response.choices[0].message
if hasattr(response_message, "tool_calls"):
tool_calls = response_message.tool_calls tool_calls = response_message.tool_calls
assert isinstance(response.choices[0].message.tool_calls[0].function.name, str) assert isinstance(
response.choices[0].message.tool_calls[0].function.name, str
)
assert isinstance( assert isinstance(
response.choices[0].message.tool_calls[0].function.arguments, str response.choices[0].message.tool_calls[0].function.arguments, str
) )