9c1c32e058
Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai> Co-authored-by: Vincent G <10739306+VinceOPS@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
108 lines
4 KiB
Python
108 lines
4 KiB
Python
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
from typing import Any
|
|
|
|
from textual.app import ComposeResult
|
|
from textual.containers import Horizontal, Vertical
|
|
from textual.reactive import reactive
|
|
from textual.widgets import Static
|
|
|
|
from vibe import __version__
|
|
from vibe.cli.textual_ui.widgets.banner.petit_chat import PetitChat
|
|
from vibe.cli.textual_ui.widgets.no_markup_static import NoMarkupStatic
|
|
from vibe.core.config import VibeConfig
|
|
from vibe.core.skills.manager import SkillManager
|
|
from vibe.core.tools.mcp.registry import MCPRegistry
|
|
|
|
|
|
@dataclass
|
|
class BannerState:
|
|
active_model: str = ""
|
|
models_count: int = 0
|
|
mcp_servers_count: int = 0
|
|
skills_count: int = 0
|
|
plan_description: str | None = None
|
|
|
|
|
|
class Banner(Static):
|
|
state = reactive(BannerState(), init=False)
|
|
|
|
def __init__(
|
|
self,
|
|
config: VibeConfig,
|
|
skill_manager: SkillManager,
|
|
mcp_registry: MCPRegistry,
|
|
**kwargs: Any,
|
|
) -> None:
|
|
super().__init__(**kwargs)
|
|
self.can_focus = False
|
|
self._initial_state = BannerState(
|
|
active_model=config.active_model,
|
|
models_count=len(config.models),
|
|
mcp_servers_count=mcp_registry.count_loaded(config.mcp_servers),
|
|
skills_count=len(skill_manager.available_skills),
|
|
plan_description=None,
|
|
)
|
|
self._animated = not config.disable_welcome_banner_animation
|
|
|
|
def compose(self) -> ComposeResult:
|
|
with Horizontal(id="banner-container"):
|
|
yield PetitChat(animate=self._animated)
|
|
|
|
with Vertical(id="banner-info"):
|
|
with Horizontal(classes="banner-line"):
|
|
yield NoMarkupStatic("Mistral Vibe", id="banner-brand")
|
|
yield NoMarkupStatic(" ", classes="banner-spacer")
|
|
yield NoMarkupStatic(f"v{__version__} · ", classes="banner-meta")
|
|
yield NoMarkupStatic("", id="banner-model")
|
|
yield NoMarkupStatic("", id="banner-user-plan")
|
|
with Horizontal(classes="banner-line"):
|
|
yield NoMarkupStatic("", id="banner-meta-counts")
|
|
with Horizontal(classes="banner-line"):
|
|
yield NoMarkupStatic("Type ", classes="banner-meta")
|
|
yield NoMarkupStatic("/help", classes="banner-cmd")
|
|
yield NoMarkupStatic(" for more information", classes="banner-meta")
|
|
|
|
def on_mount(self) -> None:
|
|
self.state = self._initial_state
|
|
|
|
def watch_state(self) -> None:
|
|
self.query_one("#banner-model", NoMarkupStatic).update(self.state.active_model)
|
|
self.query_one("#banner-meta-counts", NoMarkupStatic).update(
|
|
self._format_meta_counts()
|
|
)
|
|
self.query_one("#banner-user-plan", NoMarkupStatic).update(self._format_plan())
|
|
|
|
def freeze_animation(self) -> None:
|
|
if self._animated:
|
|
self.query_one(PetitChat).freeze_animation()
|
|
|
|
def set_state(
|
|
self,
|
|
config: VibeConfig,
|
|
skill_manager: SkillManager,
|
|
mcp_registry: MCPRegistry,
|
|
plan_description: str | None = None,
|
|
) -> None:
|
|
self.state = BannerState(
|
|
active_model=config.active_model,
|
|
models_count=len(config.models),
|
|
mcp_servers_count=mcp_registry.count_loaded(config.mcp_servers),
|
|
skills_count=len(skill_manager.available_skills),
|
|
plan_description=plan_description,
|
|
)
|
|
|
|
def _format_meta_counts(self) -> str:
|
|
return (
|
|
f"{self.state.models_count} model{'s' if self.state.models_count != 1 else ''}"
|
|
f" · {self.state.mcp_servers_count} MCP server{'s' if self.state.mcp_servers_count != 1 else ''}"
|
|
f" · {self.state.skills_count} skill{'s' if self.state.skills_count != 1 else ''}"
|
|
)
|
|
|
|
def _format_plan(self) -> str:
|
|
return (
|
|
""
|
|
if self.state.plan_description is None
|
|
else f" · {self.state.plan_description}"
|
|
)
|