unique_toolkit 0.5.45__py3-none-any.whl → 0.5.47__py3-none-any.whl

unique_toolkit/language_model/builder.py

@@ -0,0 +1,50 @@
+from unique_toolkit.language_model import (
+    LanguageModelAssistantMessage,
+    LanguageModelMessage,
+    LanguageModelMessages,
+    LanguageModelSystemMessage,
+    LanguageModelToolMessage,
+    LanguageModelUserMessage,
+)
+
+
+class MessagesBuilder:
+    def __init__(self):
+        self.messages: list[LanguageModelMessage] = []
+
+    def system_message_append(self, content):
+        """Appends a user message to the messages list."""
+        message = LanguageModelSystemMessage(content=content)
+        self.messages.append(message)
+        return self  # Return self to allow method chaining
+
+    def user_message_append(self, content):
+        """Appends a user message to the messages list."""
+        message = LanguageModelUserMessage(content=content)
+        self.messages.append(message)
+        return self  # Return self to allow method chaining
+
+    def assistant_message_append(self, content):
+        """Appends an assistant message to the messages list."""
+        message = LanguageModelAssistantMessage(content=content)
+        self.messages.append(message)
+        return self  # Return self to allow method chaining
+
+    def tool_message_append(self, name: str, tool_call_id: str, content: str):
+        """Appends a tool message to the messages list."""
+        message = LanguageModelToolMessage(
+            name=name, tool_call_id=tool_call_id, content=content
+        )
+        self.messages.append(message)
+        return self  # Return self to allow method chaining
+
+    def build(self, reset: bool = True) -> LanguageModelMessages:
+        """Returns the list of messages and resets the builder"""
+        messages = LanguageModelMessages(root=self.messages)
+        if reset:
+            self.messages = []
+        return messages
+
+    def model_dump(self, **kwargs):
+        """Dumps the LanguageModelMessages model"""
+        return self.build().model_dump(**kwargs)

unique_toolkit/language_model/infos.py

@@ -11,6 +11,7 @@ class LanguageModelName(StrEnum):
     AZURE_GPT_35_TURBO_0613 = "AZURE_GPT_35_TURBO_0613"
     AZURE_GPT_35_TURBO = "AZURE_GPT_35_TURBO"
     AZURE_GPT_35_TURBO_16K = "AZURE_GPT_35_TURBO_16K"
+    AZURE_GPT_35_TURBO_0125 = "AZURE_GPT_35_TURBO_0125"
     AZURE_GPT_4_0613 = "AZURE_GPT_4_0613"
     AZURE_GPT_4_TURBO_1106 = "AZURE_GPT_4_TURBO_1106"
     AZURE_GPT_4_VISION_PREVIEW = "AZURE_GPT_4_VISION_PREVIEW"
@@ -33,6 +34,7 @@ def get_encoder_name(model_name: LanguageModelName) -> Optional[EncoderName]:
             LMN.AZURE_GPT_35_TURBO
             | LMN.AZURE_GPT_35_TURBO_16K
             | LMN.AZURE_GPT_35_TURBO_0613
+            | LMN.AZURE_GPT_35_TURBO_0125
         ):
             return EncoderName.CL100K_BASE
         case (
@@ -278,7 +280,7 @@ AzureGpt35Turbo0613 = create_language_model(
     token_limit=8192,
     info_cutoff_at=date(2021, 9, 1),
     published_at=date(2023, 6, 13),
-    retirement_at=date(2024, 10, 1),
+    retirement_at=date(2025, 2, 13),
 )
 
 AzureGpt35Turbo = create_language_model(
@@ -289,6 +291,7 @@ AzureGpt35Turbo = create_language_model(
     token_limit=4096,
     info_cutoff_at=date(2021, 9, 1),
     published_at=date(2023, 3, 1),
+    retirement_at=date(2025, 2, 13),
 )
 
 
@@ -300,7 +303,20 @@ AzureGpt35Turbo16k = create_language_model(
     token_limit=16382,
     info_cutoff_at=date(2021, 9, 1),
     published_at=date(2023, 6, 13),
-    retirement_at=date(2024, 10, 1),
+    retirement_at=date(2025, 2, 13),
+)
+
+
+AzureGpt35Turbo0125 = create_language_model(
+    name=LanguageModelName.AZURE_GPT_35_TURBO_0125,
+    provider=LanguageModelProvider.AZURE,
+    version="0125",
+    encoder_name=get_encoder_name(LanguageModelName.AZURE_GPT_35_TURBO_0125),
+    token_limit_input=16385,
+    token_limit_output=4096,
+    info_cutoff_at=date(2021, 9, 1),
+    published_at=date(2023, 1, 25),
+    retirement_at=date(5, 3, 31),
 )
 
 
@@ -313,7 +329,7 @@ AzureGpt40613 = create_language_model(
     info_cutoff_at=date(2021, 9, 1),
     published_at=date(2023, 6, 13),
     deprecated_at=date(2024, 10, 1),
-    retirement_at=date(2025, 6, 1),
+    retirement_at=date(2025, 6, 6),
 )
 
 
@@ -349,7 +365,7 @@ AzureGpt432k0613 = create_language_model(
     info_cutoff_at=date(2021, 9, 1),
     published_at=date(2023, 6, 13),
     deprecated_at=date(2024, 10, 1),
-    retirement_at=date(2025, 6, 1),
+    retirement_at=date(2025, 6, 6),
 )
 
 AzureGpt4Turbo20240409 = create_language_model(

unique_toolkit/language_model/schemas.py

@@ -89,7 +89,6 @@ class LanguageModelMessage(BaseModel):
     model_config = model_config
     role: LanguageModelMessageRole
     content: Optional[str | list[dict]] = None
-    tool_calls: Optional[list[LanguageModelFunctionCall]] = None
 
     def __str__(self):
         return format_message(self.role.capitalize(), message=self.content, num_tabs=1)
@@ -113,6 +112,7 @@ class LanguageModelUserMessage(LanguageModelMessage):
 
 class LanguageModelAssistantMessage(LanguageModelMessage):
     role: LanguageModelMessageRole = LanguageModelMessageRole.ASSISTANT
+    tool_calls: Optional[list[LanguageModelFunctionCall]] = None
 
     @field_validator("role", mode="before")
     def set_role(cls, value):
@@ -159,7 +159,7 @@ class LanguageModelCompletionChoice(BaseModel):
     model_config = model_config
 
     index: int
-    message: LanguageModelMessage
+    message: LanguageModelAssistantMessage
     finish_reason: str
 
 

{unique_toolkit-0.5.45.dist-info → unique_toolkit-0.5.47.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.5.45
+Version: 0.5.47
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -79,7 +79,7 @@ The `unique_toolkit.embedding` module encompasses all embedding related function
 The `unique_toolkit.language_model` module encompasses all language model related functionality and information on the different language models deployed through the 
 Unique platform.
 
-- `infos.py` comprises the information on all language models deployed through the Unique platform. We recommend to use the LanguageModel class, initialized with the LanguageModelName, e.g., LanguageModel(LanguageModelName.AZURE_GPT_35_TURBO_16K) to get the information on the specific language model like the name, version, token limits or retirement date.
+- `infos.py` comprises the information on all language models deployed through the Unique platform. We recommend to use the LanguageModel class, initialized with the LanguageModelName, e.g., LanguageModel(LanguageModelName.AZURE_GPT_35_TURBO_0125) to get the information on the specific language model like the name, version, token limits or retirement date.
 - `service.py` comprises the LanguageModelService and provides an interface to interact with the language models, e.g., complete or stream_complete. 
 - `schemas.py` comprises all relevant schemas, e.g., LanguageModelResponse, used in the LanguageModelService.
 - `utils.py` comprises utility functions to parse the output of the language model, e.g., convert_string_to_json finds and parses the last json object in a string.
@@ -100,6 +100,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.5.47] - 2025-01-07
+- Added a message builder to build language model messages conveniently without importing all different messages.
+- Move tool_calls to assistant message as not needed anywhere else.
+
+## [0.5.46] - 2025-01-07
+- Added `AZURE_GPT_35_TURBO_0125` as new model into toolkit.
+
 ## [0.5.45] - 2025-01-03
 - Added `ShortTermMemoryService` class to handle short term memory.
 

{unique_toolkit-0.5.45.dist-info → unique_toolkit-0.5.47.dist-info}/RECORD

@@ -36,14 +36,15 @@ unique_toolkit/evaluators/hallucination/utils.py,sha256=507BsX1mFTEne1-LdRCNMgBj
 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=hgk5yiFF4SpIcE2QSoki9YknFxmcKnq2LCJ1cK9de9I,1830
-unique_toolkit/language_model/infos.py,sha256=kQK6F3r8xTN7oT6b39J7rxW-Y4iPXjx_Fr9bCOVQdm0,12509
+unique_toolkit/language_model/builder.py,sha256=nsRqWO_2dgFehK5CgtqR5aqXgYUU0QL6mR0lALPrQXM,1898
+unique_toolkit/language_model/infos.py,sha256=NgoV05ausVWMqrYqgH6i3s7tYG7mejupROIF_bwEGZo,13050
 unique_toolkit/language_model/prompt.py,sha256=JSawaLjQg3VR-E2fK8engFyJnNdk21zaO8pPIodzN4Q,3991
-unique_toolkit/language_model/schemas.py,sha256=MBwEAFnnCYjSpbZDUYLrqkCc-0bsarsepSQEj1v6YEY,6991
+unique_toolkit/language_model/schemas.py,sha256=3en2aV9I3mD6IzugCRsq33vh9-JPNobszwBTw-LsaJQ,7000
 unique_toolkit/language_model/service.py,sha256=brNCPRA0XxgqHi2rI5i2lyFCkUiw4MNMe1VaR3UgWmY,15500
 unique_toolkit/language_model/utils.py,sha256=bPQ4l6_YO71w-zaIPanUUmtbXC1_hCvLK0tAFc3VCRc,1902
 unique_toolkit/short_term_memory/schemas.py,sha256=OhfcXyF6ACdwIXW45sKzjtZX_gkcJs8FEZXcgQTNenw,1406
 unique_toolkit/short_term_memory/service.py,sha256=Jd9P72-VvJy7hnqNrjmrmB5BHmsKuOpTiT0Jr-dBbsQ,1682
-unique_toolkit-0.5.45.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-0.5.45.dist-info/METADATA,sha256=hHpirT_B8QwGgqPLxXFkOC0UDBYRcHmmcDdhjLGXqKw,15248
-unique_toolkit-0.5.45.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.5.45.dist-info/RECORD,,
+unique_toolkit-0.5.47.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-0.5.47.dist-info/METADATA,sha256=idzuuHNWe1HqAkrEbASKasyAlzWwpsm6_3tI5er7b2c,15544
+unique_toolkit-0.5.47.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-0.5.47.dist-info/RECORD,,