diff --git a/tests/hermes_cli/test_tools_config.py b/tests/hermes_cli/test_tools_config.py index 4ac32d007..d7a2f7c7c 100644 --- a/tests/hermes_cli/test_tools_config.py +++ b/tests/hermes_cli/test_tools_config.py @@ -657,3 +657,16 @@ def test_discord_admin_in_configurable_toolsets(): def test_discord_admin_in_default_off(): assert "discord_admin" in _DEFAULT_OFF_TOOLSETS + + +def test_get_platform_tools_feishu_includes_doc_and_drive(): + enabled = _get_platform_tools({}, "feishu") + assert "feishu_doc" in enabled + assert "feishu_drive" in enabled + + +def test_get_platform_tools_feishu_tools_not_on_other_platforms(): + for plat in ["cli", "telegram", "discord"]: + enabled = _get_platform_tools({}, plat) + assert "feishu_doc" not in enabled, f"feishu_doc leaked onto {plat}" + assert "feishu_drive" not in enabled, f"feishu_drive leaked onto {plat}" diff --git a/toolsets.py b/toolsets.py index a9e119358..4543b06c0 100644 --- a/toolsets.py +++ b/toolsets.py @@ -391,7 +391,13 @@ TOOLSETS = { "hermes-feishu": { "description": "Feishu/Lark bot toolset - enterprise messaging via Feishu/Lark (full access)", - "tools": _HERMES_CORE_TOOLS, + "tools": _HERMES_CORE_TOOLS + [ + "feishu_doc_read", + "feishu_drive_list_comments", + "feishu_drive_list_comment_replies", + "feishu_drive_reply_comment", + "feishu_drive_add_comment", + ], "includes": [] },