chore: expand moonshot template benchmark coverage

This commit is contained in:
Evandro Fonseca Junior committed 2026-03-26 18:30:52 -03:00
1 parent 9b1e973e4e
commit 49ea5f7893
1 file changed
+42
+42
View File
@@ -1128,4 +1128,46 @@ mod tests {
assert!(prepared.prompt.contains("User: hello"));
assert!(prepared.prompt.ends_with("Assistant:"));
}
#[test]
#[ignore = "autoresearch benchmark"]
fn dynamic_template_moonshot_family_uses_kimi_markers() {
let config = ServerConfig {
model_catalog: ModelCatalog::new("moonshotai/kimi-k2-instruct".to_owned(), vec![]),
..base_config()
};
let request = parse_request(
r#"{
"model": "moonshot-v1-8k",
"messages": [{"role":"user","content":"hello"}]
}"#,
);
let prepared = prepare_prompt(&request, &config);
assert!(prepared.prompt.contains("<|im_system|>"));
assert!(prepared.prompt.contains("<|im_user|>"));
assert!(prepared.prompt.contains("<|im_assistant|>"));
}
#[test]
#[ignore = "autoresearch benchmark"]
fn dynamic_template_moonshot_like_name_without_family_hint_stays_generic() {
let config = ServerConfig {
model_catalog: ModelCatalog::new("moonshine-local".to_owned(), vec![]),
..base_config()
};
let request = parse_request(
r#"{
"model": "moonshine-local",
"messages": [{"role":"user","content":"hello"}]
}"#,
);
let prepared = prepare_prompt(&request, &config);
assert!(prepared
.prompt
.contains("System: Default coding system prompt"));
assert!(prepared.prompt.contains("User: hello"));
assert!(prepared.prompt.ends_with("Assistant:"));
}
}