unique_toolkit 2026.40.0.dev9__py3-none-any.whl → 2026.40.0.dev11__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (4) hide show
  1. unique_toolkit/language_model/infos.py +122 -0
  2. {unique_toolkit-2026.40.0.dev9.dist-info → unique_toolkit-2026.40.0.dev11.dist-info}/METADATA +2 -2
  3. {unique_toolkit-2026.40.0.dev9.dist-info → unique_toolkit-2026.40.0.dev11.dist-info}/RECORD +4 -4
  4. {unique_toolkit-2026.40.0.dev9.dist-info → unique_toolkit-2026.40.0.dev11.dist-info}/WHEEL +0 -0
unique_toolkit/language_model/infos.py CHANGED
@@ -66,6 +66,8 @@ class LanguageModelName(StrEnum):
66
66
  AZURE_GPT_56_TERRA_2026_0709 = "AZURE_GPT_56_TERRA_2026_0709"
67
67
  AZURE_GPT_56_LUNA_2026_0709 = "AZURE_GPT_56_LUNA_2026_0709"
68
68
  AZURE_GPT_6_ASTRA_2026_0903 = "AZURE_GPT_6_ASTRA_2026_0903"
69
+ AZURE_GPT_6_LUNA_2026_0922 = "AZURE_GPT_6_LUNA_2026_0922"
70
+ AZURE_GPT_6_SOL_2026_0922 = "AZURE_GPT_6_SOL_2026_0922"
69
71
  AZURE_GPT_4o_2024_0513 = "AZURE_GPT_4o_2024_0513"
70
72
  AZURE_GPT_4o_2024_0806 = "AZURE_GPT_4o_2024_0806"
71
73
  AZURE_GPT_4o_2024_1120 = "AZURE_GPT_4o_2024_1120"
@@ -108,6 +110,7 @@ class LanguageModelName(StrEnum):
108
110
  GEMINI_2_5_PRO_PREVIEW_0605 = "litellm:gemini-2-5-pro-preview-06-05"
109
111
  GEMINI_3_1_PRO_PREVIEW = "litellm:gemini-3-1-pro-preview"
110
112
  GEMINI_3_6_FLASH = "litellm:gemini-3-6-flash"
113
+ GEMINI_3_8_FLASH = "litellm:gemini-3-8-flash"
111
114
  GEMINI_3_FLASH_PREVIEW = "litellm:gemini-3-flash-preview"
112
115
  GEMINI_3_PRO_PREVIEW = "litellm:gemini-3-pro-preview"
113
116
  GROK_4_1_FAST_NON_REASONING = "litellm:grok-4-1-fast-non-reasoning"
@@ -132,6 +135,8 @@ class LanguageModelName(StrEnum):
132
135
  LITELLM_OPENAI_GPT_56_TERRA = "litellm:openai-gpt-5-6-terra"
133
136
  LITELLM_OPENAI_GPT_56_LUNA = "litellm:openai-gpt-5-6-luna"
134
137
  LITELLM_OPENAI_GPT_6_ASTRA = "litellm:openai-gpt-6-astra"
138
+ LITELLM_OPENAI_GPT_6_LUNA = "litellm:openai-gpt-6-luna"
139
+ LITELLM_OPENAI_GPT_6_SOL = "litellm:openai-gpt-6-sol"
135
140
  LITELLM_OPENAI_O1 = "litellm:openai-o1"
136
141
  LITELLM_OPENAI_O3 = "litellm:openai-o3"
137
142
  LITELLM_OPENAI_O3_DEEP_RESEARCH = "litellm:openai-o3-deep-research"
@@ -228,6 +233,8 @@ def get_encoder_name(model_name: LanguageModelName) -> EncoderName:
228
233
  | LMN.AZURE_GPT_56_TERRA_2026_0709
229
234
  | LMN.AZURE_GPT_56_LUNA_2026_0709
230
235
  | LMN.AZURE_GPT_6_ASTRA_2026_0903
236
+ | LMN.AZURE_GPT_6_LUNA_2026_0922
237
+ | LMN.AZURE_GPT_6_SOL_2026_0922
231
238
  | LMN.AZURE_MODEL_ROUTER_2025_1118
232
239
  | LMN.LITELLM_OPENAI_GPT_5
233
240
  | LMN.LITELLM_OPENAI_GPT_5_MINI
@@ -246,6 +253,8 @@ def get_encoder_name(model_name: LanguageModelName) -> EncoderName:
246
253
  | LMN.LITELLM_OPENAI_GPT_56_TERRA
247
254
  | LMN.LITELLM_OPENAI_GPT_56_LUNA
248
255
  | LMN.LITELLM_OPENAI_GPT_6_ASTRA
256
+ | LMN.LITELLM_OPENAI_GPT_6_LUNA
257
+ | LMN.LITELLM_OPENAI_GPT_6_SOL
249
258
  | LMN.LITELLM_OPENAI_O1
250
259
  | LMN.LITELLM_OPENAI_O3
251
260
  | LMN.LITELLM_OPENAI_O3_DEEP_RESEARCH
@@ -1286,6 +1295,49 @@ class LanguageModelInfo(BaseModel):
1286
1295
  "xhigh",
1287
1296
  ],
1288
1297
  )
1298
+ case (
1299
+ LanguageModelName.AZURE_GPT_6_LUNA_2026_0922
1300
+ | LanguageModelName.AZURE_GPT_6_SOL_2026_0922
1301
+ ):
1302
+ info_cutoff_at = {
1303
+ LanguageModelName.AZURE_GPT_6_LUNA_2026_0922: date(2026, 5, 18),
1304
+ LanguageModelName.AZURE_GPT_6_SOL_2026_0922: date(2026, 4, 20),
1305
+ }[model_name]
1306
+ return cls(
1307
+ name=model_name,
1308
+ provider=LanguageModelProvider.AZURE,
1309
+ family=ModelFamily.OPENAI,
1310
+ version="2026-09-22",
1311
+ encoder_name=EncoderName.O200K_BASE,
1312
+ capabilities=[
1313
+ ModelCapabilities.CHAT_COMPLETIONS_API,
1314
+ ModelCapabilities.FUNCTION_CALLING,
1315
+ ModelCapabilities.PARALLEL_FUNCTION_CALLING,
1316
+ ModelCapabilities.REASONING,
1317
+ ModelCapabilities.RESPONSES_API,
1318
+ ModelCapabilities.STREAMING,
1319
+ ModelCapabilities.STRUCTURED_OUTPUT,
1320
+ ModelCapabilities.VISION,
1321
+ ],
1322
+ token_limits=LanguageModelTokenLimits(
1323
+ token_limit_input=922_000, token_limit_output=128_000
1324
+ ),
1325
+ info_cutoff_at=info_cutoff_at,
1326
+ published_at=date(2026, 9, 22),
1327
+ temperature_bounds=TemperatureBounds(
1328
+ min_temperature=0.0, max_temperature=1.0
1329
+ ),
1330
+ default_options={
1331
+ "reasoning_effort": "medium",
1332
+ },
1333
+ supported_reasoning_efforts=[
1334
+ "none",
1335
+ "low",
1336
+ "medium",
1337
+ "high",
1338
+ "xhigh",
1339
+ ],
1340
+ )
1289
1341
  case LanguageModelName.AZURE_GPT_4_TURBO_2024_0409:
1290
1342
  return cls(
1291
1343
  name=model_name,
@@ -2214,6 +2266,33 @@ class LanguageModelInfo(BaseModel):
2214
2266
  published_at=date(2026, 7, 21),
2215
2267
  supported_reasoning_efforts=["minimal", "low", "medium", "high"],
2216
2268
  )
2269
+ case LanguageModelName.GEMINI_3_8_FLASH:
2270
+ return cls(
2271
+ name=model_name,
2272
+ capabilities=[
2273
+ ModelCapabilities.FUNCTION_CALLING,
2274
+ ModelCapabilities.STREAMING,
2275
+ ModelCapabilities.VISION,
2276
+ ModelCapabilities.STRUCTURED_OUTPUT,
2277
+ ModelCapabilities.REASONING,
2278
+ ],
2279
+ provider=LanguageModelProvider.LITELLM,
2280
+ family=ModelFamily.GOOGLE,
2281
+ version="gemini-3-8-flash",
2282
+ encoder_name=EncoderName.O200K_BASE, # TODO: Update encoder with litellm
2283
+ token_limits=LanguageModelTokenLimits(
2284
+ # https://ai.google.dev/gemini-api/docs/models/gemini-3.8-flash
2285
+ token_limit_input=1_048_576,
2286
+ token_limit_output=65_536,
2287
+ ),
2288
+ # https://deepmind.google/models/model-cards/gemini-3-8-flash/
2289
+ info_cutoff_at=date(2026, 3, 1),
2290
+ # https://ai.google.dev/gemini-api/docs/changelog
2291
+ published_at=date(2026, 9, 2),
2292
+ # minimal is rejected; default thinking level is medium when unset.
2293
+ # https://ai.google.dev/gemini-api/docs/models/gemini-3.8-flash
2294
+ supported_reasoning_efforts=["low", "medium", "high"],
2295
+ )
2217
2296
  case LanguageModelName.GEMINI_3_FLASH_PREVIEW:
2218
2297
  return cls(
2219
2298
  name=model_name,
@@ -2857,6 +2936,49 @@ class LanguageModelInfo(BaseModel):
2857
2936
  "xhigh",
2858
2937
  ],
2859
2938
  )
2939
+ case (
2940
+ LanguageModelName.LITELLM_OPENAI_GPT_6_LUNA
2941
+ | LanguageModelName.LITELLM_OPENAI_GPT_6_SOL
2942
+ ):
2943
+ info_cutoff_at = {
2944
+ LanguageModelName.LITELLM_OPENAI_GPT_6_LUNA: date(2026, 5, 18),
2945
+ LanguageModelName.LITELLM_OPENAI_GPT_6_SOL: date(2026, 4, 20),
2946
+ }[model_name]
2947
+ return cls(
2948
+ name=model_name,
2949
+ provider=LanguageModelProvider.LITELLM,
2950
+ family=ModelFamily.OPENAI,
2951
+ version="2026-09-22",
2952
+ encoder_name=EncoderName.O200K_BASE,
2953
+ capabilities=[
2954
+ ModelCapabilities.CHAT_COMPLETIONS_API,
2955
+ ModelCapabilities.FUNCTION_CALLING,
2956
+ ModelCapabilities.PARALLEL_FUNCTION_CALLING,
2957
+ ModelCapabilities.REASONING,
2958
+ ModelCapabilities.RESPONSES_API,
2959
+ ModelCapabilities.STREAMING,
2960
+ ModelCapabilities.STRUCTURED_OUTPUT,
2961
+ ModelCapabilities.VISION,
2962
+ ],
2963
+ token_limits=LanguageModelTokenLimits(
2964
+ token_limit_input=922_000, token_limit_output=128_000
2965
+ ),
2966
+ info_cutoff_at=info_cutoff_at,
2967
+ published_at=date(2026, 9, 22),
2968
+ temperature_bounds=TemperatureBounds(
2969
+ min_temperature=0.0, max_temperature=1.0
2970
+ ),
2971
+ default_options={
2972
+ "reasoning_effort": "medium",
2973
+ },
2974
+ supported_reasoning_efforts=[
2975
+ "none",
2976
+ "low",
2977
+ "medium",
2978
+ "high",
2979
+ "xhigh",
2980
+ ],
2981
+ )
2860
2982
  case LanguageModelName.LITELLM_OPENAI_O1:
2861
2983
  return cls(
2862
2984
  name=model_name,
{unique_toolkit-2026.40.0.dev9.dist-info → unique_toolkit-2026.40.0.dev11.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: unique-toolkit
3
- Version: 2026.40.0.dev9
3
+ Version: 2026.40.0.dev11
4
4
  Summary:
5
5
  Author: Cedric Klinkert, Martin Fadler, Sadique Sheik, Fabian Schläpfer, Pascal Hauri, Ejiroghene Oki-Peter, Rami Azouz, Ricardo Visini, Sanzio Monti
6
6
  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>
@@ -16,7 +16,7 @@ Requires-Dist: sseclient>=0.0.27,<0.0.28
16
16
  Requires-Dist: openai>=1.105.0,<3
17
17
  Requires-Dist: platformdirs>=4.0.0,<5
18
18
  Requires-Dist: pillow>=12.1.1
19
- Requires-Dist: unique-sdk>=2026.40.0.dev6,<2026.40.0rc0
19
+ Requires-Dist: unique-sdk>=2026.40.0.dev7,<2026.40.0rc0
20
20
  Requires-Dist: jambo>=0.1.2,<0.2
21
21
  Requires-Dist: jinja2>=3.1.6,<4
22
22
  Requires-Dist: docxtpl>=0.20.1,<0.21
{unique_toolkit-2026.40.0.dev9.dist-info → unique_toolkit-2026.40.0.dev11.dist-info}/RECORD RENAMED
@@ -394,7 +394,7 @@ unique_toolkit/language_model/builder.py,sha256=KwBzsD6-fvQnB4p2I7M-5Tk6j4tmQ973
394
394
  unique_toolkit/language_model/constants.py,sha256=B-topqW0r83dkC_25DeQfnPk3n53qzIHUCBS7YJ0-1U,119
395
395
  unique_toolkit/language_model/default_language_model.py,sha256=mcyuFs_V6tp52C8xJSHiEHO4kFGj6xed1gvozvCRD7k,1181
396
396
  unique_toolkit/language_model/functions.py,sha256=W6JC4LUh-LZ9waDgph--xVmKeQ7Ls6NRAqDx9DwQdeY,26691
397
- unique_toolkit/language_model/infos.py,sha256=J6E8ciy6f8jm4CjT_3Zct8IWoqYRL4GdDK96n1rqvAM,161736
397
+ unique_toolkit/language_model/infos.py,sha256=YHTGaptzuVf65rI0S4T8QWTXluQJKKrQiNLG309eHo4,167622
398
398
  unique_toolkit/language_model/invocation_stats.py,sha256=CQbpUz394IsVd8eNTB1gPSMt5f0UXmkN1amNsqtAW6U,5857
399
399
  unique_toolkit/language_model/model_costs.py,sha256=ShuDAzMew1fUkJ3tzPgU5Zqjp6ivUZaG97BFh93tEn4,6358
400
400
  unique_toolkit/language_model/prompt.py,sha256=JSawaLjQg3VR-E2fK8engFyJnNdk21zaO8pPIodzN4Q,3991
@@ -423,6 +423,6 @@ unique_toolkit/short_term_memory/service.py,sha256=2JMBEBn6GoyK-a0xvdcWpPft-3C74
423
423
  unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
424
424
  unique_toolkit/smart_rules/compile.py,sha256=vRe0aMJWuVIK-VsqomgxQImzke7MDmwgSxxJfYCa7zw,1386
425
425
  unique_toolkit/test_utilities/events.py,sha256=FLrhMh0A_mQutAUjoD0FdMcVQ4Gq31r35OVGzRjvLro,6760
426
- unique_toolkit-2026.40.0.dev9.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
427
- unique_toolkit-2026.40.0.dev9.dist-info/METADATA,sha256=ciTtZWBWSiZiomhVa7XHox6eylrK0RVkCpzoFdUF5e8,7821
428
- unique_toolkit-2026.40.0.dev9.dist-info/RECORD,,
426
+ unique_toolkit-2026.40.0.dev11.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
427
+ unique_toolkit-2026.40.0.dev11.dist-info/METADATA,sha256=3IjKcTmSpHYdPsDY_lhCjGNhgATTFqhy6elkOUG5ng0,7822
428
+ unique_toolkit-2026.40.0.dev11.dist-info/RECORD,,