plato-sdk-v2 2.7.3__py3-none-any.whl → 2.7.5__py3-none-any.whl

plato/_generated/models/__init__.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
-#   filename:  tmph5p8qjzo.json
-#   timestamp: 2026-01-28T09:03:29+00:00
+#   filename:  tmpio7emwyr.json
+#   timestamp: 2026-01-29T20:21:27+00:00
 
 from __future__ import annotations
 
@@ -36,18 +36,6 @@ class Outcome(Enum):
     reject = "reject"
 
 
-class AddReviewRequest(BaseModel):
-    model_config = ConfigDict(
-        extra="allow",
-    )
-    review_type: Annotated[ReviewType, Field(title="Review Type")]
-    outcome: Annotated[Outcome, Field(title="Outcome")]
-    artifact_id: Annotated[str, Field(title="Artifact Id")]
-    video_s3_path: Annotated[str | None, Field(title="Video S3 Path")] = None
-    events_s3_path: Annotated[str | None, Field(title="Events S3 Path")] = None
-    comments: Annotated[str | None, Field(title="Comments")] = None
-
-
 class AddSSHKeyRequest(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -458,6 +446,57 @@ class BulkArchiveTestCasesRequest(BaseModel):
     test_case_ids: Annotated[list[int], Field(title="Test Case Ids")]
 
 
+class CacheClearAllResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    caches_cleared: Annotated[dict[str, int], Field(title="Caches Cleared")]
+    total_cleared: Annotated[int, Field(title="Total Cleared")]
+
+
+class CacheClearResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    cache_name: Annotated[str, Field(title="Cache Name")]
+    entries_cleared: Annotated[int, Field(title="Entries Cleared")]
+
+
+class CacheKeyDeleteResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    cache_name: Annotated[str, Field(title="Cache Name")]
+    key: Annotated[str, Field(title="Key")]
+    deleted: Annotated[bool, Field(title="Deleted")]
+
+
+class CacheName(Enum):
+    artifact = "artifact"
+    artifact_tag = "artifact_tag"
+    simulator = "simulator"
+    handler = "handler"
+    service_type = "service_type"
+    organization = "organization"
+    api_key = "api_key"
+    access = "access"
+
+
+class CacheStats(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    size: Annotated[int, Field(title="Size")]
+    ttl_seconds: Annotated[int, Field(title="Ttl Seconds")]
+
+
+class CacheStatsResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    caches: Annotated[dict[str, CacheStats], Field(title="Caches")]
+
+
 class Capability(Enum):
     forms_and_inputs = "forms_and_inputs"
     search_and_extract = "search_and_extract"
@@ -630,13 +669,6 @@ class CloseSessionResponse(BaseModel):
     """
 
 
-class ConnectNetworkRequest(BaseModel):
-    model_config = ConfigDict(
-        extra="allow",
-    )
-    host_only: Annotated[bool | None, Field(title="Host Only")] = False
-
-
 class ConnectRoutingInfoResult(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -855,7 +887,6 @@ class CreateNetworkResponse(BaseModel):
     )
     session_id: Annotated[str, Field(title="Session Id")]
     name: Annotated[str | None, Field(title="Name")] = None
-    subnet: Annotated[str, Field(title="Subnet")]
     dns_suffix: Annotated[str, Field(title="Dns Suffix")]
     status: Annotated[str, Field(title="Status")]
 
@@ -1774,7 +1805,6 @@ class NetworkResponse(BaseModel):
     )
     session_id: Annotated[str, Field(title="Session Id")]
     name: Annotated[str | None, Field(title="Name")] = None
-    subnet: Annotated[str, Field(title="Subnet")]
     dns_suffix: Annotated[str, Field(title="Dns Suffix")]
     status: Annotated[str, Field(title="Status")]
     member_count: Annotated[int, Field(title="Member Count")]
@@ -1885,6 +1915,18 @@ class NodeSessionNetworkConfig(BaseModel):
     """
     Enable session networking (WireGuard-based VM-to-VM communication).
     """
+    gateway_public_key: Annotated[str | None, Field(title="Gateway Public Key")] = None
+    """
+    WireGuard public key of the session network gateway.
+    """
+    gateway_endpoint: Annotated[str | None, Field(title="Gateway Endpoint")] = None
+    """
+    Gateway WireGuard endpoint (ip:port), e.g., '10.0.0.1:51820'.
+    """
+    gateway_ip: Annotated[str | None, Field(title="Gateway Ip")] = "10.255.0.1"
+    """
+    Gateway's WireGuard IP address.
+    """
 
 
 class NodeSnapshotStoreConfig(BaseModel):
@@ -2700,18 +2742,12 @@ class SimResetLogInput(BaseModel):
     """
 
 
-class SimReview(BaseModel):
+class SimReviewComment(BaseModel):
     model_config = ConfigDict(
         extra="allow",
     )
-    timestamp_iso: Annotated[str, Field(title="Timestamp Iso")]
-    review_type: Annotated[ReviewType, Field(title="Review Type")]
-    outcome: Annotated[Outcome, Field(title="Outcome")]
-    artifact_id: Annotated[str, Field(title="Artifact Id")]
-    video_s3_path: Annotated[str | None, Field(title="Video S3 Path")] = None
-    events_s3_path: Annotated[str | None, Field(title="Events S3 Path")] = None
-    comments: Annotated[str | None, Field(title="Comments")] = None
-    reviewer_user_id: Annotated[int | None, Field(title="Reviewer User Id")] = None
+    comment: Annotated[str, Field(title="Comment")]
+    s3_path: Annotated[str, Field(title="S3 Path")]
 
 
 class SimStatusHistory(BaseModel):
@@ -2760,27 +2796,6 @@ class Status1(Enum):
     out_of_service = "out_of_service"
 
 
-class SimulatorConfig(BaseModel):
-    model_config = ConfigDict(
-        extra="allow",
-    )
-    type: Annotated[Type8, Field(title="Type")]
-    cookies: Annotated[list[ChromeCookie] | None, Field(title="Cookies")] = None
-    authentication: Authentication | None = None
-    default_start_path: Annotated[str | None, Field(title="Default Start Path")] = "/"
-    status: Annotated[Status1 | None, Field(title="Status")] = Status1.not_started
-    env_assignees: Annotated[list[int] | None, Field(title="Env Assignees")] = None
-    env_review_assignees: Annotated[list[int] | None, Field(title="Env Review Assignees")] = None
-    data_assignees: Annotated[list[int] | None, Field(title="Data Assignees")] = None
-    data_review_assignees: Annotated[list[int] | None, Field(title="Data Review Assignees")] = None
-    status_history: Annotated[list[SimStatusHistory] | None, Field(title="Status History")] = None
-    reviews: Annotated[list[SimReview] | None, Field(title="Reviews")] = None
-    assigned_user_id: Annotated[int | None, Field(title="Assigned User Id")] = None
-    base_artifact_id: Annotated[str | None, Field(title="Base Artifact Id")] = None
-    data_artifact_id: Annotated[str | None, Field(title="Data Artifact Id")] = None
-    notes: Annotated[str | None, Field(title="Notes")] = None
-
-
 class SimulatorIdResponse(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -2804,16 +2819,6 @@ class SimulatorListItem(BaseModel):
     image_uri: Annotated[str | None, Field(title="Image Uri")] = None
 
 
-class SimulatorMetadata(BaseModel):
-    model_config = ConfigDict(
-        extra="allow",
-    )
-    id: Annotated[int, Field(title="Id")]
-    name: Annotated[str, Field(title="Name")]
-    imgUrl: Annotated[str | None, Field(title="Imgurl")] = None
-    config: SimulatorConfig | None = None
-
-
 class SimulatorRepositoryResponse(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -3673,23 +3678,6 @@ class AppApiV1EnvRoutesBatchLogRequest(BaseModel):
     count: Annotated[int, Field(title="Count")]
 
 
-class AppApiV1EnvRoutesUpdateSimulatorRequest(BaseModel):
-    model_config = ConfigDict(
-        extra="allow",
-    )
-    name: Annotated[str | None, Field(title="Name")] = None
-    url: Annotated[str | None, Field(title="Url")] = None
-    description: Annotated[str | None, Field(title="Description")] = None
-    imgUrl: Annotated[str | None, Field(title="Imgurl")] = None
-    config: SimulatorConfig | None = None
-    ancestors: Annotated[list[str] | None, Field(title="Ancestors")] = None
-    enabled: Annotated[bool | None, Field(title="Enabled")] = None
-    simType: Annotated[str | None, Field(title="Simtype")] = None
-    jobName: Annotated[str | None, Field(title="Jobname")] = None
-    internalAppPort: Annotated[int | None, Field(title="Internalappport")] = None
-    supportedProviders: Annotated[list[str] | None, Field(title="Supportedproviders")] = None
-
-
 class AppApiV1PublicBuildRoutesCreateSnapshotRequest(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -4150,6 +4138,33 @@ class ModelsScoringConfigJSONSchemaDraft7(BaseModel):
     """
 
 
+class AddJobResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    session_id: Annotated[str, Field(title="Session Id")]
+    """
+    Session ID the job was added to
+    """
+    env: EnvInfo
+    """
+    Created environment info
+    """
+
+
+class AddReviewRequest(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    review_type: Annotated[ReviewType, Field(title="Review Type")]
+    outcome: Annotated[Outcome, Field(title="Outcome")]
+    artifact_id: Annotated[str, Field(title="Artifact Id")]
+    video_s3_path: Annotated[str | None, Field(title="Video S3 Path")] = None
+    events_s3_path: Annotated[str | None, Field(title="Events S3 Path")] = None
+    comments: Annotated[str | None, Field(title="Comments")] = None
+    sim_comments: Annotated[list[SimReviewComment] | None, Field(title="Sim Comments")] = None
+
+
 class AddSSHKeyResponse(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -4168,18 +4183,6 @@ class AddSSHKeyResponse(BaseModel):
     """
 
 
-class AnnotatorAssignmentListItem(BaseModel):
-    model_config = ConfigDict(
-        extra="allow",
-    )
-    assignment_id: Annotated[int, Field(title="Assignment Id")]
-    type: AssignmentType
-    stage: AssignmentStage
-    simulator: SimulatorMetadata | None = None
-    createdAt: Annotated[AwareDatetime, Field(title="Createdat")]
-    test_case: TestCaseMetadata | None = None
-
-
 class AutoVerifyRequest(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -4293,23 +4296,6 @@ class CreateSessionResponse(BaseModel):
     """
 
 
-class CreateSimulatorRequest(BaseModel):
-    model_config = ConfigDict(
-        extra="allow",
-    )
-    name: Annotated[str, Field(title="Name")]
-    url: Annotated[str | None, Field(title="Url")] = None
-    description: Annotated[str | None, Field(title="Description")] = None
-    imgUrl: Annotated[str | None, Field(title="Imgurl")] = None
-    config: SimulatorConfig
-    ancestors: Annotated[list[str] | None, Field(title="Ancestors")] = None
-    enabled: Annotated[bool | None, Field(title="Enabled")] = False
-    simType: Annotated[str, Field(title="Simtype")]
-    jobName: Annotated[str | None, Field(title="Jobname")] = None
-    internalAppPort: Annotated[int | None, Field(title="Internalappport")] = None
-    supportedProviders: Annotated[list[str] | None, Field(title="Supportedproviders")] = None
-
-
 class DispatchersResponse(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -4618,6 +4604,42 @@ class SetDateResponse(BaseModel):
     results: Annotated[dict[str, SetDateResult], Field(title="Results")]
 
 
+class SimReview(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    timestamp_iso: Annotated[str, Field(title="Timestamp Iso")]
+    review_type: Annotated[ReviewType, Field(title="Review Type")]
+    outcome: Annotated[Outcome, Field(title="Outcome")]
+    artifact_id: Annotated[str, Field(title="Artifact Id")]
+    reviewer_user_id: Annotated[int | None, Field(title="Reviewer User Id")] = None
+    sim_comments: Annotated[list[SimReviewComment] | None, Field(title="Sim Comments")] = None
+
+
+class SimulatorConfigInput(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    type: Annotated[Type8, Field(title="Type")]
+    cookies: Annotated[list[ChromeCookie] | None, Field(title="Cookies")] = None
+    authentication: Authentication | None = None
+    default_start_path: Annotated[str | None, Field(title="Default Start Path")] = "/"
+    status: Annotated[Status1 | None, Field(title="Status")] = Status1.not_started
+    env_assignees: Annotated[list[int] | None, Field(title="Env Assignees")] = None
+    env_review_assignees: Annotated[list[int] | None, Field(title="Env Review Assignees")] = None
+    data_assignees: Annotated[list[int] | None, Field(title="Data Assignees")] = None
+    data_review_assignees: Annotated[list[int] | None, Field(title="Data Review Assignees")] = None
+    status_history: Annotated[list[SimStatusHistory] | None, Field(title="Status History")] = None
+    reviews: Annotated[list[SimReview] | None, Field(title="Reviews")] = None
+    assigned_user_id: Annotated[int | None, Field(title="Assigned User Id")] = None
+    base_artifact_id: Annotated[str | None, Field(title="Base Artifact Id")] = None
+    data_artifact_id: Annotated[str | None, Field(title="Data Artifact Id")] = None
+    notes: Annotated[str | None, Field(title="Notes")] = None
+
+
+SimulatorConfigOutput = SimulatorConfigInput
+
+
 class SimulatorInfoResponse(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -4627,6 +4649,16 @@ class SimulatorInfoResponse(BaseModel):
     tags: Annotated[list[SimulatorTagInfo], Field(title="Tags")]
 
 
+class SimulatorMetadata(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    id: Annotated[int, Field(title="Id")]
+    name: Annotated[str, Field(title="Name")]
+    imgUrl: Annotated[str | None, Field(title="Imgurl")] = None
+    config: SimulatorConfigOutput | None = None
+
+
 class SnapshotStatusResponse(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -4715,6 +4747,23 @@ class WaitForReadyResponse(BaseModel):
     """
 
 
+class AppApiV1EnvRoutesUpdateSimulatorRequest(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    name: Annotated[str | None, Field(title="Name")] = None
+    url: Annotated[str | None, Field(title="Url")] = None
+    description: Annotated[str | None, Field(title="Description")] = None
+    imgUrl: Annotated[str | None, Field(title="Imgurl")] = None
+    config: SimulatorConfigInput | None = None
+    ancestors: Annotated[list[str] | None, Field(title="Ancestors")] = None
+    enabled: Annotated[bool | None, Field(title="Enabled")] = None
+    simType: Annotated[str | None, Field(title="Simtype")] = None
+    jobName: Annotated[str | None, Field(title="Jobname")] = None
+    internalAppPort: Annotated[int | None, Field(title="Internalappport")] = None
+    supportedProviders: Annotated[list[str] | None, Field(title="Supportedproviders")] = None
+
+
 class AppApiV2SchemasArtifactSimConfigDataset(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -4766,6 +4815,39 @@ class AppSchemasBuildModelsSimConfigDataset(BaseModel):
     """
 
 
+class AddJobRequest(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    env: Annotated[
+        EnvFromSimulator | EnvFromArtifact | EnvFromResource,
+        Field(discriminator="type", title="Env"),
+    ]
+    """
+    Environment configuration for the new job
+    """
+    timeout: Annotated[int | None, Field(title="Timeout")] = 1800
+    """
+    VM timeout in seconds (default: 30 minutes)
+    """
+    heartbeat_timeout: Annotated[int | None, Field(title="Heartbeat Timeout")] = None
+    """
+    Per-VM heartbeat timeout. None=use default (300s), 0=disabled.
+    """
+
+
+class AnnotatorAssignmentListItem(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    assignment_id: Annotated[int, Field(title="Assignment Id")]
+    type: AssignmentType
+    stage: AssignmentStage
+    simulator: SimulatorMetadata | None = None
+    createdAt: Annotated[AwareDatetime, Field(title="Createdat")]
+    test_case: TestCaseMetadata | None = None
+
+
 class Envs(RootModel[EnvFromSimulator | EnvFromArtifact | EnvFromResource]):
     root: Annotated[
         EnvFromSimulator | EnvFromArtifact | EnvFromResource,
@@ -4799,6 +4881,23 @@ class CreateSessionFromEnvs(BaseModel):
     """
 
 
+class CreateSimulatorRequest(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    name: Annotated[str, Field(title="Name")]
+    url: Annotated[str | None, Field(title="Url")] = None
+    description: Annotated[str | None, Field(title="Description")] = None
+    imgUrl: Annotated[str | None, Field(title="Imgurl")] = None
+    config: SimulatorConfigInput
+    ancestors: Annotated[list[str] | None, Field(title="Ancestors")] = None
+    enabled: Annotated[bool | None, Field(title="Enabled")] = False
+    simType: Annotated[str, Field(title="Simtype")]
+    jobName: Annotated[str | None, Field(title="Jobname")] = None
+    internalAppPort: Annotated[int | None, Field(title="Internalappport")] = None
+    supportedProviders: Annotated[list[str] | None, Field(title="Supportedproviders")] = None
+
+
 class CreateVMRequest(BaseModel):
     model_config = ConfigDict(
         extra="allow",

plato/chronos/api/agents/get_agent_schema.py

@@ -7,7 +7,7 @@ from typing import Any
 import httpx
 
 from plato.chronos.errors import raise_for_status
-from plato.chronos.models import ChronosModelsAgentAgentSchemaResponse
+from plato.chronos.models import AgentSchemaResponse
 
 
 def _build_request_args(
@@ -39,7 +39,7 @@ def sync(
     name: str,
     version: str | None = None,
     x_api_key: str | None = None,
-) -> ChronosModelsAgentAgentSchemaResponse:
+) -> AgentSchemaResponse:
     """Get agent schema by name and optional version."""
 
     request_args = _build_request_args(
@@ -50,7 +50,7 @@ def sync(
 
     response = client.request(**request_args)
     raise_for_status(response)
-    return ChronosModelsAgentAgentSchemaResponse.model_validate(response.json())
+    return AgentSchemaResponse.model_validate(response.json())
 
 
 async def asyncio(
@@ -58,7 +58,7 @@ async def asyncio(
     name: str,
     version: str | None = None,
     x_api_key: str | None = None,
-) -> ChronosModelsAgentAgentSchemaResponse:
+) -> AgentSchemaResponse:
     """Get agent schema by name and optional version."""
 
     request_args = _build_request_args(
@@ -69,4 +69,4 @@ async def asyncio(
 
     response = await client.request(**request_args)
     raise_for_status(response)
-    return ChronosModelsAgentAgentSchemaResponse.model_validate(response.json())
+    return AgentSchemaResponse.model_validate(response.json())

plato/chronos/api/agents/get_agent_versions.py

@@ -7,7 +7,7 @@ from typing import Any
 import httpx
 
 from plato.chronos.errors import raise_for_status
-from plato.chronos.models import ChronosModelsAgentAgentVersionsResponse
+from plato.chronos.models import AgentVersionsResponse
 
 
 def _build_request_args(
@@ -32,7 +32,7 @@ def sync(
     client: httpx.Client,
     name: str,
     x_api_key: str | None = None,
-) -> ChronosModelsAgentAgentVersionsResponse:
+) -> AgentVersionsResponse:
     """Get all versions of an agent by name."""
 
     request_args = _build_request_args(
@@ -42,14 +42,14 @@ def sync(
 
     response = client.request(**request_args)
     raise_for_status(response)
-    return ChronosModelsAgentAgentVersionsResponse.model_validate(response.json())
+    return AgentVersionsResponse.model_validate(response.json())
 
 
 async def asyncio(
     client: httpx.AsyncClient,
     name: str,
     x_api_key: str | None = None,
-) -> ChronosModelsAgentAgentVersionsResponse:
+) -> AgentVersionsResponse:
     """Get all versions of an agent by name."""
 
     request_args = _build_request_args(
@@ -59,4 +59,4 @@ async def asyncio(
 
     response = await client.request(**request_args)
     raise_for_status(response)
-    return ChronosModelsAgentAgentVersionsResponse.model_validate(response.json())
+    return AgentVersionsResponse.model_validate(response.json())

plato/chronos/api/agents/list_agents.py

@@ -7,7 +7,7 @@ from typing import Any
 import httpx
 
 from plato.chronos.errors import raise_for_status
-from plato.chronos.models import ChronosModelsAgentAgentListResponse
+from plato.chronos.models import AgentListResponse
 
 
 def _build_request_args(
@@ -30,7 +30,7 @@ def _build_request_args(
 def sync(
     client: httpx.Client,
     x_api_key: str | None = None,
-) -> ChronosModelsAgentAgentListResponse:
+) -> AgentListResponse:
     """List all agents for the org."""
 
     request_args = _build_request_args(
@@ -39,13 +39,13 @@ def sync(
 
     response = client.request(**request_args)
     raise_for_status(response)
-    return ChronosModelsAgentAgentListResponse.model_validate(response.json())
+    return AgentListResponse.model_validate(response.json())
 
 
 async def asyncio(
     client: httpx.AsyncClient,
     x_api_key: str | None = None,
-) -> ChronosModelsAgentAgentListResponse:
+) -> AgentListResponse:
     """List all agents for the org."""
 
     request_args = _build_request_args(
@@ -54,4 +54,4 @@ async def asyncio(
 
     response = await client.request(**request_args)
     raise_for_status(response)
-    return ChronosModelsAgentAgentListResponse.model_validate(response.json())
+    return AgentListResponse.model_validate(response.json())

plato/chronos/api/registry/get_agent_schema_api_registry_agents__agent_name__schema_get.py

@@ -7,7 +7,7 @@ from typing import Any
 import httpx
 
 from plato.chronos.errors import raise_for_status
-from plato.chronos.models import AgentSchemaResponse
+from plato.chronos.models import ChronosApiRegistryAgentSchemaResponse
 
 
 def _build_request_args(
@@ -32,7 +32,7 @@ def sync(
     client: httpx.Client,
     agent_name: str,
     version: str | None = None,
-) -> AgentSchemaResponse:
+) -> ChronosApiRegistryAgentSchemaResponse:
     """Get schema for an agent version from the registry."""
 
     request_args = _build_request_args(
@@ -42,14 +42,14 @@ def sync(
 
     response = client.request(**request_args)
     raise_for_status(response)
-    return AgentSchemaResponse.model_validate(response.json())
+    return ChronosApiRegistryAgentSchemaResponse.model_validate(response.json())
 
 
 async def asyncio(
     client: httpx.AsyncClient,
     agent_name: str,
     version: str | None = None,
-) -> AgentSchemaResponse:
+) -> ChronosApiRegistryAgentSchemaResponse:
     """Get schema for an agent version from the registry."""
 
     request_args = _build_request_args(
@@ -59,4 +59,4 @@ async def asyncio(
 
     response = await client.request(**request_args)
     raise_for_status(response)
-    return AgentSchemaResponse.model_validate(response.json())
+    return ChronosApiRegistryAgentSchemaResponse.model_validate(response.json())

plato/chronos/api/registry/get_agent_versions_api_registry_agents__agent_name__versions_get.py

@@ -7,7 +7,7 @@ from typing import Any
 import httpx
 
 from plato.chronos.errors import raise_for_status
-from plato.chronos.models import AgentVersionsResponse
+from plato.chronos.models import ChronosApiRegistryAgentVersionsResponse
 
 
 def _build_request_args(
@@ -25,7 +25,7 @@ def _build_request_args(
 def sync(
     client: httpx.Client,
     agent_name: str,
-) -> AgentVersionsResponse:
+) -> ChronosApiRegistryAgentVersionsResponse:
     """Get versions for an agent from the registry."""
 
     request_args = _build_request_args(
@@ -34,13 +34,13 @@ def sync(
 
     response = client.request(**request_args)
     raise_for_status(response)
-    return AgentVersionsResponse.model_validate(response.json())
+    return ChronosApiRegistryAgentVersionsResponse.model_validate(response.json())
 
 
 async def asyncio(
     client: httpx.AsyncClient,
     agent_name: str,
-) -> AgentVersionsResponse:
+) -> ChronosApiRegistryAgentVersionsResponse:
     """Get versions for an agent from the registry."""
 
     request_args = _build_request_args(
@@ -49,4 +49,4 @@ async def asyncio(
 
     response = await client.request(**request_args)
     raise_for_status(response)
-    return AgentVersionsResponse.model_validate(response.json())
+    return ChronosApiRegistryAgentVersionsResponse.model_validate(response.json())

plato/chronos/api/registry/list_registry_agents_api_registry_agents_get.py

@@ -7,7 +7,7 @@ from typing import Any
 import httpx
 
 from plato.chronos.errors import raise_for_status
-from plato.chronos.models import AgentListResponse
+from plato.chronos.models import ChronosApiRegistryAgentListResponse
 
 
 def _build_request_args() -> dict[str, Any]:
@@ -22,23 +22,23 @@ def _build_request_args() -> dict[str, Any]:
 
 def sync(
     client: httpx.Client,
-) -> AgentListResponse:
+) -> ChronosApiRegistryAgentListResponse:
     """List all agents from the registry with their ECR image URIs."""
 
     request_args = _build_request_args()
 
     response = client.request(**request_args)
     raise_for_status(response)
-    return AgentListResponse.model_validate(response.json())
+    return ChronosApiRegistryAgentListResponse.model_validate(response.json())
 
 
 async def asyncio(
     client: httpx.AsyncClient,
-) -> AgentListResponse:
+) -> ChronosApiRegistryAgentListResponse:
     """List all agents from the registry with their ECR image URIs."""
 
     request_args = _build_request_args()
 
     response = await client.request(**request_args)
     raise_for_status(response)
-    return AgentListResponse.model_validate(response.json())
+    return ChronosApiRegistryAgentListResponse.model_validate(response.json())