langchain 0.0.78 → 0.0.79

package/dist/vectorstores/milvus.js

@@ -59,24 +59,6 @@ export class Milvus extends VectorStore {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "colMgr", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "idxMgr", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "dataMgr", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
         Object.defineProperty(this, "indexParams", {
             enumerable: true,
             configurable: true,
@@ -123,9 +105,6 @@ export class Milvus extends VectorStore {
             throw new Error("Milvus URL address is not provided.");
         }
         this.client = new MilvusClient(url, args.ssl, args.username, args.password);
-        this.colMgr = this.client.collectionManager;
-        this.idxMgr = this.client.indexManager;
-        this.dataMgr = this.client.dataManager;
     }
     async addDocuments(documents) {
         const texts = documents.map(({ pageContent }) => pageContent);
@@ -176,17 +155,17 @@ export class Milvus extends VectorStore {
             });
             insertDatas.push(data);
         }
-        const insertResp = await this.dataMgr.insert({
+        const insertResp = await this.client.insert({
             collection_name: this.collectionName,
             fields_data: insertDatas,
         });
         if (insertResp.status.error_code !== ErrorCode.SUCCESS) {
             throw new Error(`Error inserting data: ${JSON.stringify(insertResp)}`);
         }
-        await this.dataMgr.flushSync({ collection_names: [this.collectionName] });
+        await this.client.flushSync({ collection_names: [this.collectionName] });
     }
     async similaritySearchVectorWithScore(query, k) {
-        const hasColResp = await this.colMgr.hasCollection({
+        const hasColResp = await this.client.hasCollection({
             collection_name: this.collectionName,
         });
         if (hasColResp.status.error_code !== ErrorCode.SUCCESS) {
@@ -196,7 +175,7 @@ export class Milvus extends VectorStore {
             throw new Error(`Collection not found: ${this.collectionName}, please create collection before search.`);
         }
         await this.grabCollectionFields();
-        const loadResp = await this.colMgr.loadCollectionSync({
+        const loadResp = await this.client.loadCollectionSync({
             collection_name: this.collectionName,
         });
         if (loadResp.error_code !== ErrorCode.SUCCESS) {
@@ -204,7 +183,7 @@ export class Milvus extends VectorStore {
         }
         // clone this.field and remove vectorField
         const outputFields = this.fields.filter((field) => field !== this.vectorField);
-        const searchResp = await this.dataMgr.search({
+        const searchResp = await this.client.search({
             collection_name: this.collectionName,
             search_params: {
                 anns_field: this.vectorField,
@@ -243,7 +222,7 @@ export class Milvus extends VectorStore {
         return results;
     }
     async ensureCollection(vectors, documents) {
-        const hasColResp = await this.colMgr.hasCollection({
+        const hasColResp = await this.client.hasCollection({
             collection_name: this.collectionName,
         });
         if (hasColResp.status.error_code !== ErrorCode.SUCCESS) {
@@ -288,7 +267,7 @@ export class Milvus extends VectorStore {
                 this.fields.push(field.name);
             }
         });
-        const createRes = await this.colMgr.createCollection({
+        const createRes = await this.client.createCollection({
             collection_name: this.collectionName,
             fields: fieldList,
         });
@@ -296,7 +275,7 @@ export class Milvus extends VectorStore {
             console.log(createRes);
             throw new Error(`Failed to create collection: ${createRes}`);
         }
-        await this.idxMgr.createIndex({
+        await this.client.createIndex({
             collection_name: this.collectionName,
             field_name: this.vectorField,
             extra_params: this.indexCreateParams,
@@ -312,7 +291,7 @@ export class Milvus extends VectorStore {
             this.fields.length > 0) {
             return;
         }
-        const desc = await this.colMgr.describeCollection({
+        const desc = await this.client.describeCollection({
             collection_name: this.collectionName,
         });
         desc.schema.fields.forEach((field) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langchain",
-  "version": "0.0.78",
+  "version": "0.0.79",
   "description": "Typescript bindings for langchain",
   "type": "module",
   "engines": {
@@ -354,7 +354,7 @@
     "@aws-sdk/client-sagemaker-runtime": "^3.310.0",
     "@clickhouse/client": "^0.0.14",
     "@faker-js/faker": "^7.6.0",
-    "@getmetal/metal-sdk": "^2.0.1",
+    "@getmetal/metal-sdk": "^4.0.0",
     "@huggingface/inference": "^1.5.1",
     "@jest/globals": "^29.5.0",
     "@opensearch-project/opensearch": "^2.2.0",
@@ -373,6 +373,7 @@
     "@types/uuid": "^9",
     "@typescript-eslint/eslint-plugin": "^5.58.0",
     "@typescript-eslint/parser": "^5.58.0",
+    "@zilliz/milvus2-sdk-node": ">=2.2.7",
     "apify-client": "^2.7.1",
     "axios": "^0.26.0",
     "cheerio": "^1.0.0-rc.12",
@@ -426,7 +427,7 @@
     "@tensorflow-models/universal-sentence-encoder": "*",
     "@tensorflow/tfjs-converter": "*",
     "@tensorflow/tfjs-core": "*",
-    "@zilliz/milvus2-sdk-node": "^2.2.0",
+    "@zilliz/milvus2-sdk-node": ">=2.2.7",
     "apify-client": "^2.7.1",
     "axios": "*",
     "cheerio": "^1.0.0-rc.12",