3f8487f761
Co-authored-by: Alexis Tacnet <alexis@mistral.ai> Co-authored-by: Clément Drouin <clement.drouin@mistral.ai> Co-authored-by: Guillaume LE GOFF <guillaume.lgf@gmail.com> Co-authored-by: Lucas Marandat <31749711+lucasmrdt@users.noreply.github.com> Co-authored-by: Maxime Dolores <maxime.dolores@ext.mistral.ai> Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai> Co-authored-by: Quentin <quentin.torroba@mistral.ai> Co-authored-by: Val <102326092+vdeva@users.noreply.github.com> Co-authored-by: Vincent G <10739306+VinceOPS@users.noreply.github.com> Co-authored-by: p.vezia <166131032+le-codeur-rapide@users.noreply.github.com> Co-authored-by: Hiba Chaabnia <Hiba-Chaabnia@users.noreply.github.com> Co-authored-by: Nikhil Bhima <nikhilbhima@users.noreply.github.com> Co-authored-by: Nkipohcs <Nkipohcs@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
35 lines
1 KiB
Python
35 lines
1 KiB
Python
from __future__ import annotations
|
|
|
|
from tests.conftest import build_test_vibe_config
|
|
from vibe.acp.acp_agent_loop import (
|
|
NON_INTERACTIVE_DISABLED_TOOLS,
|
|
_merge_non_interactive_disabled_tools,
|
|
)
|
|
|
|
|
|
def test_merge_preserves_toml_disabled_tools():
|
|
config = build_test_vibe_config(disabled_tools=["task"])
|
|
|
|
_merge_non_interactive_disabled_tools(config)
|
|
|
|
assert "task" in config.disabled_tools
|
|
for tool in NON_INTERACTIVE_DISABLED_TOOLS:
|
|
assert tool in config.disabled_tools
|
|
|
|
|
|
def test_merge_deduplicates():
|
|
config = build_test_vibe_config(disabled_tools=["ask_user_question", "task"])
|
|
|
|
_merge_non_interactive_disabled_tools(config)
|
|
|
|
assert config.disabled_tools.count("ask_user_question") == 1
|
|
assert "task" in config.disabled_tools
|
|
assert "exit_plan_mode" in config.disabled_tools
|
|
|
|
|
|
def test_merge_on_empty_disabled_tools():
|
|
config = build_test_vibe_config()
|
|
|
|
_merge_non_interactive_disabled_tools(config)
|
|
|
|
assert set(config.disabled_tools) == set(NON_INTERACTIVE_DISABLED_TOOLS)
|