unique_toolkit 0.7.3__py3-none-any.whl → 0.7.6__py3-none-any.whl

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(

{unique_toolkit-0.7.3.dist-info → unique_toolkit-0.7.6.dist-info}/METADATA

@@ -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.dist-info → unique_toolkit-0.7.6.dist-info}/RECORD

@@ -48,7 +48,7 @@ unique_toolkit/language_model/__init__.py,sha256=jWko_vQj48wjnpTtlkg8iNdef0SMI3F
 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=0Lx9M0gumczOpzbS5Zaqz_T-e_eV4MOb6gJVz5X1KF8,17339
 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.3.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-0.7.3.dist-info/METADATA,sha256=ynA_2RASnYvrs5DSqAcq7-taxnu1iZQe2gSRBiJupd4,20944
-unique_toolkit-0.7.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.7.3.dist-info/RECORD,,
+unique_toolkit-0.7.6.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-0.7.6.dist-info/METADATA,sha256=M2N81nXjCCmfZfZiWyj-jLwonMTt5TjMZxz_F7FLXCw,21181
+unique_toolkit-0.7.6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-0.7.6.dist-info/RECORD,,