Co-authored-by: Antoine <33425718+anth2o@users.noreply.github.com>
Co-authored-by: Bastien <bastien.baret@gmail.com>
Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai>
Co-authored-by: Kim-Adeline Miguel <51720070+kimadeline@users.noreply.github.com>
Co-authored-by: Mathias Gesbert <mathias.gesbert@mistral.ai>
Co-authored-by: Maxime Dolores <maxime.dolores@ext.mistral.ai>
Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com>
Co-authored-by: Nelson PROIA <144663685+Nelson-PROIA@users.noreply.github.com>
Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai>
Co-authored-by: Quentin <quentin.torroba@mistral.ai>
Co-authored-by: Robin Gullo <robin.gullo@mistral.ai>
Co-authored-by: Mistral Vibe <vibe@mistral.ai>
This commit is contained in:
Clément Drouin 2026-04-28 17:44:07 +02:00 committed by GitHub
parent a83c81ecf5
commit 632ea8c032
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
253 changed files with 13965 additions and 2525 deletions

View file

@ -75,11 +75,11 @@ class TestACPSetConfigOptionMode:
assert response is not None
assert response.config_options is not None
assert len(response.config_options) == 2
assert len(response.config_options) == 3
assert (
acp_session.agent_loop.agent_profile.name == BuiltinAgentName.AUTO_APPROVE
)
assert acp_session.agent_loop.auto_approve is True
assert acp_session.agent_loop.bypass_tool_permissions is True
# Verify config_options reflect the new state
mode_config = response.config_options[0]
@ -126,10 +126,10 @@ class TestACPSetConfigOptionMode:
assert response is not None
assert response.config_options is not None
assert len(response.config_options) == 2
assert len(response.config_options) == 3
assert acp_session.agent_loop.agent_profile.name == BuiltinAgentName.CHAT
assert (
acp_session.agent_loop.auto_approve is True
acp_session.agent_loop.bypass_tool_permissions is True
) # Chat mode auto-approves read-only tools
mode_config = response.config_options[0]
@ -200,7 +200,7 @@ class TestACPSetConfigOptionModel:
assert response is not None
assert response.config_options is not None
assert len(response.config_options) == 2
assert len(response.config_options) == 3
assert acp_session.agent_loop.config.active_model == "devstral-small"
# Verify config_options reflect the new state
@ -315,3 +315,87 @@ class TestACPSetConfigOptionInvalidConfigId:
)
assert response is None
class TestACPSetConfigOptionThinking:
@pytest.mark.asyncio
async def test_set_config_option_thinking_success(
self, acp_agent_loop: VibeAcpAgentLoop
) -> None:
session_response = await acp_agent_loop.new_session(
cwd=str(Path.cwd()), mcp_servers=[]
)
session_id = session_response.session_id
acp_session = next(
(s for s in acp_agent_loop.sessions.values() if s.id == session_id), None
)
assert acp_session is not None
assert acp_session.agent_loop.config.get_active_model().thinking == "off"
response = await acp_agent_loop.set_config_option(
session_id=session_id, config_id="thinking", value="high"
)
assert response is not None
assert response.config_options is not None
assert len(response.config_options) == 3
assert acp_session.agent_loop.config.get_active_model().thinking == "high"
thinking_config = response.config_options[2]
assert thinking_config.id == "thinking"
assert thinking_config.current_value == "high"
@pytest.mark.asyncio
async def test_set_config_option_thinking_all_levels(
self, acp_agent_loop: VibeAcpAgentLoop
) -> None:
session_response = await acp_agent_loop.new_session(
cwd=str(Path.cwd()), mcp_servers=[]
)
session_id = session_response.session_id
acp_session = next(
(s for s in acp_agent_loop.sessions.values() if s.id == session_id), None
)
assert acp_session is not None
for level in ["low", "medium", "high", "max", "off"]:
response = await acp_agent_loop.set_config_option(
session_id=session_id, config_id="thinking", value=level
)
assert response is not None
assert acp_session.agent_loop.config.get_active_model().thinking == level
@pytest.mark.asyncio
async def test_set_config_option_thinking_invalid_returns_none(
self, acp_agent_loop: VibeAcpAgentLoop
) -> None:
session_response = await acp_agent_loop.new_session(
cwd=str(Path.cwd()), mcp_servers=[]
)
session_id = session_response.session_id
acp_session = next(
(s for s in acp_agent_loop.sessions.values() if s.id == session_id), None
)
assert acp_session is not None
response = await acp_agent_loop.set_config_option(
session_id=session_id, config_id="thinking", value="ultra"
)
assert response is None
assert acp_session.agent_loop.config.get_active_model().thinking == "off"
@pytest.mark.asyncio
async def test_set_config_option_thinking_empty_string_returns_none(
self, acp_agent_loop: VibeAcpAgentLoop
) -> None:
session_response = await acp_agent_loop.new_session(
cwd=str(Path.cwd()), mcp_servers=[]
)
session_id = session_response.session_id
response = await acp_agent_loop.set_config_option(
session_id=session_id, config_id="thinking", value=""
)
assert response is None