Source code for taimoe.platform.types.sync

"""Platform-to-runtime sync models."""

from __future__ import annotations

from datetime import datetime
from typing import Any

from pydantic import BaseModel, ConfigDict, Field




class AgentRuntimeConfig(BaseModel):
    """Resolved runtime config for a single agent.

    The platform owns composition. The SDK caches and applies this shape.
    """

    model_config = ConfigDict(frozen=True)

    runtime_agent_id: str
    version: int
    instruction: str | None = None
    model_alias: str | None = None
    generation_config: dict[str, Any] = Field(default_factory=dict)
    enable_google_search: bool = False
    knowledge_base_ids: tuple[str, ...] = ()
    tools: tuple[str, ...] = ()





class RuntimeSyncResponse(BaseModel):
    """Response returned by the platform sync endpoint."""

    model_config = ConfigDict(frozen=True)

    synced_at: datetime
    agents: tuple[AgentRuntimeConfig, ...] = ()