Initial commit

Co-Authored-By: Quentin Torroba <quentin.torroba@mistral.ai>
Co-Authored-By: Laure Hugo <laure.hugo@mistral.ai>
Co-Authored-By: Benjamin Trom <benjamin.trom@mistral.ai>
Co-Authored-By: Mathias Gesbert <mathias.gesbert@ext.mistral.ai>
Co-Authored-By: Michel Thomazo <michel.thomazo@mistral.ai>
Co-Authored-By: Clément Drouin <clement.drouin@mistral.ai>
Co-Authored-By: Vincent Guilloux <vincent.guilloux@mistral.ai>
Co-Authored-By: Valentin Berard <val@mistral.ai>
Co-Authored-By: Mistral Vibe <vibe@mistral.ai>

tests/snapshots/test_ui_snapshot_basic_conversation.py

@@ -0,0 +1,43 @@
+from __future__ import annotations
+
+from textual.pilot import Pilot
+
+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.core.agent import Agent
+
+
+class SnapshotTestAppWithConversation(BaseSnapshotTestApp):
+    def __init__(self) -> None:
+        config = default_config()
+        fake_backend = FakeBackend(
+            results=[
+                mock_llm_chunk(
+                    content="I'm the Vibe agent and I'm ready to help.",
+                    prompt_tokens=10_000,
+                    completion_tokens=2_500,
+                )
+            ]
+        )
+        super().__init__(config=config)
+        self.agent = Agent(
+            config,
+            auto_approve=self.auto_approve,
+            enable_streaming=self.enable_streaming,
+            backend=fake_backend,
+        )
+
+
+def test_snapshot_shows_basic_conversation(snap_compare: SnapCompare) -> None:
+    async def run_before(pilot: Pilot) -> None:
+        await pilot.press(*"Hello there, who are you?")
+        await pilot.press("enter")
+        await pilot.pause(0.4)
+
+    assert snap_compare(
+        "test_ui_snapshot_basic_conversation.py:SnapshotTestAppWithConversation",
+        terminal_size=(120, 36),
+        run_before=run_before,
+    )