mainsequence 8.1.2__tar.gz → 8.1.4__tar.gz

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

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

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

@@ -1992,6 +1992,14 @@ class AgentSession(OwnerLogMixin, BaseObjectOrm, BasePydanticModel):
             "harness. Capability names are backend-extensible."
         ),
     )
+    catalog_digest: str | None = Field(
+        ...,
+        pattern=r"^sha256:[0-9a-f]{64}$",
+        description=(
+            "Read-only digest of the model-provider catalog used to resolve the "
+            "session, or null when no provider-control projection is available."
+        ),
+    )
     agent_uid: str | None = Field(
         None, description="Public UID of the agent definition used for this session."
     )
@@ -2028,6 +2036,20 @@ class AgentSession(OwnerLogMixin, BaseObjectOrm, BasePydanticModel):
         "",
         description="Read-only helper with the canonical machine-readable agent runtime or workflow type.",
     )
+    organization_environment_uid: str = Field(
+        ...,
+        description=(
+            "Read-only public UID of the Organization Environment that owns the "
+            "session through its Agent."
+        ),
+    )
+    organization_environment_name: str = Field(
+        ...,
+        description=(
+            "Read-only name of the Organization Environment that owns the session "
+            "through its Agent."
+        ),
+    )
     harness: AgentHarnessKind | None = Field(
         None,
         description=(

{mainsequence-8.1.2 → mainsequence-8.1.4}/mainsequence/client/base.py

@@ -130,6 +130,8 @@ class BaseObjectOrm:
     DESTROY_QUERY_PARAM_DESCRIPTIONS: ClassVar[dict[str, str] | None] = None
     PUBLIC_LOOKUP_FIELD: ClassVar[str] = "uid"
     SDK_OWNED_CONTEXT_FIELDS: ClassVar[frozenset[str]] = frozenset()
+    # Opt in when collection responses intentionally omit detail-only fields.
+    HYDRATE_FILTER_GET_FROM_DETAIL: ClassVar[bool] = False
 
     END_POINTS = {
         "User": "users",
@@ -633,7 +635,14 @@ class BaseObjectOrm:
         if len(candidates) > 1:
             raise ApiError(f"Multiple objects returned for {cls.__name__} with filters={filters}")
 
-        return candidates[0]
+        candidate = candidates[0]
+        if cls.HYDRATE_FILTER_GET_FROM_DETAIL:
+            return cls.get(
+                pk=candidate._public_detail_reference(),
+                timeout=timeout,
+            )
+
+        return candidate
 
     @classmethod
     def get_by_uid(cls, uid: str, timeout=None, **filters):

{mainsequence-8.1.2 → mainsequence-8.1.4}/mainsequence/client/models_foundry.py

@@ -1083,6 +1083,7 @@ class Secret(
     BasePydanticModel,
     BaseObjectOrm,
 ):
+    HYDRATE_FILTER_GET_FROM_DETAIL: ClassVar[bool] = True
     FILTERSET_FIELDS: ClassVar[dict[str, list[str]]] = {
         "uid": ["in", "exact"],
         "name": ["in", "exact"],

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

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

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

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

{mainsequence-8.1.2 → mainsequence-8.1.4}/tests/test_environment_resource_context.py

@@ -55,6 +55,7 @@ def test_environment_resource_reads_use_sdk_owned_scope(monkeypatch):
     detail = models_module.Secret.get_by_uid(_secret_payload()["uid"])
 
     assert listed[0].organization_environment_uid == ENVIRONMENT_UID
+    assert listed[0].value is None
     assert detail.organization_environment_name == "Development"
     assert requests[0]["payload"]["params"] == {
         "name": "API_KEY",
@@ -65,6 +66,55 @@ def test_environment_resource_reads_use_sdk_owned_scope(monkeypatch):
     }
 
 
+def test_secret_get_by_name_hydrates_value_from_scoped_detail(monkeypatch):
+    requests = []
+
+    def fake_make_request(**kwargs):
+        requests.append(kwargs)
+        if kwargs["url"].rstrip("/").endswith("secrets"):
+            return _Response([_secret_payload()])
+        return _Response(_secret_payload() | {"value": "provider-secret"})
+
+    monkeypatch.setattr(base_module, "make_request", fake_make_request)
+
+    secret = models_module.Secret.get(name="API_KEY", timeout=17)
+
+    assert secret.value is not None
+    assert secret.value.get_secret_value() == "provider-secret"
+    assert "value" not in secret.model_dump()
+    assert len(requests) == 2
+    assert requests[0]["payload"]["params"] == {
+        "name": "API_KEY",
+        "organization_environment_uid": ENVIRONMENT_UID,
+    }
+    assert requests[1]["url"].endswith(f"/secrets/{_secret_payload()['uid']}/")
+    assert requests[1]["payload"]["params"] == {
+        "organization_environment_uid": ENVIRONMENT_UID,
+    }
+    assert [request["time_out"] for request in requests] == [17, 17]
+
+
+@pytest.mark.parametrize("list_payload", [[], [_secret_payload(), _secret_payload()]])
+def test_secret_get_by_name_preserves_unique_lookup_errors_without_detail_request(
+    monkeypatch,
+    list_payload,
+):
+    requests = []
+
+    def fake_make_request(**kwargs):
+        requests.append(kwargs)
+        return _Response(list_payload)
+
+    monkeypatch.setattr(base_module, "make_request", fake_make_request)
+
+    expected_error = base_module.DoesNotExist if not list_payload else base_module.ApiError
+    with pytest.raises(expected_error):
+        models_module.Secret.get(name="API_KEY")
+
+    assert len(requests) == 1
+    assert requests[0]["url"].rstrip("/").endswith("secrets")
+
+
 def test_environment_resource_create_delete_and_detail_actions_are_scoped(monkeypatch):
     requests = []
 

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

@@ -25,6 +25,12 @@ RUNTIME_UPDATE_FIELD = "runtime_update"
             {ENVIRONMENT_UID_FIELD, ENVIRONMENT_NAME_FIELD, RUNTIME_UPDATE_FIELD},
             id="agent-semantic-search-result",
         ),
+        pytest.param(
+            ("AgentSessionSerializer",),
+            agent_models.AgentSession,
+            {ENVIRONMENT_UID_FIELD, ENVIRONMENT_NAME_FIELD},
+            id="agent-session",
+        ),
         pytest.param(
             ("JobSerializer",),
             helper_models.Job,

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

@@ -1870,6 +1870,8 @@ def test_agent_runtime_models_deserialize_backend_uid_payloads():
             "agent_uid": agent_uid,
             "agent_name": "Research Copilot",
             "agent_type": "custom",
+            "organization_environment_uid": environment_uid,
+            "organization_environment_name": "production",
             "harness": "tau",
             "harness_protocol": "tau-session-v1",
             "harness_version": "0.3.1",
@@ -1912,10 +1914,14 @@ def test_agent_runtime_models_deserialize_backend_uid_payloads():
                 "tau_activity_sequence": "v1",
                 "tau_turn_commit": "v1",
             },
+            "catalog_digest": "sha256:" + ("a" * 64),
         }
     )
     assert session.uid == session_uid
     assert session.agent_uid == agent_uid
+    assert session.organization_environment_uid == environment_uid
+    assert session.organization_environment_name == "production"
+    assert session.catalog_digest == "sha256:" + ("a" * 64)
     assert session.name == "Research follow-up"
     assert session.harness is agent_models_mod.AgentHarnessKind.TAU
     assert session.harness_protocol is agent_models_mod.AgentHarnessProtocol.TAU_SESSION_V1
@@ -2246,6 +2252,8 @@ def test_agent_session_list_and_detail_parse_runtime_capabilities(monkeypatch):
         "agent_uid": agent_uid,
         "agent_name": "Research Copilot",
         "agent_type": "custom",
+        "organization_environment_uid": ENVIRONMENT_UID,
+        "organization_environment_name": "production",
         "harness": "tau",
         "harness_protocol": "tau-session-v1",
         "harness_version": "0.3.1",
@@ -2266,6 +2274,7 @@ def test_agent_session_list_and_detail_parse_runtime_capabilities(monkeypatch):
             "tau_activity_sequence": "v1",
             "tau_turn_commit": "v1",
         },
+        "catalog_digest": "sha256:" + ("b" * 64),
     }
     responses = iter(
         [
@@ -2295,6 +2304,9 @@ def test_agent_session_list_and_detail_parse_runtime_capabilities(monkeypatch):
     assert listed[0].runtime_capabilities["tau_turn_commit"] == "v1"
     assert detailed.runtime_capabilities == listed[0].runtime_capabilities
     assert detailed.observability.application_logs_url.endswith(f"/{session_uid}/logs/")
+    assert listed[0].organization_environment_uid == ENVIRONMENT_UID
+    assert detailed.organization_environment_name == "production"
+    assert detailed.catalog_digest == "sha256:" + ("b" * 64)
 
     with pytest.raises(ValidationError, match="unexpected_projection"):
         agent_models_mod.AgentSession.model_validate(
@@ -2302,6 +2314,41 @@ def test_agent_session_list_and_detail_parse_runtime_capabilities(monkeypatch):
         )
 
 
+def test_agent_session_environment_and_catalog_projection_contract_is_strict():
+    payload = {
+        "organization_environment_uid": ENVIRONMENT_UID,
+        "organization_environment_name": "production",
+        "catalog_digest": None,
+        "llm_provider": "openai",
+        "llm_model": "gpt-5.4",
+    }
+
+    session = agent_models_mod.AgentSession.model_validate(payload)
+    assert session.catalog_digest is None
+
+    for field_name in (
+        "organization_environment_uid",
+        "organization_environment_name",
+        "catalog_digest",
+    ):
+        incomplete = dict(payload)
+        incomplete.pop(field_name)
+        with pytest.raises(ValidationError, match=field_name):
+            agent_models_mod.AgentSession.model_validate(incomplete)
+
+    for field_name in (
+        "organization_environment_uid",
+        "organization_environment_name",
+    ):
+        with pytest.raises(ValidationError, match=field_name):
+            agent_models_mod.AgentSession.model_validate(payload | {field_name: None})
+
+    with pytest.raises(ValidationError, match="catalog_digest"):
+        agent_models_mod.AgentSession.model_validate(
+            payload | {"catalog_digest": "not-a-catalog-digest"}
+        )
+
+
 def test_agent_session_get_insights_parses_pi_response(monkeypatch):
     session_uid = "3f1cc452-43ec-49cb-b2ba-87dbac164d29"
     captured = {}
@@ -2461,6 +2508,8 @@ def test_agent_session_archive_actions_return_current_session_contract(monkeypat
                 "agent_uid": "9d81d63f-b8c9-404d-9f1a-5f2ad29dbf16",
                 "agent_name": "Research Copilot",
                 "agent_type": "custom",
+                "organization_environment_uid": ENVIRONMENT_UID,
+                "organization_environment_name": "production",
                 "harness": "tau",
                 "harness_protocol": "tau-session-v1",
                 "harness_version": "0.3.1",
@@ -2484,6 +2533,7 @@ def test_agent_session_archive_actions_return_current_session_contract(monkeypat
                 "thread_id": "thread-123",
                 "session_metadata": {},
                 "bound_handle": None,
+                "catalog_digest": None,
             }
 
     def _fake_make_request(*, s, loaders, r_type, url, payload, time_out=None):
@@ -3671,6 +3721,16 @@ def test_agent_session_runtime_access_accepts_minimal_image_drift(monkeypatch):
 def test_agent_session_send_a2a_message_posts_standard_contract(monkeypatch):
     captured = {"resolve_count": 0, "runtime": {}}
     session_uid = "3f1cc452-43ec-49cb-b2ba-87dbac164d29"
+    session = agent_models_mod.AgentSession.model_validate(
+        {
+            "uid": session_uid,
+            "organization_environment_uid": ENVIRONMENT_UID,
+            "organization_environment_name": "production",
+            "catalog_digest": "sha256:" + ("e" * 64),
+            "llm_provider": "openai",
+            "llm_model": "gpt-5.4",
+        }
+    )
 
     agent_models_mod.AgentSession.clear_cached_runtime_access(session_uid)
 
@@ -3733,7 +3793,7 @@ def test_agent_session_send_a2a_message_posts_standard_contract(monkeypatch):
     )
 
     payload = agent_models_mod.AgentSession.send_a2a_message(
-        session_uid,
+        session,
         message="What can this agent do?",
         timeout=15,
     )
@@ -4455,6 +4515,8 @@ def test_agent_get_or_create_session_posts_new_contract(monkeypatch):
                 "agent_uid": agent_uid,
                 "agent_name": "Research Copilot",
                 "agent_type": "custom",
+                "organization_environment_uid": ENVIRONMENT_UID,
+                "organization_environment_name": "production",
                 "created_by_user_uid": user_uid,
                 "parent_session_uid": parent_session_uid,
                 "name": "Quarterly portfolio review",
@@ -4477,6 +4539,7 @@ def test_agent_get_or_create_session_posts_new_contract(monkeypatch):
                     "owner_user_uid": user_uid,
                     "is_locked": False,
                 },
+                "catalog_digest": "sha256:" + ("c" * 64),
             }
 
     def _fake_make_request(*, s, loaders, r_type, url, payload, time_out=None):
@@ -4501,6 +4564,8 @@ def test_agent_get_or_create_session_posts_new_contract(monkeypatch):
     assert session.uid == session_uid
     assert session.name == "Quarterly portfolio review"
     assert session.parent_session_uid == parent_session_uid
+    assert session.organization_environment_uid == ENVIRONMENT_UID
+    assert session.catalog_digest == "sha256:" + ("c" * 64)
     assert captured == {
         "r_type": "POST",
         "url": (
@@ -4551,6 +4616,8 @@ def test_agent_get_or_create_session_parses_reused_handle_capabilities(monkeypat
                 "agent_uid": agent_uid,
                 "agent_name": "Research Copilot",
                 "agent_type": "custom",
+                "organization_environment_uid": ENVIRONMENT_UID,
+                "organization_environment_name": "production",
                 "harness": "tau",
                 "harness_protocol": "tau-session-v1",
                 "harness_version": "0.3.1",
@@ -4577,6 +4644,7 @@ def test_agent_get_or_create_session_parses_reused_handle_capabilities(monkeypat
                     "tau_activity_sequence": "v1",
                     "tau_turn_commit": "v1",
                 },
+                "catalog_digest": "sha256:" + ("d" * 64),
             }
 
     def _fake_make_request(*, s, loaders, r_type, url, payload, time_out=None):
@@ -4591,6 +4659,8 @@ def test_agent_get_or_create_session_parses_reused_handle_capabilities(monkeypat
     assert session.bound_handle["handle_unique_id"] == handle_unique_id
     assert session.runtime_capabilities["tau_runtime_bootstrap"] == "v1"
     assert session.observability.application_logs_url.endswith(f"/{session_uid}/logs/")
+    assert session.organization_environment_name == "production"
+    assert session.catalog_digest == "sha256:" + ("d" * 64)
     assert captured["payload"] == {"json": {"handle_unique_id": handle_unique_id}}
     assert captured["timeout"] == 13
 
@@ -4625,12 +4695,15 @@ def test_agent_get_or_create_session_by_uid_sends_only_session_uid(monkeypatch):
                 "agent_uid": agent_uid,
                 "agent_name": "Research Copilot",
                 "agent_type": "custom",
+                "organization_environment_uid": ENVIRONMENT_UID,
+                "organization_environment_name": "production",
                 "name": "Existing session",
                 "status": "running",
                 "llm_provider": "openai",
                 "llm_model": "gpt-5.4",
                 "llm_thinking": "",
                 "bound_handle": None,
+                "catalog_digest": None,
             }
 
     def _fake_make_request(*, s, loaders, r_type, url, payload, time_out=None):