mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-06-09 17:12:01 +00:00
3b6dd0a4e3
* feat(subagents): extend deferred MCP tool loading to subagents (#3341) Subagents now reuse the lead agent's deferred-tool path: when tool_search.enabled, MCP tool schemas are withheld from the model and surfaced by name in <available-deferred-tools>, fetched on demand via the generated tool_search helper. DeferredToolFilterMiddleware deterministically rewrites request.tools to hide the deferred schemas (the prompt section is discovery only, not enforcement). Consolidates the assembly into deerflow.tools.builtins.tool_search, now the single home for both assemble_deferred_tools (centralized fail-closed guard, replacing the lead-only private _assemble_deferred) and the relocated get_deferred_tools_prompt_section. Shared by every build path: lead agent, embedded client, and subagent executor. tool_search is appended after the subagent's name-level tool policy and is treated as infrastructure: its catalog is built from the already policy-filtered list, so it can never surface a tool the policy denied. Follow-up to #3370. Fixes #3341. * test(subagents): assert the real middleware builder emits a working deferred filter (#3341) The existing recipe test hand-constructs DeferredToolFilterMiddleware, so it cannot catch a regression in how build_subagent_runtime_middlewares (the call executor._create_agent actually makes) wires the deferred setup into the filter. Add a test that sources the filter from the real builder given a real setup and runs it through a graph: a wrong catalog hash would silently stop promotion, a dropped filter would stop hiding — both now caught. Running the full real middleware stack is intentionally avoided (the other runtime middlewares need sandbox/thread infra to execute, which would make the test flaky); their attachment + ordering before Safety stays locked in test_tool_error_handling_middleware.py. * test(subagents): keep executor tests config-free in CI * chore: trigger ci * Potential fix for pull request finding Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com> --------- Co-authored-by: Willem Jiang <willem.jiang@gmail.com> Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>
39 lines
1.4 KiB
Python
39 lines
1.4 KiB
Python
"""Tests for the tool_search (deferred tool loading) config + prompt section.
|
|
|
|
Catalog search, setup assembly, the Command-writing tool_search tool, and the
|
|
filter middleware are covered by:
|
|
- tests/test_deferred_catalog.py
|
|
- tests/test_deferred_setup.py
|
|
- tests/test_deferred_filter_middleware.py
|
|
- tests/test_thread_state_promoted.py
|
|
"""
|
|
|
|
from deerflow.config.tool_search_config import ToolSearchConfig, load_tool_search_config_from_dict
|
|
from deerflow.tools.builtins.tool_search import get_deferred_tools_prompt_section
|
|
|
|
|
|
class TestToolSearchConfig:
|
|
def test_default_disabled(self):
|
|
assert ToolSearchConfig().enabled is False
|
|
|
|
def test_enabled(self):
|
|
assert ToolSearchConfig(enabled=True).enabled is True
|
|
|
|
def test_load_from_dict(self):
|
|
assert load_tool_search_config_from_dict({"enabled": True}).enabled is True
|
|
|
|
def test_load_from_empty_dict(self):
|
|
assert load_tool_search_config_from_dict({}).enabled is False
|
|
|
|
|
|
class TestDeferredToolsPromptSection:
|
|
def test_empty_without_names(self):
|
|
assert get_deferred_tools_prompt_section() == ""
|
|
|
|
def test_empty_with_empty_frozenset(self):
|
|
assert get_deferred_tools_prompt_section(deferred_names=frozenset()) == ""
|
|
|
|
def test_lists_sorted_names(self):
|
|
out = get_deferred_tools_prompt_section(deferred_names=frozenset({"b_tool", "a_tool"}))
|
|
assert out == "<available-deferred-tools>\na_tool\nb_tool\n</available-deferred-tools>"
|