assistant-runtime-sdk 1.9.0__tar.gz → 1.11.0__tar.gz

{assistant_runtime_sdk-1.9.0 → assistant_runtime_sdk-1.11.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: assistant_runtime_sdk
-Version: 1.9.0
+Version: 1.11.0
 Summary: Python SDK for FAC Cloud - the Assistant Runtime AI assistant backend
 Keywords: frappe,ai,assistant,llm,mcp,anthropic,openai,streaming
 Author: Paul Clinton
@@ -158,7 +158,8 @@ surface. A representative selection:
 
 **Billing** — `get_plan_comparison()`, `get_usage_dashboard()`, `get_usage_history()`,
 `get_credit_balance()`, `initiate_checkout()`, `create_hosted_checkout()`,
-`verify_checkout()`, `upgrade_plan()`, `cancel_subscription()`, `get_invoices()`,
+`get_checkout_session_status()`, `verify_checkout()`, `upgrade_plan()`,
+`cancel_subscription()`, `get_invoices()`,
 `get_payment_instrument()`, `update_payment_method()`
 
 **Users & seats** — `register_user()`, `get_user()`, `list_users()`, `invite_user()`,

{assistant_runtime_sdk-1.9.0 → assistant_runtime_sdk-1.11.0}/README.md

@@ -125,7 +125,8 @@ surface. A representative selection:
 
 **Billing** — `get_plan_comparison()`, `get_usage_dashboard()`, `get_usage_history()`,
 `get_credit_balance()`, `initiate_checkout()`, `create_hosted_checkout()`,
-`verify_checkout()`, `upgrade_plan()`, `cancel_subscription()`, `get_invoices()`,
+`get_checkout_session_status()`, `verify_checkout()`, `upgrade_plan()`,
+`cancel_subscription()`, `get_invoices()`,
 `get_payment_instrument()`, `update_payment_method()`
 
 **Users & seats** — `register_user()`, `get_user()`, `list_users()`, `invite_user()`,

{assistant_runtime_sdk-1.9.0 → assistant_runtime_sdk-1.11.0}/assistant_runtime_sdk/__init__.py

@@ -53,7 +53,7 @@ Standalone Functions:
 
 """
 
-__version__ = "1.9.0"
+__version__ = "1.11.0"
 __author__ = "Paul Clinton"
 __license__ = "AGPL-3.0"
 
@@ -61,7 +61,14 @@ __license__ = "AGPL-3.0"
 # Core Client Classes
 # =============================================================================
 
-from .client import AssistantRuntimeClient, get_terms, register_tenant, get_registration_state
+from .client import (
+    AssistantRuntimeClient,
+    change_pending_owner_email,
+    get_registration_state,
+    get_terms,
+    register_tenant,
+    resend_owner_verification,
+)
 
 # Async client - import lazily to avoid requiring aiohttp
 # Skill providers - import lazily to avoid requiring strands-agents
@@ -173,6 +180,8 @@ __all__ = [
     "get_terms",
     "register_tenant",
     "get_registration_state",
+    "resend_owner_verification",
+    "change_pending_owner_email",
     # Exceptions - new names
     "ARError",
     "ARAuthenticationError",

{assistant_runtime_sdk-1.9.0 → assistant_runtime_sdk-1.11.0}/assistant_runtime_sdk/async_client.py

@@ -384,6 +384,7 @@ class AsyncAssistantRuntimeClient(BaseAssistantRuntimeClient):
         *,
         web_search: Optional[bool] = None,
         thinking_enabled: Optional[bool] = None,
+        reasoning_effort: Optional[str] = None,
     ) -> AsyncGenerator[Dict[str, Any], None]:
         """Async version of AssistantRuntimeClient.stream_chat.
 
@@ -400,6 +401,7 @@ class AsyncAssistantRuntimeClient(BaseAssistantRuntimeClient):
             continue_from_message_id=continue_from_message_id,
             web_search=web_search,
             thinking_enabled=thinking_enabled,
+            reasoning_effort=reasoning_effort,
         )
         url = self._build_endpoint_url("streaming.stream_chat")
         payload = self._with_site_url(payload)
@@ -947,6 +949,11 @@ class AsyncAssistantRuntimeClient(BaseAssistantRuntimeClient):
         endpoint, payload = self._prepare_create_hosted_checkout(purpose, params, return_url)
         return await self._request_post_json(endpoint, payload, api_base=self.billing_api_base)
 
+    async def get_checkout_session_status(self, session: str) -> Optional[Dict[str, Any]]:
+        """Async version of AssistantRuntimeClient.get_checkout_session_status."""
+        endpoint, payload = self._prepare_get_checkout_session_status(session)
+        return await self._request_post_json(endpoint, payload, api_base=self.billing_api_base)
+
     async def verify_checkout(self, session_id: Optional[str] = None) -> Optional[Dict[str, Any]]:
         """Async version of AssistantRuntimeClient.verify_checkout."""
         endpoint, payload = self._prepare_verify_checkout(session_id)

{assistant_runtime_sdk-1.9.0 → assistant_runtime_sdk-1.11.0}/assistant_runtime_sdk/base.py

@@ -316,6 +316,7 @@ class BaseAssistantRuntimeClient:
         continue_from_message_id: Optional[str] = None,
         web_search: Optional[bool] = None,
         thinking_enabled: Optional[bool] = None,
+        reasoning_effort: Optional[str] = None,
     ) -> Dict[str, Any]:
         """
         Prepare JSON payload for stream_chat POST request.
@@ -347,6 +348,8 @@ class BaseAssistantRuntimeClient:
                 previous truncated response from
             web_search: Optional explicit web-search toggle for this turn
             thinking_enabled: Optional explicit extended-thinking toggle for this turn
+            reasoning_effort: Optional thinking level for this turn — "off", "low", "medium",
+                "high", "xhigh" or "max"
 
         Returns:
             Payload dict ready for JSON body
@@ -400,6 +403,9 @@ class BaseAssistantRuntimeClient:
         if thinking_enabled is not None:
             payload["thinking_enabled"] = bool(thinking_enabled)
 
+        if reasoning_effort is not None:
+            payload["reasoning_effort"] = str(reasoning_effort).strip().lower()
+
         return payload
 
     def _prepare_resource_params(
@@ -987,6 +993,15 @@ class BaseAssistantRuntimeClient:
             payload["return_url"] = return_url
         return "create_hosted_checkout", payload
 
+    def _prepare_get_checkout_session_status(self, session: str) -> tuple:
+        self._require_billing()
+        if not session:
+            raise ValueError("session is required")
+        return "get_checkout_session_status", {
+            "tenant_id": self.tenant_id,
+            "session": session,
+        }
+
     def _prepare_verify_checkout(self, session_id: Optional[str] = None) -> tuple:
         self._require_billing()
         payload: Dict[str, Any] = {"tenant_id": self.tenant_id}

{assistant_runtime_sdk-1.9.0 → assistant_runtime_sdk-1.11.0}/assistant_runtime_sdk/client.py

@@ -329,6 +329,7 @@ class AssistantRuntimeClient(BaseAssistantRuntimeClient):
         *,
         web_search: Optional[bool] = None,
         thinking_enabled: Optional[bool] = None,
+        reasoning_effort: Optional[str] = None,
     ) -> Generator[Dict[str, Any], None, None]:
         """
         Stream chat response from Assistant Runtime.
@@ -361,6 +362,9 @@ class AssistantRuntimeClient(BaseAssistantRuntimeClient):
                 truncated response from
             web_search: Optional explicit web-search toggle for this turn
             thinking_enabled: Optional explicit extended-thinking toggle for this turn
+            reasoning_effort: Optional thinking level for this turn — "off", "low", "medium",
+                "high", "xhigh" or "max". Servers that predate it ignore it, so callers
+                should also send thinking_enabled.
 
         Yields:
             Parsed SSE events with structure:
@@ -390,6 +394,7 @@ class AssistantRuntimeClient(BaseAssistantRuntimeClient):
             continue_from_message_id=continue_from_message_id,
             web_search=web_search,
             thinking_enabled=thinking_enabled,
+            reasoning_effort=reasoning_effort,
         )
         url = self._build_endpoint_url("streaming.stream_chat")
         payload = self._with_site_url(payload)
@@ -1394,6 +1399,27 @@ class AssistantRuntimeClient(BaseAssistantRuntimeClient):
         endpoint, payload = self._prepare_create_hosted_checkout(purpose, params, return_url)
         return self._request_post_json(endpoint, payload, api_base=self.billing_api_base)
 
+    def get_checkout_session_status(self, session: str) -> Optional[Dict[str, Any]]:
+        """
+        Ask whether the purchase a hosted checkout was opened for has landed.
+
+        The checkout page returns the user with ``fac_checkout=<session>``
+        (and ``result=success|processing|cancelled|failed``) on the query
+        string. Poll this with that session until ``done`` is true. It only
+        reads, so polling it never touches the payment gateway.
+
+        Args:
+            session: The ``fac_checkout`` value from the return URL — the
+                ``session`` that :meth:`create_hosted_checkout` returned.
+
+        Returns:
+            Dict with status, purpose, target_plan, plan, pending_plan,
+            payment_status, done (bool) and outcome — one of "applied",
+            "processing", "failed" or "cancelled".
+        """
+        endpoint, payload = self._prepare_get_checkout_session_status(session)
+        return self._request_post_json(endpoint, payload, api_base=self.billing_api_base)
+
     def verify_checkout(self, session_id: Optional[str] = None) -> Optional[Dict[str, Any]]:
         """Verify payment completion after checkout."""
         endpoint, payload = self._prepare_verify_checkout(session_id)
@@ -3308,7 +3334,11 @@ def register_tenant(
             registration cap exactly once.
 
     Returns:
-        Registration result with tenant_id (secret delivered via email link), or error
+        Registration result with tenant_id (secret delivered via email link), or
+        error. A newly created pending tenant also carries ``pending_token``,
+        returned only this once: store it securely, because
+        ``resend_owner_verification`` and ``change_pending_owner_email``
+        require it.
     """
     url = f"{ar_url.rstrip('/')}/api/method/assistant_runtime.api.register_tenant"
 
@@ -3368,6 +3398,70 @@ def get_registration_state(
         return {"error": str(e)}
 
 
+def _post_guest(ar_url: str, method: str, payload: Dict[str, Any], timeout: int = 30) -> Dict[str, Any]:
+    """POST to a guest AR method, returning AR's own message when it fails.
+
+    Failures come back as ``{"error", "status_code", "exc_type"}``. ``error`` is
+    the message AR raised when the body carries one, so a caller can show it
+    instead of ``str(HTTPError)``; ``exc_type`` is set only in that case.
+    """
+    url = f"{ar_url.rstrip('/')}/api/method/assistant_runtime.api.{method}"
+    try:
+        response = requests.post(
+            url, json=payload, headers={"Content-Type": "application/json"}, timeout=timeout,
+        )
+        response.raise_for_status()
+        return response.json().get("message", response.json())
+    except requests.exceptions.HTTPError as e:
+        status_code = e.response.status_code if e.response is not None else None
+        try:
+            data = e.response.json() if e.response is not None else None
+        except ValueError:
+            data = None
+        message = BaseAssistantRuntimeClient._extract_error_from_data(data) if isinstance(data, dict) else None
+        if not message:
+            return {"error": str(e), "status_code": status_code}
+        return {"error": message, "status_code": status_code, "exc_type": data.get("exc_type")}
+    except requests.exceptions.RequestException as e:
+        return {"error": str(e), "status_code": None}
+
+
+def resend_owner_verification(ar_url: str, site_url: str, pending_token: str) -> Dict[str, Any]:
+    """Ask AR to re-send the pending verification link. No terms, no secret.
+
+    Args:
+        pending_token: The token ``register_tenant`` returned when this site's
+            tenant was created. AR refuses the call without it.
+
+    Returns:
+        AR's result (``success``, ``retry_after``, ``already_verified``,
+        ``owner_email_masked``), or ``{"error", "status_code", "exc_type"}``.
+    """
+    return _post_guest(
+        ar_url, "resend_owner_verification", {"site_url": site_url, "pending_token": pending_token}
+    )
+
+
+def change_pending_owner_email(
+    ar_url: str, site_url: str, owner_email: str, pending_token: str
+) -> Dict[str, Any]:
+    """Correct the owner address while the tenant is still unverified.
+
+    Args:
+        pending_token: The token ``register_tenant`` returned when this site's
+            tenant was created. AR refuses the call without it.
+
+    Returns:
+        AR's result (``success``, ``retry_after``, ``owner_email_masked``), or
+        ``{"error", "status_code", "exc_type"}``.
+    """
+    return _post_guest(
+        ar_url,
+        "change_pending_owner_email",
+        {"site_url": site_url, "owner_email": owner_email, "pending_token": pending_token},
+    )
+
+
 def get_initial_secret(ar_url: str, verification_token: str) -> Dict[str, Any]:
     """One-shot retrieval of the freshly-minted tenant_secret after the
     owner has confirmed their email via the verification link.

{assistant_runtime_sdk-1.9.0 → assistant_runtime_sdk-1.11.0}/assistant_runtime_sdk/types.py

@@ -16,7 +16,15 @@ from typing import TypedDict, List, Optional, Any
 # =============================================================================
 
 
-class ModelInfo(TypedDict):
+class _ModelInfoOptional(TypedDict, total=False):
+    """Keys newer runtimes add to each model; absent on older ones."""
+
+    thinking_effective: bool
+    reasoning_levels: List[str]
+    reasoning_off_floor: bool
+
+
+class ModelInfo(_ModelInfoOptional):
     """Information about an available AI model."""
 
     model_id: str
@@ -104,7 +112,7 @@ class RoutingReceiptData(TypedDict):
     shortlist_size: int
     pick_reason: str
     notices: List[str]
-    thinking: dict                     # requested, applied, effort, not_applied_reason
+    thinking: dict                     # requested, applied, effort {requested, applied}, effective_budget, not_applied_reason
     credits: dict                      # actual
     cycles: int
     also_ran: List[str]

{assistant_runtime_sdk-1.9.0 → assistant_runtime_sdk-1.11.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "assistant_runtime_sdk"
-version = "1.9.0"
+version = "1.11.0"
 description = "Python SDK for FAC Cloud - the Assistant Runtime AI assistant backend"
 readme = "README.md"
 license = {text = "AGPL-3.0"}