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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: assistant_runtime_sdk
-Version: 1.11.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.11.0 → assistant_runtime_sdk-1.12.0}/assistant_runtime_sdk/__init__.py

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

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

@@ -608,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,
@@ -621,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.11.0 → assistant_runtime_sdk-1.12.0}/assistant_runtime_sdk/base.py

@@ -609,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:
@@ -625,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'"
             )
@@ -645,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",
@@ -660,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),
@@ -669,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.11.0 → assistant_runtime_sdk-1.12.0}/assistant_runtime_sdk/client.py

@@ -682,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.
@@ -699,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,
@@ -707,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,
@@ -715,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.
@@ -728,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.11.0 → assistant_runtime_sdk-1.12.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "assistant_runtime_sdk"
-version = "1.11.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"}