mainsequence 8.1.0__tar.gz → 8.1.2__tar.gz

{mainsequence-8.1.0/mainsequence.egg-info → mainsequence-8.1.2}/PKG-INFO

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

{mainsequence-8.1.0 → mainsequence-8.1.2}/mainsequence/client/agent_runtime_models.py

@@ -122,6 +122,23 @@ class AgentHarnessProtocol(str, Enum):
     TAU_SESSION_V1 = "tau-session-v1"
 
 
+class AgentRuntimeUpdateRemediation(BasePydanticModel):
+    tool: Literal["agent.update_runtime"]
+
+
+class AgentRuntimeUpdate(BasePydanticModel):
+    state: Literal[
+        "current",
+        "update_required",
+        "updating",
+        "update_failed",
+        "not_deployed",
+        "unknown",
+    ]
+    needs_redeploy: bool | None
+    remediation: AgentRuntimeUpdateRemediation | None
+
+
 class AgentSemanticSearchResult(BasePydanticModel):
     uid: str = Field(..., description="Public UID of the matched agent.")
     name: str = Field(..., description="Human-readable display name of the matched agent.")
@@ -150,6 +167,10 @@ class AgentSemanticSearchResult(BasePydanticModel):
         ...,
         description="Name of the Organization Environment that scoped the search result.",
     )
+    runtime_update: AgentRuntimeUpdate = Field(
+        ...,
+        description="Backend-owned runtime-currency state for the matched Agent.",
+    )
     semantic_score: float = Field(
         ...,
         description="Vector similarity component of the semantic-search ranking.",
@@ -550,6 +571,10 @@ class Agent(
             "CodeRepositoryBranch, or null when the Agent is not environment-scoped."
         ),
     )
+    runtime_update: AgentRuntimeUpdate = Field(
+        ...,
+        description="Read-only backend-owned runtime-currency state for this Agent.",
+    )
     observability: ObservabilityLinks | None = Field(
         default=None,
         description="Backend-owned runtime observability and related-resource capabilities.",
@@ -2126,6 +2151,8 @@ __all__ = [
     "AgentA2AProfile",
     "AgentRuntimeImageDrift",
     "AgentRuntimeImageDriftCheck",
+    "AgentRuntimeUpdate",
+    "AgentRuntimeUpdateRemediation",
     "AgentHarnessKind",
     "AgentHarnessProtocol",
     "AgentSessionInsights",

{mainsequence-8.1.0 → mainsequence-8.1.2}/mainsequence/client/models_helpers.py

@@ -1397,6 +1397,29 @@ class DeploymentRunPipeline(BaseModel):
     steps: list[DeploymentRunStep] = Field(default_factory=list)
 
 
+class DeploymentRunBillingComponents(BasePydanticModel):
+    image_build: Decimal | None = Field(max_digits=18, decimal_places=6)
+    image_registry_storage: Decimal | None = Field(max_digits=18, decimal_places=6)
+    image_registry_service: Decimal | None = Field(max_digits=18, decimal_places=6)
+
+
+class DeploymentRunBilling(BasePydanticModel):
+    scope: Literal["image_lifecycle"]
+    total_cost: Decimal | None = Field(max_digits=18, decimal_places=6)
+    currency: str = Field(min_length=1)
+    pricing_state: Literal[
+        "priced",
+        "partial",
+        "pending",
+        "unavailable",
+        "failed",
+    ]
+    components: DeploymentRunBillingComponents
+    priced_rows: int = Field(ge=0)
+    unpriced_rows: int = Field(ge=0)
+    reused_image_count: int = Field(ge=0)
+
+
 class DeploymentRunLogEntry(BaseModel):
     sequence: int
     timestamp: datetime.datetime | None = None
@@ -1473,6 +1496,7 @@ class DeploymentRun(CurrentCodeRepositoryBranchCollectionMixin, BaseObjectOrm, B
     builder_runtime: str = ""
     logs: DeploymentRunLogReference
     error: DeploymentRunError | None = None
+    billing: DeploymentRunBilling
 
     def get_logs(
         self,

{mainsequence-8.1.0 → mainsequence-8.1.2/mainsequence.egg-info}/PKG-INFO

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

{mainsequence-8.1.0 → mainsequence-8.1.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "mainsequence"
-version = "8.1.0"
+version = "8.1.2"
 description = "Main Sequence SDK "
 readme = "README.md"
 requires-python = ">=3.13"

{mainsequence-8.1.0 → mainsequence-8.1.2}/tests/test_environment_response_contracts.py

@@ -7,6 +7,7 @@ import mainsequence.client.models_helpers as helper_models
 
 ENVIRONMENT_UID_FIELD = "organization_environment_uid"
 ENVIRONMENT_NAME_FIELD = "organization_environment_name"
+RUNTIME_UPDATE_FIELD = "runtime_update"
 
 
 @pytest.mark.parametrize(
@@ -15,13 +16,13 @@ ENVIRONMENT_NAME_FIELD = "organization_environment_name"
         pytest.param(
             ("AgentSerializer",),
             agent_models.Agent,
-            {ENVIRONMENT_UID_FIELD, ENVIRONMENT_NAME_FIELD},
+            {ENVIRONMENT_UID_FIELD, ENVIRONMENT_NAME_FIELD, RUNTIME_UPDATE_FIELD},
             id="agent",
         ),
         pytest.param(
             ("AgentSemanticSearchResultSerializer",),
             agent_models.AgentSemanticSearchResult,
-            {ENVIRONMENT_UID_FIELD, ENVIRONMENT_NAME_FIELD},
+            {ENVIRONMENT_UID_FIELD, ENVIRONMENT_NAME_FIELD, RUNTIME_UPDATE_FIELD},
             id="agent-semantic-search-result",
         ),
         pytest.param(

{mainsequence-8.1.0 → mainsequence-8.1.2}/tests/test_filter_normalization.py

@@ -3,6 +3,7 @@ import datetime
 import json
 import pathlib
 import uuid
+from decimal import Decimal
 from types import SimpleNamespace
 
 import pytest
@@ -22,6 +23,67 @@ CODE_REPOSITORY_BRANCH_UID = "5a28020a-0f1b-47ee-aab8-334286234bea"
 ENVIRONMENT_UID = "58218213-5e4e-43de-a5bd-6757f4e1c8f6"
 
 
+def _agent_runtime_update_contract(state: str = "current") -> dict:
+    values = {
+        "current": (False, None),
+        "update_required": (True, {"tool": "agent.update_runtime"}),
+        "updating": (True, None),
+        "update_failed": (True, {"tool": "agent.update_runtime"}),
+        "not_deployed": (False, None),
+        "unknown": (None, None),
+    }
+    needs_redeploy, remediation = values[state]
+    return {
+        "state": state,
+        "needs_redeploy": needs_redeploy,
+        "remediation": remediation,
+    }
+
+
+@pytest.mark.parametrize(
+    ("state", "needs_redeploy", "remediation_tool"),
+    [
+        ("current", False, None),
+        ("update_required", True, "agent.update_runtime"),
+        ("updating", True, None),
+        ("update_failed", True, "agent.update_runtime"),
+        ("not_deployed", False, None),
+        ("unknown", None, None),
+    ],
+)
+def test_agent_runtime_update_parses_every_backend_state(
+    state,
+    needs_redeploy,
+    remediation_tool,
+):
+    runtime_update = agent_models_mod.AgentRuntimeUpdate.model_validate(
+        _agent_runtime_update_contract(state)
+    )
+
+    assert runtime_update.state == state
+    assert runtime_update.needs_redeploy is needs_redeploy
+    assert (
+        runtime_update.remediation.tool if runtime_update.remediation is not None else None
+    ) == remediation_tool
+
+
+def test_agent_runtime_update_contract_is_complete_and_closed():
+    payload = _agent_runtime_update_contract("unknown")
+    for field_name in ("state", "needs_redeploy", "remediation"):
+        incomplete = dict(payload)
+        incomplete.pop(field_name)
+        with pytest.raises(ValidationError, match=field_name):
+            agent_models_mod.AgentRuntimeUpdate.model_validate(incomplete)
+
+    with pytest.raises(ValidationError, match="extra_forbidden"):
+        agent_models_mod.AgentRuntimeUpdate.model_validate(payload | {"future_state": "value"})
+
+    invalid_remediation = _agent_runtime_update_contract("update_required")
+    invalid_remediation["remediation"] = {"tool": "agent.update_session"}
+    with pytest.raises(ValidationError, match="agent.update_runtime"):
+        agent_models_mod.AgentRuntimeUpdate.model_validate(invalid_remediation)
+
+
 @pytest.fixture(autouse=True)
 def _resolved_code_repository_context(monkeypatch):
     code_repository_context._reset_code_repository_context()
@@ -1760,6 +1822,7 @@ def test_agent_runtime_models_deserialize_backend_uid_payloads():
             "repository_branch": "main",
             "organization_environment_uid": environment_uid,
             "organization_environment_name": "production",
+            "runtime_update": _agent_runtime_update_contract(),
         }
     )
     assert agent.uid == agent_uid
@@ -1770,6 +1833,8 @@ def test_agent_runtime_models_deserialize_backend_uid_payloads():
     assert agent.repository_branch == "main"
     assert agent.organization_environment_uid == environment_uid
     assert agent.organization_environment_name == "production"
+    assert agent.runtime_update.state == "current"
+    assert agent.runtime_update.needs_redeploy is False
     assert agent.a2a_profile.supported_response_kinds == [agent_models_mod.A2AResponseKind.MESSAGE]
 
     search_result = agent_models_mod.AgentSemanticSearchResult.model_validate(
@@ -1782,6 +1847,7 @@ def test_agent_runtime_models_deserialize_backend_uid_payloads():
             "repository_branch": "main",
             "organization_environment_uid": environment_uid,
             "organization_environment_name": "production",
+            "runtime_update": _agent_runtime_update_contract("update_required"),
             "semantic_score": 0.91,
             "text_score": 0.74,
             "combined_score": 0.85,
@@ -1792,6 +1858,8 @@ def test_agent_runtime_models_deserialize_backend_uid_payloads():
     assert search_result.repository_branch == "main"
     assert search_result.organization_environment_uid == environment_uid
     assert search_result.organization_environment_name == "production"
+    assert search_result.runtime_update.state == "update_required"
+    assert search_result.runtime_update.remediation.tool == "agent.update_runtime"
     assert search_result.a2a_profile.default_response_kind == (
         agent_models_mod.A2AResponseKind.MESSAGE
     )
@@ -1913,6 +1981,7 @@ def test_agent_scope_code_repositoryion_is_required_but_nullable():
         "repository_branch": None,
         "organization_environment_uid": None,
         "organization_environment_name": None,
+        "runtime_update": _agent_runtime_update_contract("not_deployed"),
     }
 
     agent = agent_models_mod.Agent.model_validate(payload)
@@ -1946,6 +2015,7 @@ def test_agent_filter_sends_environment_read_scope_and_parses_code_repositoryion
                         "repository_branch": "main",
                         "organization_environment_uid": str(environment_uid),
                         "organization_environment_name": "production",
+                        "runtime_update": _agent_runtime_update_contract("current"),
                     }
                 ],
                 "next": None,
@@ -1989,6 +2059,63 @@ def test_agent_filter_sends_environment_read_scope_and_parses_code_repositoryion
     assert len(agents) == 1
     assert agents[0].repository_branch == "main"
     assert agents[0].organization_environment_uid == str(environment_uid)
+    assert agents[0].runtime_update.state == "current"
+
+
+def test_agent_get_parses_runtime_update_projection(monkeypatch):
+    captured = {}
+    agent_uid = "e0e75693-4110-464c-93e0-82c7fd9c9a23"
+    environment_uid = uuid.UUID("22222222-2222-4222-8222-222222222222")
+
+    class FakeResponse:
+        status_code = 200
+
+        @staticmethod
+        def json():
+            return {
+                "uid": agent_uid,
+                "name": "CodeRepository Executor",
+                "llm_thinking": "medium",
+                "code_repository_branch_uid": "9d81d63f-b8c9-404d-9f1a-5f2ad29dbf16",
+                "repository_branch": "main",
+                "organization_environment_uid": str(environment_uid),
+                "organization_environment_name": "production",
+                "runtime_update": _agent_runtime_update_contract("update_failed"),
+            }
+
+    def _fake_make_request(*, s, loaders, r_type, url, payload, time_out=None):
+        captured.update(
+            {
+                "r_type": r_type,
+                "url": url,
+                "payload": payload,
+                "timeout": time_out,
+            }
+        )
+        return FakeResponse()
+
+    monkeypatch.setattr(base_mod, "make_request", _fake_make_request)
+    monkeypatch.setattr(
+        agent_models_mod.Agent,
+        "build_session",
+        classmethod(lambda cls: object()),
+    )
+
+    agent = agent_models_mod.Agent.get(
+        pk=agent_uid,
+        organization_environment_uid=environment_uid,
+        timeout=12,
+    )
+
+    assert captured == {
+        "r_type": "GET",
+        "url": f"{agent_models_mod.Agent.get_object_url()}/{agent_uid}/",
+        "payload": {"params": {"organization_environment_uid": str(environment_uid)}},
+        "timeout": 12,
+    }
+    assert agent.runtime_update.state == "update_failed"
+    assert agent.runtime_update.needs_redeploy is True
+    assert agent.runtime_update.remediation.tool == "agent.update_runtime"
 
 
 def test_agent_semantic_search_sends_environment_scope_and_parses_code_repositoryion(monkeypatch):
@@ -2010,6 +2137,7 @@ def test_agent_semantic_search_sends_environment_scope_and_parses_code_repositor
                     "repository_branch": "main",
                     "organization_environment_uid": str(environment_uid),
                     "organization_environment_name": "production",
+                    "runtime_update": _agent_runtime_update_contract("unknown"),
                     "semantic_score": 0.91,
                     "text_score": 0.74,
                     "combined_score": 0.85,
@@ -2056,6 +2184,8 @@ def test_agent_semantic_search_sends_environment_scope_and_parses_code_repositor
     assert len(results) == 1
     assert results[0].code_repository_branch_uid == "9d81d63f-b8c9-404d-9f1a-5f2ad29dbf16"
     assert results[0].organization_environment_name == "production"
+    assert results[0].runtime_update.state == "unknown"
+    assert results[0].runtime_update.needs_redeploy is None
 
 
 def test_agent_session_filter_supports_archive_history_query(monkeypatch):
@@ -3016,6 +3146,83 @@ def _resource_release_pipeline_payload(*, running_step: str | None = None):
     }
 
 
+def _deployment_run_billing_payload(
+    *,
+    pricing_state: str = "priced",
+    total_cost: str | None = "0.000000",
+) -> dict:
+    return {
+        "scope": "image_lifecycle",
+        "total_cost": total_cost,
+        "currency": "USD",
+        "pricing_state": pricing_state,
+        "components": {
+            "image_build": total_cost,
+            "image_registry_storage": "0.000000" if total_cost is not None else None,
+            "image_registry_service": "0.000000" if total_cost is not None else None,
+        },
+        "priced_rows": 3 if pricing_state == "priced" else 0,
+        "unpriced_rows": 0 if pricing_state == "priced" else 3,
+        "reused_image_count": 0,
+    }
+
+
+@pytest.mark.parametrize(
+    "pricing_state",
+    ["priced", "partial", "pending", "unavailable", "failed"],
+)
+def test_deployment_run_billing_parses_every_backend_pricing_state(pricing_state):
+    total_cost = "1.250000" if pricing_state in {"priced", "partial"} else None
+
+    billing = models_helpers_mod.DeploymentRunBilling.model_validate(
+        _deployment_run_billing_payload(
+            pricing_state=pricing_state,
+            total_cost=total_cost,
+        )
+    )
+
+    assert billing.pricing_state == pricing_state
+    assert billing.total_cost == (Decimal(total_cost) if total_cost is not None else None)
+    assert billing.components.image_build == billing.total_cost
+
+
+def test_deployment_run_billing_contract_is_complete_and_closed():
+    payload = _deployment_run_billing_payload()
+    for field_name in (
+        "scope",
+        "total_cost",
+        "currency",
+        "pricing_state",
+        "components",
+        "priced_rows",
+        "unpriced_rows",
+        "reused_image_count",
+    ):
+        incomplete = dict(payload)
+        incomplete.pop(field_name)
+        with pytest.raises(ValidationError, match=field_name):
+            models_helpers_mod.DeploymentRunBilling.model_validate(incomplete)
+
+    with pytest.raises(ValidationError, match="extra_forbidden"):
+        models_helpers_mod.DeploymentRunBilling.model_validate(
+            payload | {"future_billing_field": "value"}
+        )
+
+    with pytest.raises(ValidationError, match="extra_forbidden"):
+        models_helpers_mod.DeploymentRunBilling.model_validate(
+            payload
+            | {
+                "components": payload["components"]
+                | {"future_component": "0.000000"}
+            }
+        )
+
+    with pytest.raises(ValidationError, match="greater_than_equal"):
+        models_helpers_mod.DeploymentRunBilling.model_validate(
+            payload | {"unpriced_rows": -1}
+        )
+
+
 def test_resource_release_deploy_current_version_posts_detail_action(monkeypatch):
     captured = {}
     release_uid = "2f4c4c3d-5669-4da5-9d86-b84633c1e6ed"
@@ -3060,6 +3267,7 @@ def test_resource_release_deploy_current_version_posts_detail_action(monkeypatch
                     "retention_expires_at": None,
                 },
                 "error": None,
+                "billing": _deployment_run_billing_payload(),
             }
 
     def _fake_make_request(*, s, loaders, r_type, url, payload, time_out=None):
@@ -3084,6 +3292,7 @@ def test_resource_release_deploy_current_version_posts_detail_action(monkeypatch
     assert run.builder_runtime == ""
     assert run.logs.state == "available"
     assert run.error is None
+    assert run.billing.total_cost == Decimal("0.000000")
     assert captured == {
         "r_type": "POST",
         "url": (
@@ -3127,6 +3336,7 @@ def test_deployment_run_collection_and_detail_use_unified_resource_release_contr
             "retention_expires_at": None,
         },
         "error": None,
+        "billing": _deployment_run_billing_payload(),
     }
 
     collection_payload = {
@@ -3182,6 +3392,8 @@ def test_deployment_run_collection_and_detail_use_unified_resource_release_contr
     assert detail.builder_runtime == ""
     assert detail.logs.state == "available"
     assert detail.error is None
+    assert runs[0].billing.pricing_state == "priced"
+    assert detail.billing.components.image_registry_storage == Decimal("0.000000")
     assert captured[0]["payload"] == {
         "params": {
             "code_repository_branch_uid": CODE_REPOSITORY_BRANCH_UID,
@@ -3221,6 +3433,10 @@ def test_unified_deployment_run_models_and_filters(monkeypatch):
                 "retention_expires_at": None,
             },
             "error": None,
+            "billing": _deployment_run_billing_payload(
+                pricing_state="pending",
+                total_cost=None,
+            ),
         }
     )
 
@@ -3238,6 +3454,8 @@ def test_unified_deployment_run_models_and_filters(monkeypatch):
     assert run.pipeline.steps[4].state == "pending"
     assert run.builder_image == "us-docker.pkg.dev/platform/static-builder:latest"
     assert run.builder_runtime == "nodejs22"
+    assert run.billing.pricing_state == "pending"
+    assert run.billing.total_cost is None
     assert normalized == {
         "code_repository_branch_uid": code_repository_branch_uid,
         "target_type__in": ["resource_release", "static_site"],
@@ -4024,6 +4242,7 @@ def test_agent_respond_uses_agent_scoped_sessionless_contract(monkeypatch):
         repository_branch=None,
         organization_environment_uid=None,
         organization_environment_name=None,
+        runtime_update=_agent_runtime_update_contract(),
     )
 
     class FakeAccessResponse:
@@ -4131,6 +4350,7 @@ def test_agent_respond_waits_for_transient_runtime_interaction(monkeypatch):
         repository_branch=None,
         organization_environment_uid=None,
         organization_environment_name=None,
+        runtime_update=_agent_runtime_update_contract(),
     )
     accesses = iter(
         [
@@ -4221,6 +4441,7 @@ def test_agent_get_or_create_session_posts_new_contract(monkeypatch):
         repository_branch=None,
         organization_environment_uid=None,
         organization_environment_name=None,
+        runtime_update=_agent_runtime_update_contract(),
     )
 
     class FakeResponse:
@@ -4315,6 +4536,7 @@ def test_agent_get_or_create_session_parses_reused_handle_capabilities(monkeypat
         repository_branch=None,
         organization_environment_uid=None,
         organization_environment_name=None,
+        runtime_update=_agent_runtime_update_contract(),
     )
     captured = {}
 
@@ -4389,6 +4611,7 @@ def test_agent_get_or_create_session_by_uid_sends_only_session_uid(monkeypatch):
         repository_branch=None,
         organization_environment_uid=None,
         organization_environment_name=None,
+        runtime_update=_agent_runtime_update_contract(),
     )
 
     class FakeResponse:

{mainsequence-8.1.0 → mainsequence-8.1.2}/tests/test_observability_client.py

@@ -223,6 +223,11 @@ def test_agent_logs_support_optional_session_filter_without_environment_override
         repository_branch=None,
         organization_environment_uid=ENVIRONMENT_UID,
         organization_environment_name="Development",
+        runtime_update={
+            "state": "current",
+            "needs_redeploy": False,
+            "remediation": None,
+        },
         observability={
             "application_logs_url": (
                 "/api/v1/agents/e0e75693-4110-464c-93e0-82c7fd9c9a23/logs/"