unique-toolkit 2026.28.0.dev14__tar.gz → 2026.28.0.dev16__tar.gz

{unique_toolkit-2026.28.0.dev14 → unique_toolkit-2026.28.0.dev16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.28.0.dev14
+Version: 2026.28.0.dev16
 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.dev15,<2026.28.0rc0
+Requires-Dist: unique-sdk>=2026.28.0.dev18,<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.dev14 → unique_toolkit-2026.28.0.dev16}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "2026.28.0.dev14"
+version = "2026.28.0.dev16"
 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.dev15,<2026.28.0rc0",
+    "unique-sdk>=2026.28.0.dev18,<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.dev14 → unique_toolkit-2026.28.0.dev16}/unique_toolkit/chat/functions.py

@@ -66,6 +66,7 @@ def modify_message(
     debug_info: dict[str, Any] | None = None,
     message_id: str | None = None,
     set_completed_at: bool = False,
+    segment_kind: str | None = None,
 ) -> ChatMessage:
     """Modifies a chat message synchronously.
 
@@ -83,6 +84,7 @@ def modify_message(
         references (list[ContentReference]): list of ContentReference objects. Defaults to None.
         debug_info (dict[str, Any]], optional): Debug information. Defaults to None.
         set_completed_at (bool, optional): 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") to relabel the segment to in place. Defaults to None (kind unchanged).
 
     Returns:
         ChatMessage: The modified message.
@@ -106,6 +108,7 @@ def modify_message(
             debug_info=debug_info,
             message_id=message_id,
             set_completed_at=set_completed_at,
+            segment_kind=segment_kind,
         )
         message = unique_sdk.Message.modify(**params)
         return ChatMessage(**message)
@@ -128,6 +131,7 @@ async def modify_message_async(
     debug_info: dict[str, Any] | None = None,
     message_id: str | None = None,
     set_completed_at: bool = False,
+    segment_kind: str | None = None,
 ) -> ChatMessage:
     """Modifies a chat message asynchronously.
 
@@ -145,6 +149,7 @@ async def modify_message_async(
         references (list[ContentReference]): list of ContentReference objects. Defaults to None.
         debug_info (dict[str, Any]], optional): Debug information. Defaults to None.
         set_completed_at (bool, optional): 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") to relabel the segment to in place. Defaults to None (kind unchanged).
 
     Returns:
         ChatMessage: The modified message.
@@ -168,6 +173,7 @@ async def modify_message_async(
             debug_info=debug_info,
             message_id=message_id,
             set_completed_at=set_completed_at,
+            segment_kind=segment_kind,
         )
         message = await unique_sdk.Message.modify_async(**params)
         return ChatMessage(**message)
@@ -225,6 +231,7 @@ def _construct_message_modify_params(
     debug_info: dict[str, Any] | None = None,
     message_id: str | None = None,
     set_completed_at: bool = False,
+    segment_kind: str | None = None,
 ) -> dict[str, Any]:
     completed_at_datetime = None
 
@@ -242,7 +249,7 @@ def _construct_message_modify_params(
     if set_completed_at:
         completed_at_datetime = _time_utils.get_datetime_now()
 
-    params = {
+    params: dict[str, Any] = {
         "user_id": user_id,
         "company_id": company_id,
         "id": message_id,
@@ -253,6 +260,8 @@ def _construct_message_modify_params(
         "debugInfo": debug_info,
         "completedAt": completed_at_datetime,
     }
+    if segment_kind is not None:
+        params["segmentKind"] = segment_kind
     return params
 
 

{unique_toolkit-2026.28.0.dev14 → unique_toolkit-2026.28.0.dev16}/unique_toolkit/language_model/infos.py

@@ -83,12 +83,14 @@ class LanguageModelName(StrEnum):
     ANTHROPIC_CLAUDE_SONNET_4 = "litellm:anthropic-claude-sonnet-4"
     ANTHROPIC_CLAUDE_SONNET_4_5 = "litellm:anthropic-claude-sonnet-4-5"
     ANTHROPIC_CLAUDE_SONNET_4_6 = "litellm:anthropic-claude-sonnet-4-6"
+    ANTHROPIC_CLAUDE_SONNET_5 = "litellm:anthropic-claude-sonnet-5"
     ANTHROPIC_CLAUDE_OPUS_4 = "litellm:anthropic-claude-opus-4"
     ANTHROPIC_CLAUDE_OPUS_4_1 = "litellm:anthropic-claude-opus-4-1"
     ANTHROPIC_CLAUDE_OPUS_4_5 = "litellm:anthropic-claude-opus-4-5"
     ANTHROPIC_CLAUDE_OPUS_4_6 = "litellm:anthropic-claude-opus-4-6"
     ANTHROPIC_CLAUDE_OPUS_4_7 = "litellm:anthropic-claude-opus-4-7"
     ANTHROPIC_CLAUDE_OPUS_4_8 = "litellm:anthropic-claude-opus-4-8"
+    ANTHROPIC_CLAUDE_FABLE_5 = "litellm:anthropic-claude-fable-5"
     GEMINI_2_0_FLASH = "litellm:gemini-2-0-flash"
     GEMINI_2_5_FLASH = "litellm:gemini-2-5-flash"
     GEMINI_2_5_FLASH_LITE = "litellm:gemini-2-5-flash-lite"
@@ -125,12 +127,17 @@ class LanguageModelName(StrEnum):
     LITELLM_OPENAI_GPT_4_1_NANO = "litellm:openai-gpt-4-1-nano"
     LITELLM_DEEPSEEK_R1 = "litellm:deepseek-r1"
     LITELLM_DEEPSEEK_V3 = "litellm:deepseek-v3-1"
+    LITELLM_DEEPSEEK_V4_PRO = "litellm:deepseek-v4-pro"
+    LITELLM_GLM_5_1 = "litellm:glm-5.1"
+    LITELLM_GLM_5_2 = "litellm:glm-5.2"
     LITELLM_QWEN_3 = "litellm:qwen-3-235B-A22B"
     LITELLM_QWEN_3_THINKING = "litellm:qwen-3-235B-A22B-thinking"
     VERTEX_CLAUDE_SONNET_4_6 = "litellm:vertex-claude-sonnet-4-6"
+    VERTEX_CLAUDE_SONNET_5 = "litellm:vertex-claude-sonnet-5"
     VERTEX_CLAUDE_OPUS_4_6 = "litellm:vertex-claude-opus-4-6"
     VERTEX_CLAUDE_OPUS_4_7 = "litellm:vertex-claude-opus-4-7"
     VERTEX_CLAUDE_OPUS_4_8 = "litellm:vertex-claude-opus-4-8"
+    VERTEX_CLAUDE_FABLE_5 = "litellm:vertex-claude-fable-5"
 
 
 class EncoderName(StrEnum):
@@ -1590,6 +1597,33 @@ class LanguageModelInfo(BaseModel):
                     published_at=date(2026, 2, 17),
                     supported_reasoning_efforts=[],
                 )
+            case (
+                LanguageModelName.ANTHROPIC_CLAUDE_SONNET_5
+                | LanguageModelName.VERTEX_CLAUDE_SONNET_5
+            ):
+                return cls(
+                    name=model_name,
+                    capabilities=[
+                        ModelCapabilities.FUNCTION_CALLING,
+                        ModelCapabilities.STREAMING,
+                        ModelCapabilities.VISION,
+                        ModelCapabilities.REASONING,
+                    ],
+                    provider=LanguageModelProvider.LITELLM,
+                    family=ModelFamily.ANTHROPIC,
+                    version="claude-sonnet-5",
+                    encoder_name=EncoderName.O200K_BASE,  # TODO: Update encoder with litellm
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=1_000_000,
+                        token_limit_output=128_000,
+                    ),
+                    info_cutoff_at=date(2026, 1, 1),
+                    published_at=date(2026, 7, 1),
+                    temperature_bounds=TemperatureBounds(
+                        min_temperature=1.0, max_temperature=1.0
+                    ),
+                    supported_reasoning_efforts=[],
+                )
             case LanguageModelName.ANTHROPIC_CLAUDE_OPUS_4:
                 return cls(
                     name=model_name,
@@ -1725,6 +1759,30 @@ class LanguageModelInfo(BaseModel):
                     published_at=date(2026, 5, 28),
                     supported_reasoning_efforts=[],
                 )
+            case (
+                LanguageModelName.ANTHROPIC_CLAUDE_FABLE_5
+                | LanguageModelName.VERTEX_CLAUDE_FABLE_5
+            ):
+                return cls(
+                    name=model_name,
+                    capabilities=[
+                        ModelCapabilities.FUNCTION_CALLING,
+                        ModelCapabilities.STREAMING,
+                        ModelCapabilities.VISION,
+                        ModelCapabilities.REASONING,
+                    ],
+                    provider=LanguageModelProvider.LITELLM,
+                    family=ModelFamily.ANTHROPIC,
+                    version="claude-fable-5",
+                    encoder_name=EncoderName.O200K_BASE,  # TODO: Update encoder with litellm
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=1_000_000,
+                        token_limit_output=128_000,
+                    ),
+                    info_cutoff_at=date(2026, 1, 1),
+                    published_at=date(2026, 7, 1),
+                    supported_reasoning_efforts=[],
+                )
             case LanguageModelName.GEMINI_2_0_FLASH:
                 return cls(
                     name=model_name,
@@ -2627,6 +2685,63 @@ class LanguageModelInfo(BaseModel):
                     published_at=date(2025, 8, 1),
                     supported_reasoning_efforts=[],
                 )
+            case LanguageModelName.LITELLM_DEEPSEEK_V4_PRO:
+                return cls(
+                    name=model_name,
+                    provider=LanguageModelProvider.LITELLM,
+                    family=ModelFamily.DEEPSEEK,
+                    version="deepseek-v4-pro",
+                    encoder_name=EncoderName.DEEPSEEK,
+                    capabilities=[
+                        ModelCapabilities.FUNCTION_CALLING,
+                        ModelCapabilities.STRUCTURED_OUTPUT,
+                        ModelCapabilities.REASONING,
+                        ModelCapabilities.STREAMING,
+                    ],
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=1_000_000, token_limit_output=384_000
+                    ),
+                    published_at=date(2026, 1, 1),
+                    supported_reasoning_efforts=[],
+                )
+            case LanguageModelName.LITELLM_GLM_5_1:
+                return cls(
+                    name=model_name,
+                    provider=LanguageModelProvider.LITELLM,
+                    family=ModelFamily.ZAI,
+                    version="glm-5.1",
+                    encoder_name=EncoderName.O200K_BASE,
+                    capabilities=[
+                        ModelCapabilities.FUNCTION_CALLING,
+                        ModelCapabilities.STRUCTURED_OUTPUT,
+                        ModelCapabilities.REASONING,
+                        ModelCapabilities.STREAMING,
+                    ],
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=200_000, token_limit_output=128_000
+                    ),
+                    published_at=date(2026, 1, 1),
+                    supported_reasoning_efforts=[],
+                )
+            case LanguageModelName.LITELLM_GLM_5_2:
+                return cls(
+                    name=model_name,
+                    provider=LanguageModelProvider.LITELLM,
+                    family=ModelFamily.ZAI,
+                    version="glm-5.2",
+                    encoder_name=EncoderName.O200K_BASE,
+                    capabilities=[
+                        ModelCapabilities.FUNCTION_CALLING,
+                        ModelCapabilities.STRUCTURED_OUTPUT,
+                        ModelCapabilities.REASONING,
+                        ModelCapabilities.STREAMING,
+                    ],
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=1_000_000, token_limit_output=128_000
+                    ),
+                    published_at=date(2026, 1, 1),
+                    supported_reasoning_efforts=[],
+                )
             case LanguageModelName.LITELLM_QWEN_3:
                 return cls(
                     name=model_name,

{unique_toolkit-2026.28.0.dev14 → unique_toolkit-2026.28.0.dev16}/unique_toolkit/language_model/settings.py

@@ -22,6 +22,7 @@ class ModelFamily(StrEnum):
     DEEPSEEK = "deepseek"
     QWEN = "qwen"
     MISTRAL = "mistral"
+    ZAI = "zai"
     UNKNOWN = "unknown"
 
 

{unique_toolkit-2026.28.0.dev14 → unique_toolkit-2026.28.0.dev16}/unique_toolkit/services/chat_service.py

@@ -387,6 +387,7 @@ class ChatService(ChatServiceDeprecated):
         debug_info: dict[str, Any] | None = None,
         message_id: str | None = None,
         set_completed_at: bool | None = None,
+        segment_kind: str | None = None,
     ) -> ChatMessage:
         """Modifies a message in the chat session synchronously if parameter is not specified the corresponding field will remain as is.
 
@@ -397,6 +398,7 @@ class ChatService(ChatServiceDeprecated):
             debug_info (dict[str, Any]]]): Debug information. Defaults to {}.
             message_id (Optional[str]): The message ID. 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") to relabel the segment to in place, e.g. relabeling M0 from PROCESS to PREFACE once content is set. Defaults to None (kind unchanged).
 
         Returns:
             ChatMessage: The modified message.
@@ -420,6 +422,7 @@ class ChatService(ChatServiceDeprecated):
             debug_info=debug_info,
             message_id=message_id,
             set_completed_at=set_completed_at or False,
+            segment_kind=segment_kind,
         )
 
     async def modify_assistant_message_async(
@@ -430,6 +433,7 @@ class ChatService(ChatServiceDeprecated):
         debug_info: dict[str, Any] | None = None,
         message_id: str | None = None,
         set_completed_at: bool | None = False,
+        segment_kind: str | None = None,
     ) -> ChatMessage:
         """Modifies a message in the chat session asynchronously.
 
@@ -440,6 +444,7 @@ class ChatService(ChatServiceDeprecated):
             references (list[ContentReference]): list of ContentReference objects. Defaults to None.
             debug_info (dict[str, Any]], optional): Debug information. Defaults to None.
             set_completed_at (bool, optional): 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") to relabel the segment to in place, e.g. relabeling M0 from PROCESS to PREFACE once content is set. Defaults to None (kind unchanged).
 
         Returns:
             ChatMessage: The modified message.
@@ -462,6 +467,7 @@ class ChatService(ChatServiceDeprecated):
             debug_info=debug_info,
             message_id=message_id,
             set_completed_at=set_completed_at or False,
+            segment_kind=segment_kind,
         )
 
     def create_assistant_message(