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

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.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_sdk
-Version: 0.10.18
+Version: 0.10.19
 Summary: 
 License: MIT
 Author: Martin Fadler
@@ -1466,11 +1466,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 +1499,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 +1575,13 @@ 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.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.19.dist-info}/RECORD

@@ -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.19.dist-info/LICENSE,sha256=EJCWoHgrXVBUb47PnjeV4MFIEOR71MAdCOIgv61J-4k,1065
+unique_sdk-0.10.19.dist-info/METADATA,sha256=xppurAY1RbWZ-Z8BlMws7i_yZZhYY04pqxDj676FTpU,54448
+unique_sdk-0.10.19.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_sdk-0.10.19.dist-info/RECORD,,