unique_toolkit 0.7.2__py3-none-any.whl → 0.7.4__py3-none-any.whl

unique_toolkit/language_model/builder.py

@@ -2,6 +2,8 @@ from typing_extensions import Self
 
 from unique_toolkit.language_model import (
     LanguageModelAssistantMessage,
+    LanguageModelFunction,
+    LanguageModelFunctionCall,
     LanguageModelMessage,
     LanguageModelMessageRole,
     LanguageModelMessages,
@@ -48,11 +50,22 @@ class MessagesBuilder:
         self.messages.append(message)
         return self
 
-    def assistant_message_append(self, content: str) -> Self:
+    def assistant_message_append(
+        self, content: str, tool_calls: list[LanguageModelFunction] | None = None
+    ) -> Self:
         """Appends an assistant message to the messages list."""
         message = LanguageModelAssistantMessage(content=content)
+        if tool_calls:
+            message.tool_calls = [
+                LanguageModelFunctionCall(
+                    id=tool_call.id,
+                    type="function",
+                    function=tool_call,
+                )
+                for tool_call in tool_calls
+            ]
         self.messages.append(message)
-        return self  # Return self to allow method chaining
+        return self
 
     def tool_message_append(self, name: str, tool_call_id: str, content: str) -> Self:
         """Appends a tool message to the messages list."""
@@ -60,7 +73,7 @@ class MessagesBuilder:
             name=name, tool_call_id=tool_call_id, content=content
         )
         self.messages.append(message)
-        return self  # Return self to allow method chaining
+        return self
 
     def build(self, reset: bool = True) -> LanguageModelMessages:
         """Returns the list of messages and resets the builder"""

unique_toolkit/language_model/infos.py

@@ -20,6 +20,7 @@ class LanguageModelName(StrEnum):
     AZURE_o1_2024_1217 = "AZURE_o1_2024_1217"
     AZURE_o1_MINI_2024_0912 = "AZURE_o1_MINI_2024_0912"
     AZURE_o3_MINI_2025_0131 = "AZURE_o3_MINI_2025_0131"
+    AZURE_GPT_45_PREVIEW_2025_0227 = "AZURE_GPT_45_PREVIEW_2025_0227"
 
 
 class EncoderName(StrEnum):
@@ -275,6 +276,24 @@ class LanguageModelInfo(BaseModel):
                     info_cutoff_at=date(2023, 10, 1),
                     published_at=date(2025, 1, 31),
                 )
+            case LanguageModelName.AZURE_GPT_45_PREVIEW_2025_0227:
+                return cls(
+                    name=model_name,
+                    capabilities=[
+                        ModelCapabilities.STRUCTURED_OUTPUT,
+                        ModelCapabilities.FUNCTION_CALLING,
+                        ModelCapabilities.STREAMING,
+                        ModelCapabilities.VISION,
+                    ],
+                    provider=LanguageModelProvider.AZURE,
+                    version="2025-02-27",
+                    encoder_name=EncoderName.O200K_BASE,
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=128_000, token_limit_output=16_384
+                    ),
+                    info_cutoff_at=date(2023, 10, 1),
+                    published_at=date(2025, 2, 27),
+                )
             case _:
                 if isinstance(model_name, LanguageModelName):
                     raise ValueError(

{unique_toolkit-0.7.2.dist-info → unique_toolkit-0.7.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.7.2
+Version: 0.7.4
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -111,6 +111,11 @@ 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.7.4] - 2025-03-20
+- add `AZURE_GPT_45_PREVIEW_2025_0227` as part of the models
+
+## [0.7.3] - 2025-03-20
+- Enable handling tool calls in message builder
 
 ## [0.7.2] - 2025-03-17
 - HotFix `ContentService.search_content_chunks` to use `chat_id` from event if provided.

{unique_toolkit-0.7.2.dist-info → unique_toolkit-0.7.4.dist-info}/RECORD

@@ -45,10 +45,10 @@ unique_toolkit/evaluators/hallucination/utils.py,sha256=4KTJH8low_fBzOcuVlcHB2FR
 unique_toolkit/evaluators/output_parser.py,sha256=eI72qkzK1dZyUvnfP2SOAQCGBj_-PwX5wy_aLPMsJMY,883
 unique_toolkit/evaluators/schemas.py,sha256=Jaue6Uhx75X1CyHKWj8sT3RE1JZXTqoLtfLt2xQNCX8,2507
 unique_toolkit/language_model/__init__.py,sha256=jWko_vQj48wjnpTtlkg8iNdef0SMI3FN2kGywXRTMzg,1880
-unique_toolkit/language_model/builder.py,sha256=_lUL2shMN1Je2LLYu2PiehOvfWtn5pg1yUuYyiw3f_c,2710
+unique_toolkit/language_model/builder.py,sha256=aIAXWWUoB5G-HONJiAt3MdRGd4jdP8nA-HYX2D2WlSI,3048
 unique_toolkit/language_model/constants.py,sha256=B-topqW0r83dkC_25DeQfnPk3n53qzIHUCBS7YJ0-1U,119
 unique_toolkit/language_model/functions.py,sha256=I5jHhHsKoq7GwEQyTrM8LXB2n_6dvMAk7UklenjuHSY,7945
-unique_toolkit/language_model/infos.py,sha256=HRdQLG3oetwc-bPHtsXvDRRLsK7fSsC1q8FubvxLMsw,16459
+unique_toolkit/language_model/infos.py,sha256=2X5Y6jBKYAysa9Gmz2YYhjySlJn5uyOD4wZJ0r_Im9w,17383
 unique_toolkit/language_model/prompt.py,sha256=JSawaLjQg3VR-E2fK8engFyJnNdk21zaO8pPIodzN4Q,3991
 unique_toolkit/language_model/schemas.py,sha256=rrwzUgKANFOrdehCULW8Hh03uRW3tsE5dXpWqxmClfg,8618
 unique_toolkit/language_model/service.py,sha256=GupYD4uDZjy1TfVQW3jichmgQwiSgQCj350FtL4O0W4,5569
@@ -58,7 +58,7 @@ unique_toolkit/short_term_memory/constants.py,sha256=698CL6-wjup2MvU19RxSmQk3gX7
 unique_toolkit/short_term_memory/functions.py,sha256=3WiK-xatY5nh4Dr5zlDUye1k3E6kr41RiscwtTplw5k,4484
 unique_toolkit/short_term_memory/schemas.py,sha256=OhfcXyF6ACdwIXW45sKzjtZX_gkcJs8FEZXcgQTNenw,1406
 unique_toolkit/short_term_memory/service.py,sha256=0wrwuo6K17Edcuiwp5LMCvxFuBSDP82HF0_5qwi0nYc,5307
-unique_toolkit-0.7.2.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-0.7.2.dist-info/METADATA,sha256=grlxS6ihrKZ5fk-IeNN7iGqMK0iKVkAPMOzN8Kx-u28,20871
-unique_toolkit-0.7.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.7.2.dist-info/RECORD,,
+unique_toolkit-0.7.4.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-0.7.4.dist-info/METADATA,sha256=PxjJfBaWJRBkKB0HxMbsOuuAHacJatew_91BK7IK6Ow,21029
+unique_toolkit-0.7.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-0.7.4.dist-info/RECORD,,