fix: milvus错误处理

IdeaLeap · Oct 20, 2023 · 321c523 · 321c523
1 parent 34bd662
commit 321c523
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "llm-ops",
-  "version": "0.0.10",
+  "version": "0.0.11",
   "description": "高效制作llm workflow的低代码框架",
   "main": "package/index.js",
   "author": "Marlene && IdeaLeap",

diff --git a/package/db/milvus.ts b/package/db/milvus.ts
@@ -151,11 +151,15 @@ export class milvusVectorDB {
   async upload(params: milvusVectorDBUploadSchema) {
     const { fields_data, index, partition_name } = params;
     console.time("Upload time");
-    await this.milvusClient.insert({
+    const insertRes = await this.milvusClient.insert({
       collection_name: this.COLLECTION_NAME,
       fields_data: fields_data,
       partition_name: partition_name || undefined,
     });
+    if (insertRes.status.error_code != "Success") {
+      console.error(insertRes.status.reason);
+      throw new Error(insertRes.status.reason);
+    }
     await this.milvusClient.createIndex({
       collection_name: this.COLLECTION_NAME,
       field_name: index.field_name,
@@ -172,7 +176,7 @@ export class milvusVectorDB {
       this.llm = new LLM({});
     }
     const vector = await this.llm?.embedding(data);
-    if (!vector?.data[0]?.embedding) {
+    if (!vector|| !vector?.data[0] || !vector?.data[0]?.embedding) {
       throw new Error("生成向量失败");
     }
     return vector?.data[0]?.embedding;