forked from phoenix/litellm-mirror
test(test_streaming.py): fix openai streaming function calling test
This commit is contained in:
parent
01e6c7ed89
commit
ca03c57277
1 changed files with 20 additions and 11 deletions
|
@ -1613,6 +1613,8 @@ def streaming_and_function_calling_format_tests(idx, chunk):
|
|||
def test_openai_streaming_and_function_calling():
|
||||
tools = [
|
||||
{
|
||||
"type": "function",
|
||||
"function": {
|
||||
"name": "get_current_weather",
|
||||
"description": "Get the current weather in a given location",
|
||||
"parameters": {
|
||||
|
@ -1626,6 +1628,7 @@ def test_openai_streaming_and_function_calling():
|
|||
},
|
||||
"required": ["location"],
|
||||
},
|
||||
},
|
||||
}
|
||||
]
|
||||
messages = [{"role": "user", "content": "What is the weather like in Boston?"}]
|
||||
|
@ -1638,7 +1641,13 @@ def test_openai_streaming_and_function_calling():
|
|||
)
|
||||
# Add any assertions here to check the response
|
||||
for idx, chunk in enumerate(response):
|
||||
streaming_and_function_calling_format_tests(idx=idx, chunk=chunk)
|
||||
if idx == 0:
|
||||
assert (
|
||||
chunk.choices[0].delta.tool_calls[0].function.arguments is not None
|
||||
)
|
||||
assert isinstance(
|
||||
chunk.choices[0].delta.tool_calls[0].function.arguments, str
|
||||
)
|
||||
except Exception as e:
|
||||
pytest.fail(f"Error occurred: {e}")
|
||||
raise e
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue