mainsequence 8.1.20.dev13__tar.gz → 8.1.20.dev15__tar.gz

{mainsequence-8.1.20.dev13/mainsequence.egg-info → mainsequence-8.1.20.dev15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mainsequence
-Version: 8.1.20.dev13
+Version: 8.1.20.dev15
 Summary: Main Sequence SDK 
 Author-email: Main Sequence GmbH <dev@main-sequence.io>
 License-Expression: LicenseRef-MainSequence-Proprietary

{mainsequence-8.1.20.dev13 → mainsequence-8.1.20.dev15}/mainsequence/client/agent_runtime_models.py

@@ -6,7 +6,6 @@ import json
 import pathlib
 import time
 import uuid
-from enum import Enum
 from typing import Any, ClassVar, Literal
 
 import requests
@@ -31,6 +30,7 @@ from .observability import (
     PublicLogLevel,
 )
 from .utils import make_request, serialize_to_json
+from .value_sets import OpenStrEnum, OpenValueSet
 
 DEFAULT_AGENT_SESSION_LONG_REQUEST_TIMEOUT = (5.0, 900.0)
 DEFAULT_AGENT_RUNTIME_ACCESS_CACHE_TTL_SECONDS = 60.0
@@ -50,7 +50,7 @@ MIN_RUNTIME_INTERACTION_RETRY_SECONDS = 0.5
 MAX_RUNTIME_INTERACTION_RETRY_SECONDS = 30.0
 
 
-class AgentSessionStatus(str, Enum):
+class AgentSessionStatus(OpenStrEnum):
     PENDING = "pending"
     RUNNING = "running"
     COMPLETED = "completed"
@@ -58,7 +58,7 @@ class AgentSessionStatus(str, Enum):
     CANCELED = "canceled"
 
 
-class A2AResponseKind(str, Enum):
+class A2AResponseKind(OpenStrEnum):
     MESSAGE = "message"
     TASK = "task"
 
@@ -121,12 +121,12 @@ class A2AMessageSendResult(BasePydanticModel):
         return A2AResponseKind.TASK if self.task is not None else A2AResponseKind.MESSAGE
 
 
-class AgentHarnessKind(str, Enum):
+class AgentHarnessKind(OpenStrEnum):
     PI = "pi"
     TAU = "tau"
 
 
-class AgentHarnessProtocol(str, Enum):
+class AgentHarnessProtocol(OpenStrEnum):
     PI_CHECKPOINT_V1 = "pi-checkpoint-v1"
     TAU_SESSION_V1 = "tau-session-v1"
 
@@ -136,13 +136,15 @@ class AgentRuntimeUpdateRemediation(BasePydanticModel):
 
 
 class AgentRuntimeUpdate(BasePydanticModel):
-    state: Literal[
-        "current",
-        "update_required",
-        "updating",
-        "update_failed",
-        "not_deployed",
-        "unknown",
+    state: OpenValueSet[
+        Literal[
+            "current",
+            "update_required",
+            "updating",
+            "update_failed",
+            "not_deployed",
+            "unknown",
+        ]
     ]
     needs_redeploy: bool | None
     remediation: AgentRuntimeUpdateRemediation | None
@@ -279,7 +281,7 @@ class AgentRuntimeInteractionNotice(BasePydanticModel):
 class AgentRuntimeInteractionAction(BasePydanticModel):
     model_config = ConfigDict(extra="allow")
 
-    type: Literal["update_agent_runtime", "retry_runtime_wake"]
+    type: OpenValueSet[Literal["update_agent_runtime", "retry_runtime_wake"]]
     label: str
     confirmation_message: str
     interaction_revision: str
@@ -299,16 +301,18 @@ class AgentRuntimeInteractionOperation(BasePydanticModel):
 class AgentRuntimeInteraction(BasePydanticModel):
     model_config = ConfigDict(extra="allow")
 
-    state: Literal[
-        "ready",
-        "warning",
-        "checking",
-        "starting",
-        "waking",
-        "update_required",
-        "updating",
-        "update_failed",
-        "unavailable",
+    state: OpenValueSet[
+        Literal[
+            "ready",
+            "warning",
+            "checking",
+            "starting",
+            "waking",
+            "update_required",
+            "updating",
+            "update_failed",
+            "unavailable",
+        ]
     ]
     can_submit: bool
     notice: AgentRuntimeInteractionNotice | None
@@ -324,29 +328,33 @@ class AgentRuntimePresenceReplicas(BasePydanticModel):
 
 class AgentRuntimePresenceWake(BasePydanticModel):
     operation_uid: str
-    state: Literal[
-        "requested",
-        "in_progress",
-        "serving",
-        "failed",
-        "expired",
-        "superseded",
+    state: OpenValueSet[
+        Literal[
+            "requested",
+            "in_progress",
+            "serving",
+            "failed",
+            "expired",
+            "superseded",
+        ]
     ]
     requested_at: str
     deadline_at: str
 
 
 class AgentRuntimePresence(BasePydanticModel):
-    phase: Literal[
-        "not_deployed",
-        "observing",
-        "idle",
-        "provisioning",
-        "pulling_image",
-        "starting",
-        "serving",
-        "redeploying",
-        "failed",
+    phase: OpenValueSet[
+        Literal[
+            "not_deployed",
+            "observing",
+            "idle",
+            "provisioning",
+            "pulling_image",
+            "starting",
+            "serving",
+            "redeploying",
+            "failed",
+        ]
     ]
     replicas: AgentRuntimePresenceReplicas
     detail: str
@@ -361,7 +369,7 @@ class AgentSessionRuntimeAccess(BasePydanticModel):
         None,
         description="Canonical public UID of the coding-agent service that owns the runtime.",
     )
-    mode: Literal["token", "unavailable"] = Field(
+    mode: OpenValueSet[Literal["token", "unavailable"]] = Field(
         "token",
         description="Runtime access mode returned by the backend.",
     )
@@ -862,7 +870,7 @@ class TauAgentSessionInsightsSession(BasePydanticModel):
     agent_session_id: str = Field(..., alias="agentSessionId")
     session_id: str | None = Field(None, alias="sessionId")
     thread_id: str | None = Field(None, alias="threadId")
-    status: Literal["running", "completed", "error"]
+    status: OpenValueSet[Literal["running", "completed", "error"]]
     started_at: datetime.datetime | None = Field(None, alias="startedAt")
     updated_at: datetime.datetime | None = Field(None, alias="updatedAt")
     last_error: str | None = Field(None, alias="lastError")

{mainsequence-8.1.20.dev13 → mainsequence-8.1.20.dev15}/mainsequence/client/github_issues.py

@@ -11,11 +11,16 @@ from . import utils as client_utils
 from .base import BasePydanticModel
 from .exceptions import raise_for_response
 from .utils import make_request
+from .value_sets import OpenValueSet
 
 GitHubIssueState = Literal["open", "closed"]
 GitHubIssueStateReason = Literal["completed", "not_planned", "reopened"]
-GitHubIssueOperationType = Literal["create_issue", "update_issue", "create_comment"]
-GitHubIssueOperationStatus = Literal["pending", "succeeded", "failed", "unknown"]
+GitHubIssueOperationType = OpenValueSet[
+    Literal["create_issue", "update_issue", "create_comment"]
+]
+GitHubIssueOperationStatus = OpenValueSet[
+    Literal["pending", "succeeded", "failed", "unknown"]
+]
 
 _IDEMPOTENCY_KEY_PATTERN = re.compile(r"^[A-Za-z0-9._:-]+$")
 
@@ -115,7 +120,7 @@ class _GitHubIssueProjection(BasePydanticModel):
     issue_number: int
     html_url: str
     title: str
-    state: GitHubIssueState
+    state: OpenValueSet[GitHubIssueState]
     state_reason: str | None
     created_at: datetime.datetime
     updated_at: datetime.datetime

{mainsequence-8.1.20.dev13 → mainsequence-8.1.20.dev15}/mainsequence/client/metatables/core.py

@@ -61,6 +61,7 @@ from ..utils import (
     make_request,
     serialize_to_json,
 )
+from ..value_sets import OpenValueSet
 
 DUCK_DB = "duck_db"
 SQLITE = "sqlite"
@@ -1059,7 +1060,7 @@ class MetaTableCascadeDeletedMetaTable(BasePydanticModel):
     meta_table_uid: str
     physical_schema: str | None = None
     physical_table_name: str | None = None
-    management_mode: MetaTableManagementMode
+    management_mode: OpenValueSet[MetaTableManagementMode]
     physical_table_drop_scheduled: bool
     physical_table_drop_cascade: bool
     schema_management_protection_overridden: bool
@@ -1109,7 +1110,7 @@ class DataSource(BasePydanticModel, BaseObjectOrm):
     )
     class_type: str | None = None
     status: str | None = None
-    storage_access_mode: StorageAccessMode | None = Field(
+    storage_access_mode: OpenValueSet[StorageAccessMode] | None = Field(
         default="read_write",
         description="Runtime storage access gate: read_write, read_only, or disabled.",
     )
@@ -1369,9 +1370,9 @@ class MetaTable(
     namespace: str | None = None
     description: str | None = None
     labels: list[str] = Field(default_factory=list)
-    management_mode: MetaTableManagementMode
-    provisioning_status: Literal["reserved", "active"] = "active"
-    schema_management_mode: MetaTableSchemaManagementMode = "backend_managed"
+    management_mode: OpenValueSet[MetaTableManagementMode]
+    provisioning_status: OpenValueSet[Literal["reserved", "active"]] = "active"
+    schema_management_mode: OpenValueSet[MetaTableSchemaManagementMode] = "backend_managed"
     schema_management: dict[str, Any] = Field(default_factory=dict)
     migration_package: str | None = None
     migration_namespace: str | None = None

{mainsequence-8.1.20.dev13 → mainsequence-8.1.20.dev15}/mainsequence/client/models_foundry.py

@@ -30,6 +30,7 @@ from .metatables import DataSource as _DataSource
 from .utils import (
     make_request,
 )
+from .value_sets import OpenValueSet
 
 if TYPE_CHECKING:
     from .github_issues import GitHubIssueMutationResult, GitHubIssuePage, GitHubIssueState
@@ -508,7 +509,7 @@ class CodeRepositoryBranch(BasePydanticModel, BaseObjectOrm):
     github_repository_binding_uid: str | None = None
     latest_git_version: str = ""
     is_initialized: bool
-    provisioning_status: Literal["CREATING", "READY", "FAILED"]
+    provisioning_status: OpenValueSet[Literal["CREATING", "READY", "FAILED"]]
     provisioning_job_run_uid: str | None
     provisioning_error_code: str
     provisioning_error_detail: str
@@ -726,7 +727,7 @@ class GitHubRepositoryBinding(BasePydanticModel, BaseObjectOrm):
 
 
 class CodeRepositoryImageSourceProvenance(BasePydanticModel):
-    verification_state: Literal["verified", "unverified", "building", "failed"]
+    verification_state: OpenValueSet[Literal["verified", "unverified", "building", "failed"]]
     code_repository_uid: str | None = None
     code_repository_branch_uid: str | None = None
     github_repository_binding_uid: str | None = None

{mainsequence-8.1.20.dev13 → mainsequence-8.1.20.dev15}/mainsequence/client/models_helpers.py

@@ -5,7 +5,6 @@ import json
 import time
 from collections.abc import Collection
 from decimal import Decimal
-from enum import Enum
 from pathlib import PurePosixPath
 from typing import Any, ClassVar, Literal
 from uuid import UUID
@@ -41,6 +40,7 @@ from .observability import (
     PublicLogLevel,
 )
 from .utils import make_request
+from .value_sets import OpenStrEnum, OpenValueSet
 
 
 def get_model_class(model_class: str):
@@ -1156,27 +1156,34 @@ class CodeRepositoryResource(CurrentCodeRepositoryBranchCollectionMixin, BaseObj
         )
 
 
-class ResourceReleaseKind(str, Enum):
+class ResourceReleaseKind(OpenStrEnum):
+    """The backend's canonical release-kind vocabulary.
+
+    Open (ADR 0033): a kind a later backend adds is read as sent, rather than
+    failing the release that carries it and the listing that release is in.
+    """
+
     AGENT = "agent"
     FAST_API = "fastapi"
+    HARNESS_AGENT = "harness_agent"
     STATIC_SITE = "static_site"
 
 
 class ResourceReleaseRuntimeRouting(BasePydanticModel):
-    state: Literal["routable", "unavailable"]
+    state: OpenValueSet[Literal["routable", "unavailable"]]
     active_revision_uid: str | None = None
 
 
 class ResourceReleaseRuntimeNotice(BasePydanticModel):
     code: str
-    severity: Literal["info", "warning", "error"]
+    severity: OpenValueSet[Literal["info", "warning", "error"]]
     title: str
     message: str
 
 
 class ResourceReleaseRuntimeOperation(BasePydanticModel):
     uid: str
-    status: Literal["queued", "running", "succeeded", "failed", "superseded"]
+    status: OpenValueSet[Literal["queued", "running", "succeeded", "failed", "superseded"]]
     created_at: datetime.datetime
     started_at: datetime.datetime | None = None
     finished_at: datetime.datetime | None = None
@@ -1184,7 +1191,7 @@ class ResourceReleaseRuntimeOperation(BasePydanticModel):
 
 
 class ResourceReleaseRuntimeAdmission(BasePydanticModel):
-    state: Literal["ready", "waking", "unavailable"]
+    state: OpenValueSet[Literal["ready", "waking", "unavailable"]]
     can_request: bool
     notice: ResourceReleaseRuntimeNotice | None = None
     operation: ResourceReleaseRuntimeOperation | None = None
@@ -1198,24 +1205,28 @@ class ResourceReleaseRuntimeReplicas(BasePydanticModel):
 
 class ResourceReleaseRuntimeWake(BasePydanticModel):
     operation_uid: str
-    state: Literal[
-        "requested", "in_progress", "serving", "failed", "expired", "superseded"
+    state: OpenValueSet[
+        Literal[
+            "requested", "in_progress", "serving", "failed", "expired", "superseded"
+        ]
     ]
     requested_at: datetime.datetime
     deadline_at: datetime.datetime
 
 
 class ResourceReleaseRuntimePresence(BasePydanticModel):
-    phase: Literal[
-        "not_deployed",
-        "idle",
-        "observing",
-        "provisioning",
-        "pulling_image",
-        "starting",
-        "serving",
-        "redeploying",
-        "failed",
+    phase: OpenValueSet[
+        Literal[
+            "not_deployed",
+            "idle",
+            "observing",
+            "provisioning",
+            "pulling_image",
+            "starting",
+            "serving",
+            "redeploying",
+            "failed",
+        ]
     ]
     replicas: ResourceReleaseRuntimeReplicas
     detail: str
@@ -1225,7 +1236,9 @@ class ResourceReleaseRuntimePresence(BasePydanticModel):
 
 class ResourceReleaseRuntimeAccess(BasePydanticModel):
     resource_release_uid: str
-    release_kind: Literal["fastapi", "harness_agent"]
+    # The canonical vocabulary ResourceRelease.release_kind reads (#119): this
+    # payload names a release kind, so it names it the same way.
+    release_kind: ResourceReleaseKind
     routing: ResourceReleaseRuntimeRouting
     runtime_access: ResourceReleaseRuntimeAdmission
     runtime_presence: ResourceReleaseRuntimePresence
@@ -1512,11 +1525,14 @@ class ResourceRelease(
             normalized_release_kind = release_kind.value
         else:
             normalized_release_kind = Job._normalize_str(release_kind)
-            allowed_release_kinds = {kind.value for kind in ResourceReleaseKind}
-            if normalized_release_kind not in allowed_release_kinds:
-                raise ValueError(
-                    "release_kind must be one of: " + ", ".join(sorted(allowed_release_kinds)) + "."
-                )
+        # What the SDK sends stays closed (ADR 0033). ResourceReleaseKind reads an
+        # undeclared kind out of a response, but iterating it yields only the kinds
+        # this release declares, so a kind built here is still checked against them.
+        allowed_release_kinds = {kind.value for kind in ResourceReleaseKind}
+        if normalized_release_kind not in allowed_release_kinds:
+            raise ValueError(
+                "release_kind must be one of: " + ", ".join(sorted(allowed_release_kinds)) + "."
+            )
 
         normalized_compute = Job._validate_and_normalize_compute_fields(
             cpu_request=cpu_request,
@@ -1617,26 +1633,30 @@ class DeploymentRunStep(BaseModel):
     sequence: PositiveInt
     key: str
     name: str = ""
-    kind: Literal[
-        "source",
-        "image_build",
-        "validation",
-        "runtime_deploy",
-        "runtime_readiness",
-        "publish",
-        "cleanup",
-        "orchestration",
+    kind: OpenValueSet[
+        Literal[
+            "source",
+            "image_build",
+            "validation",
+            "runtime_deploy",
+            "runtime_readiness",
+            "publish",
+            "cleanup",
+            "orchestration",
+        ]
     ]
     required: bool = True
-    state: Literal[
-        "pending",
-        "running",
-        "succeeded",
-        "failed",
-        "cancelled",
-        "skipped",
-        "blocked",
-        "superseded",
+    state: OpenValueSet[
+        Literal[
+            "pending",
+            "running",
+            "succeeded",
+            "failed",
+            "cancelled",
+            "skipped",
+            "blocked",
+            "superseded",
+        ]
     ]
     outcome: str = ""
     artifact_context: dict[str, Any] = Field(default_factory=dict)
@@ -1658,12 +1678,14 @@ class DeploymentRunBillingComponents(BasePydanticModel):
     image_registry_service: Decimal | None = Field(max_digits=18, decimal_places=6)
 
 
-DeploymentRunPricingState = Literal[
-    "priced",
-    "partial",
-    "pending",
-    "unavailable",
-    "failed",
+DeploymentRunPricingState = OpenValueSet[
+    Literal[
+        "priced",
+        "partial",
+        "pending",
+        "unavailable",
+        "failed",
+    ]
 ]
 
 
@@ -1700,14 +1722,14 @@ class DeploymentRunLogEntry(BaseModel):
     timestamp: datetime.datetime | None = None
     step_uid: str | None = None
     source: str
-    stream: Literal["stdout", "stderr"]
-    level: Literal["debug", "info", "warning", "error"]
+    stream: OpenValueSet[Literal["stdout", "stderr"]]
+    level: OpenValueSet[Literal["debug", "info", "warning", "error"]]
     text: str = ""
 
 
 class DeploymentRunLogSource(BaseModel):
     source: str
-    state: Literal["pending", "available", "partial", "expired", "unavailable"]
+    state: OpenValueSet[Literal["pending", "available", "partial", "expired", "unavailable"]]
 
 
 class DeploymentRunLogPage(BaseModel):

{mainsequence-8.1.20.dev13 → mainsequence-8.1.20.dev15}/mainsequence/client/models_user.py

@@ -22,6 +22,7 @@ from .utils import (
     DEFAULT_TIMEOUT,
     make_request,
 )
+from .value_sets import OpenValueSet
 
 _CURRENT_AUTH_HEADERS: ContextVar[Mapping[str, Any] | None] = ContextVar(
     "_CURRENT_AUTH_HEADERS",
@@ -528,7 +529,7 @@ class Notification(DetailActionObjectMixin, BasePydanticModel, UserApiBaseObject
         title="Notification UID",
         description="Stable public unique identifier of the notification.",
     )
-    source: Literal["system", "organization"] | None = Field(
+    source: OpenValueSet[Literal["system", "organization"]] | None = Field(
         None,
         title="Notification Source",
         description="Source scope of the notification.",
@@ -544,7 +545,7 @@ class Notification(DetailActionObjectMixin, BasePydanticModel, UserApiBaseObject
         title="Source Organization UID",
         description="Public UID of the organization that owns the notification.",
     )
-    type: Literal["UR", "IM", "IN"] = Field(
+    type: OpenValueSet[Literal["UR", "IM", "IN"]] = Field(
         "IN",
         title="Notification Type",
         description="Stored notification priority code.",
@@ -914,7 +915,7 @@ class ShareableAccessState(BasePydanticModel):
         description="Backend object type identifier for the shareable object.",
         examples=["tdag.constant"],
     )
-    access_level: Literal["view", "edit"] = Field(
+    access_level: OpenValueSet[Literal["view", "edit"]] = Field(
         ...,
         title="Access Level",
         description="Permission level represented by this access-state payload.",

{mainsequence-8.1.20.dev13 → mainsequence-8.1.20.dev15}/mainsequence/client/observability.py

@@ -10,6 +10,7 @@ from pydantic import ConfigDict, Field
 from .base import BasePydanticModel
 from .exceptions import ApiError, raise_for_response
 from .utils import make_request
+from .value_sets import OpenValueSet
 
 PublicLogLevel = Literal[
     "debug",
@@ -55,7 +56,7 @@ class OwnerLogRow(BasePydanticModel):
     time: int | float | None = None
     timestamp: str | None = None
     severity: str | None = None
-    level: PublicLogLevel | None = None
+    level: OpenValueSet[PublicLogLevel] | None = None
     source: str | None = None
     event: str | None = None
     event_id: str | None = None
@@ -110,16 +111,18 @@ class OwnerLogPage(BasePydanticModel):
 class EnvironmentLogSearchRow(OwnerLogRow):
     """One sanitized row returned by an Environment-scoped collection search."""
 
-    owner_type: Literal[
-        "deployment_run",
-        "job_run",
-        "resource_release",
-        "agent",
-        "agent_session",
+    owner_type: OpenValueSet[
+        Literal[
+            "deployment_run",
+            "job_run",
+            "resource_release",
+            "agent",
+            "agent_session",
+        ]
     ]
     owner_uid: str
     occurred_at: datetime.datetime
-    level: PublicLogLevel
+    level: OpenValueSet[PublicLogLevel]
 
 
 class EnvironmentLogSearchPage(BasePydanticModel):
@@ -134,7 +137,9 @@ class EnvironmentLogSearchPage(BasePydanticModel):
     result_limit: int = Field(ge=0)
     next_cursor: str | None = None
     truncated: bool
-    truncation_reason: Literal["page_limit", "result_limit", "candidate_limit"] | None = None
+    truncation_reason: (
+        OpenValueSet[Literal["page_limit", "result_limit", "candidate_limit"]] | None
+    ) = None
     rows: list[EnvironmentLogSearchRow] = Field(default_factory=list)