mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 07:03:15 +08:00
oliver
dbbe3add6a
同时收敛启动与运维脚本默认行为（含 wiki worker）、更新 Admin 可观测性与相关测试，降低首轮时延并提高运行稳定性。 Made-with: Cursor
49 lines
2 KiB
Python
49 lines
2 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
from pathlib import Path
|
|
|
|
from oclaw.runtime.agents.factory import build_gateway_executor
|
|
from oclaw.platform.persistence.sqlite_store import SqliteStore
|
|
|
|
|
|
def test_build_gateway_executor_defaults_to_generalist(tmp_path: Path) -> None:
|
|
store = SqliteStore(str(tmp_path / "ops.sqlite"))
|
|
ex = build_gateway_executor(store, lang="zh")
|
|
assert type(ex).__name__ == "Agent"
|
|
assert hasattr(ex, "tools")
|
|
assert hasattr(ex, "model")
|
|
|
|
|
|
def test_build_gateway_executor_generalist_run_command_is_disabled_by_default(tmp_path: Path) -> None:
|
|
os.environ.pop("AIA_ENABLE_RUN_COMMAND", None)
|
|
os.environ["OPS_ASSISTANT_DB_PATH"] = str(tmp_path / "ops.sqlite")
|
|
try:
|
|
store = SqliteStore(str(tmp_path / "ops.sqlite"))
|
|
ex = build_gateway_executor(store, lang="zh", specialist="generalist")
|
|
run_tool = next((t for t in ex.tools.list() if t.name == "run_command"), None)
|
|
assert run_tool is not None
|
|
out = run_tool.handler({"command": "echo hi"})
|
|
assert isinstance(out, dict)
|
|
assert out.get("ok") is False
|
|
assert out.get("error") == "disabled"
|
|
finally:
|
|
os.environ.pop("OPS_ASSISTANT_DB_PATH", None)
|
|
|
|
|
|
def test_build_gateway_executor_generalist_run_command_prefers_db_setting(tmp_path: Path) -> None:
|
|
os.environ["AIA_ENABLE_RUN_COMMAND"] = "0"
|
|
os.environ["OPS_ASSISTANT_DB_PATH"] = str(tmp_path / "ops.sqlite")
|
|
try:
|
|
store = SqliteStore(str(tmp_path / "ops.sqlite"))
|
|
store.set_setting("AIA_ENABLE_RUN_COMMAND", "1")
|
|
ex = build_gateway_executor(store, lang="zh", specialist="generalist")
|
|
run_tool = next((t for t in ex.tools.list() if t.name == "run_command"), None)
|
|
assert run_tool is not None
|
|
out = run_tool.handler({"command": "echo hi"})
|
|
assert isinstance(out, dict)
|
|
assert out.get("ok") is True
|
|
finally:
|
|
os.environ.pop("AIA_ENABLE_RUN_COMMAND", None)
|
|
os.environ.pop("OPS_ASSISTANT_DB_PATH", None)
|
|
|