@mastra/rag 1.0.6 → 1.0.7-alpha.0

package/src/tools/document-chunker.ts

@@ -3,22 +3,26 @@ import { z } from 'zod';
 
 import type { MDocument, ChunkParams } from '../document';
 
+const DEFAULT_CHUNK_PARAMS = {
+  strategy: 'recursive' as const,
+  maxSize: 512,
+  overlap: 50,
+  separators: ['\n'],
+} satisfies ChunkParams;
+
 export const createDocumentChunkerTool = ({
   doc,
-  params = {
-    strategy: 'recursive',
-    size: 512,
-    overlap: 50,
-    separator: '\n',
-  },
+  params = DEFAULT_CHUNK_PARAMS,
 }: {
   doc: MDocument;
   params?: ChunkParams;
 }): ReturnType<typeof createTool> => {
   return createTool({
-    id: `Document Chunker ${params.strategy} ${params.size}`,
+    id: `Document Chunker ${params.strategy} ${params.maxSize}`,
     inputSchema: z.object({}),
-    description: `Chunks document using ${params.strategy} strategy with size ${params.size} and ${params.overlap} overlap`,
+    description: `Chunks document using ${params.strategy} strategy with maxSize ${params.maxSize} and ${
+      params.overlap || 0
+    } overlap`,
     execute: async () => {
       const chunks = await doc.chunk(params);