agentpool 2.1.9__py3-none-any.whl → 2.5.0__py3-none-any.whl

agentpool_server/opencode_server/models/agent.py

@@ -0,0 +1,53 @@
+"""Agent and command models."""
+
+from __future__ import annotations
+
+from typing import Literal
+
+from pydantic import Field
+
+from agentpool_server.opencode_server.models.base import OpenCodeBaseModel
+
+
+class AgentPermission(OpenCodeBaseModel):
+    """Agent permission settings."""
+
+    edit: Literal["ask", "allow", "deny"] = "ask"
+    bash: dict[str, Literal["ask", "allow", "deny"]] = Field(default_factory=dict)
+    skill: dict[str, Literal["ask", "allow", "deny"]] = Field(default_factory=dict)
+    webfetch: Literal["ask", "allow", "deny"] | None = None
+    doom_loop: Literal["ask", "allow", "deny"] | None = None
+    external_directory: Literal["ask", "allow", "deny"] | None = None
+
+
+class AgentModel(OpenCodeBaseModel):
+    """Agent model configuration."""
+
+    model_id: str
+    provider_id: str
+
+
+class Agent(OpenCodeBaseModel):
+    """Agent information matching SDK type."""
+
+    name: str
+    description: str | None = None
+    mode: Literal["subagent", "primary", "all"] = "primary"
+    native: bool | None = None
+    hidden: bool | None = None
+    default: bool | None = None
+    top_p: float | None = None
+    temperature: float | None = None
+    color: str | None = None
+    permission: AgentPermission = Field(default_factory=AgentPermission)
+    model: AgentModel | None = None
+    prompt: str | None = None
+    tools: dict[str, bool] = Field(default_factory=dict)
+    options: dict[str, str] = Field(default_factory=dict)
+
+
+class Command(OpenCodeBaseModel):
+    """Slash command."""
+
+    name: str
+    description: str = ""

agentpool_server/opencode_server/models/app.py

@@ -0,0 +1,72 @@
+"""App, project, and path related models."""
+
+from typing import Any
+
+from agentpool_server.opencode_server.models.base import OpenCodeBaseModel
+
+
+class HealthResponse(OpenCodeBaseModel):
+    """Response for /global/health endpoint."""
+
+    healthy: bool = True
+    version: str
+
+
+class PathInfo(OpenCodeBaseModel):
+    """Path information for app/project."""
+
+    config: str = ""
+    cwd: str
+    data: str = ""
+    root: str
+    state: str = ""
+
+
+class AppTimeInfo(OpenCodeBaseModel):
+    """App time information."""
+
+    initialized: float | None = None
+
+
+class App(OpenCodeBaseModel):
+    """App information response."""
+
+    git: bool = False
+    hostname: str = "localhost"
+    path: PathInfo
+    time: AppTimeInfo
+
+
+class ProjectTime(OpenCodeBaseModel):
+    """Project time information."""
+
+    created: int
+    initialized: int | None = None
+
+
+class Project(OpenCodeBaseModel):
+    """Project information."""
+
+    id: str
+    worktree: str
+    vcs_dir: str | None = None
+    vcs: str | None = None  # "git" or None
+    time: ProjectTime
+
+
+class VcsInfo(OpenCodeBaseModel):
+    """VCS (git) information."""
+
+    branch: str | None = None
+    dirty: bool = False
+    commit: str | None = None
+
+
+class ProjectUpdateRequest(OpenCodeBaseModel):
+    """Request to update project metadata."""
+
+    name: str | None = None
+    """Optional friendly name for the project."""
+
+    settings: dict[str, Any] | None = None
+    """Optional project-specific settings to update."""

agentpool_server/opencode_server/models/base.py

@@ -0,0 +1,26 @@
+"""Base model for all OpenCode API models."""
+
+from pydantic import BaseModel, ConfigDict
+from pydantic.alias_generators import to_camel
+
+
+def convert(text: str) -> str:
+    val = to_camel(text)
+    if val.endswith("Id"):
+        return val.rstrip("Id") + "ID"
+    return val
+
+
+class OpenCodeBaseModel(BaseModel):
+    """Base model with OpenCode-compatible configuration.
+
+    All OpenCode models should inherit from this to ensure:
+    - Fields can be populated by their alias (camelCase) or Python name (snake_case)
+    - Serialization uses aliases by default for API compatibility
+    """
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        alias_generator=convert,
+        use_attribute_docstrings=True,
+    )

agentpool_server/opencode_server/models/common.py

@@ -0,0 +1,23 @@
+"""Common/shared models used across multiple domains."""
+
+from agentpool_server.opencode_server.models.base import OpenCodeBaseModel
+
+
+class TimeCreatedUpdated(OpenCodeBaseModel):
+    """Timestamp with created and updated fields (milliseconds)."""
+
+    created: int
+    updated: int
+
+
+class TimeCreated(OpenCodeBaseModel):
+    """Timestamp with created field only (milliseconds)."""
+
+    created: int
+
+
+class TimeStartEnd(OpenCodeBaseModel):
+    """Timestamp with start and optional end (milliseconds)."""
+
+    start: int
+    end: int | None = None

agentpool_server/opencode_server/models/config.py

@@ -0,0 +1,37 @@
+"""Config models."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from agentpool_server.opencode_server.models.base import OpenCodeBaseModel
+
+
+class Config(OpenCodeBaseModel):
+    """Server configuration.
+
+    This is a simplified version - we only include fields the TUI needs.
+    """
+
+    # Model settings
+    model: str | None = None
+    small_model: str | None = None
+
+    # Theme and UI
+    theme: str | None = None
+    username: str | None = None
+
+    # Sharing
+    share: str | None = None  # "manual", "auto", "disabled"
+
+    # Provider configurations (simplified)
+    provider: dict[str, Any] | None = None
+
+    # MCP configurations
+    mcp: dict[str, Any] | None = None
+
+    # Instructions
+    instructions: list[str] | None = None
+
+    # Auto-update
+    autoupdate: bool | None = None