702d0f412e
Co-authored-by: Clément Drouin <clement.drouin@mistral.ai> Co-authored-by: Gauthier Guinet <43207538+Gguinet@users.noreply.github.com> Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com> Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai> Co-authored-by: Stanislas <stanislas.lange@mistral.ai> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
121 lines
3.9 KiB
Python
121 lines
3.9 KiB
Python
from __future__ import annotations
|
|
|
|
from collections.abc import Awaitable, Callable
|
|
from dataclasses import dataclass, field
|
|
|
|
type OnCommandsChanged = Callable[[], Awaitable[None]]
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class AcpCommandAvailabilityContext:
|
|
"""Context used to decide whether a command should be advertised."""
|
|
|
|
vibe_code_enabled: bool = False
|
|
|
|
def is_teleport_available(self) -> bool:
|
|
return self.vibe_code_enabled
|
|
|
|
|
|
type CommandAvailability = Callable[[AcpCommandAvailabilityContext], bool]
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class AcpCommand:
|
|
"""Command advertised to ACP clients via available_commands_update."""
|
|
|
|
name: str
|
|
description: str
|
|
handler: str
|
|
input_hint: str | None = None
|
|
is_available: CommandAvailability | None = None
|
|
|
|
|
|
@dataclass
|
|
class AcpCommandRegistry:
|
|
"""Registry of ACP commands. Notifies listeners when commands change."""
|
|
|
|
availability_context: AcpCommandAvailabilityContext = field(
|
|
default_factory=AcpCommandAvailabilityContext
|
|
)
|
|
_commands: dict[str, AcpCommand] = field(default_factory=dict)
|
|
_on_changed: OnCommandsChanged | None = None
|
|
|
|
def __post_init__(self) -> None:
|
|
if not self._commands:
|
|
self._commands = {
|
|
name: command
|
|
for name, command in _build_commands().items()
|
|
if self._is_available(command)
|
|
}
|
|
|
|
def _is_available(self, command: AcpCommand) -> bool:
|
|
if command.is_available is None:
|
|
return True
|
|
return command.is_available(self.availability_context)
|
|
|
|
def set_on_changed(self, callback: OnCommandsChanged) -> None:
|
|
self._on_changed = callback
|
|
|
|
@property
|
|
def commands(self) -> dict[str, AcpCommand]:
|
|
return self._commands
|
|
|
|
def get(self, name: str) -> AcpCommand | None:
|
|
return self._commands.get(name)
|
|
|
|
async def notify_changed(self) -> None:
|
|
if self._on_changed is not None:
|
|
await self._on_changed()
|
|
|
|
|
|
def _build_commands() -> dict[str, AcpCommand]:
|
|
return {
|
|
"help": AcpCommand(
|
|
name="help",
|
|
description="Show available commands and keyboard shortcuts",
|
|
handler="_handle_help",
|
|
),
|
|
"compact": AcpCommand(
|
|
name="compact",
|
|
description="Compact conversation history by summarizing. Optionally pass instructions to guide the summary",
|
|
handler="_handle_compact",
|
|
input_hint="Optional instructions to guide the compaction summary",
|
|
),
|
|
"reload": AcpCommand(
|
|
name="reload",
|
|
description="Reload configuration, agent instructions, and skills from disk",
|
|
handler="_handle_reload",
|
|
),
|
|
"log": AcpCommand(
|
|
name="log",
|
|
description="Show path to current session log directory",
|
|
handler="_handle_log",
|
|
),
|
|
"teleport": AcpCommand(
|
|
name="teleport",
|
|
description="Teleport session to Vibe Code Web",
|
|
handler="_handle_teleport",
|
|
is_available=AcpCommandAvailabilityContext.is_teleport_available,
|
|
),
|
|
"proxy-setup": AcpCommand(
|
|
name="proxy-setup",
|
|
description="Configure proxy and SSL certificate settings",
|
|
handler="_handle_proxy_setup",
|
|
input_hint="KEY value to set, KEY to unset, or empty for help",
|
|
),
|
|
"leanstall": AcpCommand(
|
|
name="leanstall",
|
|
description="Install the Lean 4 agent (leanstral)",
|
|
handler="_handle_leanstall",
|
|
),
|
|
"unleanstall": AcpCommand(
|
|
name="unleanstall",
|
|
description="Uninstall the Lean 4 agent",
|
|
handler="_handle_unleanstall",
|
|
),
|
|
"data-retention": AcpCommand(
|
|
name="data-retention",
|
|
description="Show data retention information",
|
|
handler="_handle_data_retention",
|
|
),
|
|
}
|