mirrors/zed
1
0
Fork 0
mirror of https://github.com/zed-industries/zed.git synced 2025-01-29 05:33:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

updated vector_store to include extra context for semantic search modal

Browse source
This commit is contained in:
KCaverly 2023-07-06 16:33:54 -04:00
parent e57f6f21fe
commit 7d634f66e2
1 changed files with 14 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -418,21 +418,34 @@ impl VectorStore {
) {
let mut item_range = None;
let mut name_range = None;
let mut context_range = None;
for capture in mat.captures {
if capture.index == embedding_config.item_capture_ix {
item_range = Some(capture.node.byte_range());
} else if capture.index == embedding_config.name_capture_ix {
name_range = Some(capture.node.byte_range());
}
if let Some(context_capture_ix) = embedding_config.context_capture_ix {
if capture.index == context_capture_ix {
context_range = Some(capture.node.byte_range());
}
}
}
if let Some((item_range, name_range)) = item_range.zip(name_range) {
let mut context_data = String::new();
if let Some(context_range) = context_range {
if let Some(context) = content.get(context_range.clone()) {
context_data.push_str(context);
}
}
if let Some((item, name)) =
content.get(item_range.clone()).zip(content.get(name_range))
{
context_spans.push(item.to_string());
documents.push(Document {
name: name.to_string(),
name: format!("{} {}", context_data.to_string(), name.to_string()),
offset: item_range.start,
embedding: Vec::new(),
});