Shun Miyazawa 1 год назад
Родитель
Сommit
d6ed29cda9
1 измененных файлов с 3 добавлено и 4 удалено
  1. 3 4
      apps/app/src/features/openai/server/services/openai.ts

+ 3 - 4
apps/app/src/features/openai/server/services/openai.ts

@@ -239,7 +239,7 @@ class OpenaiService implements IOpenaiService {
 
 
   }
   }
 
 
-  private async deleteDeletedVectorStoresWithoutVectorStoreFileRelations(): Promise<void> {
+  private async deleteObsolatedVectorStoreRelations(): Promise<void> {
     const result: VectorStoreDocument[] = await VectorStoreModel.aggregate([
     const result: VectorStoreDocument[] = await VectorStoreModel.aggregate([
       { $match: { isDeleted: true } },
       { $match: { isDeleted: true } },
       {
       {
@@ -257,8 +257,7 @@ class OpenaiService implements IOpenaiService {
       return;
       return;
     }
     }
 
 
-    const idToDelete = result.map(vectorStore => vectorStore._id);
-    await VectorStoreModel.deleteMany({ _id: { $in: idToDelete } });
+    await VectorStoreModel.deleteMany({ _id: { $in: result.map(vectorStore => vectorStore._id) } });
   }
   }
 
 
   async deleteVectorStoreFile(vectorStoreRelationId: Types.ObjectId, pageId: Types.ObjectId, apiCallInterval?: number): Promise<void> {
   async deleteVectorStoreFile(vectorStoreRelationId: Types.ObjectId, pageId: Types.ObjectId, apiCallInterval?: number): Promise<void> {
@@ -298,7 +297,7 @@ class OpenaiService implements IOpenaiService {
 
 
   async deleteObsoleteVectorStoreFile(limit: number, apiCallInterval: number): Promise<void> {
   async deleteObsoleteVectorStoreFile(limit: number, apiCallInterval: number): Promise<void> {
     // Deletes all VectorStore documents that are marked as deleted (isDeleted: true) and have no associated VectorStoreFileRelation documents
     // Deletes all VectorStore documents that are marked as deleted (isDeleted: true) and have no associated VectorStoreFileRelation documents
-    await this.deleteDeletedVectorStoresWithoutVectorStoreFileRelations();
+    await this.deleteObsolatedVectorStoreRelations();
 
 
     // Retrieves all VectorStore documents that are marked as deleted
     // Retrieves all VectorStore documents that are marked as deleted
     const deletedVectorStores = await VectorStoreModel.find({ isDeleted: true });
     const deletedVectorStores = await VectorStoreModel.find({ isDeleted: true });