Mirrors
/
hf_text-generation-inference
mirror of https://github.com/huggingface/text-generation-inference.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat: return the generated text when parsing fails (#2353)

This commit is contained in:
drbh 2024-08-06 13:10:19 -04:00 committed by GitHub
parent f8a5b381fe
commit 1768c00b9f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
router/src/server.rs
View File

@ -1246,9 +1246,13 @@ async fn chat_completions(
.as_secs(); .as_secs();
let (tool_calls, output) = if tool_grammar.is_some() { let (tool_calls, output) = if tool_grammar.is_some() {
let gen_text_value: Value = serde_json::from_str(&generation.generated_text) let gen_text_value: Value =
.map_err(|e| InferError::ToolError(e.to_string()))?; serde_json::from_str(&generation.generated_text).map_err(|e| {
InferError::ToolError(format!(
"Failed to parse generated text: {} {:?}",
e, generation.generated_text
))
})?;
let function = gen_text_value.get("function").ok_or(InferError::ToolError( let function = gen_text_value.get("function").ok_or(InferError::ToolError(
"No function found in generated text".to_string(), "No function found in generated text".to_string(),
))?; ))?;