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

idun_agent_schema/engine/__init__.py

@@ -2,8 +2,7 @@
 
 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 .config import AgentConfig, AgentFramework, EngineConfig  # noqa: F401
 from .langgraph import (  # noqa: F401
     CheckpointConfig,
     LangGraphAgentConfig,

idun_agent_schema/engine/config.py

@@ -1,25 +1,30 @@
 """Top-level engine configuration models."""
 
-from typing import Literal
-
+from enum import Enum
 from pydantic import BaseModel, Field
 
-from idun_agent_schema.engine.haystack import HaystackAgentConfig
-
 from .agent import BaseAgentConfig
 from .langgraph import LangGraphAgentConfig
 from .server import ServerConfig
 
 
-class AgentConfig(BaseModel):
-    """Configuration for agent specification and settings."""
+class AgentFramework(str, Enum):
+    """Supported agent frameworks for engine."""
 
-    type: Literal["langgraph", "ADK", "CREWAI", "haystack"] = Field(default="langgraph")
+    LANGGRAPH = "langgraph"
+    ADK = "ADK"
+    CREWAI = "CREWAI"
+    HAYSTACK = "haystack"
+    CUSTOM = "custom"
 
 
-config: BaseAgentConfig | HaystackAgentConfig | LangGraphAgentConfig = Field(
-    default_factory=BaseAgentConfig
-)
+class AgentConfig(BaseModel):
+    """Configuration for agent specification and settings."""
+
+    type: AgentFramework = Field(default=AgentFramework.LANGGRAPH)
+    config: BaseAgentConfig | LangGraphAgentConfig = Field(
+        default_factory=BaseAgentConfig
+    )
 
 
 class EngineConfig(BaseModel):

idun_agent_schema/engine/haystack.py

@@ -2,7 +2,7 @@
 
 from typing import Literal
 
-from .agent import BaseAgentConfig
+from idun_agent_engine.core.engine_config import BaseAgentConfig
 
 
 class HaystackAgentConfig(BaseAgentConfig):

idun_agent_schema/manager/__init__.py

@@ -1,14 +1,17 @@
 """Manager-related schemas."""
 
 from .api import (  # noqa: F401
-    AgentCreateRequest,
+    Agent,
+    AgentCreate,
+    AgentPatch,
+    AgentReplace,
     AgentResponse,
     AgentRunRequest,
     AgentRunResponse,
     AgentRunSummaryResponse,
     AgentStatsResponse,
     AgentSummaryResponse,
-    AgentUpdateRequest,
+    AgentUpdate,
     PaginatedAgentsResponse,
     PaginatedResponse,
     PaginatedRunsResponse,

idun_agent_schema/manager/api.py

@@ -4,30 +4,12 @@ from datetime import datetime
 from typing import Any
 from uuid import UUID
 
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, Field, field_validator
 
+from ..engine.config import AgentConfig
 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):
@@ -156,3 +138,149 @@ class AgentStatsResponse(BaseModel):
     total_runs_this_month: int
     avg_success_rate: float | None
     avg_response_time_ms: float | None
+
+
+class AgentCreate(BaseModel):
+    """Schema for creating a new agent.
+
+    Framework is inferred from config.type (e.g., config.type = "langgraph").
+    """
+
+    name: str = Field(..., min_length=1, max_length=100, description="Agent name")
+    description: str | None = Field(
+        None, max_length=500, description="Agent description"
+    )
+    # Use schema's AgentConfig instead of AgentPayload
+    config: AgentConfig = Field(
+        ..., description="Framework-specific agent configuration"
+    )
+
+    @field_validator("name")  # noqa: N805 - Pydantic validator uses `cls` by convention
+    @classmethod
+    def validate_name(cls, v):
+        if not v.strip():
+            raise ValueError("Name cannot be empty or whitespace only")
+        return v.strip()
+
+    class Config:
+        json_schema_extra = {
+            "example": {
+                "name": "My AI Assistant",
+                "description": "A helpful AI assistant for customer support",
+                "config": {
+                    "type": "langgraph",
+                    "config": {
+                        "name": "My AI Assistant",
+                        "graph_definition": "./examples/01_basic_config_file/example_agent.py:app",
+                        "checkpointer": None,
+                        "store": None,
+                        "input_schema_definition": {},
+                        "output_schema_definition": {},
+                        "observability": None,
+                    },
+                },
+            }
+        }
+
+
+class AgentUpdate(BaseModel):
+    """Schema for updating an existing agent."""
+
+    name: str | None = Field(
+        None, min_length=1, max_length=100, description="Agent name"
+    )
+    description: str | None = Field(
+        None, max_length=500, description="Agent description"
+    )
+    framework: AgentFramework | None = Field(None, description="Agent framework")
+
+    @field_validator("name")  # noqa: N805
+    @classmethod
+    def validate_name(cls, v):
+        if v is not None and not v.strip():
+            raise ValueError("Name cannot be empty or whitespace only")
+        return v.strip() if v else v
+
+    class Config:
+        json_schema_extra = {
+            "example": {
+                "name": "Updated Agent Name",
+                "description": "Updated description",
+                "framework": "langchain",
+            }
+        }
+
+
+class Agent(BaseModel):
+    """Complete agent model for responses."""
+
+    id: str = Field(..., description="Unique agent identifier")
+    name: str = Field(..., description="Agent name")
+    description: str | None = Field(None, description="Agent description")
+    framework: AgentFramework = Field(..., description="Agent framework")
+    status: AgentStatus = Field(AgentStatus.DRAFT, description="Agent status")
+    created_at: datetime = Field(..., description="Creation timestamp")
+    updated_at: datetime = Field(..., description="Last update timestamp")
+
+    class Config:
+        json_schema_extra = {
+            "example": {
+                "id": "550e8400-e29b-41d4-a716-446655440000",
+                "name": "My AI Assistant",
+                "description": "A helpful AI assistant",
+                "framework": "langgraph",
+                "status": "draft",
+                "created_at": "2024-01-01T00:00:00",
+                "updated_at": "2024-01-01T00:00:00",
+            }
+        }
+
+
+class AgentReplace(BaseModel):
+    """Full replacement schema for PUT of an agent.
+
+    Represents the complete updatable representation of an agent.
+    Server-managed fields like id, status, and timestamps are excluded.
+    """
+
+    name: str = Field(..., min_length=1, max_length=100, description="Agent name")
+    description: str | None = Field(
+        None, max_length=500, description="Agent description"
+    )
+    config: AgentConfig = Field(
+        ..., description="Framework-specific agent configuration"
+    )
+
+    @field_validator("name")  # noqa: N805
+    @classmethod
+    def validate_name(cls, v):
+        if not v.strip():
+            raise ValueError("Name cannot be empty or whitespace only")
+        return v.strip()
+
+
+class AgentPatch(BaseModel):
+    """Partial update schema for PATCH of an agent.
+
+    Only provided fields will be updated. If config is provided, the
+    framework will be inferred from config.type.
+    """
+
+    name: str | None = Field(
+        None, min_length=1, max_length=100, description="Agent name"
+    )
+    description: str | None = Field(
+        None, max_length=500, description="Agent description"
+    )
+    config: AgentConfig | None = Field(
+        None, description="Framework-specific agent configuration"
+    )
+
+    @field_validator("name")  # noqa: N805
+    @classmethod
+    def validate_name(cls, v):
+        if v is not None and not v.strip():
+            raise ValueError("Name cannot be empty or whitespace only")
+        return v.strip() if v else v
+
+

{idun_agent_schema-0.1.41.dist-info → idun_agent_schema-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: idun-agent-schema
-Version: 0.1.41
+Version: 0.2.0
 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

{idun_agent_schema-0.1.41.dist-info → idun_agent_schema-0.2.0.dist-info}/RECORD

@@ -1,14 +1,14 @@
 idun_agent_schema/__init__.py,sha256=eRy0qwee7_nRSubm2X6jUnZyf55Dn9qztWtDocgIIJg,345
 idun_agent_schema/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-idun_agent_schema/engine/__init__.py,sha256=48Ow-dWqLI0DlJnZhCrs0z3X1AVxlHk4O0y0K9XJ57s,420
+idun_agent_schema/engine/__init__.py,sha256=pLcOjL793n7E5Nf63L5NubA9Flfi8mcKoDzZ2DCKav8,394
 idun_agent_schema/engine/agent.py,sha256=KNzNMJcqjlczQL23Uqdkpcf2-b37p8vHkI5EZIdTHmE,569
 idun_agent_schema/engine/api.py,sha256=sN1QGQAuUb4znhR8QJ434sKblEJbDoXQfC4K7L9yyPQ,626
-idun_agent_schema/engine/config.py,sha256=bhROHb4rms97F2Ge8rRNaC7R-75LklSo2PIj787Y-Nc,790
-idun_agent_schema/engine/haystack.py,sha256=knpJTW4Er2JxRfuqGgDEjcxfHzaQrNXgLfLvsXeXo_s,332
+idun_agent_schema/engine/config.py,sha256=XAwvz-c2Vilae8_uX86RcBNvP5pbut-cpYoQhGpu8d8,882
+idun_agent_schema/engine/haystack.py,sha256=EtOYnsWRufcrQufTRMeB3V-rZVQqfnmwKwPsYGfZdCs,362
 idun_agent_schema/engine/langgraph.py,sha256=m6_xvMaCgkO1O3pf2M6NaBz3ASkjn8XGFeEcXvIr4sQ,1276
 idun_agent_schema/engine/server.py,sha256=uEuTi2ow2WOzCW4okqZmgwqEBtekgeA928Yq_gc48jU,338
-idun_agent_schema/manager/__init__.py,sha256=cGm8vg2CjGYsAyG4mM73JfWfg16u61YSGvcQVnp-XMA,1083
-idun_agent_schema/manager/api.py,sha256=wSGhkm_jkLwTiiPTMCambiBrKqR0AqlRVRkbfcmCnP8,3879
+idun_agent_schema/manager/__init__.py,sha256=XwkaFY1EtMSuD3A0titNvFX5ZM2Hdyu8Ms4JSGUQhag,1114
+idun_agent_schema/manager/api.py,sha256=mieoHzwOjqCRpDyRj7fuFX0rS0hp5-zjbKiFZdrIa6k,8078
 idun_agent_schema/manager/deployments.py,sha256=qpS0cTm6H8eWAfFFJ3w3jXBFVVal8dfVB79yBnLq9ow,213
 idun_agent_schema/manager/deps.py,sha256=Z3Kgq0CkCSmeInjROVebOrok3LsMzcs0LO8KyxiCsrM,315
 idun_agent_schema/manager/domain.py,sha256=MDXsFEN_jiCAMr1J3vAoZD_I1tIq6WZjJwvHcin0yvs,8840
@@ -17,6 +17,6 @@ idun_agent_schema/manager/errors.py,sha256=wPb6DHMd7mU9MnGBZw6U51JxJfKxuVKEHSbS1
 idun_agent_schema/manager/settings.py,sha256=Z3j18AayZxjhVy2JwbqKOgVA-GGBiGUsUj6Q0e9jCN0,5791
 idun_agent_schema/shared/__init__.py,sha256=C0g2hEWT5EwYdLya7oHmXdMTFOGHVINIxeFV2j2MpXw,151
 idun_agent_schema/shared/observability.py,sha256=EL0h0qSJ7gByMCASYkQRlPv5szffJ6aYY6699TZ6uRM,1741
-idun_agent_schema-0.1.41.dist-info/METADATA,sha256=k6PUhqp9HN_YfVDly-LEVde4zMvJTGQb9fmr8EZI0QI,1332
-idun_agent_schema-0.1.41.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-idun_agent_schema-0.1.41.dist-info/RECORD,,
+idun_agent_schema-0.2.0.dist-info/METADATA,sha256=Aljj6pUqkX1ix-O16GTRRz6XNzB8DWz579Ok8pKseJ0,1331
+idun_agent_schema-0.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+idun_agent_schema-0.2.0.dist-info/RECORD,,