@mcampa/ai-context-core 0.0.1-beta.05e8984

package/dist/vectordb/milvus-restful-vectordb.d.ts

@@ -0,0 +1,75 @@
+/**
+ * Milvus RESTful Vector Database Implementation
+ *
+ * This RESTful implementation of Milvus vector database is specifically designed for
+ * environments with strict dependency constraints, e.g. VSCode Extensions, Chrome Extensions, etc.
+ *
+ * The standard Milvus gRPC implementation requires some dependencies and modules
+ * that are not available or restricted in these constrained environments. This RESTful
+ * implementation uses only HTTP requests, making it compatible with them.
+ */
+import { VectorDocument, SearchOptions, VectorSearchResult, VectorDatabase, HybridSearchRequest, HybridSearchOptions, HybridSearchResult } from "./types";
+export interface MilvusRestfulConfig {
+    address?: string;
+    token?: string;
+    username?: string;
+    password?: string;
+    database?: string;
+}
+/**
+ * Milvus Vector Database implementation using REST API
+ * This implementation is designed for environments where gRPC is not available,
+ * such as VSCode extensions or browser environments.
+ */
+export declare class MilvusRestfulVectorDatabase implements VectorDatabase {
+    protected config: MilvusRestfulConfig;
+    private baseUrl;
+    protected initializationPromise: Promise<void>;
+    constructor(config: MilvusRestfulConfig);
+    private initialize;
+    private initializeClient;
+    /**
+     * Resolve address from config or token
+     * Common logic for both gRPC and REST implementations
+     */
+    protected resolveAddress(): Promise<string>;
+    /**
+     * Ensure initialization is complete before method execution
+     */
+    protected ensureInitialized(): Promise<void>;
+    /**
+     * Ensure collection is loaded before search/query operations
+     */
+    protected ensureLoaded(collectionName: string): Promise<void>;
+    /**
+     * Make HTTP request to Milvus REST API
+     */
+    private makeRequest;
+    createCollection(collectionName: string, dimension: number, _description?: string): Promise<void>;
+    /**
+     * Create index for vector field using the Index Create API
+     */
+    private createIndex;
+    /**
+     * Load collection to memory for searching
+     */
+    private loadCollection;
+    dropCollection(collectionName: string): Promise<void>;
+    hasCollection(collectionName: string): Promise<boolean>;
+    listCollections(): Promise<string[]>;
+    insert(collectionName: string, documents: VectorDocument[]): Promise<void>;
+    search(collectionName: string, queryVector: number[], options?: SearchOptions): Promise<VectorSearchResult[]>;
+    delete(collectionName: string, ids: string[]): Promise<void>;
+    query(collectionName: string, filter: string, outputFields: string[], limit?: number): Promise<Record<string, unknown>[]>;
+    createHybridCollection(collectionName: string, dimension: number, _description?: string): Promise<void>;
+    private createHybridIndexes;
+    insertHybrid(collectionName: string, documents: VectorDocument[]): Promise<void>;
+    hybridSearch(collectionName: string, searchRequests: HybridSearchRequest[], options?: HybridSearchOptions): Promise<HybridSearchResult[]>;
+    /**
+     * Check collection limit
+     * Returns true if collection can be created, false if limit exceeded
+     * TODO: Implement proper collection limit checking for REST API
+     */
+    checkCollectionLimit(): Promise<boolean>;
+}
+//# sourceMappingURL=milvus-restful-vectordb.d.ts.map

package/dist/vectordb/milvus-restful-vectordb.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"milvus-restful-vectordb.d.ts","sourceRoot":"","sources":["../../src/vectordb/milvus-restful-vectordb.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,EACL,cAAc,EACd,aAAa,EACb,kBAAkB,EAClB,cAAc,EACd,mBAAmB,EACnB,mBAAmB,EACnB,kBAAkB,EAEnB,MAAM,SAAS,CAAC;AAGjB,MAAM,WAAW,mBAAmB;IAClC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAqDD;;;;GAIG;AACH,qBAAa,2BAA4B,YAAW,cAAc;IAChE,SAAS,CAAC,MAAM,EAAE,mBAAmB,CAAC;IACtC,OAAO,CAAC,OAAO,CAAuB;IACtC,SAAS,CAAC,qBAAqB,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;gBAEnC,MAAM,EAAE,mBAAmB;YAOzB,UAAU;YAKV,gBAAgB;IAe9B;;;OAGG;cACa,cAAc,IAAI,OAAO,CAAC,MAAM,CAAC;IAmBjD;;OAEG;cACa,iBAAiB,IAAI,OAAO,CAAC,IAAI,CAAC;IAOlD;;OAEG;cACa,YAAY,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IA8BnE;;OAEG;YACW,WAAW;IAmDnB,gBAAgB,CACpB,cAAc,EAAE,MAAM,EACtB,SAAS,EAAE,MAAM,EACjB,YAAY,CAAC,EAAE,MAAM,GACpB,OAAO,CAAC,IAAI,CAAC;IAyFhB;;OAEG;YACW,WAAW;IA0BzB;;OAEG;YACW,cAAc;IAgBtB,cAAc,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAkBrD,aAAa,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAsBvD,eAAe,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAiBpC,MAAM,CACV,cAAc,EAAE,MAAM,EACtB,SAAS,EAAE,cAAc,EAAE,GAC1B,OAAO,CAAC,IAAI,CAAC;IAkCV,MAAM,CACV,cAAc,EAAE,MAAM,EACtB,WAAW,EAAE,MAAM,EAAE,EACrB,OAAO,CAAC,EAAE,aAAa,GACtB,OAAO,CAAC,kBAAkB,EAAE,CAAC;IAyF1B,MAAM,CAAC,cAAc,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IA0B5D,KAAK,CACT,cAAc,EAAE,MAAM,EACtB,MAAM,EAAE,MAAM,EACd,YAAY,EAAE,MAAM,EAAE,EACtB,KAAK,CAAC,EAAE,MAAM,GACb,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC;IAqC/B,sBAAsB,CAC1B,cAAc,EAAE,MAAM,EACtB,SAAS,EAAE,MAAM,EACjB,YAAY,CAAC,EAAE,MAAM,GACpB,OAAO,CAAC,IAAI,CAAC;YAoGF,mBAAmB;IA0C3B,YAAY,CAChB,cAAc,EAAE,MAAM,EACtB,SAAS,EAAE,cAAc,EAAE,GAC1B,OAAO,CAAC,IAAI,CAAC;IA4CV,YAAY,CAChB,cAAc,EAAE,MAAM,EACtB,cAAc,EAAE,mBAAmB,EAAE,EACrC,OAAO,CAAC,EAAE,mBAAmB,GAC5B,OAAO,CAAC,kBAAkB,EAAE,CAAC;IAsJhC;;;;OAIG;IACG,oBAAoB,IAAI,OAAO,CAAC,OAAO,CAAC;CAQ/C"}