3f8487f761
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>
116 lines
3.6 KiB
Python
116 lines
3.6 KiB
Python
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,
|
|
)
|