fa15fc977b
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>
52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
from rich.style import Style
|
|
from textual.widgets.text_area import TextAreaTheme
|
|
|
|
from tests.stubs.fake_backend import FakeBackend
|
|
from vibe.cli.textual_ui.app import VibeApp
|
|
from vibe.cli.textual_ui.widgets.chat_input import ChatTextArea
|
|
from vibe.core.agent import Agent
|
|
from vibe.core.config import SessionLoggingConfig, VibeConfig
|
|
|
|
|
|
def default_config() -> VibeConfig:
|
|
"""Default configuration for snapshot testing.
|
|
Remove as much interference as possible from the snapshot comparison, in order to get a clean pixel-to-pixel comparison.
|
|
- Injects a fake backend to prevent (or stub) LLM calls.
|
|
- Disables the welcome banner animation.
|
|
- Forces a value for the displayed workdir
|
|
- Hides the chat input cursor (as the blinking animation is not deterministic).
|
|
"""
|
|
return VibeConfig(
|
|
session_logging=SessionLoggingConfig(enabled=False),
|
|
textual_theme="gruvbox",
|
|
disable_welcome_banner_animation=True,
|
|
displayed_workdir="/test/workdir",
|
|
)
|
|
|
|
|
|
class BaseSnapshotTestApp(VibeApp):
|
|
CSS_PATH = "../../vibe/cli/textual_ui/app.tcss"
|
|
|
|
def __init__(self, config: VibeConfig | None = None, **kwargs):
|
|
config = config or default_config()
|
|
|
|
super().__init__(config=config, **kwargs)
|
|
|
|
self.agent = Agent(
|
|
config,
|
|
auto_approve=self.auto_approve,
|
|
enable_streaming=self.enable_streaming,
|
|
backend=FakeBackend(),
|
|
)
|
|
|
|
async def on_mount(self) -> None:
|
|
await super().on_mount()
|
|
self._hide_chat_input_cursor()
|
|
|
|
def _hide_chat_input_cursor(self) -> None:
|
|
text_area = self.query_one(ChatTextArea)
|
|
hidden_cursor_theme = TextAreaTheme(name="hidden_cursor", cursor_style=Style())
|
|
text_area.register_theme(hidden_cursor_theme)
|
|
text_area.theme = "hidden_cursor"
|