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

unique_sdk/__init__.py

@@ -7,15 +7,15 @@ from typing import Literal, Optional
 # Konstantin Krauss<konstantin@unique.ch>
 from unique_sdk._api_version import _ApiVersion
 
-api_key: Optional[str] = None
-app_id: Optional[str] = None
+api_key: str | None = None
+app_id: str | None = None
 api_base: str = "https://gateway.unique.app/public/chat-gen2"
 api_version: str = _ApiVersion.CURRENT
 api_verify_mode: bool = True
 default_http_client: Optional["HTTPClient"] = None
 
 # Set to either 'debug' or 'info', controls console logging
-log: Optional[Literal["debug", "info"]] = None
+log: Literal["debug", "info"] | None = None
 
 # Webhooks
 from unique_sdk._api_requestor import APIRequestor as APIRequestor
@@ -92,3 +92,18 @@ from unique_sdk.api_resources._mcp import MCP as MCP
 # Unique QL
 from unique_sdk._unique_ql import UQLOperator as UQLOperator
 from unique_sdk._unique_ql import UQLCombinator as UQLCombinator
+
+# Agentic Table
+from .api_resources._agentic_table import (
+    AgenticTable as AgenticTable,
+    AgenticTableCell as AgenticTableCell,
+    AgenticTableSheet as AgenticTableSheet,
+    AgenticTableSheetState as AgenticTableSheetState,
+    LogEntry as LogEntry,
+    LogDetail as LogDetail,
+    FilterTypes as FilterTypes,
+    CellRendererTypes as CellRendererTypes,
+    SelectionMethod as SelectionMethod,
+    AgreementStatus as AgreementStatus,
+    RowVerificationStatus as RowVerificationStatus,
+)

unique_sdk/api_resources/_agentic_table.py

@@ -0,0 +1,367 @@
+from enum import StrEnum
+from typing import (
+    Literal,
+    NotRequired,
+    TypedDict,
+    cast,
+)
+
+from typing_extensions import Unpack
+
+from unique_sdk._api_resource import APIResource
+from unique_sdk._request_options import RequestOptions
+
+
+class AgenticTableSheetState(StrEnum):
+    STOPPED_BY_USER = "STOPPED_BY_USER"
+    PROCESSING = "PROCESSING"
+    IDLE = "IDLE"
+
+
+class LogDetail(TypedDict, total=False):
+    text: str
+    messageId: str | None
+
+
+class LogEntry(TypedDict):
+    text: str
+    createdAt: str
+    actorType: Literal["USER", "SYSTEM", "ASSISTANT", "TOOL"]
+    messageId: NotRequired[str]
+    details: NotRequired[list[LogDetail]]
+
+
+class AgenticTableCell(TypedDict, total=False):
+    sheetId: str
+    rowOrder: int
+    columnOrder: int
+    rowLocked: bool
+    text: str
+    logEntries: list[LogEntry]
+
+
+class ColumnMetadataUpdateStatus(TypedDict, total=False):
+    status: bool
+    message: str | None
+
+
+class AgenticTableSheet(TypedDict):
+    sheetId: str
+    name: str
+    state: AgenticTableSheetState
+    chatId: str
+    createdBy: str
+    companyId: str
+    createdAt: str
+    magicTableRowCount: int
+    magicTableCells: NotRequired[list[AgenticTableCell]]
+
+
+class FilterTypes(StrEnum):
+    VALUE_MATCH_FILTER = "ValueMatchFilter"
+    PARTIAL_MATCH_FILTER = "PartialMatchFilter"
+    REFERENCE_FILTER = "ReferenceFilter"
+    HALLUCINATION_FILTER = "HallucinationFilter"
+    REVIEW_STATUS_FILTER = "ReviewStatusFilter"
+    ASSIGNEE_FILTER = "AssigneeFilter"
+
+
+class CellRendererTypes(StrEnum):
+    CHECKBOX_LOCK_CELL_RENDERER = "CheckboxLockCellRenderer"
+    COLLABORATOR_DROPDOWN = "CollaboratorDropdown"
+    REVIEW_STATUS_DROPDOWN = "ReviewStatusDropdown"
+    CUSTOM_CELL_RENDERER = "CustomCellRenderer"
+    SELECTABLE_CELL_RENDERER = "SelectableCellRenderer"
+
+
+class SelectionMethod(StrEnum):
+    DEFAULT = "DEFAULT"
+    MANUAL = "MANUAL"
+
+
+class AgreementStatus(StrEnum):
+    MATCH = "MATCH"
+    NO_MATCH = "NO_MATCH"
+
+
+class RowVerificationStatus(StrEnum):
+    NEED_REVIEW = "NEED_REVIEW"
+    READY_FOR_VERIFICATION = "READY_FOR_VERIFICATION"
+    VERIFIED = "VERIFIED"
+
+
+class AgenticTable(APIResource["AgenticTable"]):
+    """
+    This object represents the magic table route. It is used to run complex APIs on the Unique platform
+    """
+
+    class SetCell(RequestOptions):
+        tableId: str
+        rowOrder: int
+        columnOrder: int
+        text: str
+        logEntries: NotRequired[list[LogEntry]]
+
+    class GetCell(RequestOptions):
+        tableId: str
+        rowOrder: int
+        columnOrder: int
+
+    class SetActivityStatus(RequestOptions):
+        tableId: str
+        activity: Literal[
+            "DeleteRow",
+            "DeleteColumn",
+            "UpdateCell",
+            "AddQuestionText",
+            "AddMetaData",
+            "GenerateArtifact",
+            "SheetCompleted",
+            "LibrarySheetRowVerified",
+        ]
+        status: Literal[
+            "IN_PROGRESS",
+            "COMPLETED",
+            "FAILED",
+        ]
+        text: str
+
+    class SetArtifact(RequestOptions):
+        tableId: str
+        name: str
+        contentId: str
+        mimeType: str
+        artifactType: Literal["QUESTIONS", "FULL_REPORT"]
+
+    class UpdateSheet(RequestOptions):
+        tableId: str
+        state: NotRequired[AgenticTableSheetState]
+        name: NotRequired[str]
+
+    class UpdateSheetResponse(RequestOptions):
+        status: bool
+        message: str
+
+    class SetColumnMetadata(RequestOptions):
+        tableId: str
+        columnOrder: int
+        columnWidth: NotRequired[int]
+        filter: NotRequired[FilterTypes]
+        cellRenderer: NotRequired[str]
+        editable: NotRequired[bool]
+
+    class GetSheetData(RequestOptions):
+        tableId: str
+        includeCells: NotRequired[bool]
+        includeLogHistory: NotRequired[bool]
+        includeRowCount: NotRequired[bool]
+        includeCellMetaData: NotRequired[bool]
+        startRow: NotRequired[int]
+        endRow: NotRequired[int]
+
+    class SetCellMetadata(RequestOptions):
+        tableId: str
+        rowOrder: int
+        columnOrder: int
+        selected: NotRequired[bool]
+        selectionMethod: NotRequired[SelectionMethod]
+        agreementStatus: NotRequired[AgreementStatus]
+
+    class BulkUpdateStatus(RequestOptions):
+        tableId: str
+        rowOrders: list[int]
+        status: RowVerificationStatus
+
+    @classmethod
+    async def set_cell(
+        cls,
+        user_id: str,
+        company_id: str,
+        **params: Unpack["AgenticTable.SetCell"],
+    ) -> "AgenticTableCell":
+        """ """
+        url = f"/magic-table/{params['tableId']}/cell"
+        return cast(
+            "AgenticTableCell",
+            cls._static_request(
+                "post",
+                url,
+                user_id,
+                company_id,
+                params,
+            ),
+        )
+
+    @classmethod
+    async def get_cell(
+        cls,
+        user_id: str,
+        company_id: str,
+        **params: Unpack["AgenticTable.GetCell"],
+    ) -> "AgenticTableCell":
+        """ """
+        url = f"/magic-table/{params['tableId']}/cell?rowOrder={params['rowOrder']}&columnOrder={params['columnOrder']}"
+        return cast(
+            "AgenticTableCell",
+            await cls._static_request_async(
+                "get",
+                url,
+                user_id,
+                company_id,
+                params={},
+            ),
+        )
+
+    @classmethod
+    async def set_activity(
+        cls,
+        user_id: str,
+        company_id: str,
+        **params: Unpack["AgenticTable.SetActivityStatus"],
+    ) -> "AgenticTableCell":
+        """ """
+        url = f"/magic-table/{params['tableId']}/activity"
+        return cast(
+            "AgenticTableCell",
+            cls._static_request(
+                "post",
+                url,
+                user_id,
+                company_id,
+                params,
+            ),
+        )
+
+    @classmethod
+    async def set_artifact(
+        cls,
+        user_id: str,
+        company_id: str,
+        **params: Unpack["AgenticTable.SetArtifact"],
+    ) -> "AgenticTableCell":
+        """ """
+        url = f"/magic-table/{params['tableId']}/artifact"
+        return cast(
+            "AgenticTableCell",
+            cls._static_request(
+                "post",
+                url,
+                user_id,
+                company_id,
+                params,
+            ),
+        )
+
+    @classmethod
+    async def update_sheet_state(
+        cls,
+        user_id: str,
+        company_id: str,
+        **params: Unpack["AgenticTable.UpdateSheet"],
+    ) -> "AgenticTable.UpdateSheetResponse":
+        url = f"/magic-table/{params['tableId']}"
+        return cast(
+            "AgenticTable.UpdateSheetResponse",
+            cls._static_request("post", url, user_id, company_id, params),
+        )
+
+    @classmethod
+    async def set_column_metadata(
+        cls,
+        user_id: str,
+        company_id: str,
+        **params: Unpack["AgenticTable.SetColumnMetadata"],
+    ) -> ColumnMetadataUpdateStatus:
+        url = f"/magic-table/{params['tableId']}/column/metadata"
+        # Remove tableId from params
+        params.pop("tableId")
+        response = cls._static_request("post", url, user_id, company_id, params)
+        return cast(
+            ColumnMetadataUpdateStatus,
+            response,
+        )
+
+    @classmethod
+    async def get_sheet_data(
+        cls,
+        user_id: str,
+        company_id: str,
+        **params: Unpack["AgenticTable.GetSheetData"],
+    ) -> AgenticTableSheet:
+        url = f"/magic-table/{params['tableId']}"
+        return cast(
+            AgenticTableSheet,
+            cls._static_request("get", url, user_id, company_id, params),
+        )
+
+    @classmethod
+    async def get_sheet_state(
+        cls, user_id: str, company_id: str, tableId: str
+    ) -> AgenticTableSheetState:
+        sheet = await cls.get_sheet_data(
+            user_id=user_id,
+            company_id=company_id,
+            tableId=tableId,
+            includeCells=False,
+        )
+        return sheet["state"]
+
+    @classmethod
+    async def set_cell_metadata(
+        cls,
+        user_id: str,
+        company_id: str,
+        **params: Unpack["AgenticTable.SetCellMetadata"],
+    ) -> ColumnMetadataUpdateStatus:
+        url = f"/magic-table/{params['tableId']}/cell/metadata"
+        return cast(
+            "ColumnMetadataUpdateStatus",
+            cls._static_request("post", url, user_id, company_id, params),
+        )
+
+    @classmethod
+    async def set_multiple_cells(
+        cls,
+        user_id: str,
+        company_id: str,
+        tableId: str,
+        cells: list[AgenticTableCell],
+    ) -> ColumnMetadataUpdateStatus:
+        url = f"/magic-table/{tableId}/cells/bulk-upsert"
+        # Map AgenticTableCell to PublicAgenticTableCellDto
+        try:
+            params_api = {
+                "cells": [
+                    {
+                        "rowOrder": cell["rowOrder"],
+                        "columnOrder": cell["columnOrder"],
+                        "data": cell["text"],
+                    }
+                    for cell in cells
+                ]
+            }
+        except Exception as e:
+            raise ValueError(f"Invalid data or missing required fields: {e}")
+        return cast(
+            "ColumnMetadataUpdateStatus",
+            cls._static_request("post", url, user_id, company_id, params=params_api),
+        )
+
+    @classmethod
+    async def bulk_update_status(
+        cls,
+        user_id: str,
+        company_id: str,
+        **params: Unpack["AgenticTable.BulkUpdateStatus"],
+    ) -> ColumnMetadataUpdateStatus:
+        url = f"/magic-table/{params['tableId']}/rows/bulk-update-status"
+        return cast(
+            "ColumnMetadataUpdateStatus",
+            cls._static_request(
+                "post",
+                url,
+                user_id,
+                company_id,
+                params,
+            ),
+        )

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.7.dist-info → unique_sdk-0.10.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_sdk
-Version: 0.10.7
+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,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.9] - 2025-08-15
+- Add script to wait for content ingestion finished.
+
+## [0.10.8] - 2025-08-13
+- Add support for Agentic Table.
+
 ## [0.10.7] - 2025-08-13
 - Make metadata optional when uploading a file.
 

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

@@ -1,4 +1,4 @@
-unique_sdk/__init__.py,sha256=n_drdw9PkmJpQUt-7DjeUFinRl95gBSks2xsMLEeE9M,3350
+unique_sdk/__init__.py,sha256=kD6dQiLHr0eJKy5yzN0iRNO7N2v4n9TDv6h6DTHMzoo,3840
 unique_sdk/_api_requestor.py,sha256=i4gCpzx8zP95sv-AhJfpQxKvWR0U-I6lclHyV55RPtg,14397
 unique_sdk/_api_resource.py,sha256=ytjomI-IVJwsbvdPyuZCfF-bl-Abgf66bu1D34YxCu8,6244
 unique_sdk/_api_version.py,sha256=Ku4JPdeyJtnX5eJJvRCEc1_u44UObdVrvrL1T-WwWCs,46
@@ -15,6 +15,7 @@ unique_sdk/_version.py,sha256=j4_tPC6t3enRds7LqiRuWSyfrYHfEo6CXIDARAWW98I,19
 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/_agentic_table.py,sha256=8-_f7t-m_iiiOj2835iESoxz91YRxl4-tkxpzQbgdcI,9958
 unique_sdk/api_resources/_chat_completion.py,sha256=ILCAffxkbkfh2iV9L4KKnfe80gZmT9pWfkNmf3mq68U,2172
 unique_sdk/api_resources/_content.py,sha256=KnrfH8UFItNjnNejyWpvxBb9kABwjaxSiK9J8CPf67o,10725
 unique_sdk/api_resources/_embedding.py,sha256=C6qak7cCUBMBINfPhgH8taCJZ9n6w1MUElqDJJ8dG10,1281
@@ -29,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.7.dist-info/LICENSE,sha256=EJCWoHgrXVBUb47PnjeV4MFIEOR71MAdCOIgv61J-4k,1065
-unique_sdk-0.10.7.dist-info/METADATA,sha256=79pPucHAbyeI3kp8Y2bmWCDs350XiTLqAQQtxF2Ygbg,49144
-unique_sdk-0.10.7.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_sdk-0.10.7.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,,