idun-agent-schema 0.1.41__py3-none-any.whl → 0.1.42__py3-none-any.whl

idun_agent_schema/manager/dto.py

@@ -1,131 +1,131 @@
-"""DTOs for Manager operations."""
-
-from datetime import datetime
-from typing import Any
-from uuid import UUID
-
-from pydantic import BaseModel, Field
-
-from .domain import AgentFramework, TenantPlan
-
-
-class AgentCreateDTO(BaseModel):
-    """DTO for creating an agent in the application layer."""
-
-    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)
-    tenant_id: UUID
-
-
-class AgentUpdateDTO(BaseModel):
-    """DTO for updating an existing agent."""
-
-    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 AgentDeploymentDTO(BaseModel):
-    """DTO describing deployment details for an agent."""
-
-    agent_id: UUID
-    container_id: str
-    endpoint: str
-    status: str
-    framework: str
-    deployed_at: datetime | None = None
-
-
-class AgentHealthDTO(BaseModel):
-    """DTO representing health metrics for an agent instance."""
-
-    agent_id: UUID
-    status: str
-    uptime: str | None = None
-    cpu_usage: str | None = None
-    memory_usage: str | None = None
-    last_activity: str | None = None
-    error: str | None = None
-
-
-class AgentRunCreateDTO(BaseModel):
-    """DTO for creating a new agent run."""
-
-    agent_id: UUID
-    tenant_id: UUID
-    input_data: dict[str, Any]
-    trace_id: str | None = None
-
-
-class AgentMetricsDTO(BaseModel):
-    """DTO for aggregated performance metrics of an agent."""
-
-    agent_id: UUID
-    total_runs: int
-    success_rate: float | None = Field(None, ge=0.0, le=1.0)
-    avg_response_time_ms: float | None = Field(None, ge=0)
-    last_run_at: datetime | None = None
-
-
-class TenantCreateDTO(BaseModel):
-    """DTO for creating a tenant."""
-
-    name: str = Field(..., min_length=1, max_length=255)
-    slug: str = Field(..., min_length=1, max_length=100)
-    email: str = Field(..., description="Primary contact email")
-    website: str | None = None
-    plan: TenantPlan = Field(default=TenantPlan.FREE)
-
-
-class TenantUpdateDTO(BaseModel):
-    """DTO for updating tenant metadata and settings."""
-
-    name: str | None = Field(None, min_length=1, max_length=255)
-    email: str | None = None
-    website: str | None = None
-    settings: dict[str, Any] | None = None
-
-
-class TenantUsageDTO(BaseModel):
-    """DTO summarizing a tenant's resource usage and quotas."""
-
-    tenant_id: UUID
-    current_agents: int
-    max_agents: int
-    current_runs_this_month: int
-    max_runs_per_month: int
-    current_storage_mb: float
-    max_storage_mb: int
-    usage_percentage: dict[str, float]
-
-
-class TenantQuotaDTO(BaseModel):
-    """DTO for updating tenant quotas."""
-
-    max_agents: int | None = None
-    max_runs_per_month: int | None = None
-    max_storage_mb: int | None = None
-
-
-class TenantUserCreateDTO(BaseModel):
-    """DTO for adding a user to a tenant."""
-
-    tenant_id: UUID
-    user_id: str
-    email: str
-    role: str = Field(default="member")
-    permissions: list[str] = Field(default_factory=list)
-
-
-class TenantUserUpdateDTO(BaseModel):
-    """DTO for updating tenant user fields."""
-
-    role: str | None = None
-    permissions: list[str] | None = None
-    is_active: bool | None = None
+"""DTOs for Manager operations."""
+
+from datetime import datetime
+from typing import Any
+from uuid import UUID
+
+from pydantic import BaseModel, Field
+
+from .domain import AgentFramework, TenantPlan
+
+
+class AgentCreateDTO(BaseModel):
+    """DTO for creating an agent in the application layer."""
+
+    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)
+    tenant_id: UUID
+
+
+class AgentUpdateDTO(BaseModel):
+    """DTO for updating an existing agent."""
+
+    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 AgentDeploymentDTO(BaseModel):
+    """DTO describing deployment details for an agent."""
+
+    agent_id: UUID
+    container_id: str
+    endpoint: str
+    status: str
+    framework: str
+    deployed_at: datetime | None = None
+
+
+class AgentHealthDTO(BaseModel):
+    """DTO representing health metrics for an agent instance."""
+
+    agent_id: UUID
+    status: str
+    uptime: str | None = None
+    cpu_usage: str | None = None
+    memory_usage: str | None = None
+    last_activity: str | None = None
+    error: str | None = None
+
+
+class AgentRunCreateDTO(BaseModel):
+    """DTO for creating a new agent run."""
+
+    agent_id: UUID
+    tenant_id: UUID
+    input_data: dict[str, Any]
+    trace_id: str | None = None
+
+
+class AgentMetricsDTO(BaseModel):
+    """DTO for aggregated performance metrics of an agent."""
+
+    agent_id: UUID
+    total_runs: int
+    success_rate: float | None = Field(None, ge=0.0, le=1.0)
+    avg_response_time_ms: float | None = Field(None, ge=0)
+    last_run_at: datetime | None = None
+
+
+class TenantCreateDTO(BaseModel):
+    """DTO for creating a tenant."""
+
+    name: str = Field(..., min_length=1, max_length=255)
+    slug: str = Field(..., min_length=1, max_length=100)
+    email: str = Field(..., description="Primary contact email")
+    website: str | None = None
+    plan: TenantPlan = Field(default=TenantPlan.FREE)
+
+
+class TenantUpdateDTO(BaseModel):
+    """DTO for updating tenant metadata and settings."""
+
+    name: str | None = Field(None, min_length=1, max_length=255)
+    email: str | None = None
+    website: str | None = None
+    settings: dict[str, Any] | None = None
+
+
+class TenantUsageDTO(BaseModel):
+    """DTO summarizing a tenant's resource usage and quotas."""
+
+    tenant_id: UUID
+    current_agents: int
+    max_agents: int
+    current_runs_this_month: int
+    max_runs_per_month: int
+    current_storage_mb: float
+    max_storage_mb: int
+    usage_percentage: dict[str, float]
+
+
+class TenantQuotaDTO(BaseModel):
+    """DTO for updating tenant quotas."""
+
+    max_agents: int | None = None
+    max_runs_per_month: int | None = None
+    max_storage_mb: int | None = None
+
+
+class TenantUserCreateDTO(BaseModel):
+    """DTO for adding a user to a tenant."""
+
+    tenant_id: UUID
+    user_id: str
+    email: str
+    role: str = Field(default="member")
+    permissions: list[str] = Field(default_factory=list)
+
+
+class TenantUserUpdateDTO(BaseModel):
+    """DTO for updating tenant user fields."""
+
+    role: str | None = None
+    permissions: list[str] | None = None
+    is_active: bool | None = None

idun_agent_schema/manager/errors.py

@@ -1,22 +1,22 @@
-"""RFC 9457 Problem Details model (shared)."""
-
-from typing import Any
-
-from pydantic import BaseModel, Field
-
-
-class ProblemDetail(BaseModel):
-    """RFC 9457 Problem Details model."""
-
-    type: str = Field(default="about:blank")
-    title: str = Field()
-    status: int = Field()
-    detail: str | None = Field(default=None)
-    instance: str | None = Field(default=None)
-
-    # Extension members
-    timestamp: str | None = Field(default=None)
-    request_id: str | None = Field(default=None)
-    errors: dict[str, Any] | None = Field(default=None)
-
-    model_config = {"extra": "allow"}
+"""RFC 9457 Problem Details model (shared)."""
+
+from typing import Any
+
+from pydantic import BaseModel, Field
+
+
+class ProblemDetail(BaseModel):
+    """RFC 9457 Problem Details model."""
+
+    type: str = Field(default="about:blank")
+    title: str = Field()
+    status: int = Field()
+    detail: str | None = Field(default=None)
+    instance: str | None = Field(default=None)
+
+    # Extension members
+    timestamp: str | None = Field(default=None)
+    request_id: str | None = Field(default=None)
+    errors: dict[str, Any] | None = Field(default=None)
+
+    model_config = {"extra": "allow"}

idun_agent_schema/manager/settings.py

@@ -1,161 +1,161 @@
-"""Application settings schemas using Pydantic Settings v2."""
-
-from typing import Literal
-
-from pydantic import Field, field_validator
-from pydantic_settings import BaseSettings, SettingsConfigDict
-
-
-class DatabaseSettings(BaseSettings):
-    """Database configuration settings for the Manager service."""
-
-    url: str = Field(
-        default="postgresql+asyncpg://postgres:postgres@localhost:55432/idun_agents"
-    )
-    echo: bool = Field(default=False)
-    pool_size: int = Field(default=10)
-    max_overflow: int = Field(default=20)
-    pool_pre_ping: bool = Field(default=True)
-
-    model_config = SettingsConfigDict(env_prefix="DATABASE_", env_file=".env")
-
-
-class RedisSettings(BaseSettings):
-    """Redis cache settings."""
-
-    url: str = Field(default="redis://localhost:6379/0")
-    max_connections: int = Field(default=20)
-
-    model_config = SettingsConfigDict(env_prefix="REDIS_", env_file=".env")
-
-
-class AuthSettings(BaseSettings):
-    """Authentication and OIDC-related configuration."""
-
-    provider_type: Literal["okta", "auth0", "entra", "google"] = Field(default="auth0")
-    issuer: str = Field(default="")
-    client_id: str = Field(default="")
-    client_secret: str = Field(default="")
-    audience: str | None = Field(default=None)
-    redirect_uri: str | None = Field(default=None)
-    scopes: list[str] = Field(default_factory=lambda: ["openid", "profile", "email"])
-
-    allowed_algs: list[str] = Field(default_factory=lambda: ["RS256", "RS512", "ES256"])
-    jwks_cache_ttl: int = Field(default=300)
-    clock_skew_seconds: int = Field(default=60)
-    expected_audiences: list[str] = Field(default_factory=list)
-
-    claim_user_id_path: list[str] | None = Field(default=None)
-    claim_email_path: list[str] | None = Field(default=None)
-    claim_roles_paths: list[list[str]] | None = Field(default=None)
-    claim_groups_paths: list[list[str]] | None = Field(default=None)
-    claim_workspace_ids_paths: list[list[str]] | None = Field(default=None)
-
-    @field_validator("claim_user_id_path", "claim_email_path", mode="before")
-    @classmethod
-    def _string_to_list(cls, v):
-        if isinstance(v, str) and v.strip().startswith("#"):
-            return None
-        return v
-
-    @field_validator(
-        "claim_roles_paths",
-        "claim_groups_paths",
-        "claim_workspace_ids_paths",
-        mode="before",
-    )
-    @classmethod
-    def _string_to_list_of_lists(cls, v):
-        if isinstance(v, str) and v.strip().startswith("#"):
-            return None
-        return v
-
-    model_config = SettingsConfigDict(
-        env_prefix="AUTH_", env_file=".env", extra="ignore"
-    )
-
-    @field_validator("issuer")
-    @classmethod
-    def validate_issuer(cls, v: str) -> str:
-        """Validate that issuer is either empty or a URL."""
-        if not v:
-            return v
-        if not v.startswith("http"):
-            raise ValueError("issuer must be a URL")
-        return v
-
-
-class ObservabilitySettings(BaseSettings):
-    """OpenTelemetry and logging settings."""
-
-    otel_service_name: str = Field(default="idun-agent-manager")
-    otel_exporter_endpoint: str | None = Field(default=None)
-    otel_exporter_headers: str | None = Field(default=None)
-    log_level: str = Field(default="INFO")
-    log_format: str = Field(default="json")
-
-    model_config = SettingsConfigDict(env_prefix="OTEL_", env_file=".env")
-
-
-class CelerySettings(BaseSettings):
-    """Celery broker/result backend settings."""
-
-    broker_url: str = Field(default="redis://localhost:6379/1")
-    result_backend: str = Field(default="redis://localhost:6379/2")
-    task_serializer: str = Field(default="json")
-    result_serializer: str = Field(default="json")
-    accept_content: list[str] = Field(default=["json"])
-    timezone: str = Field(default="UTC")
-
-    model_config = SettingsConfigDict(env_prefix="CELERY_", env_file=".env")
-
-
-class APISettings(BaseSettings):
-    """API metadata, CORS, and rate limiting settings."""
-
-    title: str = Field(default="Idun Agent Manager API")
-    description: str = Field(default="Modern FastAPI backend for managing AI agents")
-    version: str = Field(default="0.1.0")
-    docs_url: str = Field(default="/docs")
-    redoc_url: str = Field(default="/redoc")
-    openapi_url: str = Field(default="/openapi.json")
-    cors_origins: list[str] = Field(
-        default=["http://localhost:3000", "http://localhost:8080"]
-    )
-    cors_methods: list[str] = Field(default=["*"])
-    cors_headers: list[str] = Field(default=["*"])
-    rate_limit_enabled: bool = Field(default=True)
-    rate_limit_requests: int = Field(default=100)
-    rate_limit_window: int = Field(default=60)
-
-    model_config = SettingsConfigDict(
-        env_prefix="API_", env_file=".env", extra="ignore"
-    )
-
-
-class Settings(BaseSettings):
-    """Top-level application settings composed of sub-settings."""
-
-    environment: str = Field(default="development")
-    debug: bool = Field(default=False)
-    testing: bool = Field(default=False)
-    host: str = Field(default="0.0.0.0")
-    port: int = Field(default=8000)
-    workers: int = Field(default=1)
-    reload: bool = Field(default=False)
-    is_development: bool = Field(default=True)
-
-    database: DatabaseSettings = Field(default_factory=DatabaseSettings)
-    redis: RedisSettings = Field(default_factory=RedisSettings)
-    auth: AuthSettings = Field(default_factory=AuthSettings)
-    observability: ObservabilitySettings = Field(default_factory=ObservabilitySettings)
-    celery: CelerySettings = Field(default_factory=CelerySettings)
-    api: APISettings = Field(default_factory=APISettings)
-
-    model_config = SettingsConfigDict(
-        env_file=".env",
-        env_file_encoding="utf-8",
-        env_nested_delimiter="__",
-        case_sensitive=False,
-        extra="ignore",
-    )
+"""Application settings schemas using Pydantic Settings v2."""
+
+from typing import Literal
+
+from pydantic import Field, field_validator
+from pydantic_settings import BaseSettings, SettingsConfigDict
+
+
+class DatabaseSettings(BaseSettings):
+    """Database configuration settings for the Manager service."""
+
+    url: str = Field(
+        default="postgresql+asyncpg://postgres:postgres@localhost:55432/idun_agents"
+    )
+    echo: bool = Field(default=False)
+    pool_size: int = Field(default=10)
+    max_overflow: int = Field(default=20)
+    pool_pre_ping: bool = Field(default=True)
+
+    model_config = SettingsConfigDict(env_prefix="DATABASE_", env_file=".env")
+
+
+class RedisSettings(BaseSettings):
+    """Redis cache settings."""
+
+    url: str = Field(default="redis://localhost:6379/0")
+    max_connections: int = Field(default=20)
+
+    model_config = SettingsConfigDict(env_prefix="REDIS_", env_file=".env")
+
+
+class AuthSettings(BaseSettings):
+    """Authentication and OIDC-related configuration."""
+
+    provider_type: Literal["okta", "auth0", "entra", "google"] = Field(default="auth0")
+    issuer: str = Field(default="")
+    client_id: str = Field(default="")
+    client_secret: str = Field(default="")
+    audience: str | None = Field(default=None)
+    redirect_uri: str | None = Field(default=None)
+    scopes: list[str] = Field(default_factory=lambda: ["openid", "profile", "email"])
+
+    allowed_algs: list[str] = Field(default_factory=lambda: ["RS256", "RS512", "ES256"])
+    jwks_cache_ttl: int = Field(default=300)
+    clock_skew_seconds: int = Field(default=60)
+    expected_audiences: list[str] = Field(default_factory=list)
+
+    claim_user_id_path: list[str] | None = Field(default=None)
+    claim_email_path: list[str] | None = Field(default=None)
+    claim_roles_paths: list[list[str]] | None = Field(default=None)
+    claim_groups_paths: list[list[str]] | None = Field(default=None)
+    claim_workspace_ids_paths: list[list[str]] | None = Field(default=None)
+
+    @field_validator("claim_user_id_path", "claim_email_path", mode="before")
+    @classmethod
+    def _string_to_list(cls, v):
+        if isinstance(v, str) and v.strip().startswith("#"):
+            return None
+        return v
+
+    @field_validator(
+        "claim_roles_paths",
+        "claim_groups_paths",
+        "claim_workspace_ids_paths",
+        mode="before",
+    )
+    @classmethod
+    def _string_to_list_of_lists(cls, v):
+        if isinstance(v, str) and v.strip().startswith("#"):
+            return None
+        return v
+
+    model_config = SettingsConfigDict(
+        env_prefix="AUTH_", env_file=".env", extra="ignore"
+    )
+
+    @field_validator("issuer")
+    @classmethod
+    def validate_issuer(cls, v: str) -> str:
+        """Validate that issuer is either empty or a URL."""
+        if not v:
+            return v
+        if not v.startswith("http"):
+            raise ValueError("issuer must be a URL")
+        return v
+
+
+class ObservabilitySettings(BaseSettings):
+    """OpenTelemetry and logging settings."""
+
+    otel_service_name: str = Field(default="idun-agent-manager")
+    otel_exporter_endpoint: str | None = Field(default=None)
+    otel_exporter_headers: str | None = Field(default=None)
+    log_level: str = Field(default="INFO")
+    log_format: str = Field(default="json")
+
+    model_config = SettingsConfigDict(env_prefix="OTEL_", env_file=".env")
+
+
+class CelerySettings(BaseSettings):
+    """Celery broker/result backend settings."""
+
+    broker_url: str = Field(default="redis://localhost:6379/1")
+    result_backend: str = Field(default="redis://localhost:6379/2")
+    task_serializer: str = Field(default="json")
+    result_serializer: str = Field(default="json")
+    accept_content: list[str] = Field(default=["json"])
+    timezone: str = Field(default="UTC")
+
+    model_config = SettingsConfigDict(env_prefix="CELERY_", env_file=".env")
+
+
+class APISettings(BaseSettings):
+    """API metadata, CORS, and rate limiting settings."""
+
+    title: str = Field(default="Idun Agent Manager API")
+    description: str = Field(default="Modern FastAPI backend for managing AI agents")
+    version: str = Field(default="0.1.0")
+    docs_url: str = Field(default="/docs")
+    redoc_url: str = Field(default="/redoc")
+    openapi_url: str = Field(default="/openapi.json")
+    cors_origins: list[str] = Field(
+        default=["http://localhost:3000", "http://localhost:8080"]
+    )
+    cors_methods: list[str] = Field(default=["*"])
+    cors_headers: list[str] = Field(default=["*"])
+    rate_limit_enabled: bool = Field(default=True)
+    rate_limit_requests: int = Field(default=100)
+    rate_limit_window: int = Field(default=60)
+
+    model_config = SettingsConfigDict(
+        env_prefix="API_", env_file=".env", extra="ignore"
+    )
+
+
+class Settings(BaseSettings):
+    """Top-level application settings composed of sub-settings."""
+
+    environment: str = Field(default="development")
+    debug: bool = Field(default=False)
+    testing: bool = Field(default=False)
+    host: str = Field(default="0.0.0.0")
+    port: int = Field(default=8000)
+    workers: int = Field(default=1)
+    reload: bool = Field(default=False)
+    is_development: bool = Field(default=True)
+
+    database: DatabaseSettings = Field(default_factory=DatabaseSettings)
+    redis: RedisSettings = Field(default_factory=RedisSettings)
+    auth: AuthSettings = Field(default_factory=AuthSettings)
+    observability: ObservabilitySettings = Field(default_factory=ObservabilitySettings)
+    celery: CelerySettings = Field(default_factory=CelerySettings)
+    api: APISettings = Field(default_factory=APISettings)
+
+    model_config = SettingsConfigDict(
+        env_file=".env",
+        env_file_encoding="utf-8",
+        env_nested_delimiter="__",
+        case_sensitive=False,
+        extra="ignore",
+    )

idun_agent_schema/shared/__init__.py

@@ -1,5 +1,5 @@
-"""Shared cross-cutting schemas used by both Engine and Manager."""
-
-from .observability import ObservabilityConfig
-
-__all__ = ["ObservabilityConfig"]
+"""Shared cross-cutting schemas used by both Engine and Manager."""
+
+from .observability import ObservabilityConfig
+
+__all__ = ["ObservabilityConfig"]