Commit 04c50f71 by light5980 Committed by GitHub

debug: workflow create knowledge base error (#7050)

parent 1d16842b
......@@ -3,7 +3,6 @@ import { DatasetSearchModeEnum, DatasetTypeEnum } from '../../../core/dataset/co
import { ApiDatasetServerSchema } from '../../../core/dataset/apiDataset/type';
import { ObjectIdSchema } from '../../../common/type/mongo';
import { ParentIdSchema } from '../../../common/parentFolder/type';
import { EmbeddingModelItemSchema } from '../../../core/ai/model.schema';
import {
ChunkSettingsSchema,
DatasetItemSchema,
......@@ -132,9 +131,16 @@ export const CreateDatasetWithFilesResponseSchema = z.object({
example: '/imgs/dataset/avatar.png',
description: '知识库头像'
}),
vectorModel: EmbeddingModelItemSchema.meta({
description: '向量模型信息'
})
vectorModel: z
.object({
model: z.string().meta({
example: 'text-embedding-3-small',
description: '向量模型名称'
})
})
.meta({
description: '向量模型选择信息'
})
});
export type CreateDatasetWithFilesResponse = z.infer<typeof CreateDatasetWithFilesResponseSchema>;
......
......@@ -154,7 +154,9 @@ async function handler(req: ApiRequestProps): Promise<CreateDatasetWithFilesResp
datasetId: dataset._id,
name: dataset.name,
avatar: dataset.avatar,
vectorModel: getEmbeddingModel(dataset.vectorModel)
vectorModel: {
model: getEmbeddingModel(dataset.vectorModel)?.model || dataset.vectorModel
}
};
});
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment