unique_toolkit 0.7.23__py3-none-any.whl → 0.7.24__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.
@@ -29,8 +29,11 @@ class LanguageModelName(StrEnum):
29
29
  ANTHROPIC_CLAUDE_3_7_SONNET_THINKING = (
30
30
  "litellm:anthropic-claude-3-7-sonnet-thinking"
31
31
  )
32
+ ANTHROPIC_CLAUDE_SONNET_4 = "litellm:anthropic-claude-sonnet-4"
33
+ ANTHROPIC_CLAUDE_OPUS_4 = "litellm:anthropic-claude-opus-4"
32
34
  GEMINI_2_0_FLASH = "litellm:gemini-2-0-flash"
33
35
  GEMINI_2_5_FLASH_PREVIEW_0417 = "litellm:gemini-2-5-flash-preview-04-17"
36
+ GEMINI_2_5_FLASH_PREVIEW_0520 = "litellm:gemini-2-5-flash-preview-05-20"
34
37
  GEMINI_2_5_PRO_EXP_0325 = "litellm:gemini-2-5-pro-exp-03-25"
35
38
 
36
39
 
@@ -410,6 +413,42 @@ class LanguageModelInfo(BaseModel):
410
413
  info_cutoff_at=date(2024, 10, 31),
411
414
  published_at=date(2025, 2, 24),
412
415
  )
416
+ case LanguageModelName.ANTHROPIC_CLAUDE_SONNET_4:
417
+ return cls(
418
+ name=model_name,
419
+ capabilities=[
420
+ ModelCapabilities.FUNCTION_CALLING,
421
+ ModelCapabilities.STREAMING,
422
+ ModelCapabilities.VISION,
423
+ ModelCapabilities.REASONING,
424
+ ],
425
+ provider=LanguageModelProvider.LITELLM,
426
+ version="claude-sonnet-4",
427
+ encoder_name=EncoderName.O200K_BASE, # TODO: Update encoder with litellm
428
+ token_limits=LanguageModelTokenLimits(
429
+ token_limit_input=200_000, token_limit_output=64_000
430
+ ),
431
+ info_cutoff_at=date(2025, 3, 1),
432
+ published_at=date(2025, 5, 1),
433
+ )
434
+ case LanguageModelName.ANTHROPIC_CLAUDE_OPUS_4:
435
+ return cls(
436
+ name=model_name,
437
+ capabilities=[
438
+ ModelCapabilities.FUNCTION_CALLING,
439
+ ModelCapabilities.STREAMING,
440
+ ModelCapabilities.VISION,
441
+ ModelCapabilities.REASONING,
442
+ ],
443
+ provider=LanguageModelProvider.LITELLM,
444
+ version="claude-opus-4",
445
+ encoder_name=EncoderName.O200K_BASE, # TODO: Update encoder with litellm
446
+ token_limits=LanguageModelTokenLimits(
447
+ token_limit_input=200_000, token_limit_output=32_000
448
+ ),
449
+ info_cutoff_at=date(2025, 3, 1),
450
+ published_at=date(2025, 5, 1),
451
+ )
413
452
  case LanguageModelName.GEMINI_2_0_FLASH:
414
453
  return cls(
415
454
  name=model_name,
@@ -448,6 +487,25 @@ class LanguageModelInfo(BaseModel):
448
487
  info_cutoff_at=date(2025, 1, day=1),
449
488
  published_at=date(2025, 4, 1),
450
489
  )
490
+ case LanguageModelName.GEMINI_2_5_FLASH_PREVIEW_0520:
491
+ return cls(
492
+ name=model_name,
493
+ capabilities=[
494
+ ModelCapabilities.FUNCTION_CALLING,
495
+ ModelCapabilities.STREAMING,
496
+ ModelCapabilities.VISION,
497
+ ModelCapabilities.STRUCTURED_OUTPUT,
498
+ ModelCapabilities.REASONING,
499
+ ],
500
+ provider=LanguageModelProvider.LITELLM,
501
+ version="gemini-2-5-flash-preview-05-20",
502
+ encoder_name=EncoderName.O200K_BASE, # TODO:Replace with LLM tokenizer
503
+ token_limits=LanguageModelTokenLimits(
504
+ token_limit_input=1_048_576, token_limit_output=65_536
505
+ ),
506
+ info_cutoff_at=date(2025, 1, day=1),
507
+ published_at=date(2025, 4, 1),
508
+ )
451
509
  case LanguageModelName.GEMINI_2_5_PRO_EXP_0325:
452
510
  return cls(
453
511
  name=model_name,
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: unique_toolkit
3
- Version: 0.7.23
3
+ Version: 0.7.24
4
4
  Summary:
5
5
  License: Proprietary
6
6
  Author: Martin Fadler
@@ -111,6 +111,9 @@ All notable changes to this project will be documented in this file.
111
111
  The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
112
112
  and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
113
113
 
114
+ ## [0.7.24] - 2025-05-30
115
+ - Adding litellm model `gemini-2-5-flash-preview-05-20`, `anthropic-claude-sonnet-4` and `anthropic-claude-opus-4`
116
+
114
117
  ## [0.7.23] - 2025-05-22
115
118
  - add encoder for `AZURE_GPT_4o_2024_1120` to be part of the encoder function returns.
116
119
 
@@ -48,7 +48,7 @@ unique_toolkit/language_model/__init__.py,sha256=lRQyLlbwHbNFf4-0foBU13UGb09lwEe
48
48
  unique_toolkit/language_model/builder.py,sha256=69WCcmkm2rMP2-YEH_EjHiEp6OzwjwCs8VbhjVJaCe0,3168
49
49
  unique_toolkit/language_model/constants.py,sha256=B-topqW0r83dkC_25DeQfnPk3n53qzIHUCBS7YJ0-1U,119
50
50
  unique_toolkit/language_model/functions.py,sha256=koCAfhtkIGSiy8pSdDpIw9xRbwJ20EeLhDQMUXc8KZk,8049
51
- unique_toolkit/language_model/infos.py,sha256=oMm8TkK6YIjR1Uy8_EQCO25WqKcEhMMpwFlnIsHv0E4,25951
51
+ unique_toolkit/language_model/infos.py,sha256=c44oyI0IuYILVmvxndGlKr_OecYz0Zvn8orCAtDv8WQ,28889
52
52
  unique_toolkit/language_model/prompt.py,sha256=JSawaLjQg3VR-E2fK8engFyJnNdk21zaO8pPIodzN4Q,3991
53
53
  unique_toolkit/language_model/schemas.py,sha256=DJD2aoMfs2Irnc4rzOrVuV4Fbt84LQAiDGG5rse1dgk,12770
54
54
  unique_toolkit/language_model/service.py,sha256=9LS3ouRNtzqZaKrMFagLZS9gBvNC5e46Ut86YWHBBHY,8470
@@ -59,7 +59,7 @@ unique_toolkit/short_term_memory/constants.py,sha256=698CL6-wjup2MvU19RxSmQk3gX7
59
59
  unique_toolkit/short_term_memory/functions.py,sha256=3WiK-xatY5nh4Dr5zlDUye1k3E6kr41RiscwtTplw5k,4484
60
60
  unique_toolkit/short_term_memory/schemas.py,sha256=OhfcXyF6ACdwIXW45sKzjtZX_gkcJs8FEZXcgQTNenw,1406
61
61
  unique_toolkit/short_term_memory/service.py,sha256=vEKFxP1SScPrFniso492fVthWR1sosdFibhiNF3zRvI,8081
62
- unique_toolkit-0.7.23.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
63
- unique_toolkit-0.7.23.dist-info/METADATA,sha256=Gku_wy6SKZskvVjQ2u-nd5W0mkGpr0bDNnEFdvgzgE4,23317
64
- unique_toolkit-0.7.23.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
65
- unique_toolkit-0.7.23.dist-info/RECORD,,
62
+ unique_toolkit-0.7.24.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
63
+ unique_toolkit-0.7.24.dist-info/METADATA,sha256=s-kBSrSHAM1Qh3g4b_hGvB4EoBFA2jrjlrNu5PzH4I8,23458
64
+ unique_toolkit-0.7.24.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
65
+ unique_toolkit-0.7.24.dist-info/RECORD,,