Skip to content

Commit 1768c00

Browse files
authored
feat: return the generated text when parsing fails (#2353)
1 parent f8a5b38 commit 1768c00

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

router/src/server.rs

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1246,9 +1246,13 @@ async fn chat_completions(
12461246
.as_secs();
12471247

12481248
let (tool_calls, output) = if tool_grammar.is_some() {
1249-
let gen_text_value: Value = serde_json::from_str(&generation.generated_text)
1250-
.map_err(|e| InferError::ToolError(e.to_string()))?;
1251-
1249+
let gen_text_value: Value =
1250+
serde_json::from_str(&generation.generated_text).map_err(|e| {
1251+
InferError::ToolError(format!(
1252+
"Failed to parse generated text: {} {:?}",
1253+
e, generation.generated_text
1254+
))
1255+
})?;
12521256
let function = gen_text_value.get("function").ok_or(InferError::ToolError(
12531257
"No function found in generated text".to_string(),
12541258
))?;

0 commit comments

Comments
 (0)