chore: expand moonshot template benchmark coverage
This commit is contained in:
1 parent
9b1e973e4e
commit
49ea5f7893
1 file changed
+42
@@ -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:"));
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user