huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-12 04:44:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: remove dev tests

Browse Source
This commit is contained in:
drbh 2024-07-18 17:48:51 +00:00
parent 21dc6776b1
commit 062f91ad60
1 changed files with 0 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
router/src/lib.rs
View File

@ -1371,47 +1371,4 @@ mod tests {
r#"{"role":"assistant","tool_calls":[{"id":"0","type":"function","function":{"description":null,"name":"myfn","arguments":{"format":"csv"}}}]}"#
);
}
#[test]
fn tool_deserialize() {
// Test ToolCall deserialization
let json = r#"{"id":"0","type":"function","function":{"description":null,"name":"myfn","arguments":{"format":"csv"}}}"#;
let tool: ToolCall = serde_json::from_str(json).unwrap();
assert_eq!(
tool,
ToolCall {
id: "0".to_string(),
r#type: "function".to_string(),
function: FunctionDefinition {
description: None,
name: "myfn".to_string(),
arguments: json!({
"format": "csv"
}),
},
}
);
// Test ToolChoice deserialization with "auto"
let auto_json = r#""auto""#;
let auto_choice: ToolChoice = serde_json::from_str(auto_json).unwrap();
assert_eq!(auto_choice, ToolChoice(Some(ToolType::OneOf)));
// Test ToolChoice deserialization with "none"
let none_json = r#""none""#;
let none_choice: ToolChoice = serde_json::from_str(none_json).unwrap();
assert_eq!(none_choice, ToolChoice(None));
// Test ToolChoice deserialization with a specific function name
let function_json = r#""my_function""#;
let function_choice: ToolChoice = serde_json::from_str(function_json).unwrap();
assert_eq!(
function_choice,
ToolChoice(Some(ToolType::FunctionName("my_function".to_string())))
);
// Test ToolChoice deserialization with no value (should default to OneOf)
let default_json = r#"null"#;
let default_choice: ToolChoice = serde_json::from_str(default_json).unwrap();
assert_eq!(default_choice, ToolChoice(Some(ToolType::OneOf)));
}
}