From c21ea483f4875737705d7acc39eccc5096b846fb Mon Sep 17 00:00:00 2001 From: Meng Zhang Date: Fri, 3 Nov 2023 17:35:06 -0700 Subject: [PATCH] fix: collect_snippet should handle NotReady error --- crates/tabby/src/serve/completions/prompt.rs | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/crates/tabby/src/serve/completions/prompt.rs b/crates/tabby/src/serve/completions/prompt.rs index d78f4b8..b538030 100644 --- a/crates/tabby/src/serve/completions/prompt.rs +++ b/crates/tabby/src/serve/completions/prompt.rs @@ -109,8 +109,12 @@ fn collect_snippets(index_server: &IndexServer, language: &str, text: &str) -> V let mut ret = Vec::new(); let mut tokens = tokenize_text(text); - let language_query = index_server.language_query(language).unwrap(); - let body_query = index_server.body_query(&tokens).unwrap(); + let Ok(language_query) = index_server.language_query(language) else { + return vec![]; + }; + let Ok(body_query) = index_server.body_query(&tokens) else { + return vec![]; + }; let query = BooleanQuery::new(vec![ (Occur::Must, language_query), (Occur::Must, body_query),