use crate::provider::{CompletionRequest, ContentPart, Message, Role};
use super::prompt::{ROUTER_SYSTEM, build_user_prompt};
pub fn build_router_request(model: &str, query: &str, top_n: usize) -> CompletionRequest {
CompletionRequest {
model: model.to_string(),
messages: vec![
Message {
role: Role::System,
content: vec![ContentPart::Text {
text: ROUTER_SYSTEM.to_string(),
}],
},
Message {
role: Role::User,
content: vec![ContentPart::Text {
text: build_user_prompt(query, top_n),
}],
},
],
tools: Vec::new(),
temperature: Some(1.0),
top_p: Some(0.9),
max_tokens: Some(400),
stop: Vec::new(),
}
}