mirrors/zed
1
0
Fork 0
mirror of https://github.com/zed-industries/zed.git synced 2025-02-10 12:19:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

Debounce searches in semantic search modal

Browse source
This commit is contained in:
Max Brunsfeld 2023-06-28 15:02:20 -07:00
parent 85e71415fe
commit fd68a2afae
2 changed files with 26 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
crates/vector_store/src/modal.rs
View file

@ -1,15 +1,16 @@
use std::sync::Arc;
use crate::{SearchResult, VectorStore};
use gpui::{
actions, elements::*, AnyElement, AppContext, ModelHandle, MouseState, Task, ViewContext,
WeakViewHandle,
};
use picker::{Picker, PickerDelegate, PickerEvent};
use project::Project;
use std::{sync::Arc, time::Duration};
use util::ResultExt;
use workspace::Workspace;
use crate::{SearchResult, VectorStore};
const MIN_QUERY_LEN: usize = 5;
const EMBEDDING_DEBOUNCE_INTERVAL: Duration = Duration::from_millis(500);
actions!(semantic_search, [Toggle]);
@ -68,18 +69,26 @@ impl PickerDelegate for SemanticSearchDelegate {
}
fn update_matches(&mut self, query: String, cx: &mut ViewContext<SemanticSearch>) -> Task<()> {
let task = self.vector_store.update(cx, |store, cx| {
store.search(&self.project, query.to_string(), 10, cx)
});
if query.len() < MIN_QUERY_LEN {
return Task::ready(());
}
let vector_store = self.vector_store.clone();
let project = self.project.clone();
cx.spawn(|this, mut cx| async move {
let results = task.await.log_err();
this.update(&mut cx, |this, cx| {
if let Some(results) = results {
cx.background().timer(EMBEDDING_DEBOUNCE_INTERVAL).await;
let task = vector_store.update(&mut cx, |store, cx| {
store.search(&project, query.to_string(), 10, cx)
});
if let Some(results) = task.await.log_err() {
this.update(&mut cx, |this, _| {
let delegate = this.delegate_mut();
delegate.matches = results;
}
});
})
.ok();
}
})
}

11
crates/vector_store/src/vector_store.rs
View file

@ -31,14 +31,14 @@ pub fn init(
language_registry: Arc<LanguageRegistry>,
cx: &mut AppContext,
) {
let vector_store = cx.add_model(|cx| {
let vector_store = cx.add_model(|_| {
VectorStore::new(
fs,
VECTOR_DB_URL.to_string(),
// Arc::new(OpenAIEmbeddings {
// client: http_client,
// }),
Arc::new(DummyEmbeddings {}),
// Arc::new(DummyEmbeddings {}),
Arc::new(OpenAIEmbeddings {
client: http_client,
}),
language_registry,
)
});
@ -74,6 +74,7 @@ pub fn init(
})
}
});
SemanticSearch::init(cx);
}