mirror of
https://github.com/andrewyng/openworker.git
synced 2026-09-04 07:20:10 +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>
45 lines
1.6 KiB
Python
45 lines
1.6 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. Defaults keep non-persona callers behaving as before.
|
|
family: str = "knowledge"
|
|
messaging: bool = False
|
|
connectors: bool = False
|
|
|
|
def build_tools(self, context: AgentContext) -> list:
|
|
return list(self.tool_factory(context)) if self.tool_factory else []
|