mirror of
https://github.com/andrewyng/openworker.git
synced 2026-09-03 23:03:22 +00:00
Rohit C PrasadandDevika
2b45018ffa
Imported from andrewyng/aisuite@1b4bbf303e (contents of its platform/ directory, hoisted to the repo root). Development history prior to this commit lives in that repository. Co-authored-by: Devika <devikaverma11@gmail.com>
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").family == "code"
|
|
|
|
|
|
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 a.family == "knowledge" and a.messaging and a.connectors
|
|
assert "read_file_lines" in _names(a, ctx) # multi-root knowledge files
|
|
|
|
|
|
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
|