unique_toolkit 0.6.8__tar.gz → 0.6.9__tar.gz

{unique_toolkit-0.6.8 → unique_toolkit-0.6.9}/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.6.9] - 2025-03-11
+- Add o1-preview as part of the language model info, make the name consistent across board.
+
 ## [0.6.8] - 2025-03-11
 - Add `verify_request_and_construct_event` to `verification.py`
 

{unique_toolkit-0.6.8 → unique_toolkit-0.6.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.6.8
+Version: 0.6.9
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -111,6 +111,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.6.9] - 2025-03-11
+- Add o1-preview as part of the language model info, make the name consistent across board.
+
 ## [0.6.8] - 2025-03-11
 - Add `verify_request_and_construct_event` to `verification.py`
 

{unique_toolkit-0.6.8 → unique_toolkit-0.6.9}/pyproject.toml

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

{unique_toolkit-0.6.8 → unique_toolkit-0.6.9}/unique_toolkit/language_model/infos.py

@@ -16,9 +16,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_GPT_o1_2024_1217 = "AZURE_GPT_o1_2024_1217"
-    AZURE_GPT_o1_MINI_2024_0912 = "AZURE_GPT_o1_MINI_2024_0912"
-    AZURE_GPT_o3_MINI_2025_0131 = "AZURE_GPT_o3_MINI_2025_0131"
+    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_o3_MINI_2025_0131 = "AZURE_o3_MINI_2025_0131"
 
 
 class EncoderName(StrEnum):
@@ -215,7 +216,7 @@ class LanguageModelInfo(BaseModel):
                     info_cutoff_at=date(2023, 10, 1),
                     published_at=date(2024, 7, 18),
                 )
-            case LanguageModelName.AZURE_GPT_o1_2024_1217:
+            case LanguageModelName.AZURE_o1_2024_1217:
                 return cls(
                     name=model_name,
                     capabilities=[
@@ -234,7 +235,10 @@ class LanguageModelInfo(BaseModel):
                     info_cutoff_at=date(2023, 10, 1),
                     published_at=date(2024, 12, 17),
                 )
-            case LanguageModelName.AZURE_GPT_o1_MINI_2024_0912:
+            case (
+                LanguageModelName.AZURE_o1_MINI_2024_0912
+                | LanguageModelName.AZURE_o1_PREVIEW_2024_0912
+            ):
                 return cls(
                     name=model_name,
                     capabilities=[
@@ -253,7 +257,7 @@ class LanguageModelInfo(BaseModel):
                     info_cutoff_at=date(2023, 10, 1),
                     published_at=date(2024, 9, 12),
                 )
-            case LanguageModelName.AZURE_GPT_o3_MINI_2025_0131:
+            case LanguageModelName.AZURE_o3_MINI_2025_0131:
                 return cls(
                     name=model_name,
                     capabilities=[