EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

[Router] Refactor protocol definitions: split spec.rs into modular files (#11677)

Browse Source 
Co-authored-by: Chang Su <chang.s.su@oracle.com>
This commit is contained in:
Keyang Ru
2025-10-16 13:44:44 -07:00
committed by GitHub
parent 86b04d25b3
commit 4c9bcb9d56
56 changed files with 2939 additions and 2914 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
sgl-router/tests/chat_template_loading.rs
View File

@@ -1,6 +1,6 @@
#[cfg(test)]
mod tests {
use sglang_router_rs::protocols::spec;
use sglang_router_rs::protocols::chat::{ChatMessage, UserMessageContent};
use sglang_router_rs::tokenizer::chat_template::ChatTemplateParams;
use sglang_router_rs::tokenizer::huggingface::HuggingFaceTokenizer;
use std::fs;
@@ -58,11 +58,11 @@ mod tests {
.unwrap();
let messages = [
spec::ChatMessage::User {
content: spec::UserMessageContent::Text("Hello".to_string()),
ChatMessage::User {
content: UserMessageContent::Text("Hello".to_string()),
name: None,
},
spec::ChatMessage::Assistant {
ChatMessage::Assistant {
content: Some("Hi there".to_string()),
name: None,
tool_calls: None,
@@ -140,8 +140,8 @@ mod tests {
)
.unwrap();
let messages = [spec::ChatMessage::User {
content: spec::UserMessageContent::Text("Test".to_string()),
let messages = [ChatMessage::User {
content: UserMessageContent::Text("Test".to_string()),
name: None,
}];
@@ -199,11 +199,11 @@ mod tests {
tokenizer.set_chat_template(new_template.to_string());
let messages = [
spec::ChatMessage::User {
content: spec::UserMessageContent::Text("Hello".to_string()),
ChatMessage::User {
content: UserMessageContent::Text("Hello".to_string()),
name: None,
},
spec::ChatMessage::Assistant {
ChatMessage::Assistant {
content: Some("World".to_string()),
name: None,
tool_calls: None,