From 6796f2409c8f598ef2ca0314a0dce7f3f6846834 Mon Sep 17 00:00:00 2001
From: archer <545436317@qq.com>
Date: Mon, 8 Dec 2025 00:49:25 +0800
Subject: [PATCH] index

---
 packages/service/core/dataset/data/schema.ts | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/packages/service/core/dataset/data/schema.ts b/packages/service/core/dataset/data/schema.ts
index d291f3cec..7100d06ce 100644
--- a/packages/service/core/dataset/data/schema.ts
+++ b/packages/service/core/dataset/data/schema.ts
@@ -107,6 +107,16 @@ try {
 
   // Cron clear invalid data
   DatasetDataSchema.index({ updateTime: 1 });
+
+  // fileId index
+  DatasetDataSchema.index(
+    { fileId: 1 },
+    { unique: true, partialFilterExpression: { fileId: { $exists: true } } }
+  );
+  DatasetDataSchema.index(
+    { imageId: 1 },
+    { unique: true, partialFilterExpression: { imageId: { $exists: true } } }
+  );
 } catch (error) {
   console.log(error);
 }