unique_toolkit 0.7.3__tar.gz → 0.7.6__tar.gz

{unique_toolkit-0.7.3 → unique_toolkit-0.7.6}/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.7.6] - 2025-04-08
+- De provisioning o1-preview
+
+## [0.7.5] - 2025-04-07
+- Skip None values when serializing to json schema for LanguageModelInfo
+
+## [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

{unique_toolkit-0.7.3 → unique_toolkit-0.7.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.7.3
+Version: 0.7.6
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -111,6 +111,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.7.6] - 2025-04-08
+- De provisioning o1-preview
+
+## [0.7.5] - 2025-04-07
+- Skip None values when serializing to json schema for LanguageModelInfo
+
+## [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

{unique_toolkit-0.7.3 → unique_toolkit-0.7.6}/pyproject.toml

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

{unique_toolkit-0.7.3 → unique_toolkit-0.7.6}/unique_toolkit/language_model/infos.py

@@ -3,6 +3,7 @@ from enum import StrEnum
 from typing import ClassVar, Optional, Self
 
 from pydantic import BaseModel
+from pydantic.json_schema import SkipJsonSchema
 from typing_extensions import deprecated
 
 from unique_toolkit.language_model.schemas import LanguageModelTokenLimits
@@ -16,10 +17,10 @@ class LanguageModelName(StrEnum):
     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"
-    AZURE_o1_PREVIEW_2024_0912 = "AZURE_o1_PREVIEW_2024_0912"
-    AZURE_o1_2024_1217 = "AZURE_o1_2024_1217"
     AZURE_o1_MINI_2024_0912 = "AZURE_o1_MINI_2024_0912"
+    AZURE_o1_2024_1217 = "AZURE_o1_2024_1217"
     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):
@@ -79,12 +80,12 @@ class LanguageModelInfo(BaseModel):
     )
     capabilities: list[ModelCapabilities] = [ModelCapabilities.STREAMING]
 
-    info_cutoff_at: Optional[date] = None
-    published_at: Optional[date] = None
-    retirement_at: Optional[date] = None
+    info_cutoff_at: date | SkipJsonSchema[None] = None
+    published_at: date | SkipJsonSchema[None] = None
+    retirement_at: date | SkipJsonSchema[None] = None
 
-    deprecated_at: Optional[date] = None
-    retirement_text: Optional[str] = None
+    deprecated_at: date | SkipJsonSchema[None] = None
+    retirement_text: str | SkipJsonSchema[None] = None
 
     @classmethod
     def from_name(cls, model_name: LanguageModelName) -> Self:
@@ -216,7 +217,7 @@ class LanguageModelInfo(BaseModel):
                     info_cutoff_at=date(2023, 10, 1),
                     published_at=date(2024, 7, 18),
                 )
-            case LanguageModelName.AZURE_o1_2024_1217:
+            case LanguageModelName.AZURE_o1_MINI_2024_0912:
                 return cls(
                     name=model_name,
                     capabilities=[
@@ -227,18 +228,15 @@ class LanguageModelInfo(BaseModel):
                         ModelCapabilities.REASONING,
                     ],
                     provider=LanguageModelProvider.AZURE,
-                    version="2024-12-17",
+                    version="2024-09-12",
                     encoder_name=EncoderName.O200K_BASE,
                     token_limits=LanguageModelTokenLimits(
-                        token_limit_input=200_000, token_limit_output=100_000
+                        token_limit_input=128_000, token_limit_output=65_536
                     ),
                     info_cutoff_at=date(2023, 10, 1),
-                    published_at=date(2024, 12, 17),
+                    published_at=date(2024, 9, 12),
                 )
-            case (
-                LanguageModelName.AZURE_o1_MINI_2024_0912
-                | LanguageModelName.AZURE_o1_PREVIEW_2024_0912
-            ):
+            case LanguageModelName.AZURE_o1_2024_1217:
                 return cls(
                     name=model_name,
                     capabilities=[
@@ -249,13 +247,13 @@ class LanguageModelInfo(BaseModel):
                         ModelCapabilities.REASONING,
                     ],
                     provider=LanguageModelProvider.AZURE,
-                    version="2024-09-12",
+                    version="2024-12-17",
                     encoder_name=EncoderName.O200K_BASE,
                     token_limits=LanguageModelTokenLimits(
-                        token_limit_input=128_000, token_limit_output=65_536
+                        token_limit_input=200_000, token_limit_output=100_000
                     ),
                     info_cutoff_at=date(2023, 10, 1),
-                    published_at=date(2024, 9, 12),
+                    published_at=date(2024, 12, 17),
                 )
             case LanguageModelName.AZURE_o3_MINI_2025_0131:
                 return cls(
@@ -275,6 +273,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(