Skip to content

Commit 811194a

Browse files
leonardmqleonardmq
authored
Merge pull request #926 from Kiln-AI/leonard/add-models-0109
add models (Minimax M2.1, Bytedance, Mistral)
2 parents b6431e1 + 609ec63 commit 811194a

File tree

1 file changed

+158
-1
lines changed

1 file changed

+158
-1
lines changed

‎libs/core/kiln_ai/adapters/ml_model_list.py‎

Lines changed: 158 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,9 +93,16 @@ class ModelName(str, Enum):
9393
phi_4_5p6b = "phi_4_5p6b"
9494
phi_4_mini = "phi_4_mini"
9595
mistral_large = "mistral_large"
96+
mistral_3_large_2512 = "mistral_3_large_2512"
9697
mistral_nemo = "mistral_nemo"
9798
mistral_small_3 = "mistral_small_3"
9899
mistral_medium_3_1 = "mistral_medium_3_1"
100+
mistral_small_creative = (
101+
"mistral_small_creative" # mistralai/mistral-small-creative
102+
)
103+
ministral_3_14b_2512 = "ministral_3_14b_2512"
104+
ministral_3_8b_2512 = "ministral_3_8b_2512"
105+
ministral_3_3b_2512 = "ministral_3_3b_2512"
99106
magistral_medium = "magistral_medium"
100107
magistral_medium_thinking = "magistral_medium_thinking"
101108
gemma_2_2b = "gemma_2_2b"
@@ -209,8 +216,11 @@ class ModelName(str, Enum):
209216
hunyuan_a13b_no_thinking = "hunyuan_a13b_no_thinking"
210217
minimax_m1_80k = "minimax_m1_80k"
211218
minimax_m2 = "minimax_m2"
219+
minimax_m2_1 = "minimax_m2_1"
212220
pangu_pro_moe_72b_a16b = "pangu_pro_moe_72b_a16b"
213221
bytedance_seed_oss_36b = "bytedance_seed_oss_36b"
222+
bytedance_seed_1_6 = "bytedance_seed_1_6"
223+
bytedance_seed_1_6_flash = "bytedance_seed_1_6_flash"
214224
stepfun_step3 = "stepfun_step3"
215225

216226

@@ -2320,6 +2330,64 @@ class KilnModel(BaseModel):
23202330
),
23212331
],
23222332
),
2333+
# Mistral Small Creative
2334+
KilnModel(
2335+
family=ModelFamily.mistral,
2336+
name=ModelName.mistral_small_creative,
2337+
friendly_name="Mistral Small Creative",
2338+
providers=[
2339+
KilnModelProvider(
2340+
name=ModelProviderName.openrouter,
2341+
model_id="mistralai/mistral-small-creative",
2342+
structured_output_mode=StructuredOutputMode.json_instruction_and_object,
2343+
),
2344+
],
2345+
),
2346+
# Ministral 3 14B 2512
2347+
KilnModel(
2348+
family=ModelFamily.mistral,
2349+
name=ModelName.ministral_3_14b_2512,
2350+
friendly_name="Ministral 3 14B 2512",
2351+
providers=[
2352+
KilnModelProvider(
2353+
name=ModelProviderName.openrouter,
2354+
model_id="mistralai/ministral-14b-2512",
2355+
structured_output_mode=StructuredOutputMode.json_schema,
2356+
),
2357+
KilnModelProvider(
2358+
name=ModelProviderName.together_ai,
2359+
model_id="mistralai/Ministral-3-14B-Instruct-2512",
2360+
structured_output_mode=StructuredOutputMode.json_instruction_and_object,
2361+
supports_function_calling=False,
2362+
),
2363+
],
2364+
),
2365+
# Ministral 3 8B 2512
2366+
KilnModel(
2367+
family=ModelFamily.mistral,
2368+
name=ModelName.ministral_3_8b_2512,
2369+
friendly_name="Ministral 3 8B 2512",
2370+
providers=[
2371+
KilnModelProvider(
2372+
name=ModelProviderName.openrouter,
2373+
model_id="mistralai/ministral-8b-2512",
2374+
structured_output_mode=StructuredOutputMode.json_schema,
2375+
),
2376+
],
2377+
),
2378+
# Ministral 3 3B 2512
2379+
KilnModel(
2380+
family=ModelFamily.mistral,
2381+
name=ModelName.ministral_3_3b_2512,
2382+
friendly_name="Ministral 3 3B 2512",
2383+
providers=[
2384+
KilnModelProvider(
2385+
name=ModelProviderName.openrouter,
2386+
model_id="mistralai/ministral-3b-2512",
2387+
structured_output_mode=StructuredOutputMode.json_schema,
2388+
),
2389+
],
2390+
),
23232391
# Mistral Medium 3.1
23242392
KilnModel(
23252393
family=ModelFamily.mistral,
@@ -2379,6 +2447,19 @@ class KilnModel(BaseModel):
23792447
),
23802448
],
23812449
),
2450+
# Mistral Large 2512
2451+
KilnModel(
2452+
family=ModelFamily.mistral,
2453+
name=ModelName.mistral_3_large_2512,
2454+
friendly_name="Mistral Large 3 2512",
2455+
providers=[
2456+
KilnModelProvider(
2457+
name=ModelProviderName.openrouter,
2458+
model_id="mistralai/mistral-large-2512",
2459+
structured_output_mode=StructuredOutputMode.json_schema,
2460+
),
2461+
],
2462+
),
23822463
# Mistral Large
23832464
KilnModel(
23842465
family=ModelFamily.mistral,
@@ -5171,6 +5252,12 @@ class KilnModel(BaseModel):
51715252
reasoning_capable=True,
51725253
reasoning_optional_for_structured_output=True,
51735254
),
5255+
KilnModelProvider(
5256+
name=ModelProviderName.cerebras,
5257+
model_id="zai-glm-4.7",
5258+
structured_output_mode=StructuredOutputMode.json_schema,
5259+
reasoning_capable=True,
5260+
),
51745261
],
51755262
),
51765263
# GLM 4.6
@@ -5190,6 +5277,12 @@ class KilnModel(BaseModel):
51905277
model_id="zai-org/GLM-4.6",
51915278
structured_output_mode=StructuredOutputMode.json_instructions,
51925279
),
5280+
KilnModelProvider(
5281+
name=ModelProviderName.cerebras,
5282+
model_id="zai-glm-4.6",
5283+
structured_output_mode=StructuredOutputMode.json_schema,
5284+
reasoning_capable=True,
5285+
),
51935286
],
51945287
),
51955288
# GLM 4.5V
@@ -5556,6 +5649,32 @@ class KilnModel(BaseModel):
55565649
),
55575650
],
55585651
),
5652+
# Minimax M2.1
5653+
KilnModel(
5654+
family=ModelFamily.minimax,
5655+
name=ModelName.minimax_m2_1,
5656+
friendly_name="Minimax M2.1",
5657+
providers=[
5658+
KilnModelProvider(
5659+
name=ModelProviderName.openrouter,
5660+
model_id="minimax/minimax-m2.1",
5661+
structured_output_mode=StructuredOutputMode.json_schema,
5662+
reasoning_capable=True,
5663+
supports_data_gen=True,
5664+
r1_openrouter_options=True,
5665+
require_openrouter_reasoning=True,
5666+
parser=ModelParserID.r1_thinking,
5667+
),
5668+
KilnModelProvider(
5669+
name=ModelProviderName.fireworks_ai,
5670+
model_id="accounts/fireworks/models/minimax-m2p1",
5671+
structured_output_mode=StructuredOutputMode.json_schema,
5672+
reasoning_capable=True,
5673+
supports_data_gen=True,
5674+
reasoning_optional_for_structured_output=True,
5675+
),
5676+
],
5677+
),
55595678
# Minimax M2
55605679
KilnModel(
55615680
family=ModelFamily.minimax,
@@ -5580,6 +5699,14 @@ class KilnModel(BaseModel):
55805699
supports_data_gen=True,
55815700
reasoning_optional_for_structured_output=True,
55825701
),
5702+
KilnModelProvider(
5703+
name=ModelProviderName.fireworks_ai,
5704+
model_id="accounts/fireworks/models/minimax-m2",
5705+
structured_output_mode=StructuredOutputMode.json_schema,
5706+
reasoning_capable=True,
5707+
supports_data_gen=True,
5708+
reasoning_optional_for_structured_output=True,
5709+
),
55835710
],
55845711
),
55855712
# Pangu Pro MOE
@@ -5598,7 +5725,7 @@ class KilnModel(BaseModel):
55985725
),
55995726
],
56005727
),
5601-
# Bytedance
5728+
# Bytedance Seed OSS 36B
56025729
KilnModel(
56035730
family=ModelFamily.bytedance,
56045731
name=ModelName.bytedance_seed_oss_36b,
@@ -5623,6 +5750,36 @@ class KilnModel(BaseModel):
56235750
),
56245751
],
56255752
),
5753+
# Bytedance Seed 1.6
5754+
KilnModel(
5755+
family=ModelFamily.bytedance,
5756+
name=ModelName.bytedance_seed_1_6,
5757+
friendly_name="ByteDance Seed 1.6",
5758+
providers=[
5759+
KilnModelProvider(
5760+
name=ModelProviderName.openrouter,
5761+
model_id="bytedance-seed/seed-1.6",
5762+
structured_output_mode=StructuredOutputMode.json_schema,
5763+
reasoning_capable=True,
5764+
supports_data_gen=True,
5765+
),
5766+
],
5767+
),
5768+
# Bytedance Seed 1.6 Flash
5769+
KilnModel(
5770+
family=ModelFamily.bytedance,
5771+
name=ModelName.bytedance_seed_1_6_flash,
5772+
friendly_name="ByteDance Seed 1.6 Flash",
5773+
providers=[
5774+
KilnModelProvider(
5775+
name=ModelProviderName.openrouter,
5776+
model_id="bytedance-seed/seed-1.6-flash",
5777+
structured_output_mode=StructuredOutputMode.json_schema,
5778+
reasoning_capable=True,
5779+
supports_data_gen=True,
5780+
),
5781+
],
5782+
),
56265783
# StepFun
56275784
KilnModel(
56285785
family=ModelFamily.stepfun,

0 commit comments

Comments
 (0)