@aiconnect/easy-rag 0.3.0

package/dist/parsers/csv.js

@@ -0,0 +1,63 @@
+import fs from 'fs/promises';
+export async function parseCSV(filePath) {
+    try {
+        const content = await fs.readFile(filePath, 'utf-8');
+        const lines = content.trim().split('\n');
+        if (lines.length === 0) {
+            return [];
+        }
+        const headers = parseCSVLine(lines[0]);
+        const documents = [];
+        for (let i = 1; i < lines.length; i++) {
+            const values = parseCSVLine(lines[i]);
+            let documentContent;
+            if (headers.length > 0 && values.length === headers.length) {
+                documentContent = headers.map((header, index) => `${header}: ${values[index]}`).join('\n');
+            }
+            else {
+                documentContent = values.join(', ');
+            }
+            documents.push({
+                content: documentContent,
+                metadata: {
+                    sourcePath: filePath,
+                    contentType: 'csv',
+                    rowNumber: i
+                }
+            });
+        }
+        return documents;
+    }
+    catch (error) {
+        if (error instanceof Error) {
+            throw new Error(`Failed to parse CSV file: ${error.message}`);
+        }
+        throw new Error('Failed to parse CSV file: Unknown error');
+    }
+}
+function parseCSVLine(line) {
+    const result = [];
+    let current = '';
+    let inQuotes = false;
+    for (let i = 0; i < line.length; i++) {
+        const char = line[i];
+        if (char === '"') {
+            inQuotes = !inQuotes;
+        }
+        else if (char === ',' && !inQuotes) {
+            result.push(current.trim());
+            current = '';
+        }
+        else {
+            current += char;
+        }
+    }
+    result.push(current.trim());
+    return result.map(value => {
+        if (value.startsWith('"') && value.endsWith('"')) {
+            return value.slice(1, -1);
+        }
+        return value;
+    });
+}
+//# sourceMappingURL=csv.js.map

package/dist/parsers/csv.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"csv.js","sourceRoot":"","sources":["../../src/parsers/csv.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,aAAa,CAAC;AAG7B,MAAM,CAAC,KAAK,UAAU,QAAQ,CAAC,QAAgB;IAC7C,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACrD,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAEzC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,OAAO,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,SAAS,GAAkB,EAAE,CAAC;QAEpC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YACtC,IAAI,eAAuB,CAAC;YAE5B,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,MAAM,CAAC,MAAM,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC;gBAC3D,eAAe,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,MAAM,KAAK,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7F,CAAC;iBAAM,CAAC;gBACN,eAAe,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtC,CAAC;YAED,SAAS,CAAC,IAAI,CAAC;gBACb,OAAO,EAAE,eAAe;gBACxB,QAAQ,EAAE;oBACR,UAAU,EAAE,QAAQ;oBACpB,WAAW,EAAE,KAAK;oBAClB,SAAS,EAAE,CAAC;iBACb;aACF,CAAC,CAAC;QACL,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,6BAA6B,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QAChE,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;IAC7D,CAAC;AACH,CAAC;AAED,SAAS,YAAY,CAAC,IAAY;IAChC,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,OAAO,GAAG,EAAE,CAAC;IACjB,IAAI,QAAQ,GAAG,KAAK,CAAC;IAErB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACrC,MAAM,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QAErB,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;YACjB,QAAQ,GAAG,CAAC,QAAQ,CAAC;QACvB,CAAC;aAAM,IAAI,IAAI,KAAK,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;YACrC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;YAC5B,OAAO,GAAG,EAAE,CAAC;QACf,CAAC;aAAM,CAAC;YACN,OAAO,IAAI,IAAI,CAAC;QAClB,CAAC;IACH,CAAC;IAED,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IAE5B,OAAO,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE;QACxB,IAAI,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YACjD,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC5B,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/parsers/fileTypeDetector.d.ts

@@ -0,0 +1,3 @@
+import type { FileType } from './types.js';
+export declare function detectFileType(filePath: string): FileType;
+//# sourceMappingURL=fileTypeDetector.d.ts.map

package/dist/parsers/fileTypeDetector.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"fileTypeDetector.d.ts","sourceRoot":"","sources":["../../src/parsers/fileTypeDetector.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAE3C,wBAAgB,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,QAAQ,CAczD"}

package/dist/parsers/fileTypeDetector.js

@@ -0,0 +1,16 @@
+import path from 'path';
+export function detectFileType(filePath) {
+    const ext = path.extname(filePath).toLowerCase();
+    switch (ext) {
+        case '.pdf':
+            return 'pdf';
+        case '.md':
+        case '.markdown':
+            return 'markdown';
+        case '.csv':
+            return 'csv';
+        default:
+            throw new Error(`Unsupported file extension: ${ext}. Supported formats: .pdf, .md, .markdown, .csv`);
+    }
+}
+//# sourceMappingURL=fileTypeDetector.js.map

package/dist/parsers/fileTypeDetector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"fileTypeDetector.js","sourceRoot":"","sources":["../../src/parsers/fileTypeDetector.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,MAAM,CAAC;AAGxB,MAAM,UAAU,cAAc,CAAC,QAAgB;IAC7C,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;IAEjD,QAAQ,GAAG,EAAE,CAAC;QACZ,KAAK,MAAM;YACT,OAAO,KAAK,CAAC;QACf,KAAK,KAAK,CAAC;QACX,KAAK,WAAW;YACd,OAAO,UAAU,CAAC;QACpB,KAAK,MAAM;YACT,OAAO,KAAK,CAAC;QACf;YACE,MAAM,IAAI,KAAK,CAAC,+BAA+B,GAAG,iDAAiD,CAAC,CAAC;IACzG,CAAC;AACH,CAAC"}

package/dist/parsers/index.d.ts

@@ -0,0 +1,3 @@
+import type { ParsedDocument, CSVDocument } from './types.js';
+export declare function parseDocument(filePath: string): Promise<ParsedDocument | CSVDocument[]>;
+//# sourceMappingURL=index.d.ts.map

package/dist/parsers/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/parsers/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAM9D,wBAAsB,aAAa,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,GAAG,WAAW,EAAE,CAAC,CAa7F"}

package/dist/parsers/index.js

@@ -0,0 +1,18 @@
+import { detectFileType } from './fileTypeDetector.js';
+import { parsePDF } from './pdf.js';
+import { parseMarkdown } from './markdown.js';
+import { parseCSV } from './csv.js';
+export async function parseDocument(filePath) {
+    const fileType = detectFileType(filePath);
+    switch (fileType) {
+        case 'pdf':
+            return await parsePDF(filePath);
+        case 'markdown':
+            return await parseMarkdown(filePath);
+        case 'csv':
+            return await parseCSV(filePath);
+        default:
+            throw new Error(`Unsupported file type: ${fileType}`);
+    }
+}
+//# sourceMappingURL=index.js.map

package/dist/parsers/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/parsers/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAC;AACpC,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAC9C,OAAO,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAC;AAEpC,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,QAAgB;IAClD,MAAM,QAAQ,GAAG,cAAc,CAAC,QAAQ,CAAC,CAAC;IAE1C,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,KAAK;YACR,OAAO,MAAM,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAClC,KAAK,UAAU;YACb,OAAO,MAAM,aAAa,CAAC,QAAQ,CAAC,CAAC;QACvC,KAAK,KAAK;YACR,OAAO,MAAM,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAClC;YACE,MAAM,IAAI,KAAK,CAAC,0BAA0B,QAAQ,EAAE,CAAC,CAAC;IAC1D,CAAC;AACH,CAAC"}

package/dist/parsers/markdown.d.ts

@@ -0,0 +1,3 @@
+import type { ParsedDocument } from './types.js';
+export declare function parseMarkdown(filePath: string): Promise<ParsedDocument>;
+//# sourceMappingURL=markdown.d.ts.map

package/dist/parsers/markdown.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"markdown.d.ts","sourceRoot":"","sources":["../../src/parsers/markdown.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAEjD,wBAAsB,aAAa,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,CAAC,CA4B7E"}

package/dist/parsers/markdown.js

@@ -0,0 +1,30 @@
+import fs from 'fs/promises';
+export async function parseMarkdown(filePath) {
+    try {
+        const content = await fs.readFile(filePath, 'utf-8');
+        const lines = content.split('\n');
+        const sections = [];
+        const headingRegex = /^(#{1,3})\s+(.+)$/;
+        for (const line of lines) {
+            const match = line.match(headingRegex);
+            if (match) {
+                sections.push(match[2]);
+            }
+        }
+        return {
+            content,
+            metadata: {
+                sourcePath: filePath,
+                contentType: 'markdown',
+                sections: sections.length > 0 ? sections : undefined
+            }
+        };
+    }
+    catch (error) {
+        if (error instanceof Error) {
+            throw new Error(`Failed to parse Markdown file: ${error.message}`);
+        }
+        throw new Error('Failed to parse Markdown file: Unknown error');
+    }
+}
+//# sourceMappingURL=markdown.js.map

package/dist/parsers/markdown.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"markdown.js","sourceRoot":"","sources":["../../src/parsers/markdown.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,aAAa,CAAC;AAG7B,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,QAAgB;IAClD,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACrD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,QAAQ,GAAa,EAAE,CAAC;QAC9B,MAAM,YAAY,GAAG,mBAAmB,CAAC;QAEzC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YACvC,IAAI,KAAK,EAAE,CAAC;gBACV,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;QAED,OAAO;YACL,OAAO;YACP,QAAQ,EAAE;gBACR,UAAU,EAAE,QAAQ;gBACpB,WAAW,EAAE,UAAU;gBACvB,QAAQ,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;aACrD;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,kCAAkC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QACrE,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;IAClE,CAAC;AACH,CAAC"}

package/dist/parsers/pdf.d.ts

@@ -0,0 +1,3 @@
+import type { ParsedDocument } from './types.js';
+export declare function parsePDF(filePath: string): Promise<ParsedDocument>;
+//# sourceMappingURL=pdf.d.ts.map

package/dist/parsers/pdf.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pdf.d.ts","sourceRoot":"","sources":["../../src/parsers/pdf.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAEjD,wBAAsB,QAAQ,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,CAAC,CAkBxE"}

package/dist/parsers/pdf.js

@@ -0,0 +1,22 @@
+import fs from 'fs/promises';
+import pdfParse from 'pdf-parse';
+export async function parsePDF(filePath) {
+    try {
+        const dataBuffer = await fs.readFile(filePath);
+        const data = await pdfParse(dataBuffer);
+        return {
+            content: data.text,
+            metadata: {
+                sourcePath: filePath,
+                contentType: 'pdf'
+            }
+        };
+    }
+    catch (error) {
+        if (error instanceof Error) {
+            throw new Error(`Failed to parse PDF file: ${error.message}`);
+        }
+        throw new Error('Failed to parse PDF file: Unknown error');
+    }
+}
+//# sourceMappingURL=pdf.js.map

package/dist/parsers/pdf.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pdf.js","sourceRoot":"","sources":["../../src/parsers/pdf.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,aAAa,CAAC;AAC7B,OAAO,QAAQ,MAAM,WAAW,CAAC;AAGjC,MAAM,CAAC,KAAK,UAAU,QAAQ,CAAC,QAAgB;IAC7C,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC/C,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,UAAU,CAAC,CAAC;QAExC,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,IAAI;YAClB,QAAQ,EAAE;gBACR,UAAU,EAAE,QAAQ;gBACpB,WAAW,EAAE,KAAK;aACnB;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,6BAA6B,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QAChE,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;IAC7D,CAAC;AACH,CAAC"}

package/dist/parsers/types.d.ts

@@ -0,0 +1,17 @@
+export interface ParsedDocument {
+    content: string;
+    metadata: DocumentMetadata;
+}
+export interface DocumentMetadata {
+    sourcePath: string;
+    contentType: 'pdf' | 'markdown' | 'csv';
+    sections?: string[];
+    rowNumber?: number;
+}
+export type FileType = 'pdf' | 'markdown' | 'csv';
+export interface CSVDocument extends ParsedDocument {
+    metadata: DocumentMetadata & {
+        rowNumber: number;
+    };
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/parsers/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/parsers/types.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,cAAc;IAC7B,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,gBAAgB,CAAC;CAC5B;AAED,MAAM,WAAW,gBAAgB;IAC/B,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,KAAK,GAAG,UAAU,GAAG,KAAK,CAAC;IACxC,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,MAAM,QAAQ,GAAG,KAAK,GAAG,UAAU,GAAG,KAAK,CAAC;AAElD,MAAM,WAAW,WAAY,SAAQ,cAAc;IACjD,QAAQ,EAAE,gBAAgB,GAAG;QAAE,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC;CACpD"}

package/dist/parsers/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/parsers/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/parsers/types.ts"],"names":[],"mappings":""}

package/dist/query/embedding.d.ts

@@ -0,0 +1,2 @@
+export declare function generateQueryEmbedding(query: string): Promise<number[]>;
+//# sourceMappingURL=embedding.d.ts.map

package/dist/query/embedding.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embedding.d.ts","sourceRoot":"","sources":["../../src/query/embedding.ts"],"names":[],"mappings":"AAEA,wBAAsB,sBAAsB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAG7E"}

package/dist/query/embedding.js

@@ -0,0 +1,6 @@
+import { generateEmbeddings } from '../embeddings/openai.js';
+export async function generateQueryEmbedding(query) {
+    const embeddings = await generateEmbeddings([query]);
+    return embeddings[0];
+}
+//# sourceMappingURL=embedding.js.map

package/dist/query/embedding.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"embedding.js","sourceRoot":"","sources":["../../src/query/embedding.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAE7D,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAAC,KAAa;IACxD,MAAM,UAAU,GAAG,MAAM,kBAAkB,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;IACrD,OAAO,UAAU,CAAC,CAAC,CAAC,CAAC;AACvB,CAAC"}

package/dist/query/index.d.ts

@@ -0,0 +1,3 @@
+import type { QueryOptions } from './types.js';
+export declare function executeQuery(question: string, options: QueryOptions): Promise<void>;
+//# sourceMappingURL=index.d.ts.map

package/dist/query/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/query/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAe,YAAY,EAAE,MAAM,YAAY,CAAC;AAI5D,wBAAsB,YAAY,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC,CAiEzF"}

package/dist/query/index.js

@@ -0,0 +1,86 @@
+import { generateQueryEmbedding } from './embedding.js';
+import { listAllCollections, getCollection, searchChromaDB } from './search.js';
+export async function executeQuery(question, options) {
+    if (!process.env.OPENAI_API_KEY) {
+        console.error('Error: OPENAI_API_KEY environment variable is required');
+        console.error('\nTo set your API key:');
+        console.error('  export OPENAI_API_KEY=your-api-key-here');
+        console.error('\nOr pass it when running the command:');
+        console.error('  OPENAI_API_KEY=your-api-key-here easy-rag query "<question>"');
+        process.exit(1);
+    }
+    const queryEmbedding = await generateQueryEmbedding(question);
+    const allResults = [];
+    if (options.collection) {
+        try {
+            const collection = await getCollection(options.collection);
+            const results = await searchChromaDB(collection, queryEmbedding, options.top);
+            allResults.push(...results);
+        }
+        catch (error) {
+            if (error instanceof Error && error.message.includes('Collection')) {
+                console.error(`Error: Collection "${options.collection}" not found`);
+                console.error('\nAvailable collections:');
+                const collections = await listAllCollections();
+                if (collections.length === 0) {
+                    console.error('  (no collections found)');
+                }
+                else {
+                    collections.forEach((c) => console.error(`  - ${c}`));
+                }
+                process.exit(1);
+            }
+            throw error;
+        }
+    }
+    else {
+        const collectionNames = await listAllCollections();
+        if (collectionNames.length === 0) {
+            console.error('Error: No collections found in ChromaDB');
+            console.error('\nPlease run "easy-rag index <folder>" to create a collection first.');
+            process.exit(1);
+        }
+        for (const name of collectionNames) {
+            try {
+                const collection = await getCollection(name);
+                const results = await searchChromaDB(collection, queryEmbedding, options.top);
+                allResults.push(...results);
+            }
+            catch (error) {
+                console.error(`Warning: Failed to search collection "${name}": ${error instanceof Error ? error.message : String(error)}`);
+            }
+        }
+    }
+    if (allResults.length === 0) {
+        console.log('No results found.');
+        return;
+    }
+    allResults.sort((a, b) => b.similarity - a.similarity);
+    const topResults = allResults.slice(0, options.top);
+    if (options.includeMetadata) {
+        displayResultsWithMetadata(topResults);
+    }
+    else {
+        displayResults(topResults);
+    }
+}
+function displayResults(results) {
+    console.log();
+    results.forEach((result, index) => {
+        console.log(`--- Result ${index + 1} ---`);
+        console.log(result.content);
+        console.log();
+    });
+}
+function displayResultsWithMetadata(results) {
+    console.log();
+    results.forEach((result, index) => {
+        console.log(`--- Result ${index + 1} ---`);
+        console.log(`Source: ${result.metadata.sourcePath}`);
+        console.log(`Similarity: ${(result.similarity * 100).toFixed(1)}%`);
+        console.log(`Chunk: ${result.metadata.chunkIndex}`);
+        console.log(`\nContent:\n${result.content}`);
+        console.log();
+    });
+}
+//# sourceMappingURL=index.js.map

package/dist/query/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/query/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,sBAAsB,EAAE,MAAM,gBAAgB,CAAC;AACxD,OAAO,EAAE,kBAAkB,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAEhF,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,QAAgB,EAAE,OAAqB;IACxE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC;QAChC,OAAO,CAAC,KAAK,CAAC,wDAAwD,CAAC,CAAC;QACxE,OAAO,CAAC,KAAK,CAAC,wBAAwB,CAAC,CAAC;QACxC,OAAO,CAAC,KAAK,CAAC,2CAA2C,CAAC,CAAC;QAC3D,OAAO,CAAC,KAAK,CAAC,wCAAwC,CAAC,CAAC;QACxD,OAAO,CAAC,KAAK,CAAC,gEAAgE,CAAC,CAAC;QAChF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,MAAM,cAAc,GAAG,MAAM,sBAAsB,CAAC,QAAQ,CAAC,CAAC;IAE9D,MAAM,UAAU,GAAkB,EAAE,CAAC;IAErC,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;QACvB,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,MAAM,aAAa,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAC3D,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,UAAU,EAAE,cAAc,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC;YAC9E,UAAU,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,CAAC;QAC9B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;gBACnE,OAAO,CAAC,KAAK,CAAC,sBAAsB,OAAO,CAAC,UAAU,aAAa,CAAC,CAAC;gBACrE,OAAO,CAAC,KAAK,CAAC,0BAA0B,CAAC,CAAC;gBAC1C,MAAM,WAAW,GAAG,MAAM,kBAAkB,EAAE,CAAC;gBAC/C,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBAC7B,OAAO,CAAC,KAAK,CAAC,0BAA0B,CAAC,CAAC;gBAC5C,CAAC;qBAAM,CAAC;oBACN,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;gBACxD,CAAC;gBACD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAClB,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;SAAM,CAAC;QACN,MAAM,eAAe,GAAG,MAAM,kBAAkB,EAAE,CAAC;QACnD,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACjC,OAAO,CAAC,KAAK,CAAC,yCAAyC,CAAC,CAAC;YACzD,OAAO,CAAC,KAAK,CAAC,sEAAsE,CAAC,CAAC;YACtF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,KAAK,MAAM,IAAI,IAAI,eAAe,EAAE,CAAC;YACnC,IAAI,CAAC;gBACH,MAAM,UAAU,GAAG,MAAM,aAAa,CAAC,IAAI,CAAC,CAAC;gBAC7C,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,UAAU,EAAE,cAAc,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC;gBAC9E,UAAU,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,CAAC;YAC9B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,yCAAyC,IAAI,MAAM,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAC7H,CAAC;QACH,CAAC;IACH,CAAC;IAED,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5B,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;QACjC,OAAO;IACT,CAAC;IAED,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC;IACvD,MAAM,UAAU,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC;IAEpD,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;QAC5B,0BAA0B,CAAC,UAAU,CAAC,CAAC;IACzC,CAAC;SAAM,CAAC;QACN,cAAc,CAAC,UAAU,CAAC,CAAC;IAC7B,CAAC;AACH,CAAC;AAED,SAAS,cAAc,CAAC,OAAsB;IAC5C,OAAO,CAAC,GAAG,EAAE,CAAC;IACd,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE;QAChC,OAAO,CAAC,GAAG,CAAC,cAAc,KAAK,GAAG,CAAC,MAAM,CAAC,CAAC;QAC3C,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC5B,OAAO,CAAC,GAAG,EAAE,CAAC;IAChB,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,0BAA0B,CAAC,OAAsB;IACxD,OAAO,CAAC,GAAG,EAAE,CAAC;IACd,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE;QAChC,OAAO,CAAC,GAAG,CAAC,cAAc,KAAK,GAAG,CAAC,MAAM,CAAC,CAAC;QAC3C,OAAO,CAAC,GAAG,CAAC,WAAW,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QACrD,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,MAAM,CAAC,UAAU,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QACpE,OAAO,CAAC,GAAG,CAAC,UAAU,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QACpD,OAAO,CAAC,GAAG,CAAC,eAAe,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;QAC7C,OAAO,CAAC,GAAG,EAAE,CAAC;IAChB,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/query/search.d.ts

@@ -0,0 +1,6 @@
+import { type Collection } from 'chromadb';
+import type { QueryResult } from './types.js';
+export declare function listAllCollections(): Promise<string[]>;
+export declare function getCollection(collectionName: string): Promise<Collection>;
+export declare function searchChromaDB(collection: Collection, queryEmbedding: number[], top: number): Promise<QueryResult[]>;
+//# sourceMappingURL=search.d.ts.map

package/dist/query/search.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"search.d.ts","sourceRoot":"","sources":["../../src/query/search.ts"],"names":[],"mappings":"AAAA,OAAO,EAAgB,KAAK,UAAU,EAAE,MAAM,UAAU,CAAC;AACzD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAM9C,wBAAsB,kBAAkB,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC,CAG5D;AAED,wBAAsB,aAAa,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC,CAK/E;AAED,wBAAsB,cAAc,CAClC,UAAU,EAAE,UAAU,EACtB,cAAc,EAAE,MAAM,EAAE,EACxB,GAAG,EAAE,MAAM,GACV,OAAO,CAAC,WAAW,EAAE,CAAC,CAiCxB"}

package/dist/query/search.js

@@ -0,0 +1,45 @@
+import { ChromaClient } from 'chromadb';
+const client = new ChromaClient({
+    path: process.env.CHROMA_URL || 'http://localhost:8000',
+});
+export async function listAllCollections() {
+    const collections = await client.listCollections();
+    return collections.map((c) => String(c));
+}
+export async function getCollection(collectionName) {
+    const sanitized = collectionName
+        .replace(/[^a-zA-Z0-9_-]/g, '_')
+        .toLowerCase();
+    return await client.getCollection({ name: sanitized });
+}
+export async function searchChromaDB(collection, queryEmbedding, top) {
+    const result = await collection.query({
+        queryEmbeddings: [queryEmbedding],
+        nResults: top,
+    });
+    const queryResults = [];
+    if (result.documents && result.documents[0]) {
+        for (let i = 0; i < result.documents[0].length; i++) {
+            const content = result.documents[0][i];
+            const metadata = result.metadatas?.[0]?.[i];
+            const distance = result.distances?.[0]?.[i];
+            if (content && metadata) {
+                queryResults.push({
+                    content,
+                    metadata: {
+                        sourcePath: String(metadata.sourcePath),
+                        contentType: String(metadata.contentType),
+                        chunkIndex: Number(metadata.chunkIndex),
+                        sectionTitle: metadata.sectionTitle !== undefined ? String(metadata.sectionTitle) : undefined,
+                        startPosition: metadata.startPosition !== undefined ? Number(metadata.startPosition) : undefined,
+                        endPosition: metadata.endPosition !== undefined ? Number(metadata.endPosition) : undefined,
+                        rowNumber: metadata.rowNumber !== undefined ? Number(metadata.rowNumber) : undefined,
+                    },
+                    similarity: distance !== undefined ? Math.max(0, 1 - distance) : 0,
+                });
+            }
+        }
+    }
+    return queryResults;
+}
+//# sourceMappingURL=search.js.map

package/dist/query/search.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"search.js","sourceRoot":"","sources":["../../src/query/search.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAmB,MAAM,UAAU,CAAC;AAGzD,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC;IAC9B,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,UAAU,IAAI,uBAAuB;CACxD,CAAC,CAAC;AAEH,MAAM,CAAC,KAAK,UAAU,kBAAkB;IACtC,MAAM,WAAW,GAAG,MAAM,MAAM,CAAC,eAAe,EAAE,CAAC;IACnD,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;AAC3C,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,cAAsB;IACxD,MAAM,SAAS,GAAG,cAAc;SAC7B,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC;SAC/B,WAAW,EAAE,CAAC;IACjB,OAAO,MAAM,MAAM,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;AACzD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,UAAsB,EACtB,cAAwB,EACxB,GAAW;IAEX,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC;QACpC,eAAe,EAAE,CAAC,cAAc,CAAC;QACjC,QAAQ,EAAE,GAAG;KACd,CAAC,CAAC;IAEH,MAAM,YAAY,GAAkB,EAAE,CAAC;IAEvC,IAAI,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;QAC5C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACpD,MAAM,OAAO,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACvC,MAAM,QAAQ,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAA4B,CAAC;YACvE,MAAM,QAAQ,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAE5C,IAAI,OAAO,IAAI,QAAQ,EAAE,CAAC;gBACxB,YAAY,CAAC,IAAI,CAAC;oBAChB,OAAO;oBACP,QAAQ,EAAE;wBACR,UAAU,EAAE,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC;wBACvC,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC;wBACzC,UAAU,EAAE,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC;wBACvC,YAAY,EAAE,QAAQ,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,SAAS;wBAC7F,aAAa,EAAE,QAAQ,CAAC,aAAa,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,SAAS;wBAChG,WAAW,EAAE,QAAQ,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,SAAS;wBAC1F,SAAS,EAAE,QAAQ,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS;qBACrF;oBACD,UAAU,EAAE,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;iBACnE,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,YAAY,CAAC;AACtB,CAAC"}

package/dist/query/types.d.ts

@@ -0,0 +1,19 @@
+export interface QueryResult {
+    content: string;
+    metadata: {
+        sourcePath: string;
+        contentType: string;
+        chunkIndex: number;
+        sectionTitle?: string;
+        startPosition?: number;
+        endPosition?: number;
+        rowNumber?: number;
+    };
+    similarity: number;
+}
+export interface QueryOptions {
+    top: number;
+    includeMetadata: boolean;
+    collection?: string;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/query/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/query/types.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE;QACR,UAAU,EAAE,MAAM,CAAC;QACnB,WAAW,EAAE,MAAM,CAAC;QACpB,UAAU,EAAE,MAAM,CAAC;QACnB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,SAAS,CAAC,EAAE,MAAM,CAAC;KACpB,CAAC;IACF,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,YAAY;IAC3B,GAAG,EAAE,MAAM,CAAC;IACZ,eAAe,EAAE,OAAO,CAAC;IACzB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB"}

package/dist/query/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/query/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/query/types.ts"],"names":[],"mappings":""}

package/dist/vector-store/chroma-server.d.ts

@@ -0,0 +1,10 @@
+/**
+ * Ensures ChromaDB is running. If not, starts it automatically.
+ * Returns `true` if ChromaDB is ready, `false` otherwise.
+ */
+export declare function ensureChromaRunning(): Promise<boolean>;
+/**
+ * Stops the ChromaDB server if it was started by this process.
+ */
+export declare function stopChroma(): void;
+//# sourceMappingURL=chroma-server.d.ts.map

package/dist/vector-store/chroma-server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chroma-server.d.ts","sourceRoot":"","sources":["../../src/vector-store/chroma-server.ts"],"names":[],"mappings":"AAwDA;;;GAGG;AACH,wBAAsB,mBAAmB,IAAI,OAAO,CAAC,OAAO,CAAC,CA4C5D;AAED;;GAEG;AACH,wBAAgB,UAAU,IAAI,IAAI,CAKjC"}

package/dist/vector-store/chroma-server.js

@@ -0,0 +1,102 @@
+import { spawn } from 'child_process';
+import { fileURLToPath } from 'url';
+import path from 'path';
+import fs from 'fs';
+import { ensureChromaDataDirectory } from './utils.js';
+const CHROMA_URL = process.env.CHROMA_URL || 'http://localhost:8000';
+let chromaProcess = null;
+/**
+ * Checks if the ChromaDB server is reachable.
+ */
+async function isChromaRunning() {
+    try {
+        const response = await fetch(`${CHROMA_URL}/api/v2/heartbeat`);
+        return response.ok;
+    }
+    catch {
+        return false;
+    }
+}
+/**
+ * Waits until ChromaDB responds to heartbeat, with timeout.
+ */
+async function waitForChroma(timeoutMs = 15_000) {
+    const start = Date.now();
+    while (Date.now() - start < timeoutMs) {
+        if (await isChromaRunning())
+            return true;
+        await new Promise((r) => setTimeout(r, 500));
+    }
+    return false;
+}
+/**
+ * Resolves the chroma binary path.
+ * Looks in the package's own node_modules first, then cwd's node_modules, then falls back to npx.
+ */
+function resolveChromaBin() {
+    // 1. Try the package's own node_modules (works when installed globally)
+    const __dirname = path.dirname(fileURLToPath(import.meta.url));
+    const packageRoot = path.resolve(__dirname, '..', '..');
+    const packageBin = path.join(packageRoot, 'node_modules', '.bin', 'chroma');
+    if (fs.existsSync(packageBin)) {
+        return { command: packageBin, args: [] };
+    }
+    // 2. Try cwd's node_modules (works during development)
+    const cwdBin = path.join(process.cwd(), 'node_modules', '.bin', 'chroma');
+    if (fs.existsSync(cwdBin)) {
+        return { command: cwdBin, args: [] };
+    }
+    // 3. Fall back to npx
+    return { command: 'npx', args: ['chroma'] };
+}
+/**
+ * Ensures ChromaDB is running. If not, starts it automatically.
+ * Returns `true` if ChromaDB is ready, `false` otherwise.
+ */
+export async function ensureChromaRunning() {
+    if (await isChromaRunning()) {
+        return true;
+    }
+    console.log('ChromaDB is not running. Starting server...');
+    const chromadbPath = await ensureChromaDataDirectory();
+    const port = new URL(CHROMA_URL).port || '8000';
+    const { command, args } = resolveChromaBin();
+    chromaProcess = spawn(command, [...args, 'run', '--path', chromadbPath, '--port', port], {
+        stdio: 'ignore',
+        detached: false,
+    });
+    chromaProcess.on('error', (err) => {
+        console.error(`Failed to start ChromaDB: ${err.message}`);
+    });
+    chromaProcess.on('exit', (code) => {
+        if (code !== null && code !== 0) {
+            console.error(`ChromaDB exited with code ${code}`);
+        }
+        chromaProcess = null;
+    });
+    // Register cleanup handlers
+    const cleanup = () => stopChroma();
+    process.on('exit', cleanup);
+    process.on('SIGINT', () => { cleanup(); process.exit(0); });
+    process.on('SIGTERM', () => { cleanup(); process.exit(0); });
+    const ready = await waitForChroma();
+    if (ready) {
+        console.log('ChromaDB server started successfully.\n');
+    }
+    else {
+        console.error('ChromaDB failed to start within 15 seconds.');
+        stopChroma();
+        return false;
+    }
+    return true;
+}
+/**
+ * Stops the ChromaDB server if it was started by this process.
+ */
+export function stopChroma() {
+    if (chromaProcess && !chromaProcess.killed) {
+        chromaProcess.kill('SIGTERM');
+        chromaProcess = null;
+    }
+}
+//# sourceMappingURL=chroma-server.js.map

package/dist/vector-store/chroma-server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chroma-server.js","sourceRoot":"","sources":["../../src/vector-store/chroma-server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAqB,MAAM,eAAe,CAAC;AACzD,OAAO,EAAE,aAAa,EAAE,MAAM,KAAK,CAAC;AACpC,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,EAAE,yBAAyB,EAAE,MAAM,YAAY,CAAC;AAEvD,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,CAAC,UAAU,IAAI,uBAAuB,CAAC;AACrE,IAAI,aAAa,GAAwB,IAAI,CAAC;AAE9C;;GAEG;AACH,KAAK,UAAU,eAAe;IAC5B,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,UAAU,mBAAmB,CAAC,CAAC;QAC/D,OAAO,QAAQ,CAAC,EAAE,CAAC;IACrB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,aAAa,CAAC,SAAS,GAAG,MAAM;IAC7C,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACzB,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,GAAG,SAAS,EAAE,CAAC;QACtC,IAAI,MAAM,eAAe,EAAE;YAAE,OAAO,IAAI,CAAC;QACzC,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;IAC/C,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;;GAGG;AACH,SAAS,gBAAgB;IACvB,wEAAwE;IACxE,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IAC/D,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IACxD,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,cAAc,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;IAC5E,IAAI,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;QAC9B,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;IAC3C,CAAC;IAED,uDAAuD;IACvD,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;IAC1E,IAAI,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1B,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;IACvC,CAAC;IAED,sBAAsB;IACtB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;AAC9C,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB;IACvC,IAAI,MAAM,eAAe,EAAE,EAAE,CAAC;QAC5B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,6CAA6C,CAAC,CAAC;IAE3D,MAAM,YAAY,GAAG,MAAM,yBAAyB,EAAE,CAAC;IACvD,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,CAAC,IAAI,IAAI,MAAM,CAAC;IAChD,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,gBAAgB,EAAE,CAAC;IAE7C,aAAa,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC,GAAG,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE;QACvF,KAAK,EAAE,QAAQ;QACf,QAAQ,EAAE,KAAK;KAChB,CAAC,CAAC;IAEH,aAAa,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;QAChC,OAAO,CAAC,KAAK,CAAC,6BAA6B,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,aAAa,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;QAChC,IAAI,IAAI,KAAK,IAAI,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;YAChC,OAAO,CAAC,KAAK,CAAC,6BAA6B,IAAI,EAAE,CAAC,CAAC;QACrD,CAAC;QACD,aAAa,GAAG,IAAI,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,4BAA4B;IAC5B,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC,UAAU,EAAE,CAAC;IACnC,OAAO,CAAC,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC5B,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5D,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAE7D,MAAM,KAAK,GAAG,MAAM,aAAa,EAAE,CAAC;IAEpC,IAAI,KAAK,EAAE,CAAC;QACV,OAAO,CAAC,GAAG,CAAC,yCAAyC,CAAC,CAAC;IACzD,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,KAAK,CAAC,6CAA6C,CAAC,CAAC;QAC7D,UAAU,EAAE,CAAC;QACb,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,UAAU;IACxB,IAAI,aAAa,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC;QAC3C,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9B,aAAa,GAAG,IAAI,CAAC;IACvB,CAAC;AACH,CAAC"}

package/dist/vector-store/chromadb.d.ts

@@ -0,0 +1,8 @@
+import { type Collection } from 'chromadb';
+import type { VectorChunk, CollectionInfo } from './types.js';
+export declare function getOrCreateCollection(baseName: string): Promise<Collection>;
+export declare function storeEmbeddings(collection: Collection, chunks: VectorChunk[]): Promise<void>;
+export declare function initializeChromaDB(): Promise<void>;
+export declare function listCollections(): Promise<CollectionInfo[]>;
+export declare function deleteCollection(collectionName: string): Promise<void>;
+//# sourceMappingURL=chromadb.d.ts.map

package/dist/vector-store/chromadb.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chromadb.d.ts","sourceRoot":"","sources":["../../src/vector-store/chromadb.ts"],"names":[],"mappings":"AAAA,OAAO,EAAgB,KAAK,UAAU,EAAE,MAAM,UAAU,CAAC;AACzD,OAAO,KAAK,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAO9D,wBAAsB,qBAAqB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC,CA6BjF;AAED,wBAAsB,eAAe,CACnC,UAAU,EAAE,UAAU,EACtB,MAAM,EAAE,WAAW,EAAE,GACpB,OAAO,CAAC,IAAI,CAAC,CAqCf;AAED,wBAAsB,kBAAkB,IAAI,OAAO,CAAC,IAAI,CAAC,CAExD;AAED,wBAAsB,eAAe,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC,CAqBjE;AAED,wBAAsB,gBAAgB,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAK5E"}