@zilliz/claude-context-core 0.0.1

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

@@ -0,0 +1,34 @@
+import { VectorDocument, SearchOptions, VectorSearchResult, VectorDatabase } from './types';
+export interface MilvusConfig {
+    address?: string;
+    token?: string;
+    username?: string;
+    password?: string;
+    ssl?: boolean;
+}
+export declare class MilvusVectorDatabase implements VectorDatabase {
+    protected config: MilvusConfig;
+    private client;
+    protected initializationPromise: Promise<void>;
+    constructor(config: MilvusConfig);
+    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>;
+    createCollection(collectionName: string, dimension: number, description?: string): Promise<void>;
+    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, any>[]>;
+}
+//# sourceMappingURL=milvus-vectordb.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"milvus-vectordb.d.ts","sourceRoot":"","sources":["../../src/vectordb/milvus-vectordb.ts"],"names":[],"mappings":"AACA,OAAO,EACH,cAAc,EACd,aAAa,EACb,kBAAkB,EAClB,cAAc,EAEjB,MAAM,SAAS,CAAC;AAGjB,MAAM,WAAW,YAAY;IACzB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,GAAG,CAAC,EAAE,OAAO,CAAC;CACjB;AAwBD,qBAAa,oBAAqB,YAAW,cAAc;IACvD,SAAS,CAAC,MAAM,EAAE,YAAY,CAAC;IAC/B,OAAO,CAAC,MAAM,CAA6B;IAC3C,SAAS,CAAC,qBAAqB,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;gBAEnC,MAAM,EAAE,YAAY;YAOlB,UAAU;YAKV,gBAAgB;IAY9B;;;OAGG;cACa,cAAc,IAAI,OAAO,CAAC,MAAM,CAAC;IAejD;;OAEG;cACa,iBAAiB,IAAI,OAAO,CAAC,IAAI,CAAC;IAO5C,gBAAgB,CAAC,cAAc,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAoFhG,cAAc,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQrD,aAAa,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAUvD,eAAe,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IASpC,MAAM,CAAC,cAAc,EAAE,MAAM,EAAE,SAAS,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAqB1E,MAAM,CAAC,cAAc,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,EAAE,OAAO,CAAC,EAAE,aAAa,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;IA+B7G,MAAM,CAAC,cAAc,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAS5D,KAAK,CAAC,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAC;CA8B9H"}

package/dist/vectordb/milvus-vectordb.js

@@ -0,0 +1,248 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MilvusVectorDatabase = void 0;
+const milvus2_sdk_node_1 = require("@zilliz/milvus2-sdk-node");
+const types_1 = require("./types");
+const zilliz_utils_1 = require("./zilliz-utils");
+/**
+ * Wrapper function to handle collection creation with limit detection for gRPC client
+ * This is the single point where collection limit errors are detected and handled
+ */
+async function createCollectionWithLimitCheck(client, createCollectionParams) {
+    try {
+        await client.createCollection(createCollectionParams);
+    }
+    catch (error) {
+        // Check if the error message contains the collection limit exceeded pattern
+        const errorMessage = error.message || error.toString() || '';
+        if (/exceeded the limit number of collections/i.test(errorMessage)) {
+            // Throw the exact message string, not an Error object
+            throw types_1.COLLECTION_LIMIT_MESSAGE;
+        }
+        // Re-throw other errors as-is
+        throw error;
+    }
+}
+class MilvusVectorDatabase {
+    constructor(config) {
+        this.client = null;
+        this.config = config;
+        // Start initialization asynchronously without waiting
+        this.initializationPromise = this.initialize();
+    }
+    async initialize() {
+        const resolvedAddress = await this.resolveAddress();
+        await this.initializeClient(resolvedAddress);
+    }
+    async initializeClient(address) {
+        const milvusConfig = this.config;
+        console.log('🔌 Connecting to vector database at: ', address);
+        this.client = new milvus2_sdk_node_1.MilvusClient({
+            address: address,
+            username: milvusConfig.username,
+            password: milvusConfig.password,
+            token: milvusConfig.token,
+            ssl: milvusConfig.ssl || false,
+        });
+    }
+    /**
+     * Resolve address from config or token
+     * Common logic for both gRPC and REST implementations
+     */
+    async resolveAddress() {
+        let finalConfig = { ...this.config };
+        // If address is not provided, get it using token
+        if (!finalConfig.address && finalConfig.token) {
+            finalConfig.address = await zilliz_utils_1.ClusterManager.getAddressFromToken(finalConfig.token);
+        }
+        if (!finalConfig.address) {
+            throw new Error('Address is required and could not be resolved from token');
+        }
+        return finalConfig.address;
+    }
+    /**
+     * Ensure initialization is complete before method execution
+     */
+    async ensureInitialized() {
+        await this.initializationPromise;
+        if (!this.client) {
+            throw new Error('Client not initialized');
+        }
+    }
+    async createCollection(collectionName, dimension, description) {
+        await this.ensureInitialized();
+        console.log('Beginning collection creation:', collectionName);
+        console.log('Collection dimension:', dimension);
+        const schema = [
+            {
+                name: 'id',
+                description: 'Document ID',
+                data_type: milvus2_sdk_node_1.DataType.VarChar,
+                max_length: 512,
+                is_primary_key: true,
+            },
+            {
+                name: 'vector',
+                description: 'Embedding vector',
+                data_type: milvus2_sdk_node_1.DataType.FloatVector,
+                dim: dimension,
+            },
+            {
+                name: 'content',
+                description: 'Document content',
+                data_type: milvus2_sdk_node_1.DataType.VarChar,
+                max_length: 65535,
+            },
+            {
+                name: 'relativePath',
+                description: 'Relative path to the codebase',
+                data_type: milvus2_sdk_node_1.DataType.VarChar,
+                max_length: 1024,
+            },
+            {
+                name: 'startLine',
+                description: 'Start line number of the chunk',
+                data_type: milvus2_sdk_node_1.DataType.Int64,
+            },
+            {
+                name: 'endLine',
+                description: 'End line number of the chunk',
+                data_type: milvus2_sdk_node_1.DataType.Int64,
+            },
+            {
+                name: 'fileExtension',
+                description: 'File extension',
+                data_type: milvus2_sdk_node_1.DataType.VarChar,
+                max_length: 32,
+            },
+            {
+                name: 'metadata',
+                description: 'Additional document metadata as JSON string',
+                data_type: milvus2_sdk_node_1.DataType.VarChar,
+                max_length: 65535,
+            },
+        ];
+        const createCollectionParams = {
+            collection_name: collectionName,
+            description: description || `Code context collection: ${collectionName}`,
+            fields: schema,
+        };
+        await createCollectionWithLimitCheck(this.client, createCollectionParams);
+        // Create index
+        const indexParams = {
+            collection_name: collectionName,
+            field_name: 'vector',
+            index_type: 'AUTOINDEX',
+            metric_type: milvus2_sdk_node_1.MetricType.COSINE,
+        };
+        await this.client.createIndex(indexParams);
+        // Load collection to memory
+        await this.client.loadCollection({
+            collection_name: collectionName,
+        });
+        // Verify collection is created correctly
+        await this.client.describeCollection({
+            collection_name: collectionName,
+        });
+    }
+    async dropCollection(collectionName) {
+        await this.ensureInitialized();
+        await this.client.dropCollection({
+            collection_name: collectionName,
+        });
+    }
+    async hasCollection(collectionName) {
+        await this.ensureInitialized();
+        const result = await this.client.hasCollection({
+            collection_name: collectionName,
+        });
+        return Boolean(result.value);
+    }
+    async listCollections() {
+        await this.ensureInitialized();
+        const result = await this.client.showCollections();
+        // Handle the response format - cast to any to avoid type errors
+        const collections = result.collection_names || result.collections || [];
+        return Array.isArray(collections) ? collections : [];
+    }
+    async insert(collectionName, documents) {
+        await this.ensureInitialized();
+        console.log('Inserting documents into collection:', collectionName);
+        const data = documents.map(doc => ({
+            id: doc.id,
+            vector: doc.vector,
+            content: doc.content,
+            relativePath: doc.relativePath,
+            startLine: doc.startLine,
+            endLine: doc.endLine,
+            fileExtension: doc.fileExtension,
+            metadata: JSON.stringify(doc.metadata),
+        }));
+        await this.client.insert({
+            collection_name: collectionName,
+            data: data,
+        });
+    }
+    async search(collectionName, queryVector, options) {
+        await this.ensureInitialized();
+        const searchParams = {
+            collection_name: collectionName,
+            data: [queryVector],
+            limit: options?.topK || 10,
+            output_fields: ['id', 'content', 'relativePath', 'startLine', 'endLine', 'fileExtension', 'metadata'],
+        };
+        const searchResult = await this.client.search(searchParams);
+        if (!searchResult.results || searchResult.results.length === 0) {
+            return [];
+        }
+        return searchResult.results.map((result) => ({
+            document: {
+                id: result.id,
+                vector: queryVector,
+                content: result.content,
+                relativePath: result.relativePath,
+                startLine: result.startLine,
+                endLine: result.endLine,
+                fileExtension: result.fileExtension,
+                metadata: JSON.parse(result.metadata || '{}'),
+            },
+            score: result.score,
+        }));
+    }
+    async delete(collectionName, ids) {
+        await this.ensureInitialized();
+        await this.client.delete({
+            collection_name: collectionName,
+            filter: `id in [${ids.map(id => `"${id}"`).join(', ')}]`,
+        });
+    }
+    async query(collectionName, filter, outputFields, limit) {
+        await this.ensureInitialized();
+        try {
+            const queryParams = {
+                collection_name: collectionName,
+                filter: filter,
+                output_fields: outputFields,
+            };
+            // Add limit if provided, or default for empty filter expressions
+            if (limit !== undefined) {
+                queryParams.limit = limit;
+            }
+            else if (filter === '' || filter.trim() === '') {
+                // Milvus requires limit when using empty expressions
+                queryParams.limit = 16384; // Default limit for empty filters
+            }
+            const result = await this.client.query(queryParams);
+            if (result.status.error_code !== 'Success') {
+                throw new Error(`Failed to query Milvus: ${result.status.reason}`);
+            }
+            return result.data || [];
+        }
+        catch (error) {
+            console.error(`❌ Failed to query collection '${collectionName}':`, error);
+            throw error;
+        }
+    }
+}
+exports.MilvusVectorDatabase = MilvusVectorDatabase;
+//# sourceMappingURL=milvus-vectordb.js.map

package/dist/vectordb/milvus-vectordb.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"milvus-vectordb.js","sourceRoot":"","sources":["../../src/vectordb/milvus-vectordb.ts"],"names":[],"mappings":";;;AAAA,+DAA8E;AAC9E,mCAMiB;AACjB,iDAAgD;AAUhD;;;GAGG;AACH,KAAK,UAAU,8BAA8B,CACzC,MAAoB,EACpB,sBAA2B;IAE3B,IAAI,CAAC;QACD,MAAM,MAAM,CAAC,gBAAgB,CAAC,sBAAsB,CAAC,CAAC;IAC1D,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QAClB,4EAA4E;QAC5E,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC;QAC7D,IAAI,2CAA2C,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;YACjE,sDAAsD;YACtD,MAAM,gCAAwB,CAAC;QACnC,CAAC;QACD,8BAA8B;QAC9B,MAAM,KAAK,CAAC;IAChB,CAAC;AACL,CAAC;AAED,MAAa,oBAAoB;IAK7B,YAAY,MAAoB;QAHxB,WAAM,GAAwB,IAAI,CAAC;QAIvC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAErB,sDAAsD;QACtD,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;IACnD,CAAC;IAEO,KAAK,CAAC,UAAU;QACpB,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QACpD,MAAM,IAAI,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC;IACjD,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,OAAe;QAC1C,MAAM,YAAY,GAAG,IAAI,CAAC,MAAsB,CAAC;QACjD,OAAO,CAAC,GAAG,CAAC,uCAAuC,EAAE,OAAO,CAAC,CAAC;QAC9D,IAAI,CAAC,MAAM,GAAG,IAAI,+BAAY,CAAC;YAC3B,OAAO,EAAE,OAAO;YAChB,QAAQ,EAAE,YAAY,CAAC,QAAQ;YAC/B,QAAQ,EAAE,YAAY,CAAC,QAAQ;YAC/B,KAAK,EAAE,YAAY,CAAC,KAAK;YACzB,GAAG,EAAE,YAAY,CAAC,GAAG,IAAI,KAAK;SACjC,CAAC,CAAC;IACP,CAAC;IAED;;;OAGG;IACO,KAAK,CAAC,cAAc;QAC1B,IAAI,WAAW,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QAErC,iDAAiD;QACjD,IAAI,CAAC,WAAW,CAAC,OAAO,IAAI,WAAW,CAAC,KAAK,EAAE,CAAC;YAC5C,WAAW,CAAC,OAAO,GAAG,MAAM,6BAAc,CAAC,mBAAmB,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QACtF,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;QAChF,CAAC;QAED,OAAO,WAAW,CAAC,OAAO,CAAC;IAC/B,CAAC;IAED;;OAEG;IACO,KAAK,CAAC,iBAAiB;QAC7B,MAAM,IAAI,CAAC,qBAAqB,CAAC;QACjC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAC9C,CAAC;IACL,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,cAAsB,EAAE,SAAiB,EAAE,WAAoB;QAClF,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,OAAO,CAAC,GAAG,CAAC,gCAAgC,EAAE,cAAc,CAAC,CAAC;QAC9D,OAAO,CAAC,GAAG,CAAC,uBAAuB,EAAE,SAAS,CAAC,CAAC;QAChD,MAAM,MAAM,GAAG;YACX;gBACI,IAAI,EAAE,IAAI;gBACV,WAAW,EAAE,aAAa;gBAC1B,SAAS,EAAE,2BAAQ,CAAC,OAAO;gBAC3B,UAAU,EAAE,GAAG;gBACf,cAAc,EAAE,IAAI;aACvB;YACD;gBACI,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,kBAAkB;gBAC/B,SAAS,EAAE,2BAAQ,CAAC,WAAW;gBAC/B,GAAG,EAAE,SAAS;aACjB;YACD;gBACI,IAAI,EAAE,SAAS;gBACf,WAAW,EAAE,kBAAkB;gBAC/B,SAAS,EAAE,2BAAQ,CAAC,OAAO;gBAC3B,UAAU,EAAE,KAAK;aACpB;YACD;gBACI,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,+BAA+B;gBAC5C,SAAS,EAAE,2BAAQ,CAAC,OAAO;gBAC3B,UAAU,EAAE,IAAI;aACnB;YACD;gBACI,IAAI,EAAE,WAAW;gBACjB,WAAW,EAAE,gCAAgC;gBAC7C,SAAS,EAAE,2BAAQ,CAAC,KAAK;aAC5B;YACD;gBACI,IAAI,EAAE,SAAS;gBACf,WAAW,EAAE,8BAA8B;gBAC3C,SAAS,EAAE,2BAAQ,CAAC,KAAK;aAC5B;YACD;gBACI,IAAI,EAAE,eAAe;gBACrB,WAAW,EAAE,gBAAgB;gBAC7B,SAAS,EAAE,2BAAQ,CAAC,OAAO;gBAC3B,UAAU,EAAE,EAAE;aACjB;YACD;gBACI,IAAI,EAAE,UAAU;gBAChB,WAAW,EAAE,6CAA6C;gBAC1D,SAAS,EAAE,2BAAQ,CAAC,OAAO;gBAC3B,UAAU,EAAE,KAAK;aACpB;SACJ,CAAC;QAEF,MAAM,sBAAsB,GAAG;YAC3B,eAAe,EAAE,cAAc;YAC/B,WAAW,EAAE,WAAW,IAAI,4BAA4B,cAAc,EAAE;YACxE,MAAM,EAAE,MAAM;SACjB,CAAC;QAEF,MAAM,8BAA8B,CAAC,IAAI,CAAC,MAAO,EAAE,sBAAsB,CAAC,CAAC;QAE3E,eAAe;QACf,MAAM,WAAW,GAAG;YAChB,eAAe,EAAE,cAAc;YAC/B,UAAU,EAAE,QAAQ;YACpB,UAAU,EAAE,WAAW;YACvB,WAAW,EAAE,6BAAU,CAAC,MAAM;SACjC,CAAC;QAEF,MAAM,IAAI,CAAC,MAAO,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;QAE5C,4BAA4B;QAC5B,MAAM,IAAI,CAAC,MAAO,CAAC,cAAc,CAAC;YAC9B,eAAe,EAAE,cAAc;SAClC,CAAC,CAAC;QAEH,yCAAyC;QACzC,MAAM,IAAI,CAAC,MAAO,CAAC,kBAAkB,CAAC;YAClC,eAAe,EAAE,cAAc;SAClC,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,cAAsB;QACvC,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,MAAM,IAAI,CAAC,MAAO,CAAC,cAAc,CAAC;YAC9B,eAAe,EAAE,cAAc;SAClC,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,cAAsB;QACtC,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAO,CAAC,aAAa,CAAC;YAC5C,eAAe,EAAE,cAAc;SAClC,CAAC,CAAC;QAEH,OAAO,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,eAAe;QACjB,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAO,CAAC,eAAe,EAAE,CAAC;QACpD,gEAAgE;QAChE,MAAM,WAAW,GAAI,MAAc,CAAC,gBAAgB,IAAK,MAAc,CAAC,WAAW,IAAI,EAAE,CAAC;QAC1F,OAAO,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC;IACzD,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,cAAsB,EAAE,SAA2B;QAC5D,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,OAAO,CAAC,GAAG,CAAC,sCAAsC,EAAE,cAAc,CAAC,CAAC;QACpE,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC/B,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,OAAO,EAAE,GAAG,CAAC,OAAO;YACpB,YAAY,EAAE,GAAG,CAAC,YAAY;YAC9B,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,OAAO,EAAE,GAAG,CAAC,OAAO;YACpB,aAAa,EAAE,GAAG,CAAC,aAAa;YAChC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC;SACzC,CAAC,CAAC,CAAC;QAEJ,MAAM,IAAI,CAAC,MAAO,CAAC,MAAM,CAAC;YACtB,eAAe,EAAE,cAAc;YAC/B,IAAI,EAAE,IAAI;SACb,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,cAAsB,EAAE,WAAqB,EAAE,OAAuB;QAC/E,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,MAAM,YAAY,GAAG;YACjB,eAAe,EAAE,cAAc;YAC/B,IAAI,EAAE,CAAC,WAAW,CAAC;YACnB,KAAK,EAAE,OAAO,EAAE,IAAI,IAAI,EAAE;YAC1B,aAAa,EAAE,CAAC,IAAI,EAAE,SAAS,EAAE,cAAc,EAAE,WAAW,EAAE,SAAS,EAAE,eAAe,EAAE,UAAU,CAAC;SACxG,CAAC;QAEF,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,MAAO,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAE7D,IAAI,CAAC,YAAY,CAAC,OAAO,IAAI,YAAY,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7D,OAAO,EAAE,CAAC;QACd,CAAC;QAED,OAAO,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAW,EAAE,EAAE,CAAC,CAAC;YAC9C,QAAQ,EAAE;gBACN,EAAE,EAAE,MAAM,CAAC,EAAE;gBACb,MAAM,EAAE,WAAW;gBACnB,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,YAAY,EAAE,MAAM,CAAC,YAAY;gBACjC,SAAS,EAAE,MAAM,CAAC,SAAS;gBAC3B,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,aAAa,EAAE,MAAM,CAAC,aAAa;gBACnC,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,IAAI,IAAI,CAAC;aAChD;YACD,KAAK,EAAE,MAAM,CAAC,KAAK;SACtB,CAAC,CAAC,CAAC;IACR,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,cAAsB,EAAE,GAAa;QAC9C,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,MAAM,IAAI,CAAC,MAAO,CAAC,MAAM,CAAC;YACtB,eAAe,EAAE,cAAc;YAC/B,MAAM,EAAE,UAAU,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG;SAC3D,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,cAAsB,EAAE,MAAc,EAAE,YAAsB,EAAE,KAAc;QACtF,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,IAAI,CAAC;YACD,MAAM,WAAW,GAAQ;gBACrB,eAAe,EAAE,cAAc;gBAC/B,MAAM,EAAE,MAAM;gBACd,aAAa,EAAE,YAAY;aAC9B,CAAC;YAEF,iEAAiE;YACjE,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACtB,WAAW,CAAC,KAAK,GAAG,KAAK,CAAC;YAC9B,CAAC;iBAAM,IAAI,MAAM,KAAK,EAAE,IAAI,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;gBAC/C,qDAAqD;gBACrD,WAAW,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC,kCAAkC;YACjE,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAO,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAErD,IAAI,MAAM,CAAC,MAAM,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;gBACzC,MAAM,IAAI,KAAK,CAAC,2BAA2B,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;YACvE,CAAC;YAED,OAAO,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;QAC7B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,iCAAiC,cAAc,IAAI,EAAE,KAAK,CAAC,CAAC;YAC1E,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;CACJ;AApQD,oDAoQC"}

package/dist/vectordb/types.d.ts

@@ -0,0 +1,75 @@
+export interface VectorDocument {
+    id: string;
+    vector: number[];
+    content: string;
+    relativePath: string;
+    startLine: number;
+    endLine: number;
+    fileExtension: string;
+    metadata: Record<string, any>;
+}
+export interface SearchOptions {
+    topK?: number;
+    filter?: Record<string, any>;
+    threshold?: number;
+}
+export interface VectorSearchResult {
+    document: VectorDocument;
+    score: number;
+}
+export interface VectorDatabase {
+    /**
+     * Create collection
+     * @param collectionName Collection name
+     * @param dimension Vector dimension
+     * @param description Collection description
+     */
+    createCollection(collectionName: string, dimension: number, description?: string): Promise<void>;
+    /**
+     * Drop collection
+     * @param collectionName Collection name
+     */
+    dropCollection(collectionName: string): Promise<void>;
+    /**
+     * Check if collection exists
+     * @param collectionName Collection name
+     */
+    hasCollection(collectionName: string): Promise<boolean>;
+    /**
+     * List all collections
+     */
+    listCollections(): Promise<string[]>;
+    /**
+     * Insert vector documents
+     * @param collectionName Collection name
+     * @param documents Document array
+     */
+    insert(collectionName: string, documents: VectorDocument[]): Promise<void>;
+    /**
+     * Search similar vectors
+     * @param collectionName Collection name
+     * @param queryVector Query vector
+     * @param options Search options
+     */
+    search(collectionName: string, queryVector: number[], options?: SearchOptions): Promise<VectorSearchResult[]>;
+    /**
+     * Delete documents
+     * @param collectionName Collection name
+     * @param ids Document ID array
+     */
+    delete(collectionName: string, ids: string[]): Promise<void>;
+    /**
+     * Query documents by filter
+     * @param collectionName Collection name
+     * @param filter Filter expression string
+     * @param outputFields Fields to return
+     * @param limit Maximum number of results to return (optional)
+     */
+    query(collectionName: string, filter: string, outputFields: string[], limit?: number): Promise<Record<string, any>[]>;
+}
+/**
+ * Special error message for collection limit exceeded
+ * This allows us to distinguish it from other errors across all Milvus implementations
+ */
+export declare const COLLECTION_LIMIT_MESSAGE = "[Error]: Your Zilliz Cloud account has hit its collection limit. To continue creating collections, you'll need to expand your capacity. We recommend visiting https://zilliz.com/pricing to explore options for dedicated or serverless clusters.";
+//# sourceMappingURL=types.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/vectordb/types.ts"],"names":[],"mappings":"AACA,MAAM,WAAW,cAAc;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,aAAa,EAAE,MAAM,CAAC;IACtB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CACjC;AAED,MAAM,WAAW,aAAa;IAC1B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC7B,SAAS,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,kBAAkB;IAC/B,QAAQ,EAAE,cAAc,CAAC;IACzB,KAAK,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,cAAc;IAC3B;;;;;OAKG;IACH,gBAAgB,CAAC,cAAc,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAEjG;;;OAGG;IACH,cAAc,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAEtD;;;OAGG;IACH,aAAa,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAExD;;OAEG;IACH,eAAe,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAErC;;;;OAIG;IACH,MAAM,CAAC,cAAc,EAAE,MAAM,EAAE,SAAS,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAE3E;;;;;OAKG;IACH,MAAM,CAAC,cAAc,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,EAAE,OAAO,CAAC,EAAE,aAAa,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAAC;IAE9G;;;;OAIG;IACH,MAAM,CAAC,cAAc,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAE7D;;;;;;OAMG;IACH,KAAK,CAAC,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;CACzH;AAED;;;GAGG;AACH,eAAO,MAAM,wBAAwB,sPAAsP,CAAC"}

package/dist/vectordb/types.js

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.COLLECTION_LIMIT_MESSAGE = void 0;
+/**
+ * Special error message for collection limit exceeded
+ * This allows us to distinguish it from other errors across all Milvus implementations
+ */
+exports.COLLECTION_LIMIT_MESSAGE = "[Error]: Your Zilliz Cloud account has hit its collection limit. To continue creating collections, you'll need to expand your capacity. We recommend visiting https://zilliz.com/pricing to explore options for dedicated or serverless clusters.";
+//# sourceMappingURL=types.js.map

package/dist/vectordb/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/vectordb/types.ts"],"names":[],"mappings":";;;AAiFA;;;GAGG;AACU,QAAA,wBAAwB,GAAG,mPAAmP,CAAC"}

package/dist/vectordb/zilliz-utils.d.ts

@@ -0,0 +1,135 @@
+export interface ZillizConfig {
+    baseUrl?: string;
+    token?: string;
+}
+export interface Project {
+    projectId: string;
+    projectName: string;
+    instanceCount: number;
+    createTime: string;
+}
+export interface Cluster {
+    clusterId: string;
+    clusterName: string;
+    description: string;
+    regionId: string;
+    plan: string;
+    cuType: string;
+    cuSize: number;
+    status: string;
+    connectAddress: string;
+    privateLinkAddress: string;
+    projectId: string;
+    createTime: string;
+}
+export interface CreateFreeClusterRequest {
+    clusterName: string;
+    projectId: string;
+    regionId: string;
+}
+export interface CreateFreeClusterResponse {
+    clusterId: string;
+    username: string;
+    password: string;
+    prompt: string;
+}
+export interface CreateFreeClusterWithDetailsResponse extends CreateFreeClusterResponse {
+    clusterDetails: DescribeClusterResponse;
+}
+export interface ListProjectsResponse {
+    code: number;
+    data: Project[];
+}
+export interface ListClustersResponse {
+    code: number;
+    data: {
+        count: number;
+        currentPage: number;
+        pageSize: number;
+        clusters: Cluster[];
+    };
+}
+export interface CreateFreeClusterApiResponse {
+    code: number;
+    data: CreateFreeClusterResponse;
+}
+export interface DescribeClusterResponse {
+    clusterId: string;
+    clusterName: string;
+    projectId: string;
+    description: string;
+    regionId: string;
+    cuType: string;
+    plan: string;
+    status: string;
+    connectAddress: string;
+    privateLinkAddress: string;
+    createTime: string;
+    cuSize: number;
+    storageSize: number;
+    snapshotNumber: number;
+    createProgress: number;
+}
+export interface DescribeClusterApiResponse {
+    code: number;
+    data: DescribeClusterResponse;
+}
+export interface ErrorResponse {
+    code: number;
+    message: string;
+}
+/**
+ * Zilliz Cloud cluster manager
+ * For managing Zilliz Cloud projects and clusters
+ * See https://docs.zilliz.com/reference/restful/list-clusters-v2 for more details
+ */
+export declare class ClusterManager {
+    private baseUrl;
+    private token;
+    constructor(config?: ZillizConfig);
+    /**
+     * Generic method for sending HTTP requests
+     */
+    private makeRequest;
+    /**
+     * List all projects
+     * @returns List of projects
+     */
+    listProjects(): Promise<Project[]>;
+    /**
+     * List all clusters
+     * @param projectId Optional project ID filter
+     * @param pageSize Page size, default 10
+     * @param currentPage Current page number, default 1
+     * @returns Cluster list with pagination info
+     */
+    listClusters(projectId?: string, pageSize?: number, currentPage?: number): Promise<{
+        clusters: Cluster[];
+        count: number;
+        currentPage: number;
+        pageSize: number;
+    }>;
+    /**
+ * Describe cluster details
+ * @param clusterId Cluster ID to describe
+ * @returns Cluster details
+ */
+    describeCluster(clusterId: string): Promise<DescribeClusterResponse>;
+    /**
+ * Create free cluster and wait for it to be ready
+ * @param request Request parameters for creating cluster
+ * @param timeoutMs Timeout in milliseconds, default 5 minutes
+ * @param pollIntervalMs Polling interval in milliseconds, default 5 seconds
+ * @returns Creation result including cluster ID, username, password and cluster details
+ */
+    createFreeCluster(request: CreateFreeClusterRequest, timeoutMs?: number, // 5 minutes default
+    pollIntervalMs?: number): Promise<CreateFreeClusterWithDetailsResponse>;
+    /**
+     * Static utility method to get address from token using Zilliz Cloud API
+     * This method will find or create a cluster and return its connect address
+     * @param token Zilliz Cloud API token
+     * @returns Connect address for the cluster
+     */
+    static getAddressFromToken(token?: string): Promise<string>;
+}
+//# sourceMappingURL=zilliz-utils.d.ts.map

package/dist/vectordb/zilliz-utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"zilliz-utils.d.ts","sourceRoot":"","sources":["../../src/vectordb/zilliz-utils.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,YAAY;IACzB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,OAAO;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,OAAO;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,cAAc,EAAE,MAAM,CAAC;IACvB,kBAAkB,EAAE,MAAM,CAAC;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,wBAAwB;IACrC,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,yBAAyB;IACtC,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,oCAAqC,SAAQ,yBAAyB;IACnF,cAAc,EAAE,uBAAuB,CAAC;CAC3C;AAED,MAAM,WAAW,oBAAoB;IACjC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,OAAO,EAAE,CAAC;CACnB;AAED,MAAM,WAAW,oBAAoB;IACjC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE;QACF,KAAK,EAAE,MAAM,CAAC;QACd,WAAW,EAAE,MAAM,CAAC;QACpB,QAAQ,EAAE,MAAM,CAAC;QACjB,QAAQ,EAAE,OAAO,EAAE,CAAC;KACvB,CAAC;CACL;AAED,MAAM,WAAW,4BAA4B;IACzC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,yBAAyB,CAAC;CACnC;AAED,MAAM,WAAW,uBAAuB;IACpC,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,cAAc,EAAE,MAAM,CAAC;IACvB,kBAAkB,EAAE,MAAM,CAAC;IAC3B,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;IACpB,cAAc,EAAE,MAAM,CAAC;IACvB,cAAc,EAAE,MAAM,CAAC;CAC1B;AAED,MAAM,WAAW,0BAA0B;IACvC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,uBAAuB,CAAC;CACjC;AAED,MAAM,WAAW,aAAa;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;CACnB;AAED;;;;GAIG;AACH,qBAAa,cAAc;IACvB,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,KAAK,CAAS;gBAEV,MAAM,CAAC,EAAE,YAAY;IAUjC;;OAEG;YACW,WAAW;IA4CzB;;;OAGG;IACG,YAAY,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;IAUxC;;;;;;OAMG;IACG,YAAY,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,QAAQ,GAAE,MAAW,EAAE,WAAW,GAAE,MAAU,GAAG,OAAO,CAAC;QAC5F,QAAQ,EAAE,OAAO,EAAE,CAAC;QACpB,KAAK,EAAE,MAAM,CAAC;QACd,WAAW,EAAE,MAAM,CAAC;QACpB,QAAQ,EAAE,MAAM,CAAC;KACpB,CAAC;IAeF;;;;GAID;IACO,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,uBAAuB,CAAC;IAU1E;;;;;;GAMD;IACO,iBAAiB,CACnB,OAAO,EAAE,wBAAwB,EACjC,SAAS,GAAE,MAAsB,EAAE,oBAAoB;IACvD,cAAc,GAAE,MAAiB,GAClC,OAAO,CAAC,oCAAoC,CAAC;IA4ChD;;;;;OAKG;WACU,mBAAmB,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;CAwCpE"}