idun-agent-schema 0.2.3__py3-none-any.whl

idun_agent_schema/__init__.py

@@ -0,0 +1,9 @@
+"""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
+"""
+
+__version__ = "0.2.3"

idun_agent_schema/engine/__init__.py

@@ -0,0 +1,13 @@
+"""Engine-related schemas."""
+
+from .agent import BaseAgentConfig,AgentConfig  # noqa: F401
+from .api import ChatRequest, ChatResponse  # noqa: F401
+from .agent_framework import AgentFramework  # noqa: F401
+from .engine import EngineConfig  # noqa: F401
+from .langgraph import (  # noqa: F401
+    CheckpointConfig,
+    LangGraphAgentConfig,
+    SqliteCheckpointConfig,
+)
+from .server import ServerAPIConfig, ServerConfig  # noqa: F401
+from .observability import ObservabilityConfig  # noqa: F401

idun_agent_schema/engine/agent.py

@@ -0,0 +1,42 @@
+"""Common agent model definitions (engine)."""
+
+from typing import Any
+
+from pydantic import BaseModel, Field
+
+from idun_agent_schema.engine.agent_framework import AgentFramework
+from idun_agent_schema.engine.langgraph import LangGraphAgentConfig
+from idun_agent_schema.engine.haystack import HaystackAgentConfig
+from idun_agent_schema.engine.base_agent import BaseAgentConfig
+from pydantic import model_validator
+
+
+class AgentConfig(BaseModel):
+    """Configuration for agent specification and settings."""
+
+    type: AgentFramework
+    config: BaseAgentConfig | LangGraphAgentConfig | HaystackAgentConfig
+
+    @model_validator(mode="after")
+    def _validate_framework_config(self) -> "AgentConfig":
+        """Ensure the `config` type matches the selected framework.
+
+        - LANGGRAPH  -> LangGraphAgentConfig
+        - HAYSTACK   -> HaystackAgentConfig
+        - ADK/CREWAI/CUSTOM -> BaseAgentConfig (or subclass)
+        """
+        expected_type: type[BaseAgentConfig] | None = None
+
+        if self.type == AgentFramework.LANGGRAPH:
+            expected_type = LangGraphAgentConfig
+        elif self.type == AgentFramework.HAYSTACK:
+            expected_type = HaystackAgentConfig
+        elif self.type in {AgentFramework.ADK, AgentFramework.CREWAI, AgentFramework.CUSTOM}:
+            expected_type = BaseAgentConfig
+
+        if expected_type is not None and not isinstance(self.config, expected_type):
+            raise ValueError(
+                f"config must be {expected_type.__name__} when type is {self.type}"
+            )
+
+        return self

idun_agent_schema/engine/agent_framework.py

@@ -0,0 +1,13 @@
+"""Agent framework enumeration (engine)."""
+
+from enum import Enum
+
+
+class AgentFramework(str, Enum):
+    """Supported agent frameworks for engine."""
+
+    LANGGRAPH = "LANGGRAPH"
+    ADK = "ADK"
+    CREWAI = "CREWAI"
+    HAYSTACK = "HAYSTACK"
+    CUSTOM = "CUSTOM"

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/engine/base_agent.py

@@ -0,0 +1,15 @@
+"""Common agent model definitions (engine)."""
+
+from typing import Any
+
+from pydantic import BaseModel, Field
+
+from idun_agent_schema.engine.observability import ObservabilityConfig
+
+class BaseAgentConfig(BaseModel):
+    """Base model for agent configurations. Extend for specific frameworks."""
+
+    name: str
+    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/engine.py

@@ -0,0 +1,13 @@
+"""Main engine configuration model."""
+
+from pydantic import BaseModel, Field
+
+from .server import ServerConfig
+from .agent import AgentConfig
+
+
+class EngineConfig(BaseModel):
+    """Main engine configuration model for the entire Idun Agent Engine."""
+
+    server: ServerConfig = Field(default_factory=ServerConfig)
+    agent: AgentConfig

idun_agent_schema/engine/haystack.py

@@ -0,0 +1,13 @@
+"""Configuration models for Haystack agents."""
+
+from typing import Literal
+
+from .base_agent 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

@@ -0,0 +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 .base_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/observability.py

@@ -0,0 +1,56 @@
+"""Provider-agnostic observability configuration model (engine-scoped)."""
+
+from __future__ import annotations
+
+import os
+from typing import Any
+
+from pydantic import BaseModel, Field
+
+
+def _resolve_env(value: Any) -> Any:
+    """Resolve environment placeholders in strings.
+
+    Supports patterns ${VAR} and $VAR. Non-strings are returned unchanged.
+    """
+    if isinstance(value, str):
+        if value.startswith("${") and value.endswith("}"):
+            return os.getenv(value[2:-1])
+        if value.startswith("$"):
+            return os.getenv(value[1:])
+    return value
+
+
+class ObservabilityConfig(BaseModel):
+    """Provider-agnostic observability configuration based on Pydantic.
+
+    Example YAML:
+      observability:
+        provider: "langfuse"  # or "phoenix"
+        enabled: true
+        options:
+          host: ${LANGFUSE_HOST}
+          public_key: ${LANGFUSE_PUBLIC_KEY}
+          secret_key: ${LANGFUSE_SECRET_KEY}
+          run_name: "my-run"
+    """
+
+    provider: str | None = Field(default=None)
+    enabled: bool = Field(default=False)
+    options: dict[str, Any] = Field(default_factory=dict)
+
+    def _resolve_value(self, value: Any) -> Any:
+        if isinstance(value, dict):
+            return {k: self._resolve_value(v) for k, v in value.items()}
+        if isinstance(value, list):
+            return [self._resolve_value(v) for v in value]
+        return _resolve_env(value)
+
+    def resolved(self) -> ObservabilityConfig:
+        """Return a copy with env placeholders resolved in options."""
+        resolved_options = self._resolve_value(self.options)
+        return ObservabilityConfig(
+            provider=self.provider,
+            enabled=self.enabled,
+            options=resolved_options,
+        )

idun_agent_schema/engine/server.py

@@ -0,0 +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)

idun_agent_schema/manager/__init__.py

@@ -0,0 +1,9 @@
+"""Manager-related schemas."""
+
+from .managed_agent import (  # noqa: F401
+    ManagedAgentCreate,
+    ManagedAgentRead,
+    ManagedAgentPatch,
+    AgentStatus,
+)
+from .api import ApiKeyResponse  # noqa: F401

idun_agent_schema/manager/api.py

@@ -0,0 +1,17 @@
+"""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, field_validator
+
+class ApiKeyResponse(BaseModel):
+    """Response shape for a single agent resource."""
+
+    api_key: str
+
+    class Config:
+        """Pydantic configuration for ORM compatibility."""
+
+        from_attributes = True

idun_agent_schema/manager/managed_agent.py

@@ -0,0 +1,58 @@
+"""Main managed agent configuration model."""
+
+from datetime import datetime
+from typing import Any
+from uuid import UUID
+
+from pydantic import BaseModel, Field, field_validator, ConfigDict
+
+from idun_agent_schema.engine import EngineConfig
+from enum import Enum
+
+
+class AgentStatus(str, Enum):
+    """Agent status enumeration."""
+
+    DRAFT = "draft"
+    ACTIVE = "active"
+    INACTIVE = "inactive"
+    DEPRECATED = "deprecated"
+    ERROR = "error"
+
+# class ManagedAgentBase(BaseModel):
+#     """Base model for managed agent configuration."""
+
+#     id: UUID = Field(, description="Agent UUID")
+#     name: str
+#     status: AgentStatus = Field(AgentStatus.DRAFT, description="Agent status")
+#     version: str | None = Field(None, description="Agent version")
+#     engine_config: EngineConfig = Field(..., description="Idun Agent Engine configuration")
+#     created_at: datetime = Field(..., description="Creation timestamp")
+#     updated_at: datetime = Field(..., description="Last update timestamp")
+#     agent_hash: str | None = Field(default=None, description="Agent hash")
+
+class ManagedAgentCreate(BaseModel):
+    """Create managed agent model for requests."""
+
+    name: str
+    version: str | None = Field(None, description="Agent version")
+    engine_config: EngineConfig = Field(..., description="Idun Agent Engine configuration")
+
+
+class ManagedAgentRead(BaseModel):
+    """Complete managed agent model for responses."""
+    model_config = ConfigDict(from_attributes=True)
+
+    id: UUID
+    name: str
+    status: AgentStatus = Field(AgentStatus.DRAFT, description="Agent status")
+    version: str | None = Field(None, description="Agent version")
+    engine_config: EngineConfig = Field(..., description="Idun Agent Engine configuration")
+    created_at: datetime = Field(..., description="Creation timestamp")
+    updated_at: datetime = Field(..., description="Last update timestamp")
+
+
+class ManagedAgentPatch(BaseModel):
+    """Full replacement schema for PUT of a managed agent."""
+    name: str
+    engine_config: EngineConfig = Field(..., description="Idun Agent Engine configuration")

idun_agent_schema/shared/__init__.py

@@ -0,0 +1 @@
+"""Shared cross-cutting schemas used by both Engine and Manager."""

idun_agent_schema-0.2.3.dist-info/METADATA

@@ -0,0 +1,39 @@
+Metadata-Version: 2.4
+Name: idun-agent-schema
+Version: 0.2.3
+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
+Project-URL: Issues, https://github.com/geoffreyharrazi/idun-agent-platform/issues
+Author-email: Idun Group <contact@idun-group.com>
+License-Expression: MIT
+Keywords: fastapi,idun,langgraph,pydantic,schemas
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Typing :: Typed
+Requires-Python: <3.14,>=3.12
+Requires-Dist: pydantic-settings<3.0.0,>=2.7.0
+Requires-Dist: pydantic<3.0.0,>=2.11.7
+Description-Content-Type: text/markdown
+
+# Idun Agent Schema
+
+Centralized Pydantic schema library shared by Idun Agent Engine and Idun Agent Manager.
+
+## Install
+
+```bash
+pip install idun-agent-schema
+```
+
+## Usage
+
+```python
+from idun_agent_schema.engine import EngineConfig
+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.2.3.dist-info/RECORD

@@ -0,0 +1,19 @@
+idun_agent_schema/__init__.py,sha256=fbyxVjqxXFJSYpXiczCrqhxwhmTm6_1H3xyEMuyIVzo,261
+idun_agent_schema/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+idun_agent_schema/engine/__init__.py,sha256=jK6jfjb1SlOagLEJmvDMjGebgIcyAu-FHyOsDqRj40U,496
+idun_agent_schema/engine/agent.py,sha256=M6QHYa2yFDXgCDmKaUwZJDdCCNNSGoIow_JqF67b084,1564
+idun_agent_schema/engine/agent_framework.py,sha256=kCrOZl4jf0Om5YW4Ot32pzynatO9orVmn68O0xU8nKQ,266
+idun_agent_schema/engine/api.py,sha256=sN1QGQAuUb4znhR8QJ434sKblEJbDoXQfC4K7L9yyPQ,626
+idun_agent_schema/engine/base_agent.py,sha256=LLFarbDfBVPOztICvRtyqJE2v-H8-j2OeCiBnlqQBRY,542
+idun_agent_schema/engine/engine.py,sha256=ddfVryb5CV6hJ7K9QtGERYpwhgU6obYrs4zEzMFzwyU,339
+idun_agent_schema/engine/haystack.py,sha256=zeJIL1Hc6tUtgO146ZaScWZFPfV-b1bdGDC8BawIFaM,337
+idun_agent_schema/engine/langgraph.py,sha256=j2210QgxWv1SXaDx4gtmMgSiwQtHThUsXDEXDTifehY,1281
+idun_agent_schema/engine/observability.py,sha256=bpZupgXzczwg-fp4QANuho_8yUmvcjaeZR_m6r6lh_M,1748
+idun_agent_schema/engine/server.py,sha256=uEuTi2ow2WOzCW4okqZmgwqEBtekgeA928Yq_gc48jU,338
+idun_agent_schema/manager/__init__.py,sha256=dRhtVlGpse6kcIdSFggWQ3iBsh6Z-Q9jUVvP-XEbRQQ,209
+idun_agent_schema/manager/api.py,sha256=jS1RkUmLmNfNzCNkWyrVMW31VH8FseVe_smq6jAfz-Y,399
+idun_agent_schema/manager/managed_agent.py,sha256=0e523-MFFlqNZkHmC_zLQh8cFyK59D-mHDDykTfZIjM,2109
+idun_agent_schema/shared/__init__.py,sha256=3nb5NzRGfS3B3Q9w_scRa4rdRYY020bBI0fs8nlLOtw,68
+idun_agent_schema-0.2.3.dist-info/METADATA,sha256=CLu4iFE42MOjyVyFZcIY9g7fTcivCmKSX64LJPYqOFQ,1392
+idun_agent_schema-0.2.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+idun_agent_schema-0.2.3.dist-info/RECORD,,

idun_agent_schema-0.2.3.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.27.0
+Root-Is-Purelib: true
+Tag: py3-none-any