unique_sdk 0.10.0__py3-none-any.whl → 0.10.2__py3-none-any.whl

unique_sdk/api_resources/_content.py

@@ -64,6 +64,7 @@ class Content(APIResource["Content"]):
     class SearchParams(RequestOptions):
         where: "Content.ContentWhereInput"
         chatId: NotRequired[str]
+        includeFailedContent: NotRequired[bool]
 
     class ContentInfoParams(TypedDict):
         """
@@ -163,11 +164,11 @@ class Content(APIResource["Content"]):
     class MagicTableSheetTable(TypedDict):
         rowId: str
         columns: List["Content.MagicTableSheetTableColumn"]
-    
+
     class MagicTableSheetIngestionConfiguration(TypedDict):
         columnIdsInMetadata: List[str]
         columnIdsInChunkText: List[str]
-    
+
     class MagicTableSheetIngestParams(TypedDict):
         data: List["Content.MagicTableSheetTable"]
         ingestionConfiguration: "Content.MagicTableSheetIngestionConfiguration"
@@ -298,10 +299,10 @@ class Content(APIResource["Content"]):
 
     @classmethod
     def ingest_magic_table_sheets(
-            cls, 
-            user_id: str, 
-            company_id: str, 
-            **params: Unpack["Content.MagicTableSheetIngestParams"]
+        cls,
+        user_id: str,
+        company_id: str,
+        **params: Unpack["Content.MagicTableSheetIngestParams"],
     ) -> "Content.MagicTableSheetResponse":
         return cast(
             Content.MagicTableSheetResponse,
@@ -316,10 +317,10 @@ class Content(APIResource["Content"]):
 
     @classmethod
     async def ingest_magic_table_sheets_async(
-            cls, 
-            user_id: str, 
-            company_id: str, 
-            **params: Unpack["Content.MagicTableSheetIngestParams"]
+        cls,
+        user_id: str,
+        company_id: str,
+        **params: Unpack["Content.MagicTableSheetIngestParams"],
     ) -> "Content.MagicTableSheetResponse":
         return cast(
             Content.MagicTableSheetResponse,

unique_sdk/api_resources/_space.py

@@ -67,6 +67,13 @@ class Space(APIResource["Space"]):
         assessment: Optional[List["Space.Reference"]]
         messageAssessment: Optional[List["Space.Assessment"]]
 
+    class DeleteChatResponse(TypedDict):
+        """
+        Response for deleting a chat in a space.
+        """
+
+        chat_id: str
+
     @classmethod
     def create_message(
         cls,
@@ -142,3 +149,43 @@ class Space(APIResource["Space"]):
                 company_id,
             ),
         )
+
+    @classmethod
+    def delete_chat(
+        cls,
+        user_id: str,
+        company_id: str,
+        chat_id: str,
+    ) -> "Space.DeleteChatResponse":
+        """
+        Delete a chat in a space.
+        """
+        return cast(
+            "Space.DeleteChatResponse",
+            cls._static_request(
+                "delete",
+                f"/space/chat/{chat_id}",
+                user_id,
+                company_id,
+            ),
+        )
+
+    @classmethod
+    async def delete_chat_async(
+        cls,
+        user_id: str,
+        company_id: str,
+        chat_id: str,
+    ) -> "Space.DeleteChatResponse":
+        """
+        Async delete a chat in a space.
+        """
+        return cast(
+            "Space.DeleteChatResponse",
+            await cls._static_request_async(
+                "delete",
+                f"/space/chat/{chat_id}",
+                user_id,
+                company_id,
+            ),
+        )

unique_sdk/utils/chat_in_space.py

@@ -1,7 +1,9 @@
 import asyncio
 from typing import List
 
+from unique_sdk.api_resources._content import Content
 from unique_sdk.api_resources._space import Space
+from unique_sdk.utils.file_io import upload_file
 
 
 async def send_message_and_wait_for_completion(
@@ -30,7 +32,6 @@ async def send_message_and_wait_for_completion(
     Returns:
         The completed Space.Message.
     """
-    # Send the prompt asynchronously
     response = await Space.create_message_async(
         user_id=user_id,
         company_id=company_id,
@@ -44,10 +45,123 @@ async def send_message_and_wait_for_completion(
 
     max_attempts = int(max_wait // poll_interval)
     for _ in range(max_attempts):
-        # Poll for the answer
         answer = Space.get_latest_message(user_id, company_id, chat_id)
         if answer.get("stoppedStreamingAt") is not None:
             return answer
         await asyncio.sleep(poll_interval)
 
     raise TimeoutError("Timed out waiting for prompt completion.")
+
+
+async def chat_against_file(
+    user_id: str,
+    company_id: str,
+    assistant_id: str,
+    path_to_file: str,
+    displayed_filename: str,
+    mime_type: str,
+    text: str,
+    poll_interval: float = 1.0,
+    max_wait: float = 60.0,
+) -> "Space.Message":
+    """
+    Chat against a file by uploading it, sending a message and waiting for a reply.
+    Args:
+
+        user_id: The user ID.
+        company_id: The company ID.
+        assistant_id: The assistant ID.
+        path_to_file: Path to the PDF file to upload.
+        displayed_filename: Name to display for the uploaded file.
+        mime_type: MIME type of the file (e.g., "application/pdf").
+        text: Text to send after uploading the file and chat against.
+        poll_interval: Seconds between polls for file ingestion.
+        max_wait: Maximum seconds to wait for the final response.
+
+    Returns:
+        The final message response.
+    """
+    chat_id = None
+
+    try:
+        response = await send_message_and_wait_for_completion(
+            user_id=user_id,
+            company_id=company_id,
+            assistant_id=assistant_id,
+            text="I'm going to upload a file for analysis.",
+        )
+        chat_id = response.get("chatId")
+
+        upload_response = upload_file(
+            userId=user_id,
+            companyId=company_id,
+            path_to_file=path_to_file,
+            displayed_filename=displayed_filename,
+            mime_type=mime_type,
+            chat_id=chat_id,
+        )
+        content_id = upload_response.get("id")
+
+        await wait_for_ingestion_completion(
+            user_id=user_id,
+            company_id=company_id,
+            content_id=content_id,
+            chat_id=chat_id,
+            poll_interval=poll_interval,
+            max_wait=max_wait,
+        )
+
+        final_response = await send_message_and_wait_for_completion(
+            user_id=user_id,
+            company_id=company_id,
+            assistant_id=assistant_id,
+            text=text,
+            chat_id=chat_id,
+            poll_interval=poll_interval,
+            max_wait=max_wait,
+        )
+
+        return final_response
+
+    except Exception as err:
+        print(f"Error during chat against file: {err}")
+        raise
+    finally:
+        if chat_id:
+            Space.delete_chat(
+                user_id=user_id,
+                company_id=company_id,
+                chat_id=chat_id,
+            )
+
+
+async def wait_for_ingestion_completion(
+    user_id: str,
+    company_id: str,
+    content_id: str,
+    chat_id: str,
+    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.
+    """
+    max_attempts = int(max_wait // poll_interval)
+    for _ in range(max_attempts):
+        searched_content = Content.search(
+            user_id=user_id,
+            company_id=company_id,
+            where={"id": {"equals": content_id}},
+            chatId=chat_id,
+            includeFailedContent=True,
+        )
+        if searched_content:
+            ingestion_state = searched_content[0].get("ingestionState")
+            if ingestion_state == "FINISHED":
+                return
+            if isinstance(ingestion_state, str) and ingestion_state.startswith(
+                "FAILED"
+            ):
+                raise RuntimeError(f"Ingestion failed with state: {ingestion_state}")
+        await asyncio.sleep(poll_interval)
+    raise TimeoutError("Timed out waiting for file ingestion to finish.")

{unique_sdk-0.10.0.dist-info → unique_sdk-0.10.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_sdk
-Version: 0.10.0
+Version: 0.10.2
 Summary: 
 License: MIT
 Author: Martin Fadler
@@ -37,6 +37,7 @@ The Unique Python SDK provides access to the public API of Unique FinanceGPT. It
    - [Short Term Memory](#short-term-memory)
    - [Message Assessment](#message-assessment)
    - [Folder](#folder)
+   - [Space](#space)
 6. [UniqueQL](#uniqueql)
    - [Query Structure](#uniqueql-query-structure)
    - [Metadata Filtering](#metadata-filtering)
@@ -251,6 +252,7 @@ unique_sdk.Message.modify(
 - [Short Term Memory](#short-term-memory)
 - [Message Assessment](#message-assessment)
 - [Folder](#folder)
+- [Space](#space)
 
 Most of the API services provide an asynchronous version of the method. The async methods are suffixed with `_async`.
 
@@ -950,6 +952,20 @@ unique_sdk.Folder.remove_access(
 )
 ```
 
+### Space
+
+#### `unique_sdk.Space.delete_chat`
+
+Delete a space chat by id. If the chat does not exist, the function will return an error.
+
+```python
+unique_sdk.Space.delete_chat(
+    user_id=user_id,
+    company_id=company_id,
+    chat_id="chat_dejfhe729br398",
+)
+```
+
 ## UniqueQL
 
 [UniqueQL](https://unique-ch.atlassian.net/wiki/x/coAXHQ) is an advanced query language designed to enhance search capabilities within various search modes such as Vector, Full-Text Search (FTS), and Combined. This query language enables users to perform detailed searches by filtering through metadata attributes like filenames, URLs, dates, and more. UniqueQL is versatile and can be translated into different query formats for various database systems, including PostgreSQL and Qdrant.
@@ -1008,6 +1024,12 @@ A metadata filter such as the one designed above can be used in a `Search.create
 
 ## Utils
 
+- [Chat History](#chat-history)
+- [File Io](#file-io)
+- [Sources](#sources)
+- [token](#token)
+- [Chat In Space](#chat-in-space)
+
 ### Chat History
 
 #### `unique_sdk.utils.chat_history.load_history`
@@ -1237,33 +1259,64 @@ The script ensures you can flexibly interact with spaces in new or ongoing chats
 
 ```python
 latest_message = await unique_sdk.utils.chat_in_space.send_message_and_wait_for_completion(
-        user_id=user_id,
-        company_id=company_id,
-        assistant_id=assistant_id,
-        text="Tell me a short story.",
-        chat_id=chat_id,                # Optional - if no chat id is specified, a new chat will be created
-        tool_choices=["WebSearch"],
-        scope_rules={
-            "or": [
-                {
-                    "operator": "in",
-                    "path": [
-                        "contentId"
-                    ],
-                    "value": [
-                        "cont_u888z7cazxxm4lugfdjq7pks"
-                    ]
-                },
-                {
-                    "operator": "contains",
-                    "path": [
-                        "folderIdPath"
-                    ],
-                    "value": "uniquepathid://scope_btfo28b3eeelwh5obwgea71bl/scope_fn56ta67knd6w4medgq3028fx"
-                }
-            ]
-        },
-    )
+    user_id=user_id,
+    company_id=company_id,
+    assistant_id=assistant_id,
+    text="Tell me a short story.",
+    chat_id=chat_id,                # Optional - if no chat id is specified, a new chat will be created
+    tool_choices=["WebSearch"],
+    scope_rules={
+        "or": [
+            {
+                "operator": "in",
+                "path": [
+                    "contentId"
+                ],
+                "value": [
+                    "cont_u888z7cazxxm4lugfdjq7pks"
+                ]
+            },
+            {
+                "operator": "contains",
+                "path": [
+                    "folderIdPath"
+                ],
+                "value": "uniquepathid://scope_btfo28b3eeelwh5obwgea71bl/scope_fn56ta67knd6w4medgq3028fx"
+            }
+        ]
+    },
+)
+```
+
+#### `unique_sdk.utils.chat_in_space.chat_against_file`
+
+The following script enables you to chat against a file.
+
+You must provide the following parameters:
+- `assistantId`: The assistant to be used for the chat.
+- `path_to_file`: The local path of the file to be uploaded.
+- `displayed_filename`: The name of the file to be displayed.
+- `mime_type`: The mime type of the ifle to be uploaded.
+- `text`: The text to be sent to the chat for chatting against the file.
+
+The script creates a chat and uploads the file to it. It then keeps polling the `ingestionState` field of the message, waiting for it to reach `FINISHED`, signaling the upload is complete. Once the file uploads successfully, the script sends the text, continues polling for completion, and finally retrieves the response message.
+
+**Optional parameters:**
+- `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 chatting against a PDF. (The usage can be extended to any supported file type)
+
+```python
+latest_message = await unique_sdk.utils.chat_in_space.chat_against_file(
+    user_id=user_id,
+    company_id=company_id,
+    assistant_id="assistant_hjcdga64bkcjnhu4",
+    path_to_file="/files/hello.pdf",
+    displayed_filename="hello.pdf"
+    mime_type="application/pdf"
+    text="Give me a bullet point summary of the file.",
+)
 ```
 
 ## Error Handling
@@ -1283,6 +1336,12 @@ 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.2] - 2025-08-05
+- Add script to chat against file.
+
+## [0.10.1] - 2025-08-05
+- Allow deletion of a space chat.
+
 ## [0.10.0] - 2025-08-04
 - Add MCP support
 

{unique_sdk-0.10.0.dist-info → unique_sdk-0.10.2.dist-info}/RECORD

@@ -16,7 +16,7 @@ unique_sdk/_webhook.py,sha256=GYxbUibQN_W4XlNTHaMIksT9FQJk4LJmlKcxOu3jqiU,2855
 unique_sdk/api_resources/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_sdk/api_resources/_acronyms.py,sha256=GIU1XH1flGWQYcpsFqTYwg4ioIGxVmb15tux84nmhEg,891
 unique_sdk/api_resources/_chat_completion.py,sha256=ILCAffxkbkfh2iV9L4KKnfe80gZmT9pWfkNmf3mq68U,2172
-unique_sdk/api_resources/_content.py,sha256=KIbAjXcf_-YiXAByW4wRYqLAqst47Bup8YXTIQ-PnTo,9550
+unique_sdk/api_resources/_content.py,sha256=Rl1Pb0vUQ9EkEQa0XTGOLZamAYlZaz7h6jNWE-zqFaw,9554
 unique_sdk/api_resources/_embedding.py,sha256=C6qak7cCUBMBINfPhgH8taCJZ9n6w1MUElqDJJ8dG10,1281
 unique_sdk/api_resources/_event.py,sha256=bpWF9vstdoAWbUzr-iiGP713ceP0zPk77GJXiImf9zg,374
 unique_sdk/api_resources/_folder.py,sha256=dmPorUoOJSe-hcxlJq3RnkQeZg3gPqGF37ZsLgnloT0,8335
@@ -27,13 +27,13 @@ unique_sdk/api_resources/_message_assessment.py,sha256=SSfx6eW7zb_GKe8cFJzCqW-t-
 unique_sdk/api_resources/_search.py,sha256=pAVMXL2AdJNP5JG-7LlaU2Uw1152iUaMZ2YO7fHnNWc,1906
 unique_sdk/api_resources/_search_string.py,sha256=4Idw6exgZdA8qksz9WkiA68k1hTU-7yFkgT_OLU_GkE,1662
 unique_sdk/api_resources/_short_term_memory.py,sha256=vPRN-Y0WPx74E6y-A3LocGc0TxJdzT-xGL66WzZwKRg,2820
-unique_sdk/api_resources/_space.py,sha256=LAK-ZbHemIIfX51anthml4ojxhPUPDOu_EbQs64rOFA,3726
+unique_sdk/api_resources/_space.py,sha256=9e-_3oBr_25JeRLF7GfaXnIPE33tZ19g0UWxPp_yR4k,4805
 unique_sdk/utils/chat_history.py,sha256=5UqL9hF1O9pV7skbNOlEibF5rHdYsmG3m5-YEPUowOs,3037
-unique_sdk/utils/chat_in_space.py,sha256=kQA43P9XZ6Kdf05JZiLhVJ5ejVtrgzZSb8osyNHSZmc,1580
+unique_sdk/utils/chat_in_space.py,sha256=nYmgQYhIxqQex_6zjdCfNuGnjoU14WkxUN6_zmSOiCQ,5169
 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.0.dist-info/LICENSE,sha256=EJCWoHgrXVBUb47PnjeV4MFIEOR71MAdCOIgv61J-4k,1065
-unique_sdk-0.10.0.dist-info/METADATA,sha256=VcwHI6WA3JGiFekfjs411NWLxE3lj22a7FmReyJv_Xw,42872
-unique_sdk-0.10.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_sdk-0.10.0.dist-info/RECORD,,
+unique_sdk-0.10.2.dist-info/LICENSE,sha256=EJCWoHgrXVBUb47PnjeV4MFIEOR71MAdCOIgv61J-4k,1065
+unique_sdk-0.10.2.dist-info/METADATA,sha256=L2sOYupGkEPz5rw9qOtVc5nUtjUBu14pgtC-CpLH9Os,44785
+unique_sdk-0.10.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_sdk-0.10.2.dist-info/RECORD,,