unique-toolkit 2026.28.0.dev10__tar.gz → 2026.28.0.dev11__tar.gz

{unique_toolkit-2026.28.0.dev10 → unique_toolkit-2026.28.0.dev11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.28.0.dev10
+Version: 2026.28.0.dev11
 Summary: 
 Author: Cedric Klinkert, Martin Fadler, Sadique Sheik, Fabian Schläpfer, Pascal Hauri, Ejiroghene Oki-Peter, Rami Azouz, Ricardo Visini, Sanzio Monti
 Author-email: Cedric Klinkert <cedric.klinkert@unique.ch>, Martin Fadler <martin.fadler@unique.ch>, Sadique Sheik <sadique@unique.ch>, Fabian Schläpfer <fabian@unique.ch>, Pascal Hauri <pascal@unique.ch>, Ejiroghene Oki-Peter <oki-peter.ejiroghene@unique.ai>, Rami Azouz <rami.ext@unique.ch>, Ricardo Visini <ricardo.visini@unique.ch>, Sanzio Monti <sanzio.monti@unique.ai>
@@ -16,7 +16,7 @@ Requires-Dist: sseclient>=0.0.27,<0.0.28
 Requires-Dist: openai>=1.105.0,<3
 Requires-Dist: platformdirs>=4.0.0,<5
 Requires-Dist: pillow>=12.1.1
-Requires-Dist: unique-sdk>=2026.28.0.dev11,<2026.28.0rc0
+Requires-Dist: unique-sdk>=2026.28.0.dev12,<2026.28.0rc0
 Requires-Dist: jambo>=0.1.2,<0.2
 Requires-Dist: jinja2>=3.1.6,<4
 Requires-Dist: docxtpl>=0.20.1,<0.21

{unique_toolkit-2026.28.0.dev10 → unique_toolkit-2026.28.0.dev11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "2026.28.0.dev10"
+version = "2026.28.0.dev11"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"
@@ -28,7 +28,7 @@ dependencies = [
     "openai>=1.105.0,<3",
     "platformdirs>=4.0.0,<5",
     "pillow>=12.1.1",
-    "unique-sdk>=2026.28.0.dev11,<2026.28.0rc0",
+    "unique-sdk>=2026.28.0.dev12,<2026.28.0rc0",
     "jambo>=0.1.2,<0.2",
     "jinja2>=3.1.6,<4",
     "docxtpl>=0.20.1,<0.21",

{unique_toolkit-2026.28.0.dev10 → unique_toolkit-2026.28.0.dev11}/unique_toolkit/chat/functions.py

@@ -267,6 +267,9 @@ def create_message(
     references: list[ContentReference] | None = None,
     debug_info: dict[str, Any] | None = None,
     set_completed_at: bool | None = False,
+    segment_kind: str | None = None,
+    response_turn_id: str | None = None,
+    segment_index: int | None = None,
 ) -> ChatMessage:
     """Creates a message in the chat session synchronously.
 
@@ -281,6 +284,9 @@ def create_message(
         references (list[ContentReference], optional): list of ContentReference objects. Defaults to None.
         debug_info (dict[str, Any]], optional): Debug information. Defaults to None.
         set_completed_at (Optional[bool]): Whether to set the completedAt field with the current date time. Defaults to False.
+        segment_kind (str, optional): The segment kind (e.g. "PROCESS", "PREFACE", "ELICITATION", "ANSWER"). Defaults to None.
+        response_turn_id (str, optional): The response turn ID grouping segments of the same assistant turn. Defaults to None.
+        segment_index (int, optional): The segment index within the turn. Defaults to None (derived by node-chat).
 
     Returns:
         ChatMessage: The created message.
@@ -304,6 +310,9 @@ def create_message(
             references=references,
             debug_info=debug_info,
             set_completed_at=set_completed_at,
+            segment_kind=segment_kind,
+            response_turn_id=response_turn_id,
+            segment_index=segment_index,
         )
 
         message = unique_sdk.Message.create(**params)
@@ -324,8 +333,11 @@ async def create_message_async(
     references: list[ContentReference] | None = None,
     debug_info: dict[str, Any] | None = None,
     set_completed_at: bool | None = False,
+    segment_kind: str | None = None,
+    response_turn_id: str | None = None,
+    segment_index: int | None = None,
 ):
-    """Creates a message in the chat session synchronously.
+    """Creates a message in the chat session asynchronously.
 
     Args:
         user_id (str): The user ID.
@@ -338,6 +350,9 @@ async def create_message_async(
         references (list[ContentReference], optional): list of ContentReference objects. Defaults to None.
         debug_info (dict[str, Any]], optional): Debug information. Defaults to None.
         set_completed_at (Optional[bool]): Whether to set the completedAt field with the current date time. Defaults to False.
+        segment_kind (str, optional): The segment kind (e.g. "PROCESS", "PREFACE", "ELICITATION", "ANSWER"). Defaults to None.
+        response_turn_id (str, optional): The response turn ID grouping segments of the same assistant turn. Defaults to None.
+        segment_index (int, optional): The segment index within the turn. Defaults to None (derived by node-chat).
 
     Returns:
         ChatMessage: The created message.
@@ -361,6 +376,9 @@ async def create_message_async(
             references=references,
             debug_info=debug_info,
             set_completed_at=set_completed_at,
+            segment_kind=segment_kind,
+            response_turn_id=response_turn_id,
+            segment_index=segment_index,
         )
 
         message = await unique_sdk.Message.create_async(**params)
@@ -381,6 +399,9 @@ def _construct_message_create_params(
     references: list[ContentReference] | None = None,
     debug_info: dict[str, Any] | None = None,
     set_completed_at: bool | None = False,
+    segment_kind: str | None = None,
+    response_turn_id: str | None = None,
+    segment_index: int | None = None,
 ) -> dict[str, Any]:
     if original_content is None:
         original_content = content
@@ -399,6 +420,12 @@ def _construct_message_create_params(
         "debugInfo": debug_info or {},
         "completedAt": _time_utils.get_datetime_now() if set_completed_at else None,
     }
+    if segment_kind is not None:
+        params["segmentKind"] = segment_kind
+    if response_turn_id is not None:
+        params["responseTurnId"] = response_turn_id
+    if segment_index is not None:
+        params["segmentIndex"] = segment_index
     return params
 
 

{unique_toolkit-2026.28.0.dev10 → unique_toolkit-2026.28.0.dev11}/unique_toolkit/chat/schemas.py

@@ -146,6 +146,9 @@ class ChatMessage(BaseModel):
     updated_at: datetime | None = None
     references: list[ContentReference] | None = []
     assessment: list[ChatMessageAssessment] | None = None
+    segment_kind: str | None = None
+    response_turn_id: str | None = None
+    segment_index: int | None = None
 
     # TODO make sdk return role consistently in lowercase
     # Currently needed as sdk returns role in uppercase
@@ -213,8 +216,9 @@ class MessageLogUncitedReferences(BaseModel):
 
 class MessageLogEvent(BaseModel):
     model_config = model_config
-    type: Literal["WebSearch", "InternalSearch"]
+    type: Literal["WebSearch", "InternalSearch", "Elicitation", "Thinking", "ToolCall"]
     text: str
+    step_type: str | None = None
 
 
 class MessageLogDetails(BaseModel):
@@ -223,6 +227,10 @@ class MessageLogDetails(BaseModel):
     status: str | None = Field(
         default=None, description="Overarching status of the current message log"
     )
+    debug: dict[str, Any] | None = Field(
+        default=None,
+        description="Raw tool tracing: tool name, input args (stdin), truncated output (stdout) for debugging",
+    )
 
 
 class MessageLog(BaseModel):

{unique_toolkit-2026.28.0.dev10 → unique_toolkit-2026.28.0.dev11}/unique_toolkit/services/chat_service.py

@@ -471,6 +471,9 @@ class ChatService(ChatServiceDeprecated):
         references: list[ContentReference] | None = None,
         debug_info: dict[str, Any] | None = None,
         set_completed_at: bool | None = False,
+        segment_kind: str | None = None,
+        response_turn_id: str | None = None,
+        segment_index: int | None = None,
     ) -> ChatMessage:
         """Creates a message in the chat session synchronously.
 
@@ -480,6 +483,9 @@ class ChatService(ChatServiceDeprecated):
             references (list[ContentReference]): list of ContentReference objects. Defaults to None.
             debug_info (dict[str, Any]]): Debug information. Defaults to None.
             set_completed_at (Optional[bool]): Whether to set the completedAt field with the current date time. Defaults to False.
+            segment_kind (str, optional): The segment kind (e.g. "PROCESS", "PREFACE", "ELICITATION", "ANSWER"). Defaults to None.
+            response_turn_id (str, optional): The response turn ID grouping segments of the same assistant turn. Defaults to None.
+            segment_index (int, optional): The segment index within the turn. Defaults to None (derived by node-chat).
 
         Returns:
             ChatMessage: The created message.
@@ -499,6 +505,9 @@ class ChatService(ChatServiceDeprecated):
             references=references,
             debug_info=debug_info,
             set_completed_at=set_completed_at,
+            segment_kind=segment_kind,
+            response_turn_id=response_turn_id,
+            segment_index=segment_index,
         )
         # Update the assistant message id
         self._assistant_message_id = chat_message.id or "unknown"
@@ -511,6 +520,9 @@ class ChatService(ChatServiceDeprecated):
         references: list[ContentReference] | None = None,
         debug_info: dict[str, Any] | None = None,
         set_completed_at: bool | None = False,
+        segment_kind: str | None = None,
+        response_turn_id: str | None = None,
+        segment_index: int | None = None,
     ) -> ChatMessage:
         """Creates a message in the chat session asynchronously.
 
@@ -520,6 +532,9 @@ class ChatService(ChatServiceDeprecated):
             references (list[ContentReference]): list of references. Defaults to None.
             debug_info (dict[str, Any]]): Debug information. Defaults to None.
             set_completed_at (Optional[bool]): Whether to set the completedAt field with the current date time. Defaults to False.
+            segment_kind (str, optional): The segment kind (e.g. "PROCESS", "PREFACE", "ELICITATION", "ANSWER"). Defaults to None.
+            response_turn_id (str, optional): The response turn ID grouping segments of the same assistant turn. Defaults to None.
+            segment_index (int, optional): The segment index within the turn. Defaults to None (derived by node-chat).
 
         Returns:
             ChatMessage: The created message.
@@ -539,6 +554,9 @@ class ChatService(ChatServiceDeprecated):
             references=references,
             debug_info=debug_info,
             set_completed_at=set_completed_at,
+            segment_kind=segment_kind,
+            response_turn_id=response_turn_id,
+            segment_index=segment_index,
         )
         # Update the assistant message id
         self._assistant_message_id = chat_message.id or "unknown"