diff --git a/web/src/pages/agent/form/title-chunker-form/index.tsx b/web/src/pages/agent/form/title-chunker-form/index.tsx index 2cff6679f3..5cc8efcd23 100644 --- a/web/src/pages/agent/form/title-chunker-form/index.tsx +++ b/web/src/pages/agent/form/title-chunker-form/index.tsx @@ -69,7 +69,9 @@ export enum TitleChunkerRulesField { Group = 'groupRules', } -export type TitleChunkerFormSchemaType = z.infer; +export type TitleChunkerFormSchemaType = z.infer & { + rules: z.infer; +}; type LevelItemProps = { index: number; diff --git a/web/src/pages/agent/utils.ts b/web/src/pages/agent/utils.ts index bf46a2a120..5ae01936e7 100644 --- a/web/src/pages/agent/utils.ts +++ b/web/src/pages/agent/utils.ts @@ -355,9 +355,9 @@ function transformTokenChunkerParams(params: TokenChunkerFormSchemaType) { function transformTitleChunkerParams(params: TitleChunkerFormSchemaType) { const activeRules = - params.method === TitleChunkerMethod.Group + (params.method === TitleChunkerMethod.Group ? params.groupRules - : params.hierarchyRules; + : params.hierarchyRules) ?? params.rules; const levels = (activeRules || []).map((rule) => transformObjectArrayToPureArray(rule.levels, 'expression'),