Просмотр исходного кода

pdateOrCreateDocument -> upsertVectorStoreFileRelations

Shun Miyazawa 1 год назад
Родитель
Сommit
6336e8f58d

+ 2 - 2
apps/app/src/features/openai/server/models/vector-store-file-relation.ts

@@ -12,7 +12,7 @@ export interface VectorStoreFileRelation {
 interface VectorStoreFileRelationDocument extends VectorStoreFileRelation, Document {}
 
 interface VectorStoreFileRelationModel extends Model<VectorStoreFileRelation> {
-  updateOrCreateDocument(requestData: VectorStoreFileRelation[]): Promise<void>;
+  upsertVectorStoreFileRelations(vectorStoreFileRelations: VectorStoreFileRelation[]): Promise<void>;
 }
 
 export const prepareDocumentData = (pageId: Types.ObjectId, fileId: string, vectorStoreFileRelations: VectorStoreFileRelation[]): VectorStoreFileRelation[] => {
@@ -46,7 +46,7 @@ const schema = new Schema<VectorStoreFileRelationDocument, VectorStoreFileRelati
   }],
 });
 
-schema.statics.updateOrCreateDocument = async function(vectorStoreFileRelations: VectorStoreFileRelation[]): Promise<void> {
+schema.statics.upsertVectorStoreFileRelations = async function(vectorStoreFileRelations: VectorStoreFileRelation[]): Promise<void> {
   await this.bulkWrite(
     vectorStoreFileRelations.map((data) => {
       return {

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

@@ -98,7 +98,7 @@ class OpenaiService implements IOpenaiService {
       logger.debug('Create vector store file', createVectorStoreFileBatchRes);
 
       // Save vector store file relation
-      await VectorStoreFileRelationModel.updateOrCreateDocument(vectorStoreFileRelations);
+      await VectorStoreFileRelationModel.upsertVectorStoreFileRelations(vectorStoreFileRelations);
     }
     catch (err) {
       logger.error(err);