[router] Support complex assistant and tool messages in /chat/completions (#12860)
Co-authored-by: Chang Su <chang.s.su@oracle.com> Co-authored-by: Simo Lin <linsimo.mark@gmail.com>
This commit is contained in:
co-authored by
Chang Su
Simo Lin
parent
ad8d24c39e
commit
d28caaf60a
@@ -3,7 +3,7 @@ mod tests {
|
||||
use std::fs;
|
||||
|
||||
use sglang_router_rs::{
|
||||
protocols::chat::{ChatMessage, UserMessageContent},
|
||||
protocols::chat::{ChatMessage, MessageContent},
|
||||
tokenizer::{chat_template::ChatTemplateParams, huggingface::HuggingFaceTokenizer},
|
||||
};
|
||||
use tempfile::TempDir;
|
||||
@@ -61,11 +61,11 @@ mod tests {
|
||||
|
||||
let messages = [
|
||||
ChatMessage::User {
|
||||
content: UserMessageContent::Text("Hello".to_string()),
|
||||
content: MessageContent::Text("Hello".to_string()),
|
||||
name: None,
|
||||
},
|
||||
ChatMessage::Assistant {
|
||||
content: Some("Hi there".to_string()),
|
||||
content: Some(MessageContent::Text("Hi there".to_string())),
|
||||
name: None,
|
||||
tool_calls: None,
|
||||
reasoning_content: None,
|
||||
@@ -143,7 +143,7 @@ mod tests {
|
||||
.unwrap();
|
||||
|
||||
let messages = [ChatMessage::User {
|
||||
content: UserMessageContent::Text("Test".to_string()),
|
||||
content: MessageContent::Text("Test".to_string()),
|
||||
name: None,
|
||||
}];
|
||||
|
||||
@@ -202,11 +202,11 @@ mod tests {
|
||||
|
||||
let messages = [
|
||||
ChatMessage::User {
|
||||
content: UserMessageContent::Text("Hello".to_string()),
|
||||
content: MessageContent::Text("Hello".to_string()),
|
||||
name: None,
|
||||
},
|
||||
ChatMessage::Assistant {
|
||||
content: Some("World".to_string()),
|
||||
content: Some(MessageContent::Text("World".to_string())),
|
||||
name: None,
|
||||
tool_calls: None,
|
||||
reasoning_content: None,
|
||||
|
||||
Reference in New Issue
Block a user