v2.17.0 (#822)

Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai> Co-authored-by: Guillaume LE GOFF <guillaume.lgf@gmail.com> Co-authored-by: Hdandria <henri.dandria@mistral.ai> Co-authored-by: Ivana Dunisijevic <ivana.dunisijevic@mistral.ai> Co-authored-by: Jean Burellier <sheplu@users.noreply.github.com> Co-authored-by: Mathias Gesbert <mathias.gesbert@mistral.ai> Co-authored-by: Mert Unsal <mert.unsal@mistral.ai> Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com> Co-authored-by: Paul VEZIA <166131032+le-codeur-rapide@users.noreply.github.com> Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai> Co-authored-by: Val <102326092+vdeva@users.noreply.github.com> Co-authored-by: Vincent G <10739306+VinceOPS@users.noreply.github.com> Co-authored-by: renovate-mistral[bot] <253709520+renovate-mistral[bot]@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
2026-06-19 11:01:24 +02:00 · 2026-06-19 11:01:24 +02:00 · 6bedf271ce
commit 6bedf271ce
parent 564a14365e
223 changed files with 10533 additions and 6947 deletions
--- a/tests/e2e/test_mock_server.py
+++ b/tests/e2e/test_mock_server.py
@ -0,0 +1,73 @@
+from __future__ import annotations
+
+import httpx
+
+from tests.e2e.mock_server import ChatCompletionsRequestPayload, StreamingMockServer
+
+
+def _split_tool_call_factory(
+    _request_index: int, _payload: ChatCompletionsRequestPayload
+) -> list[dict[str, object]]:
+    return [
+        StreamingMockServer.build_chunk(
+            created=10,
+            delta={
+                "role": "assistant",
+                "tool_calls": [
+                    {
+                        "index": 0,
+                        "id": "call_split",
+                        "type": "function",
+                        "function": {"name": "bash", "arguments": '{"command":"echo'},
+                    }
+                ],
+            },
+            finish_reason=None,
+        ),
+        StreamingMockServer.build_chunk(
+            created=11,
+            delta={"tool_calls": [{"function": {"arguments": ' ok"}'}}]},
+            finish_reason=None,
+        ),
+        StreamingMockServer.build_chunk(
+            created=12,
+            delta={},
+            finish_reason="tool_calls",
+            usage={"prompt_tokens": 7, "completion_tokens": 8},
+        ),
+    ]
+
+
+def test_non_streaming_completion_merges_split_tool_call_deltas() -> None:
+    server = StreamingMockServer(chunk_factory=_split_tool_call_factory)
+    server.start()
+    try:
+        response = httpx.post(
+            f"{server.api_base}/chat/completions",
+            json={"model": "mock-model", "messages": [], "stream": False},
+            timeout=5,
+        )
+    finally:
+        server.stop()
+
+    response.raise_for_status()
+    response_json = response.json()
+    choices = response_json["choices"]
+    assert isinstance(choices, list)
+    choice = choices[0]
+    assert isinstance(choice, dict)
+    message = choice["message"]
+    assert isinstance(message, dict)
+    tool_calls = message["tool_calls"]
+    assert isinstance(tool_calls, list)
+    assert tool_calls == [
+        {
+            "index": 0,
+            "id": "call_split",
+            "type": "function",
+            "function": {"name": "bash", "arguments": '{"command":"echo ok"}'},
+        }
+    ]
+    assert choice["finish_reason"] == "tool_calls"
+    assert response_json["usage"] == {"prompt_tokens": 7, "completion_tokens": 8}
+    assert server.requests == [{"model": "mock-model", "messages": [], "stream": False}]