228f3c65a9
Co-authored-by: Clément Drouin <clement.drouin@mistral.ai> Co-authored-by: Corentin André <corentin.andre@mistral.ai> Co-authored-by: Guillaume LE GOFF <guillaume.lgf@gmail.com> Co-authored-by: Kim-Adeline Miguel <51720070+kimadeline@users.noreply.github.com> Co-authored-by: Maxime Dolores <maxime.dolores@ext.mistral.ai> Co-authored-by: Nelson PROIA <144663685+Nelson-PROIA@users.noreply.github.com> Co-authored-by: Peter Evers <pevers90@gmail.com> Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai> Co-authored-by: Quentin <quentin.torroba@mistral.ai> Co-authored-by: Vincent G <10739306+VinceOPS@users.noreply.github.com> Co-authored-by: MichisGitIsKing <MichisGitIsKing@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
79 lines
3 KiB
Python
79 lines
3 KiB
Python
from __future__ import annotations
|
|
|
|
from acp import PROTOCOL_VERSION
|
|
from acp.schema import (
|
|
AgentCapabilities,
|
|
ClientCapabilities,
|
|
Implementation,
|
|
PromptCapabilities,
|
|
SessionCapabilities,
|
|
SessionCloseCapabilities,
|
|
SessionForkCapabilities,
|
|
SessionListCapabilities,
|
|
)
|
|
import pytest
|
|
|
|
from vibe.acp.acp_agent_loop import VibeAcpAgentLoop
|
|
|
|
|
|
class TestACPInitialize:
|
|
@pytest.mark.asyncio
|
|
async def test_initialize(self, acp_agent_loop: VibeAcpAgentLoop) -> None:
|
|
response = await acp_agent_loop.initialize(protocol_version=PROTOCOL_VERSION)
|
|
|
|
assert response.protocol_version == PROTOCOL_VERSION
|
|
assert response.agent_capabilities == AgentCapabilities(
|
|
load_session=True,
|
|
prompt_capabilities=PromptCapabilities(
|
|
audio=False, embedded_context=True, image=False
|
|
),
|
|
session_capabilities=SessionCapabilities(
|
|
close=SessionCloseCapabilities(),
|
|
list=SessionListCapabilities(),
|
|
fork=SessionForkCapabilities(),
|
|
),
|
|
)
|
|
assert response.agent_info == Implementation(
|
|
name="@mistralai/mistral-vibe", title="Mistral Vibe", version="2.10.0"
|
|
)
|
|
|
|
assert response.auth_methods == []
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_initialize_with_terminal_auth(
|
|
self, acp_agent_loop: VibeAcpAgentLoop
|
|
) -> None:
|
|
"""Test initialize with terminal-auth capabilities to check it was included."""
|
|
client_capabilities = ClientCapabilities(field_meta={"terminal-auth": True})
|
|
response = await acp_agent_loop.initialize(
|
|
protocol_version=PROTOCOL_VERSION, client_capabilities=client_capabilities
|
|
)
|
|
|
|
assert response.protocol_version == PROTOCOL_VERSION
|
|
assert response.agent_capabilities == AgentCapabilities(
|
|
load_session=True,
|
|
prompt_capabilities=PromptCapabilities(
|
|
audio=False, embedded_context=True, image=False
|
|
),
|
|
session_capabilities=SessionCapabilities(
|
|
close=SessionCloseCapabilities(),
|
|
list=SessionListCapabilities(),
|
|
fork=SessionForkCapabilities(),
|
|
),
|
|
)
|
|
assert response.agent_info == Implementation(
|
|
name="@mistralai/mistral-vibe", title="Mistral Vibe", version="2.10.0"
|
|
)
|
|
|
|
assert response.auth_methods is not None
|
|
assert len(response.auth_methods) == 1
|
|
auth_method = response.auth_methods[0]
|
|
assert auth_method.id == "vibe-setup"
|
|
assert auth_method.name == "Register your API Key"
|
|
assert auth_method.description == "Register your API Key inside Mistral Vibe"
|
|
assert auth_method.field_meta is not None
|
|
assert "terminal-auth" in auth_method.field_meta
|
|
terminal_auth_meta = auth_method.field_meta["terminal-auth"]
|
|
assert "command" in terminal_auth_meta
|
|
assert "args" in terminal_auth_meta
|
|
assert terminal_auth_meta["label"] == "Mistral Vibe Setup"
|