unique_sdk 0.10.19__py3-none-any.whl → 0.10.71__py3-none-any.whl

unique_sdk/__init__.py

@@ -83,6 +83,9 @@ from unique_sdk.api_resources._short_term_memory import (
 from unique_sdk.api_resources._folder import Folder as Folder
 from unique_sdk.api_resources._embedding import Embeddings as Embeddings
 from unique_sdk.api_resources._acronyms import Acronyms as Acronyms
+from unique_sdk.api_resources._llm_models import LLMModels as LLMModels
+from unique_sdk.api_resources._user import User as User
+from unique_sdk.api_resources._group import Group as Group
 from unique_sdk.api_resources._message_assessment import (
     MessageAssessment as MessageAssessment,
 )

unique_sdk/api_resources/_agentic_table.py

@@ -19,8 +19,7 @@ class AgenticTableSheetState(StrEnum):
 
 
 class LogDetail(TypedDict, total=False):
-    text: str
-    messageId: str | None
+    llmRequest: list[dict] | None
 
 
 class LogEntry(TypedDict):
@@ -28,7 +27,7 @@ class LogEntry(TypedDict):
     createdAt: str
     actorType: Literal["USER", "SYSTEM", "ASSISTANT", "TOOL"]
     messageId: NotRequired[str]
-    details: NotRequired[list[LogDetail]]
+    details: NotRequired[LogDetail]
 
 
 class AgenticTableCell(TypedDict, total=False):
@@ -37,7 +36,7 @@ class AgenticTableCell(TypedDict, total=False):
     columnOrder: int
     rowLocked: bool
     text: str
-    logEntries: list[LogEntry]
+    logEntries: list[LogEntry] | None
 
 
 class ColumnMetadataUpdateStatus(TypedDict, total=False):
@@ -74,6 +73,29 @@ class CellRendererTypes(StrEnum):
     SELECTABLE_CELL_RENDERER = "SelectableCellRenderer"
 
 
+class MagicTableAction(StrEnum):
+    DELETE_ROW = "DeleteRow"
+    DELETE_COLUMN = "DeleteColumn"
+    UPDATE_CELL = "UpdateCell"
+    ADD_QUESTION_TEXT = "AddQuestionText"
+    ADD_META_DATA = "AddMetaData"
+    GENERATE_ARTIFACT = "GenerateArtifact"
+    SHEET_COMPLETED = "SheetCompleted"
+    LIBRARY_SHEET_ROW_VERIFIED = "LibrarySheetRowVerified"
+    SHEET_CREATED = "SheetCreated"
+
+
+class ActivityStatus(StrEnum):
+    IN_PROGRESS = "IN_PROGRESS"
+    COMPLETED = "COMPLETED"
+    FAILED = "FAILED"
+
+
+class SheetType(StrEnum):
+    DEFAULT = "DEFAULT"
+    LIBRARY = "LIBRARY"
+
+
 class SelectionMethod(StrEnum):
     DEFAULT = "DEFAULT"
     MANUAL = "MANUAL"
@@ -106,24 +128,12 @@ class AgenticTable(APIResource["AgenticTable"]):
         tableId: str
         rowOrder: int
         columnOrder: int
+        includeRowMetadata: NotRequired[bool]
 
     class SetActivityStatus(RequestOptions):
         tableId: str
-        activity: Literal[
-            "DeleteRow",
-            "DeleteColumn",
-            "UpdateCell",
-            "AddQuestionText",
-            "AddMetaData",
-            "GenerateArtifact",
-            "SheetCompleted",
-            "LibrarySheetRowVerified",
-        ]
-        status: Literal[
-            "IN_PROGRESS",
-            "COMPLETED",
-            "FAILED",
-        ]
+        activity: MagicTableAction
+        status: ActivityStatus
         text: str
 
     class SetArtifact(RequestOptions):
@@ -201,6 +211,9 @@ class AgenticTable(APIResource["AgenticTable"]):
     ) -> "AgenticTableCell":
         """ """
         url = f"/magic-table/{params['tableId']}/cell?rowOrder={params['rowOrder']}&columnOrder={params['columnOrder']}"
+        params.pop("tableId")
+        params.pop("rowOrder")
+        params.pop("columnOrder")
         return cast(
             "AgenticTableCell",
             await cls._static_request_async(
@@ -208,7 +221,7 @@ class AgenticTable(APIResource["AgenticTable"]):
                 url,
                 user_id,
                 company_id,
-                params={},
+                params=params,
             ),
         )
 

unique_sdk/api_resources/_content.py

@@ -12,6 +12,7 @@ from typing import (
 
 from typing_extensions import NotRequired, Unpack
 
+import unique_sdk
 from unique_sdk._api_resource import APIResource
 from unique_sdk._request_options import RequestOptions
 
@@ -19,6 +20,18 @@ from unique_sdk._request_options import RequestOptions
 class Content(APIResource["Content"]):
     OBJECT_NAME: ClassVar[Literal["content.search"]] = "content.search"
 
+    id: str
+    key: str
+    url: Optional[str]
+    title: Optional[str]
+    updatedAt: str
+    chunks: Optional[List["Content.Chunk"]]
+    metadata: Optional[Dict[str, Any]]
+    writeUrl: Optional[str]
+    readUrl: Optional[str]
+    expiredAt: Optional[str]
+    appliedIngestionConfig: Optional[Dict[str, Any]]
+
     class QueryMode(Enum):
         Default = "default"
         Insensitive = "insensitive"
@@ -72,7 +85,7 @@ class Content(APIResource["Content"]):
 
     class SearchParams(RequestOptions):
         where: "Content.ContentWhereInput"
-        chatId: NotRequired[str]
+        chatId: NotRequired[str | None]
         includeFailedContent: NotRequired[bool]
 
     class ContentInfoParams(TypedDict):
@@ -81,10 +94,24 @@ class Content(APIResource["Content"]):
         This is used to retrieve information about content based on various filters.
         """
 
-        metadataFilter: dict
-        skip: NotRequired[int]
-        take: NotRequired[int]
-        filePath: NotRequired[str]
+        metadataFilter: NotRequired[dict[str, Any] | None]
+        skip: NotRequired[int | None]
+        take: NotRequired[int | None]
+        filePath: NotRequired[str | None]
+        contentId: NotRequired[str | None]
+        chatId: NotRequired[str | None]
+
+    class ContentInfosParams(TypedDict):
+        """
+        Parameters for the content infos endpoint.
+        This is used to retrieve information about contents based on various filters.
+        """
+
+        metadataFilter: NotRequired[dict[str, Any] | None]
+        skip: NotRequired[int | None]
+        take: NotRequired[int | None]
+        parentId: NotRequired[str | None]
+        parentFolderPath: NotRequired[str | None]
 
     class CustomApiOptions(TypedDict):
         apiIdentifier: str
@@ -108,29 +135,36 @@ class Content(APIResource["Content"]):
         uniqueIngestionMode: str
         vttConfig: Optional["Content.VttConfig"]
         wordReadMode: Optional[str]
+        hideInChat: Optional[bool]
 
     class Input(TypedDict):
         key: str
         title: Optional[str]
         mimeType: str
-        ownerType: str
-        ownerId: str
-        byteSize: Optional[int]
-        ingestionConfig: "Content.IngestionConfig"
-        metadata: dict[str, Any] | None = None
+        description: NotRequired[str | None]
+        ownerType: NotRequired[str | None]
+        ownerId: NotRequired[str | None]
+        byteSize: NotRequired[int | None]
+        ingestionConfig: NotRequired["Content.IngestionConfig | None"]
+        metadata: NotRequired[dict[str, Any] | None]
 
     class UpsertParams(RequestOptions):
         input: "Content.Input"
-        scopeId: Optional[str]
-        chatId: Optional[str]
-        sourceOwnerType: str
-        storeInternally: bool
-        fileUrl: Optional[str]
+        scopeId: NotRequired[str | None]
+        parentFolderPath: NotRequired[str | None]
+        createFolderIfNotExists: NotRequired[bool | None]
+        chatId: NotRequired[str | None]
+        sourceOwnerType: NotRequired[str | None]
+        storeInternally: NotRequired[bool | None]
+        fileUrl: NotRequired[str | None]
 
     class UpdateParams(RequestOptions):
-        contentId: str | None = None
-        ownerId: str | None = None
-        title: str | None = None
+        contentId: NotRequired[str]
+        filePath: NotRequired[str]
+        ownerId: NotRequired[str]
+        parentFolderPath: NotRequired[str]
+        title: NotRequired[str]
+        metadata: NotRequired[dict[str, str | None]]
 
     class Chunk(TypedDict):
         id: str
@@ -151,6 +185,7 @@ class Content(APIResource["Content"]):
         title: str | None
         metadata: Dict[str, Any] | None
         mimeType: str
+        description: str | None
         byteSize: int
         ownerId: str
         createdAt: str
@@ -160,43 +195,43 @@ class Content(APIResource["Content"]):
         expiredAt: str | None
 
     class PaginatedContentInfo(TypedDict):
+        contentInfo: List["Content.ContentInfo"]
+        totalCount: int
+
+    class PaginatedContentInfos(TypedDict):
         contentInfos: List["Content.ContentInfo"]
         totalCount: int
 
     class DeleteParams(RequestOptions):
         contentId: NotRequired[str]
+        filePath: NotRequired[str]
         chatId: NotRequired[str]
 
     class DeleteResponse(TypedDict):
         id: str
 
-    id: str
-    key: str
-    url: Optional[str]
-    title: Optional[str]
-    updatedAt: str
-    chunks: List[Chunk]
-    metadata: Optional[Dict[str, Any]]
-
     class MagicTableSheetTableColumn(TypedDict):
         columnId: str
         columnName: str
         content: str
 
-    class MagicTableSheetTable(TypedDict):
+    class MagicTableRow(TypedDict):
         rowId: str
         columns: List["Content.MagicTableSheetTableColumn"]
+        context: NotRequired[str]
+        rowMetadata: NotRequired[str]
 
     class MagicTableSheetIngestionConfiguration(TypedDict):
         columnIdsInMetadata: List[str]
         columnIdsInChunkText: List[str]
 
     class MagicTableSheetIngestParams(TypedDict):
-        data: List["Content.MagicTableSheetTable"]
+        data: List["Content.MagicTableRow"]
         ingestionConfiguration: "Content.MagicTableSheetIngestionConfiguration"
         metadata: Dict[str, Optional[str]]
         scopeId: str
         sheetName: str
+        context: NotRequired[str]
 
     class MagicTableSheetRowIdToContentId(TypedDict):
         rowId: str
@@ -282,10 +317,20 @@ class Content(APIResource["Content"]):
         cls,
         user_id: str,
         company_id: str,
-        **params: Unpack["Content.ContentInfoParams"],
-    ) -> "Content.PaginatedContentInfo":
+        **params: Unpack["Content.ContentInfosParams"],
+    ) -> "Content.PaginatedContentInfos":
+        parent_id = unique_sdk.Folder.resolve_scope_id_from_folder_path(
+            user_id=user_id,
+            company_id=company_id,
+            scope_id=params.get("parentId"),
+            folder_path=params.get("parentFolderPath"),
+        )
+        params.pop("parentFolderPath", None)
+        if parent_id:
+            params["parentId"] = parent_id
+
         return cast(
-            Content.PaginatedContentInfo,
+            Content.PaginatedContentInfos,
             cls._static_request(
                 "post",
                 "/content/infos",
@@ -300,10 +345,20 @@ class Content(APIResource["Content"]):
         cls,
         user_id: str,
         company_id: str,
-        **params: Unpack["Content.ContentInfoParams"],
-    ) -> "Content.PaginatedContentInfo":
+        **params: Unpack["Content.ContentInfosParams"],
+    ) -> "Content.PaginatedContentInfos":
+        parent_id = await unique_sdk.Folder.resolve_scope_id_from_folder_path_async(
+            user_id=user_id,
+            company_id=company_id,
+            scope_id=params.get("parentId"),
+            folder_path=params.get("parentFolderPath"),
+        )
+        params.pop("parentFolderPath", None)
+        if parent_id:
+            params["parentId"] = parent_id
+
         return cast(
-            Content.PaginatedContentInfo,
+            Content.PaginatedContentInfos,
             await cls._static_request_async(
                 "post",
                 "/content/infos",
@@ -325,6 +380,22 @@ class Content(APIResource["Content"]):
         """
         if "input" in params:
             params["input"]["metadata"] = params["input"].get("metadata") or {}
+            if "description" in params["input"] and not params["input"]["description"]:
+                params["input"].pop("description")
+
+        create_folder = params.get("createFolderIfNotExists")
+        scope_id = unique_sdk.Folder.resolve_scope_id_from_folder_path_with_create(
+            user_id=user_id,
+            company_id=company_id,
+            scope_id=params.get("scopeId"),
+            folder_path=params.get("parentFolderPath"),
+            create_if_not_exists=create_folder if create_folder is not None else True,
+        )
+        params.pop("parentFolderPath", None)
+        params.pop("createFolderIfNotExists", None)
+        if scope_id:
+            params["scopeId"] = scope_id
+
         return cast(
             "Content",
             cls._static_request(
@@ -348,6 +419,26 @@ class Content(APIResource["Content"]):
         """
         if "input" in params:
             params["input"]["metadata"] = params["input"].get("metadata") or {}
+            if "description" in params["input"] and not params["input"]["description"]:
+                params["input"].pop("description")
+
+        create_folder = params.get("createFolderIfNotExists")
+        scope_id = (
+            await unique_sdk.Folder.resolve_scope_id_from_folder_path_with_create_async(
+                user_id=user_id,
+                company_id=company_id,
+                scope_id=params.get("scopeId"),
+                folder_path=params.get("parentFolderPath"),
+                create_if_not_exists=create_folder
+                if create_folder is not None
+                else True,
+            )
+        )
+        params.pop("parentFolderPath", None)
+        params.pop("createFolderIfNotExists", None)
+        if scope_id:
+            params["scopeId"] = scope_id
+
         return cast(
             "Content",
             await cls._static_request_async(
@@ -402,11 +493,29 @@ class Content(APIResource["Content"]):
         company_id: str,
         **params: Unpack["Content.UpdateParams"],
     ) -> "Content.ContentInfo":
+        content_id = cls.resolve_content_id_from_file_path(
+            user_id=user_id,
+            company_id=company_id,
+            content_id=params.get("contentId"),
+            file_path=params.get("filePath"),
+        )
+        owner_id = unique_sdk.Folder.resolve_scope_id_from_folder_path(
+            user_id,
+            company_id,
+            params.get("ownerId"),
+            params.get("parentFolderPath"),
+        )
+        params.pop("contentId", None)
+        params.pop("filePath", None)
+        params.pop("parentFolderPath", None)
+        if owner_id is not None:
+            params["ownerId"] = owner_id
+
         return cast(
             "Content.ContentInfo",
             cls._static_request(
                 "patch",
-                f"/content/{params.get('contentId')}",
+                f"/content/{content_id}",
                 user_id,
                 company_id,
                 params=params,
@@ -420,11 +529,29 @@ class Content(APIResource["Content"]):
         company_id: str,
         **params: Unpack["Content.UpdateParams"],
     ) -> "Content.ContentInfo":
+        content_id = cls.resolve_content_id_from_file_path(
+            user_id,
+            company_id,
+            params.get("contentId"),
+            params.get("filePath"),
+        )
+        owner_id = await unique_sdk.Folder.resolve_scope_id_from_folder_path_async(
+            user_id,
+            company_id,
+            params.get("ownerId"),
+            params.get("parentFolderPath"),
+        )
+        params.pop("contentId", None)
+        params.pop("filePath", None)
+        params.pop("parentFolderPath", None)
+        if owner_id is not None:
+            params["ownerId"] = owner_id
+
         return cast(
             "Content.ContentInfo",
             await cls._static_request_async(
                 "patch",
-                f"/content/{params.get('contentId')}",
+                f"/content/{content_id}",
                 user_id,
                 company_id,
                 params=params,
@@ -436,17 +563,25 @@ class Content(APIResource["Content"]):
         cls,
         user_id: str,
         company_id: str,
-        **params: "Content.DeleteParams",
+        **params: Unpack["Content.DeleteParams"],
     ) -> "Content.DeleteResponse":
         """
         Deletes a content by its id or file path.
         """
+        content_id = cls.resolve_content_id_from_file_path(
+            user_id,
+            company_id,
+            params.get("contentId"),
+            params.get("filePath"),
+        )
+        params.pop("contentId", None)
+        params.pop("filePath", None)
 
         return cast(
             "Content.DeleteResponse",
             cls._static_request(
                 "delete",
-                f"/content/{params.get('contentId')}",
+                f"/content/{content_id}",
                 user_id,
                 company_id,
                 params=params,
@@ -458,17 +593,25 @@ class Content(APIResource["Content"]):
         cls,
         user_id: str,
         company_id: str,
-        **params: "Content.DeleteParams",
+        **params: Unpack["Content.DeleteParams"],
     ) -> "Content.DeleteResponse":
         """
         Async deletes a content by its id or file path.
         """
+        content_id = cls.resolve_content_id_from_file_path(
+            user_id,
+            company_id,
+            params.get("contentId"),
+            params.get("filePath"),
+        )
+        params.pop("contentId", None)
+        params.pop("filePath", None)
 
         return cast(
             "Content.DeleteResponse",
             await cls._static_request_async(
                 "delete",
-                f"/content/{params.get('contentId')}",
+                f"/content/{content_id}",
                 user_id,
                 company_id,
                 params=params,
@@ -482,10 +625,14 @@ class Content(APIResource["Content"]):
         company_id: str,
         content_id: str | None = None,
         file_path: str | None = None,
-    ) -> str:
+    ) -> str | None:
         """
         Returns the contentId to use: if content_id is provided, returns it;
         if not, but file_path is provided, resolves and returns the id for that file path.
+
+        Returns:
+            str: The resolved content ID.
+            None: Failed to resolve a content ID (e.g., file_path not found or not provided).
         """
         if content_id:
             return content_id
@@ -495,9 +642,12 @@ class Content(APIResource["Content"]):
                 company_id=company_id,
                 filePath=file_path,
             )
+            content_infos = file_info.get("contentInfo")
             resolved_id = (
-                file_info.get("contentInfo")[0].get("id")
+                content_infos[0].get("id")
                 if file_info.get("totalCount", 0) > 0
+                and content_infos is not None
+                and len(content_infos) > 0
                 else None
             )
             if not resolved_id: