@mastra/rag 2.7.0-alpha.2 → 2.7.0

package/dist/index.js

@@ -17,7 +17,7 @@ var __getOwnPropDesc$1 = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __commonJSMin = (cb, mod) => () => (mod || (cb((mod = { exports: {} }).exports, mod), cb = null), mod.exports);
+var __commonJSMin = (cb, mod) => () => (mod || (cb((mod = { exports: {} }).exports, mod, mod.exports), cb = null), mod.exports);
 var __copyProps = (to, from, except, desc) => {
 	if (from && typeof from === "object" || typeof from === "function") for (var keys = __getOwnPropNames(from), i = 0, n = keys.length, key; i < n; i++) {
 		key = keys[i];
@@ -28,7 +28,7 @@ var __copyProps = (to, from, except, desc) => {
 	}
 	return to;
 };
-var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create$1(__getProtoOf(mod)) : {}, __copyProps(isNodeMode || !mod || !mod.__esModule ? __defProp$1(target, "default", {
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create$1(__getProtoOf(mod)) : {}, __copyProps(isNodeMode || !mod || !mod.__esModule || !__hasOwnProp.call(mod, "default") ? __defProp$1(target, "default", {
 	value: mod,
 	enumerable: true
 }) : target, mod));
@@ -776,7 +776,6 @@ function createEventSourceParserStream() {
 			case "retry":
 				const parsedRetry = parseInt(value, 10);
 				if (!isNaN(parsedRetry)) retry = parsedRetry;
-				break;
 		}
 	}
 	return new TransformStream({
@@ -1275,16 +1274,14 @@ function convertToOpenAIChatMessages({ prompt, useLegacyFunctionCalling = false,
 				case "text":
 					text += part.text;
 					break;
-				case "tool-call":
-					toolCalls.push({
-						id: part.toolCallId,
-						type: "function",
-						function: {
-							name: part.toolName,
-							arguments: JSON.stringify(part.args)
-						}
-					});
-					break;
+				case "tool-call": toolCalls.push({
+					id: part.toolCallId,
+					type: "function",
+					function: {
+						name: part.toolName,
+						arguments: JSON.stringify(part.args)
+					}
+				});
 			}
 			if (useLegacyFunctionCalling) {
 				if (toolCalls.length > 1) throw new UnsupportedFunctionalityError({ functionality: "useLegacyFunctionCalling with multiple tool calls in one message" });
@@ -2641,14 +2638,12 @@ function convertToOpenAIResponsesMessages({ prompt, systemMessageMode }) {
 						}]
 					});
 					break;
-				case "tool-call":
-					messages.push({
-						type: "function_call",
-						call_id: part.toolCallId,
-						name: part.toolName,
-						arguments: JSON.stringify(part.args)
-					});
-					break;
+				case "tool-call": messages.push({
+					type: "function_call",
+					call_id: part.toolCallId,
+					name: part.toolName,
+					arguments: JSON.stringify(part.args)
+				});
 			}
 			break;
 		case "tool":
@@ -2704,20 +2699,16 @@ function prepareResponsesTools({ mode, strict }) {
 						user_location: tool.args.userLocation
 					});
 					break;
-				default:
-					toolWarnings.push({
-						type: "unsupported-tool",
-						tool
-					});
-					break;
+				default: toolWarnings.push({
+					type: "unsupported-tool",
+					tool
+				});
 			}
 			break;
-		default:
-			toolWarnings.push({
-				type: "unsupported-tool",
-				tool
-			});
-			break;
+		default: toolWarnings.push({
+			type: "unsupported-tool",
+			tool
+		});
 	}
 	if (toolChoice == null) return {
 		tools: openaiTools2,
@@ -3332,19 +3323,21 @@ var OpenAISpeechModel = class {
 			speed,
 			instructions
 		};
-		if (outputFormat) if ([
-			"mp3",
-			"opus",
-			"aac",
-			"flac",
-			"wav",
-			"pcm"
-		].includes(outputFormat)) requestBody.response_format = outputFormat;
-		else warnings.push({
-			type: "unsupported-setting",
-			setting: "outputFormat",
-			details: `Unsupported output format: ${outputFormat}. Using mp3 instead.`
-		});
+		if (outputFormat) {
+			if ([
+				"mp3",
+				"opus",
+				"aac",
+				"flac",
+				"wav",
+				"pcm"
+			].includes(outputFormat)) requestBody.response_format = outputFormat;
+			else warnings.push({
+				type: "unsupported-setting",
+				setting: "outputFormat",
+				details: `Unsupported output format: ${outputFormat}. Using mp3 instead.`
+			});
+		}
 		if (openAIOptions) {
 			const speechModelOptions = {};
 			for (const key in speechModelOptions) {
@@ -4948,21 +4941,22 @@ var RecursiveJsonTransformer = class RecursiveJsonTransformer {
 	*/
 	handlePrimitiveValue(value, key, currentChunk, chunks, fullPath) {
 		const testValue = { [key]: value };
-		if (RecursiveJsonTransformer.jsonSize(testValue) <= this.maxSize) if (RecursiveJsonTransformer.jsonSize({
-			...currentChunk,
-			...testValue
-		}) <= this.maxSize) return {
-			currentChunk: {
+		if (RecursiveJsonTransformer.jsonSize(testValue) <= this.maxSize) {
+			if (RecursiveJsonTransformer.jsonSize({
 				...currentChunk,
 				...testValue
-			},
-			chunks
-		};
-		else return {
-			currentChunk: testValue,
-			chunks: [...chunks, currentChunk]
-		};
-		else if (typeof value === "string") {
+			}) <= this.maxSize) return {
+				currentChunk: {
+					...currentChunk,
+					...testValue
+				},
+				chunks
+			};
+			else return {
+				currentChunk: testValue,
+				chunks: [...chunks, currentChunk]
+			};
+		} else if (typeof value === "string") {
 			const newChunks = this.splitLongString(value).map((chunk) => {
 				return this.createChunk(chunk, fullPath);
 			}).filter((chunk) => RecursiveJsonTransformer.jsonSize(chunk) <= this.maxSize);
@@ -6896,9 +6890,10 @@ const createGraphRAGTool = (options) => {
 	};
 	const graphRag = new GraphRAG(graphOptions.dimension, graphOptions.threshold);
 	let isInitialized = false;
+	const inputSchema = options.enableFilter ? filterSchema : z.object(baseSchema).passthrough();
 	return createTool({
 		id: toolId,
-		inputSchema: options.enableFilter ? filterSchema : z.object(baseSchema).passthrough(),
+		inputSchema,
 		outputSchema: outputSchema$1,
 		description: toolDescription,
 		execute: async (inputData, context) => {
@@ -7042,10 +7037,13 @@ const createGraphRAGTool = (options) => {
 const createVectorQueryTool = (options) => {
 	const { id, description } = options;
 	const storeName = options["vectorStoreName"] ? options.vectorStoreName : "DirectVectorStore";
+	const toolId = id || `VectorQuery ${storeName} ${options.indexName} Tool`;
+	const toolDescription = description || defaultVectorQueryDescription();
+	const inputSchema = options.enableFilter ? filterSchema : z.object(baseSchema).passthrough();
 	return createTool({
-		id: id || `VectorQuery ${storeName} ${options.indexName} Tool`,
-		description: description || defaultVectorQueryDescription(),
-		inputSchema: options.enableFilter ? filterSchema : z.object(baseSchema).passthrough(),
+		id: toolId,
+		description: toolDescription,
+		inputSchema,
 		outputSchema: outputSchema$1,
 		execute: async (inputData, context) => {
 			const { requestContext, mastra, tracingContext } = context || {};