unique_toolkit 0.7.4__tar.gz → 0.7.6__tar.gz

{unique_toolkit-0.7.4 → unique_toolkit-0.7.6}/CHANGELOG.md

@@ -5,6 +5,12 @@ 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
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.7.4
+Version: 0.7.6
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -111,6 +111,12 @@ 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
 

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

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

{unique_toolkit-0.7.4 → 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,9 +17,8 @@ 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"
 
@@ -80,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:
@@ -217,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=[
@@ -228,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=[
@@ -250,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(