fix: milvus (#2968)

labring · Oct 22, 2024 · 718108a · 718108a
1 parent 025facb
commit 718108a
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/packages/service/common/vectorStore/milvus/class.ts b/packages/service/common/vectorStore/milvus/class.ts
@@ -24,7 +24,10 @@ export class MilvusCtrl {
 
     global.milvusClient = new MilvusClient({
       address: MILVUS_ADDRESS,
-      token: MILVUS_TOKEN
+      token: MILVUS_TOKEN,
+      loaderOptions: {
+        longs: Function
+      }
     });
 
     addLog.info(`Milvus connected`);
@@ -326,10 +329,19 @@ export class MilvusCtrl {
       id: string;
       teamId: string;
       datasetId: string;
+      int64: {
+        low: bigint;
+        high: bigint;
+        unsigned: boolean;
+      };
     }[];
 
     return rows.map((item) => ({
-      id: String(item.id),
+      id: String({
+        low: BigInt(item.int64.low),
+        high: BigInt(item.int64.high),
+        unsigned: item.int64.unsigned
+      }),
       teamId: item.teamId,
       datasetId: item.datasetId
     }));