소스 검색

Imprv deleteVectorStoreFile()

Shun Miyazawa 1 년 전
부모
커밋
9dd0f416dc
1개의 변경된 파일2개의 추가작업 그리고 1개의 파일을 삭제
  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;
     }