[unit tests] Adding a space in the model output

avigny · avigny · commit 83e86c5b8d96 · 2025-12-03T11:32:31.000+01:00
Signed-off-by: avigny &lt;47987522+avigny@users.noreply.github.com&gt;
diff --git a/tests/tool_use/test_mistral_tool_parser.py b/tests/tool_use/test_mistral_tool_parser.py
@@ -494,7 +494,7 @@ def _test_extract_tool_calls_streaming(
             "",
         ),
         (
-            """[TOOL_CALLS][{"name": "add", "arguments": {"a": 3.5, "b": 4}}, {"name": "get_current_weather", "arguments":{"city": "San Francisco", "state": "CA", "unit": "celsius"}}]""",  # noqa: E501
+            """[TOOL_CALLS] [{"name": "add", "arguments": {"a": 3.5, "b": 4}}, {"name": "get_current_weather", "arguments":{"city": "San Francisco", "state": "CA", "unit": "celsius"}}]""",  # noqa: E501
             [
                 ToolCall(
                     function=FunctionCall(
@@ -790,7 +790,7 @@ def test_extract_tool_calls_streaming_one_chunk(
             "",
         ),
         (
-            """[TOOL_CALLS][{"arguments": {"a": 3.5, "b": 4}, "name": "add"}, {"arguments":{"city": "San Francisco", "state": "CA", "unit": "celsius"}, "name": "get_current_weather"}]""",  # noqa: E501
+            """[TOOL_CALLS] [{"arguments": {"a": 3.5, "b": 4}, "name": "add"}, {"arguments":{"city": "San Francisco", "state": "CA", "unit": "celsius"}, "name": "get_current_weather"}]""",  # noqa: E501
             [
                 ToolCall(
                     function=FunctionCall(

Original file line number	Diff line number	Diff line change
`@@ -494,7 +494,7 @@ def _test_extract_tool_calls_streaming(`
`494`	`494`	`"",`
`495`	`495`	`),`
`496`	`496`	`(`
`497`		`- """[TOOL_CALLS][{"name": "add", "arguments": {"a": 3.5, "b": 4}}, {"name": "get_current_weather", "arguments":{"city": "San Francisco", "state": "CA", "unit": "celsius"}}]""", # noqa: E501`
	`497`	`+ """[TOOL_CALLS] [{"name": "add", "arguments": {"a": 3.5, "b": 4}}, {"name": "get_current_weather", "arguments":{"city": "San Francisco", "state": "CA", "unit": "celsius"}}]""", # noqa: E501`
`498`	`498`	`[`
`499`	`499`	`ToolCall(`
`500`	`500`	`function=FunctionCall(`
`@@ -790,7 +790,7 @@ def test_extract_tool_calls_streaming_one_chunk(`
`790`	`790`	`"",`
`791`	`791`	`),`
`792`	`792`	`(`
`793`		`- """[TOOL_CALLS][{"arguments": {"a": 3.5, "b": 4}, "name": "add"}, {"arguments":{"city": "San Francisco", "state": "CA", "unit": "celsius"}, "name": "get_current_weather"}]""", # noqa: E501`
	`793`	`+ """[TOOL_CALLS] [{"arguments": {"a": 3.5, "b": 4}, "name": "add"}, {"arguments":{"city": "San Francisco", "state": "CA", "unit": "celsius"}, "name": "get_current_weather"}]""", # noqa: E501`
`794`	`794`	`[`
`795`	`795`	`ToolCall(`
`796`	`796`	`function=FunctionCall(`