Files
openworker/coworker/agents/base.py
T
Rohit C Prasad 3e4fafead0 Team wake plumbing: trait-gated verbs, durable queues, staffing gate, digests (OPE-97)
team: manifest trait gates lead/worker toolsets; propose_team pre-spawns worker sessions on approval (fail closed on solo personas).
Deliveries + lead subscriptions are cursor-consumed projections; turns end with a queue kick, ticks replay; timer wakes carry the code-computed staleness digest; hourly wake cap is the budget gate.
2026-08-16 08:37:07 -07:00

50 lines
2.0 KiB
Python

"""Agent — a top-level surface (Code / Chat / Cowork).
An agent owns its system prompt + base toolset + whether it needs a workspace. Distinct
from a Skill: skills are Anthropic-format, loadable capabilities that ANY agent can pull
in (see coworker.skills).
"""
from __future__ import annotations
from dataclasses import dataclass
from pathlib import Path
from typing import Any, Callable, Optional
from ..tools.todo import TodoList
@dataclass
class AgentContext:
workspace: Optional[Path] = None
executor: Optional[Any] = None
todo: Optional[TodoList] = None
# Shared, mutable list of RootDir the session may touch (primary scratch + added folders).
# When None, tools fall back to the single `workspace` root. Held by reference so runtime
# add/remove of folders is seen by the file tools built from it.
roots: Optional[list] = None
@dataclass
class Agent:
name: str
title: str
system_prompt: str
needs_workspace: bool = False
tool_factory: Optional[Callable[[AgentContext], list]] = None
# Traits that replace the old per-agent-name branching in build_engine / manager.
# family: "code" gets explorer subagents; "knowledge" gets scheduling / request_directory /
# roots context (when it has a workspace). messaging: exposes send_message. connectors:
# loads the integration toolset — True = every connected connector (general builtins
# only), a tuple = allowlist (session gets declared ∩ connected; OPE-93), False = none.
# Defaults keep non-persona callers behaving as before.
family: str = "knowledge"
messaging: bool = False
connectors: bool | tuple[str, ...] = False
# Team identity: "lead" | "worker" | None (solo-only). Gates the board/journal
# toolsets and staffing eligibility — solo personas are never team-staffable.
team: Optional[str] = None
def build_tools(self, context: AgentContext) -> list:
return list(self.tool_factory(context)) if self.tool_factory else []