From bf07cdf256e1e3008578972f5916a4d6763554f5 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Tue, 29 Oct 2024 14:15:02 +0800
Subject: refactor: improve code quality (#956)

---
 src/rag/mod.rs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'src/rag/mod.rs')

diff --git a/src/rag/mod.rs b/src/rag/mod.rs
index 48ace8e..fd734e2 100644
--- a/src/rag/mod.rs
+++ b/src/rag/mod.rs
@@ -93,7 +93,7 @@ impl Rag {
                 spinner.stop();
                 ret?;
             }
-            _ = watch_abort_signal(abort_signal) => {
+            _ = wait_abort_signal(&abort_signal) => {
                 spinner.stop();
                 bail!("Aborted!")
             },
@@ -142,7 +142,7 @@ impl Rag {
                 spinner.stop();
                 ret?;
             }
-            _ = watch_abort_signal(abort_signal) => {
+            _ = wait_abort_signal(&abort_signal) => {
                 spinner.stop();
                 bail!("Aborted!")
             },
@@ -320,7 +320,7 @@ impl Rag {
             ret = self.hybird_search(text, top_k, min_score_vector_search, min_score_keyword_search, rerank_model) => {
                 ret
             }
-            _ = watch_abort_signal(abort_signal) => {
+            _ = wait_abort_signal(&abort_signal) => {
                 bail!("Aborted!")
             },
         };
-- 
cgit v1.2.3