orxt 0.1.0__py3-none-any.whl

orxt/agent/__init__.py

@@ -0,0 +1,18 @@
+from __future__ import annotations
+
+from orxt.agent._categories import load_categories, resolve_category
+from orxt.agent._loader import load_agent, load_agents
+from orxt.agent._prompt import resolve_includes, resolve_prompt
+from orxt.agent._types import Agent, ExecToolConfig, ShellConfig
+
+__all__ = [
+    "Agent",
+    "ExecToolConfig",
+    "ShellConfig",
+    "load_agent",
+    "load_agents",
+    "load_categories",
+    "resolve_category",
+    "resolve_includes",
+    "resolve_prompt",
+]

orxt/agent/_categories.py

@@ -0,0 +1,29 @@
+from __future__ import annotations
+
+import tomllib
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from pathlib import Path
+
+    from orxt.agent._types import Agent
+
+
+def load_categories(path: Path) -> dict[str, str]:
+    if not path.is_file():
+        msg = f"Categories file not found: {path}"
+        raise FileNotFoundError(msg)
+    with path.open("rb") as f:
+        data = tomllib.load(f)
+    if "categories" not in data:
+        msg = f"Missing [categories] section in {path}"
+        raise ValueError(msg)
+    categories: dict[str, str] = data["categories"]
+    return categories
+
+
+def resolve_category(agent: Agent, categories: dict[str, str]) -> str:
+    if agent.category not in categories:
+        msg = f"Unknown category: {agent.category}"
+        raise ValueError(msg)
+    return categories[agent.category]

orxt/agent/_loader.py

@@ -0,0 +1,61 @@
+from __future__ import annotations
+
+import tomllib
+from typing import TYPE_CHECKING, Any
+
+from orxt.agent._prompt import resolve_includes
+from orxt.agent._types import Agent
+
+if TYPE_CHECKING:
+    from pathlib import Path
+
+
+def load_agent(path: Path) -> Agent:
+    if not path.is_file():
+        msg = f"Agent file not found: {path}"
+        raise FileNotFoundError(msg)
+    with path.open("rb") as f:
+        data = tomllib.load(f)
+
+    agent_section: dict[str, Any] = dict(data.get("agent", {}))
+    if "tools" not in data:
+        msg = f"Missing [tools] section in {path}"
+        raise ValueError(msg)
+    tools_section: dict[str, Any] = data["tools"]
+
+    prompt_rel = agent_section.pop("prompt", "")
+    prompt_path = (path.parent / prompt_rel).resolve()
+    if not prompt_path.is_file():
+        msg = f"Prompt file not found: {prompt_path}"
+        raise FileNotFoundError(msg)
+    prompt_text = prompt_path.read_text()
+    prompt_text = resolve_includes(prompt_text, prompt_path.parent)
+
+    agent_section["prompt"] = prompt_text
+    if "allow" not in tools_section:
+        msg = f"Missing 'allow' key in [tools] section in {path}"
+        raise ValueError(msg)
+    unknown_keys = set(tools_section.keys()) - {"allow"}
+    if unknown_keys:
+        names = ", ".join(sorted(unknown_keys))
+        msg = f"Unknown keys in [tools] section: {names}"
+        raise ValueError(msg)
+    agent_section["allow"] = tools_section["allow"]
+
+    exec_configs: list[dict[str, Any]] = data.get("exec", [])
+    shell_section: dict[str, Any] | None = data.get("shell")
+    agent_section["exec_tools"] = exec_configs
+    agent_section["shell_config"] = shell_section
+
+    return Agent(**agent_section)
+
+
+def load_agents(directory: Path) -> dict[str, Agent]:
+    agents: dict[str, Agent] = {}
+    for toml_path in sorted(directory.glob("*.toml")):
+        agent = load_agent(toml_path)
+        if agent.name in agents:
+            msg = f"Duplicate agent name: {agent.name}"
+            raise ValueError(msg)
+        agents[agent.name] = agent
+    return agents

orxt/agent/_prompt.py

@@ -0,0 +1,57 @@
+from __future__ import annotations
+
+import re
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from pathlib import Path
+
+_INCLUDE_RE = re.compile(r"\{include:([^}]+)\}")
+_VAR_RE = re.compile(r"\{([a-zA-Z_][a-zA-Z0-9_]*)\}")
+
+
+def resolve_includes(template: str, base_dir: Path) -> str:
+    return _resolve_includes(template, base_dir, frozenset())
+
+
+def _resolve_includes(
+    template: str, base_dir: Path, seen: frozenset[Path]
+) -> str:
+    def replacer(match: re.Match[str]) -> str:
+        rel_path = match.group(1)
+        abs_path = (base_dir / rel_path).resolve()
+        if abs_path in seen:
+            msg = f"Circular include detected: {abs_path}"
+            raise ValueError(msg)
+        if not abs_path.is_file():
+            msg = f"Include file not found: {abs_path}"
+            raise FileNotFoundError(msg)
+        content = abs_path.read_text()
+        return _resolve_includes(content, abs_path.parent, seen | {abs_path})
+
+    return _INCLUDE_RE.sub(replacer, template)
+
+
+def resolve_prompt(template: str, variables: dict[str, str]) -> str:
+    placeholders = set(_VAR_RE.findall(template))
+    var_keys = set(variables.keys())
+
+    unresolved = placeholders - var_keys
+    if unresolved:
+        name = sorted(unresolved)[0]
+        msg = f"Unresolved placeholder: {{{name}}}"
+        raise ValueError(msg)
+
+    unused = var_keys - placeholders
+    if unused:
+        name = sorted(unused)[0]
+        msg = f"Unused variable: {name}"
+        raise ValueError(msg)
+
+    if not placeholders:
+        return template
+
+    def replacer(match: re.Match[str]) -> str:
+        return variables[match.group(1)]
+
+    return _VAR_RE.sub(replacer, template)

orxt/agent/_types.py

@@ -0,0 +1,29 @@
+from __future__ import annotations
+
+from pydantic import BaseModel, ConfigDict
+
+
+class ExecToolConfig(BaseModel):
+    model_config = ConfigDict(frozen=True, strict=True, extra="forbid")
+    name: str
+    executable: str
+    description: str
+    timeout_ceiling: int = 300
+
+
+class ShellConfig(BaseModel):
+    model_config = ConfigDict(frozen=True, strict=True, extra="forbid")
+    allowed_binaries: list[str]
+    description: str = "Execute shell commands with whitelisted binaries"
+    timeout_ceiling: int = 300
+
+
+class Agent(BaseModel):
+    model_config = ConfigDict(frozen=True, strict=True, extra="forbid")
+    name: str
+    description: str
+    prompt: str
+    category: str
+    allow: list[str]
+    exec_tools: list[ExecToolConfig] = []
+    shell_config: ShellConfig | None = None