@oomkapwn/enquire-mcp 3.0.1 → 3.1.0

package/dist/tools.js

@@ -2313,10 +2313,29 @@ export async function semanticSearch(vault, args) {
     }
     return { query: args.query, total_docs: docs.length, method: "tfidf-cosine", matches };
 }
+/**
+ * v3.1.0 — pick the text that should be embedded for an embeddings-search
+ * call. HyDE-augmented retrieval prefers the agent-supplied
+ * `hypothetical_answer` (Gao et al 2023); falls back to the raw query
+ * when that's absent / empty / whitespace-only.
+ *
+ * Pure helper so we can unit-test the decision in isolation (the real
+ * `embeddingsSearch` function loads the @huggingface/transformers
+ * embedder, which is out of scope for unit tests).
+ */
+export function pickEmbedTextForHyde(args) {
+    const ha = args.hypothetical_answer?.trim() ?? "";
+    if (ha.length > 0)
+        return { text: ha, usedHyde: true };
+    return { text: args.query, usedHyde: false };
+}
 export async function embeddingsSearch(vault, args, embedFile, hnsw) {
     await vault.ensureExists();
     if (!args.query.trim())
         throw new Error("query must not be empty");
+    // v3.1.0 — pick the actual text to embed. HyDE prefers the
+    // hypothetical answer when present; otherwise fall back to the query.
+    const { text: embedText, usedHyde } = pickEmbedTextForHyde(args);
     const limit = args.limit ?? 10;
     const minScore = args.min_score ?? 0.3;
     // Lazy-load embed-db + embeddings only when the tool is actually called.
@@ -2346,7 +2365,7 @@ export async function embeddingsSearch(vault, args, embedFile, hnsw) {
             return { query: args.query, method: "embeddings-cosine", model: model.alias, total_chunks: 0, matches: [] };
         }
         const embedder = await loadEmbedder(args.model);
-        const [qVec] = await embedder.embed([args.query]);
+        const [qVec] = await embedder.embed([embedText]);
         if (!qVec)
             throw new Error("Embedder returned no vectors for the query");
         // v2.0.0-beta.2 P0 fix: filter excluded paths from the embedding-index
@@ -2388,7 +2407,14 @@ export async function embeddingsSearch(vault, args, embedFile, hnsw) {
             line_end: h.line_end,
             kind: h.kind
         }));
-        return { query: args.query, method: "embeddings-cosine", model: model.alias, total_chunks: total, matches };
+        return {
+            query: args.query,
+            method: "embeddings-cosine",
+            model: model.alias,
+            total_chunks: total,
+            matches,
+            ...(usedHyde ? { hyde: true } : {})
+        };
     }
     finally {
         db.close();