@ixo/data-store 1.0.1

package/dist/chroma/chroma-data-store.js

@@ -0,0 +1,196 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChromaDataStore = void 0;
+const chromadb_1 = require("chromadb");
+const vector_db_data_store_1 = require("../types/vector-db-data-store");
+const embedding_function_1 = require("./embedding-function");
+const utils_1 = require("./utils");
+/**
+ * ChromaDataStore class for managing vector storage and retrieval using ChromaDB
+ */
+class ChromaDataStore extends vector_db_data_store_1.VectorDBDataStore {
+    client;
+    collection;
+    /**
+     * Creates a new ChromaDataStore instance
+     * @param options - Configuration options for the vector store
+     * @throws Error if OPENAI_API_KEY is not set and no embedding function is provided
+     */
+    constructor(options) {
+        if (typeof process.env.OPENAI_API_KEY !== 'string' &&
+            !options.embeddingFunction) {
+            throw new Error('OPENAI_API_KEY is not set and no embedding function is provided');
+        }
+        options.embeddingFunction =
+            options.embeddingFunction ||
+                new embedding_function_1.OpenAIEmbeddingFunction({
+                    openai_api_key: process.env.OPENAI_API_KEY ?? '',
+                    openai_model: 'text-embedding-3-small',
+                });
+        options.url = options.url || 'http://localhost:8000';
+        options.collectionName = options.collectionName || 'default-vector-store';
+        super(options);
+        this.client = new chromadb_1.ChromaClient({
+            path: options.url,
+        });
+        // this is a workaround to avoid setting collection as optional
+        this.collection = undefined;
+    }
+    /**
+     * Checks if the ChromaDB collection is initialized
+     * @throws Error if collection is not initialized
+     */
+    checkIsInitialized() {
+        if (!this.collection) {
+            throw new Error('ChromaDataStore is not initialized');
+        }
+    }
+    /**
+     * Initializes the ChromaDB collection
+     */
+    async init() {
+        if (this.collection) {
+            return;
+        }
+        const collection = await this.client.getOrCreateCollection({
+            name: this.options.collectionName,
+            embeddingFunction: this.options.embeddingFunction,
+        });
+        this.collection = collection;
+    }
+    /**
+     * Queries the vector store using text
+     * @param query - Text query to search for
+     * @param options - Query options including filters and top-k results
+     * @returns Array of matching documents
+     */
+    async query(query, options) {
+        this.checkIsInitialized();
+        const params = {
+            queryTexts: [query],
+            nResults: options?.topK || 10,
+        };
+        if (options?.filters) {
+            params.where = options.filters;
+        }
+        const result = await this.collection.query(params);
+        return (0, utils_1.createVectorStoreSearchResult)(result);
+    }
+    /**
+     * Queries the vector store and filters results by similarity threshold
+     * @param query - Text query to search for
+     * @param options - Query options including similarity threshold
+     * @returns Array of documents meeting the similarity threshold
+     */
+    async queryWithSimilarity(query, options) {
+        this.checkIsInitialized();
+        const result = await this.query(query, options);
+        return result.filter((doc) => (doc.score && doc.score >= (options?.similarityThreshold ?? 0.5)) ??
+            false);
+    }
+    /**
+     * Upserts (inserts or updates) documents into the vector store
+     * @param documents - Array of documents to upsert
+     * @throws Error if any document is missing an ID
+     */
+    async upsert(documents) {
+        this.checkIsInitialized();
+        const [ids, contents, metadatas] = documents.reduce((acc, doc) => {
+            const id = doc.id;
+            const metadata = doc.metadata ?? {};
+            const content = doc.content;
+            if (!id) {
+                throw new Error('Document ID is required');
+            }
+            acc[0].push(id);
+            acc[1].push(content);
+            acc[2].push(metadata);
+            return acc;
+        }, [[], [], []]);
+        await this.collection.upsert({
+            ids,
+            documents: contents,
+            metadatas,
+        });
+    }
+    async addDocumentsWithEmbeddings(documents) {
+        this.checkIsInitialized();
+        const [ids, embeddings, metadatas, contents] = documents.reduce((acc, doc) => {
+            const id = doc.id;
+            const metadata = doc.metadata ?? {};
+            const embedding = doc.embedding;
+            if (!id) {
+                throw new Error('Document ID is required');
+            }
+            if (embedding.length < 1536) {
+                throw new Error('Embedding is required');
+            }
+            acc[0].push(id);
+            acc[1].push(embedding);
+            acc[2].push(metadata);
+            acc[3].push(doc.content);
+            return acc;
+        }, [[], [], [], []]);
+        await this.collection.add({
+            ids,
+            embeddings,
+            metadatas,
+            documents: contents,
+        });
+    }
+    /**
+     * Deletes documents from the vector store by their IDs
+     * @param ids - Array of document IDs to delete
+     */
+    async delete(ids) {
+        this.checkIsInitialized();
+        await this.collection.delete({
+            ids,
+        });
+    }
+    /**
+     * Retrieves a document by its ID
+     * @param id - Document ID to retrieve
+     * @returns The document if found, null otherwise
+     */
+    async getById(id) {
+        const result = await this.collection.get({
+            ids: [id],
+        });
+        return result.documents.at(0)
+            ? {
+                id: result.ids.at(0)?.toString() ?? '',
+                content: result.documents.at(0) ?? '',
+                metadata: result.metadatas.at(0) ?? {},
+                score: undefined,
+            }
+            : null;
+    }
+    /**
+     * Queries the vector store using a pre-computed embedding vector
+     * @param vector - The embedding vector to search with
+     * @param options - Query options including top-k results
+     * @returns Array of matching documents
+     */
+    async queryByVector(vector, options) {
+        this.checkIsInitialized();
+        const result = await this.collection.query({
+            queryEmbeddings: [vector],
+            nResults: options?.topK || 10,
+        });
+        return (0, utils_1.createVectorStoreSearchResult)(result);
+    }
+    /**
+     * Updates the metadata for a document by its ID
+     * @param ids - Array of document IDs to update
+     * @param metadata - Metadata to update
+     */
+    async updateMetadata(ids, metadatas) {
+        this.checkIsInitialized();
+        await this.collection.update({
+            ids,
+            metadatas,
+        });
+    }
+}
+exports.ChromaDataStore = ChromaDataStore;

package/dist/chroma/chroma-data-store.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=chroma-data-store.test.d.ts.map

package/dist/chroma/chroma-data-store.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chroma-data-store.test.d.ts","sourceRoot":"","sources":["../../src/chroma/chroma-data-store.test.ts"],"names":[],"mappings":""}

package/dist/chroma/chroma-data-store.test.js

@@ -0,0 +1,176 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+/* eslint-disable jest/unbound-method  -- ignore this for now */
+const chromadb_1 = require("chromadb");
+const chroma_data_store_1 = require("./chroma-data-store");
+jest.mock('chromadb');
+const embeddingFunction = {
+    generate: jest.fn(),
+};
+jest
+    .mocked(chromadb_1.ChromaClient.prototype.getOrCreateCollection)
+    .mockResolvedValue(new chromadb_1.Collection('test', 'test', jest.fn(), embeddingFunction));
+describe('ChromaDataStore', () => {
+    beforeAll(() => {
+        process.env.OPENAI_API_KEY = 'test';
+    });
+    afterAll(() => {
+        process.env.OPENAI_API_KEY = undefined;
+    });
+    it('should be defined', () => {
+        expect(chroma_data_store_1.ChromaDataStore).toBeDefined();
+    });
+    describe('Initialization', () => {
+        it('should construct and initialize without errors', async () => {
+            // Should not throw during construction
+            const chromaDataStore = new chroma_data_store_1.ChromaDataStore({
+                collectionName: 'test',
+                url: 'http://localhost:8000',
+            });
+            expect(chromaDataStore).toBeInstanceOf(chroma_data_store_1.ChromaDataStore);
+            // Should not throw during initialization
+            await expect(chromaDataStore.init()).resolves.not.toThrow();
+        });
+        it('should throw error when OPENAI_API_KEY is not set and no embedding function is provided', () => {
+            // Temporarily remove API key
+            const originalKey = process.env.OPENAI_API_KEY;
+            delete process.env.OPENAI_API_KEY;
+            // Should throw during construction
+            expect(() => new chroma_data_store_1.ChromaDataStore({
+                collectionName: 'test',
+                url: 'http://localhost:8000',
+            })).toThrow('OPENAI_API_KEY is not set');
+            // Restore API key
+            process.env.OPENAI_API_KEY = originalKey;
+        });
+        it('should create collection or use existing collection', async () => {
+            jest.spyOn(chromadb_1.ChromaClient.prototype, 'getOrCreateCollection');
+            const chromaDataStore = new chroma_data_store_1.ChromaDataStore({
+                collectionName: 'test',
+                url: 'http://localhost:8000',
+                embeddingFunction,
+            });
+            await chromaDataStore.init();
+            expect(chromadb_1.ChromaClient.prototype.getOrCreateCollection).toHaveBeenCalledWith({
+                name: 'test',
+                embeddingFunction,
+            });
+        });
+    });
+    describe('Query', () => {
+        let chromaDataStore;
+        beforeAll(async () => {
+            chromaDataStore = new chroma_data_store_1.ChromaDataStore({
+                collectionName: 'test',
+                url: 'http://localhost:8000',
+                embeddingFunction,
+            });
+            await chromaDataStore.init();
+            await chromaDataStore.init();
+            jest.mocked(chromadb_1.Collection.prototype.query).mockResolvedValue({
+                ids: [['1', '2']],
+                embeddings: null,
+                documents: [['Hello, world!', 'Hello, world! 2']],
+                metadatas: [[{ title: 'Hello, world!' }, { title: 'Hello, world! 2' }]],
+                distances: [[3.3435163194430493e-16, 0.30934086831133445]],
+                included: [],
+            });
+        });
+        it('should query documents', async () => {
+            const documents = await chromaDataStore.query('Hello, world!');
+            expect(documents).toHaveLength(2);
+            expect(documents[0]?.id).toBe('1');
+            expect(documents[0]?.content).toBe('Hello, world!');
+        });
+        it('should query documents by vector', async () => {
+            const vector = [1, 2, 3];
+            const documents = await chromaDataStore.queryByVector(vector);
+            expect(documents).toHaveLength(2);
+            expect(documents[0]?.id).toBe('1');
+            expect(documents[0]?.content).toBe('Hello, world!');
+        });
+        it('should query documents by similarity', async () => {
+            const documents = await chromaDataStore.queryWithSimilarity('Hello, world!', { similarityThreshold: 0.9 });
+            expect(documents).toHaveLength(1);
+            expect(documents[0]?.id).toBe('1');
+            expect(documents[0]?.content).toBe('Hello, world!');
+        });
+        it('should get document by id', async () => {
+            jest.spyOn(chromadb_1.Collection.prototype, 'get');
+            jest.mocked(chromadb_1.Collection.prototype.get).mockResolvedValue({
+                ids: ['1'],
+                embeddings: null,
+                documents: ['Hello, world!'],
+                metadatas: [{ title: 'Hello, world!' }],
+                included: [],
+            });
+            const document = await chromaDataStore.getById('1');
+            expect(chromadb_1.Collection.prototype.get).toHaveBeenCalledWith({
+                ids: ['1'],
+            });
+            expect(document?.id).toBe('1');
+            expect(document?.content).toBe('Hello, world!');
+        });
+        it('should handle empty results', async () => {
+            jest.mocked(chromadb_1.Collection.prototype.query).mockResolvedValue({
+                ids: [],
+                embeddings: null,
+                documents: [],
+                metadatas: [],
+                distances: [],
+                included: [],
+            });
+            const documents = await chromaDataStore.query('Hello, world!');
+            expect(documents).toHaveLength(0);
+        });
+    });
+    describe('Upsert', () => {
+        let chromaDataStore;
+        beforeAll(async () => {
+            chromaDataStore = new chroma_data_store_1.ChromaDataStore({
+                collectionName: 'test',
+                url: 'http://localhost:8000',
+                embeddingFunction,
+            });
+            await chromaDataStore.init();
+        });
+        it('should upsert documents', async () => {
+            jest.spyOn(chromadb_1.Collection.prototype, 'upsert');
+            await chromaDataStore.upsert([
+                {
+                    id: '1',
+                    content: 'Hello, world!',
+                    metadata: { title: 'Hello, world!' },
+                },
+                {
+                    id: '2',
+                    content: 'Hello, world! 2',
+                    metadata: { title: 'Hello, world! 2' },
+                },
+            ]);
+            expect(chromadb_1.Collection.prototype.upsert).toHaveBeenCalledWith({
+                ids: ['1', '2'],
+                documents: ['Hello, world!', 'Hello, world! 2'],
+                metadatas: [{ title: 'Hello, world!' }, { title: 'Hello, world! 2' }],
+            });
+        });
+    });
+    describe('Delete', () => {
+        let chromaDataStore;
+        beforeAll(async () => {
+            chromaDataStore = new chroma_data_store_1.ChromaDataStore({
+                collectionName: 'test',
+                url: 'http://localhost:8000',
+                embeddingFunction,
+            });
+            await chromaDataStore.init();
+        });
+        it('should delete documents', async () => {
+            jest.spyOn(chromadb_1.Collection.prototype, 'delete');
+            await chromaDataStore.delete(['1', '2']);
+            expect(chromadb_1.Collection.prototype.delete).toHaveBeenCalledWith({
+                ids: ['1', '2'],
+            });
+        });
+    });
+});

package/dist/chroma/chroma.types.d.ts

@@ -0,0 +1,24 @@
+/**
+ * Chroma-specific document content filter operators
+ */
+export interface IChromaDocumentFilter {
+    $contains?: string;
+    $not_contains?: string;
+}
+/**
+ * Chroma-specific metadata filter operators
+ */
+export type IChromaMetadataFilter = Record<string, string | number | boolean | null | IChromaMetadataOperators>;
+export interface IChromaMetadataOperators {
+    $eq?: unknown;
+    $ne?: unknown;
+    $gt?: number;
+    $gte?: number;
+    $lt?: number;
+    $lte?: number;
+    $in?: unknown[];
+    $nin?: unknown[];
+    $and?: IChromaMetadataFilter[];
+    $or?: IChromaMetadataFilter[];
+}
+//# sourceMappingURL=chroma.types.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"chroma.types.d.ts","sourceRoot":"","sources":["../../src/chroma/chroma.types.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAAG,MAAM,CACxC,MAAM,EACN,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,IAAI,GAAG,wBAAwB,CAC5D,CAAC;AAEF,MAAM,WAAW,wBAAwB;IACvC,GAAG,CAAC,EAAE,OAAO,CAAC;IACd,GAAG,CAAC,EAAE,OAAO,CAAC;IACd,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,GAAG,CAAC,EAAE,OAAO,EAAE,CAAC;IAChB,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC;IACjB,IAAI,CAAC,EAAE,qBAAqB,EAAE,CAAC;IAC/B,GAAG,CAAC,EAAE,qBAAqB,EAAE,CAAC;CAC/B"}

package/dist/chroma/chroma.types.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/dist/chroma/embedding-function.d.ts

@@ -0,0 +1,14 @@
+import { type IEmbeddingFunction } from 'chromadb';
+interface IOpenAIEmbeddingFunctionOptions {
+    openai_api_key: string;
+    openai_model: string;
+    openai_organization_id?: string;
+}
+export declare class OpenAIEmbeddingFunction implements IEmbeddingFunction {
+    private readonly options;
+    private readonly openai;
+    constructor(options: IOpenAIEmbeddingFunctionOptions);
+    generate(texts: string[]): Promise<number[][]>;
+}
+export {};
+//# sourceMappingURL=embedding-function.d.ts.map

package/dist/chroma/embedding-function.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embedding-function.d.ts","sourceRoot":"","sources":["../../src/chroma/embedding-function.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,kBAAkB,EAAE,MAAM,UAAU,CAAC;AAGnD,UAAU,+BAA+B;IACvC,cAAc,EAAE,MAAM,CAAC;IACvB,YAAY,EAAE,MAAM,CAAC;IACrB,sBAAsB,CAAC,EAAE,MAAM,CAAC;CACjC;AAED,qBAAa,uBAAwB,YAAW,kBAAkB;IAEpD,OAAO,CAAC,QAAQ,CAAC,OAAO;IADpC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;gBACH,OAAO,EAAE,+BAA+B;IAO/D,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;CAOrD"}

package/dist/chroma/embedding-function.js

@@ -0,0 +1,26 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OpenAIEmbeddingFunction = void 0;
+const openai_1 = __importDefault(require("openai"));
+class OpenAIEmbeddingFunction {
+    options;
+    openai;
+    constructor(options) {
+        this.options = options;
+        this.openai = new openai_1.default({
+            apiKey: options.openai_api_key,
+            organization: options.openai_organization_id,
+        });
+    }
+    async generate(texts) {
+        const embeddings = await this.openai.embeddings.create({
+            model: this.options.openai_model,
+            input: texts,
+        });
+        return embeddings.data.map((embedding) => embedding.embedding);
+    }
+}
+exports.OpenAIEmbeddingFunction = OpenAIEmbeddingFunction;

package/dist/chroma/index.d.ts

@@ -0,0 +1,3 @@
+export * from './chroma-data-store';
+export * from './chroma.types';
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/chroma/index.ts"],"names":[],"mappings":"AAAA,cAAc,qBAAqB,CAAC;AACpC,cAAc,gBAAgB,CAAC"}

package/dist/chroma/index.js

@@ -0,0 +1,18 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./chroma-data-store"), exports);
+__exportStar(require("./chroma.types"), exports);

package/dist/chroma/utils.d.ts

@@ -0,0 +1,6 @@
+import { type Collection } from 'chromadb';
+import { type IVectorStoreDocument } from 'types/vector-db-data-store';
+type MultiQueryResponse = Awaited<ReturnType<typeof Collection.prototype.query>>;
+export declare const createVectorStoreSearchResult: (result: MultiQueryResponse) => IVectorStoreDocument[];
+export {};
+//# sourceMappingURL=utils.d.ts.map

package/dist/chroma/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/chroma/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,UAAU,EAAE,MAAM,UAAU,CAAC;AAC3C,OAAO,EAAE,KAAK,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;AAEvE,KAAK,kBAAkB,GAAG,OAAO,CAC/B,UAAU,CAAC,OAAO,UAAU,CAAC,SAAS,CAAC,KAAK,CAAC,CAC9C,CAAC;AAEF,eAAO,MAAM,6BAA6B,GACxC,QAAQ,kBAAkB,KACzB,oBAAoB,EAWtB,CAAC"}

package/dist/chroma/utils.js

@@ -0,0 +1,31 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createVectorStoreSearchResult = void 0;
+const createVectorStoreSearchResult = (result) => {
+    const scores = result.distances
+        ? convertCosineDistancesToScores(result.distances)
+        : undefined;
+    const metadatas = result.metadatas.flat().filter(Boolean);
+    return result.documents.flat().map((document, idx) => ({
+        id: result.ids.flat().at(idx)?.toString() ?? '',
+        content: document ?? '',
+        metadata: metadatas.at(idx) ?? {},
+        score: scores?.at(idx),
+    }));
+};
+exports.createVectorStoreSearchResult = createVectorStoreSearchResult;
+/**
+ * Converts a 2D array of cosine distances into a flattened array of similarity scores.
+ * @param distances - A 2D array of cosine distances (e.g., [[1.04, 1.24], [0.8, 1.5]]).
+ * @returns A flattened array of similarity scores (e.g., [0.48, 0.38, 0.6, 0.25]).
+ */
+function convertCosineDistancesToScores(distances) {
+    return distances.flat().map((distance) => {
+        // 1. Validate that the distance is within the valid range [0, 2].
+        if (distance < 0 || distance > 2) {
+            throw new Error(`Cosine distance must be between 0 and 2. Received: ${distance}`);
+        }
+        // 2. Convert cosine distance (d) to similarity score (s): s = 1 - (d / 2).
+        return 1 - distance / 2;
+    });
+}

package/dist/index.d.ts

@@ -0,0 +1,4 @@
+export * from './airtable-store';
+export * from './chroma';
+export * from './types';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,kBAAkB,CAAC;AACjC,cAAc,UAAU,CAAC;AAEzB,cAAc,SAAS,CAAC"}

package/dist/index.js

@@ -0,0 +1,19 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./airtable-store"), exports);
+__exportStar(require("./chroma"), exports);
+__exportStar(require("./types"), exports);

package/dist/types/index.d.ts

@@ -0,0 +1,3 @@
+export * from './structured-data-store';
+export * from './vector-db-data-store';
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/types/index.ts"],"names":[],"mappings":"AAAA,cAAc,yBAAyB,CAAC;AACxC,cAAc,wBAAwB,CAAC"}

package/dist/types/index.js

@@ -0,0 +1,18 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./structured-data-store"), exports);
+__exportStar(require("./vector-db-data-store"), exports);

package/dist/types/structured-data-store.d.ts

@@ -0,0 +1,24 @@
+interface IQueryParams<T> {
+    filterByFormula?: string;
+    maxRecords?: number;
+    pageSize?: number;
+    fields?: (keyof T)[];
+    view?: string;
+}
+export interface IDataStore<T> {
+    getAllRecords: (tableName: string, selectOptions: IQueryParams<T>) => Promise<T[]>;
+    /**
+     * Get a record by its ID or throw an error if not found
+     */
+    getRecord: (tableName: string, recordId: string) => Promise<T>;
+    createRecord: (tableName: string, recordData: T) => Promise<T>;
+    updateRecord: (tableName: string, recordId: string, recordData: T) => Promise<T>;
+    batchUpdateRecords: (tableName: string, records: {
+        id: string;
+        fields: T;
+    }[]) => Promise<T[]>;
+    deleteRecord: (tableName: string, recordId: string) => Promise<T>;
+    getRecordByField: (tableName: string, fieldName: string, fieldValue: string) => Promise<T[]>;
+}
+export {};
+//# sourceMappingURL=structured-data-store.d.ts.map

package/dist/types/structured-data-store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"structured-data-store.d.ts","sourceRoot":"","sources":["../../src/types/structured-data-store.ts"],"names":[],"mappings":"AAAA,UAAU,YAAY,CAAC,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC;IACrB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,UAAU,CAAC,CAAC;IAC3B,aAAa,EAAE,CACb,SAAS,EAAE,MAAM,EACjB,aAAa,EAAE,YAAY,CAAC,CAAC,CAAC,KAC3B,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC;IAClB;;OAEG;IACH,SAAS,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,CAAC,CAAC,CAAC;IAC/D,YAAY,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,KAAK,OAAO,CAAC,CAAC,CAAC,CAAC;IAC/D,YAAY,EAAE,CACZ,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,CAAC,KACV,OAAO,CAAC,CAAC,CAAC,CAAC;IAChB,kBAAkB,EAAE,CAClB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,CAAC,CAAA;KAAE,EAAE,KACjC,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC;IAClB,YAAY,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,CAAC,CAAC,CAAC;IAClE,gBAAgB,EAAE,CAChB,SAAS,EAAE,MAAM,EACjB,SAAS,EAAE,MAAM,EACjB,UAAU,EAAE,MAAM,KACf,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC;CACnB"}

package/dist/types/structured-data-store.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });