Return the right message format

This commit is contained in:
Clément Renault 2025-05-14 12:03:43 +02:00
parent af482d8ee9
commit 24050f06e4
No known key found for this signature in database
GPG Key ID: F250A4C4E3AE5F5F

View File

@ -301,7 +301,7 @@ async fn streamed_chat(
let mut global_tool_calls = HashMap::<u32, Call>::new(); let mut global_tool_calls = HashMap::<u32, Call>::new();
actix_web_lab::sse::Sse::from_stream(response.map(move |response| { actix_web_lab::sse::Sse::from_stream(response.map(move |response| {
response.map(|mut r| { response.map(|mut r| {
let delta = r.choices.pop().unwrap().delta; let delta = &r.choices[0].delta;
let ChatCompletionStreamResponseDelta { let ChatCompletionStreamResponseDelta {
ref content, ref content,
ref function_call, ref function_call,
@ -330,10 +330,10 @@ async fn streamed_chat(
None if !global_tool_calls.is_empty() => { None if !global_tool_calls.is_empty() => {
dbg!(&global_tool_calls); dbg!(&global_tool_calls);
} }
_ => (), None => (),
} }
Event::Data(sse::Data::new_json(delta).unwrap()) Event::Data(sse::Data::new_json(r).unwrap())
}) })
})) }))
} }