Shun Miyazawa 1 год назад
Родитель
Сommit
16b9a92ab3
1 измененных файлов с 7 добавлено и 1 удалено
  1. 7 1
      apps/app/src/server/service/openai/openai.ts

+ 7 - 1
apps/app/src/server/service/openai/openai.ts

@@ -71,9 +71,9 @@ class OpenaiService implements IOpenaiService {
       }
     });
 
+    const uploadedFileIds = preparedVectorStoreFileRelations.map(data => data.fileIds).flat();
     try {
       // Create vector store file
-      const uploadedFileIds = preparedVectorStoreFileRelations.map(data => data.fileIds).flat();
       const createVectorStoreFileBatchResponse = await this.client.createVectorStoreFileBatch(uploadedFileIds);
       logger.debug('Create vector store file', createVectorStoreFileBatchResponse);
 
@@ -82,6 +82,12 @@ class OpenaiService implements IOpenaiService {
     }
     catch (err) {
       logger.error(err);
+
+      // Delete all uploaded files if createVectorStoreFileBatch fails
+      uploadedFileIds.forEach(async(fileId) => {
+        const deleteFileResponse = await this.client.deleteFile(fileId);
+        logger.debug('Delete vector store file (Due to createVectorStoreFileBatch failure)', deleteFileResponse);
+      });
     }
 
   }