From 21d0f16da6977de19b996a4730e96f68843e14d7 Mon Sep 17 00:00:00 2001
From: cryscan <cryscan@umich.edu>
Date: Tue, 21 May 2024 15:06:10 +0800
Subject: [PATCH] Make `model_route` exit gracefully.

---
 crates/ai00-core/src/lib.rs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/crates/ai00-core/src/lib.rs b/crates/ai00-core/src/lib.rs
index 261d72f1..aee93531 100644
--- 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());