assistant-runtime-sdk 1.10.0__tar.gz → 1.12.0__tar.gz

{assistant_runtime_sdk-1.10.0 → assistant_runtime_sdk-1.12.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: assistant_runtime_sdk
-Version: 1.10.0
+Version: 1.12.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

{assistant_runtime_sdk-1.10.0 → assistant_runtime_sdk-1.12.0}/assistant_runtime_sdk/__init__.py

@@ -53,7 +53,7 @@ Standalone Functions:
 
 """
 
-__version__ = "1.10.0"
+__version__ = "1.12.0"
 __author__ = "Paul Clinton"
 __license__ = "AGPL-3.0"
 

{assistant_runtime_sdk-1.10.0 → assistant_runtime_sdk-1.12.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)
@@ -606,11 +608,13 @@ class AsyncAssistantRuntimeClient(BaseAssistantRuntimeClient):
         user_id: Optional[str] = None,
         visibility: Optional[str] = None,
         shared_with: Optional[List[str]] = None,
+        project_id: Optional[str] = None,
     ) -> Optional[Dict[str, Any]]:
         """Async version of AssistantRuntimeClient.upload_document."""
         endpoint, params, f_field, f_name, f_data, c_type = self._prepare_upload_document(
             file_path, file_data, file_name, content_type,
             user_id=user_id, visibility=visibility, shared_with=shared_with,
+            project_id=project_id,
         )
         return await self._request_post_multipart(
             endpoint, params=params, file_field=f_field,
@@ -619,12 +623,27 @@ class AsyncAssistantRuntimeClient(BaseAssistantRuntimeClient):
         )
 
     async def list_documents(
-        self, limit: int = 50, offset: int = 0, user_id: Optional[str] = None
+        self,
+        limit: int = 50,
+        offset: int = 0,
+        user_id: Optional[str] = None,
+        project_id: Optional[str] = None,
+        include_project_docs: bool = False,
     ) -> Optional[Dict[str, Any]]:
         """Async version of AssistantRuntimeClient.list_documents."""
-        endpoint, params = self._prepare_list_documents(limit, offset, user_id=user_id)
+        endpoint, params = self._prepare_list_documents(
+            limit, offset, user_id=user_id,
+            project_id=project_id, include_project_docs=include_project_docs,
+        )
         return await self._request_get(endpoint, params, api_base=self.memory_api_base)
 
+    async def unlink_project_documents(
+        self, user_id: str, project_id: str
+    ) -> Optional[Dict[str, Any]]:
+        """Async version of AssistantRuntimeClient.unlink_project_documents."""
+        endpoint, payload = self._prepare_unlink_project_documents(user_id, project_id)
+        return await self._request_post_json(endpoint, payload, api_base=self.memory_api_base)
+
     async def get_document(self, document_id: str) -> Optional[Dict[str, Any]]:
         """Async version of AssistantRuntimeClient.get_document."""
         endpoint, params = self._prepare_get_document(document_id)

{assistant_runtime_sdk-1.10.0 → assistant_runtime_sdk-1.12.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(
@@ -603,11 +609,14 @@ class BaseAssistantRuntimeClient:
         user_id: Optional[str] = None,
         visibility: Optional[str] = None,
         shared_with: Optional[List[str]] = None,
+        project_id: Optional[str] = None,
     ) -> tuple:
         """Returns (endpoint, params, file_field, file_name, file_data, content_type)."""
         import mimetypes as _mt
         import os as _os
 
+        if project_id and not user_id:
+            raise ARConfigurationError("user_id is required with project_id")
         if file_path and file_data:
             raise ARConfigurationError("Provide either file_path or file_data, not both")
         if not file_path and not file_data:
@@ -619,7 +628,7 @@ class BaseAssistantRuntimeClient:
             raise ARConfigurationError(
                 f"Invalid visibility: {visibility}. Must be public, private, or shared."
             )
-        if visibility == "shared" and not shared_with:
+        if not project_id and visibility == "shared" and not shared_with:
             raise ARConfigurationError(
                 "shared_with is required when visibility is 'shared'"
             )
@@ -639,10 +648,15 @@ class BaseAssistantRuntimeClient:
         params: Dict[str, Any] = {"tenant_id": self.tenant_id}
         if user_id:
             params["user_id"] = user_id
-        if visibility:
-            params["visibility"] = visibility
-        if shared_with:
-            params["shared_with"] = json.dumps(shared_with)
+        if project_id:
+            # The server forces project documents to private, so no visibility
+            # or sharing is sent alongside a project_id.
+            params["project_id"] = project_id
+        else:
+            if visibility:
+                params["visibility"] = visibility
+            if shared_with:
+                params["shared_with"] = json.dumps(shared_with)
 
         return (
             "documents.upload_document",
@@ -654,8 +668,15 @@ class BaseAssistantRuntimeClient:
         )
 
     def _prepare_list_documents(
-        self, limit: int = 50, offset: int = 0, user_id: Optional[str] = None
+        self,
+        limit: int = 50,
+        offset: int = 0,
+        user_id: Optional[str] = None,
+        project_id: Optional[str] = None,
+        include_project_docs: bool = False,
     ) -> tuple:
+        if project_id and not user_id:
+            raise ARConfigurationError("user_id is required with project_id")
         params: Dict[str, Any] = {
             "tenant_id": self.tenant_id,
             "limit": str(limit),
@@ -663,8 +684,20 @@ class BaseAssistantRuntimeClient:
         }
         if user_id:
             params["user_id"] = user_id
+        if project_id:
+            params["project_id"] = project_id
+        if include_project_docs:
+            params["include_project_docs"] = "1"
         return "documents.list_documents", params
 
+    def _prepare_unlink_project_documents(self, user_id: str, project_id: str) -> tuple:
+        """Returns (endpoint, payload) for unlinking a project's documents."""
+        return "documents.unlink_project_documents", {
+            "tenant_id": self.tenant_id,
+            "user_id": user_id,
+            "project_id": project_id,
+        }
+
     def _prepare_get_document(self, document_id: str) -> tuple:
         return "documents.get_document", {
             "tenant_id": self.tenant_id,

{assistant_runtime_sdk-1.10.0 → assistant_runtime_sdk-1.12.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)
@@ -677,6 +682,7 @@ class AssistantRuntimeClient(BaseAssistantRuntimeClient):
         user_id: Optional[str] = None,
         visibility: Optional[str] = None,
         shared_with: Optional[List[str]] = None,
+        project_id: Optional[str] = None,
     ) -> Optional[Dict[str, Any]]:
         """
         Upload a document for RAG processing.
@@ -694,6 +700,9 @@ class AssistantRuntimeClient(BaseAssistantRuntimeClient):
                 tenant configuration if omitted.
             shared_with: List of user IDs to share with. Required when
                 visibility is "shared".
+            project_id: Attach the document to a project. Project documents are
+                always private to the uploader, so visibility and shared_with
+                are not sent.
 
         Returns:
             {"status": "queued", "document_id": str, "file_name": str,
@@ -702,6 +711,7 @@ class AssistantRuntimeClient(BaseAssistantRuntimeClient):
         endpoint, params, f_field, f_name, f_data, c_type = self._prepare_upload_document(
             file_path, file_data, file_name, content_type,
             user_id=user_id, visibility=visibility, shared_with=shared_with,
+            project_id=project_id,
         )
         return self._request_post_multipart(
             endpoint, params=params, file_field=f_field,
@@ -710,7 +720,12 @@ class AssistantRuntimeClient(BaseAssistantRuntimeClient):
         )
 
     def list_documents(
-        self, limit: int = 50, offset: int = 0, user_id: Optional[str] = None
+        self,
+        limit: int = 50,
+        offset: int = 0,
+        user_id: Optional[str] = None,
+        project_id: Optional[str] = None,
+        include_project_docs: bool = False,
     ) -> Optional[Dict[str, Any]]:
         """
         List RAG documents for this tenant.
@@ -723,13 +738,38 @@ class AssistantRuntimeClient(BaseAssistantRuntimeClient):
             limit: Maximum number of documents to return (default 50).
             offset: Pagination offset (default 0).
             user_id: User performing the query. Enables visibility filtering.
+            project_id: Only list documents attached to this project.
+            include_project_docs: Also include documents attached to projects.
 
         Returns:
             {"documents": [...], "pagination": {...}, "storage": {...}}
         """
-        endpoint, params = self._prepare_list_documents(limit, offset, user_id=user_id)
+        endpoint, params = self._prepare_list_documents(
+            limit, offset, user_id=user_id,
+            project_id=project_id, include_project_docs=include_project_docs,
+        )
         return self._request_get(endpoint, params, api_base=self.memory_api_base)
 
+    def unlink_project_documents(
+        self, user_id: str, project_id: str
+    ) -> Optional[Dict[str, Any]]:
+        """
+        Unlink a project's documents from the project.
+
+        Documents attached to a project are removed from that project only;
+        the documents themselves are kept.
+
+        Args:
+            user_id: User performing the unlink.
+            project_id: The project whose documents should be unlinked.
+
+        Returns:
+            ``{"status": "unlinked", "count": int}``, where count is the number of
+            documents released from the project.
+        """
+        endpoint, payload = self._prepare_unlink_project_documents(user_id, project_id)
+        return self._request_post_json(endpoint, payload, api_base=self.memory_api_base)
+
     def get_document(self, document_id: str) -> Optional[Dict[str, Any]]:
         """
         Get detailed information about a specific RAG document.

{assistant_runtime_sdk-1.10.0 → assistant_runtime_sdk-1.12.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.10.0 → assistant_runtime_sdk-1.12.0}/pyproject.toml

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