diff --git a/crates/vector_store/src/parsing.rs b/crates/vector_store/src/parsing.rs index a91e87aa222f89dc6e6b04c9b5772a6e85ac18cd..91dcf699f8c3add9088b2af4f0d4df59b7551ac2 100644 --- a/crates/vector_store/src/parsing.rs +++ b/crates/vector_store/src/parsing.rs @@ -1,4 +1,4 @@ -use std::{ops::Range, path::PathBuf, sync::Arc, time::SystemTime}; +use std::{path::PathBuf, sync::Arc, time::SystemTime}; use anyhow::{anyhow, Ok, Result}; use project::Fs; @@ -61,6 +61,8 @@ impl CodeContextRetriever { tree.root_node(), content.as_bytes(), ) { + // log::info!("-----MATCH-----"); + let mut name: Vec<&str> = vec![]; let mut item: Option<&str> = None; let mut offset: Option = None; @@ -89,6 +91,12 @@ impl CodeContextRetriever { .replace("", &pending_file.language.name().to_lowercase()) .replace("", item.unwrap()); + let mut truncated_span = context_span.clone(); + truncated_span.truncate(100); + + // log::info!("Name: {:?}", name); + // log::info!("Span: {:?}", truncated_span); + context_spans.push(context_span); documents.push(Document { name: name.join(" "), diff --git a/crates/vector_store/src/vector_store.rs b/crates/vector_store/src/vector_store.rs index 6f63f07b884d6046e62f9c8a670ea693b062bf5e..a2ca90e84ef5c138f4019fba08c000cd160de450 100644 --- a/crates/vector_store/src/vector_store.rs +++ b/crates/vector_store/src/vector_store.rs @@ -386,7 +386,8 @@ impl VectorStore { let (parsing_files_tx, parsing_files_rx) = channel::unbounded::(); let mut _parsing_files_tasks = Vec::new(); - for _ in 0..cx.background().num_cpus() { + // for _ in 0..cx.background().num_cpus() { + for _ in 0..1 { let fs = fs.clone(); let parsing_files_rx = parsing_files_rx.clone(); let batch_files_tx = batch_files_tx.clone(); diff --git a/crates/zed/src/languages/javascript/embedding.scm b/crates/zed/src/languages/javascript/embedding.scm new file mode 100644 index 0000000000000000000000000000000000000000..ec6eb5ab1a8be481bc7a9987056ce2d1cb7d2474 --- /dev/null +++ b/crates/zed/src/languages/javascript/embedding.scm @@ -0,0 +1,56 @@ +; (internal_module +; "namespace" @context +; name: (_) @name) @item + +(enum_declaration + "enum" @context + name: (_) @name) @item + +(function_declaration + "async"? @context + "function" @context + name: (_) @name) @item + +(interface_declaration + "interface" @context + name: (_) @name) @item + +; (program +; (export_statement +; (lexical_declaration +; ["let" "const"] @context +; (variable_declarator +; name: (_) @name) @item))) + +(program + (lexical_declaration + ["let" "const"] @context + (variable_declarator + name: (_) @name) @item)) + +(class_declaration + "class" @context + name: (_) @name) @item + +(method_definition + [ + "get" + "set" + "async" + "*" + "readonly" + "static" + (override_modifier) + (accessibility_modifier) + ]* @context + name: (_) @name) @item + +; (public_field_definition +; [ +; "declare" +; "readonly" +; "abstract" +; "static" +; (accessibility_modifier) +; ]* @context +; name: (_) @name) @item diff --git a/crates/zed/src/languages/tsx/embedding.scm b/crates/zed/src/languages/tsx/embedding.scm index a8cde61b9e5ce51ad6ea3176b06c3b07dbaa9ee5..96c56abe9f984d05312d96dcb828ff83601d66c9 100644 --- a/crates/zed/src/languages/tsx/embedding.scm +++ b/crates/zed/src/languages/tsx/embedding.scm @@ -1,6 +1,6 @@ ; (internal_module ; "namespace" @context - name: (_) @name) @item + ; name: (_) @name) @item (enum_declaration "enum" @context