Skip to content

Commit a37d3d4

Browse files
authored
feat: add support for ministral-3 model type in capabilities manager (#1307)
* feat: add support for ministral-3 model type in capabilities manager * fixes
1 parent 2beb775 commit a37d3d4

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

shinkai-bin/shinkai-node/src/managers/model_capabilities_manager.rs

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -180,6 +180,9 @@ impl ModelCapabilitiesManager {
180180
model_type if model_type.starts_with("mistral-small3.2") || model_type.starts_with("mistral-small3.1") => {
181181
vec![ModelCapability::TextInference, ModelCapability::ImageAnalysis]
182182
}
183+
model_type if model_type.starts_with("ministral-3") => {
184+
vec![ModelCapability::TextInference, ModelCapability::ImageAnalysis]
185+
}
183186
model_type if model_type.starts_with("llama3.2-vision") => {
184187
vec![ModelCapability::TextInference, ModelCapability::ImageAnalysis]
185188
}
@@ -808,6 +811,9 @@ impl ModelCapabilitiesManager {
808811
model_type if model_type.starts_with("mistral-small3.2") || model_type.starts_with("mistral-small3.1") => {
809812
128_000
810813
}
814+
model_type if model_type.starts_with("ministral-3") => {
815+
256_000
816+
}
811817
model_type if model_type.starts_with("mistral-small") => 128_000,
812818
model_type if model_type.starts_with("mistral-large") => 128_000,
813819
model_type if model_type.starts_with("mixtral:8x7b-instruct-v0.1") => 16_000,
@@ -1158,6 +1164,7 @@ impl ModelCapabilitiesManager {
11581164
|| model.model_type.starts_with("mistral-small")
11591165
|| model.model_type.starts_with("mistral-large")
11601166
|| model.model_type.starts_with("mistral-pixtral")
1167+
|| model.model_type.starts_with("ministral-3")
11611168
|| model.model_type.starts_with("qwen2.5-coder")
11621169
|| model.model_type.starts_with("qwen2.5vl")
11631170
|| model.model_type.starts_with("qwq")
@@ -1213,6 +1220,7 @@ impl ModelCapabilitiesManager {
12131220
|| model.model_type.starts_with("mistral-small")
12141221
|| model.model_type.starts_with("mistral-large")
12151222
|| model.model_type.starts_with("mistral-pixtral")
1223+
|| model.model_type.starts_with("ministral-3")
12161224
|| model.model_type.starts_with("magistral")
12171225
|| model.model_type.starts_with("gpt-oss")
12181226
}

0 commit comments

Comments
 (0)