From 0fcc7e1377eac5ce85477c63133e3ef88a559cb0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cl=C3=A9ment=20Renault?= <clement@meilisearch.com>
Date: Tue, 20 May 2025 12:23:22 +0200
Subject: [PATCH] Stream errors

---
 crates/meilisearch/src/routes/chat.rs | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/crates/meilisearch/src/routes/chat.rs b/crates/meilisearch/src/routes/chat.rs
index 733b8ff65..5ddcb6088 100644
--- a/crates/meilisearch/src/routes/chat.rs
+++ b/crates/meilisearch/src/routes/chat.rs
@@ -400,13 +400,21 @@ async fn streamed_chat(
                         }
                     }
                     Err(err) => {
-                        // writeln!(lock, "error: {err}").unwrap();
                         tracing::error!("{err:?}");
+                        if let Err(SendError(_)) = tx.send(Event::Data(sse::Data::new_json(&json!({
+                            "object": "chat.completion.error",
+                            "tool": err.to_string(),
+                        })).unwrap())).await {
+                            return;
+                        }
+
                         break 'main;
                     }
                 }
             }
         }
+
+        let _ = tx.send(Event::Data(sse::Data::new("[DONE]")));
     });
 
     Sse::from_infallible_receiver(rx).with_retry_duration(Duration::from_secs(10))