mirror of
https://github.com/TencentCloud/Octop.git
synced 2026-09-28 19:33:02 +08:00
22bf41be14
- Add server-wide plugin enable/disable (PATCH /plugins/{id}) and a
reload endpoint to pick up CLI installs without a full restart.
- Serve read-only plugin UI assets with traversal checks.
- Persist per-agent plugin tool enable flags and hot-sync the harness
denylist so disables take effect on the next turn (no full reload).
- Add a tool catalog and plugin market UI plus a tool settings API.
- Ship demo plugins: ui-card, server-status, bilibili-anime.
- Update i18n bundles and API docs.
Co-authored-by: jubaoliang <jubaoliang@tencent.com>
Co-authored-by: CodeBuddy <noreply@codebuddy.ai>
70 lines
2.5 KiB
Python
70 lines
2.5 KiB
Python
"""Integration tests for GET/PUT /api/agents/{id}/tool-settings."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
|
|
async def test_tool_settings_list_and_put(env: Any) -> None:
|
|
client, srv, auth = env
|
|
created = await client.post(
|
|
"/api/agents",
|
|
headers=auth,
|
|
json={"name": "tool-settings-agent"},
|
|
)
|
|
assert created.status_code == 201, created.text
|
|
aid = created.json()["agent_id"]
|
|
|
|
listed = await client.get(f"/api/agents/{aid}/tool-settings", headers=auth)
|
|
assert listed.status_code == 200, listed.text
|
|
body = listed.json()
|
|
tools = body["tools"]
|
|
assert any(t["name"] == "web_fetch" and t["source"] == "builtin" for t in tools)
|
|
assert any(t["name"] == "read_file" and t["disableable"] is False for t in tools)
|
|
assert any(t["name"] == "execute" and t["disableable"] is True for t in tools)
|
|
|
|
put = await client.put(
|
|
f"/api/agents/{aid}/tool-settings",
|
|
headers=auth,
|
|
json={"disabled_builtin": ["web_fetch", "execute", "read_file"]},
|
|
)
|
|
assert put.status_code == 200, put.text
|
|
put_body = put.json()
|
|
by_name = {t["name"]: t for t in put_body["tools"] if t["source"] == "builtin"}
|
|
assert by_name["web_fetch"]["enabled"] is False
|
|
assert by_name["execute"]["enabled"] is False
|
|
assert by_name["read_file"]["enabled"] is True # critical; ignore disable
|
|
|
|
cfg = srv.app_runtime.agent_registry.get_config(aid)
|
|
assert cfg.get("tools_disabled") == ["execute", "web_fetch"]
|
|
|
|
|
|
async def test_tool_settings_patch_builtin_and_plugin(env: Any) -> None:
|
|
client, srv, auth = env
|
|
created = await client.post(
|
|
"/api/agents",
|
|
headers=auth,
|
|
json={"name": "tool-patch-agent"},
|
|
)
|
|
assert created.status_code == 201, created.text
|
|
aid = created.json()["agent_id"]
|
|
|
|
r = await client.patch(
|
|
f"/api/agents/{aid}/tool-settings/web_fetch",
|
|
headers=auth,
|
|
json={"enabled": False, "source": "builtin"},
|
|
)
|
|
assert r.status_code == 200, r.text
|
|
by_name = {t["name"]: t for t in r.json()["tools"] if t["source"] == "builtin"}
|
|
assert by_name["web_fetch"]["enabled"] is False
|
|
cfg = srv.app_runtime.agent_registry.get_config(aid)
|
|
assert "web_fetch" in set(cfg.get("tools_disabled") or [])
|
|
|
|
# Critical tool cannot be disabled.
|
|
bad = await client.patch(
|
|
f"/api/agents/{aid}/tool-settings/read_file",
|
|
headers=auth,
|
|
json={"enabled": False, "source": "builtin"},
|
|
)
|
|
assert bad.status_code == 400, bad.text
|