CalvinBackup/invariant-gateway
1
0
Fork 0
mirror of https://github.com/invariantlabs-ai/invariant-gateway.git synced 2026-05-25 00:04:01 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix test_chat_completion_with_tool_call_without_streaming.

Browse Source
This commit is contained in:
Hemang
2025-03-12 21:56:19 +01:00
parent 032d60e23e
commit c9535e0a72
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/open_ai/test_chat_with_tool_call.py
+5 -5
View File
@@ -62,12 +62,12 @@ async def test_chat_completion_with_tool_call_without_streaming(
assert chat_response.choices[0].message.role == "assistant"
# Extract tool call
assert len(chat_response.choices[0].message.tool_calls) == 1
assert chat_response.choices[0].message.tool_calls[0].function.name == "get_weather"
assert (
chat_response.choices[0].message.tool_calls[0].function.arguments
== '{"location":"New York"}'
)
tool_call = chat_response.choices[0].message.tool_calls[0]
assert tool_call.function.name == "get_weather"
assert (
"New York" in tool_call.function.arguments
and "location" in tool_call.function.arguments
)
# Mock response of tool call
tool_result = "The temperature in New York is 15°C and it is raining."