use super::*;
use crate::tool_parser::ToolDefinition;
mod cold_compile;
mod engine_state;
mod gemma4_required;
mod minimax;
mod misc;
mod native_prewarm;
mod parallel_calls;
mod param_key_constraint;
mod poolside;
mod prewarm_ordering;
mod qwen3_coder_required;
mod sanitize;
mod tools_basic;
mod value_entry;
pub(crate) fn test_vocab() -> Vec<String> {
let mut vocab = Vec::new();
for i in 0u8..128 {
vocab.push(String::from(i as char));
}
vocab.push("<tool_call>".to_string());
vocab.push("</tool_call>".to_string());
vocab.push("<eos>".to_string());
vocab
}
pub(crate) fn test_tool_defs() -> Vec<ToolDefinition> {
vec![ToolDefinition {
tool_type: "function".to_string(),
function: crate::tool_parser::FunctionDefinition {
name: "get_weather".to_string(),
description: Some("Get weather for a location".to_string()),
parameters: Some(serde_json::json!({
"type": "object",
"properties": {
"location": {
"type": "string",
"description": "City name"
}
},
"required": ["location"]
})),
},
}]
}