Selaa lähdekoodia

Imprv deleteVectorStoreFile()

Shun Miyazawa 1 vuosi sitten
vanhempi
sitoutus
9dd0f416dc
1 muutettua tiedostoa jossa 2 lisäystä ja 1 poistoa
  1. 2 1
      apps/app/src/features/openai/server/services/openai.ts

+ 2 - 1
apps/app/src/features/openai/server/services/openai.ts

@@ -215,7 +215,8 @@ class OpenaiService implements IOpenaiService {
 
   async deleteVectorStoreFile(pageId: Types.ObjectId): Promise<void> {
     // Delete vector store file and delete vector store file relation
-    const vectorStoreFileRelation = await VectorStoreFileRelationModel.findOne({ pageId });
+    const vectorStore = await this.getOrCreateVectorStoreForPublicScope();
+    const vectorStoreFileRelation = await VectorStoreFileRelationModel.findOne({ vectorStoreRelationId: vectorStore._id, pageId });
     if (vectorStoreFileRelation == null) {
       return;
     }