mainsequence 8.1.4__tar.gz → 8.1.6__tar.gz

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

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

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

@@ -30,6 +30,8 @@ DEFAULT_AGENT_RUNTIME_ACCESS_CACHE_EXPIRY_SKEW_SECONDS = 30.0
 STANDARD_A2A_MESSAGE_SEND_PATH = "/api/a2a/v1/message:send"
 STANDARD_A2A_TASKS_PATH = "/api/a2a/v1/tasks"
 STANDARD_A2A_CONTENT_TYPE = "application/a2a+json"
+STANDARD_A2A_REQUESTER_ROLE = "ROLE_REQUESTER"
+STANDARD_A2A_RESPONDER_ROLE = "ROLE_RESPONDER"
 STANDARD_A2A_RESPONSE_KIND_EXTENSION_URI = "https://mainsequence.ai/a2a/extensions/response-kind/v1"
 STANDARD_A2A_OUTPUT_CONTRACT_METADATA_KEY = (
     "https://mainsequence.ai/a2a/extensions/output-contract/v1"
@@ -105,6 +107,14 @@ class A2AMessageSendResult(BasePydanticModel):
     def _require_exact_result(self) -> A2AMessageSendResult:
         if (self.message is None) == (self.task is None):
             raise ValueError("A2A message-send result must contain exactly one of message or task")
+        if (
+            self.message is not None
+            and self.message.get("role") != STANDARD_A2A_RESPONDER_ROLE
+        ):
+            raise ValueError(
+                "A2A message result role must be "
+                f"{STANDARD_A2A_RESPONDER_ROLE}."
+            )
         return self
 
     @property
@@ -227,6 +237,13 @@ class AgentRuntimeImageDrift(BasePydanticModel):
     has_drift: bool = Field(
         False, description="Whether any included drift check is currently drifting."
     )
+    requires_user_action: bool = Field(
+        False,
+        description=(
+            "Whether at least one current drift condition requires explicit user "
+            "action instead of backend autohealing."
+        ),
+    )
     autoheal_available: bool = Field(
         False,
         description="Whether all currently drifting checks can be repaired automatically by the backend.",
@@ -670,7 +687,7 @@ class Agent(
         body: dict[str, Any] = {
             "message": {
                 "messageId": str(message_id or "").strip() or f"msg-{uuid.uuid4()}",
-                "role": "ROLE_USER",
+                "role": STANDARD_A2A_REQUESTER_ROLE,
                 "parts": parts,
             },
             "configuration": {
@@ -1519,7 +1536,7 @@ class AgentSession(OwnerLogMixin, BaseObjectOrm, BasePydanticModel):
         body: dict[str, Any] = {
             "message": {
                 "messageId": normalized_message_id,
-                "role": "ROLE_USER",
+                "role": STANDARD_A2A_REQUESTER_ROLE,
                 "contextId": agent_session_uid,
                 "parts": parts,
             },
@@ -1725,6 +1742,16 @@ class AgentSession(OwnerLogMixin, BaseObjectOrm, BasePydanticModel):
                 response=response,
                 payload=body,
             )
+        if (
+            result.message is not None
+            and result.message.get("contextId") != session_uid
+        ):
+            raise ApiError(
+                "Standard A2A message response contextId does not match the target "
+                "AgentSession.",
+                response=response,
+                payload=body,
+            )
         return result
 
     @classmethod

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

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

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

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

{mainsequence-8.1.4 → mainsequence-8.1.6}/tests/test_cli.py

@@ -1055,7 +1055,7 @@ def test_send_agent_session_a2a_message_uses_client_model(cli_mod, monkeypatch):
                 return {
                     "message": {
                         "messageId": "msg-runtime-output",
-                        "role": "ROLE_AGENT",
+                        "role": "ROLE_RESPONDER",
                         "contextId": agent_session,
                         "parts": [{"text": "Done."}],
                     }
@@ -6785,7 +6785,7 @@ def test_agent_session_a2a_send_always_returns_json(cli_mod, runner, monkeypatch
         return {
             "message": {
                 "messageId": "msg-runtime-output",
-                "role": "ROLE_AGENT",
+                "role": "ROLE_RESPONDER",
                 "contextId": agent_session_uid,
                 "parts": [{"text": '{"ok": true}'}],
             }
@@ -6860,7 +6860,7 @@ def test_agent_session_a2a_send_does_not_require_runtime_resolve(cli_mod, runner
         return {
             "message": {
                 "messageId": "msg-runtime-output",
-                "role": "ROLE_AGENT",
+                "role": "ROLE_RESPONDER",
                 "contextId": args[0],
                 "parts": [{"text": "Done."}],
             }

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

@@ -3609,6 +3609,7 @@ def test_agent_session_runtime_access_uses_session_uid_route(monkeypatch):
                     "agent_kind": "astro_orchestrator",
                     "available": True,
                     "has_drift": False,
+                    "requires_user_action": False,
                     "autoheal_available": False,
                     "autoheal_message": "No automatic drift repair is needed.",
                     "checks": [
@@ -3667,6 +3668,7 @@ def test_agent_session_runtime_access_uses_session_uid_route(monkeypatch):
     assert access.knative_service_runtime_uid == "70c6efb9-8e80-4051-ad3a-f432b2c37f5a"
     assert access.image_drift is not None
     assert access.image_drift.agent_kind == "astro_orchestrator"
+    assert access.image_drift.requires_user_action is False
     assert access.image_drift.checks[0].key == "orchestrator_image"
     assert access.image_drift.catalog_state is not None
     assert access.image_drift.catalog_state["refresh_required"] is False
@@ -3711,6 +3713,7 @@ def test_agent_session_runtime_access_accepts_minimal_image_drift(monkeypatch):
 
     assert access.image_drift is not None
     assert access.image_drift.has_drift is False
+    assert access.image_drift.requires_user_action is False
     assert access.image_drift.detail is None
     assert access.model_dump()["reconciliation"] == {
         "queued": False,
@@ -3718,6 +3721,20 @@ def test_agent_session_runtime_access_accepts_minimal_image_drift(monkeypatch):
     }
 
 
+@pytest.mark.parametrize("requires_user_action", [False, True])
+def test_agent_runtime_image_drift_parses_user_action_signal(requires_user_action):
+    image_drift = agent_models_mod.AgentRuntimeImageDrift.model_validate(
+        {"requires_user_action": requires_user_action}
+    )
+
+    assert image_drift.requires_user_action is requires_user_action
+
+    with pytest.raises(ValidationError, match="requires_user_action"):
+        agent_models_mod.AgentRuntimeImageDrift.model_validate(
+            {"requires_user_action": None}
+        )
+
+
 def test_agent_session_send_a2a_message_posts_standard_contract(monkeypatch):
     captured = {"resolve_count": 0, "runtime": {}}
     session_uid = "3f1cc452-43ec-49cb-b2ba-87dbac164d29"
@@ -3747,6 +3764,13 @@ def test_agent_session_send_a2a_message_posts_standard_contract(monkeypatch):
                 "token": "tok-secret",
                 "expires_at": "2999-01-01T00:00:00Z",
                 **_ready_runtime_contract(),
+                "image_drift": {
+                    "agent_kind": "astro_orchestrator",
+                    "available": True,
+                    "has_drift": False,
+                    "requires_user_action": False,
+                    "checks": [],
+                },
             }
 
     class FakeRuntimeResponse:
@@ -3759,7 +3783,7 @@ def test_agent_session_send_a2a_message_posts_standard_contract(monkeypatch):
             return {
                 "message": {
                     "messageId": "msg-runtime-output",
-                    "role": "ROLE_AGENT",
+                    "role": "ROLE_RESPONDER",
                     "contextId": session_uid,
                     "parts": [{"text": "I can analyze workspaces."}],
                 }
@@ -3799,6 +3823,10 @@ def test_agent_session_send_a2a_message_posts_standard_contract(monkeypatch):
     )
 
     assert captured["resolve_count"] == 1
+    cached_access = agent_models_mod.AgentSession.get_cached_runtime_access(session)
+    assert cached_access is not None
+    assert cached_access.image_drift is not None
+    assert cached_access.image_drift.requires_user_action is False
     assert captured["resolve"] == {
         "r_type": "POST",
         "url": f"{agent_models_mod.AgentSession.get_object_url()}/{session_uid}/resolve-runtime-access/",
@@ -3821,7 +3849,7 @@ def test_agent_session_send_a2a_message_posts_standard_contract(monkeypatch):
     assert request_body == {
         "message": {
             "messageId": "msg-00000000-0000-4000-8000-000000000001",
-            "role": "ROLE_USER",
+            "role": "ROLE_REQUESTER",
             "contextId": session_uid,
             "parts": [{"text": "What can this agent do?"}],
         },
@@ -3833,6 +3861,58 @@ def test_agent_session_send_a2a_message_posts_standard_contract(monkeypatch):
     assert "omit_reasoning" not in captured["runtime"]["data"]
 
 
+@pytest.mark.parametrize("invalid_role", [None, "ROLE_REQUESTER", "requester"])
+def test_a2a_message_result_requires_responder_role(invalid_role):
+    with pytest.raises(ValidationError, match="ROLE_RESPONDER"):
+        agent_models_mod.A2AMessageSendResult.model_validate(
+            {
+                "message": {
+                    "messageId": "msg-output",
+                    "role": invalid_role,
+                    "contextId": "session-uid",
+                    "parts": [{"text": "Result."}],
+                }
+            }
+        )
+
+
+def test_agent_session_send_rejects_mismatched_response_context(monkeypatch):
+    session_uid = "3f1cc452-43ec-49cb-b2ba-87dbac164d29"
+
+    class FakeResponse:
+        status_code = 200
+
+        @staticmethod
+        def json():
+            return {
+                "message": {
+                    "messageId": "msg-output",
+                    "role": "ROLE_RESPONDER",
+                    "contextId": "different-session",
+                    "parts": [{"text": "Result."}],
+                }
+            }
+
+    monkeypatch.setattr(
+        agent_models_mod.AgentSession,
+        "_resolve_runtime_access_for_message_send",
+        classmethod(lambda cls, agent_session, timeout=None: object()),
+    )
+    monkeypatch.setattr(
+        agent_models_mod.AgentSession,
+        "_post_standard_a2a_message",
+        classmethod(
+            lambda cls, access, *, body, timeout=None: FakeResponse()
+        ),
+    )
+
+    with pytest.raises(agent_models_mod.ApiError, match="contextId"):
+        agent_models_mod.AgentSession.send_a2a_message(
+            session_uid,
+            message="Return a result.",
+        )
+
+
 def test_agent_session_send_waits_only_while_runtime_interaction_is_transient(
     monkeypatch,
 ):
@@ -3887,7 +3967,7 @@ def test_agent_session_send_waits_only_while_runtime_interaction_is_transient(
             return {
                 "message": {
                     "messageId": "msg-output",
-                    "role": "ROLE_AGENT",
+                    "role": "ROLE_RESPONDER",
                     "contextId": session_uid,
                     "parts": [{"text": "Ready."}],
                 }
@@ -4170,7 +4250,7 @@ def test_agent_session_send_a2a_message_posts_strict_dictionary_contract(monkeyp
             return {
                 "message": {
                     "messageId": "msg-runtime-output",
-                    "role": "ROLE_AGENT",
+                    "role": "ROLE_RESPONDER",
                     "contextId": session_uid,
                     "parts": [{"text": '{"ok": true}'}],
                 }
@@ -4253,7 +4333,7 @@ def test_agent_session_send_a2a_message_refreshes_access_and_reuses_body(monkeyp
             return {
                 "message": {
                     "messageId": "msg-runtime-output",
-                    "role": "ROLE_AGENT",
+                    "role": "ROLE_RESPONDER",
                     "contextId": session_uid,
                     "parts": [{"text": "Done."}],
                 }
@@ -4329,7 +4409,7 @@ def test_agent_respond_uses_agent_scoped_sessionless_contract(monkeypatch):
                 "message": {
                     "kind": "message",
                     "messageId": "msg-output",
-                    "role": "ROLE_AGENT",
+                    "role": "ROLE_RESPONDER",
                     "parts": [{"data": {"answer": 42}, "mediaType": "application/json"}],
                 }
             }
@@ -4383,7 +4463,7 @@ def test_agent_respond_uses_agent_scoped_sessionless_contract(monkeypatch):
     body = runtime["body"]
     assert body["message"] == {
         "messageId": "msg-input",
-        "role": "ROLE_USER",
+        "role": "ROLE_REQUESTER",
         "parts": [{"text": "Return an answer."}],
     }
     assert "contextId" not in body["message"]
@@ -4460,7 +4540,7 @@ def test_agent_respond_waits_for_transient_runtime_interaction(monkeypatch):
             return {
                 "message": {
                     "messageId": "msg-output",
-                    "role": "ROLE_AGENT",
+                    "role": "ROLE_RESPONDER",
                     "parts": [{"text": "Ready."}],
                 }
             }