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>

tests/core/test_image_attachment.py

@@ -0,0 +1,58 @@
+from __future__ import annotations
+
+from pathlib import Path
+
+from pydantic import ValidationError
+import pytest
+
+from vibe.core.types import FileImageSource, ImageAttachment, InlineImageSource
+
+
+def test_migrates_legacy_flat_path_shape() -> None:
+    att = ImageAttachment.model_validate({
+        "path": "/tmp/a.png",
+        "alias": "a.png",
+        "mime_type": "image/png",
+    })
+
+    assert isinstance(att.source, FileImageSource)
+    assert att.source.path == Path("/tmp/a.png")
+
+
+def test_migrates_legacy_flat_data_shape() -> None:
+    att = ImageAttachment.model_validate({
+        "data": "Zm9v",
+        "alias": "pasted.png",
+        "mime_type": "image/png",
+    })
+
+    assert isinstance(att.source, InlineImageSource)
+    assert att.source.data == "Zm9v"
+
+
+def test_source_path_construction() -> None:
+    att = ImageAttachment(
+        source=FileImageSource(path=Path("/tmp/a.png")),
+        alias="a.png",
+        mime_type="image/png",
+    )
+
+    assert isinstance(att.source, FileImageSource)
+    assert att.source.path == Path("/tmp/a.png")
+
+
+def test_file_source_round_trips_through_json() -> None:
+    att = ImageAttachment(
+        source=FileImageSource(path=Path("/tmp/a.png")),
+        alias="a.png",
+        mime_type="image/png",
+    )
+
+    dumped = att.model_dump(exclude_none=True, mode="json")
+    assert dumped["source"] == {"kind": "file", "path": "/tmp/a.png"}
+    assert ImageAttachment.model_validate(dumped) == att
+
+
+def test_rejects_attachment_without_source() -> None:
+    with pytest.raises(ValidationError):
+        ImageAttachment.model_validate({"alias": "a.png", "mime_type": "image/png"})