zhaolei
6f402ffcee
- 后端: coworker 智能体框架, WS API, 文件上传, 附件处理 - 前端: Open WebUI, 文件全量走 upload API (含 MD/TXT/JSON 等文本类) - 技能: md-to-office (pandoc + wkhtmltopdf) - 修复: 上传文件路径丢失, Agent 搜索浪费, 输出文件跑到 uploads/ - 打包: PyInstaller one-dir, 预打包 pandoc/wkhtmltopdf/chromium
54 lines
2.0 KiB
Python
54 lines
2.0 KiB
Python
"""Phase 1 gate — built-in personas resolve to the same toolsets as the legacy agents.
|
|
|
|
The equivalence net: routing Code/Cowork through the persona registry must yield the exact
|
|
same tools the agent builders produce, and Ops (a markdown persona) must compose the knowledge
|
|
toolset. Ties back to the Phase 0 catalog equivalence."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from coworker.agents.base import AgentContext
|
|
from coworker.agents.code import code_agent
|
|
from coworker.agents.cowork import cowork_agent
|
|
from coworker.personas.registry import PersonaRegistry
|
|
from coworker.tools.todo import TodoList
|
|
|
|
|
|
def _ctx(tmp_path) -> AgentContext:
|
|
return AgentContext(workspace=tmp_path, executor=object(), todo=TodoList())
|
|
|
|
|
|
def _names(agent, ctx) -> set:
|
|
return {getattr(t, "__name__", "") for t in agent.build_tools(ctx)}
|
|
|
|
|
|
def test_code_persona_matches_builder(tmp_path):
|
|
reg = PersonaRegistry()
|
|
ctx = _ctx(tmp_path)
|
|
assert _names(reg.agent("code"), ctx) == _names(code_agent(), ctx)
|
|
assert reg.agent("code").requires_folder and reg.agent("code").subagents
|
|
|
|
|
|
def test_cowork_persona_matches_builder(tmp_path):
|
|
reg = PersonaRegistry()
|
|
ctx = _ctx(tmp_path)
|
|
assert _names(reg.agent("cowork"), ctx) == _names(cowork_agent(), ctx)
|
|
a = reg.agent("cowork")
|
|
assert a.messaging and a.connectors
|
|
|
|
|
|
def test_ops_persona_composes_knowledge_toolset(tmp_path):
|
|
reg = PersonaRegistry()
|
|
ctx = _ctx(tmp_path)
|
|
# Ops uses the same capability list as Cowork (files/search/shell/todo).
|
|
assert _names(reg.agent("ops"), ctx) == _names(cowork_agent(), ctx)
|
|
a = reg.agent("ops")
|
|
assert not a.requires_folder and a.scheduling and a.messaging and a.connectors
|
|
assert "read_file" in _names(a, ctx) # windowed reader, multi-root aware
|
|
|
|
|
|
def test_code_keeps_single_root_file_tools(tmp_path):
|
|
reg = PersonaRegistry()
|
|
names = _names(reg.agent("code"), _ctx(tmp_path))
|
|
assert "read_file" in names and "read_file_lines" not in names
|
|
assert "git_log" in names # code has git; cowork/ops do not
|