brainbank 0.1.3-beta.1 → 0.1.4

package/dist/{chunk-ZEUCCE6P.js → chunk-3UIWA32X.js}

@@ -7,7 +7,6 @@ import {
   isContextFieldPlugin,
   isContextFormatterPlugin,
   isDocsPlugin,
-  isExpandablePlugin,
   isFileResolvable,
   isIndexable,
   isReembeddable,
@@ -51,7 +50,6 @@ function resolveConfig(partial = {}) {
     maxElements: partial.maxElements ?? DEFAULTS.maxElements,
     embeddingProvider: partial.embeddingProvider,
     pruner: partial.pruner,
-    expander: partial.expander,
     webhookPort: partial.webhookPort,
     contextFields: partial.contextFields
   };
@@ -533,7 +531,7 @@ __name(fuseRankedLists, "fuseRankedLists");
 
 // src/lib/prune.ts
 var MAX_PREVIEW_CHARS = 8e3;
-async function pruneResults(query, results, pruner) {
+async function pruneResults(query, results, pruner, context) {
   if (results.length <= 1) return results;
   const items = results.map((r, i) => ({
     id: i,
@@ -541,7 +539,7 @@ async function pruneResults(query, results, pruner) {
     preview: _buildPreview(r.content),
     metadata: r.metadata
   }));
-  const keepIds = await pruner.prune(query, items);
+  const keepIds = await pruner.prune(query, items, context);
   const validIds = new Set(Array.from({ length: results.length }, (_, i) => i));
   return keepIds.filter((id) => validIds.has(id)).map((id) => results[id]);
 }
@@ -568,7 +566,11 @@ function filterByPath(results, prefix) {
   if (!prefix) return results;
   const prefixes = Array.isArray(prefix) ? prefix : [prefix];
   if (prefixes.length === 0) return results;
-  return results.filter((r) => prefixes.some((p) => r.filePath?.startsWith(p)));
+  const normalized = prefixes.map((p) => {
+    if (p.endsWith("/") || p.includes(".")) return p;
+    return p + "/";
+  });
+  return results.filter((r) => normalized.some((p) => r.filePath?.startsWith(p)));
 }
 __name(filterByPath, "filterByPath");
 function filterByIgnore(results, ignorePaths) {
@@ -652,20 +654,18 @@ function dbg(msg) {
 }
 __name(dbg, "dbg");
 var ContextBuilder = class {
-  constructor(_search, _registry, _pruner, _embedding, _configFields = {}, _expander) {
+  constructor(_search, _registry, _pruner, _embedding, _configFields = {}) {
     this._search = _search;
     this._registry = _registry;
     this._pruner = _pruner;
     this._embedding = _embedding;
     this._configFields = _configFields;
-    this._expander = _expander;
   }
   _search;
   _registry;
   _pruner;
   _embedding;
   _configFields;
-  _expander;
   static {
     __name(this, "ContextBuilder");
   }
@@ -673,10 +673,6 @@ var ContextBuilder = class {
   set configFields(fields) {
     this._configFields = fields;
   }
-  /** Set the expander instance. */
-  set expander(expander) {
-    this._expander = expander;
-  }
   /** Build a full context block for a task. Returns markdown for system prompt. */
   async build(task, options = {}) {
     const t0 = Date.now();
@@ -692,10 +688,20 @@ var ContextBuilder = class {
     results = filterByIgnore(results, options.ignorePaths);
     const pruner = options.pruner ?? this._pruner;
     const beforePrune = results;
+    if (pruner && results.length > 40) {
+      const topScore = Math.max(...results.map((r) => r.score));
+      const threshold = topScore * 0.35;
+      const preFiltered = results.filter((r) => r.score >= threshold);
+      if (preFiltered.length < results.length && preFiltered.length >= 3) {
+        dbg(`[pre-filter] Dropped ${results.length - preFiltered.length} low-score results (threshold: ${threshold.toFixed(3)}, top: ${topScore.toFixed(3)})`);
+        results = preFiltered;
+      }
+    }
     if (pruner && results.length > 1) {
       dbg(`[pruner] Running ${_prunerName(pruner)} on ${results.length} results...`);
       const pruneT0 = Date.now();
-      results = await pruneResults(task, results, pruner);
+      const prunerDesc = [options.context, options.prunerContext].filter(Boolean).join("\n\n") || void 0;
+      results = await pruneResults(task, results, pruner, prunerDesc);
       const pruneMs = Date.now() - pruneT0;
       const dropped = beforePrune.filter((r) => !results.includes(r));
       dbg(`[pruner] ${beforePrune.length} \u2192 ${results.length} in ${pruneMs}ms (${dropped.length} dropped)`);
@@ -714,40 +720,18 @@ var ContextBuilder = class {
       results = results.filter((r) => !r.filePath || !options.excludeFiles.has(r.filePath));
     }
     const resolvedFields = this._resolveFields(options);
-    let expanderNote;
-    if (resolvedFields.expander === true && this._expander && results.length > 0) {
-      dbg(`[expander] Running expansion on ${results.length} results...`);
-      const expansion = await this._expand(task, results);
-      dbg(`[expander] Expansion returned ${expansion.results.length} new chunks${expansion.note ? `, note: "${expansion.note}"` : ""}`);
-      if (expansion.results.length > 0) {
-        results = [...results, ...expansion.results];
-      }
-      expanderNote = expansion.note;
-    } else if (resolvedFields.expander === true && !this._expander) {
-      dbg(`[expander] Field enabled but no expander instance! Check config.json "expander" key.`);
-    }
     const parts = [`# Context for: "${task}"
 `];
     this._appendFormatterResults(results, parts, options, resolvedFields);
     await this._appendSearchableResults(task, src, minScore, parts);
-    if (expanderNote) {
-      parts.push(`
-## Expansion Notes
-
-${expanderNote}
-`);
-    }
     const prunedResults = pruner ? beforePrune.filter((r) => !results.includes(r)) : [];
-    const expanderEnabled = resolvedFields.expander === true;
-    const expandedResults = expanderNote !== void 0 ? results.filter((r) => !beforePrune.includes(r) && !prunedResults.includes(r)) : [];
     logQuery({
       source: options.source ?? "api",
       method: "getContext",
       query: task,
       embedding: this._embedding ? providerKey(this._embedding) : "unknown",
       pruner: pruner ? _prunerName(pruner) : null,
-      expander: expanderEnabled ? this._expander ? _expanderName(this._expander) : "configured-no-instance" : null,
-      expandedCount: expandedResults.length > 0 ? expandedResults.length : void 0,
+      expander: null,
       options: {
         sources: src,
         pathPrefix: options.pathPrefix,
@@ -787,44 +771,6 @@ ${expanderNote}
     }
     return { ...defaults, ...this._configFields, ...options.fields ?? {} };
   }
-  /**
-   * Run LLM expansion: build manifest of candidate chunks from files
-   * NOT already in search results, call expander, resolve selected IDs.
-   */
-  async _expand(task, results) {
-    if (!this._expander) return { results: [] };
-    const excludeFilePaths = [...new Set(
-      results.filter((r) => r.filePath).map((r) => r.filePath)
-    )];
-    const excludeIds = [];
-    for (const r of results) {
-      const meta = r.metadata;
-      const id = meta?.id;
-      if (id !== void 0) excludeIds.push(id);
-    }
-    const manifest = [];
-    let resolver;
-    for (const mod of this._registry.all) {
-      if (!isExpandablePlugin(mod)) continue;
-      manifest.push(...mod.buildManifest(excludeFilePaths, excludeIds, excludeFilePaths));
-      if (!resolver) {
-        resolver = /* @__PURE__ */ __name((ids) => mod.resolveChunks(ids), "resolver");
-      }
-    }
-    if (manifest.length === 0 || !resolver) {
-      dbg(`[expander] No manifest items (no ExpandablePlugin or all chunks excluded)`);
-      return { results: [] };
-    }
-    const priorityCount = manifest.filter((m) => m.priority).length;
-    dbg(`[expander] Manifest: ${manifest.length} chunks (${priorityCount} priority, ${manifest.length - priorityCount} other), excluding ${excludeFilePaths.length} files`);
-    try {
-      const expandResult = await this._expander.expand(task, excludeIds, manifest);
-      if (expandResult.ids.length === 0) return { results: [], note: expandResult.note };
-      return { results: resolver(expandResult.ids), note: expandResult.note };
-    } catch {
-      return { results: [] };
-    }
-  }
   /** Collect results from SearchablePlugins that don't have their own formatter. */
   async _appendSearchableResults(task, sources, minScore, parts) {
     for (const mod of this._registry.all) {
@@ -856,10 +802,6 @@ function _prunerName(pruner) {
   return pruner.constructor?.name ?? "custom";
 }
 __name(_prunerName, "_prunerName");
-function _expanderName(expander) {
-  return expander.constructor?.name ?? "custom";
-}
-__name(_expanderName, "_expanderName");
 
 // src/search/keyword/composite-bm25-search.ts
 var DEFAULT_K = 8;
@@ -1598,7 +1540,6 @@ var IGNORE_DIRS = /* @__PURE__ */ new Set([
   ".svelte-kit",
   // Auto-generated code
   "generated",
-  "sdk",
   "openapi",
   // Version control
   ".git",
@@ -2206,7 +2147,7 @@ function createSearchAPI(_db, embedding, config, registry, kvService, sharedHnsw
     embedding
   }) : void 0;
   const bm25 = new CompositeBM25Search(registry);
-  const contextBuilder = new ContextBuilder(search, registry, config.pruner, embedding, config.contextFields ?? {}, config.expander);
+  const contextBuilder = new ContextBuilder(search, registry, config.pruner, embedding, config.contextFields ?? {});
   return new SearchAPI({
     search,
     bm25,
@@ -3191,23 +3132,23 @@ async function setupProviders(brainOpts, config, flags, env) {
   if (perplexityKey) process.env.PERPLEXITY_API_KEY = perplexityKey;
   if (openaiKey) process.env.OPENAI_API_KEY = openaiKey;
   const prunerFlag = flags?.pruner ?? config?.pruner;
-  if (prunerFlag === "haiku") {
-    const { HaikuPruner } = await import("./haiku-pruner-MT4JFKUC.js");
+  if (prunerFlag === "haiku" || prunerFlag === "sonnet") {
+    const { HaikuPruner } = await import("./haiku-pruner-5KVT5AI2.js");
+    const model = prunerFlag === "sonnet" ? "claude-sonnet-4-6" : void 0;
+    brainOpts.pruner = new HaikuPruner({ apiKey: anthropicKey, model });
+  } else if (!prunerFlag && anthropicKey) {
+    const { HaikuPruner } = await import("./haiku-pruner-5KVT5AI2.js");
     brainOpts.pruner = new HaikuPruner({ apiKey: anthropicKey });
   }
-  const expanderFlag = flags?.expander ?? config?.expander;
-  if (expanderFlag === "haiku") {
-    try {
-      const { HaikuExpander } = await import("./haiku-expander-CGEHFJIA.js");
-      brainOpts.expander = new HaikuExpander({ apiKey: anthropicKey });
-    } catch {
-    }
-  }
   const embFlag = flags?.embedding ?? config?.embedding ?? env?.BRAINBANK_EMBEDDING ?? process.env.BRAINBANK_EMBEDDING;
   if (embFlag) {
     const provider = await resolveEmbeddingKey(embFlag);
     brainOpts.embeddingProvider = provider;
     brainOpts.embeddingDims = provider.dims;
+  } else if (perplexityKey) {
+    const provider = await resolveEmbeddingKey("perplexity");
+    brainOpts.embeddingProvider = provider;
+    brainOpts.embeddingDims = provider.dims;
   }
   if (config?.context) {
     brainOpts.contextFields = config.context;
@@ -3397,4 +3338,4 @@ export {
   resetFactoryCache,
   createBrain
 };
-//# sourceMappingURL=chunk-ZEUCCE6P.js.map
+//# sourceMappingURL=chunk-3UIWA32X.js.map