mainsequence 9.0.9.dev46__tar.gz → 9.0.10.dev47__tar.gz

{mainsequence-9.0.9.dev46 → mainsequence-9.0.10.dev47}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mainsequence
-Version: 9.0.9.dev46
+Version: 9.0.10.dev47
 Summary: Main Sequence SDK
 Author-email: Main Sequence GmbH <dev@main-sequence.io>
 License-Expression: MIT

{mainsequence-9.0.9.dev46 → mainsequence-9.0.10.dev47}/agent_scaffold/skills/mainsequence-inference/SKILL.md

@@ -46,6 +46,21 @@ context. Never invent an Environment, use another branch as a fallback, or add
 inference environment variables. Provider secrets remain in existing server-side
 credential storage and must not appear in messages or `provider_options`.
 
+## Select a configured provider
+
+Pass optional `custom_id="openai-work"` to `client.complete()` to select that named
+credential within the caller's Environment. Omit it to retain the first available
+owned credential, then the first shared credential. A missing or ambiguous explicit
+name fails; do not retry using a different credential. Recorded requests retain the
+exact credential UID for replay. Custom endpoints use their existing provider
+identifier and do not accept `custom_id`.
+
+Sharing a configured provider authorizes recipients, including workload users,
+to use its credentials and receive them in local Tau. They may copy those credentials;
+revoking a share cannot remove copies already delivered. Explain this before
+guiding someone through sharing. This inference client executes on the server;
+provider management remains with the existing provider clients.
+
 ## Construct a completion envelope
 
 Every call supplies complete ordered text messages: `system`, `user`, `assistant`.

{mainsequence-9.0.9.dev46 → mainsequence-9.0.10.dev47}/mainsequence/client/agent_runtime_models.py

@@ -727,6 +727,7 @@ class Agent(
         llm_provider: str | None = None,
         llm_model: str | None = None,
         llm_thinking: str | None = None,
+        custom_id: str | None = None,
         timeout=None,
     ) -> AgentSession:
         """
@@ -759,6 +760,10 @@ class Agent(
             "llm_provider": llm_provider,
             "llm_model": llm_model,
             "llm_thinking": llm_thinking,
+            "custom_id": (
+                custom_id
+                if custom_id is not None else None
+            ),
         }
         if resolved_session_uid and any(value is not None for value in creation_options.values()):
             raise ValueError("Creation options require handle_unique_id, not session_uid")
@@ -787,7 +792,14 @@ class Agent(
         session_payload = response.json()
         if not isinstance(session_payload, dict):
             raise TypeError("get_or_create_session response must be an AgentSession object")
-        return AgentSession(**session_payload)
+        session = AgentSession(**session_payload)
+        if (
+            custom_id is not None
+            and session.custom_id
+            != creation_options["custom_id"]
+        ):
+            raise ValueError("The returned session does not use the requested configured provider")
+        return session
 
 
 class AgentSessionInsightsBase(BasePydanticModel):
@@ -1745,6 +1757,10 @@ class AgentSession(
         None,
         description="Timestamp when the session was archived.",
     )
+    model_provider_credential_uid: str | None = None
+    custom_id: str | None = Field(
+        None, description="Resolved configured provider name; None for custom endpoints."
+    )
     llm_provider: str = Field(
         ...,
         description="Resolved LLM provider actually used for this session. Unlike Agent defaults, this is intended to be the authoritative runtime record.",

{mainsequence-9.0.9.dev46 → mainsequence-9.0.10.dev47}/mainsequence/client/inference.py

@@ -20,6 +20,8 @@ class InferenceResult(BaseModel):
     sequence: int
     provider: str
     model: str
+    custom_id: str | None = None
+    model_provider_credential_uid: str | None = None
     status: str
     error_code: str
     content_available: bool
@@ -125,6 +127,7 @@ class InferenceClient:
         messages: list[dict[str, str]],
         idempotency_key: str,
         conversation_uid: str | UUID | None = None,
+        custom_id: str | None = None,
         thinking_level: str | None = None,
         provider_options: dict[str, Any] | None = None,
         provider_storage: dict[str, Any] | None = None,
@@ -142,6 +145,8 @@ class InferenceClient:
                 "idempotency_key must contain 1–128 letters, digits, dots, underscores, colons or hyphens."
             )
         body = {"provider": provider, "model": model, "messages": deepcopy(messages)}
+        if custom_id is not None:
+            body["custom_id"] = custom_id
         if conversation_uid is not None:
             body["conversation_uid"] = str(UUID(str(conversation_uid)))
         for name, value in (
@@ -153,9 +158,15 @@ class InferenceClient:
         ):
             if value is not None:
                 body[name] = deepcopy(value)
-        return self._request(
+        result = self._request(
             "POST", "inference/completions/", body=body, key=idempotency_key, timeout=timeout
         )
+        if (
+            custom_id is not None
+            and result.custom_id != body["custom_id"]
+        ):
+            raise InferenceTransportError(idempotency_key)
+        return result
 
     def conversations(self, *, limit: int = 25, offset: int = 0) -> dict[str, Any]:
         return self._request(

{mainsequence-9.0.9.dev46 → mainsequence-9.0.10.dev47}/mainsequence.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mainsequence
-Version: 9.0.9.dev46
+Version: 9.0.10.dev47
 Summary: Main Sequence SDK
 Author-email: Main Sequence GmbH <dev@main-sequence.io>
 License-Expression: MIT

{mainsequence-9.0.9.dev46 → mainsequence-9.0.10.dev47}/pyproject.toml

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

{mainsequence-9.0.9.dev46 → mainsequence-9.0.10.dev47}/tests/test_inference_client.py

@@ -165,3 +165,33 @@ def test_invalid_success_is_uncertain_and_never_retried(transport, invalid):
     assert exc.value.idempotency_key == "stable"
     assert session.request.call_count == 1
     assert session.request.call_args.kwargs["allow_redirects"] is False
+
+
+@pytest.mark.parametrize("subscription_uid", [None, "11111111-1111-4111-8111-111111111111"])
+def test_completion_subscription_is_optional_and_preserves_request_identity(transport, subscription_uid):
+    session, response, _ = transport
+    response.json.return_value = result(custom_id=subscription_uid)
+    answer = inference.InferenceClient().complete(
+        provider="openai", model="configured-model", messages=[], idempotency_key="stable",
+        custom_id=subscription_uid,
+    )
+    body = session.request.call_args.kwargs["json"]
+    assert answer.custom_id == subscription_uid
+    assert session.request.call_args.kwargs["headers"]["Idempotency-Key"] == "stable"
+    if subscription_uid is None:
+        assert "custom_id" not in body
+    else:
+        assert body["custom_id"] == subscription_uid
+
+
+@pytest.mark.parametrize("resolved", [None, "22222222-2222-4222-8222-222222222222"])
+def test_completion_does_not_accept_ignored_or_different_subscription(transport, resolved):
+    session, response, _ = transport
+    response.json.return_value = result(custom_id=resolved)
+    with pytest.raises(inference.InferenceTransportError) as error:
+        inference.InferenceClient().complete(
+            provider="openai", model="configured-model", messages=[], idempotency_key="stable",
+            custom_id="11111111-1111-4111-8111-111111111111",
+        )
+    assert error.value.idempotency_key == "stable"
+    assert session.request.call_count == 1