unique-toolkit 2026.40.0.dev8__tar.gz → 2026.40.0.dev10__tar.gz

{unique_toolkit-2026.40.0.dev8 → unique_toolkit-2026.40.0.dev10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.40.0.dev8
+Version: 2026.40.0.dev10
 Summary: 
 Author: Cedric Klinkert, Martin Fadler, Sadique Sheik, Fabian Schläpfer, Pascal Hauri, Ejiroghene Oki-Peter, Rami Azouz, Ricardo Visini, Sanzio Monti
 Author-email: Cedric Klinkert <cedric.klinkert@unique.ch>, Martin Fadler <martin.fadler@unique.ch>, Sadique Sheik <sadique@unique.ch>, Fabian Schläpfer <fabian@unique.ch>, Pascal Hauri <pascal@unique.ch>, Ejiroghene Oki-Peter <oki-peter.ejiroghene@unique.ai>, Rami Azouz <rami.ext@unique.ch>, Ricardo Visini <ricardo.visini@unique.ch>, Sanzio Monti <sanzio.monti@unique.ai>

{unique_toolkit-2026.40.0.dev8 → unique_toolkit-2026.40.0.dev10}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "2026.40.0.dev8"
+version = "2026.40.0.dev10"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"

{unique_toolkit-2026.40.0.dev8 → unique_toolkit-2026.40.0.dev10}/unique_toolkit/language_model/infos.py

@@ -66,6 +66,8 @@ class LanguageModelName(StrEnum):
     AZURE_GPT_56_TERRA_2026_0709 = "AZURE_GPT_56_TERRA_2026_0709"
     AZURE_GPT_56_LUNA_2026_0709 = "AZURE_GPT_56_LUNA_2026_0709"
     AZURE_GPT_6_ASTRA_2026_0903 = "AZURE_GPT_6_ASTRA_2026_0903"
+    AZURE_GPT_6_LUNA_2026_0922 = "AZURE_GPT_6_LUNA_2026_0922"
+    AZURE_GPT_6_SOL_2026_0922 = "AZURE_GPT_6_SOL_2026_0922"
     AZURE_GPT_4o_2024_0513 = "AZURE_GPT_4o_2024_0513"
     AZURE_GPT_4o_2024_0806 = "AZURE_GPT_4o_2024_0806"
     AZURE_GPT_4o_2024_1120 = "AZURE_GPT_4o_2024_1120"
@@ -96,6 +98,7 @@ class LanguageModelName(StrEnum):
     ANTHROPIC_CLAUDE_OPUS_4_7 = "litellm:anthropic-claude-opus-4-7"
     ANTHROPIC_CLAUDE_OPUS_4_8 = "litellm:anthropic-claude-opus-4-8"
     ANTHROPIC_CLAUDE_OPUS_5 = "litellm:anthropic-claude-opus-5"
+    ANTHROPIC_CLAUDE_OPUS_5_5 = "litellm:anthropic-claude-opus-5-5"
     ANTHROPIC_CLAUDE_FABLE_5 = "litellm:anthropic-claude-fable-5"
     GEMINI_2_0_FLASH = "litellm:gemini-2-0-flash"
     GEMINI_2_5_FLASH = "litellm:gemini-2-5-flash"
@@ -131,6 +134,8 @@ class LanguageModelName(StrEnum):
     LITELLM_OPENAI_GPT_56_TERRA = "litellm:openai-gpt-5-6-terra"
     LITELLM_OPENAI_GPT_56_LUNA = "litellm:openai-gpt-5-6-luna"
     LITELLM_OPENAI_GPT_6_ASTRA = "litellm:openai-gpt-6-astra"
+    LITELLM_OPENAI_GPT_6_LUNA = "litellm:openai-gpt-6-luna"
+    LITELLM_OPENAI_GPT_6_SOL = "litellm:openai-gpt-6-sol"
     LITELLM_OPENAI_O1 = "litellm:openai-o1"
     LITELLM_OPENAI_O3 = "litellm:openai-o3"
     LITELLM_OPENAI_O3_DEEP_RESEARCH = "litellm:openai-o3-deep-research"
@@ -158,6 +163,7 @@ class LanguageModelName(StrEnum):
     VERTEX_CLAUDE_OPUS_4_7 = "litellm:vertex-claude-opus-4-7"
     VERTEX_CLAUDE_OPUS_4_8 = "litellm:vertex-claude-opus-4-8"
     VERTEX_CLAUDE_OPUS_5 = "litellm:vertex-claude-opus-5"
+    VERTEX_CLAUDE_OPUS_5_5 = "litellm:vertex-claude-opus-5-5"
     VERTEX_CLAUDE_FABLE_5 = "litellm:vertex-claude-fable-5"
 
 
@@ -226,6 +232,8 @@ def get_encoder_name(model_name: LanguageModelName) -> EncoderName:
             | LMN.AZURE_GPT_56_TERRA_2026_0709
             | LMN.AZURE_GPT_56_LUNA_2026_0709
             | LMN.AZURE_GPT_6_ASTRA_2026_0903
+            | LMN.AZURE_GPT_6_LUNA_2026_0922
+            | LMN.AZURE_GPT_6_SOL_2026_0922
             | LMN.AZURE_MODEL_ROUTER_2025_1118
             | LMN.LITELLM_OPENAI_GPT_5
             | LMN.LITELLM_OPENAI_GPT_5_MINI
@@ -244,6 +252,8 @@ def get_encoder_name(model_name: LanguageModelName) -> EncoderName:
             | LMN.LITELLM_OPENAI_GPT_56_TERRA
             | LMN.LITELLM_OPENAI_GPT_56_LUNA
             | LMN.LITELLM_OPENAI_GPT_6_ASTRA
+            | LMN.LITELLM_OPENAI_GPT_6_LUNA
+            | LMN.LITELLM_OPENAI_GPT_6_SOL
             | LMN.LITELLM_OPENAI_O1
             | LMN.LITELLM_OPENAI_O3
             | LMN.LITELLM_OPENAI_O3_DEEP_RESEARCH
@@ -1284,6 +1294,49 @@ class LanguageModelInfo(BaseModel):
                         "xhigh",
                     ],
                 )
+            case (
+                LanguageModelName.AZURE_GPT_6_LUNA_2026_0922
+                | LanguageModelName.AZURE_GPT_6_SOL_2026_0922
+            ):
+                info_cutoff_at = {
+                    LanguageModelName.AZURE_GPT_6_LUNA_2026_0922: date(2026, 5, 18),
+                    LanguageModelName.AZURE_GPT_6_SOL_2026_0922: date(2026, 4, 20),
+                }[model_name]
+                return cls(
+                    name=model_name,
+                    provider=LanguageModelProvider.AZURE,
+                    family=ModelFamily.OPENAI,
+                    version="2026-09-22",
+                    encoder_name=EncoderName.O200K_BASE,
+                    capabilities=[
+                        ModelCapabilities.CHAT_COMPLETIONS_API,
+                        ModelCapabilities.FUNCTION_CALLING,
+                        ModelCapabilities.PARALLEL_FUNCTION_CALLING,
+                        ModelCapabilities.REASONING,
+                        ModelCapabilities.RESPONSES_API,
+                        ModelCapabilities.STREAMING,
+                        ModelCapabilities.STRUCTURED_OUTPUT,
+                        ModelCapabilities.VISION,
+                    ],
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=922_000, token_limit_output=128_000
+                    ),
+                    info_cutoff_at=info_cutoff_at,
+                    published_at=date(2026, 9, 22),
+                    temperature_bounds=TemperatureBounds(
+                        min_temperature=0.0, max_temperature=1.0
+                    ),
+                    default_options={
+                        "reasoning_effort": "medium",
+                    },
+                    supported_reasoning_efforts=[
+                        "none",
+                        "low",
+                        "medium",
+                        "high",
+                        "xhigh",
+                    ],
+                )
             case LanguageModelName.AZURE_GPT_4_TURBO_2024_0409:
                 return cls(
                     name=model_name,
@@ -1950,6 +2003,32 @@ class LanguageModelInfo(BaseModel):
                     published_at=date(2026, 7, 24),
                     supported_reasoning_efforts=[],
                 )
+            case (
+                LanguageModelName.ANTHROPIC_CLAUDE_OPUS_5_5
+                | LanguageModelName.VERTEX_CLAUDE_OPUS_5_5
+            ):
+                return cls(
+                    name=model_name,
+                    capabilities=[
+                        ModelCapabilities.FUNCTION_CALLING,
+                        ModelCapabilities.STREAMING,
+                        ModelCapabilities.VISION,
+                        ModelCapabilities.REASONING,
+                    ],
+                    provider=LanguageModelProvider.LITELLM,
+                    family=ModelFamily.ANTHROPIC,
+                    version="claude-opus-5-5",
+                    encoder_name=EncoderName.O200K_BASE,  # TODO: Update encoder with litellm
+                    token_limits=LanguageModelTokenLimits(
+                        # TODO: Remove the 1.3 adjustment once a proper Claude tokenizer is implemented. UN-24123
+                        # 1M context / 128K max output. https://platform.claude.com/docs/en/about-claude/models/overview
+                        token_limit_input=int(1_000_000 / 1.3),
+                        token_limit_output=int(128_000 / 1.3),
+                    ),
+                    info_cutoff_at=date(2026, 6, 1),
+                    published_at=date(2026, 9, 22),
+                    supported_reasoning_efforts=[],
+                )
             case (
                 LanguageModelName.ANTHROPIC_CLAUDE_FABLE_5
                 | LanguageModelName.VERTEX_CLAUDE_FABLE_5
@@ -2829,6 +2908,49 @@ class LanguageModelInfo(BaseModel):
                         "xhigh",
                     ],
                 )
+            case (
+                LanguageModelName.LITELLM_OPENAI_GPT_6_LUNA
+                | LanguageModelName.LITELLM_OPENAI_GPT_6_SOL
+            ):
+                info_cutoff_at = {
+                    LanguageModelName.LITELLM_OPENAI_GPT_6_LUNA: date(2026, 5, 18),
+                    LanguageModelName.LITELLM_OPENAI_GPT_6_SOL: date(2026, 4, 20),
+                }[model_name]
+                return cls(
+                    name=model_name,
+                    provider=LanguageModelProvider.LITELLM,
+                    family=ModelFamily.OPENAI,
+                    version="2026-09-22",
+                    encoder_name=EncoderName.O200K_BASE,
+                    capabilities=[
+                        ModelCapabilities.CHAT_COMPLETIONS_API,
+                        ModelCapabilities.FUNCTION_CALLING,
+                        ModelCapabilities.PARALLEL_FUNCTION_CALLING,
+                        ModelCapabilities.REASONING,
+                        ModelCapabilities.RESPONSES_API,
+                        ModelCapabilities.STREAMING,
+                        ModelCapabilities.STRUCTURED_OUTPUT,
+                        ModelCapabilities.VISION,
+                    ],
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=922_000, token_limit_output=128_000
+                    ),
+                    info_cutoff_at=info_cutoff_at,
+                    published_at=date(2026, 9, 22),
+                    temperature_bounds=TemperatureBounds(
+                        min_temperature=0.0, max_temperature=1.0
+                    ),
+                    default_options={
+                        "reasoning_effort": "medium",
+                    },
+                    supported_reasoning_efforts=[
+                        "none",
+                        "low",
+                        "medium",
+                        "high",
+                        "xhigh",
+                    ],
+                )
             case LanguageModelName.LITELLM_OPENAI_O1:
                 return cls(
                     name=model_name,