Make model_route exit gracefully.

Ai00-X · May 21, 2024 · 21d0f16 · 21d0f16
1 parent 3228cc7
commit 21d0f16
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/crates/ai00-core/src/lib.rs b/crates/ai00-core/src/lib.rs
@@ -476,8 +476,13 @@ pub async fn model_route(receiver: Receiver<ThreadRequest>) -> Result<()> {
     tokio::spawn(dequeue);
 
     loop {
+        let Ok(request) = receiver.recv_async().await else {
+            log::info!("core exit");
+            break Ok(());
+        };
+
         let listen = async {
-            match receiver.recv_async().await.unwrap() {
+            match request {
                 ThreadRequest::Adapter(sender) => {
                     tokio::spawn(async move {
                         let _ = sender.send(list_adapters());