unique_sdk 0.10.18__py3-none-any.whl → 0.10.20__py3-none-any.whl

unique_sdk/api_resources/_integrated.py

@@ -1,15 +1,29 @@
 from typing import (
+    TYPE_CHECKING,
     Any,
     ClassVar,
-    Dict,
     List,
     Literal,
+    NotRequired,
     Optional,
-    TypedDict,
+    Union,
+    Unpack,
     cast,
 )
 
-from typing_extensions import NotRequired, Unpack
+from typing_extensions import TypedDict
+
+# Avoid introducing a dependency on the openai sdk as it's only used for type hints
+if TYPE_CHECKING:
+    from openai.types.responses import (
+        ResponseIncludable,
+        ResponseInputParam,
+        ResponseOutputItem,
+        ResponseTextConfigParam,
+        ToolParam,
+        response_create_params,
+    )
+    from openai.types.shared_params import Metadata, Reasoning
 
 from unique_sdk._api_resource import APIResource
 from unique_sdk._request_options import RequestOptions
@@ -35,63 +49,49 @@ class Integrated(APIResource["Integrated"]):
         content: str
         name: Optional[str]
 
-    class CreateStream(RequestOptions):
-        model: NotRequired[
-            Literal[
-                "AZURE_GPT_4_0613",
-                "AZURE_GPT_4_32K_0613",
-            ]
-        ]
-        timeout: NotRequired["int"]
-        messages: List["Integrated.ChatCompletionRequestMessage"]
+    class CommonIntegratedParams(RequestOptions):
+        model: NotRequired[str]
         searchContext: NotRequired[List["Integrated.SearchResult"]]
         chatId: str
         assistantId: str
         assistantMessageId: str
         userMessageId: str
         startText: NotRequired["str"]
-        debugInfo: NotRequired[Dict[str, Any]]
+        debugInfo: NotRequired[dict[str, Any]]
+
+    class CreateStream(CommonIntegratedParams):
+        timeout: NotRequired["int"]
+        messages: List["Integrated.ChatCompletionRequestMessage"]
 
     # For further details about the responses parameters, see the OpenAI API documentation.
-    class CreateStreamResponseParams(TypedDict):
-        debugInfo: Optional[Dict[str, Any]] = None
-        input: Any
-        model: str
-        searchContext: Optional[List["Integrated.SearchResult"]] = None
-        chatId: str
-        assistantMessageId: str
-        userMessageId: str
-        startText: str | None = None
-        include: Optional[
-            list[
-                Literal[
-                    "computer_call_output.output.image_url",
-                    "file_search_call.results",
-                    "message.input_image.image_url",
-                    "reasoning.encrypted_content",
-                ]
-            ]
-        ] = None
-        instructions: str | None = None
-        max_output_tokens: int | None = None
-        metadata: Optional[Dict[str, str]] = None
-        parallel_tool_calls: float | None = None
-        temperature: float | None = None
-        text: Any
-        tool_choice: Any
-        tools: Any
-        top_p: float | None = None
-        reasoning: Any
+    # Note that other parameters from openai.resources.responses.Response.create can be passed
+    class CreateStreamResponsesOpenaiParams(TypedDict):
+        include: NotRequired[list["ResponseIncludable"] | None]
+        instructions: NotRequired[str | None]
+        max_output_tokens: NotRequired[int | None]
+        metadata: NotRequired[Union["Metadata", None]]
+        parallel_tool_calls: NotRequired[bool | None]
+        temperature: NotRequired[float | None]
+        text: NotRequired["ResponseTextConfigParam"]
+        tool_choice: NotRequired["response_create_params.ToolChoice"]
+        tools: NotRequired[list["ToolParam"]]
+        top_p: NotRequired[float | None]
+        reasoning: NotRequired["Reasoning"]
+
+    class CreateStreamResponsesParams(CommonIntegratedParams):
+        input: Union[str, "ResponseInputParam"]
+        options: NotRequired["Integrated.CreateStreamResponsesOpenaiParams"]
 
     class ToolCall(TypedDict):
         id: str
-        name: str | None = None
-        arguments: str | None = None
+        name: str | None
+        arguments: str | None
 
     class ResponsesStreamResult(TypedDict):
         id: str
         message: Message
         toolCalls: List["Integrated.ToolCall"]
+        output: list["ResponseOutputItem"]
 
     @classmethod
     def chat_stream_completion(
@@ -146,7 +146,7 @@ class Integrated(APIResource["Integrated"]):
         cls,
         user_id: str,
         company_id: str,
-        **params: Unpack["Integrated.CreateStreamResponseParams"],
+        **params: Unpack["Integrated.CreateStreamResponsesParams"],
     ) -> "Integrated.ResponsesStreamResult":
         """
         Executes a call to the language model and streams to the chat in real-time.
@@ -154,7 +154,7 @@ class Integrated(APIResource["Integrated"]):
         In the form of [sourceX]. The reference documents must be given as a list in searchContext.
         """
         return cast(
-            "Integrated.Responses",
+            "Integrated.ResponsesStreamResult",
             cls._static_request(
                 "post",
                 "/integrated/chat/stream-responses",
@@ -169,7 +169,7 @@ class Integrated(APIResource["Integrated"]):
         cls,
         user_id: str,
         company_id: str,
-        **params: Unpack["Integrated.CreateStreamResponseParams"],
+        **params: Unpack["Integrated.CreateStreamResponsesParams"],
     ) -> "Integrated.ResponsesStreamResult":
         """
         Executes a call to the language model and streams to the chat in real-time.
@@ -177,7 +177,7 @@ class Integrated(APIResource["Integrated"]):
         In the form of [sourceX]. The reference documents must be given as a list in searchContext.
         """
         return cast(
-            "Integrated.Responses",
+            "Integrated.ResponsesStreamResult",
             cls._static_request(
                 "post",
                 "/integrated/chat/stream-responses",

unique_sdk/utils/chat_in_space.py

@@ -1,7 +1,8 @@
 import asyncio
-from typing import List
+from typing import List, Literal
 
 from unique_sdk.api_resources._content import Content
+from unique_sdk.api_resources._message import Message
 from unique_sdk.api_resources._space import Space
 from unique_sdk.utils.file_io import upload_file
 
@@ -16,6 +17,7 @@ async def send_message_and_wait_for_completion(
     chat_id: str = None,
     poll_interval: float = 1.0,
     max_wait: float = 60.0,
+    stop_condition: Literal["stoppedStreamingAt", "completedAt"] = "stoppedStreamingAt",
 ) -> "Space.Message":
     """
     Sends a prompt asynchronously and polls for completion. (until stoppedStreamingAt is not None)
@@ -27,6 +29,7 @@ async def send_message_and_wait_for_completion(
         text: The prompt text.
         poll_interval: Seconds between polls.
         max_wait: Maximum seconds to wait for completion.
+        stop_condition: Defines when to expect a response back, when the assistant stop streaming or when it completes the message. (default: "stoppedStreamingAt")
         **kwargs: Additional parameters for the prompt.
 
     Returns:
@@ -42,11 +45,21 @@ async def send_message_and_wait_for_completion(
         scopeRules=scope_rules,
     )
     chat_id = response.get("chatId")
+    message_id = response.get("id")
 
     max_attempts = int(max_wait // poll_interval)
     for _ in range(max_attempts):
         answer = Space.get_latest_message(user_id, company_id, chat_id)
-        if answer.get("stoppedStreamingAt") is not None:
+        if answer.get(stop_condition) is not None:
+            try:
+                user_message = Message.retrieve(
+                    user_id, company_id, message_id, chatId=chat_id
+                )
+                debug_info = user_message.get("debugInfo")
+                answer["debugInfo"] = debug_info
+            except Exception as e:
+                print(f"Failed to load debug info from user message: {e}")
+
             return answer
         await asyncio.sleep(poll_interval)
 

{unique_sdk-0.10.18.dist-info → unique_sdk-0.10.20.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_sdk
-Version: 0.10.18
+Version: 0.10.20
 Summary: 
 License: MIT
 Author: Martin Fadler
@@ -10,6 +10,8 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Provides-Extra: openai
+Requires-Dist: openai (>=1.105.0,<2.0.0) ; extra == "openai"
 Requires-Dist: requests (>=2.32.3,<3.0.0)
 Requires-Dist: typing-extensions (>=4.9.0,<5.0.0)
 Description-Content-Type: text/markdown
@@ -1466,11 +1468,13 @@ The script sends a prompt asynchronously and continuously polls for completion,
 - `chat_id`: The ID of the chat where the message should be sent. If omitted, a new chat will be created.
 - `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).
+- `stop_condition`: Defines when to expect a response back, when the assistant stop streaming or when it completes the message. (default: "stoppedStreamingAt")
 
 The script ensures you can flexibly interact with spaces in new or ongoing chats, with fine-grained control over tools, context, and polling behavior.
 
 ```python
-latest_message = await unique_sdk.utils.chat_in_space.send_message_and_wait_for_completion(
+from unique_sdk.utils.chat_in_space import send_message_and_wait_for_completion
+latest_message = await send_message_and_wait_for_completion(
     user_id=user_id,
     company_id=company_id,
     assistant_id=assistant_id,
@@ -1497,6 +1501,7 @@ latest_message = await unique_sdk.utils.chat_in_space.send_message_and_wait_for_
             }
         ]
     },
+    stop_condition = "completedAt" # If not specified, stoppedStreamingAt will be set by default
 )
 ```
 
@@ -1572,8 +1577,16 @@ 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.20] - 2025-09-04
+- Update Responses API types
+
+## [0.10.19] - 2025-09-02
+- Improve `send_message_and_wait_for_completion`:
+    - Add option to select stop_condition `["stoppedStreamingAt", "completedAt"]`. 
+    - Load `debugInfo` from `last_user_message` for better developer experience.
+
 ## [0.10.18] - 2025-09-02
-- Temporarily remove support for udpate and delete files by filePath.
+- Temporarily remove support for update and delete files by filePath.
 
 ## [0.10.17] - 2025-09-01
 - Add function to update a file

{unique_sdk-0.10.18.dist-info → unique_sdk-0.10.20.dist-info}/RECORD

@@ -21,7 +21,7 @@ unique_sdk/api_resources/_content.py,sha256=Vi-wZ-T5f-OqBGXkA3B9dALoFHer5F8LAQlc
 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=mIyWaxJtIHlDLPFZ0FY1U9b3dmtmIcjDEbgOZtLA-DI,12871
-unique_sdk/api_resources/_integrated.py,sha256=z_DrftwjgVCi10QQqRYnG5_-95kD7Kfjogbb-dmnJuA,5854
+unique_sdk/api_resources/_integrated.py,sha256=O8e673z-RB7FRFMQYn_YEuHijebr5W7KJxkUnymbBZk,6164
 unique_sdk/api_resources/_mcp.py,sha256=zKh0dyn0QnkKk57N2zlGVN_GQoxEp5T2CS38vVm6jQY,3341
 unique_sdk/api_resources/_message.py,sha256=gEDIzg3METZU2k7m69meAuf0IWmZxnYOjbBKPRMwPYE,7688
 unique_sdk/api_resources/_message_assessment.py,sha256=SSfx6eW7zb_GKe8cFJzCqW-t-_eWEXxKP5cnIb0DhIc,2276
@@ -32,11 +32,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=6789zLwkoZqrEESiTJIBVaNi8kAKAZnqR0KMmW1AzgI,4905
 unique_sdk/utils/chat_history.py,sha256=5UqL9hF1O9pV7skbNOlEibF5rHdYsmG3m5-YEPUowOs,3037
-unique_sdk/utils/chat_in_space.py,sha256=3NeBjOu7p43V_6PrjwxyaTkgknUS10KE4QRuTlFDU_4,5232
+unique_sdk/utils/chat_in_space.py,sha256=NrH9e2lvXtj_oePG0RWUqFoTanMblF8-VgtnVfszPS8,5949
 unique_sdk/utils/file_io.py,sha256=YY8B7VJcTLOPmCXByiOfNerXGlAtjCC5EVNmAbQJ3dQ,4306
 unique_sdk/utils/sources.py,sha256=DoxxhMLcLhmDfNarjXa41H4JD2GSSDywr71hiC-4pYc,4952
 unique_sdk/utils/token.py,sha256=AzKuAA1AwBtnvSFxGcsHLpxXr_wWE5Mj4jYBbOz2ljA,1740
-unique_sdk-0.10.18.dist-info/LICENSE,sha256=EJCWoHgrXVBUb47PnjeV4MFIEOR71MAdCOIgv61J-4k,1065
-unique_sdk-0.10.18.dist-info/METADATA,sha256=aduy4luaw6fs5a2fInb5ws8VfGb_glx3CG1RrgFrnmM,53900
-unique_sdk-0.10.18.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_sdk-0.10.18.dist-info/RECORD,,
+unique_sdk-0.10.20.dist-info/LICENSE,sha256=EJCWoHgrXVBUb47PnjeV4MFIEOR71MAdCOIgv61J-4k,1065
+unique_sdk-0.10.20.dist-info/METADATA,sha256=mbVWlT-JFrSvRV5W7AjWT3a1HoZvlnvbReiu_f1HMpI,54588
+unique_sdk-0.10.20.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_sdk-0.10.20.dist-info/RECORD,,