mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-10 23:30:53 +08:00
oliver
ba3836f00f
- 在 oclaw/ 下重新初始化 Git 仓库 - 补齐子仓库 .gitignore，避免提交本地运行态数据（_local、node_modules、logs 等） - 提交当前工程代码与配置 Made-with: Cursor
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
from oclaw.platform.persistence.sqlite_store import SqliteStore
|
|
from oclaw.tools.catalog import materialize_tool_specs
|
|
|
|
|
|
def _seed_single_mcp_tool(store: SqliteStore) -> None:
|
|
store.upsert_mcp_server(
|
|
server_id="echo",
|
|
source_type="github",
|
|
source_ref="https://github.com/acme/echo",
|
|
entry_command="python",
|
|
entry_args=["-m", "echo_server"],
|
|
enabled=True,
|
|
)
|
|
store.replace_mcp_server_tools(
|
|
server_id="echo",
|
|
tools=[{"tool_name": "ping", "description": "Ping", "parameters": {"type": "object", "properties": {}}}],
|
|
)
|
|
|
|
|
|
def test_catalog_materializes_mcp_tools_when_enabled(tmp_path: Path) -> None:
|
|
store = SqliteStore(str(tmp_path / "ops.sqlite"))
|
|
_seed_single_mcp_tool(store)
|
|
store.set_setting("AIA_ENABLE_MCP_TOOLS", "1")
|
|
specs = materialize_tool_specs(store=store, specialist="generalist")
|
|
names = {x.name for x in specs}
|
|
assert "system_time" in names
|
|
assert "mcp__echo__ping" in names
|
|
|
|
|
|
def test_catalog_skips_mcp_tools_when_disabled(tmp_path: Path) -> None:
|
|
store = SqliteStore(str(tmp_path / "ops.sqlite"))
|
|
_seed_single_mcp_tool(store)
|
|
store.set_setting("AIA_ENABLE_MCP_TOOLS", "0")
|
|
specs = materialize_tool_specs(store=store, specialist="generalist")
|
|
names = {x.name for x in specs}
|
|
assert "system_time" in names
|
|
assert "mcp__echo__ping" not in names
|
|
|