cohere-compass-sdk 2.2.2__tar.gz → 2.3.3__tar.gz

{cohere_compass_sdk-2.2.2 → cohere_compass_sdk-2.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cohere-compass-sdk
-Version: 2.2.2
+Version: 2.3.3
 Summary: Cohere Compass SDK
 Requires-Python: >=3.11,<4.0
 Classifier: Programming Language :: Python :: 3

{cohere_compass_sdk-2.2.2 → cohere_compass_sdk-2.3.3}/cohere_compass/clients/compass.py

@@ -635,6 +635,8 @@ class CompassClient:
         attributes: DocumentAttributes = DocumentAttributes(),
         config: ParseableDocumentConfig = ParseableDocumentConfig(),
         content_type: ContentTypeEnum | None = None,
+        authorized_groups: list[str] | None = None,
+        merge_groups_on_conflict: bool = False,
         max_retries: int | None = None,
         retry_wait: timedelta | None = None,
         timeout: timedelta | None = None,
@@ -650,6 +652,11 @@ class CompassClient:
         :param document_id: The ID to assign to the document.
         :param attributes: Additional attributes to add to the document.
         :param config: Configuration for the document parsing.
+        :param authorized_groups: When Document-level Security is enabled,
+            Set authorized user groups to access the document.
+            These groups should exist in RBAC. Default will make the document public.
+        :param merge_groups_on_conflict: When Document-level Security is enabled,
+            allows merging authorized groups if document already exists.
         :param max_retries: Maximum number of retries for failed requests. If not
             provided, the default from the client will be used.
         :param retry_wait: Time to wait between retries. If not provided, the default
@@ -674,7 +681,11 @@ class CompassClient:
 
         result = self._send_request(
             api_name="upload_documents",
-            data=UploadDocumentsInput(documents=[doc]),
+            data=UploadDocumentsInput(
+                documents=[doc],
+                authorized_groups=authorized_groups,
+                merge_groups_on_conflict=merge_groups_on_conflict,
+            ),
             index_name=index_name,
             max_retries=max_retries,
             retry_wait=retry_wait,

{cohere_compass_sdk-2.2.2 → cohere_compass_sdk-2.3.3}/cohere_compass/clients/compass_async.py

@@ -494,6 +494,8 @@ class CompassAsyncClient:
         attributes: DocumentAttributes = DocumentAttributes(),
         config: ParseableDocumentConfig = ParseableDocumentConfig(),
         content_type: ContentTypeEnum | None = None,
+        authorized_groups: list[str] | None = None,
+        merge_groups_on_conflict: bool = False,
         max_retries: int | None = None,
         retry_wait: timedelta | None = None,
         timeout: timedelta | None = None,
@@ -509,6 +511,11 @@ class CompassAsyncClient:
         :param document_id: the id of the document (optional)
         :param attributes: Additional attributes to add to the document.
         :param config: Configuration for the document parsing.
+        :param authorized_groups: The groups that are authorized to access the
+            document. These groups should exist in RBAC. None passed will make the
+            document public.
+        :param merge_groups_on_conflict: Allows combining authorized group field
+          when document already exists, and Document Level Security is enabled.
         :param max_retries: Maximum number of retries for failed requests. If not
             provided, the default from the client will be used.
         :param retry_wait: Time to wait between retries. If not provided, the default
@@ -533,7 +540,11 @@ class CompassAsyncClient:
 
         result = await self._send_request(
             api_name="upload_documents",
-            data=UploadDocumentsInput(documents=[doc]),
+            data=UploadDocumentsInput(
+                documents=[doc],
+                authorized_groups=authorized_groups,
+                merge_groups_on_conflict=merge_groups_on_conflict,
+            ),
             index_name=index_name,
             max_retries=max_retries,
             retry_wait=retry_wait,

{cohere_compass_sdk-2.2.2 → cohere_compass_sdk-2.3.3}/cohere_compass/models/documents.py

@@ -311,6 +311,8 @@ class UploadDocumentsInput(BaseModel):
     """A model for the input of a call to upload_documents API."""
 
     documents: list[ParseableDocument]
+    authorized_groups: list[str] | None = None
+    merge_groups_on_conflict: bool = False
 
 
 class UploadDocumentsResult(BaseModel):

cohere_compass_sdk-2.3.3/cohere_compass/utils/__init__.py

@@ -0,0 +1 @@
+"""Utility functions for the Cohere Compass SDK."""

{cohere_compass_sdk-2.2.2 → cohere_compass_sdk-2.3.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cohere-compass-sdk"
-version = "2.2.2"
+version = "2.3.3"
 authors = []
 description = "Cohere Compass SDK"
 readme = "README.md"