idun-agent-schema 0.1.4__py3-none-any.whl → 0.1.5__py3-none-any.whl

idun_agent_schema/manager/api.py

@@ -1,158 +1,158 @@
-"""Pydantic schemas for Agent Manager API I/O."""
-
-from datetime import datetime
-from typing import Any
-from uuid import UUID
-
-from pydantic import BaseModel, Field
-
-from .domain import AgentFramework, AgentStatus
-
-
-class AgentCreateRequest(BaseModel):
-    """Request payload to create a new agent."""
-
-    name: str = Field(..., min_length=1, max_length=255)
-    description: str | None = Field(None, max_length=1000)
-    framework: AgentFramework
-    config: dict[str, Any] = Field(default_factory=dict)
-    environment_variables: dict[str, str] = Field(default_factory=dict)
-    tags: list[str] = Field(default_factory=list)
-
-
-class AgentUpdateRequest(BaseModel):
-    """Request payload to update an existing agent (partial)."""
-
-    name: str | None = Field(None, min_length=1, max_length=255)
-    description: str | None = Field(None, max_length=1000)
-    config: dict[str, Any] | None = None
-    environment_variables: dict[str, str] | None = None
-    tags: list[str] | None = None
-
-
-class AgentRunRequest(BaseModel):
-    """Request payload to execute an agent run."""
-
-    input_data: dict[str, Any]
-    trace_id: str | None = Field(None, max_length=100)
-
-
-class AgentResponse(BaseModel):
-    """Response shape for a single agent resource."""
-
-    id: UUID
-    name: str
-    description: str | None
-    framework: AgentFramework
-    status: AgentStatus
-    config: dict[str, Any]
-    environment_variables: dict[str, str]
-    version: str
-    tags: list[str]
-    tenant_id: UUID
-    created_at: datetime
-    updated_at: datetime
-    deployed_at: datetime | None
-    total_runs: int
-    success_rate: float | None
-    avg_response_time_ms: float | None
-
-    class Config:
-        """Pydantic configuration for ORM compatibility."""
-
-        from_attributes = True
-
-
-class AgentSummaryResponse(BaseModel):
-    """Reduced agent fields for listing views."""
-
-    id: UUID
-    name: str
-    description: str | None
-    framework: AgentFramework
-    status: AgentStatus
-    version: str
-    tags: list[str]
-    created_at: datetime
-    updated_at: datetime
-    total_runs: int
-    success_rate: float | None
-
-    class Config:
-        """Pydantic configuration for ORM compatibility."""
-
-        from_attributes = True
-
-
-class AgentRunResponse(BaseModel):
-    """Detailed run record returned after execution."""
-
-    id: UUID
-    agent_id: UUID
-    tenant_id: UUID
-    input_data: dict[str, Any]
-    output_data: dict[str, Any] | None
-    status: str
-    started_at: datetime
-    completed_at: datetime | None
-    error_message: str | None
-    response_time_ms: float | None
-    tokens_used: int | None
-    cost_usd: float | None
-    trace_id: str | None
-    span_id: str | None
-
-    class Config:
-        """Pydantic configuration for ORM compatibility."""
-
-        from_attributes = True
-
-
-class AgentRunSummaryResponse(BaseModel):
-    """Reduced run fields for list views."""
-
-    id: UUID
-    agent_id: UUID
-    status: str
-    started_at: datetime
-    completed_at: datetime | None
-    response_time_ms: float | None
-    tokens_used: int | None
-    cost_usd: float | None
-
-    class Config:
-        """Pydantic configuration for ORM compatibility."""
-
-        from_attributes = True
-
-
-class PaginatedResponse(BaseModel):
-    """Base pagination container used by list endpoints."""
-
-    total: int
-    limit: int
-    offset: int
-    has_more: bool
-
-
-class PaginatedAgentsResponse(PaginatedResponse):
-    """Paginated list of agents."""
-
-    items: list[AgentSummaryResponse]
-
-
-class PaginatedRunsResponse(PaginatedResponse):
-    """Paginated list of agent runs."""
-
-    items: list[AgentRunSummaryResponse]
-
-
-class AgentStatsResponse(BaseModel):
-    """Aggregated statistics across all agents."""
-
-    total_agents: int
-    active_agents: int
-    total_runs_today: int
-    total_runs_this_month: int
-    avg_success_rate: float | None
-    avg_response_time_ms: float | None
+"""Pydantic schemas for Agent Manager API I/O."""
+
+from datetime import datetime
+from typing import Any
+from uuid import UUID
+
+from pydantic import BaseModel, Field
+
+from .domain import AgentFramework, AgentStatus
+
+
+class AgentCreateRequest(BaseModel):
+    """Request payload to create a new agent."""
+
+    name: str = Field(..., min_length=1, max_length=255)
+    description: str | None = Field(None, max_length=1000)
+    framework: AgentFramework
+    config: dict[str, Any] = Field(default_factory=dict)
+    environment_variables: dict[str, str] = Field(default_factory=dict)
+    tags: list[str] = Field(default_factory=list)
+
+
+class AgentUpdateRequest(BaseModel):
+    """Request payload to update an existing agent (partial)."""
+
+    name: str | None = Field(None, min_length=1, max_length=255)
+    description: str | None = Field(None, max_length=1000)
+    config: dict[str, Any] | None = None
+    environment_variables: dict[str, str] | None = None
+    tags: list[str] | None = None
+
+
+class AgentRunRequest(BaseModel):
+    """Request payload to execute an agent run."""
+
+    input_data: dict[str, Any]
+    trace_id: str | None = Field(None, max_length=100)
+
+
+class AgentResponse(BaseModel):
+    """Response shape for a single agent resource."""
+
+    id: UUID
+    name: str
+    description: str | None
+    framework: AgentFramework
+    status: AgentStatus
+    config: dict[str, Any]
+    environment_variables: dict[str, str]
+    version: str
+    tags: list[str]
+    tenant_id: UUID
+    created_at: datetime
+    updated_at: datetime
+    deployed_at: datetime | None
+    total_runs: int
+    success_rate: float | None
+    avg_response_time_ms: float | None
+
+    class Config:
+        """Pydantic configuration for ORM compatibility."""
+
+        from_attributes = True
+
+
+class AgentSummaryResponse(BaseModel):
+    """Reduced agent fields for listing views."""
+
+    id: UUID
+    name: str
+    description: str | None
+    framework: AgentFramework
+    status: AgentStatus
+    version: str
+    tags: list[str]
+    created_at: datetime
+    updated_at: datetime
+    total_runs: int
+    success_rate: float | None
+
+    class Config:
+        """Pydantic configuration for ORM compatibility."""
+
+        from_attributes = True
+
+
+class AgentRunResponse(BaseModel):
+    """Detailed run record returned after execution."""
+
+    id: UUID
+    agent_id: UUID
+    tenant_id: UUID
+    input_data: dict[str, Any]
+    output_data: dict[str, Any] | None
+    status: str
+    started_at: datetime
+    completed_at: datetime | None
+    error_message: str | None
+    response_time_ms: float | None
+    tokens_used: int | None
+    cost_usd: float | None
+    trace_id: str | None
+    span_id: str | None
+
+    class Config:
+        """Pydantic configuration for ORM compatibility."""
+
+        from_attributes = True
+
+
+class AgentRunSummaryResponse(BaseModel):
+    """Reduced run fields for list views."""
+
+    id: UUID
+    agent_id: UUID
+    status: str
+    started_at: datetime
+    completed_at: datetime | None
+    response_time_ms: float | None
+    tokens_used: int | None
+    cost_usd: float | None
+
+    class Config:
+        """Pydantic configuration for ORM compatibility."""
+
+        from_attributes = True
+
+
+class PaginatedResponse(BaseModel):
+    """Base pagination container used by list endpoints."""
+
+    total: int
+    limit: int
+    offset: int
+    has_more: bool
+
+
+class PaginatedAgentsResponse(PaginatedResponse):
+    """Paginated list of agents."""
+
+    items: list[AgentSummaryResponse]
+
+
+class PaginatedRunsResponse(PaginatedResponse):
+    """Paginated list of agent runs."""
+
+    items: list[AgentRunSummaryResponse]
+
+
+class AgentStatsResponse(BaseModel):
+    """Aggregated statistics across all agents."""
+
+    total_agents: int
+    active_agents: int
+    total_runs_today: int
+    total_runs_this_month: int
+    avg_success_rate: float | None
+    avg_response_time_ms: float | None

idun_agent_schema/manager/deployments.py

@@ -1,12 +1,12 @@
-"""Deployment-related enums and schemas."""
-
-from enum import Enum
-
-
-class DeploymentMode(str, Enum):
-    """Supported deployment modes."""
-
-    LOCAL = "local"
-    AWS = "aws"
-    AZURE = "azure"
-    GCP = "gcp"
+"""Deployment-related enums and schemas."""
+
+from enum import Enum
+
+
+class DeploymentMode(str, Enum):
+    """Supported deployment modes."""
+
+    LOCAL = "local"
+    AWS = "aws"
+    AZURE = "azure"
+    GCP = "gcp"

idun_agent_schema/manager/deps.py

@@ -1,14 +1,14 @@
-"""Shared dependency models used by FastAPI deps in Manager."""
-
-from uuid import UUID
-
-from pydantic import BaseModel
-
-
-class Principal(BaseModel):
-    """Authenticated caller identity and authorization context."""
-
-    user_id: str
-    tenant_id: UUID
-    roles: list[str] = []
-    workspace_ids: list[UUID] = []
+"""Shared dependency models used by FastAPI deps in Manager."""
+
+from uuid import UUID
+
+from pydantic import BaseModel
+
+
+class Principal(BaseModel):
+    """Authenticated caller identity and authorization context."""
+
+    user_id: str
+    tenant_id: UUID
+    roles: list[str] = []
+    workspace_ids: list[UUID] = []