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

idun_agent_schema/__init__.py

@@ -1,10 +1,10 @@
-"""Idun Agent Schema - Centralized Pydantic schemas.
-
-Public namespaces:
-- idun_agent_schema.engine: Engine-related schemas
-- idun_agent_schema.manager: Manager-related schemas
-- idun_agent_schema.shared: Shared cross-cutting schemas
-"""
-
-# Re-export key types for convenience
-from .shared.observability import ObservabilityConfig  # noqa: F401
+"""Idun Agent Schema - Centralized Pydantic schemas.
+
+Public namespaces:
+- idun_agent_schema.engine: Engine-related schemas
+- idun_agent_schema.manager: Manager-related schemas
+- idun_agent_schema.shared: Shared cross-cutting schemas
+"""
+
+# Re-export key types for convenience
+from .shared.observability import ObservabilityConfig  # noqa: F401

idun_agent_schema/engine/__init__.py

@@ -1,12 +1,11 @@
-"""Engine-related schemas."""
-
-from .agent import BaseAgentConfig  # noqa: F401
-from .api import ChatRequest, ChatResponse  # noqa: F401
-from .config import AgentConfig, EngineConfig  # noqa: F401
-from .haystack import HaystackAgentConfig
-from .langgraph import (  # noqa: F401
-    CheckpointConfig,
-    LangGraphAgentConfig,
-    SqliteCheckpointConfig,
-)
-from .server import ServerAPIConfig, ServerConfig  # noqa: F401
+"""Engine-related schemas."""
+
+from .agent import BaseAgentConfig  # noqa: F401
+from .api import ChatRequest, ChatResponse  # noqa: F401
+from .config import AgentConfig, EngineConfig  # noqa: F401
+from .langgraph import (  # noqa: F401
+    CheckpointConfig,
+    LangGraphAgentConfig,
+    SqliteCheckpointConfig,
+)
+from .server import ServerAPIConfig, ServerConfig  # noqa: F401

idun_agent_schema/engine/agent.py

@@ -1,16 +1,16 @@
-"""Common agent model definitions (engine)."""
-
-from typing import Any
-
-from pydantic import BaseModel, Field
-
-from idun_agent_schema.shared import ObservabilityConfig
-
-
-class BaseAgentConfig(BaseModel):
-    """Base model for agent configurations. Extend for specific frameworks."""
-
-    name: str | None = Field(default="Unnamed Agent")
-    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)
+"""Common agent model definitions (engine)."""
+
+from typing import Any
+
+from pydantic import BaseModel, Field
+
+from idun_agent_schema.shared import ObservabilityConfig
+
+
+class BaseAgentConfig(BaseModel):
+    """Base model for agent configurations. Extend for specific frameworks."""
+
+    name: str | None = Field(default="Unnamed Agent")
+    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/engine/api.py

@@ -1,29 +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
+"""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/engine/config.py

@@ -1,6 +1,6 @@
 """Top-level engine configuration models."""
 
-from typing import Literal, Any
+from typing import Literal
 
 from pydantic import BaseModel, Field
 
@@ -15,12 +15,7 @@ class AgentConfig(BaseModel):
     """Configuration for agent specification and settings."""
 
     type: Literal["langgraph", "ADK", "CREWAI", "haystack"] = Field(default="langgraph")
-    config: Any = Field(description="Specific agent config")
-
-
-config: BaseAgentConfig | HaystackAgentConfig | LangGraphAgentConfig = Field(
-    default_factory=BaseAgentConfig
-)
+    config: BaseAgentConfig | HaystackAgentConfig | LangGraphAgentConfig
 
 
 class EngineConfig(BaseModel):

idun_agent_schema/engine/haystack.py

@@ -1,13 +1,13 @@
-"""Configuration models for Haystack agents."""
-
-from typing import Literal
-
-from .agent import BaseAgentConfig
-
-
-class HaystackAgentConfig(BaseAgentConfig):
-    """Configuration model for Haystack Agents."""
-
-    type: Literal["haystack"] = "haystack"
-    component_type: Literal["pipeline", "agent"]
-    component_definition: str
+"""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/engine/langgraph.py

@@ -1,47 +1,47 @@
-"""Configuration models for LangGraph agents (engine)."""
-
-from typing import Any, Literal
-from urllib.parse import urlparse
-
-from pydantic import BaseModel, field_validator
-
-from .agent import BaseAgentConfig
-
-
-class SqliteCheckpointConfig(BaseModel):
-    """Configuration for SQLite checkpointer."""
-
-    type: Literal["sqlite"]
-    db_url: str
-
-    @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("/")
-        return path
-
-
-CheckpointConfig = SqliteCheckpointConfig
-
-
-class LangGraphAgentConfig(BaseAgentConfig):
-    """Configuration model for LangGraph agents."""
-
-    graph_definition: str
-    checkpointer: CheckpointConfig | None = None
-    store: dict[str, Any] | None = None
+"""Configuration models for LangGraph agents (engine)."""
+
+from typing import Any, Literal
+from urllib.parse import urlparse
+
+from pydantic import BaseModel, field_validator
+
+from .agent import BaseAgentConfig
+
+
+class SqliteCheckpointConfig(BaseModel):
+    """Configuration for SQLite checkpointer."""
+
+    type: Literal["sqlite"]
+    db_url: str
+
+    @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("/")
+        return path
+
+
+CheckpointConfig = SqliteCheckpointConfig
+
+
+class LangGraphAgentConfig(BaseAgentConfig):
+    """Configuration model for LangGraph agents."""
+
+    graph_definition: str
+    checkpointer: CheckpointConfig | None = None
+    store: dict[str, Any] | None = None

idun_agent_schema/engine/server.py

@@ -1,15 +1,15 @@
-"""Server configuration models (engine)."""
-
-from pydantic import BaseModel, Field
-
-
-class ServerAPIConfig(BaseModel):
-    """API server configuration."""
-
-    port: int = 8000
-
-
-class ServerConfig(BaseModel):
-    """Configuration for the Engine's universal settings."""
-
-    api: ServerAPIConfig = Field(default_factory=ServerAPIConfig)
+"""Server configuration models (engine)."""
+
+from pydantic import BaseModel, Field
+
+
+class ServerAPIConfig(BaseModel):
+    """API server configuration."""
+
+    port: int = 8000
+
+
+class ServerConfig(BaseModel):
+    """Configuration for the Engine's universal settings."""
+
+    api: ServerAPIConfig = Field(default_factory=ServerAPIConfig)

idun_agent_schema/manager/__init__.py

@@ -1,50 +1,50 @@
-"""Manager-related schemas."""
-
-from .api import (  # noqa: F401
-    AgentCreateRequest,
-    AgentResponse,
-    AgentRunRequest,
-    AgentRunResponse,
-    AgentRunSummaryResponse,
-    AgentStatsResponse,
-    AgentSummaryResponse,
-    AgentUpdateRequest,
-    PaginatedAgentsResponse,
-    PaginatedResponse,
-    PaginatedRunsResponse,
-)
-from .deps import Principal  # noqa: F401
-from .domain import (  # noqa: F401
-    AgentEntity,
-    AgentFramework,
-    AgentRunEntity,
-    AgentStatus,
-    TenantEntity,
-    TenantPlan,
-    TenantStatus,
-    TenantUserEntity,
-)
-from .dto import (  # noqa: F401
-    AgentCreateDTO,
-    AgentDeploymentDTO,
-    AgentHealthDTO,
-    AgentMetricsDTO,
-    AgentRunCreateDTO,
-    AgentUpdateDTO,
-    TenantCreateDTO,
-    TenantQuotaDTO,
-    TenantUpdateDTO,
-    TenantUsageDTO,
-    TenantUserCreateDTO,
-    TenantUserUpdateDTO,
-)
-from .errors import ProblemDetail  # noqa: F401
-from .settings import (  # noqa: F401
-    APISettings,
-    AuthSettings,
-    CelerySettings,
-    DatabaseSettings,
-    ObservabilitySettings,
-    RedisSettings,
-    Settings,
-)
+"""Manager-related schemas."""
+
+from .api import (  # noqa: F401
+    AgentCreateRequest,
+    AgentResponse,
+    AgentRunRequest,
+    AgentRunResponse,
+    AgentRunSummaryResponse,
+    AgentStatsResponse,
+    AgentSummaryResponse,
+    AgentUpdateRequest,
+    PaginatedAgentsResponse,
+    PaginatedResponse,
+    PaginatedRunsResponse,
+)
+from .deps import Principal  # noqa: F401
+from .domain import (  # noqa: F401
+    AgentEntity,
+    AgentFramework,
+    AgentRunEntity,
+    AgentStatus,
+    TenantEntity,
+    TenantPlan,
+    TenantStatus,
+    TenantUserEntity,
+)
+from .dto import (  # noqa: F401
+    AgentCreateDTO,
+    AgentDeploymentDTO,
+    AgentHealthDTO,
+    AgentMetricsDTO,
+    AgentRunCreateDTO,
+    AgentUpdateDTO,
+    TenantCreateDTO,
+    TenantQuotaDTO,
+    TenantUpdateDTO,
+    TenantUsageDTO,
+    TenantUserCreateDTO,
+    TenantUserUpdateDTO,
+)
+from .errors import ProblemDetail  # noqa: F401
+from .settings import (  # noqa: F401
+    APISettings,
+    AuthSettings,
+    CelerySettings,
+    DatabaseSettings,
+    ObservabilitySettings,
+    RedisSettings,
+    Settings,
+)