v2.14.1 (#756)

Co-authored-by: Clément Drouin <clement.drouin@mistral.ai>
Co-authored-by: Gauthier Guinet <43207538+Gguinet@users.noreply.github.com>
Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com>
Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai>
Co-authored-by: Stanislas <stanislas.lange@mistral.ai>
Co-authored-by: Mistral Vibe <vibe@mistral.ai>

tests/core/test_build_attachment_counts.py

@@ -0,0 +1,34 @@
+from __future__ import annotations
+
+from pathlib import Path
+
+from vibe.core.telemetry.build_metadata import build_attachment_counts
+from vibe.core.telemetry.types import AttachmentKind
+from vibe.core.types import ImageAttachment, LLMMessage, Role
+
+
+def _msg(images: list[ImageAttachment] | None) -> LLMMessage:
+    return LLMMessage(role=Role.user, content="hi", images=images)
+
+
+def _image() -> ImageAttachment:
+    return ImageAttachment(
+        path=Path("/tmp/x.png"), alias="x.png", mime_type="image/png"
+    )
+
+
+def test_returns_empty_when_message_is_none() -> None:
+    assert build_attachment_counts(None, supports_images=True) == {}
+
+
+def test_returns_empty_when_no_images() -> None:
+    assert build_attachment_counts(_msg(None), supports_images=True) == {}
+
+
+def test_counts_images_when_model_supports_them() -> None:
+    counts = build_attachment_counts(_msg([_image(), _image()]), supports_images=True)
+    assert counts == {AttachmentKind.IMAGE: 2}
+
+
+def test_drops_images_when_model_does_not_support_them() -> None:
+    assert build_attachment_counts(_msg([_image()]), supports_images=False) == {}