@mastra/rag 2.7.3-alpha.2 → 2.8.0

package/dist/index.js

@@ -9,6 +9,7 @@ import { createSimilarityPrompt } from "@mastra/core/relevance";
 import ZeroEntropy from "zeroentropy";
 import { createTool } from "@mastra/core/tools";
 import { embedV1, embedV2, embedV3, embedV4 } from "@mastra/core/vector";
+import { getRequestUser } from "@mastra/core/utils";
 import { AgenticRetrieveStreamCommand, BedrockAgentRuntimeClient, RetrieveCommand } from "@aws-sdk/client-bedrock-agent-runtime";
 // \0rolldown/runtime.js
 var __create$1 = Object.create;
@@ -4847,6 +4848,7 @@ var RecursiveJsonTransformer = class RecursiveJsonTransformer {
 	ensureAscii;
 	convertLists;
 	constructor({ maxSize = 2e3, ensureAscii = false, convertLists = true }) {
+		if (!Number.isFinite(maxSize) || maxSize < 21) throw new RangeError("JSON maxSize must be at least 21");
 		this.maxSize = maxSize;
 		this.ensureAscii = ensureAscii;
 		this.convertLists = convertLists;
@@ -5779,6 +5781,7 @@ const validationSchemas = {
 		disallowedSpecial: setOrAllSchema
 	}).strict(),
 	json: baseChunkOptionsSchema.extend({
+		maxSize: z.number().min(21).optional(),
 		minSize: z.number().positive().optional(),
 		ensureAscii: z.boolean().optional(),
 		convertLists: z.boolean().optional()
@@ -6746,7 +6749,8 @@ async function resolveVectorStore(options, context, logger) {
 		if (typeof vectorStoreOption === "function") {
 			const resolved = await vectorStoreOption({
 				requestContext,
-				mastra
+				mastra,
+				user: getRequestUser(requestContext)
 			});
 			if (!isValidMastraVector(resolved)) {
 				const contextStr = buildContextString(context);
@@ -6850,8 +6854,6 @@ const createGraphRAGTool = (options) => {
 		...defaultGraphOptions,
 		...options.graphOptions || {}
 	};
-	const graphRag = new GraphRAG(graphOptions.dimension, graphOptions.threshold);
-	let isInitialized = false;
 	const inputSchema = options.enableFilter ? filterSchema : z.object(baseSchema).passthrough();
 	return createTool({
 		id: toolId,
@@ -6911,38 +6913,40 @@ const createGraphRAGTool = (options) => {
 					observabilityContext
 				});
 				if (logger) logger.debug("vectorQuerySearch returned results", { count: results.length });
-				if (!isInitialized) {
-					const chunks = results.map((result) => ({
-						text: result?.metadata?.text,
-						metadata: result.metadata ?? {}
-					}));
-					const embeddings = results.map((result) => ({ vector: result.vector || [] }));
-					if (logger) logger.debug("Initializing graph", {
-						chunkCount: chunks.length,
-						embeddingCount: embeddings.length
-					});
-					const buildSpan = parentSpan?.createChildSpan({
-						type: SpanType.GRAPH_ACTION,
-						name: "graph build",
-						input: { nodeCount: chunks.length },
-						attributes: {
-							action: "build",
-							nodeCount: chunks.length,
-							threshold: graphOptions.threshold
-						}
-					});
-					try {
-						graphRag.createGraph(chunks, embeddings);
-					} catch (err) {
-						buildSpan?.error({
-							error: err,
-							endSpan: true
-						});
-						throw err;
+				if (results.length === 0) return {
+					relevantContext: [],
+					sources: []
+				};
+				const graphRag = new GraphRAG(graphOptions.dimension, graphOptions.threshold);
+				const chunks = results.map((result) => ({
+					text: result?.metadata?.text,
+					metadata: result.metadata ?? {}
+				}));
+				const embeddings = results.map((result) => ({ vector: result.vector || [] }));
+				if (logger) logger.debug("Initializing graph", {
+					chunkCount: chunks.length,
+					embeddingCount: embeddings.length
+				});
+				const buildSpan = parentSpan?.createChildSpan({
+					type: SpanType.GRAPH_ACTION,
+					name: "graph build",
+					input: { nodeCount: chunks.length },
+					attributes: {
+						action: "build",
+						nodeCount: chunks.length,
+						threshold: graphOptions.threshold
 					}
-					buildSpan?.end();
-					isInitialized = true;
-				} else if (logger) logger.debug("Graph already initialized, skipping graph construction");
+				});
+				try {
+					graphRag.createGraph(chunks, embeddings);
+				} catch (err) {
+					buildSpan?.error({
+						error: err,
+						endSpan: true
+					});
+					throw err;
+				}
+				buildSpan?.end();
 				const traverseSpan = parentSpan?.createChildSpan({
 					type: SpanType.GRAPH_ACTION,
 					name: "graph traverse",