v2.14.0 (#743)

Co-authored-by: Alexis Tacnet <alexis@mistral.ai>
Co-authored-by: Clément Drouin <clement.drouin@mistral.ai>
Co-authored-by: Guillaume LE GOFF <guillaume.lgf@gmail.com>
Co-authored-by: Lucas Marandat <31749711+lucasmrdt@users.noreply.github.com>
Co-authored-by: Maxime Dolores <maxime.dolores@ext.mistral.ai>
Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai>
Co-authored-by: Quentin <quentin.torroba@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: p.vezia <166131032+le-codeur-rapide@users.noreply.github.com>
Co-authored-by: Hiba Chaabnia <Hiba-Chaabnia@users.noreply.github.com>
Co-authored-by: Nikhil Bhima <nikhilbhima@users.noreply.github.com>
Co-authored-by: Nkipohcs <Nkipohcs@users.noreply.github.com>
Co-authored-by: Mistral Vibe <vibe@mistral.ai>

tests/snapshots/test_ui_snapshot_image_attachments.py

@@ -0,0 +1,116 @@
+from __future__ import annotations
+
+from pathlib import Path
+
+import pytest
+from textual.pilot import Pilot
+
+from tests.conftest import build_test_vibe_config
+from tests.mock.utils import mock_llm_chunk
+from tests.snapshots.base_snapshot_test_app import BaseSnapshotTestApp, default_config
+from tests.snapshots.snap_compare import SnapCompare
+from tests.stubs.fake_backend import FakeBackend
+from vibe.cli.textual_ui.widgets.chat_input import paste_path
+from vibe.core.config import ModelConfig, ProviderConfig, VibeConfig
+from vibe.core.types import Backend
+
+PNG_BYTES = b"\x89PNG\r\n\x1a\n" + b"\x00" * 16
+SNAP_ROOT = "/snap"
+
+
+def _vision_config() -> VibeConfig:
+    base = default_config()
+    models = [
+        ModelConfig(
+            name="mistral-vibe-cli-latest",
+            provider="mistral",
+            alias="devstral-latest",
+            supports_images=True,
+        )
+    ]
+    providers = [
+        ProviderConfig(
+            name="mistral",
+            api_base="https://api.mistral.ai/v1",
+            api_key_env_var="MISTRAL_API_KEY",
+            backend=Backend.MISTRAL,
+        )
+    ]
+    return build_test_vibe_config(
+        disable_welcome_banner_animation=True,
+        displayed_workdir=base.displayed_workdir,
+        active_model="devstral-latest",
+        models=models,
+        providers=providers,
+    )
+
+
+@pytest.fixture(autouse=True)
+def _accept_snap_paths_as_images(monkeypatch: pytest.MonkeyPatch) -> None:
+    def fake_is_image_file(candidate: str) -> bool:
+        return candidate.startswith(f"{SNAP_ROOT}/") and candidate.endswith((
+            ".png",
+            ".jpg",
+            ".jpeg",
+            ".gif",
+            ".webp",
+        ))
+
+    monkeypatch.setattr(paste_path, "_is_image_file", fake_is_image_file)
+
+
+class _ImageAttachmentApp(BaseSnapshotTestApp):
+    def __init__(self) -> None:
+        super().__init__(
+            config=_vision_config(),
+            backend=FakeBackend(mock_llm_chunk(content="Got it.")),
+        )
+
+
+def test_snapshot_textbox_rewrites_typed_image_path(snap_compare: SnapCompare) -> None:
+    async def run_before(pilot: Pilot) -> None:
+        await pilot.press(*f"look {SNAP_ROOT}/shot.png")
+        await pilot.pause(0.2)
+
+    assert snap_compare(
+        "test_ui_snapshot_image_attachments.py:_ImageAttachmentApp",
+        terminal_size=(120, 36),
+        run_before=run_before,
+    )
+
+
+def test_snapshot_user_message_with_image_footer_singular(
+    snap_compare: SnapCompare, tmp_working_directory: Path
+) -> None:
+    (tmp_working_directory / "shot.png").write_bytes(PNG_BYTES)
+
+    async def run_before(pilot: Pilot) -> None:
+        # Trailing space dismisses the @-mention autocomplete popup so that
+        # `enter` submits the message instead of accepting a suggestion.
+        await pilot.press(*"look @shot.png ")
+        await pilot.press("enter")
+        await pilot.pause(0.4)
+
+    assert snap_compare(
+        "test_ui_snapshot_image_attachments.py:_ImageAttachmentApp",
+        terminal_size=(120, 36),
+        run_before=run_before,
+    )
+
+
+def test_snapshot_user_message_with_image_footer_plural(
+    snap_compare: SnapCompare, tmp_working_directory: Path
+) -> None:
+    (tmp_working_directory / "a.png").write_bytes(PNG_BYTES)
+    (tmp_working_directory / "b.png").write_bytes(PNG_BYTES)
+
+    async def run_before(pilot: Pilot) -> None:
+        await pilot.press(*"compare @a.png @b.png ")
+        await pilot.press("enter")
+        await pilot.pause(0.4)
+
+    assert snap_compare(
+        "test_ui_snapshot_image_attachments.py:_ImageAttachmentApp",
+        terminal_size=(120, 36),
+        run_before=run_before,
+    )