unique_toolkit 2026.30.0.dev0__py3-none-any.whl → 2026.30.0.dev2__py3-none-any.whl

unique_toolkit/language_model/infos.py

@@ -130,6 +130,7 @@ class LanguageModelName(StrEnum):
     LITELLM_DEEPSEEK_V4_PRO = "litellm:deepseek-v4-pro"
     LITELLM_GLM_5_1 = "litellm:glm-5.1"
     LITELLM_GLM_5_2 = "litellm:glm-5.2"
+    LITELLM_KIMI_K2_6 = "litellm:kimi-k2.6"
     LITELLM_QWEN_3 = "litellm:qwen-3-235B-A22B"
     LITELLM_QWEN_3_THINKING = "litellm:qwen-3-235B-A22B-thinking"
     VERTEX_CLAUDE_SONNET_4_6 = "litellm:vertex-claude-sonnet-4-6"
@@ -2737,7 +2738,26 @@ class LanguageModelInfo(BaseModel):
                         ModelCapabilities.STREAMING,
                     ],
                     token_limits=LanguageModelTokenLimits(
-                        token_limit_input=1_000_000, token_limit_output=128_000
+                        token_limit_input=256_000, token_limit_output=128_000
+                    ),  # GLM-5.2 supports up to a 1_000_000 token context window. Together AI is supporting however only up to 256_000 tokens.
+                    published_at=date(2026, 1, 1),
+                    supported_reasoning_efforts=[],
+                )
+            case LanguageModelName.LITELLM_KIMI_K2_6:
+                return cls(
+                    name=model_name,
+                    provider=LanguageModelProvider.LITELLM,
+                    family=ModelFamily.KIMI,
+                    version="kimi-k2.6",
+                    encoder_name=EncoderName.O200K_BASE,
+                    capabilities=[
+                        ModelCapabilities.FUNCTION_CALLING,
+                        ModelCapabilities.STRUCTURED_OUTPUT,
+                        ModelCapabilities.REASONING,
+                        ModelCapabilities.STREAMING,
+                    ],
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=256_000, token_limit_output=32_000
                     ),
                     published_at=date(2026, 1, 1),
                     supported_reasoning_efforts=[],

unique_toolkit/language_model/settings.py

@@ -20,6 +20,7 @@ class ModelFamily(StrEnum):
     GOOGLE = "google"
     XAI = "xai"
     DEEPSEEK = "deepseek"
+    KIMI = "kimi"
     QWEN = "qwen"
     MISTRAL = "mistral"
     ZAI = "zai"

{unique_toolkit-2026.30.0.dev0.dist-info → unique_toolkit-2026.30.0.dev2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.30.0.dev0
+Version: 2026.30.0.dev2
 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.30.0.dev0.dist-info → unique_toolkit-2026.30.0.dev2.dist-info}/RECORD

@@ -381,12 +381,12 @@ unique_toolkit/language_model/builder.py,sha256=KwBzsD6-fvQnB4p2I7M-5Tk6j4tmQ973
 unique_toolkit/language_model/constants.py,sha256=B-topqW0r83dkC_25DeQfnPk3n53qzIHUCBS7YJ0-1U,119
 unique_toolkit/language_model/default_language_model.py,sha256=mcyuFs_V6tp52C8xJSHiEHO4kFGj6xed1gvozvCRD7k,1181
 unique_toolkit/language_model/functions.py,sha256=iTxCpcavUp1kAitJi16SIL33Y442aWKWH4W2qPpLfts,23054
-unique_toolkit/language_model/infos.py,sha256=OSxn81oMo4_Pq1OGOSyjrQQVD-sAXizWFrTVy6OpIfY,135540
+unique_toolkit/language_model/infos.py,sha256=q1SLAffrJDK56xD6dgcgZ-m_0g6uO9pvYPtUx18pEGk,136590
 unique_toolkit/language_model/prompt.py,sha256=JSawaLjQg3VR-E2fK8engFyJnNdk21zaO8pPIodzN4Q,3991
 unique_toolkit/language_model/reference.py,sha256=T3wr2jP_uRg5wKVIbKqLbxJlcgxfbZQGq3XdjbQ2kSo,6984
 unique_toolkit/language_model/schemas.py,sha256=JZbIT3w1cj5fKrYaQ55X7sjXmCo6i_GqgJ198V3RXSg,29474
 unique_toolkit/language_model/service.py,sha256=hDoJBBgG_wWs8WuC_c8tnmK8cNu8o5EhJJXRvgwhkFQ,11956
-unique_toolkit/language_model/settings.py,sha256=qoJoSJTlmu-cN6EffmwbnND-ZhLt22zzNAjaa-eb618,1506
+unique_toolkit/language_model/settings.py,sha256=7szohExI-qmTfUf2Zp5AB5TzrsZy1UekAGyXx_UaPMc,1524
 unique_toolkit/language_model/stream_transform.py,sha256=iNNGpLY_7G6Bpp9Y4KtbIkQm3WB4yZt75BXcqx0DVkI,2775
 unique_toolkit/language_model/utils.py,sha256=bPQ4l6_YO71w-zaIPanUUmtbXC1_hCvLK0tAFc3VCRc,1902
 unique_toolkit/monitoring/__init__.py,sha256=z1gkOCLxGcg936MFuLSeWFa0tREcBFSUbV09NxilBw4,1065
@@ -406,6 +406,6 @@ unique_toolkit/short_term_memory/service.py,sha256=2JMBEBn6GoyK-a0xvdcWpPft-3C74
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=vRe0aMJWuVIK-VsqomgxQImzke7MDmwgSxxJfYCa7zw,1386
 unique_toolkit/test_utilities/events.py,sha256=FLrhMh0A_mQutAUjoD0FdMcVQ4Gq31r35OVGzRjvLro,6760
-unique_toolkit-2026.30.0.dev0.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
-unique_toolkit-2026.30.0.dev0.dist-info/METADATA,sha256=-gy54Cx8xyxMVhqLP93lhU14tDJ2jqf49i3fPfDavXw,7426
-unique_toolkit-2026.30.0.dev0.dist-info/RECORD,,
+unique_toolkit-2026.30.0.dev2.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
+unique_toolkit-2026.30.0.dev2.dist-info/METADATA,sha256=5dzAWd8OVSASrP4GIyR8e1NDX86GpCYYE8tzzZC4W3s,7426
+unique_toolkit-2026.30.0.dev2.dist-info/RECORD,,