unique_toolkit 0.5.31__tar.gz → 0.5.32__tar.gz

{unique_toolkit-0.5.31 → unique_toolkit-0.5.32}/CHANGELOG.md

@@ -5,6 +5,9 @@ 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.32] - 2024-10-30
+- Extending `LanguageModelName` with GPT-4o-2024-0806. This model is invoked using `AZURE_GPT_4o_2024_0806`.
+
 ## [0.5.31] - 2024-10-29
 - Adding support for function calling. Assistant message for tool calls can be directly created with `LanguageModelFunctionCall.create_assistant_message_from_tool_calls`. Better separation of schemas for different types of `LanguageModelMessages`.
 

{unique_toolkit-0.5.31 → unique_toolkit-0.5.32}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.5.31
+Version: 0.5.32
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -100,6 +100,9 @@ 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.32] - 2024-10-30
+- Extending `LanguageModelName` with GPT-4o-2024-0806. This model is invoked using `AZURE_GPT_4o_2024_0806`.
+
 ## [0.5.31] - 2024-10-29
 - Adding support for function calling. Assistant message for tool calls can be directly created with `LanguageModelFunctionCall.create_assistant_message_from_tool_calls`. Better separation of schemas for different types of `LanguageModelMessages`.
 

{unique_toolkit-0.5.31 → unique_toolkit-0.5.32}/pyproject.toml

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

{unique_toolkit-0.5.31 → unique_toolkit-0.5.32}/unique_toolkit/language_model/infos.py

@@ -17,6 +17,7 @@ class LanguageModelName(StrEnum):
     AZURE_GPT_4_32K_0613 = "AZURE_GPT_4_32K_0613"
     AZURE_GPT_4_TURBO_2024_0409 = "AZURE_GPT_4_TURBO_2024_0409"
     AZURE_GPT_4o_2024_0513 = "AZURE_GPT_4o_2024_0513"
+    AZURE_GPT_4o_2024_0806 = "AZURE_GPT_4o_2024_0806"
     AZURE_GPT_4o_MINI_2024_0718 = "AZURE_GPT_4o_MINI_2024_0718"
 
 
@@ -42,7 +43,11 @@ def get_encoder_name(model_name: LanguageModelName) -> Optional[EncoderName]:
             | LMN.AZURE_GPT_4_TURBO_2024_0409
         ):
             return EncoderName.CL100K_BASE
-        case LMN.AZURE_GPT_4o_2024_0513 | LMN.AZURE_GPT_4o_MINI_2024_0718:
+        case (
+            LMN.AZURE_GPT_4o_2024_0513
+            | LMN.AZURE_GPT_4o_2024_0806
+            | LMN.AZURE_GPT_4o_MINI_2024_0718
+        ):
             return EncoderName.O200K_BASE
         case _:
             print(f"{model_name} is not supported. Please add encoder information.")
@@ -369,6 +374,17 @@ AzureGpt4o20240513 = create_language_model(
     published_at=date(2024, 5, 13),
 )
 
+AzureGpt4o20240806 = create_language_model(
+    name=LanguageModelName.AZURE_GPT_4o_2024_0806,
+    encoder_name=get_encoder_name(LanguageModelName.AZURE_GPT_4o_2024_0806),
+    provider=LanguageModelProvider.AZURE,
+    version="2024-08-06",
+    token_limit_input=128000,
+    token_limit_output=16384,
+    info_cutoff_at=date(2023, 10, 1),
+    published_at=date(2024, 8, 6),
+)
+
 AzureGpt4oMini20240718 = create_language_model(
     name=LanguageModelName.AZURE_GPT_4o_MINI_2024_0718,
     provider=LanguageModelProvider.AZURE,