diff options
| author | sigoden <sigoden@gmail.com> | 2024-10-29 14:15:02 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-10-29 14:15:02 +0800 |
| commit | bf07cdf256e1e3008578972f5916a4d6763554f5 (patch) | |
| tree | 3f48ead051403a7f270198578e86a92672bb4208 /src/rag | |
| parent | bb542f6e92fd782b70c44fddb21d013c289d0a3c (diff) | |
| download | aichat-bf07cdf256e1e3008578972f5916a4d6763554f5.tar.gz | |
refactor: improve code quality (#956)
Diffstat (limited to 'src/rag')
| -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!") }, }; |