Ver Fonte

Simpler

Shun Miyazawa há 1 ano atrás
pai
commit
93e6e9d142

+ 5 - 5
apps/app/src/server/service/openai/openai-client-delegator.ts

@@ -9,8 +9,6 @@ export default class OpenaiClient {
 
 
   private client: OpenAI | AzureOpenAI;
   private client: OpenAI | AzureOpenAI;
 
 
-  private isOpenai: boolean;
-
   private openaiVectorStoreId: string;
   private openaiVectorStoreId: string;
 
 
   constructor() {
   constructor() {
@@ -25,10 +23,8 @@ export default class OpenaiClient {
       throw new Error('AI_SERVICE_TYPE is missing or contains an invalid value');
       throw new Error('AI_SERVICE_TYPE is missing or contains an invalid value');
     }
     }
 
 
-    this.isOpenai = aiServiceType === serviceType.OPEN_AI;
-
     // Retrieve OpenAI related values from environment variables
     // Retrieve OpenAI related values from environment variables
-    if (this.isOpenai) {
+    if (aiServiceType === serviceType.OPEN_AI) {
       const apiKey = configManager.getConfig('crowi', 'app:openaiApiKey');
       const apiKey = configManager.getConfig('crowi', 'app:openaiApiKey');
       const vectorStoreId = configManager.getConfig('crowi', 'app:openaiVectorStoreId');
       const vectorStoreId = configManager.getConfig('crowi', 'app:openaiVectorStoreId');
 
 
@@ -60,6 +56,10 @@ export default class OpenaiClient {
     return this.client.beta.vectorStores.files.del(this.openaiVectorStoreId, fileId);
     return this.client.beta.vectorStores.files.del(this.openaiVectorStoreId, fileId);
   }
   }
 
 
+  async deleteFile(fileId: string): Promise<OpenAI.Files.FileDeleted> {
+    return this.client.files.del(fileId);
+  }
+
   async uploadAndPoll(files: Uploadable[]): Promise<OpenAI.Beta.VectorStores.FileBatches.VectorStoreFileBatch> {
   async uploadAndPoll(files: Uploadable[]): Promise<OpenAI.Beta.VectorStores.FileBatches.VectorStoreFileBatch> {
     return this.client.beta.vectorStores.fileBatches.uploadAndPoll(this.openaiVectorStoreId, { files });
     return this.client.beta.vectorStores.fileBatches.uploadAndPoll(this.openaiVectorStoreId, { files });
   }
   }