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

agentpool_server/opencode_server/models/pty.py

@@ -0,0 +1,43 @@
+"""PTY (Pseudo-Terminal) models."""
+
+from __future__ import annotations
+
+from typing import Literal
+
+from agentpool_server.opencode_server.models.base import OpenCodeBaseModel
+
+
+class PtyInfo(OpenCodeBaseModel):
+    """PTY session information."""
+
+    id: str
+    title: str
+    command: str
+    args: list[str]
+    cwd: str
+    status: Literal["running", "exited"]
+    pid: int
+
+
+class PtyCreateRequest(OpenCodeBaseModel):
+    """Request to create a PTY session."""
+
+    command: str | None = None
+    args: list[str] | None = None
+    cwd: str | None = None
+    title: str | None = None
+    env: dict[str, str] | None = None
+
+
+class PtySize(OpenCodeBaseModel):
+    """Terminal size."""
+
+    rows: int
+    cols: int
+
+
+class PtyUpdateRequest(OpenCodeBaseModel):
+    """Request to update a PTY session."""
+
+    title: str | None = None
+    size: PtySize | None = None

agentpool_server/opencode_server/models/question.py

@@ -0,0 +1,56 @@
+"""Question models for OpenCode compatibility."""
+
+from __future__ import annotations
+
+from pydantic import Field
+
+from agentpool_server.opencode_server.models.base import OpenCodeBaseModel
+
+
+class QuestionOption(OpenCodeBaseModel):
+    """A single option for a question."""
+
+    label: str
+    """Display text (1-5 words, concise)."""
+
+    description: str
+    """Explanation of choice."""
+
+
+class QuestionInfo(OpenCodeBaseModel):
+    """Information about a single question."""
+
+    question: str
+    """Complete question."""
+
+    header: str = Field(max_length=12)
+    """Very short label (max 12 chars)."""
+
+    options: list[QuestionOption]
+    """Available choices."""
+
+    multiple: bool | None = None
+    """Allow selecting multiple choices."""
+
+
+class QuestionRequest(OpenCodeBaseModel):
+    """A pending question request."""
+
+    id: str
+    """Unique question identifier."""
+
+    session_id: str
+    """Session identifier."""
+
+    questions: list[QuestionInfo]
+    """List of questions to ask."""
+
+    tool: dict[str, str] | None = None
+    """Optional tool context: {message_id, call_id}."""
+
+
+class QuestionReply(OpenCodeBaseModel):
+    """Reply to a question request."""
+
+    answers: list[list[str]]
+    """User answers in order of questions (each answer is an array of selected labels)."""

agentpool_server/opencode_server/models/session.py

@@ -0,0 +1,111 @@
+"""Session related models."""
+
+from __future__ import annotations
+
+from typing import Any, Literal
+
+from pydantic import Field
+
+from agentpool_server.opencode_server.models.base import OpenCodeBaseModel
+from agentpool_server.opencode_server.models.common import TimeCreatedUpdated  # noqa: TC001
+
+
+class SessionSummary(OpenCodeBaseModel):
+    """Summary information for a session."""
+
+    additions: int | None = None
+    deletions: int | None = None
+    files: int | None = None
+    diffs: list[Any] = Field(default_factory=list)
+
+
+class SessionRevert(OpenCodeBaseModel):
+    """Revert information for a session."""
+
+    message_id: str
+    diff: str | None = None
+    part_id: str | None = None
+    snapshot: str | None = None
+
+
+class SessionShare(OpenCodeBaseModel):
+    """Share information for a session."""
+
+    url: str
+
+
+class Session(OpenCodeBaseModel):
+    """Session information."""
+
+    id: str
+    project_id: str
+    directory: str
+    title: str
+    version: str = "1"
+    time: TimeCreatedUpdated
+    parent_id: str | None = None
+    summary: SessionSummary | None = None
+    revert: SessionRevert | None = None
+    share: SessionShare | None = None
+
+
+class SessionCreateRequest(OpenCodeBaseModel):
+    """Request body for creating a session."""
+
+    parent_id: str | None = None
+    title: str | None = None
+
+
+class SessionUpdateRequest(OpenCodeBaseModel):
+    """Request body for updating a session."""
+
+    title: str | None = None
+
+
+class SessionForkRequest(OpenCodeBaseModel):
+    """Request body for forking a session."""
+
+    message_id: str | None = None
+    """Optional message ID to fork from. If provided, only messages up to and including
+    this message will be copied to the forked session. If None, all messages are copied."""
+
+
+class SessionInitRequest(OpenCodeBaseModel):
+    """Request body for initializing a session (creating AGENTS.md)."""
+
+    model_id: str | None = None
+    """Optional model ID to use for the init task."""
+
+    provider_id: str | None = None
+    """Optional provider ID to use for the init task."""
+
+
+class SummarizeRequest(OpenCodeBaseModel):
+    """Request body for summarizing a session.
+
+    Matches OpenCode's compaction API. If model info is provided, uses that model
+    for the summary generation. If 'auto' is True, automatically selects the model.
+    """
+
+    model_id: str | None = None
+    """Optional model ID to use for summary generation."""
+
+    provider_id: str | None = None
+    """Optional provider ID to use for summary generation."""
+
+    auto: bool | None = None
+    """If True, automatically select the model for summarization."""
+
+
+class SessionStatus(OpenCodeBaseModel):
+    """Status of a session."""
+
+    type: Literal["idle", "busy", "retry"] = "idle"
+
+
+class Todo(OpenCodeBaseModel):
+    """Todo item for a session."""
+
+    id: str
+    content: str
+    status: Literal["pending", "in_progress", "completed"] = "pending"

agentpool_server/opencode_server/routes/__init__.py

@@ -0,0 +1,29 @@
+"""Route handlers for the OpenCode API."""
+
+from agentpool_server.opencode_server.routes.global_routes import router as global_router
+from agentpool_server.opencode_server.routes.app_routes import router as app_router
+from agentpool_server.opencode_server.routes.config_routes import router as config_router
+from agentpool_server.opencode_server.routes.session_routes import router as session_router
+from agentpool_server.opencode_server.routes.message_routes import router as message_router
+from agentpool_server.opencode_server.routes.file_routes import router as file_router
+from agentpool_server.opencode_server.routes.agent_routes import router as agent_router
+from agentpool_server.opencode_server.routes.permission_routes import router as permission_router
+from agentpool_server.opencode_server.routes.question_routes import router as question_router
+from agentpool_server.opencode_server.routes.pty_routes import router as pty_router
+from agentpool_server.opencode_server.routes.tui_routes import router as tui_router
+from agentpool_server.opencode_server.routes.lsp_routes import router as lsp_router
+
+__all__ = [
+    "agent_router",
+    "app_router",
+    "config_router",
+    "file_router",
+    "global_router",
+    "lsp_router",
+    "message_router",
+    "permission_router",
+    "pty_router",
+    "question_router",
+    "session_router",
+    "tui_router",
+]