unique_toolkit 0.8.9__tar.gz → 0.8.11__tar.gz

{unique_toolkit-0.8.9 → unique_toolkit-0.8.11}/CHANGELOG.md

@@ -5,6 +5,14 @@ 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.8.11] - 2025-08-15
+- Fix no tool call respoonse in ChatMessage -> Open Ai messages translation
+- Add simple append method to OpenAIMessageBuilder
+
+## [0.8.10] - 2025-08-15
+- Add min and max temperature to `LanguageModelInfo`: temperature will be clamped to the min and max temperature
+- Add default options to `LanguageModelInfo`: These are used by default
+
 ## [0.8.9] - 2025-08-15
 - Reduce input token limits for `ANTHROPIC_CLAUDE_3_7_SONNET_THINKING`, `ANTHROPIC_CLAUDE_3_7_SONNET`, `ANTHROPIC_CLAUDE_OPUS_4` and `ANTHROPIC_CLAUDE_SONNET_4` to 180_000 from 200_000
 

{unique_toolkit-0.8.9 → unique_toolkit-0.8.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.8.9
+Version: 0.8.11
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -114,6 +114,14 @@ 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.8.11] - 2025-08-15
+- Fix no tool call respoonse in ChatMessage -> Open Ai messages translation
+- Add simple append method to OpenAIMessageBuilder
+
+## [0.8.10] - 2025-08-15
+- Add min and max temperature to `LanguageModelInfo`: temperature will be clamped to the min and max temperature
+- Add default options to `LanguageModelInfo`: These are used by default
+
 ## [0.8.9] - 2025-08-15
 - Reduce input token limits for `ANTHROPIC_CLAUDE_3_7_SONNET_THINKING`, `ANTHROPIC_CLAUDE_3_7_SONNET`, `ANTHROPIC_CLAUDE_OPUS_4` and `ANTHROPIC_CLAUDE_SONNET_4` to 180_000 from 200_000
 

{unique_toolkit-0.8.9 → unique_toolkit-0.8.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_toolkit"
-version = "0.8.9"
+version = "0.8.11"
 description = ""
 authors = [
     "Martin Fadler <martin.fadler@unique.ch>",

{unique_toolkit-0.8.9 → unique_toolkit-0.8.11}/unique_toolkit/chat/schemas.py

@@ -4,6 +4,7 @@ from enum import StrEnum
 from humps import camelize
 from openai.types.chat import (
     ChatCompletionAssistantMessageParam,
+    ChatCompletionMessageParam,
     ChatCompletionUserMessageParam,
 )
 from openai.types.chat.chat_completion_message_function_tool_call_param import (
@@ -96,7 +97,7 @@ class ChatMessage(BaseModel):
             raise ValueError("tool_call_ids is required when role is 'tool'")
         return self
 
-    def to_openai_param(self):
+    def to_openai_param(self) -> ChatCompletionMessageParam:
         match self.role:
             case ChatMessageRole.USER:
                 return ChatCompletionUserMessageParam(
@@ -105,14 +106,25 @@ class ChatMessage(BaseModel):
                 )
 
             case ChatMessageRole.ASSISTANT:
-                return ChatCompletionAssistantMessageParam(
-                    role="assistant",
-                    audio=None,
-                    content=self.content or "",
-                    function_call=None,
-                    refusal=None,
-                    tool_calls=[t.to_openai_param() for t in self.tool_calls or []],
-                )
+                if self.tool_calls:
+                    assistant_message = ChatCompletionAssistantMessageParam(
+                        role="assistant",
+                        audio=None,
+                        content=self.content or "",
+                        function_call=None,
+                        refusal=None,
+                        tool_calls=[t.to_openai_param() for t in self.tool_calls],
+                    )
+                else:
+                    assistant_message = ChatCompletionAssistantMessageParam(
+                        role="assistant",
+                        audio=None,
+                        content=self.content or "",
+                        function_call=None,
+                        refusal=None,
+                    )
+
+                return assistant_message
 
             case ChatMessageRole.TOOL:
                 raise NotImplementedError

{unique_toolkit-0.8.9 → unique_toolkit-0.8.11}/unique_toolkit/chat/service.py

@@ -402,9 +402,9 @@ class ChatService:
         references: list[ContentReference] | None = None,
         debug_info: dict | None = None,
         message_id: str | None = None,
-        set_completed_at: bool = False,
+        set_completed_at: bool | None = None,
     ) -> ChatMessage:
-        """Modifies a message in the chat session synchronously.
+        """Modifies a message in the chat session synchronously if parameter is not specified the corresponding field will remain as is.
 
         Args:
             content (str, optional): The new content for the message.

{unique_toolkit-0.8.9 → unique_toolkit-0.8.11}/unique_toolkit/framework_utilities/openai/message_builder.py

@@ -49,6 +49,9 @@ class OpenAIMessageBuilder:
         builder._messages = messages.copy()
         return builder
 
+    def append(self, message: ChatCompletionMessageParam):
+        self._messages.append(message)
+
     def system_message_append(
         self,
         content: str | Iterable[ChatCompletionContentPartTextParam],

{unique_toolkit-0.8.9 → unique_toolkit-0.8.11}/unique_toolkit/language_model/functions.py

@@ -20,7 +20,11 @@ from unique_toolkit.language_model import (
     LanguageModelTool,
     LanguageModelToolDescription,
 )
-from unique_toolkit.language_model.infos import LanguageModelName
+from unique_toolkit.language_model.infos import (
+    LanguageModelInfo,
+    LanguageModelName,
+    TemperatureBounds,
+)
 from unique_toolkit.language_model.reference import (
     add_references_to_message,
 )
@@ -303,6 +307,24 @@ def __camelize_keys(data):
     return data
 
 
+def _clamp_temperature(
+    temperature: float, temperature_bounds: TemperatureBounds
+) -> float:
+    temperature = max(temperature_bounds.min_temperature, temperature)
+    temperature = min(temperature_bounds.max_temperature, temperature)
+    return round(temperature, 2)
+
+
+def _prepare_other_options(
+    other_options: dict | None,
+    default_options: dict,
+) -> dict:
+    options = default_options
+    if other_options is not None:
+        options.update(other_options)
+    return options
+
+
 def _prepare_all_completions_params_util(
     messages: LanguageModelMessages | list[ChatCompletionMessageParam],
     model_name: LanguageModelName | str,
@@ -318,6 +340,13 @@ def _prepare_all_completions_params_util(
     list[unique_sdk.Integrated.ChatCompletionRequestMessage],
     dict | None,
 ]:
+    model_info = None
+    if isinstance(model_name, LanguageModelName):
+        model_info = LanguageModelInfo.from_name(model_name)
+        other_options = _prepare_other_options(
+            other_options, model_info.default_options
+        )
+
     if isinstance(messages, LanguageModelMessages):
         options, model, messages_dict, search_context = _prepare_completion_params_util(
             messages=messages,
@@ -341,6 +370,15 @@ def _prepare_all_completions_params_util(
         )
         messages_dict = __camelize_keys(messages.copy())
 
+    if (
+        model_info is not None
+        and model_info.temperature_bounds is not None
+        and "temperature" in options
+    ):
+        options["temperature"] = _clamp_temperature(
+            temperature, model_info.temperature_bounds
+        )
+
     integrated_messages = cast(
         "list[unique_sdk.Integrated.ChatCompletionRequestMessage]",
         messages_dict,

{unique_toolkit-0.8.9 → unique_toolkit-0.8.11}/unique_toolkit/language_model/infos.py

@@ -1,6 +1,6 @@
 from datetime import date
 from enum import StrEnum
-from typing import ClassVar, Optional, Self
+from typing import Any, ClassVar, Optional, Self
 
 from pydantic import BaseModel
 from pydantic.json_schema import SkipJsonSchema
@@ -96,6 +96,11 @@ class ModelCapabilities(StrEnum):
     REASONING = "reasoning"
 
 
+class TemperatureBounds(BaseModel):
+    min_temperature: float
+    max_temperature: float
+
+
 class LanguageModelInfo(BaseModel):
     name: LanguageModelName | str
     version: str
@@ -116,6 +121,10 @@ class LanguageModelInfo(BaseModel):
     deprecated_at: date | SkipJsonSchema[None] = None
     retirement_text: str | SkipJsonSchema[None] = None
 
+    temperature_bounds: TemperatureBounds | None = None
+
+    default_options: dict[str, Any] = {}
+
     @classmethod
     def from_name(cls, model_name: LanguageModelName) -> Self:
         match model_name:
@@ -183,11 +192,19 @@ class LanguageModelInfo(BaseModel):
                         ModelCapabilities.STRUCTURED_OUTPUT,
                         ModelCapabilities.PARALLEL_FUNCTION_CALLING,
                     ],
-                    token_limits=LanguageModelTokenLimits(token_limit_input=272000, token_limit_output=128000),
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=272000, token_limit_output=128000
+                    ),
                     info_cutoff_at=date(2024, 10, 24),
                     published_at=date(2025, 8, 7),
                     deprecated_at=date(2026, 8, 7),
                     retirement_at=date(2026, 8, 7),
+                    temperature_bounds=TemperatureBounds(
+                        min_temperature=1.0, max_temperature=1.0
+                    ),
+                    default_options={
+                        "reasoning_effort": "minimal",
+                    },
                 )
             case LanguageModelName.AZURE_GPT_5_MINI_2025_0807:
                 return cls(
@@ -201,11 +218,19 @@ class LanguageModelInfo(BaseModel):
                         ModelCapabilities.VISION,
                         ModelCapabilities.STRUCTURED_OUTPUT,
                     ],
-                    token_limits=LanguageModelTokenLimits(token_limit_input=272000, token_limit_output=128000),
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=272000, token_limit_output=128000
+                    ),
                     info_cutoff_at=date(2024, 6, 24),
                     published_at=date(2025, 8, 7),
                     deprecated_at=date(2026, 8, 7),
                     retirement_at=date(2026, 8, 7),
+                    temperature_bounds=TemperatureBounds(
+                        min_temperature=1.0, max_temperature=1.0
+                    ),
+                    default_options={
+                        "reasoning_effort": "minimal",
+                    },
                 )
             case LanguageModelName.AZURE_GPT_5_NANO_2025_0807:
                 return cls(
@@ -219,19 +244,29 @@ class LanguageModelInfo(BaseModel):
                         ModelCapabilities.VISION,
                         ModelCapabilities.STRUCTURED_OUTPUT,
                     ],
-                    token_limits=LanguageModelTokenLimits(token_limit_input=272000, token_limit_output=128000),
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=272000, token_limit_output=128000
+                    ),
                     info_cutoff_at=date(2024, 5, 31),
                     published_at=date(2025, 8, 7),
                     deprecated_at=date(2026, 8, 7),
                     retirement_at=date(2026, 8, 7),
+                    temperature_bounds=TemperatureBounds(
+                        min_temperature=1.0, max_temperature=1.0
+                    ),
+                    default_options={
+                        "reasoning_effort": "minimal",
+                    },
                 )
             case LanguageModelName.AZURE_GPT_5_CHAT_2025_0807:
                 return cls(
                     name=model_name,
                     provider=LanguageModelProvider.AZURE,
                     version="2025-08-07",
-                    encoder_name=EncoderName.O200K_BASE, 
-                    token_limits=LanguageModelTokenLimits(token_limit_input=128000, token_limit_output=16384),
+                    encoder_name=EncoderName.O200K_BASE,
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=128000, token_limit_output=16384
+                    ),
                     info_cutoff_at=date(2024, 10, 24),
                     published_at=date(2025, 8, 7),
                     deprecated_at=date(2026, 8, 7),
@@ -347,6 +382,9 @@ class LanguageModelInfo(BaseModel):
                     ),
                     info_cutoff_at=date(2023, 10, 1),
                     published_at=date(2024, 9, 12),
+                    temperature_bounds=TemperatureBounds(
+                        min_temperature=1.0, max_temperature=1.0
+                    ),
                 )
             case LanguageModelName.AZURE_o1_2024_1217:
                 return cls(
@@ -366,6 +404,9 @@ class LanguageModelInfo(BaseModel):
                     ),
                     info_cutoff_at=date(2023, 10, 1),
                     published_at=date(2024, 12, 17),
+                    temperature_bounds=TemperatureBounds(
+                        min_temperature=1.0, max_temperature=1.0
+                    ),
                 )
             case LanguageModelName.AZURE_o3_MINI_2025_0131:
                 return cls(
@@ -384,6 +425,9 @@ class LanguageModelInfo(BaseModel):
                     ),
                     info_cutoff_at=date(2023, 10, 1),
                     published_at=date(2025, 1, 31),
+                    temperature_bounds=TemperatureBounds(
+                        min_temperature=1.0, max_temperature=1.0
+                    ),
                 )
             case LanguageModelName.AZURE_o3_2025_0416:
                 return cls(
@@ -403,6 +447,9 @@ class LanguageModelInfo(BaseModel):
                     ),
                     info_cutoff_at=date(2024, 5, 31),
                     published_at=date(2025, 4, 16),
+                    temperature_bounds=TemperatureBounds(
+                        min_temperature=1.0, max_temperature=1.0
+                    ),
                 )
             case LanguageModelName.AZURE_o4_MINI_2025_0416:
                 return cls(
@@ -422,6 +469,9 @@ class LanguageModelInfo(BaseModel):
                     ),
                     info_cutoff_at=date(2024, 5, 31),
                     published_at=date(2025, 4, 16),
+                    temperature_bounds=TemperatureBounds(
+                        min_temperature=1.0, max_temperature=1.0
+                    ),
                 )
             case LanguageModelName.AZURE_GPT_45_PREVIEW_2025_0227:
                 return cls(