Fix qmd embed crash and resolve all TypeScript errors

- Fix ReferenceError in vectorIndex(): firstResult was used but never
  defined. Added code to embed first chunk to get embedding dimensions.

- Fix 87 TypeScript errors across codebase:
  - formatter.ts: Define MultiGetFile type locally (was missing from store.ts)
  - collections.ts: Add non-null assertion for array access
  - mcp.ts: Fix StatusResult type to match store.ts CollectionInfo,
    add list parameter to ResourceTemplate, fix undefined checks
  - qmd.ts: Fix boolean/string type coercions, undefined array access
  - llm.test.ts: Update expandQuery tests for Queryable[] return type,
    fix array access assertions
  - store.test.ts: Add non-null assertions for array access in tests
  - eval-harness.ts: Fix array access assertion

src/collections.ts

@@ -344,7 +344,7 @@ export function findContextForPath(
   // Return most specific match (longest prefix)
   if (matches.length > 0) {
     matches.sort((a, b) => b.prefix.length - a.prefix.length);
-    return matches[0].context;
+    return matches[0]!.context;
   }
 
   // Fallback to global context

src/formatter.ts

@@ -6,14 +6,32 @@
  */
 
 import { extractSnippet } from "./store.js";
-import type { SearchResult, MultiGetFile, MultiGetResult, DocumentResult } from "./store.js";
+import type { SearchResult, MultiGetResult, DocumentResult } from "./store.js";
 
 // =============================================================================
 // Types
 // =============================================================================
 
 // Re-export store types for convenience
-export type { SearchResult, MultiGetFile, MultiGetResult, DocumentResult };
+export type { SearchResult, MultiGetResult, DocumentResult };
+
+// Flattened type for formatter convenience (extracts info from MultiGetResult)
+export type MultiGetFile = {
+  filepath: string;
+  displayPath: string;
+  title: string;
+  body: string;
+  context?: string | null;
+  skipped: false;
+} | {
+  filepath: string;
+  displayPath: string;
+  title: string;
+  body: string;
+  context?: string | null;
+  skipped: true;
+  skipReason: string;
+};
 
 export type OutputFormat = "cli" | "csv" | "md" | "xml" | "files" | "json";
 

src/llm.test.ts

@@ -84,7 +84,7 @@ describe("LlamaCpp Integration", () => {
 
       // Embeddings should be identical for the same input
       for (let i = 0; i < result1!.embedding.length; i++) {
-        expect(result1!.embedding[i]).toBeCloseTo(result2!.embedding[i], 5);
+        expect(result1!.embedding[i]).toBeCloseTo(result2!.embedding[i]!, 5);
       }
     });
 
@@ -100,9 +100,11 @@ describe("LlamaCpp Integration", () => {
       let norm1 = 0;
       let norm2 = 0;
       for (let i = 0; i < result1!.embedding.length; i++) {
-        dotProduct += result1!.embedding[i] * result2!.embedding[i];
-        norm1 += result1!.embedding[i] ** 2;
-        norm2 += result2!.embedding[i] ** 2;
+        const v1 = result1!.embedding[i]!;
+        const v2 = result2!.embedding[i]!;
+        dotProduct += v1 * v2;
+        norm1 += v1 ** 2;
+        norm2 += v2 ** 2;
       }
       const similarity = dotProduct / (Math.sqrt(norm1) * Math.sqrt(norm2));
 
@@ -136,7 +138,7 @@ describe("LlamaCpp Integration", () => {
         expect(batchResults[i]).not.toBeNull();
         expect(individualResults[i]).not.toBeNull();
         for (let j = 0; j < batchResults[i]!.embedding.length; j++) {
-          expect(batchResults[i]!.embedding[j]).toBeCloseTo(individualResults[i]!.embedding[j], 5);
+          expect(batchResults[i]!.embedding[j]).toBeCloseTo(individualResults[i]!.embedding[j]!, 5);
         }
       }
     });
@@ -181,15 +183,15 @@ describe("LlamaCpp Integration", () => {
       expect(result.results).toHaveLength(3);
 
       // The France document should score highest
-      expect(result.results[0].file).toBe("france.txt");
-      expect(result.results[0].score).toBeGreaterThan(0.7);
+      expect(result.results[0]!.file).toBe("france.txt");
+      expect(result.results[0]!.score).toBeGreaterThan(0.7);
 
       // Canada should be somewhat relevant (also about capitals)
-      expect(result.results[1].file).toBe("canada.txt");
+      expect(result.results[1]!.file).toBe("canada.txt");
 
       // Butterflies should score lowest
-      expect(result.results[2].file).toBe("butterflies.txt");
-      expect(result.results[2].score).toBeLessThan(0.6);
+      expect(result.results[2]!.file).toBe("butterflies.txt");
+      expect(result.results[2]!.score).toBeLessThan(0.6);
     });
 
     test("scores authentication query correctly", async () => {
@@ -227,12 +229,12 @@ describe("LlamaCpp Integration", () => {
       const result = await llm.rerank(query, documents);
 
       // JavaScript errors doc should score highest
-      expect(result.results[0].file).toBe("errors.md");
-      expect(result.results[0].score).toBeGreaterThan(0.7);
+      expect(result.results[0]!.file).toBe("errors.md");
+      expect(result.results[0]!.score).toBeGreaterThan(0.7);
 
       // Python doc might be somewhat relevant (same concept, different language)
       // Cooking should be least relevant
-      expect(result.results[2].file).toBe("cooking.md");
+      expect(result.results[2]!.file).toBe("cooking.md");
     });
 
     test("handles empty document list", async () => {
@@ -243,7 +245,7 @@ describe("LlamaCpp Integration", () => {
     test("handles single document", async () => {
       const result = await llm.rerank("test", [{ file: "doc.md", text: "content" }]);
       expect(result.results).toHaveLength(1);
-      expect(result.results[0].file).toBe("doc.md");
+      expect(result.results[0]!.file).toBe("doc.md");
     });
 
     test("preserves original file paths", async () => {
@@ -303,14 +305,17 @@ describe("LlamaCpp Integration", () => {
     test("returns at least the original query", async () => {
       const result = await llm.expandQuery("test query");
 
-      expect(result).toContain("test query");
+      // Result is now Queryable[] - check that at least one has the original query text
+      const texts = result.map(q => q.text);
+      expect(texts).toContain("test query");
       expect(result.length).toBeGreaterThanOrEqual(1);
     }, 30000); // 30s timeout for model loading
 
     test("returns original query first", async () => {
       const result = await llm.expandQuery("authentication setup");
 
-      expect(result[0]).toBe("authentication setup");
+      // First result should have the original query text
+      expect(result[0]?.text).toBe("authentication setup");
     });
   });
 });

src/mcp.ts

@@ -40,7 +40,7 @@ type StatusResult = {
   needsEmbedding: number;
   hasVectorIndex: boolean;
   collections: {
-    id: number;
+    name: string;
     path: string;
     pattern: string;
     documents: number;
@@ -104,7 +104,7 @@ export async function startMcpServer(): Promise<void> {
 
   server.registerResource(
     "document",
-    new ResourceTemplate("qmd://{+path}", {}),
+    new ResourceTemplate("qmd://{+path}", { list: undefined }),
     {
       title: "QMD Document",
       description: "A markdown document from your QMD knowledge base. Use search tools to discover documents.",
@@ -112,11 +112,12 @@ export async function startMcpServer(): Promise<void> {
     },
     async (uri, { path }) => {
       // Decode URL-encoded path (MCP clients send encoded URIs)
-      const decodedPath = decodeURIComponent(path);
+      const pathStr = Array.isArray(path) ? path.join('/') : (path || '');
+      const decodedPath = decodeURIComponent(pathStr);
 
       // Parse virtual path: collection/relative/path
       const parts = decodedPath.split('/');
-      const collection = parts[0];
+      const collection = parts[0] || '';
       const relativePath = parts.slice(1).join('/');
 
       // Find document by collection and path, join with content table
@@ -461,7 +462,7 @@ You can also access documents directly via the \`qmd://\` URI scheme:
       let parsedFromLine = fromLine;
       let lookup = file;
       const colonMatch = lookup.match(/:(\d+)$/);
-      if (colonMatch && parsedFromLine === undefined) {
+      if (colonMatch && colonMatch[1] && parsedFromLine === undefined) {
         parsedFromLine = parseInt(colonMatch[1], 10);
         lookup = lookup.slice(0, -colonMatch[0].length);
       }

src/qmd.ts

@@ -11,7 +11,6 @@ import {
   enableProductionMode,
   searchFTS,
   searchVec,
-  reciprocalRankFusion,
   extractSnippet,
   getContextForFile,
   getContextForPath,
@@ -64,7 +63,7 @@ import {
   createStore,
   getDefaultDbPath,
 } from "./store.js";
-import { getDefaultLlamaCpp, disposeDefaultLlamaCpp, type RerankDocument, type ExpandedQuery } from "./llm.js";
+import { getDefaultLlamaCpp, disposeDefaultLlamaCpp, type RerankDocument, type Queryable, type QueryType } from "./llm.js";
 import type { SearchResult, RankedResult } from "./store.js";
 import {
   formatSearchResults,
@@ -281,7 +280,7 @@ function showStatus(): void {
   try {
     const stat = Bun.file(dbPath).size;
     indexSize = stat;
-  } catch {}
+  } catch { }
 
   // Collections info (from YAML + database stats)
   const collections = listCollections(db);
@@ -350,15 +349,15 @@ function showStatus(): void {
     // Show examples of virtual paths
     console.log(`\n${c.bold}Examples${c.reset}`);
     console.log(`  ${c.dim}# List files in a collection${c.reset}`);
-    if (collections.length > 0) {
+    if (collections.length > 0 && collections[0]) {
       console.log(`  qmd ls ${collections[0].name}`);
     }
     console.log(`  ${c.dim}# Get a document${c.reset}`);
-    if (collections.length > 0) {
+    if (collections.length > 0 && collections[0]) {
       console.log(`  qmd get qmd://${collections[0].name}/path/to/file.md`);
     }
     console.log(`  ${c.dim}# Search within a collection${c.reset}`);
-    if (collections.length > 0) {
+    if (collections.length > 0 && collections[0]) {
       console.log(`  qmd search "query" -c ${collections[0].name}`);
     }
   } else {
@@ -388,6 +387,7 @@ async function updateCollections(): Promise<void> {
 
   for (let i = 0; i < collections.length; i++) {
     const col = collections[i];
+    if (!col) continue;
     console.log(`${c.cyan}[${i + 1}/${collections.length}]${c.reset} ${c.bold}${col.name}${c.reset} ${c.dim}(${col.glob_pattern})${c.reset}`);
 
     // Execute custom update command if specified in YAML
@@ -643,13 +643,14 @@ function contextCheck(): void {
 
   // Check for top-level paths without context within collections that DO have context
   const collectionsWithContext = allCollections.filter(c =>
-    !collectionsWithoutContext.some(cwc => cwc.id === c.id)
+    c && !collectionsWithoutContext.some(cwc => cwc.name === c.name)
   );
 
   let hasPathSuggestions = false;
 
   for (const coll of collectionsWithContext) {
-    const missingPaths = getTopLevelPathsWithoutContext(db, coll.id);
+    if (!coll) continue;
+    const missingPaths = getTopLevelPathsWithoutContext(db, coll.name);
 
     if (missingPaths.length > 0) {
       if (!hasPathSuggestions) {
@@ -681,8 +682,11 @@ function getDocument(filename: string, fromLine?: number, maxLines?: number, lin
   let inputPath = filename;
   const colonMatch = inputPath.match(/:(\d+)$/);
   if (colonMatch && !fromLine) {
-    fromLine = parseInt(colonMatch[1], 10);
-    inputPath = inputPath.slice(0, -colonMatch[0].length);
+    const matched = colonMatch[1];
+    if (matched) {
+      fromLine = parseInt(matched, 10);
+      inputPath = inputPath.slice(0, -colonMatch[0].length);
+    }
   }
 
   let doc: { collectionName: string; path: string; body: string } | null = null;
@@ -727,9 +731,9 @@ function getDocument(filename: string, fromLine?: number, maxLines?: number, lin
         const possiblePath = parts.slice(1).join('/');
 
         // Check if this collection exists
-        const collExists = db.prepare(`
+        const collExists = possibleCollection ? db.prepare(`
           SELECT 1 FROM documents WHERE collection = ? AND active = 1 LIMIT 1
-        `).get(possibleCollection);
+        `).get(possibleCollection) : null;
 
         if (collExists) {
           // Try exact match on collection + path
@@ -738,7 +742,7 @@ function getDocument(filename: string, fromLine?: number, maxLines?: number, lin
             FROM documents d
             JOIN content ON content.hash = d.hash
             WHERE d.collection = ? AND d.path = ? AND d.active = 1
-          `).get(possibleCollection, possiblePath) as typeof doc;
+          `).get(possibleCollection || "", possiblePath || "") as { collectionName: string; path: string; body: string } | null;
 
           if (!doc) {
             // Try fuzzy match by path ending
@@ -748,7 +752,7 @@ function getDocument(filename: string, fromLine?: number, maxLines?: number, lin
               JOIN content ON content.hash = d.hash
               WHERE d.collection = ? AND d.path LIKE ? AND d.active = 1
               LIMIT 1
-            `).get(possibleCollection, `%${possiblePath}`) as typeof doc;
+            `).get(possibleCollection || "", `%${possiblePath}`) as { collectionName: string; path: string; body: string } | null;
           }
 
           if (doc) {
@@ -782,7 +786,7 @@ function getDocument(filename: string, fromLine?: number, maxLines?: number, lin
           FROM documents d
           JOIN content ON content.hash = d.hash
           WHERE d.collection = ? AND d.path = ? AND d.active = 1
-        `).get(detected.collectionName, detected.relativePath) as typeof doc;
+        `).get(detected.collectionName, detected.relativePath) as { collectionName: string; path: string; body: string } | null;
       }
 
       // Fuzzy match by filename (last component of path)
@@ -794,7 +798,7 @@ function getDocument(filename: string, fromLine?: number, maxLines?: number, lin
           JOIN content ON content.hash = d.hash
           WHERE d.path LIKE ? AND d.active = 1
           LIMIT 1
-        `).get(`%${filename}`) as typeof doc;
+        `).get(`%${filename}`) as { collectionName: string; path: string; body: string } | null;
       }
 
       if (doc) {
@@ -805,6 +809,7 @@ function getDocument(filename: string, fromLine?: number, maxLines?: number, lin
     }
   }
 
+  // Ensure doc is not null before proceeding
   if (!doc) {
     console.error(`Document not found: ${filename}`);
     closeDb();
@@ -882,7 +887,7 @@ function multiGet(pattern: string, maxLines?: number, maxBytes: number = DEFAULT
           JOIN content ON content.hash = d.hash
           WHERE d.path = ? AND d.active = 1
           LIMIT 1
-        `).get(name) as typeof doc;
+        `).get(name) as { virtual_path: string; body_length: number; collection: string; path: string } | null;
 
         // Try suffix match
         if (!doc) {
@@ -896,7 +901,7 @@ function multiGet(pattern: string, maxLines?: number, maxBytes: number = DEFAULT
             JOIN content ON content.hash = d.hash
             WHERE d.path LIKE ? AND d.active = 1
             LIMIT 1
-          `).get(`%${name}`) as typeof doc;
+          `).get(`%${name}`) as { virtual_path: string; body_length: number; collection: string; path: string } | null;
         }
       }
 
@@ -1004,7 +1009,7 @@ function multiGet(pattern: string, maxLines?: number, maxBytes: number = DEFAULT
     }));
     console.log(JSON.stringify(output, null, 2));
   } else if (format === "csv") {
-    const escapeField = (val: string | null): string => {
+    const escapeField = (val: string | null | undefined): string => {
       if (val === null || val === undefined) return "";
       const str = String(val);
       if (str.includes(",") || str.includes('"') || str.includes("\n")) {
@@ -1014,7 +1019,7 @@ function multiGet(pattern: string, maxLines?: number, maxBytes: number = DEFAULT
     };
     console.log("file,title,context,skipped,body");
     for (const r of results) {
-      console.log([r.displayPath, r.title, r.context || "", r.skipped ? "true" : "false", r.skipped ? r.skipReason : r.body].map(escapeField).join(","));
+      console.log([r.displayPath, r.title, r.context, r.skipped ? "true" : "false", r.skipped ? r.skipReason : r.body].map(escapeField).join(","));
     }
   } else if (format === "files") {
     for (const r of results) {
@@ -1125,7 +1130,7 @@ function listFiles(pathArg?: string): void {
   } else {
     // Just collection name or collection/path
     const parts = pathArg.split('/');
-    collectionName = parts[0];
+    collectionName = parts[0] || '';
     if (parts.length > 1) {
       pathPrefix = parts.slice(1).join('/');
     }
@@ -1228,7 +1233,7 @@ function collectionList(): void {
   console.log(`${c.bold}Collections (${collections.length}):${c.reset}\n`);
 
   for (const coll of collections) {
-    const updatedAt = new Date(coll.updated_at);
+    const updatedAt = coll.last_modified ? new Date(coll.last_modified) : new Date();
     const timeAgo = formatTimeAgo(updatedAt);
 
     console.log(`${c.cyan}${coll.name}${c.reset} ${c.dim}(qmd://${coll.name}/)${c.reset}`);
@@ -1243,15 +1248,16 @@ function collectionList(): void {
 
 async function collectionAdd(pwd: string, globPattern: string, name?: string): Promise<void> {
   // If name not provided, generate from pwd basename
-  if (!name) {
+  let collName = name;
+  if (!collName) {
     const parts = pwd.split('/').filter(Boolean);
-    name = parts[parts.length - 1] || 'root';
+    collName = parts[parts.length - 1] || 'root';
   }
 
   // Check if collection with this name already exists in YAML
-  const existing = getCollectionFromYaml(name);
+  const existing = getCollectionFromYaml(collName);
   if (existing) {
-    console.error(`${c.yellow}Collection '${name}' already exists.${c.reset}`);
+    console.error(`${c.yellow}Collection '${collName}' already exists.${c.reset}`);
     console.error(`Use a different name with --name <name>`);
     process.exit(1);
   }
@@ -1270,12 +1276,12 @@ async function collectionAdd(pwd: string, globPattern: string, name?: string): P
 
   // Add to YAML config
   const { addCollection } = await import("./collections.js");
-  addCollection(name, pwd, globPattern);
+  addCollection(collName, pwd, globPattern);
 
   // Create the collection and index files
-  console.log(`Creating collection '${name}'...`);
-  await indexFiles(pwd, globPattern, name);
-  console.log(`${c.green}✓${c.reset} Collection '${name}' created successfully`);
+  console.log(`Creating collection '${collName}'...`);
+  await indexFiles(pwd, globPattern, collName);
+  console.log(`${c.green}✓${c.reset} Collection '${collName}' created successfully`);
 }
 
 function collectionRemove(name: string): void {
@@ -1492,11 +1498,11 @@ async function vectorIndex(model: string = DEFAULT_EMBED_MODEL, force: boolean =
       allChunks.push({
         hash: item.hash,
         title,
-        text: chunks[seq].text,
+        text: chunks[seq]!.text, // Chunk is guaranteed to exist by seq loop
         seq,
-        pos: chunks[seq].pos,
-        tokens: chunks[seq].tokens,
-        bytes: encoder.encode(chunks[seq].text).length,
+        pos: chunks[seq]!.pos,
+        tokens: chunks[seq]!.tokens,
+        bytes: encoder.encode(chunks[seq]!.text).length,
         displayName,
       });
     }
@@ -1524,7 +1530,11 @@ async function vectorIndex(model: string = DEFAULT_EMBED_MODEL, force: boolean =
   // Get embedding dimensions from first chunk
   progress.indeterminate();
   const llm = getDefaultLlamaCpp();
-  const firstText = formatDocForEmbedding(allChunks[0].text, allChunks[0].title);
+  const firstChunk = allChunks[0];
+  if (!firstChunk) {
+    throw new Error("No chunks available to embed");
+  }
+  const firstText = formatDocForEmbedding(firstChunk.text, firstChunk.title);
   const firstResult = await llm.embed(firstText);
   if (!firstResult) {
     throw new Error("Failed to get embedding dimensions from first chunk");
@@ -1551,7 +1561,7 @@ async function vectorIndex(model: string = DEFAULT_EMBED_MODEL, force: boolean =
 
       // Insert each embedding
       for (let i = 0; i < batch.length; i++) {
-        const chunk = batch[i];
+        const chunk = batch[i]!;
         const embedding = embeddings[i];
 
         if (embedding) {
@@ -1630,7 +1640,7 @@ function buildFTS5Query(query: string): string {
     .filter(term => term.length >= 2); // Skip single chars and empty
 
   if (terms.length === 0) return "";
-  if (terms.length === 1) return `"${terms[0].replace(/"/g, '""')}"`;
+  if (terms.length === 1) return `"${terms[0]!.replace(/"/g, '""')}"`;
 
   // Strategy: exact phrase OR proximity match OR individual terms
   // Exact phrase matches rank highest, then close proximity, then any term
@@ -1666,7 +1676,6 @@ function normalizeScores(results: SearchResult[]): SearchResult[] {
 // Reciprocal Rank Fusion: combines multiple ranked lists
 // RRF score = sum(1 / (k + rank)) across all lists where doc appears
 // k=60 is standard, provides good balance between top and lower ranks
-export type RankedResult = { file: string; displayPath: string; title: string; body: string; score: number };
 
 function reciprocalRankFusion(
   resultLists: RankedResult[][],
@@ -1677,9 +1686,11 @@ function reciprocalRankFusion(
 
   for (let listIdx = 0; listIdx < resultLists.length; listIdx++) {
     const results = resultLists[listIdx];
+    if (!results) continue;
     const weight = weights[listIdx] ?? 1.0;
     for (let rank = 0; rank < results.length; rank++) {
       const doc = results[rank];
+      if (!doc) continue; // Ensure doc is not undefined
       const rrfScore = weight / (k + rank + 1);
       const existing = scores.get(doc.file);
       if (existing) {
@@ -1711,6 +1722,7 @@ type OutputOptions = {
   all?: boolean;
   collection?: string;  // Filter by collection name (pwd suffix match)
   lineNumbers?: boolean; // Add line numbers to output
+  context?: string;      // Optional context for query expansion
 };
 
 // Highlight query terms in text (skip short words < 3 chars)
@@ -1791,6 +1803,7 @@ function outputResults(results: { file: string; displayPath: string; title: stri
   } else if (opts.format === "cli") {
     for (let i = 0; i < filtered.length; i++) {
       const row = filtered[i];
+      if (!row) continue;
       const { line, snippet } = extractSnippet(row.body, query, 500, row.chunkPos);
       const docid = row.docid || (row.hash ? row.hash.slice(0, 6) : undefined);
 
@@ -1827,7 +1840,9 @@ function outputResults(results: { file: string; displayPath: string; title: stri
       if (i < filtered.length - 1) console.log('\n');
     }
   } else if (opts.format === "md") {
-    for (const row of filtered) {
+    for (let i = 0; i < filtered.length; i++) {
+      const row = filtered[i];
+      if (!row) continue;
       const heading = row.title || row.displayPath;
       const docid = row.docid || (row.hash ? row.hash.slice(0, 6) : undefined);
       let content = opts.full ? row.body : extractSnippet(row.body, query, 500, row.chunkPos).snippet;
@@ -1859,7 +1874,8 @@ function outputResults(results: { file: string; displayPath: string; title: stri
         content = addLineNumbers(content, line);
       }
       const docid = row.docid || (row.hash ? row.hash.slice(0, 6) : "");
-      console.log(`#${docid},${row.score.toFixed(4)},${escapeCSV(toQmdPath(row.displayPath))},${escapeCSV(row.title)},${escapeCSV(row.context || "")},${line},${escapeCSV(content)}`);
+      const snippetText = content || "";
+      console.log(`#${docid},${row.score.toFixed(4)},${escapeCSV(toQmdPath(row.displayPath))},${escapeCSV(row.title || "")},${escapeCSV(row.context || "")},${line},${escapeCSV(snippetText)}`);
     }
   }
 }
@@ -1886,7 +1902,11 @@ function search(query: string, opts: OutputOptions): void {
 
   // Add context to results
   const resultsWithContext = results.map(r => ({
-    ...r,
+    file: r.filepath,
+    displayPath: r.displayPath,
+    title: r.title,
+    body: r.body || "",
+    score: r.score,
     context: getContextForFile(db, r.filepath),
   }));
 
@@ -1925,15 +1945,16 @@ async function vectorSearch(query: string, opts: OutputOptions, model: string =
   checkIndexHealth(db);
 
   // Expand query using structured output (no lexical for vector-only search)
-  const expanded = await expandQueryStructured(query, false);
+  const queryables = await expandQueryStructured(query, false, opts.context);
 
-  // Build list of queries for vector search: original, vectorQuery, and hyde
+  // Build list of queries for vector search: original, vec, and hyde
   const vectorQueries: string[] = [query];
-  if (expanded.vectorQuery && expanded.vectorQuery !== query) {
-    vectorQueries.push(expanded.vectorQuery);
-  }
-  if (expanded.hyde && expanded.hyde.length > 20) {
-    vectorQueries.push(expanded.hyde);
+  for (const q of queryables) {
+    if (q.type === 'vec' || q.type === 'hyde') {
+      if (q.text && q.text !== query) {
+        vectorQueries.push(q.text);
+      }
+    }
   }
 
   process.stderr.write(`${c.dim}Searching ${vectorQueries.length} vector queries...${c.reset}\n`);
@@ -1942,7 +1963,8 @@ async function vectorSearch(query: string, opts: OutputOptions, model: string =
   const perQueryLimit = opts.all ? 500 : 20;
   const allResults = new Map<string, { file: string; displayPath: string; title: string; body: string; score: number; hash: string }>();
 
-  for (const q of vectorQueries) {
+  // Use Promise.all for concurrent vector searches
+  await Promise.all(vectorQueries.map(async (q) => {
     const vecResults = await searchVec(db, q, model, perQueryLimit, collectionName as any);
     for (const r of vecResults) {
       const existing = allResults.get(r.filepath);
@@ -1950,7 +1972,7 @@ async function vectorSearch(query: string, opts: OutputOptions, model: string =
         allResults.set(r.filepath, { file: r.filepath, displayPath: r.displayPath, title: r.title, body: r.body || "", score: r.score, hash: r.hash });
       }
     }
-  }
+  }));
 
   // Sort by max score and limit to requested count
   const results = Array.from(allResults.values())
@@ -1967,50 +1989,50 @@ async function vectorSearch(query: string, opts: OutputOptions, model: string =
   outputResults(results, query, { ...opts, limit: results.length }); // Already limited
 }
 
-// Expand query using structured output with JSON schema grammar
-async function expandQueryStructured(query: string, includeLexical: boolean = true): Promise<ExpandedQuery> {
+// Expand query using structured output with GBNF grammar
+async function expandQueryStructured(query: string, includeLexical: boolean = true, context?: string): Promise<Queryable[]> {
   process.stderr.write(`${c.dim}Expanding query...${c.reset}\n`);
 
   const llm = getDefaultLlamaCpp();
-  const expanded = await llm.expandQueryStructured(query, includeLexical);
+  const queryables = await llm.expandQuery(query, { includeLexical, context });
 
-  // Log the expansion as a tree, starting with original query
+  // Log the expansion as a tree
   const lines: string[] = [];
   const bothLabel = includeLexical ? ' · (lexical+vector)' : ' · (vector)';
   lines.push(`${c.dim}├─ ${query}${bothLabel}${c.reset}`);
 
-  if (expanded.lexicalQuery && expanded.lexicalQuery !== query) {
-    lines.push(`${c.dim}├─ ${expanded.lexicalQuery} · (lexical)${c.reset}`);
-  }
-  if (expanded.vectorQuery && expanded.vectorQuery !== query) {
-    lines.push(`${c.dim}├─ ${expanded.vectorQuery} · (vector)${c.reset}`);
-  }
-  if (expanded.hyde && expanded.hyde.length > 20) {
-    // Truncate hyde to first ~60 chars for display
-    const hydePreview = expanded.hyde.length > 60
-      ? expanded.hyde.substring(0, 60).replace(/\n/g, ' ') + '...'
-      : expanded.hyde.replace(/\n/g, ' ');
-    lines.push(`${c.dim}├─ ${hydePreview} · (vector)${c.reset}`);
+  for (let i = 0; i < queryables.length; i++) {
+    const q = queryables[i];
+    if (!q || q.text === query) continue;
+
+    let textPreview = q.text.replace(/\n/g, ' ');
+    if (textPreview.length > 80) {
+      textPreview = textPreview.substring(0, 77) + '...';
+    }
+
+    const label = q.type === 'lex' ? 'lexical' : (q.type === 'hyde' ? 'hyde' : 'vector');
+    lines.push(`${c.dim}├─ ${textPreview} · (${label})${c.reset}`);
   }
 
   // Fix last item to use └─ instead of ├─
   if (lines.length > 0) {
-    lines[lines.length - 1] = lines[lines.length - 1].replace('├─', '└─');
+    lines[lines.length - 1] = lines[lines.length - 1]!.replace('├─', '└─');
   }
 
   for (const line of lines) {
     process.stderr.write(line + '\n');
   }
 
-  return expanded;
+  return queryables;
 }
 
 async function expandQuery(query: string, _model: string = DEFAULT_QUERY_MODEL, _db?: Database): Promise<string[]> {
-  const expanded = await expandQueryStructured(query, true);
-  const queries = [query];
-  if (expanded.lexicalQuery && expanded.lexicalQuery !== query) queries.push(expanded.lexicalQuery);
-  if (expanded.vectorQuery && expanded.vectorQuery !== query) queries.push(expanded.vectorQuery);
-  return queries;
+  const queryables = await expandQueryStructured(query, true);
+  const queries = new Set<string>([query]);
+  for (const q of queryables) {
+    queries.add(q.text);
+  }
+  return Array.from(queries);
 }
 
 async function querySearch(query: string, opts: OutputOptions, embedModel: string = DEFAULT_EMBED_MODEL, rerankModel: string = DEFAULT_RERANK_MODEL): Promise<void> {
@@ -2033,7 +2055,7 @@ async function querySearch(query: string, opts: OutputOptions, embedModel: strin
 
   // Run initial BM25 search (will be reused for retrieval)
   const initialFts = searchFTS(db, query, 20, collectionName as any);
-  const hasVectors = !!db.prepare(`SELECT name FROM sqlite_master WHERE type='table' AND name='vectors_vec'`).get();
+  let hasVectors = !!db.prepare(`SELECT name FROM sqlite_master WHERE type='table' AND name='vectors_vec'`).get();
 
   // Check if initial results have strong signals (skip expansion if so)
   // Strong signal = top result is strong AND clearly separated from runner-up.
@@ -2052,21 +2074,19 @@ async function querySearch(query: string, opts: OutputOptions, embedModel: strin
     {
       const lines: string[] = [];
       lines.push(`${c.dim}├─ ${query} · (lexical+vector)${c.reset}`);
-      lines[lines.length - 1] = lines[lines.length - 1].replace('├─', '└─');
+      lines[lines.length - 1] = lines[lines.length - 1]!.replace('├─', '└─');
       for (const line of lines) process.stderr.write(line + '\n');
     }
   } else {
     // Weak signal - expand query for better recall
-    const expanded = await expandQueryStructured(query, true);
+    const queryables = await expandQueryStructured(query, true, opts.context);
 
-    if (expanded.lexicalQuery && expanded.lexicalQuery !== query) {
-      ftsQueries.push(expanded.lexicalQuery);
-    }
-    if (expanded.vectorQuery && expanded.vectorQuery !== query) {
-      vectorQueries.push(expanded.vectorQuery);
-    }
-    if (expanded.hyde && expanded.hyde.length > 20) {
-      vectorQueries.push(expanded.hyde);
+    for (const q of queryables) {
+      if (q.type === 'lex') {
+        if (q.text && q.text !== query) ftsQueries.push(q.text);
+      } else if (q.type === 'vec' || q.type === 'hyde') {
+        if (q.text && q.text !== query) vectorQueries.push(q.text);
+      }
     }
   }
 
@@ -2078,31 +2098,40 @@ async function querySearch(query: string, opts: OutputOptions, embedModel: strin
   // Map to store hash by filepath for final results
   const hashMap = new Map<string, string>();
 
-  // FTS searches with lexical queries (reuse initial search for original query)
-  if (initialFts.length > 0) {
-    for (const r of initialFts) hashMap.set(r.filepath, r.hash);
-    rankedLists.push(initialFts.map(r => ({ file: r.filepath, displayPath: r.displayPath, title: r.title, body: r.body || "", score: r.score })));
-  }
-  // Run expanded queries (skip first which is original)
-  for (const q of ftsQueries.slice(1)) {
-    const ftsResults = searchFTS(db, q, 20, collectionName as any);
-    if (ftsResults.length > 0) {
-      for (const r of ftsResults) hashMap.set(r.filepath, r.hash);
-      rankedLists.push(ftsResults.map(r => ({ file: r.filepath, displayPath: r.displayPath, title: r.title, body: r.body || "", score: r.score })));
-    }
+  // Run all searches concurrently (FTS + Vector)
+  const searchPromises: Promise<void>[] = [];
+
+  // FTS searches
+  for (const q of ftsQueries) {
+    if (!q) continue;
+    searchPromises.push((async () => {
+      const ftsResults = searchFTS(db, q, 20, (collectionName || "") as any);
+      if (ftsResults.length > 0) {
+        for (const r of ftsResults) {
+          // Mutex for hashMap is not strictly needed as it's just adding values
+          hashMap.set(r.filepath, r.hash);
+        }
+        rankedLists.push(ftsResults.map(r => ({ file: r.filepath, displayPath: r.displayPath, title: r.title, body: r.body || "", score: r.score })));
+      }
+    })());
   }
 
-  // Vector searches with semantic queries + hyde
+  // Vector searches
   if (hasVectors) {
     for (const q of vectorQueries) {
-      const vecResults = await searchVec(db, q, embedModel, 20, collectionName as any);
-      if (vecResults.length > 0) {
-        for (const r of vecResults) hashMap.set(r.filepath, r.hash);
-        rankedLists.push(vecResults.map(r => ({ file: r.filepath, displayPath: r.displayPath, title: r.title, body: r.body || "", score: r.score })));
-      }
+      if (!q) continue;
+      searchPromises.push((async () => {
+        const vecResults = await searchVec(db, q, embedModel, 20, (collectionName || "") as any);
+        if (vecResults.length > 0) {
+          for (const r of vecResults) hashMap.set(r.filepath, r.hash);
+          rankedLists.push(vecResults.map(r => ({ file: r.filepath, displayPath: r.displayPath, title: r.title, body: r.body || "", score: r.score })));
+        }
+      })());
     }
   }
 
+  await Promise.all(searchPromises);
+
   // Apply Reciprocal Rank Fusion to combine all ranked lists
   // Give 2x weight to original query results (first 2 lists: FTS + vector)
   const weights = rankedLists.map((_, i) => i < 2 ? 2.0 : 1.0);
@@ -2215,7 +2244,12 @@ function parseCLI() {
     args: Bun.argv.slice(2), // Skip bun and script path
     options: {
       // Global options
-      index: { type: "string" },
+      context: {
+        type: "string",
+      },
+      "no-lex": {
+        type: "boolean",
+      },
       help: { type: "boolean", short: "h" },
       // Search options
       n: { type: "string" },
@@ -2246,8 +2280,9 @@ function parseCLI() {
   });
 
   // Select index name (default: "index")
-  if (values.index) {
-    setIndexName(values.index);
+  const indexName = values.index as string | undefined;
+  if (indexName) {
+    setIndexName(indexName);
   }
 
   // Determine output format
@@ -2261,16 +2296,16 @@ function parseCLI() {
   // Default limit: 20 for --files/--json, 5 otherwise
   // --all means return all results (use very large limit)
   const defaultLimit = (format === "files" || format === "json") ? 20 : 5;
-  const isAll = values.all || false;
+  const isAll = !!values.all;
 
   const opts: OutputOptions = {
     format,
-    full: values.full || false,
-    limit: isAll ? 100000 : (values.n ? parseInt(values.n, 10) || defaultLimit : defaultLimit),
-    minScore: values["min-score"] ? parseFloat(values["min-score"]) || 0 : 0,
+    full: !!values.full,
+    limit: isAll ? 100000 : (values.n ? parseInt(String(values.n), 10) || defaultLimit : defaultLimit),
+    minScore: values["min-score"] ? parseFloat(String(values["min-score"])) || 0 : 0,
     all: isAll,
     collection: values.collection as string | undefined,
-    lineNumbers: values["line-numbers"] || false,
+    lineNumbers: !!values["line-numbers"],
   };
 
   return {
@@ -2334,255 +2369,255 @@ function showHelp(): void {
 
 // Main CLI - only run if this is the main module
 if (import.meta.main) {
-const cli = parseCLI();
-
-if (!cli.command || cli.values.help) {
-  showHelp();
-  process.exit(cli.values.help ? 0 : 1);
-}
+  const cli = parseCLI();
+
+  if (!cli.command || cli.values.help) {
+    showHelp();
+    process.exit(cli.values.help ? 0 : 1);
+  }
+
+  switch (cli.command) {
+    case "context": {
+      const subcommand = cli.args[0];
+      if (!subcommand) {
+        console.error("Usage: qmd context <add|list|check|rm>");
+        console.error("");
+        console.error("Commands:");
+        console.error("  qmd context add [path] \"text\"  - Add context (defaults to current dir)");
+        console.error("  qmd context add / \"text\"       - Add global context to all collections");
+        console.error("  qmd context list                - List all contexts");
+        console.error("  qmd context check               - Check for missing contexts");
+        console.error("  qmd context rm <path>           - Remove context");
+        process.exit(1);
+      }
 
-switch (cli.command) {
-  case "context": {
-    const subcommand = cli.args[0];
-    if (!subcommand) {
-      console.error("Usage: qmd context <add|list|check|rm>");
-      console.error("");
-      console.error("Commands:");
-      console.error("  qmd context add [path] \"text\"  - Add context (defaults to current dir)");
-      console.error("  qmd context add / \"text\"       - Add global context to all collections");
-      console.error("  qmd context list                - List all contexts");
-      console.error("  qmd context check               - Check for missing contexts");
-      console.error("  qmd context rm <path>           - Remove context");
-      process.exit(1);
-    }
+      switch (subcommand) {
+        case "add": {
+          if (cli.args.length < 2) {
+            console.error("Usage: qmd context add [path] \"text\"");
+            console.error("");
+            console.error("Examples:");
+            console.error("  qmd context add \"Context for current directory\"");
+            console.error("  qmd context add . \"Context for current directory\"");
+            console.error("  qmd context add /subfolder \"Context for subfolder\"");
+            console.error("  qmd context add / \"Global context for all collections\"");
+            console.error("");
+            console.error("  Using virtual paths:");
+            console.error("  qmd context add qmd://journals/ \"Context for entire journals collection\"");
+            console.error("  qmd context add qmd://journals/2024 \"Context for 2024 journals\"");
+            process.exit(1);
+          }
 
-    switch (subcommand) {
-      case "add": {
-        if (cli.args.length < 2) {
-          console.error("Usage: qmd context add [path] \"text\"");
-          console.error("");
-          console.error("Examples:");
-          console.error("  qmd context add \"Context for current directory\"");
-          console.error("  qmd context add . \"Context for current directory\"");
-          console.error("  qmd context add /subfolder \"Context for subfolder\"");
-          console.error("  qmd context add / \"Global context for all collections\"");
-          console.error("");
-          console.error("  Using virtual paths:");
-          console.error("  qmd context add qmd://journals/ \"Context for entire journals collection\"");
-          console.error("  qmd context add qmd://journals/2024 \"Context for 2024 journals\"");
-          process.exit(1);
-        }
+          let pathArg: string | undefined;
+          let contextText: string;
 
-        let pathArg: string | undefined;
-        let contextText: string;
+          // Check if first arg looks like a path or if it's the context text
+          const firstArg = cli.args[1] || '';
+          const secondArg = cli.args[2];
 
-        // Check if first arg looks like a path or if it's the context text
-        const firstArg = cli.args[1];
-        const secondArg = cli.args[2];
+          if (secondArg) {
+            // Two args: path + context
+            pathArg = firstArg;
+            contextText = cli.args.slice(2).join(" ");
+          } else {
+            // One arg: context only (use current directory)
+            pathArg = undefined;
+            contextText = firstArg;
+          }
 
-        if (secondArg) {
-          // Two args: path + context
-          pathArg = firstArg;
-          contextText = cli.args.slice(2).join(" ");
-        } else {
-          // One arg: context only (use current directory)
-          pathArg = undefined;
-          contextText = firstArg;
+          await contextAdd(pathArg, contextText);
+          break;
         }
 
-        await contextAdd(pathArg, contextText);
-        break;
-      }
+        case "list": {
+          contextList();
+          break;
+        }
 
-      case "list": {
-        contextList();
-        break;
-      }
+        case "check": {
+          contextCheck();
+          break;
+        }
 
-      case "check": {
-        contextCheck();
-        break;
-      }
+        case "rm":
+        case "remove": {
+          if (cli.args.length < 2 || !cli.args[1]) {
+            console.error("Usage: qmd context rm <path>");
+            console.error("Examples:");
+            console.error("  qmd context rm /");
+            console.error("  qmd context rm qmd://journals/2024");
+            process.exit(1);
+          }
+          contextRemove(cli.args[1]);
+          break;
+        }
 
-      case "rm":
-      case "remove": {
-        if (cli.args.length < 2) {
-          console.error("Usage: qmd context rm <path>");
-          console.error("Examples:");
-          console.error("  qmd context rm /");
-          console.error("  qmd context rm qmd://journals/2024");
+        default:
+          console.error(`Unknown subcommand: ${subcommand}`);
+          console.error("Available: add, list, check, rm");
           process.exit(1);
-        }
-        contextRemove(cli.args[1]);
-        break;
       }
+      break;
+    }
 
-      default:
-        console.error(`Unknown subcommand: ${subcommand}`);
-        console.error("Available: add, list, check, rm");
+    case "get": {
+      if (!cli.args[0]) {
+        console.error("Usage: qmd get <filepath>[:line] [--from <line>] [-l <lines>] [--line-numbers]");
         process.exit(1);
+      }
+      const fromLine = cli.values.from ? parseInt(cli.values.from as string, 10) : undefined;
+      const maxLines = cli.values.l ? parseInt(cli.values.l as string, 10) : undefined;
+      getDocument(cli.args[0], fromLine, maxLines, cli.opts.lineNumbers);
+      break;
     }
-    break;
-  }
 
-  case "get": {
-    if (!cli.args[0]) {
-      console.error("Usage: qmd get <filepath>[:line] [--from <line>] [-l <lines>] [--line-numbers]");
-      process.exit(1);
+    case "multi-get": {
+      if (!cli.args[0]) {
+        console.error("Usage: qmd multi-get <pattern> [-l <lines>] [--max-bytes <bytes>] [--json|--csv|--md|--xml|--files]");
+        console.error("  pattern: glob (e.g., 'journals/2025-05*.md') or comma-separated list");
+        process.exit(1);
+      }
+      const maxLinesMulti = cli.values.l ? parseInt(cli.values.l as string, 10) : undefined;
+      const maxBytes = cli.values["max-bytes"] ? parseInt(cli.values["max-bytes"] as string, 10) : DEFAULT_MULTI_GET_MAX_BYTES;
+      multiGet(cli.args[0], maxLinesMulti, maxBytes, cli.opts.format);
+      break;
     }
-    const fromLine = cli.values.from ? parseInt(cli.values.from as string, 10) : undefined;
-    const maxLines = cli.values.l ? parseInt(cli.values.l as string, 10) : undefined;
-    getDocument(cli.args[0], fromLine, maxLines, cli.opts.lineNumbers);
-    break;
-  }
 
-  case "multi-get": {
-    if (!cli.args[0]) {
-      console.error("Usage: qmd multi-get <pattern> [-l <lines>] [--max-bytes <bytes>] [--json|--csv|--md|--xml|--files]");
-      console.error("  pattern: glob (e.g., 'journals/2025-05*.md') or comma-separated list");
-      process.exit(1);
+    case "ls": {
+      listFiles(cli.args[0]);
+      break;
     }
-    const maxLinesMulti = cli.values.l ? parseInt(cli.values.l as string, 10) : undefined;
-    const maxBytes = cli.values["max-bytes"] ? parseInt(cli.values["max-bytes"] as string, 10) : DEFAULT_MULTI_GET_MAX_BYTES;
-    multiGet(cli.args[0], maxLinesMulti, maxBytes, cli.opts.format);
-    break;
-  }
 
-  case "ls": {
-    listFiles(cli.args[0]);
-    break;
-  }
+    case "collection": {
+      const subcommand = cli.args[0];
+      switch (subcommand) {
+        case "list": {
+          collectionList();
+          break;
+        }
 
-  case "collection": {
-    const subcommand = cli.args[0];
-    switch (subcommand) {
-      case "list": {
-        collectionList();
-        break;
-      }
+        case "add": {
+          const pwd = cli.args[1] || getPwd();
+          const resolvedPwd = pwd === '.' ? getPwd() : getRealPath(resolve(pwd));
+          const globPattern = cli.values.mask as string || DEFAULT_GLOB;
+          const name = cli.values.name as string | undefined;
 
-      case "add": {
-        const pwd = cli.args[1] || getPwd();
-        const resolvedPwd = pwd === '.' ? getPwd() : getRealPath(resolve(pwd));
-        const globPattern = cli.values.mask as string || DEFAULT_GLOB;
-        const name = cli.values.name as string | undefined;
+          await collectionAdd(resolvedPwd, globPattern, name);
+          break;
+        }
 
-        await collectionAdd(resolvedPwd, globPattern, name);
-        break;
-      }
+        case "remove":
+        case "rm": {
+          if (!cli.args[1]) {
+            console.error("Usage: qmd collection remove <name>");
+            console.error("  Use 'qmd collection list' to see available collections");
+            process.exit(1);
+          }
+          collectionRemove(cli.args[1]);
+          break;
+        }
 
-      case "remove":
-      case "rm": {
-        if (!cli.args[1]) {
-          console.error("Usage: qmd collection remove <name>");
-          console.error("  Use 'qmd collection list' to see available collections");
-          process.exit(1);
+        case "rename":
+        case "mv": {
+          if (!cli.args[1] || !cli.args[2]) {
+            console.error("Usage: qmd collection rename <old-name> <new-name>");
+            console.error("  Use 'qmd collection list' to see available collections");
+            process.exit(1);
+          }
+          collectionRename(cli.args[1], cli.args[2]);
+          break;
         }
-        collectionRemove(cli.args[1]);
-        break;
-      }
 
-      case "rename":
-      case "mv": {
-        if (!cli.args[1] || !cli.args[2]) {
-          console.error("Usage: qmd collection rename <old-name> <new-name>");
-          console.error("  Use 'qmd collection list' to see available collections");
+        default:
+          console.error(`Unknown subcommand: ${subcommand}`);
+          console.error("Available: list, add, remove, rename");
           process.exit(1);
-        }
-        collectionRename(cli.args[1], cli.args[2]);
-        break;
       }
-
-      default:
-        console.error(`Unknown subcommand: ${subcommand}`);
-        console.error("Available: list, add, remove, rename");
-        process.exit(1);
+      break;
     }
-    break;
-  }
 
-  case "status":
-    showStatus();
-    break;
+    case "status":
+      showStatus();
+      break;
 
-  case "update":
-    await updateCollections();
-    break;
+    case "update":
+      await updateCollections();
+      break;
 
-  case "embed":
-    await vectorIndex(DEFAULT_EMBED_MODEL, cli.values.force || false);
-    break;
+    case "embed":
+      await vectorIndex(DEFAULT_EMBED_MODEL, !!cli.values.force);
+      break;
 
-  case "search":
-    if (!cli.query) {
-      console.error("Usage: qmd search [options] <query>");
-      process.exit(1);
-    }
-    search(cli.query, cli.opts);
-    break;
+    case "search":
+      if (!cli.query) {
+        console.error("Usage: qmd search [options] <query>");
+        process.exit(1);
+      }
+      search(cli.query, cli.opts);
+      break;
 
-  case "vsearch":
-    if (!cli.query) {
-      console.error("Usage: qmd vsearch [options] <query>");
-      process.exit(1);
-    }
-    // Default min-score for vector search is 0.3
-    if (!cli.values["min-score"]) {
-      cli.opts.minScore = 0.3;
-    }
-    await vectorSearch(cli.query, cli.opts);
-    break;
+    case "vsearch":
+      if (!cli.query) {
+        console.error("Usage: qmd vsearch [options] <query>");
+        process.exit(1);
+      }
+      // Default min-score for vector search is 0.3
+      if (!cli.values["min-score"]) {
+        cli.opts.minScore = 0.3;
+      }
+      await vectorSearch(cli.query, cli.opts);
+      break;
 
-  case "query":
-    if (!cli.query) {
-      console.error("Usage: qmd query [options] <query>");
-      process.exit(1);
+    case "query":
+      if (!cli.query) {
+        console.error("Usage: qmd query [options] <query>");
+        process.exit(1);
+      }
+      await querySearch(cli.query, cli.opts);
+      break;
+
+    case "mcp": {
+      const { startMcpServer } = await import("./mcp.js");
+      await startMcpServer();
+      break;
     }
-    await querySearch(cli.query, cli.opts);
-    break;
 
-  case "mcp": {
-    const { startMcpServer } = await import("./mcp.js");
-    await startMcpServer();
-    break;
-  }
+    case "cleanup": {
+      const db = getDb();
+
+      // 1. Clear llm_cache
+      const cacheCount = deleteLLMCache(db);
+      console.log(`${c.green}✓${c.reset} Cleared ${cacheCount} cached API responses`);
 
-  case "cleanup": {
-    const db = getDb();
+      // 2. Remove orphaned vectors
+      const orphanedVecs = cleanupOrphanedVectors(db);
+      if (orphanedVecs > 0) {
+        console.log(`${c.green}✓${c.reset} Removed ${orphanedVecs} orphaned embedding chunks`);
+      } else {
+        console.log(`${c.dim}No orphaned embeddings to remove${c.reset}`);
+      }
 
-    // 1. Clear llm_cache
-    const cacheCount = deleteLLMCache(db);
-    console.log(`${c.green}✓${c.reset} Cleared ${cacheCount} cached API responses`);
+      // 3. Remove inactive documents
+      const inactiveDocs = deleteInactiveDocuments(db);
+      if (inactiveDocs > 0) {
+        console.log(`${c.green}✓${c.reset} Removed ${inactiveDocs} inactive document records`);
+      }
 
-    // 2. Remove orphaned vectors
-    const orphanedVecs = cleanupOrphanedVectors(db);
-    if (orphanedVecs > 0) {
-      console.log(`${c.green}✓${c.reset} Removed ${orphanedVecs} orphaned embedding chunks`);
-    } else {
-      console.log(`${c.dim}No orphaned embeddings to remove${c.reset}`);
-    }
+      // 4. Vacuum to reclaim space
+      vacuumDatabase(db);
+      console.log(`${c.green}✓${c.reset} Database vacuumed`);
 
-    // 3. Remove inactive documents
-    const inactiveDocs = deleteInactiveDocuments(db);
-    if (inactiveDocs > 0) {
-      console.log(`${c.green}✓${c.reset} Removed ${inactiveDocs} inactive document records`);
+      closeDb();
+      break;
     }
 
-    // 4. Vacuum to reclaim space
-    vacuumDatabase(db);
-    console.log(`${c.green}✓${c.reset} Database vacuumed`);
-
-    closeDb();
-    break;
+    default:
+      console.error(`Unknown command: ${cli.command}`);
+      console.error("Run 'qmd --help' for usage.");
+      process.exit(1);
   }
 
-  default:
-    console.error(`Unknown command: ${cli.command}`);
-    console.error("Run 'qmd --help' for usage.");
-    process.exit(1);
-}
-
-// Cleanup LlamaCpp instance to prevent NAPI crash on exit
-await disposeDefaultLlamaCpp();
+  // Cleanup LlamaCpp instance to prevent NAPI crash on exit
+  await disposeDefaultLlamaCpp();
 
 } // end if (import.meta.main)

src/store.test.ts

@@ -528,8 +528,8 @@ describe("Document Chunking", () => {
     const content = "Small document content";
     const chunks = chunkDocument(content, 1000, 0);
     expect(chunks).toHaveLength(1);
-    expect(chunks[0].text).toBe(content);
-    expect(chunks[0].pos).toBe(0);
+    expect(chunks[0]!.text).toBe(content);
+    expect(chunks[0]!.pos).toBe(0);
   });
 
   test("chunkDocument splits large documents", () => {
@@ -539,9 +539,9 @@ describe("Document Chunking", () => {
 
     // All chunks should have correct positions
     for (let i = 0; i < chunks.length; i++) {
-      expect(chunks[i].pos).toBeGreaterThanOrEqual(0);
+      expect(chunks[i]!.pos).toBeGreaterThanOrEqual(0);
       if (i > 0) {
-        expect(chunks[i].pos).toBeGreaterThan(chunks[i - 1].pos);
+        expect(chunks[i]!.pos).toBeGreaterThan(chunks[i - 1]!.pos);
       }
     }
   });
@@ -554,12 +554,12 @@ describe("Document Chunking", () => {
     // With overlap, positions should be closer together than without
     // Each new chunk starts 150 chars before where the previous one ended
     for (let i = 1; i < chunks.length; i++) {
-      const prevEnd = chunks[i - 1].pos + chunks[i - 1].text.length;
-      const currentStart = chunks[i].pos;
+      const prevEnd = chunks[i - 1]!.pos + chunks[i - 1]!.text.length;
+      const currentStart = chunks[i]!.pos;
       // Current chunk should start before the previous chunk ended (overlap)
       expect(currentStart).toBeLessThan(prevEnd);
       // But should still make forward progress
-      expect(currentStart).toBeGreaterThan(chunks[i - 1].pos);
+      expect(currentStart).toBeGreaterThan(chunks[i - 1]!.pos);
     }
   });
 
@@ -594,8 +594,8 @@ describe("Document Chunking", () => {
     const chunks = chunkDocument(content);
     expect(chunks.length).toBeGreaterThan(1);
     // Each chunk should be around 3200 chars (except last)
-    expect(chunks[0].text.length).toBeGreaterThan(2500);
-    expect(chunks[0].text.length).toBeLessThanOrEqual(3200);
+    expect(chunks[0]!.text.length).toBeGreaterThan(2500);
+    expect(chunks[0]!.text.length).toBeLessThanOrEqual(3200);
   });
 });
 
@@ -604,10 +604,10 @@ describe("Token-based Chunking", () => {
     const content = "This is a small document.";
     const chunks = await chunkDocumentByTokens(content, 800, 120);
     expect(chunks).toHaveLength(1);
-    expect(chunks[0].text).toBe(content);
-    expect(chunks[0].pos).toBe(0);
-    expect(chunks[0].tokens).toBeGreaterThan(0);
-    expect(chunks[0].tokens).toBeLessThan(800);
+    expect(chunks[0]!.text).toBe(content);
+    expect(chunks[0]!.pos).toBe(0);
+    expect(chunks[0]!.tokens).toBeGreaterThan(0);
+    expect(chunks[0]!.tokens).toBeLessThan(800);
   });
 
   test("chunkDocumentByTokens splits large documents", async () => {
@@ -625,9 +625,9 @@ describe("Token-based Chunking", () => {
 
     // Chunks should have correct positions
     for (let i = 0; i < chunks.length; i++) {
-      expect(chunks[i].pos).toBeGreaterThanOrEqual(0);
+      expect(chunks[i]!.pos).toBeGreaterThanOrEqual(0);
       if (i > 0) {
-        expect(chunks[i].pos).toBeGreaterThan(chunks[i - 1].pos);
+        expect(chunks[i]!.pos).toBeGreaterThan(chunks[i - 1]!.pos);
       }
     }
   });
@@ -640,8 +640,8 @@ describe("Token-based Chunking", () => {
 
     // With overlap, consecutive chunks should have overlapping positions
     for (let i = 1; i < chunks.length; i++) {
-      const prevEnd = chunks[i - 1].pos + chunks[i - 1].text.length;
-      const currentStart = chunks[i].pos;
+      const prevEnd = chunks[i - 1]!.pos + chunks[i - 1]!.text.length;
+      const currentStart = chunks[i]!.pos;
       // Current chunk should start before the previous chunk ended (overlap)
       expect(currentStart).toBeLessThan(prevEnd);
     }
@@ -653,8 +653,8 @@ describe("Token-based Chunking", () => {
 
     expect(chunks).toHaveLength(1);
     // The token count should be reasonable (not 0, not equal to char count)
-    expect(chunks[0].tokens).toBeGreaterThan(0);
-    expect(chunks[0].tokens).toBeLessThan(content.length);  // Tokens < chars for English
+    expect(chunks[0]!.tokens).toBeGreaterThan(0);
+    expect(chunks[0]!.tokens).toBeLessThan(content.length);  // Tokens < chars for English
   });
 });
 
@@ -805,9 +805,9 @@ describe("FTS Search", () => {
 
     const results = store.searchFTS("fox", 10);
     expect(results.length).toBeGreaterThan(0);
-    expect(results[0].displayPath).toBe(`${collectionName}/test/doc1.md`);
-    expect(results[0].filepath).toBe(`qmd://${collectionName}/test/doc1.md`);
-    expect(results[0].source).toBe("fts");
+    expect(results[0]!.displayPath).toBe(`${collectionName}/test/doc1.md`);
+    expect(results[0]!.filepath).toBe(`qmd://${collectionName}/test/doc1.md`);
+    expect(results[0]!.source).toBe("fts");
 
     await cleanupTestDb(store);
   });
@@ -836,7 +836,7 @@ describe("FTS Search", () => {
     // Both documents contain "fox" in the body now, so we should get 2 results
     expect(results.length).toBe(2);
     // Title/name match should rank higher due to BM25 weights
-    expect(results[0].displayPath).toBe(`${collectionName}/test/title.md`);
+    expect(results[0]!.displayPath).toBe(`${collectionName}/test/title.md`);
 
     await cleanupTestDb(store);
   });
@@ -883,7 +883,7 @@ describe("FTS Search", () => {
     // Filter by collection name (collectionId is now treated as collection name string)
     const filtered = store.searchFTS("searchable", 10, collection1 as unknown as number);
     expect(filtered).toHaveLength(1);
-    expect(filtered[0].displayPath).toBe(`${collection1}/doc1.md`);
+    expect(filtered[0]!.displayPath).toBe(`${collection1}/doc1.md`);
 
     await cleanupTestDb(store);
   });
@@ -925,8 +925,8 @@ describe("FTS Search", () => {
 
     const results = store.searchFTS("findme", 10);
     expect(results).toHaveLength(1);
-    expect(results[0].displayPath).toBe(`${collectionName}/test/active.md`);
-    expect(results[0].filepath).toBe(`qmd://${collectionName}/test/active.md`);
+    expect(results[0]!.displayPath).toBe(`${collectionName}/test/active.md`);
+    expect(results[0]!.filepath).toBe(`qmd://${collectionName}/test/active.md`);
 
     await cleanupTestDb(store);
   });
@@ -1231,9 +1231,9 @@ describe("Document Retrieval", () => {
 
       const { docs } = store.findDocuments("large.md", { maxBytes: 10000 });
       expect(docs).toHaveLength(1);
-      expect(docs[0].skipped).toBe(true);
-      if (docs[0].skipped) {
-        expect(docs[0].skipReason).toContain("too large");
+      expect(docs[0]!.skipped).toBe(true);
+      if (docs[0]!.skipped) {
+        expect((docs[0] as { skipped: true; skipReason: string }).skipReason).toContain("too large");
       }
 
       await cleanupTestDb(store);
@@ -1251,9 +1251,9 @@ describe("Document Retrieval", () => {
       });
 
       const { docs } = store.findDocuments("doc1.md", { includeBody: true });
-      expect(docs[0].skipped).toBe(false);
-      if (!docs[0].skipped) {
-        expect(docs[0].doc.body).toBe("The content");
+      expect(docs[0]!.skipped).toBe(false);
+      if (!docs[0]!.skipped) {
+        expect((docs[0] as { doc: { body: string }; skipped: false }).doc.body).toBe("The content");
       }
 
       await cleanupTestDb(store);
@@ -1339,10 +1339,10 @@ describe("Snippet Extraction", () => {
     expect(headerMatch).not.toBeNull();
 
     const [, startLine, count, before, after] = headerMatch!;
-    expect(parseInt(startLine)).toBe(2); // Snippet starts at line 2 (B)
-    expect(parseInt(count)).toBe(4);     // 4 lines: B, C keyword, D, E
-    expect(parseInt(before)).toBe(1);    // A is before
-    expect(parseInt(after)).toBe(3);     // F, G, H are after
+    expect(parseInt(startLine!)).toBe(2); // Snippet starts at line 2 (B)
+    expect(parseInt(count!)).toBe(4);     // 4 lines: B, C keyword, D, E
+    expect(parseInt(before!)).toBe(1);    // A is before
+    expect(parseInt(after!)).toBe(3);     // F, G, H are after
   });
 
   test("extractSnippet at document start shows 0 before", () => {
@@ -1412,9 +1412,9 @@ describe("Reciprocal Rank Fusion", () => {
     const fused = reciprocalRankFusion([list1]);
 
     // Order should be preserved
-    expect(fused[0].file).toBe("doc1");
-    expect(fused[1].file).toBe("doc2");
-    expect(fused[2].file).toBe("doc3");
+    expect(fused[0]!.file).toBe("doc1");
+    expect(fused[1]!.file).toBe("doc2");
+    expect(fused[2]!.file).toBe("doc3");
   });
 
   test("RRF merges documents from multiple lists", () => {
@@ -1437,7 +1437,7 @@ describe("Reciprocal Rank Fusion", () => {
     const fused = reciprocalRankFusion([list1, list2], [2.0, 1.0]);
 
     // doc1 should rank higher due to weight
-    expect(fused[0].file).toBe("doc1");
+    expect(fused[0]!.file).toBe("doc1");
   });
 
   test("RRF adds top-rank bonus", () => {
@@ -1468,7 +1468,7 @@ describe("Reciprocal Rank Fusion", () => {
     const fused30 = reciprocalRankFusion([list], [], 30);
 
     // Lower k = higher scores for top ranks
-    expect(fused30[0].score).toBeGreaterThan(fused60[0].score);
+    expect(fused30[0]!.score).toBeGreaterThan(fused60[0]!.score);
   });
 });
 
@@ -1746,12 +1746,12 @@ describe("Integration", () => {
     const results2 = store2.searchFTS("different", 10);
 
     expect(results1).toHaveLength(1);
-    expect(results1[0].displayPath).toBe("store1/doc.md");
-    expect(results1[0].filepath).toBe("qmd://store1/doc.md");
+    expect(results1[0]!.displayPath).toBe("store1/doc.md");
+    expect(results1[0]!.filepath).toBe("qmd://store1/doc.md");
 
     expect(results2).toHaveLength(1);
-    expect(results2[0].displayPath).toBe("store2/doc.md");
-    expect(results2[0].filepath).toBe("qmd://store2/doc.md");
+    expect(results2[0]!.displayPath).toBe("store2/doc.md");
+    expect(results2[0]!.filepath).toBe("qmd://store2/doc.md");
 
     // Cross-check: store1 shouldn't find store2's content
     const cross1 = store1.searchFTS("different", 10);
@@ -1806,9 +1806,9 @@ describe("LlamaCpp Integration", () => {
 
     const results = await store.searchVec("test query", "embeddinggemma", 10);
     expect(results).toHaveLength(1);
-    expect(results[0].displayPath).toBe(`${collectionName}/doc1.md`);
-    expect(results[0].filepath).toBe(`qmd://${collectionName}/doc1.md`);
-    expect(results[0].source).toBe("vec");
+    expect(results[0]!.displayPath).toBe(`${collectionName}/doc1.md`);
+    expect(results[0]!.filepath).toBe(`qmd://${collectionName}/doc1.md`);
+    expect(results[0]!.source).toBe("vec");
 
     await cleanupTestDb(store);
   });
@@ -1849,7 +1849,7 @@ describe("LlamaCpp Integration", () => {
     const results = await store.rerank("topic", docs);
     expect(results).toHaveLength(2);
     // LlamaCpp reranker returns relevance scores
-    expect(results[0].score).toBeGreaterThan(0);
+    expect(results[0]!.score).toBeGreaterThan(0);
 
     await cleanupTestDb(store);
   });
@@ -2106,7 +2106,7 @@ describe("Content-Addressable Storage", () => {
     // All documents should point to the same hash
     const hashes = store.db.prepare(`SELECT DISTINCT hash FROM documents WHERE active = 1`).all() as { hash: string }[];
     expect(hashes).toHaveLength(1);
-    expect(hashes[0].hash).toBe(sharedHash);
+    expect(hashes[0]!.hash).toBe(sharedHash);
 
     await cleanupTestDb(store);
   });

test/eval-harness.ts

@@ -175,7 +175,7 @@ function evaluate(mode: "search" | "query") {
       .map((r, i) => ({ rank: i + 1, matches: r.file.toLowerCase().includes(expectedDoc) }))
       .filter(r => r.matches);
 
-    const firstHit = ranks.length > 0 ? ranks[0].rank : -1;
+    const firstHit = ranks.length > 0 ? ranks[0]!.rank : -1;
 
     results[difficulty].total++;
     if (firstHit === 1) results[difficulty].hit1++;