idun-agent-schema 0.1.0__tar.gz → 0.1.1__tar.gz

{idun_agent_schema-0.1.0 → idun_agent_schema-0.1.1}/.gitignore

@@ -5,7 +5,7 @@ __pycache__/
 
 # C extensions
 *.so
-
+.DS_Store
 # Distribution / packaging
 .Python
 build/
@@ -199,3 +199,8 @@ poetry.lock
 
 # Environment variables
 .env
+
+# Custom Gitignore for Idun
+*.db-wal
+.DS_Store
+*.DS_Store

{idun_agent_schema-0.1.0 → idun_agent_schema-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: idun-agent-schema
-Version: 0.1.0
+Version: 0.1.1
 Summary: Centralized Pydantic schema library for Idun Agent Engine and Manager
 Project-URL: Homepage, https://github.com/geoffreyharrazi/idun-agent-platform
 Project-URL: Repository, https://github.com/geoffreyharrazi/idun-agent-platform
@@ -36,5 +36,3 @@ from idun_agent_schema.manager.api import AgentCreateRequest
 ```
 
 This package re-exports stable schema namespaces to avoid breaking existing imports. Prefer importing from this package directly going forward.
-
-

{idun_agent_schema-0.1.0 → idun_agent_schema-0.1.1}/README.md

@@ -16,5 +16,3 @@ from idun_agent_schema.manager.api import AgentCreateRequest
 ```
 
 This package re-exports stable schema namespaces to avoid breaking existing imports. Prefer importing from this package directly going forward.
-
-

{idun_agent_schema-0.1.0 → idun_agent_schema-0.1.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "idun-agent-schema"
-version = "0.1.0"
+version = "0.1.1"
 description = "Centralized Pydantic schema library for Idun Agent Engine and Manager"
 authors = [{ name = "Idun Team", email = "team@idun.com" }]
 readme = "README.md"
@@ -72,5 +72,3 @@ show_error_codes = true
 [tool.pytest.ini_options]
 addopts = "-ra -q --strict-markers --disable-warnings"
 testpaths = ["tests"]
-
-

{idun_agent_schema-0.1.0 → idun_agent_schema-0.1.1}/src/idun_agent_schema/__init__.py

@@ -8,5 +8,3 @@ Public namespaces:
 
 # Re-export key types for convenience
 from .shared.observability import ObservabilityConfig  # noqa: F401
-
-

{idun_agent_schema-0.1.0 → idun_agent_schema-0.1.1}/src/idun_agent_schema/engine/__init__.py

@@ -9,5 +9,3 @@ from .langgraph import (  # noqa: F401
     SqliteCheckpointConfig,
 )
 from .server import ServerAPIConfig, ServerConfig  # noqa: F401
-
-

{idun_agent_schema-0.1.0 → idun_agent_schema-0.1.1}/src/idun_agent_schema/engine/agent.py

@@ -14,5 +14,3 @@ class BaseAgentConfig(BaseModel):
     input_schema_definition: dict[str, Any] | None = Field(default_factory=dict)
     output_schema_definition: dict[str, Any] | None = Field(default_factory=dict)
     observability: ObservabilityConfig | None = Field(default=None)
-
-

idun_agent_schema-0.1.1/src/idun_agent_schema/engine/api.py

@@ -0,0 +1,29 @@
+"""Schemas for engine HTTP API request/response payloads."""
+
+from pydantic import BaseModel
+
+
+class ChatRequest(BaseModel):
+    """Request payload for synchronous and streaming chat endpoints.
+
+    Attributes:
+        session_id: Client-provided session identifier for routing state.
+        query: Natural language prompt or input for the agent.
+
+    """
+
+    session_id: str
+    query: str
+
+
+class ChatResponse(BaseModel):
+    """Response payload for chat endpoints.
+
+    Attributes:
+        session_id: Echoed session identifier.
+        response: Agent's textual response.
+
+    """
+
+    session_id: str
+    response: str

{idun_agent_schema-0.1.0 → idun_agent_schema-0.1.1}/src/idun_agent_schema/engine/config.py

@@ -12,7 +12,7 @@ from .server import ServerConfig
 class AgentConfig(BaseModel):
     """Configuration for agent specification and settings."""
 
-    type: Literal["langgraph", "ADK", "CREWAI"] = Field(default="langgraph")
+    type: Literal["langgraph", "ADK", "CREWAI", "haystack"] = Field(default="langgraph")
     config: BaseAgentConfig | LangGraphAgentConfig = Field(
         default_factory=BaseAgentConfig
     )
@@ -23,5 +23,3 @@ class EngineConfig(BaseModel):
 
     server: ServerConfig = Field(default_factory=ServerConfig)
     agent: AgentConfig
-
-

idun_agent_schema-0.1.1/src/idun_agent_schema/engine/haystack.py

@@ -0,0 +1,13 @@
+"""Configuration models for Haystack agents."""
+
+from typing import Literal
+
+from idun_agent_engine.core.engine_config import BaseAgentConfig
+
+
+class HaystackAgentConfig(BaseAgentConfig):
+    """Configuration model for Haystack Agents."""
+
+    type: Literal["haystack"] = "haystack"
+    component_type: Literal["pipeline", "agent"]
+    component_definition: str

{idun_agent_schema-0.1.0 → idun_agent_schema-0.1.1}/src/idun_agent_schema/engine/langgraph.py

@@ -17,12 +17,19 @@ class SqliteCheckpointConfig(BaseModel):
     @field_validator("db_url")
     @classmethod
     def db_url_must_be_sqlite(cls, v: str) -> str:
+        """Ensure the provided database URL uses the sqlite scheme.
+
+        Raises:
+            ValueError: If the URL does not start with 'sqlite:///'.
+
+        """
         if not v.startswith("sqlite:///"):
             raise ValueError("SQLite DB URL must start with 'sqlite:///'")
         return v
 
     @property
     def db_path(self) -> str:
+        """Return filesystem path component derived from the sqlite URL."""
         path = urlparse(self.db_url).path
         if self.db_url.startswith("sqlite:///"):
             return path.lstrip("/")
@@ -38,5 +45,3 @@ class LangGraphAgentConfig(BaseAgentConfig):
     graph_definition: str
     checkpointer: CheckpointConfig | None = None
     store: dict[str, Any] | None = None
-
-

{idun_agent_schema-0.1.0 → idun_agent_schema-0.1.1}/src/idun_agent_schema/engine/server.py

@@ -13,5 +13,3 @@ class ServerConfig(BaseModel):
     """Configuration for the Engine's universal settings."""
 
     api: ServerAPIConfig = Field(default_factory=ServerAPIConfig)
-
-

{idun_agent_schema-0.1.0 → idun_agent_schema-0.1.1}/src/idun_agent_schema/manager/__init__.py

@@ -48,5 +48,3 @@ from .settings import (  # noqa: F401
     RedisSettings,
     Settings,
 )
-
-

{idun_agent_schema-0.1.0 → idun_agent_schema-0.1.1}/src/idun_agent_schema/manager/api.py

@@ -10,6 +10,8 @@ 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
@@ -19,6 +21,8 @@ class AgentCreateRequest(BaseModel):
 
 
 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
@@ -27,11 +31,15 @@ class AgentUpdateRequest(BaseModel):
 
 
 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
@@ -50,10 +58,14 @@ class AgentResponse(BaseModel):
     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
@@ -67,10 +79,14 @@ class AgentSummaryResponse(BaseModel):
     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
@@ -87,10 +103,14 @@ class AgentRunResponse(BaseModel):
     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
@@ -101,10 +121,14 @@ class AgentRunSummaryResponse(BaseModel):
     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
@@ -112,19 +136,23 @@ class PaginatedResponse(BaseModel):
 
 
 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-0.1.0 → idun_agent_schema-0.1.1}/src/idun_agent_schema/manager/deployments.py

@@ -10,5 +10,3 @@ class DeploymentMode(str, Enum):
     AWS = "aws"
     AZURE = "azure"
     GCP = "gcp"
-
-

{idun_agent_schema-0.1.0 → idun_agent_schema-0.1.1}/src/idun_agent_schema/manager/deps.py

@@ -12,5 +12,3 @@ class Principal(BaseModel):
     tenant_id: UUID
     roles: list[str] = []
     workspace_ids: list[UUID] = []
-
-

{idun_agent_schema-0.1.0 → idun_agent_schema-0.1.1}/src/idun_agent_schema/manager/domain.py

@@ -58,6 +58,7 @@ class AgentEntity(BaseModel):
     avg_response_time_ms: float | None = Field(None, ge=0)
 
     def activate(self) -> None:
+        """Transition agent from DRAFT to ACTIVE and set deployment time."""
         if self.status == AgentStatus.DRAFT:
             self.status = AgentStatus.ACTIVE
             self.deployed_at = datetime.now(UTC)
@@ -65,12 +66,14 @@ class AgentEntity(BaseModel):
             raise ValueError(f"Cannot activate agent in {self.status} status")
 
     def deactivate(self) -> None:
+        """Transition agent from ACTIVE to INACTIVE."""
         if self.status == AgentStatus.ACTIVE:
             self.status = AgentStatus.INACTIVE
         else:
             raise ValueError(f"Cannot deactivate agent in {self.status} status")
 
     def update_metrics(self, success: bool, response_time_ms: float) -> None:
+        """Update running success rate and average response time metrics."""
         self.total_runs += 1
 
         if self.success_rate is None:
@@ -88,6 +91,7 @@ class AgentEntity(BaseModel):
             self.avg_response_time_ms = (total_time + response_time_ms) / self.total_runs
 
     def can_be_deployed(self) -> bool:
+        """Return True if the agent is eligible for deployment."""
         return (
             self.status in [AgentStatus.DRAFT, AgentStatus.INACTIVE]
             and bool(self.name)
@@ -122,22 +126,26 @@ class AgentRunEntity(BaseModel):
     span_id: str | None = None
 
     def complete(self, output_data: dict[str, Any], response_time_ms: float) -> None:
+        """Mark run as completed with outputs and timing."""
         self.status = "completed"
         self.output_data = output_data
         self.completed_at = datetime.now(UTC)
         self.response_time_ms = response_time_ms
 
     def fail(self, error_message: str) -> None:
+        """Mark run as failed with an error message."""
         self.status = "failed"
         self.error_message = error_message
         self.completed_at = datetime.now(UTC)
 
     @property
     def is_completed(self) -> bool:
+        """Return True if this run has finished (completed or failed)."""
         return self.status in ["completed", "failed"]
 
     @property
     def was_successful(self) -> bool:
+        """Return True if the run completed successfully."""
         return self.status == "completed"
 
 
@@ -192,17 +200,21 @@ class TenantEntity(BaseModel):
     suspended_at: datetime | None = None
 
     def can_create_agent(self) -> bool:
+        """Return True if tenant has capacity to create a new agent."""
         return self.status == TenantStatus.ACTIVE and self.current_agents < self.max_agents
 
     def can_run_agent(self) -> bool:
+        """Return True if tenant is below monthly run quota."""
         return self.status == TenantStatus.ACTIVE and self.current_runs_this_month < self.max_runs_per_month
 
     def suspend(self, reason: str) -> None:
+        """Suspend tenant and record the reason."""
         self.status = TenantStatus.SUSPENDED
         self.suspended_at = datetime.utcnow()
         self.settings["suspension_reason"] = reason
 
     def reactivate(self) -> None:
+        """Reactivate a previously suspended tenant."""
         if self.status == TenantStatus.SUSPENDED:
             self.status = TenantStatus.ACTIVE
             self.suspended_at = None
@@ -210,6 +222,7 @@ class TenantEntity(BaseModel):
                 del self.settings["suspension_reason"]
 
     def upgrade_plan(self, new_plan: TenantPlan) -> None:
+        """Upgrade plan and adjust quotas accordingly."""
         self.plan = new_plan
         if new_plan == TenantPlan.STARTER:
             self.max_agents = 20
@@ -225,11 +238,13 @@ class TenantEntity(BaseModel):
             self.max_storage_mb = 100000
 
     def increment_usage(self, agents: int = 0, runs: int = 0, storage_mb: float = 0) -> None:
+        """Increment tracked usage counters."""
         self.current_agents += agents
         self.current_runs_this_month += runs
         self.current_storage_mb += storage_mb
 
     def reset_monthly_usage(self) -> None:
+        """Reset monthly run counter at billing cycle start."""
         self.current_runs_this_month = 0
 
 
@@ -253,9 +268,9 @@ class TenantUserEntity(BaseModel):
     last_active_at: datetime | None = None
 
     def has_permission(self, permission: str) -> bool:
+        """Return True if the user has an explicit permission or is owner."""
         return permission in self.permissions or self.role == "owner"
 
     def is_admin(self) -> bool:
+        """Return True if the user is an admin or owner."""
         return self.role in ["owner", "admin"]
-
-

{idun_agent_schema-0.1.0 → idun_agent_schema-0.1.1}/src/idun_agent_schema/manager/dto.py

@@ -10,6 +10,8 @@ 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
@@ -20,6 +22,8 @@ class AgentCreateDTO(BaseModel):
 
 
 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
@@ -28,6 +32,8 @@ class AgentUpdateDTO(BaseModel):
 
 
 class AgentDeploymentDTO(BaseModel):
+    """DTO describing deployment details for an agent."""
+
     agent_id: UUID
     container_id: str
     endpoint: str
@@ -37,6 +43,8 @@ class AgentDeploymentDTO(BaseModel):
 
 
 class AgentHealthDTO(BaseModel):
+    """DTO representing health metrics for an agent instance."""
+
     agent_id: UUID
     status: str
     uptime: str | None = None
@@ -47,6 +55,8 @@ class AgentHealthDTO(BaseModel):
 
 
 class AgentRunCreateDTO(BaseModel):
+    """DTO for creating a new agent run."""
+
     agent_id: UUID
     tenant_id: UUID
     input_data: dict[str, Any]
@@ -54,6 +64,8 @@ class AgentRunCreateDTO(BaseModel):
 
 
 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)
@@ -62,6 +74,8 @@ class AgentMetricsDTO(BaseModel):
 
 
 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")
@@ -70,6 +84,8 @@ class TenantCreateDTO(BaseModel):
 
 
 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
@@ -77,6 +93,8 @@ class TenantUpdateDTO(BaseModel):
 
 
 class TenantUsageDTO(BaseModel):
+    """DTO summarizing a tenant's resource usage and quotas."""
+
     tenant_id: UUID
     current_agents: int
     max_agents: int
@@ -88,12 +106,16 @@ class TenantUsageDTO(BaseModel):
 
 
 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
@@ -102,8 +124,8 @@ class TenantUserCreateDTO(BaseModel):
 
 
 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-0.1.0 → idun_agent_schema-0.1.1}/src/idun_agent_schema/manager/errors.py

@@ -20,5 +20,3 @@ class ProblemDetail(BaseModel):
     errors: dict[str, Any] | None = Field(default=None)
 
     model_config = {"extra": "allow"}
-
-

{idun_agent_schema-0.1.0 → idun_agent_schema-0.1.1}/src/idun_agent_schema/manager/settings.py

@@ -7,7 +7,11 @@ from pydantic_settings import BaseSettings, SettingsConfigDict
 
 
 class DatabaseSettings(BaseSettings):
-    url: str = Field(default="postgresql+asyncpg://postgres:postgres@localhost:55432/idun_agents")
+    """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)
@@ -17,6 +21,8 @@ class DatabaseSettings(BaseSettings):
 
 
 class RedisSettings(BaseSettings):
+    """Redis cache settings."""
+
     url: str = Field(default="redis://localhost:6379/0")
     max_connections: int = Field(default=20)
 
@@ -24,6 +30,8 @@ class RedisSettings(BaseSettings):
 
 
 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="")
@@ -62,11 +70,14 @@ class AuthSettings(BaseSettings):
             return None
         return v
 
-    model_config = SettingsConfigDict(env_prefix="AUTH_", env_file=".env", extra="ignore")
+    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"):
@@ -75,6 +86,8 @@ class AuthSettings(BaseSettings):
 
 
 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)
@@ -85,6 +98,8 @@ class ObservabilitySettings(BaseSettings):
 
 
 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")
@@ -96,23 +111,31 @@ class CelerySettings(BaseSettings):
 
 
 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_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")
+    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)
@@ -120,6 +143,7 @@ class Settings(BaseSettings):
     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)
@@ -135,5 +159,3 @@ class Settings(BaseSettings):
         case_sensitive=False,
         extra="ignore",
     )
-
-

{idun_agent_schema-0.1.0 → idun_agent_schema-0.1.1}/src/idun_agent_schema/shared/__init__.py

@@ -2,4 +2,4 @@
 
 from .observability import ObservabilityConfig
 
-__all__ = ["ObservabilityConfig"]
+__all__ = ["ObservabilityConfig"]

{idun_agent_schema-0.1.0 → idun_agent_schema-0.1.1}/src/idun_agent_schema/shared/observability.py

@@ -54,5 +54,3 @@ class ObservabilityConfig(BaseModel):
             enabled=self.enabled,
             options=resolved_options,
         )
-
-

idun_agent_schema-0.1.0/src/idun_agent_schema/engine/api.py

@@ -1,15 +0,0 @@
-"""Schemas for engine HTTP API request/response payloads."""
-
-from pydantic import BaseModel
-
-
-class ChatRequest(BaseModel):
-    session_id: str
-    query: str
-
-
-class ChatResponse(BaseModel):
-    session_id: str
-    response: str
-
-

idun_agent_schema-0.1.0/src/idun_agent_schema/py.typed

@@ -1,2 +0,0 @@
-
-