diff options
Diffstat (limited to 'src/rag/mod.rs')
| -rw-r--r-- | src/rag/mod.rs | 6 |
1 files changed, 3 insertions, 3 deletions
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!") }, }; |