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 ...@@ -3,7 +3,6 @@ import { DatasetSearchModeEnum, DatasetTypeEnum } from '../../../core/dataset/co
import { ApiDatasetServerSchema } from '../../../core/dataset/apiDataset/type'; import { ApiDatasetServerSchema } from '../../../core/dataset/apiDataset/type';
import { ObjectIdSchema } from '../../../common/type/mongo'; import { ObjectIdSchema } from '../../../common/type/mongo';
import { ParentIdSchema } from '../../../common/parentFolder/type'; import { ParentIdSchema } from '../../../common/parentFolder/type';
import { EmbeddingModelItemSchema } from '../../../core/ai/model.schema';
import { import {
ChunkSettingsSchema, ChunkSettingsSchema,
DatasetItemSchema, DatasetItemSchema,
...@@ -132,9 +131,16 @@ export const CreateDatasetWithFilesResponseSchema = z.object({ ...@@ -132,9 +131,16 @@ export const CreateDatasetWithFilesResponseSchema = z.object({
example: '/imgs/dataset/avatar.png', example: '/imgs/dataset/avatar.png',
description: '知识库头像' description: '知识库头像'
}), }),
vectorModel: EmbeddingModelItemSchema.meta({ vectorModel: z
description: '向量模型信息' .object({
}) model: z.string().meta({
example: 'text-embedding-3-small',
description: '向量模型名称'
})
})
.meta({
description: '向量模型选择信息'
})
}); });
export type CreateDatasetWithFilesResponse = z.infer<typeof CreateDatasetWithFilesResponseSchema>; export type CreateDatasetWithFilesResponse = z.infer<typeof CreateDatasetWithFilesResponseSchema>;
......
...@@ -154,7 +154,9 @@ async function handler(req: ApiRequestProps): Promise<CreateDatasetWithFilesResp ...@@ -154,7 +154,9 @@ async function handler(req: ApiRequestProps): Promise<CreateDatasetWithFilesResp
datasetId: dataset._id, datasetId: dataset._id,
name: dataset.name, name: dataset.name,
avatar: dataset.avatar, 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