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

unique_sdk/api_resources/_message.py

@@ -29,8 +29,10 @@ class Message(APIResource["Message"]):
 
     class Reference(TypedDict):
         name: str
+        description: Optional[str]
         url: Optional[str]
         sequenceNumber: int
+        originalIndex: Optional[list[int]]
         sourceId: str
         source: str
 
@@ -45,10 +47,14 @@ class Message(APIResource["Message"]):
 
     class ModifyParams(RequestOptions):
         chatId: str
+        originalText: NotRequired[str | None]
         text: NotRequired[Optional["str"]]
-        references: Optional[List["Message.Reference"]]
-        debugInfo: Optional[Dict[str, Any]]
-        completedAt: Optional[datetime]
+        references: NotRequired[List["Message.Reference"] | None]
+        gptRequest: NotRequired[Dict[str, Any] | None]
+        debugInfo: NotRequired[Dict[str, Any] | None]
+        startedStreamingAt: NotRequired[datetime | None]
+        stoppedStreamingAt: NotRequired[datetime | None]
+        completedAt: NotRequired[datetime | None]
 
     class DeleteParams(RequestOptions):
         chatId: str
@@ -59,11 +65,16 @@ class Message(APIResource["Message"]):
     class RetrieveParams(RequestOptions):
         chatId: str
 
+    class CreateEventParams(ModifyParams):
+        messageId: str
+
     chatId: str
     text: Optional[str]
     role: Literal["SYSTEM", "USER", "ASSISTANT"]
     gptRequest: Optional[Dict[str, Any]]
     debugInfo: Optional[Dict[str, Any]]
+    startedStreamingAt: Optional[datetime]
+    stoppedStreamingAt: Optional[datetime]
 
     @classmethod
     def list(
@@ -155,6 +166,12 @@ class Message(APIResource["Message"]):
         """
         Creates a new message object.
         """
+        # Clean up empty descriptions from references
+        if "references" in params and params["references"]:
+            for ref in params["references"]:
+                if "description" in ref and not ref["description"]:
+                    ref.pop("description")
+
         return cast(
             "Message",
             cls._static_request(
@@ -176,6 +193,12 @@ class Message(APIResource["Message"]):
         """
         Creates a new message object.
         """
+        # Clean up empty descriptions from references
+        if "references" in params and params["references"]:
+            for ref in params["references"]:
+                if "description" in ref and not ref["description"]:
+                    ref.pop("description")
+
         return cast(
             "Message",
             await cls._static_request_async(
@@ -198,6 +221,12 @@ class Message(APIResource["Message"]):
         """
         Updates an existing message object.
         """
+        # Clean up empty descriptions from references
+        if "references" in params and params["references"]:
+            for ref in params["references"]:
+                if "description" in ref and not ref["description"]:
+                    ref.pop("description")
+
         url = "%s/%s" % (cls.class_url(), quote_plus(id))
         return cast(
             "Message",
@@ -221,6 +250,12 @@ class Message(APIResource["Message"]):
         """
         Updates an existing message object.
         """
+        # Clean up empty descriptions from references
+        if "references" in params and params["references"]:
+            for ref in params["references"]:
+                if "description" in ref and not ref["description"]:
+                    ref.pop("description")
+
         url = "%s/%s" % (cls.class_url(), quote_plus(id))
         return cast(
             "Message",
@@ -303,3 +338,61 @@ class Message(APIResource["Message"]):
             company_id,
             params=params,
         )
+
+    @classmethod
+    def create_event(
+        cls,
+        user_id: str,
+        company_id: str,
+        **params: Unpack["Message.CreateEventParams"],
+    ) -> "Message":
+        """
+        Creates a new message event object.
+        """
+        # Clean up empty descriptions from references
+        if "references" in params and params["references"]:
+            for ref in params["references"]:
+                if "description" in ref and not ref["description"]:
+                    ref.pop("description")
+
+        message_id = params.get("messageId")
+        params.pop("messageId", None)
+        return cast(
+            "Message",
+            cls._static_request(
+                "post",
+                f"{cls.class_url()}/{message_id}/event",
+                user_id,
+                company_id,
+                params,
+            ),
+        )
+
+    @classmethod
+    async def create_event_async(
+        cls,
+        user_id: str,
+        company_id: str,
+        **params: Unpack["Message.CreateEventParams"],
+    ) -> "Message":
+        """
+        Creates a new message event object.
+        """
+        # Clean up empty descriptions from references
+        if "references" in params and params["references"]:
+            for ref in params["references"]:
+                if "description" in ref and not ref["description"]:
+                    ref.pop("description")
+
+        message_id = params.get("messageId")
+        params.pop("messageId", None)
+        return cast(
+            "Message",
+            await cls._static_request_async(
+                "post",
+                f"{cls.class_url()}/{message_id}/event",
+                user_id,
+                company_id,
+                params,
+            ),
+        )

unique_sdk/api_resources/_message_execution.py

@@ -1,4 +1,4 @@
-from typing import ClassVar, Literal, Unpack, cast
+from typing import ClassVar, Literal, NotRequired, Unpack, cast
 
 from unique_sdk._api_resource import APIResource
 from unique_sdk._request_options import RequestOptions
@@ -8,16 +8,21 @@ class MessageExecution(APIResource["MessageExecution"]):
     OBJECT_NAME: ClassVar[Literal["message_execution"]] = "message_execution"
     RESOURCE_URL = "/message-execution"
 
+    TypeLiteral = Literal["DEEP_RESEARCH"]
+    StatusLiteral = Literal["PENDING", "RUNNING", "COMPLETED", "FAILED"]
+    # Only COMPLETED and FAILED statuses are allowed for update
+    UpdateStatusLiteral = Literal["COMPLETED", "FAILED"]
+
     class CreateMessageExecutionParams(RequestOptions):
         """
         Parameters for creating a message execution.
         """
 
         messageId: str
-        chatId: str
-        type: Literal["DEEP_RESEARCH"] = "DEEP_RESEARCH"
-        secondsRemaining: int | None = None
-        percentageCompleted: int | None = None
+        type: "MessageExecution.TypeLiteral"
+        isQueueable: NotRequired[bool | None]
+        executionOptions: NotRequired[dict | None]
+        progressTitle: NotRequired[str | None]
 
     class GetMessageExecutionParams(RequestOptions):
         """
@@ -32,18 +37,23 @@ class MessageExecution(APIResource["MessageExecution"]):
         """
 
         messageId: str
-        status: Literal["COMPLETED", "FAILED"]
-        secondsRemaining: int | None = None
-        percentageCompleted: int | None = None
-
-    messageExecutionId: str | None
-    messageId: str | None
-    status: Literal["PENDING", "RUNNING", "COMPLETED", "FAILED"]
-    type: Literal["DEEP_RESEARCH"] = "DEEP_RESEARCH"
+        status: NotRequired["MessageExecution.UpdateStatusLiteral | None"]
+        secondsRemaining: NotRequired[int | None]
+        percentageCompleted: NotRequired[int | None]
+        progressTitle: NotRequired[str | None]
+
+    id: str
+    messageId: str
+    status: "MessageExecution.StatusLiteral"
+    type: "MessageExecution.TypeLiteral"
     secondsRemaining: int | None
     percentageCompleted: int | None
+    isQueueable: bool
+    executionOptions: dict | None
+    progressTitle: str | None
+    positionInQueue: int | None
     createdAt: str
-    updatedAt: str | None
+    updatedAt: str
 
     @classmethod
     def create(

unique_sdk/api_resources/_message_log.py

@@ -1,4 +1,4 @@
-from typing import ClassVar, Literal, TypedDict, Unpack, cast
+from typing import ClassVar, Literal, NotRequired, Optional, TypedDict, Unpack, cast
 
 from unique_sdk._api_resource import APIResource
 from unique_sdk._request_options import RequestOptions
@@ -8,11 +8,14 @@ class MessageLog(APIResource["MessageLog"]):
     OBJECT_NAME: ClassVar[Literal["message_log"]] = "message_log"
     RESOURCE_URL = "/message-log"
 
+    StatusLiteral = Literal["RUNNING", "COMPLETED", "FAILED"]
+
     class Reference(TypedDict):
         name: str
-        url: str | None
+        description: Optional[str]
+        url: Optional[str]
         sequenceNumber: int
-        originalIndex: list[int] | None
+        originalIndex: Optional[list[int]]
         sourceId: str
         source: str
 
@@ -23,33 +26,33 @@ class MessageLog(APIResource["MessageLog"]):
 
         messageId: str
         text: str
-        status: Literal["RUNNING", "COMPLETED", "FAILED"]
+        status: "MessageLog.StatusLiteral"
         order: int
-        details: dict | None = None
-        uncitedReferences: dict | None = None
-        references: list["MessageLog.Reference"] | None = None
+        details: NotRequired[dict | None]
+        uncitedReferences: NotRequired[dict | None]
+        references: NotRequired[list["MessageLog.Reference"] | None]
 
     class UpdateMessageLogParams(RequestOptions):
         """
         Parameters for updating a message log.
         """
 
-        text: str | None
-        status: Literal["RUNNING", "COMPLETED", "FAILED"] | None
-        order: int
-        details: dict | None = None
-        uncitedReferences: dict | None = None
-        references: list["MessageLog.Reference"] | None = None
-
-    messageLogId: str | None
-    messageId: str | None
-    status: Literal["RUNNING", "COMPLETED", "FAILED"]
-    text: str | None
+        text: NotRequired[str | None]
+        status: NotRequired["MessageLog.StatusLiteral | None"]
+        order: NotRequired[int | None]
+        details: NotRequired[dict | None]
+        uncitedReferences: NotRequired[dict | None]
+        references: NotRequired[list["MessageLog.Reference"] | None]
+
+    id: str
+    messageId: str
+    status: "MessageLog.StatusLiteral"
+    text: str
     details: dict
     uncitedReferences: dict
     order: int
     createdAt: str
-    updatedAt: str | None
+    updatedAt: str
 
     @classmethod
     def create(
@@ -61,6 +64,12 @@ class MessageLog(APIResource["MessageLog"]):
         """
         Create a MessageLog.
         """
+        # Clean up empty descriptions from references
+        if "references" in params and params["references"]:
+            for ref in params["references"]:
+                if "description" in ref and not ref["description"]:
+                    ref.pop("description")
+
         return cast(
             "MessageLog",
             cls._static_request(
@@ -82,6 +91,12 @@ class MessageLog(APIResource["MessageLog"]):
         """
         Async create a MessageLog.
         """
+        # Clean up empty descriptions from references
+        if "references" in params and params["references"]:
+            for ref in params["references"]:
+                if "description" in ref and not ref["description"]:
+                    ref.pop("description")
+
         return cast(
             "MessageLog",
             await cls._static_request_async(
@@ -104,6 +119,12 @@ class MessageLog(APIResource["MessageLog"]):
         """
         Update a MessageLog.
         """
+        # Clean up empty descriptions from references
+        if "references" in params and params["references"]:
+            for ref in params["references"]:
+                if "description" in ref and not ref["description"]:
+                    ref.pop("description")
+
         return cast(
             "MessageLog",
             cls._static_request(
@@ -126,6 +147,12 @@ class MessageLog(APIResource["MessageLog"]):
         """
         Async update a MessageLog.
         """
+        # Clean up empty descriptions from references
+        if "references" in params and params["references"]:
+            for ref in params["references"]:
+                if "description" in ref and not ref["description"]:
+                    ref.pop("description")
+
         return cast(
             "MessageLog",
             await cls._static_request_async(

unique_sdk/api_resources/_search_string.py

@@ -23,7 +23,6 @@ class SearchString(APIResource["SearchString"]):
                 "AZURE_GPT_4_0613",
                 "AZURE_GPT_4_32K_0613",
                 "AZURE_GPT_4_TURBO_1106",
-                "AZURE_GPT_35_TURBO_INSTRUCT_0914",
             ]
         ]