Files
openworker/tests/test_builtin_personas.py
T
Rohit C Prasad 2c9fb7490b Toolset dedup: one windowed reader, sleep_until only, browser trim/rename
files capability adopts the numbered read_file; per-turn Now: line feeds absolute scheduling.
browser_get_text/browser_read_url dropped; browser_snapshot renamed browser_read_page.
2026-08-20 23:06:36 -07:00

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