unique_sdk 0.10.8__py3-none-any.whl → 0.10.9__py3-none-any.whl

unique_sdk/utils/chat_in_space.py

@@ -139,12 +139,12 @@ async def wait_for_ingestion_completion(
     user_id: str,
     company_id: str,
     content_id: str,
-    chat_id: str,
+    chat_id: str = None,
     poll_interval: float = 1.0,
     max_wait: float = 60.0,
 ):
     """
-    Waits until the content ingestionState is FINISHED or the maximum wait time is reached and throws if case of failed status.
+    Polls until the content ingestion is finished or the maximum wait time is reached and throws in case ingestion fails. The function assumes that the content exists.
     """
     max_attempts = int(max_wait // poll_interval)
     for _ in range(max_attempts):
@@ -158,7 +158,7 @@ async def wait_for_ingestion_completion(
         if searched_content:
             ingestion_state = searched_content[0].get("ingestionState")
             if ingestion_state == "FINISHED":
-                return
+                return ingestion_state
             if isinstance(ingestion_state, str) and ingestion_state.startswith(
                 "FAILED"
             ):

{unique_sdk-0.10.8.dist-info → unique_sdk-0.10.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_sdk
-Version: 0.10.8
+Version: 0.10.9
 Summary: 
 License: MIT
 Author: Martin Fadler
@@ -1447,6 +1447,30 @@ latest_message = await unique_sdk.utils.chat_in_space.chat_against_file(
 )
 ```
 
+#### `unique_sdk.utils.chat_in_space.wait_for_ingestion_completion`
+
+The following script enables you to wait for the ingestion of a file. This should be used carefully as it continuously polls for the status. In case of bigger files, adjust the `poll_interval` and `max_waits`.
+
+You must provide the following parameter:
+- `content_id`: The id of the content to check.
+
+The script polls until the content ingestion is finished or the maximum wait time is reached and throws in case ingestion fails. The function assumes that the content exists.
+
+**Optional parameters:**
+- `chat_id`: In case the content is uploaded to a chat, the `chat_id` must be provided.
+- `poll_interval`: The number of seconds to wait between polling attempts (default: `1` second).
+- `max_wait`: The maximum number of seconds to wait for the message to complete (default: `60` seconds).
+
+Example of waiting for the ingestion of a file in the Knowledge Base.
+
+```python
+await unique_sdk.utils.chat_in_space.wait_for_ingestion_completion(
+    user_id=user_id,
+    company_id=company_id,
+    content_id="cont_ddlezvag4kzxudfr24lrjc5mx",
+)
+```
+
 ## Error Handling
 
 ## Examples
@@ -1464,6 +1488,9 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), 
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.10.9] - 2025-08-15
+- Add script to wait for content ingestion finished.
+
 ## [0.10.8] - 2025-08-13
 - Add support for Agentic Table.
 

{unique_sdk-0.10.8.dist-info → unique_sdk-0.10.9.dist-info}/RECORD

@@ -30,11 +30,11 @@ unique_sdk/api_resources/_search_string.py,sha256=4Idw6exgZdA8qksz9WkiA68k1hTU-7
 unique_sdk/api_resources/_short_term_memory.py,sha256=vPRN-Y0WPx74E6y-A3LocGc0TxJdzT-xGL66WzZwKRg,2820
 unique_sdk/api_resources/_space.py,sha256=Mzl1EUn7RrfjB8fKC6f-TW9GAOl5M6JncYLrKMDa-pQ,4933
 unique_sdk/utils/chat_history.py,sha256=5UqL9hF1O9pV7skbNOlEibF5rHdYsmG3m5-YEPUowOs,3037
-unique_sdk/utils/chat_in_space.py,sha256=nYmgQYhIxqQex_6zjdCfNuGnjoU14WkxUN6_zmSOiCQ,5169
+unique_sdk/utils/chat_in_space.py,sha256=3NeBjOu7p43V_6PrjwxyaTkgknUS10KE4QRuTlFDU_4,5232
 unique_sdk/utils/file_io.py,sha256=YY8B7VJcTLOPmCXByiOfNerXGlAtjCC5EVNmAbQJ3dQ,4306
 unique_sdk/utils/sources.py,sha256=wfboE-neMKa0Wuq9QzfAEFMkNLrIrmm0v-QF33sLo6k,4952
 unique_sdk/utils/token.py,sha256=AzKuAA1AwBtnvSFxGcsHLpxXr_wWE5Mj4jYBbOz2ljA,1740
-unique_sdk-0.10.8.dist-info/LICENSE,sha256=EJCWoHgrXVBUb47PnjeV4MFIEOR71MAdCOIgv61J-4k,1065
-unique_sdk-0.10.8.dist-info/METADATA,sha256=4BWxqpjZI_3hk-_ZWK4IIuhRn89GC8lsEQWPkprhB94,49203
-unique_sdk-0.10.8.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_sdk-0.10.8.dist-info/RECORD,,
+unique_sdk-0.10.9.dist-info/LICENSE,sha256=EJCWoHgrXVBUb47PnjeV4MFIEOR71MAdCOIgv61J-4k,1065
+unique_sdk-0.10.9.dist-info/METADATA,sha256=HkLlEA2qV_xJPt5i7obS1trzOhzdZL0aEw4mexrODP4,50398
+unique_sdk-0.10.9.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_sdk-0.10.9.dist-info/RECORD,,