unique_toolkit 1.39.0__py3-none-any.whl → 1.39.2__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 (5) hide show
  1. unique_toolkit/language_model/infos.py +138 -0
  2. {unique_toolkit-1.39.0.dist-info → unique_toolkit-1.39.2.dist-info}/METADATA +7 -1
  3. {unique_toolkit-1.39.0.dist-info → unique_toolkit-1.39.2.dist-info}/RECORD +5 -5
  4. {unique_toolkit-1.39.0.dist-info → unique_toolkit-1.39.2.dist-info}/LICENSE +0 -0
  5. {unique_toolkit-1.39.0.dist-info → unique_toolkit-1.39.2.dist-info}/WHEEL +0 -0
unique_toolkit/language_model/infos.py CHANGED
@@ -31,6 +31,8 @@ class LanguageModelName(StrEnum):
31
31
  AZURE_GPT_51_CHAT_2025_1113 = "AZURE_GPT_51_CHAT_2025_1113"
32
32
  AZURE_GPT_51_CODEX_2025_1113 = "AZURE_GPT_51_CODEX_2025_1113"
33
33
  AZURE_GPT_51_CODEX_MINI_2025_1113 = "AZURE_GPT_51_CODEX_MINI_2025_1113"
34
+ AZURE_GPT_52_2025_1211 = "AZURE_GPT_52_2025_1211"
35
+ AZURE_GPT_52_CHAT_2025_1211 = "AZURE_GPT_52_CHAT_2025_1211"
34
36
  AZURE_GPT_4o_2024_0513 = "AZURE_GPT_4o_2024_0513"
35
37
  AZURE_GPT_4o_2024_0806 = "AZURE_GPT_4o_2024_0806"
36
38
  AZURE_GPT_4o_2024_1120 = "AZURE_GPT_4o_2024_1120"
@@ -62,6 +64,7 @@ class LanguageModelName(StrEnum):
62
64
  GEMINI_2_5_PRO = "litellm:gemini-2-5-pro"
63
65
  GEMINI_2_5_PRO_EXP_0325 = "litellm:gemini-2-5-pro-exp-03-25"
64
66
  GEMINI_2_5_PRO_PREVIEW_0605 = "litellm:gemini-2-5-pro-preview-06-05"
67
+ GEMINI_3_FLASH_PREVIEW = "litellm:gemini-3-flash-preview"
65
68
  GEMINI_3_PRO_PREVIEW = "litellm:gemini-3-pro-preview"
66
69
  LITELLM_OPENAI_GPT_5 = "litellm:openai-gpt-5"
67
70
  LITELLM_OPENAI_GPT_5_MINI = "litellm:openai-gpt-5-mini"
@@ -70,6 +73,8 @@ class LanguageModelName(StrEnum):
70
73
  LITELLM_OPENAI_GPT_5_PRO = "litellm:openai-gpt-5-pro"
71
74
  LITELLM_OPENAI_GPT_51 = "litellm:openai-gpt-5-1"
72
75
  LITELLM_OPENAI_GPT_51_THINKING = "litellm:openai-gpt-5-1-thinking"
76
+ LITELLM_OPENAI_GPT_52 = "litellm:openai-gpt-5-2"
77
+ LITELLM_OPENAI_GPT_52_THINKING = "litellm:openai-gpt-5-2-thinking"
73
78
  LITELLM_OPENAI_O1 = "litellm:openai-o1"
74
79
  LITELLM_OPENAI_O3 = "litellm:openai-o3"
75
80
  LITELLM_OPENAI_O3_DEEP_RESEARCH = "litellm:openai-o3-deep-research"
@@ -115,6 +120,8 @@ def get_encoder_name(model_name: LanguageModelName) -> EncoderName:
115
120
  | LMN.AZURE_GPT_51_CHAT_2025_1113
116
121
  | LMN.AZURE_GPT_51_CODEX_2025_1113
117
122
  | LMN.AZURE_GPT_51_CODEX_MINI_2025_1113
123
+ | LMN.AZURE_GPT_52_2025_1211
124
+ | LMN.AZURE_GPT_52_CHAT_2025_1211
118
125
  | LMN.LITELLM_OPENAI_GPT_5
119
126
  | LMN.LITELLM_OPENAI_GPT_5_MINI
120
127
  | LMN.LITELLM_OPENAI_GPT_5_NANO
@@ -122,6 +129,8 @@ def get_encoder_name(model_name: LanguageModelName) -> EncoderName:
122
129
  | LMN.LITELLM_OPENAI_GPT_5_PRO
123
130
  | LMN.LITELLM_OPENAI_GPT_51
124
131
  | LMN.LITELLM_OPENAI_GPT_51_THINKING
132
+ | LMN.LITELLM_OPENAI_GPT_52
133
+ | LMN.LITELLM_OPENAI_GPT_52_THINKING
125
134
  | LMN.LITELLM_OPENAI_O1
126
135
  | LMN.LITELLM_OPENAI_O3
127
136
  | LMN.LITELLM_OPENAI_O3_DEEP_RESEARCH
@@ -611,6 +620,60 @@ class LanguageModelInfo(BaseModel):
611
620
  "reasoning_effort": "medium",
612
621
  },
613
622
  )
623
+ case LanguageModelName.AZURE_GPT_52_2025_1211:
624
+ return cls(
625
+ name=model_name,
626
+ provider=LanguageModelProvider.AZURE,
627
+ version="2025-12-11",
628
+ encoder_name=EncoderName.O200K_BASE,
629
+ capabilities=[
630
+ ModelCapabilities.CHAT_COMPLETIONS_API,
631
+ ModelCapabilities.FUNCTION_CALLING,
632
+ ModelCapabilities.PARALLEL_FUNCTION_CALLING,
633
+ ModelCapabilities.REASONING,
634
+ ModelCapabilities.RESPONSES_API,
635
+ ModelCapabilities.STREAMING,
636
+ ModelCapabilities.STRUCTURED_OUTPUT,
637
+ ModelCapabilities.VISION,
638
+ ],
639
+ token_limits=LanguageModelTokenLimits(
640
+ token_limit_input=400_000, token_limit_output=128_000
641
+ ),
642
+ info_cutoff_at=date(2024, 9, 30),
643
+ published_at=date(2025, 12, 11),
644
+ temperature_bounds=TemperatureBounds(
645
+ min_temperature=1.0, max_temperature=1.0
646
+ ),
647
+ default_options={
648
+ "reasoning_effort": None,
649
+ },
650
+ )
651
+ case LanguageModelName.AZURE_GPT_52_CHAT_2025_1211:
652
+ return cls(
653
+ name=model_name,
654
+ provider=LanguageModelProvider.AZURE,
655
+ version="2025-12-11",
656
+ encoder_name=EncoderName.O200K_BASE,
657
+ capabilities=[
658
+ ModelCapabilities.CHAT_COMPLETIONS_API,
659
+ ModelCapabilities.FUNCTION_CALLING,
660
+ ModelCapabilities.PARALLEL_FUNCTION_CALLING,
661
+ ModelCapabilities.RESPONSES_API,
662
+ ModelCapabilities.STREAMING,
663
+ ModelCapabilities.STRUCTURED_OUTPUT,
664
+ ],
665
+ token_limits=LanguageModelTokenLimits(
666
+ token_limit_input=128_000, token_limit_output=16_384
667
+ ),
668
+ info_cutoff_at=date(2024, 9, 30),
669
+ published_at=date(2025, 12, 11),
670
+ temperature_bounds=TemperatureBounds(
671
+ min_temperature=1.0, max_temperature=1.0
672
+ ),
673
+ default_options={
674
+ "reasoning_effort": "medium",
675
+ },
676
+ )
614
677
  case LanguageModelName.AZURE_GPT_4_TURBO_2024_0409:
615
678
  return cls(
616
679
  name=model_name,
@@ -1216,6 +1279,25 @@ class LanguageModelInfo(BaseModel):
1216
1279
  info_cutoff_at=date(2025, 1, day=1),
1217
1280
  published_at=date(2025, 6, 5),
1218
1281
  )
1282
+ case LanguageModelName.GEMINI_3_FLASH_PREVIEW:
1283
+ return cls(
1284
+ name=model_name,
1285
+ capabilities=[
1286
+ ModelCapabilities.FUNCTION_CALLING,
1287
+ ModelCapabilities.STREAMING,
1288
+ ModelCapabilities.VISION,
1289
+ ModelCapabilities.STRUCTURED_OUTPUT,
1290
+ ModelCapabilities.REASONING,
1291
+ ],
1292
+ provider=LanguageModelProvider.LITELLM,
1293
+ version="gemini-3-flash-preview",
1294
+ encoder_name=EncoderName.O200K_BASE, # TODO: Update encoder with litellm
1295
+ token_limits=LanguageModelTokenLimits(
1296
+ token_limit_input=1_048_576, token_limit_output=65_536
1297
+ ),
1298
+ info_cutoff_at=date(2025, 1, day=1),
1299
+ published_at=date(2025, 12, 17),
1300
+ )
1219
1301
  case LanguageModelName.GEMINI_3_PRO_PREVIEW:
1220
1302
  return cls(
1221
1303
  name=model_name,
@@ -1422,6 +1504,62 @@ class LanguageModelInfo(BaseModel):
1422
1504
  "reasoning_effort": "medium",
1423
1505
  },
1424
1506
  )
1507
+ case LanguageModelName.LITELLM_OPENAI_GPT_52:
1508
+ return cls(
1509
+ name=model_name,
1510
+ provider=LanguageModelProvider.LITELLM,
1511
+ version="2025-12-11",
1512
+ encoder_name=EncoderName.O200K_BASE,
1513
+ capabilities=[
1514
+ ModelCapabilities.CHAT_COMPLETIONS_API,
1515
+ ModelCapabilities.FUNCTION_CALLING,
1516
+ ModelCapabilities.PARALLEL_FUNCTION_CALLING,
1517
+ ModelCapabilities.REASONING,
1518
+ ModelCapabilities.RESPONSES_API,
1519
+ ModelCapabilities.STREAMING,
1520
+ ModelCapabilities.STRUCTURED_OUTPUT,
1521
+ ModelCapabilities.VISION,
1522
+ ],
1523
+ token_limits=LanguageModelTokenLimits(
1524
+ token_limit_input=400_000, token_limit_output=128_000
1525
+ ),
1526
+ info_cutoff_at=date(2024, 9, 30),
1527
+ published_at=date(2025, 12, 11),
1528
+ temperature_bounds=TemperatureBounds(
1529
+ min_temperature=1.0, max_temperature=1.0
1530
+ ),
1531
+ default_options={
1532
+ "reasoning_effort": None,
1533
+ },
1534
+ )
1535
+ case LanguageModelName.LITELLM_OPENAI_GPT_52_THINKING:
1536
+ return cls(
1537
+ name=model_name,
1538
+ provider=LanguageModelProvider.LITELLM,
1539
+ version="2025-12-11",
1540
+ encoder_name=EncoderName.O200K_BASE,
1541
+ capabilities=[
1542
+ ModelCapabilities.CHAT_COMPLETIONS_API,
1543
+ ModelCapabilities.FUNCTION_CALLING,
1544
+ ModelCapabilities.PARALLEL_FUNCTION_CALLING,
1545
+ ModelCapabilities.REASONING,
1546
+ ModelCapabilities.RESPONSES_API,
1547
+ ModelCapabilities.STREAMING,
1548
+ ModelCapabilities.STRUCTURED_OUTPUT,
1549
+ ModelCapabilities.VISION,
1550
+ ],
1551
+ token_limits=LanguageModelTokenLimits(
1552
+ token_limit_input=400_000, token_limit_output=128_000
1553
+ ),
1554
+ info_cutoff_at=date(2024, 9, 30),
1555
+ published_at=date(2025, 12, 11),
1556
+ temperature_bounds=TemperatureBounds(
1557
+ min_temperature=1.0, max_temperature=1.0
1558
+ ),
1559
+ default_options={
1560
+ "reasoning_effort": "medium",
1561
+ },
1562
+ )
1425
1563
  case LanguageModelName.LITELLM_OPENAI_O1:
1426
1564
  return cls(
1427
1565
  name=model_name,
{unique_toolkit-1.39.0.dist-info → unique_toolkit-1.39.2.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: unique_toolkit
3
- Version: 1.39.0
3
+ Version: 1.39.2
4
4
  Summary:
5
5
  License: Proprietary
6
6
  Author: Cedric Klinkert
@@ -123,6 +123,12 @@ All notable changes to this project will be documented in this file.
123
123
  The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
124
124
  and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
125
125
 
126
+ ## [1.39.2] - 2025-12-18
127
+ - Add `litellm:gemini-3-flash-preview`, `litellm:openai-gpt-5-2` and `litellm:openai-gpt-5-2-thinking` to `language_model/info.py`
128
+
129
+ ## [1.39.1] - 2025-12-17
130
+ - Add GPT-5.2, GPT-5.2_CHAT to supported models list
131
+
126
132
  ## [1.39.0] - 2025-12-17
127
133
  - Adding simpler shortterm message abilities to chat service
128
134
 
{unique_toolkit-1.39.0.dist-info → unique_toolkit-1.39.2.dist-info}/RECORD RENAMED
@@ -194,7 +194,7 @@ unique_toolkit/language_model/builder.py,sha256=4OKfwJfj3TrgO1ezc_ewIue6W7BCQ2ZY
194
194
  unique_toolkit/language_model/constants.py,sha256=B-topqW0r83dkC_25DeQfnPk3n53qzIHUCBS7YJ0-1U,119
195
195
  unique_toolkit/language_model/default_language_model.py,sha256=-_DBsJhLCsFdaU4ynAkyW0jYIl2lhrPybZm1K-GgVJs,125
196
196
  unique_toolkit/language_model/functions.py,sha256=PTBm2BBkuqISVHoyUqMIGHGXT-RMSAfz0F_Ylo2esQ8,18246
197
- unique_toolkit/language_model/infos.py,sha256=1QLqwEOdo7uKkV1zJebj0wqmNn3xwiSlAOFw2h0g_ss,82985
197
+ unique_toolkit/language_model/infos.py,sha256=73guq14ggNplXJ5mx-s71rnyA66pNYnXaiVYSKSGMtw,89733
198
198
  unique_toolkit/language_model/prompt.py,sha256=JSawaLjQg3VR-E2fK8engFyJnNdk21zaO8pPIodzN4Q,3991
199
199
  unique_toolkit/language_model/reference.py,sha256=nkX2VFz-IrUz8yqyc3G5jUMNwrNpxITBrMEKkbqqYoI,8583
200
200
  unique_toolkit/language_model/schemas.py,sha256=ATiHjhfGxoubS332XuhL9PKSoFewcWvPTUVBaNGWlJo,23994
@@ -212,7 +212,7 @@ unique_toolkit/short_term_memory/service.py,sha256=5PeVBu1ZCAfyDb2HLVvlmqSbyzBBu
212
212
  unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
213
213
  unique_toolkit/smart_rules/compile.py,sha256=Ozhh70qCn2yOzRWr9d8WmJeTo7AQurwd3tStgBMPFLA,1246
214
214
  unique_toolkit/test_utilities/events.py,sha256=_mwV2bs5iLjxS1ynDCjaIq-gjjKhXYCK-iy3dRfvO3g,6410
215
- unique_toolkit-1.39.0.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
216
- unique_toolkit-1.39.0.dist-info/METADATA,sha256=DTmUY_QVEo70IW79vzgIZb436Ptf917vl8JtwCqm8r4,45869
217
- unique_toolkit-1.39.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
218
- unique_toolkit-1.39.0.dist-info/RECORD,,
215
+ unique_toolkit-1.39.2.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
216
+ unique_toolkit-1.39.2.dist-info/METADATA,sha256=5i_0jJRn-2014qhIdULsZ2EZHAM9KUal7jjJUqis28s,46105
217
+ unique_toolkit-1.39.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
218
+ unique_toolkit-1.39.2.dist-info/RECORD,,