@langchain/core 1.2.0-dev-1765937705265 → 1.2.0

package/dist/utils/testing/stream.d.cts

@@ -0,0 +1,96 @@
+//#region src/utils/testing/stream.d.ts
+/**
+ * The `this` context that Vitest provides to custom matchers via `expect.extend`.
+ * @see https://vitest.dev/guide/extending-matchers.html
+ */
+interface ExpectExtendThis {
+  isNot?: boolean;
+  equals(a: unknown, b: unknown): boolean;
+  utils: {
+    matcherHint(name: string, received?: string, expected?: string, options?: {
+      isNot?: boolean;
+    }): string;
+    printReceived(value: unknown): string;
+    printExpected(value: unknown): string;
+  };
+}
+interface ExpectationResult {
+  pass: boolean;
+  message: () => string;
+  actual?: unknown;
+  expected?: unknown;
+}
+declare function toHaveStreamText(this: ExpectExtendThis, received: unknown, expected: string): Promise<ExpectationResult>;
+declare function toHaveStreamReasoning(this: ExpectExtendThis, received: unknown, expected: string): Promise<ExpectationResult>;
+type StreamToolCallExpectation = {
+  name: string;
+  args: Record<string, unknown>;
+};
+type StreamUsageExpectation = {
+  input_tokens?: number;
+  output_tokens?: number;
+  total_tokens?: number;
+  input_token_details?: Record<string, unknown>;
+  output_token_details?: Record<string, unknown>;
+};
+type StreamOutputExpectation = {
+  id?: string;
+  text?: string;
+  toolCalls?: StreamToolCallExpectation[];
+  usage?: StreamUsageExpectation;
+  responseMetadata?: Record<string, unknown>;
+};
+declare function toHaveStreamToolCalls(this: ExpectExtendThis, received: unknown, expected: StreamToolCallExpectation[]): Promise<ExpectationResult>;
+declare function toHaveStreamUsage(this: ExpectExtendThis, received: unknown, expected: StreamUsageExpectation): Promise<ExpectationResult>;
+declare function toHaveStreamOutput(this: ExpectExtendThis, received: unknown, expected: StreamOutputExpectation): Promise<ExpectationResult>;
+/** Stream matchers for `expect.extend()`. */
+declare const streamMatchers: {
+  toHaveStreamText: typeof toHaveStreamText;
+  toHaveStreamReasoning: typeof toHaveStreamReasoning;
+  toHaveStreamToolCalls: typeof toHaveStreamToolCalls;
+  toHaveStreamUsage: typeof toHaveStreamUsage;
+  toHaveStreamOutput: typeof toHaveStreamOutput;
+};
+/**
+ * Custom assertion helpers for values returned by `BaseChatModel.streamEvents()`.
+ *
+ * These matchers consume the stream lazily through the corresponding
+ * `ChatModelStream` promise-backed properties.
+ *
+ * @typeParam R - The assertion return type provided by the test framework.
+ */
+interface StreamMatchers<R = unknown> {
+  /**
+   * Asserts that the stream resolves to the expected concatenated text.
+   *
+   * @param expected - The exact text expected from `ChatModelStream.text`.
+   */
+  toHaveStreamText(expected: string): R;
+  /**
+   * Asserts that the stream resolves to the expected concatenated reasoning text.
+   *
+   * @param expected - The exact reasoning text expected from `ChatModelStream.reasoning`.
+   */
+  toHaveStreamReasoning(expected: string): R;
+  /**
+   * Asserts that the stream resolves to the expected ordered tool calls.
+   *
+   * @param expected - Tool call names and arguments expected from `ChatModelStream.toolCalls`.
+   */
+  toHaveStreamToolCalls(expected: StreamToolCallExpectation[]): R;
+  /**
+   * Asserts that the stream resolves to usage metadata matching the expected fields.
+   *
+   * @param expected - A partial usage metadata object expected from `ChatModelStream.usage`.
+   */
+  toHaveStreamUsage(expected: StreamUsageExpectation): R;
+  /**
+   * Asserts that the final streamed output message matches the expected fields.
+   *
+   * @param expected - A partial output expectation checked against `ChatModelStream.output`.
+   */
+  toHaveStreamOutput(expected: StreamOutputExpectation): R;
+}
+//#endregion
+export { StreamMatchers, StreamOutputExpectation, StreamToolCallExpectation, StreamUsageExpectation, streamMatchers, toHaveStreamOutput, toHaveStreamReasoning, toHaveStreamText, toHaveStreamToolCalls, toHaveStreamUsage };
+//# sourceMappingURL=stream.d.cts.map

package/dist/utils/testing/stream.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream.d.cts","names":[],"sources":["../../../src/utils/testing/stream.ts"],"mappings":";;;;;UAQU,gBAAA;EACR,KAAA;EACA,MAAA,CAAO,CAAA,WAAY,CAAA;EACnB,KAAA;IACE,WAAA,CACE,IAAA,UACA,QAAA,WACA,QAAA,WACA,OAAA;MAAY,KAAA;IAAA;IAEd,aAAA,CAAc,KAAA;IACd,aAAA,CAAc,KAAA;EAAA;AAAA;AAAA,UAIR,iBAAA;EACR,IAAA;EACA,OAAA;EACA,MAAA;EACA,QAAA;AAAA;AAAA,iBAkHoB,gBAAA,CACpB,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,QAAA,WACC,OAAA,CAAQ,iBAAA;AAAA,iBAsBW,qBAAA,CACpB,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,QAAA,WACC,OAAA,CAAQ,iBAAA;AAAA,KAsBC,yBAAA;EACV,IAAA;EACA,IAAA,EAAM,MAAA;AAAA;AAAA,KAGI,sBAAA;EACV,YAAA;EACA,aAAA;EACA,YAAA;EACA,mBAAA,GAAsB,MAAA;EACtB,oBAAA,GAAuB,MAAA;AAAA;AAAA,KAGb,uBAAA;EACV,EAAA;EACA,IAAA;EACA,SAAA,GAAY,yBAAA;EACZ,KAAA,GAAQ,sBAAA;EACR,gBAAA,GAAmB,MAAA;AAAA;AAAA,iBAGC,qBAAA,CACpB,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,QAAA,EAAU,yBAAA,KACT,OAAA,CAAQ,iBAAA;AAAA,iBA+BW,iBAAA,CACpB,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,QAAA,EAAU,sBAAA,GACT,OAAA,CAAQ,iBAAA;AAAA,iBAyBW,kBAAA,CACpB,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,QAAA,EAAU,uBAAA,GACT,OAAA,CAAQ,iBAAA;;cAmCE,cAAA;;;;;;;;;;;;;;;UAgBI,cAAA;EAlKN;;;AAsBX;;EAkJE,gBAAA,CAAiB,QAAA,WAAmB,CAAA;EAhJxB;;;;;EAuJZ,qBAAA,CAAsB,QAAA,WAAmB,CAAA;EApJ/B;;;;;EA2JV,qBAAA,CAAsB,QAAA,EAAU,yBAAA,KAA8B,CAAA;EAxJ9D;;;;;EA+JA,iBAAA,CAAkB,QAAA,EAAU,sBAAA,GAAyB,CAAA;EA7JxB;AAG/B;;;;EAiKE,kBAAA,CAAmB,QAAA,EAAU,uBAAA,GAA0B,CAAA;AAAA"}

package/dist/utils/testing/stream.d.ts

@@ -0,0 +1,96 @@
+//#region src/utils/testing/stream.d.ts
+/**
+ * The `this` context that Vitest provides to custom matchers via `expect.extend`.
+ * @see https://vitest.dev/guide/extending-matchers.html
+ */
+interface ExpectExtendThis {
+  isNot?: boolean;
+  equals(a: unknown, b: unknown): boolean;
+  utils: {
+    matcherHint(name: string, received?: string, expected?: string, options?: {
+      isNot?: boolean;
+    }): string;
+    printReceived(value: unknown): string;
+    printExpected(value: unknown): string;
+  };
+}
+interface ExpectationResult {
+  pass: boolean;
+  message: () => string;
+  actual?: unknown;
+  expected?: unknown;
+}
+declare function toHaveStreamText(this: ExpectExtendThis, received: unknown, expected: string): Promise<ExpectationResult>;
+declare function toHaveStreamReasoning(this: ExpectExtendThis, received: unknown, expected: string): Promise<ExpectationResult>;
+type StreamToolCallExpectation = {
+  name: string;
+  args: Record<string, unknown>;
+};
+type StreamUsageExpectation = {
+  input_tokens?: number;
+  output_tokens?: number;
+  total_tokens?: number;
+  input_token_details?: Record<string, unknown>;
+  output_token_details?: Record<string, unknown>;
+};
+type StreamOutputExpectation = {
+  id?: string;
+  text?: string;
+  toolCalls?: StreamToolCallExpectation[];
+  usage?: StreamUsageExpectation;
+  responseMetadata?: Record<string, unknown>;
+};
+declare function toHaveStreamToolCalls(this: ExpectExtendThis, received: unknown, expected: StreamToolCallExpectation[]): Promise<ExpectationResult>;
+declare function toHaveStreamUsage(this: ExpectExtendThis, received: unknown, expected: StreamUsageExpectation): Promise<ExpectationResult>;
+declare function toHaveStreamOutput(this: ExpectExtendThis, received: unknown, expected: StreamOutputExpectation): Promise<ExpectationResult>;
+/** Stream matchers for `expect.extend()`. */
+declare const streamMatchers: {
+  toHaveStreamText: typeof toHaveStreamText;
+  toHaveStreamReasoning: typeof toHaveStreamReasoning;
+  toHaveStreamToolCalls: typeof toHaveStreamToolCalls;
+  toHaveStreamUsage: typeof toHaveStreamUsage;
+  toHaveStreamOutput: typeof toHaveStreamOutput;
+};
+/**
+ * Custom assertion helpers for values returned by `BaseChatModel.streamEvents()`.
+ *
+ * These matchers consume the stream lazily through the corresponding
+ * `ChatModelStream` promise-backed properties.
+ *
+ * @typeParam R - The assertion return type provided by the test framework.
+ */
+interface StreamMatchers<R = unknown> {
+  /**
+   * Asserts that the stream resolves to the expected concatenated text.
+   *
+   * @param expected - The exact text expected from `ChatModelStream.text`.
+   */
+  toHaveStreamText(expected: string): R;
+  /**
+   * Asserts that the stream resolves to the expected concatenated reasoning text.
+   *
+   * @param expected - The exact reasoning text expected from `ChatModelStream.reasoning`.
+   */
+  toHaveStreamReasoning(expected: string): R;
+  /**
+   * Asserts that the stream resolves to the expected ordered tool calls.
+   *
+   * @param expected - Tool call names and arguments expected from `ChatModelStream.toolCalls`.
+   */
+  toHaveStreamToolCalls(expected: StreamToolCallExpectation[]): R;
+  /**
+   * Asserts that the stream resolves to usage metadata matching the expected fields.
+   *
+   * @param expected - A partial usage metadata object expected from `ChatModelStream.usage`.
+   */
+  toHaveStreamUsage(expected: StreamUsageExpectation): R;
+  /**
+   * Asserts that the final streamed output message matches the expected fields.
+   *
+   * @param expected - A partial output expectation checked against `ChatModelStream.output`.
+   */
+  toHaveStreamOutput(expected: StreamOutputExpectation): R;
+}
+//#endregion
+export { StreamMatchers, StreamOutputExpectation, StreamToolCallExpectation, StreamUsageExpectation, streamMatchers, toHaveStreamOutput, toHaveStreamReasoning, toHaveStreamText, toHaveStreamToolCalls, toHaveStreamUsage };
+//# sourceMappingURL=stream.d.ts.map

package/dist/utils/testing/stream.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream.d.ts","names":[],"sources":["../../../src/utils/testing/stream.ts"],"mappings":";;;;;UAQU,gBAAA;EACR,KAAA;EACA,MAAA,CAAO,CAAA,WAAY,CAAA;EACnB,KAAA;IACE,WAAA,CACE,IAAA,UACA,QAAA,WACA,QAAA,WACA,OAAA;MAAY,KAAA;IAAA;IAEd,aAAA,CAAc,KAAA;IACd,aAAA,CAAc,KAAA;EAAA;AAAA;AAAA,UAIR,iBAAA;EACR,IAAA;EACA,OAAA;EACA,MAAA;EACA,QAAA;AAAA;AAAA,iBAkHoB,gBAAA,CACpB,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,QAAA,WACC,OAAA,CAAQ,iBAAA;AAAA,iBAsBW,qBAAA,CACpB,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,QAAA,WACC,OAAA,CAAQ,iBAAA;AAAA,KAsBC,yBAAA;EACV,IAAA;EACA,IAAA,EAAM,MAAA;AAAA;AAAA,KAGI,sBAAA;EACV,YAAA;EACA,aAAA;EACA,YAAA;EACA,mBAAA,GAAsB,MAAA;EACtB,oBAAA,GAAuB,MAAA;AAAA;AAAA,KAGb,uBAAA;EACV,EAAA;EACA,IAAA;EACA,SAAA,GAAY,yBAAA;EACZ,KAAA,GAAQ,sBAAA;EACR,gBAAA,GAAmB,MAAA;AAAA;AAAA,iBAGC,qBAAA,CACpB,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,QAAA,EAAU,yBAAA,KACT,OAAA,CAAQ,iBAAA;AAAA,iBA+BW,iBAAA,CACpB,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,QAAA,EAAU,sBAAA,GACT,OAAA,CAAQ,iBAAA;AAAA,iBAyBW,kBAAA,CACpB,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,QAAA,EAAU,uBAAA,GACT,OAAA,CAAQ,iBAAA;;cAmCE,cAAA;;;;;;;;;;;;;;;UAgBI,cAAA;EAlKN;;;AAsBX;;EAkJE,gBAAA,CAAiB,QAAA,WAAmB,CAAA;EAhJxB;;;;;EAuJZ,qBAAA,CAAsB,QAAA,WAAmB,CAAA;EApJ/B;;;;;EA2JV,qBAAA,CAAsB,QAAA,EAAU,yBAAA,KAA8B,CAAA;EAxJ9D;;;;;EA+JA,iBAAA,CAAkB,QAAA,EAAU,sBAAA,GAAyB,CAAA;EA7JxB;AAG/B;;;;EAiKE,kBAAA,CAAmB,QAAA,EAAU,uBAAA,GAA0B,CAAA;AAAA"}

package/dist/utils/testing/stream.js

@@ -0,0 +1,136 @@
+//#region src/utils/testing/stream.ts
+function isChatModelStream(received) {
+	if (received == null || typeof received !== "object") return false;
+	const stream = received;
+	return typeof stream.text !== "undefined" && typeof stream.toolCalls !== "undefined" && typeof stream.reasoning !== "undefined" && typeof stream.usage !== "undefined" && typeof stream.output !== "undefined" && typeof stream[Symbol.asyncIterator] === "function";
+}
+function matchesPartialObject(actual, expected, equals) {
+	if (actual == null) return false;
+	return Object.entries(expected).every(([key, value]) => equals(actual[key], value));
+}
+function matchesStreamUsage(actual, expected, equals) {
+	if (actual == null) return false;
+	return matchesPartialObject(actual, expected, equals);
+}
+function getOutputText(message) {
+	return message.content.find((block) => block.type === "text")?.text;
+}
+function matchesStreamOutput(message, expected, equals) {
+	if (expected.id !== void 0 && message.id !== expected.id) return false;
+	if (expected.text !== void 0 && getOutputText(message) !== expected.text) return false;
+	if (expected.toolCalls !== void 0) {
+		const calls = message.tool_calls ?? [];
+		if (calls.length !== expected.toolCalls.length) return false;
+		for (let i = 0; i < expected.toolCalls.length; i++) {
+			const call = calls[i];
+			const exp = expected.toolCalls[i];
+			if (call?.name !== exp.name || !equals(call.args, exp.args)) return false;
+		}
+	}
+	if (expected.usage !== void 0 && !matchesStreamUsage(message.usage_metadata, expected.usage, equals)) return false;
+	if (expected.responseMetadata !== void 0 && !matchesPartialObject(message.response_metadata, expected.responseMetadata, equals)) return false;
+	return true;
+}
+function invalidStreamResult(received, matcherName, utils) {
+	return {
+		pass: false,
+		message: () => `${utils.matcherHint(matcherName)}\n\nExpected: ChatModelStream (return value of model.streamEvents("Hello"))\nReceived: ${utils.printReceived(received)}`,
+		actual: received,
+		expected: "ChatModelStream"
+	};
+}
+function applyNot(pass, isNot) {
+	return isNot ? !pass : pass;
+}
+async function toHaveStreamText(received, expected) {
+	const { isNot, utils } = this;
+	const matcherName = "toHaveStreamText";
+	if (!isChatModelStream(received)) return invalidStreamResult(received, matcherName, utils);
+	const actual = await received.text;
+	return {
+		pass: applyNot(actual === expected, isNot),
+		message: () => `${utils.matcherHint(matcherName, void 0, void 0, { isNot })}\n\nExpected stream text: ${isNot ? "not " : ""}${utils.printExpected(expected)}\nReceived stream text: ${utils.printReceived(actual)}`,
+		actual,
+		expected
+	};
+}
+async function toHaveStreamReasoning(received, expected) {
+	const { isNot, utils } = this;
+	const matcherName = "toHaveStreamReasoning";
+	if (!isChatModelStream(received)) return invalidStreamResult(received, matcherName, utils);
+	const actual = await received.reasoning;
+	return {
+		pass: applyNot(actual === expected, isNot),
+		message: () => `${utils.matcherHint(matcherName, void 0, void 0, { isNot })}\n\nExpected stream reasoning: ${isNot ? "not " : ""}${utils.printExpected(expected)}\nReceived stream reasoning: ${utils.printReceived(actual)}`,
+		actual,
+		expected
+	};
+}
+async function toHaveStreamToolCalls(received, expected) {
+	const { isNot, utils } = this;
+	const matcherName = "toHaveStreamToolCalls";
+	if (!isChatModelStream(received)) return invalidStreamResult(received, matcherName, utils);
+	const actual = await received.toolCalls;
+	let pass = actual.length === expected.length && expected.every((exp, i) => {
+		const call = actual[i];
+		return call?.name === exp.name && this.equals(call.args, exp.args);
+	});
+	pass = applyNot(pass, isNot);
+	return {
+		pass,
+		message: () => `${utils.matcherHint(matcherName, void 0, void 0, { isNot })}\n\nExpected stream tool calls: ${utils.printExpected(expected)}\nReceived stream tool calls: ${utils.printReceived(actual.map((tc) => ({
+			name: tc.name,
+			args: tc.args
+		})))}`,
+		actual: actual.map((tc) => ({
+			name: tc.name,
+			args: tc.args
+		})),
+		expected
+	};
+}
+async function toHaveStreamUsage(received, expected) {
+	const { isNot, utils } = this;
+	const matcherName = "toHaveStreamUsage";
+	if (!isChatModelStream(received)) return invalidStreamResult(received, matcherName, utils);
+	const actual = await received.usage;
+	return {
+		pass: applyNot(matchesStreamUsage(actual, expected, this.equals), isNot),
+		message: () => `${utils.matcherHint(matcherName, void 0, void 0, { isNot })}\n\nExpected stream usage: ${utils.printExpected(expected)}\nReceived stream usage: ${utils.printReceived(actual)}`,
+		actual,
+		expected
+	};
+}
+async function toHaveStreamOutput(received, expected) {
+	const { isNot, utils } = this;
+	const matcherName = "toHaveStreamOutput";
+	if (!isChatModelStream(received)) return invalidStreamResult(received, matcherName, utils);
+	const message = await received.output;
+	return {
+		pass: applyNot(matchesStreamOutput(message, expected, this.equals), isNot),
+		message: () => `${utils.matcherHint(matcherName, void 0, void 0, { isNot })}\n\nExpected stream output: ${utils.printExpected(expected)}\nReceived stream output: ${utils.printReceived({
+			id: message.id,
+			text: getOutputText(message),
+			tool_calls: message.tool_calls?.map((tc) => ({
+				name: tc.name,
+				args: tc.args
+			})),
+			usage_metadata: message.usage_metadata,
+			response_metadata: message.response_metadata
+		})}`,
+		actual: message,
+		expected
+	};
+}
+/** Stream matchers for `expect.extend()`. */
+const streamMatchers = {
+	toHaveStreamText,
+	toHaveStreamReasoning,
+	toHaveStreamToolCalls,
+	toHaveStreamUsage,
+	toHaveStreamOutput
+};
+//#endregion
+export { streamMatchers };
+
+//# sourceMappingURL=stream.js.map

package/dist/utils/testing/stream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream.js","names":[],"sources":["../../../src/utils/testing/stream.ts"],"sourcesContent":["import type { AIMessage } from \"../../messages/index.js\";\nimport type { UsageMetadata } from \"../../messages/metadata.js\";\nimport { ChatModelStream } from \"../../language_models/stream.js\";\n\n/**\n * The `this` context that Vitest provides to custom matchers via `expect.extend`.\n * @see https://vitest.dev/guide/extending-matchers.html\n */\ninterface ExpectExtendThis {\n  isNot?: boolean;\n  equals(a: unknown, b: unknown): boolean;\n  utils: {\n    matcherHint(\n      name: string,\n      received?: string,\n      expected?: string,\n      options?: { isNot?: boolean }\n    ): string;\n    printReceived(value: unknown): string;\n    printExpected(value: unknown): string;\n  };\n}\n\ninterface ExpectationResult {\n  pass: boolean;\n  message: () => string;\n  actual?: unknown;\n  expected?: unknown;\n}\n\nfunction isChatModelStream(received: unknown): received is ChatModelStream {\n  if (received == null || typeof received !== \"object\") {\n    return false;\n  }\n  const stream = received as ChatModelStream;\n  return (\n    typeof stream.text !== \"undefined\" &&\n    typeof stream.toolCalls !== \"undefined\" &&\n    typeof stream.reasoning !== \"undefined\" &&\n    typeof stream.usage !== \"undefined\" &&\n    typeof stream.output !== \"undefined\" &&\n    typeof stream[Symbol.asyncIterator] === \"function\"\n  );\n}\n\nfunction matchesPartialObject(\n  actual: Record<string, unknown> | undefined,\n  expected: Record<string, unknown>,\n  equals: ExpectExtendThis[\"equals\"]\n): boolean {\n  if (actual == null) {\n    return false;\n  }\n  return Object.entries(expected).every(([key, value]) =>\n    equals(actual[key], value)\n  );\n}\n\nfunction matchesStreamUsage(\n  actual: UsageMetadata | undefined,\n  expected: StreamUsageExpectation,\n  equals: ExpectExtendThis[\"equals\"]\n): boolean {\n  if (actual == null) {\n    return false;\n  }\n  return matchesPartialObject(\n    actual as Record<string, unknown>,\n    expected as Record<string, unknown>,\n    equals\n  );\n}\n\nfunction getOutputText(message: AIMessage): string | undefined {\n  const content = message.content as Array<{ type: string; text?: string }>;\n  return content.find((block) => block.type === \"text\")?.text;\n}\n\nfunction matchesStreamOutput(\n  message: AIMessage,\n  expected: StreamOutputExpectation,\n  equals: ExpectExtendThis[\"equals\"]\n): boolean {\n  if (expected.id !== undefined && message.id !== expected.id) {\n    return false;\n  }\n  if (expected.text !== undefined && getOutputText(message) !== expected.text) {\n    return false;\n  }\n  if (expected.toolCalls !== undefined) {\n    const calls = message.tool_calls ?? [];\n    if (calls.length !== expected.toolCalls.length) {\n      return false;\n    }\n    for (let i = 0; i < expected.toolCalls.length; i++) {\n      const call = calls[i];\n      const exp = expected.toolCalls[i]!;\n      if (call?.name !== exp.name || !equals(call.args, exp.args)) {\n        return false;\n      }\n    }\n  }\n  if (\n    expected.usage !== undefined &&\n    !matchesStreamUsage(message.usage_metadata, expected.usage, equals)\n  ) {\n    return false;\n  }\n  if (\n    expected.responseMetadata !== undefined &&\n    !matchesPartialObject(\n      message.response_metadata as Record<string, unknown>,\n      expected.responseMetadata,\n      equals\n    )\n  ) {\n    return false;\n  }\n  return true;\n}\n\nfunction invalidStreamResult(\n  received: unknown,\n  matcherName: string,\n  utils: ExpectExtendThis[\"utils\"]\n): ExpectationResult {\n  return {\n    pass: false,\n    message: () =>\n      `${utils.matcherHint(matcherName)}\\n\\n` +\n      `Expected: ChatModelStream (return value of model.streamEvents(\"Hello\"))\\n` +\n      `Received: ${utils.printReceived(received)}`,\n    actual: received,\n    expected: \"ChatModelStream\",\n  };\n}\n\nfunction applyNot(pass: boolean, isNot?: boolean): boolean {\n  return isNot ? !pass : pass;\n}\n\nexport async function toHaveStreamText(\n  this: ExpectExtendThis,\n  received: unknown,\n  expected: string\n): Promise<ExpectationResult> {\n  const { isNot, utils } = this;\n  const matcherName = \"toHaveStreamText\";\n\n  if (!isChatModelStream(received)) {\n    return invalidStreamResult(received, matcherName, utils);\n  }\n\n  const actual = await received.text;\n  const pass = applyNot(actual === expected, isNot);\n\n  return {\n    pass,\n    message: () =>\n      `${utils.matcherHint(matcherName, undefined, undefined, { isNot })}\\n\\n` +\n      `Expected stream text: ${isNot ? \"not \" : \"\"}${utils.printExpected(expected)}\\n` +\n      `Received stream text: ${utils.printReceived(actual)}`,\n    actual,\n    expected,\n  };\n}\n\nexport async function toHaveStreamReasoning(\n  this: ExpectExtendThis,\n  received: unknown,\n  expected: string\n): Promise<ExpectationResult> {\n  const { isNot, utils } = this;\n  const matcherName = \"toHaveStreamReasoning\";\n\n  if (!isChatModelStream(received)) {\n    return invalidStreamResult(received, matcherName, utils);\n  }\n\n  const actual = await received.reasoning;\n  const pass = applyNot(actual === expected, isNot);\n\n  return {\n    pass,\n    message: () =>\n      `${utils.matcherHint(matcherName, undefined, undefined, { isNot })}\\n\\n` +\n      `Expected stream reasoning: ${isNot ? \"not \" : \"\"}${utils.printExpected(expected)}\\n` +\n      `Received stream reasoning: ${utils.printReceived(actual)}`,\n    actual,\n    expected,\n  };\n}\n\nexport type StreamToolCallExpectation = {\n  name: string;\n  args: Record<string, unknown>;\n};\n\nexport type StreamUsageExpectation = {\n  input_tokens?: number;\n  output_tokens?: number;\n  total_tokens?: number;\n  input_token_details?: Record<string, unknown>;\n  output_token_details?: Record<string, unknown>;\n};\n\nexport type StreamOutputExpectation = {\n  id?: string;\n  text?: string;\n  toolCalls?: StreamToolCallExpectation[];\n  usage?: StreamUsageExpectation;\n  responseMetadata?: Record<string, unknown>;\n};\n\nexport async function toHaveStreamToolCalls(\n  this: ExpectExtendThis,\n  received: unknown,\n  expected: StreamToolCallExpectation[]\n): Promise<ExpectationResult> {\n  const { isNot, utils } = this;\n  const matcherName = \"toHaveStreamToolCalls\";\n\n  if (!isChatModelStream(received)) {\n    return invalidStreamResult(received, matcherName, utils);\n  }\n\n  const actual = await received.toolCalls;\n\n  let pass =\n    actual.length === expected.length &&\n    expected.every((exp, i) => {\n      const call = actual[i];\n      return call?.name === exp.name && this.equals(call.args, exp.args);\n    });\n  pass = applyNot(pass, isNot);\n\n  return {\n    pass,\n    message: () =>\n      `${utils.matcherHint(matcherName, undefined, undefined, { isNot })}\\n\\n` +\n      `Expected stream tool calls: ${utils.printExpected(expected)}\\n` +\n      `Received stream tool calls: ${utils.printReceived(\n        actual.map((tc) => ({ name: tc.name, args: tc.args }))\n      )}`,\n    actual: actual.map((tc) => ({ name: tc.name, args: tc.args })),\n    expected,\n  };\n}\n\nexport async function toHaveStreamUsage(\n  this: ExpectExtendThis,\n  received: unknown,\n  expected: StreamUsageExpectation\n): Promise<ExpectationResult> {\n  const { isNot, utils } = this;\n  const matcherName = \"toHaveStreamUsage\";\n\n  if (!isChatModelStream(received)) {\n    return invalidStreamResult(received, matcherName, utils);\n  }\n\n  const actual = await received.usage;\n  const pass = applyNot(\n    matchesStreamUsage(actual, expected, this.equals),\n    isNot\n  );\n\n  return {\n    pass,\n    message: () =>\n      `${utils.matcherHint(matcherName, undefined, undefined, { isNot })}\\n\\n` +\n      `Expected stream usage: ${utils.printExpected(expected)}\\n` +\n      `Received stream usage: ${utils.printReceived(actual)}`,\n    actual,\n    expected,\n  };\n}\n\nexport async function toHaveStreamOutput(\n  this: ExpectExtendThis,\n  received: unknown,\n  expected: StreamOutputExpectation\n): Promise<ExpectationResult> {\n  const { isNot, utils } = this;\n  const matcherName = \"toHaveStreamOutput\";\n\n  if (!isChatModelStream(received)) {\n    return invalidStreamResult(received, matcherName, utils);\n  }\n\n  const message = await received.output;\n  const pass = applyNot(\n    matchesStreamOutput(message, expected, this.equals),\n    isNot\n  );\n\n  return {\n    pass,\n    message: () =>\n      `${utils.matcherHint(matcherName, undefined, undefined, { isNot })}\\n\\n` +\n      `Expected stream output: ${utils.printExpected(expected)}\\n` +\n      `Received stream output: ${utils.printReceived({\n        id: message.id,\n        text: getOutputText(message),\n        tool_calls: message.tool_calls?.map((tc) => ({\n          name: tc.name,\n          args: tc.args,\n        })),\n        usage_metadata: message.usage_metadata,\n        response_metadata: message.response_metadata,\n      })}`,\n    actual: message,\n    expected,\n  };\n}\n\n/** Stream matchers for `expect.extend()`. */\nexport const streamMatchers = {\n  toHaveStreamText,\n  toHaveStreamReasoning,\n  toHaveStreamToolCalls,\n  toHaveStreamUsage,\n  toHaveStreamOutput,\n};\n\n/**\n * Custom assertion helpers for values returned by `BaseChatModel.streamEvents()`.\n *\n * These matchers consume the stream lazily through the corresponding\n * `ChatModelStream` promise-backed properties.\n *\n * @typeParam R - The assertion return type provided by the test framework.\n */\nexport interface StreamMatchers<R = unknown> {\n  /**\n   * Asserts that the stream resolves to the expected concatenated text.\n   *\n   * @param expected - The exact text expected from `ChatModelStream.text`.\n   */\n  toHaveStreamText(expected: string): R;\n\n  /**\n   * Asserts that the stream resolves to the expected concatenated reasoning text.\n   *\n   * @param expected - The exact reasoning text expected from `ChatModelStream.reasoning`.\n   */\n  toHaveStreamReasoning(expected: string): R;\n\n  /**\n   * Asserts that the stream resolves to the expected ordered tool calls.\n   *\n   * @param expected - Tool call names and arguments expected from `ChatModelStream.toolCalls`.\n   */\n  toHaveStreamToolCalls(expected: StreamToolCallExpectation[]): R;\n\n  /**\n   * Asserts that the stream resolves to usage metadata matching the expected fields.\n   *\n   * @param expected - A partial usage metadata object expected from `ChatModelStream.usage`.\n   */\n  toHaveStreamUsage(expected: StreamUsageExpectation): R;\n\n  /**\n   * Asserts that the final streamed output message matches the expected fields.\n   *\n   * @param expected - A partial output expectation checked against `ChatModelStream.output`.\n   */\n  toHaveStreamOutput(expected: StreamOutputExpectation): R;\n}\n"],"mappings":";AA8BA,SAAS,kBAAkB,UAAgD;AACzE,KAAI,YAAY,QAAQ,OAAO,aAAa,SAC1C,QAAO;CAET,MAAM,SAAS;AACf,QACE,OAAO,OAAO,SAAS,eACvB,OAAO,OAAO,cAAc,eAC5B,OAAO,OAAO,cAAc,eAC5B,OAAO,OAAO,UAAU,eACxB,OAAO,OAAO,WAAW,eACzB,OAAO,OAAO,OAAO,mBAAmB;;AAI5C,SAAS,qBACP,QACA,UACA,QACS;AACT,KAAI,UAAU,KACZ,QAAO;AAET,QAAO,OAAO,QAAQ,SAAS,CAAC,OAAO,CAAC,KAAK,WAC3C,OAAO,OAAO,MAAM,MAAM,CAC3B;;AAGH,SAAS,mBACP,QACA,UACA,QACS;AACT,KAAI,UAAU,KACZ,QAAO;AAET,QAAO,qBACL,QACA,UACA,OACD;;AAGH,SAAS,cAAc,SAAwC;AAE7D,QADgB,QAAQ,QACT,MAAM,UAAU,MAAM,SAAS,OAAO,EAAE;;AAGzD,SAAS,oBACP,SACA,UACA,QACS;AACT,KAAI,SAAS,OAAO,KAAA,KAAa,QAAQ,OAAO,SAAS,GACvD,QAAO;AAET,KAAI,SAAS,SAAS,KAAA,KAAa,cAAc,QAAQ,KAAK,SAAS,KACrE,QAAO;AAET,KAAI,SAAS,cAAc,KAAA,GAAW;EACpC,MAAM,QAAQ,QAAQ,cAAc,EAAE;AACtC,MAAI,MAAM,WAAW,SAAS,UAAU,OACtC,QAAO;AAET,OAAK,IAAI,IAAI,GAAG,IAAI,SAAS,UAAU,QAAQ,KAAK;GAClD,MAAM,OAAO,MAAM;GACnB,MAAM,MAAM,SAAS,UAAU;AAC/B,OAAI,MAAM,SAAS,IAAI,QAAQ,CAAC,OAAO,KAAK,MAAM,IAAI,KAAK,CACzD,QAAO;;;AAIb,KACE,SAAS,UAAU,KAAA,KACnB,CAAC,mBAAmB,QAAQ,gBAAgB,SAAS,OAAO,OAAO,CAEnE,QAAO;AAET,KACE,SAAS,qBAAqB,KAAA,KAC9B,CAAC,qBACC,QAAQ,mBACR,SAAS,kBACT,OACD,CAED,QAAO;AAET,QAAO;;AAGT,SAAS,oBACP,UACA,aACA,OACmB;AACnB,QAAO;EACL,MAAM;EACN,eACE,GAAG,MAAM,YAAY,YAAY,CAAC,yFAErB,MAAM,cAAc,SAAS;EAC5C,QAAQ;EACR,UAAU;EACX;;AAGH,SAAS,SAAS,MAAe,OAA0B;AACzD,QAAO,QAAQ,CAAC,OAAO;;AAGzB,eAAsB,iBAEpB,UACA,UAC4B;CAC5B,MAAM,EAAE,OAAO,UAAU;CACzB,MAAM,cAAc;AAEpB,KAAI,CAAC,kBAAkB,SAAS,CAC9B,QAAO,oBAAoB,UAAU,aAAa,MAAM;CAG1D,MAAM,SAAS,MAAM,SAAS;AAG9B,QAAO;EACL,MAHW,SAAS,WAAW,UAAU,MAGrC;EACJ,eACE,GAAG,MAAM,YAAY,aAAa,KAAA,GAAW,KAAA,GAAW,EAAE,OAAO,CAAC,CAAC,4BAC1C,QAAQ,SAAS,KAAK,MAAM,cAAc,SAAS,CAAC,0BACpD,MAAM,cAAc,OAAO;EACtD;EACA;EACD;;AAGH,eAAsB,sBAEpB,UACA,UAC4B;CAC5B,MAAM,EAAE,OAAO,UAAU;CACzB,MAAM,cAAc;AAEpB,KAAI,CAAC,kBAAkB,SAAS,CAC9B,QAAO,oBAAoB,UAAU,aAAa,MAAM;CAG1D,MAAM,SAAS,MAAM,SAAS;AAG9B,QAAO;EACL,MAHW,SAAS,WAAW,UAAU,MAGrC;EACJ,eACE,GAAG,MAAM,YAAY,aAAa,KAAA,GAAW,KAAA,GAAW,EAAE,OAAO,CAAC,CAAC,iCACrC,QAAQ,SAAS,KAAK,MAAM,cAAc,SAAS,CAAC,+BACpD,MAAM,cAAc,OAAO;EAC3D;EACA;EACD;;AAwBH,eAAsB,sBAEpB,UACA,UAC4B;CAC5B,MAAM,EAAE,OAAO,UAAU;CACzB,MAAM,cAAc;AAEpB,KAAI,CAAC,kBAAkB,SAAS,CAC9B,QAAO,oBAAoB,UAAU,aAAa,MAAM;CAG1D,MAAM,SAAS,MAAM,SAAS;CAE9B,IAAI,OACF,OAAO,WAAW,SAAS,UAC3B,SAAS,OAAO,KAAK,MAAM;EACzB,MAAM,OAAO,OAAO;AACpB,SAAO,MAAM,SAAS,IAAI,QAAQ,KAAK,OAAO,KAAK,MAAM,IAAI,KAAK;GAClE;AACJ,QAAO,SAAS,MAAM,MAAM;AAE5B,QAAO;EACL;EACA,eACE,GAAG,MAAM,YAAY,aAAa,KAAA,GAAW,KAAA,GAAW,EAAE,OAAO,CAAC,CAAC,kCACpC,MAAM,cAAc,SAAS,CAAC,gCAC9B,MAAM,cACnC,OAAO,KAAK,QAAQ;GAAE,MAAM,GAAG;GAAM,MAAM,GAAG;GAAM,EAAE,CACvD;EACH,QAAQ,OAAO,KAAK,QAAQ;GAAE,MAAM,GAAG;GAAM,MAAM,GAAG;GAAM,EAAE;EAC9D;EACD;;AAGH,eAAsB,kBAEpB,UACA,UAC4B;CAC5B,MAAM,EAAE,OAAO,UAAU;CACzB,MAAM,cAAc;AAEpB,KAAI,CAAC,kBAAkB,SAAS,CAC9B,QAAO,oBAAoB,UAAU,aAAa,MAAM;CAG1D,MAAM,SAAS,MAAM,SAAS;AAM9B,QAAO;EACL,MANW,SACX,mBAAmB,QAAQ,UAAU,KAAK,OAAO,EACjD,MAII;EACJ,eACE,GAAG,MAAM,YAAY,aAAa,KAAA,GAAW,KAAA,GAAW,EAAE,OAAO,CAAC,CAAC,6BACzC,MAAM,cAAc,SAAS,CAAC,2BAC9B,MAAM,cAAc,OAAO;EACvD;EACA;EACD;;AAGH,eAAsB,mBAEpB,UACA,UAC4B;CAC5B,MAAM,EAAE,OAAO,UAAU;CACzB,MAAM,cAAc;AAEpB,KAAI,CAAC,kBAAkB,SAAS,CAC9B,QAAO,oBAAoB,UAAU,aAAa,MAAM;CAG1D,MAAM,UAAU,MAAM,SAAS;AAM/B,QAAO;EACL,MANW,SACX,oBAAoB,SAAS,UAAU,KAAK,OAAO,EACnD,MAII;EACJ,eACE,GAAG,MAAM,YAAY,aAAa,KAAA,GAAW,KAAA,GAAW,EAAE,OAAO,CAAC,CAAC,8BACxC,MAAM,cAAc,SAAS,CAAC,4BAC9B,MAAM,cAAc;GAC7C,IAAI,QAAQ;GACZ,MAAM,cAAc,QAAQ;GAC5B,YAAY,QAAQ,YAAY,KAAK,QAAQ;IAC3C,MAAM,GAAG;IACT,MAAM,GAAG;IACV,EAAE;GACH,gBAAgB,QAAQ;GACxB,mBAAmB,QAAQ;GAC5B,CAAC;EACJ,QAAQ;EACR;EACD;;;AAIH,MAAa,iBAAiB;CAC5B;CACA;CACA;CACA;CACA;CACD"}

package/dist/utils/testing/tools.cjs

@@ -1,5 +1,4 @@
-const require_tools_index = require('../../tools/index.cjs');
-
+const require_tools_index = require("../../tools/index.cjs");
 //#region src/utils/testing/tools.ts
 var FakeTool = class extends require_tools_index.StructuredTool {
 	name;
@@ -15,7 +14,7 @@ var FakeTool = class extends require_tools_index.StructuredTool {
 		return JSON.stringify(arg);
 	}
 };
-
 //#endregion
 exports.FakeTool = FakeTool;
+
 //# sourceMappingURL=tools.cjs.map

package/dist/utils/testing/tools.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"tools.cjs","names":["StructuredTool","fields: FakeToolParams<T>","arg: ToolInputSchemaOutputType<T>","_runManager?: CallbackManagerForToolRun"],"sources":["../../../src/utils/testing/tools.ts"],"sourcesContent":["import { CallbackManagerForToolRun } from \"../../callbacks/manager.js\";\nimport { StructuredTool, ToolParams } from \"../../tools/index.js\";\nimport { ToolInputSchemaOutputType } from \"../../tools/types.js\";\nimport { InteropZodObject } from \"../types/zod.js\";\n\nexport interface FakeToolParams<T extends InteropZodObject = InteropZodObject>\n  extends ToolParams {\n  name: string;\n  description: string;\n  schema: T;\n}\n\nexport class FakeTool<\n  T extends InteropZodObject = InteropZodObject\n> extends StructuredTool<T> {\n  name: string;\n\n  description: string;\n\n  schema: T;\n\n  constructor(fields: FakeToolParams<T>) {\n    super(fields);\n    this.name = fields.name;\n    this.description = fields.description;\n    this.schema = fields.schema;\n  }\n\n  protected async _call(\n    arg: ToolInputSchemaOutputType<T>,\n    _runManager?: CallbackManagerForToolRun\n  ): Promise<string> {\n    return JSON.stringify(arg);\n  }\n}\n"],"mappings":";;;AAYA,IAAa,WAAb,cAEUA,mCAAkB;CAC1B;CAEA;CAEA;CAEA,YAAYC,QAA2B;EACrC,MAAM,OAAO;EACb,KAAK,OAAO,OAAO;EACnB,KAAK,cAAc,OAAO;EAC1B,KAAK,SAAS,OAAO;CACtB;CAED,MAAgB,MACdC,KACAC,aACiB;AACjB,SAAO,KAAK,UAAU,IAAI;CAC3B;AACF"}
+{"version":3,"file":"tools.cjs","names":["StructuredTool"],"sources":["../../../src/utils/testing/tools.ts"],"sourcesContent":["import { CallbackManagerForToolRun } from \"../../callbacks/manager.js\";\nimport { StructuredTool, ToolParams } from \"../../tools/index.js\";\nimport { ToolInputSchemaOutputType } from \"../../tools/types.js\";\nimport { InteropZodObject } from \"../types/zod.js\";\n\nexport interface FakeToolParams<\n  T extends InteropZodObject = InteropZodObject,\n> extends ToolParams {\n  name: string;\n  description: string;\n  schema: T;\n}\n\nexport class FakeTool<\n  T extends InteropZodObject = InteropZodObject,\n> extends StructuredTool<T> {\n  name: string;\n\n  description: string;\n\n  schema: T;\n\n  constructor(fields: FakeToolParams<T>) {\n    super(fields);\n    this.name = fields.name;\n    this.description = fields.description;\n    this.schema = fields.schema;\n  }\n\n  protected async _call(\n    arg: ToolInputSchemaOutputType<T>,\n    _runManager?: CallbackManagerForToolRun\n  ): Promise<string> {\n    return JSON.stringify(arg);\n  }\n}\n"],"mappings":";;AAaA,IAAa,WAAb,cAEUA,oBAAAA,eAAkB;CAC1B;CAEA;CAEA;CAEA,YAAY,QAA2B;AACrC,QAAM,OAAO;AACb,OAAK,OAAO,OAAO;AACnB,OAAK,cAAc,OAAO;AAC1B,OAAK,SAAS,OAAO;;CAGvB,MAAgB,MACd,KACA,aACiB;AACjB,SAAO,KAAK,UAAU,IAAI"}

package/dist/utils/testing/tools.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"tools.d.cts","names":["CallbackManagerForToolRun","StructuredTool","ToolParams","ToolInputSchemaOutputType","InteropZodObject","FakeToolParams","T","FakeTool","Promise"],"sources":["../../../src/utils/testing/tools.d.ts"],"sourcesContent":["import { CallbackManagerForToolRun } from \"../../callbacks/manager.js\";\nimport { StructuredTool, ToolParams } from \"../../tools/index.js\";\nimport { ToolInputSchemaOutputType } from \"../../tools/types.js\";\nimport { InteropZodObject } from \"../types/zod.js\";\nexport interface FakeToolParams<T extends InteropZodObject = InteropZodObject> extends ToolParams {\n    name: string;\n    description: string;\n    schema: T;\n}\nexport declare class FakeTool<T extends InteropZodObject = InteropZodObject> extends StructuredTool<T> {\n    name: string;\n    description: string;\n    schema: T;\n    constructor(fields: FakeToolParams<T>);\n    protected _call(arg: ToolInputSchemaOutputType<T>, _runManager?: CallbackManagerForToolRun): Promise<string>;\n}\n//# sourceMappingURL=tools.d.ts.map"],"mappings":";;;;;;UAIiBK,yBAAyBD,mBAAmBA,0BAA0BF;;EAAtEG,WAAAA,EAAAA,MAAc;EAAWD,MAAAA,EAG9BE,CAH8BF;;AAG9BE,cAESC,QAFTD,CAAAA,UAE4BF,gBAF5BE,GAE+CF,gBAF/CE,CAAAA,SAEyEL,cAFzEK,CAEwFA,CAFxFA,CAAAA,CAAAA;EAH2EJ,IAAAA,EAAAA,MAAAA;EAAU,WAAA,EAAA,MAAA;EAK5EK,MAAAA,EAGTD,CAHSC;EAAmBH,WAAAA,CAAAA,MAAAA,EAIhBC,cAJgBD,CAIDE,CAJCF,CAAAA;EAAmBA,UAAAA,KAAAA,CAAAA,GAAAA,EAKlCD,yBALkCC,CAKRE,CALQF,CAAAA,EAAAA,WAAAA,CAAAA,EAKUJ,yBALVI,CAAAA,EAKsCI,OALtCJ,CAAAA,MAAAA,CAAAA"}
+{"version":3,"file":"tools.d.cts","names":[],"sources":["../../../src/utils/testing/tools.ts"],"mappings":";;;;;;UAKiB,cAAA,WACL,gBAAA,GAAmB,gBAAA,UACrB,UAAA;EACR,IAAA;EACA,WAAA;EACA,MAAA,EAAQ,CAAA;AAAA;AAAA,cAGG,QAAA,WACD,gBAAA,GAAmB,gBAAA,UACrB,cAAA,CAAe,CAAA;EACvB,IAAA;EAEA,WAAA;EAEA,MAAA,EAAQ,CAAA;EAER,WAAA,CAAY,MAAA,EAAQ,cAAA,CAAe,CAAA;EAAA,UAOnB,KAAA,CACd,GAAA,EAAK,yBAAA,CAA0B,CAAA,GAC/B,WAAA,GAAc,yBAAA,GACb,OAAA;AAAA"}

package/dist/utils/testing/tools.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tools.d.ts","names":["CallbackManagerForToolRun","StructuredTool","ToolParams","ToolInputSchemaOutputType","InteropZodObject","FakeToolParams","T","FakeTool","Promise"],"sources":["../../../src/utils/testing/tools.d.ts"],"sourcesContent":["import { CallbackManagerForToolRun } from \"../../callbacks/manager.js\";\nimport { StructuredTool, ToolParams } from \"../../tools/index.js\";\nimport { ToolInputSchemaOutputType } from \"../../tools/types.js\";\nimport { InteropZodObject } from \"../types/zod.js\";\nexport interface FakeToolParams<T extends InteropZodObject = InteropZodObject> extends ToolParams {\n    name: string;\n    description: string;\n    schema: T;\n}\nexport declare class FakeTool<T extends InteropZodObject = InteropZodObject> extends StructuredTool<T> {\n    name: string;\n    description: string;\n    schema: T;\n    constructor(fields: FakeToolParams<T>);\n    protected _call(arg: ToolInputSchemaOutputType<T>, _runManager?: CallbackManagerForToolRun): Promise<string>;\n}\n//# sourceMappingURL=tools.d.ts.map"],"mappings":";;;;;;UAIiBK,yBAAyBD,mBAAmBA,0BAA0BF;;EAAtEG,WAAAA,EAAAA,MAAc;EAAWD,MAAAA,EAG9BE,CAH8BF;;AAG9BE,cAESC,QAFTD,CAAAA,UAE4BF,gBAF5BE,GAE+CF,gBAF/CE,CAAAA,SAEyEL,cAFzEK,CAEwFA,CAFxFA,CAAAA,CAAAA;EAH2EJ,IAAAA,EAAAA,MAAAA;EAAU,WAAA,EAAA,MAAA;EAK5EK,MAAAA,EAGTD,CAHSC;EAAmBH,WAAAA,CAAAA,MAAAA,EAIhBC,cAJgBD,CAIDE,CAJCF,CAAAA;EAAmBA,UAAAA,KAAAA,CAAAA,GAAAA,EAKlCD,yBALkCC,CAKRE,CALQF,CAAAA,EAAAA,WAAAA,CAAAA,EAKUJ,yBALVI,CAAAA,EAKsCI,OALtCJ,CAAAA,MAAAA,CAAAA"}
+{"version":3,"file":"tools.d.ts","names":[],"sources":["../../../src/utils/testing/tools.ts"],"mappings":";;;;;;UAKiB,cAAA,WACL,gBAAA,GAAmB,gBAAA,UACrB,UAAA;EACR,IAAA;EACA,WAAA;EACA,MAAA,EAAQ,CAAA;AAAA;AAAA,cAGG,QAAA,WACD,gBAAA,GAAmB,gBAAA,UACrB,cAAA,CAAe,CAAA;EACvB,IAAA;EAEA,WAAA;EAEA,MAAA,EAAQ,CAAA;EAER,WAAA,CAAY,MAAA,EAAQ,cAAA,CAAe,CAAA;EAAA,UAOnB,KAAA,CACd,GAAA,EAAK,yBAAA,CAA0B,CAAA,GAC/B,WAAA,GAAc,yBAAA,GACb,OAAA;AAAA"}

package/dist/utils/testing/tools.js

@@ -1,5 +1,4 @@
 import { StructuredTool } from "../../tools/index.js";
-
 //#region src/utils/testing/tools.ts
 var FakeTool = class extends StructuredTool {
 	name;
@@ -15,7 +14,7 @@ var FakeTool = class extends StructuredTool {
 		return JSON.stringify(arg);
 	}
 };
-
 //#endregion
 export { FakeTool };
+
 //# sourceMappingURL=tools.js.map

package/dist/utils/testing/tools.js.map

@@ -1 +1 @@
-{"version":3,"file":"tools.js","names":["fields: FakeToolParams<T>","arg: ToolInputSchemaOutputType<T>","_runManager?: CallbackManagerForToolRun"],"sources":["../../../src/utils/testing/tools.ts"],"sourcesContent":["import { CallbackManagerForToolRun } from \"../../callbacks/manager.js\";\nimport { StructuredTool, ToolParams } from \"../../tools/index.js\";\nimport { ToolInputSchemaOutputType } from \"../../tools/types.js\";\nimport { InteropZodObject } from \"../types/zod.js\";\n\nexport interface FakeToolParams<T extends InteropZodObject = InteropZodObject>\n  extends ToolParams {\n  name: string;\n  description: string;\n  schema: T;\n}\n\nexport class FakeTool<\n  T extends InteropZodObject = InteropZodObject\n> extends StructuredTool<T> {\n  name: string;\n\n  description: string;\n\n  schema: T;\n\n  constructor(fields: FakeToolParams<T>) {\n    super(fields);\n    this.name = fields.name;\n    this.description = fields.description;\n    this.schema = fields.schema;\n  }\n\n  protected async _call(\n    arg: ToolInputSchemaOutputType<T>,\n    _runManager?: CallbackManagerForToolRun\n  ): Promise<string> {\n    return JSON.stringify(arg);\n  }\n}\n"],"mappings":";;;AAYA,IAAa,WAAb,cAEU,eAAkB;CAC1B;CAEA;CAEA;CAEA,YAAYA,QAA2B;EACrC,MAAM,OAAO;EACb,KAAK,OAAO,OAAO;EACnB,KAAK,cAAc,OAAO;EAC1B,KAAK,SAAS,OAAO;CACtB;CAED,MAAgB,MACdC,KACAC,aACiB;AACjB,SAAO,KAAK,UAAU,IAAI;CAC3B;AACF"}
+{"version":3,"file":"tools.js","names":[],"sources":["../../../src/utils/testing/tools.ts"],"sourcesContent":["import { CallbackManagerForToolRun } from \"../../callbacks/manager.js\";\nimport { StructuredTool, ToolParams } from \"../../tools/index.js\";\nimport { ToolInputSchemaOutputType } from \"../../tools/types.js\";\nimport { InteropZodObject } from \"../types/zod.js\";\n\nexport interface FakeToolParams<\n  T extends InteropZodObject = InteropZodObject,\n> extends ToolParams {\n  name: string;\n  description: string;\n  schema: T;\n}\n\nexport class FakeTool<\n  T extends InteropZodObject = InteropZodObject,\n> extends StructuredTool<T> {\n  name: string;\n\n  description: string;\n\n  schema: T;\n\n  constructor(fields: FakeToolParams<T>) {\n    super(fields);\n    this.name = fields.name;\n    this.description = fields.description;\n    this.schema = fields.schema;\n  }\n\n  protected async _call(\n    arg: ToolInputSchemaOutputType<T>,\n    _runManager?: CallbackManagerForToolRun\n  ): Promise<string> {\n    return JSON.stringify(arg);\n  }\n}\n"],"mappings":";;AAaA,IAAa,WAAb,cAEU,eAAkB;CAC1B;CAEA;CAEA;CAEA,YAAY,QAA2B;AACrC,QAAM,OAAO;AACb,OAAK,OAAO,OAAO;AACnB,OAAK,cAAc,OAAO;AAC1B,OAAK,SAAS,OAAO;;CAGvB,MAAgB,MACd,KACA,aACiB;AACjB,SAAO,KAAK,UAAU,IAAI"}

package/dist/utils/testing/tracers.cjs

@@ -1,5 +1,4 @@
-const require_tracers_base = require('../../tracers/base.cjs');
-
+const require_tracers_base = require("../../tracers/base.cjs");
 //#region src/utils/testing/tracers.ts
 var SingleRunExtractor = class extends require_tracers_base.BaseTracer {
 	runPromiseResolver;
@@ -19,7 +18,7 @@ var SingleRunExtractor = class extends require_tracers_base.BaseTracer {
 		return this.runPromise;
 	}
 };
-
 //#endregion
 exports.SingleRunExtractor = SingleRunExtractor;
+
 //# sourceMappingURL=tracers.cjs.map

package/dist/utils/testing/tracers.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"tracers.cjs","names":["BaseTracer","run: Run"],"sources":["../../../src/utils/testing/tracers.ts"],"sourcesContent":["import { BaseTracer, Run } from \"../../tracers/base.js\";\n\nexport class SingleRunExtractor extends BaseTracer {\n  runPromiseResolver: (run: Run) => void;\n\n  runPromise: Promise<Run>;\n\n  /** The name of the callback handler. */\n  name = \"single_run_extractor\";\n\n  constructor() {\n    super();\n    this.runPromise = new Promise<Run>((extract) => {\n      this.runPromiseResolver = extract;\n    });\n  }\n\n  async persistRun(run: Run) {\n    this.runPromiseResolver(run);\n  }\n\n  async extract(): Promise<Run> {\n    return this.runPromise;\n  }\n}\n"],"mappings":";;;AAEA,IAAa,qBAAb,cAAwCA,gCAAW;CACjD;CAEA;;CAGA,OAAO;CAEP,cAAc;EACZ,OAAO;EACP,KAAK,aAAa,IAAI,QAAa,CAAC,YAAY;GAC9C,KAAK,qBAAqB;EAC3B;CACF;CAED,MAAM,WAAWC,KAAU;EACzB,KAAK,mBAAmB,IAAI;CAC7B;CAED,MAAM,UAAwB;AAC5B,SAAO,KAAK;CACb;AACF"}
+{"version":3,"file":"tracers.cjs","names":["BaseTracer"],"sources":["../../../src/utils/testing/tracers.ts"],"sourcesContent":["import { BaseTracer, Run } from \"../../tracers/base.js\";\n\nexport class SingleRunExtractor extends BaseTracer {\n  runPromiseResolver: (run: Run) => void;\n\n  runPromise: Promise<Run>;\n\n  /** The name of the callback handler. */\n  name = \"single_run_extractor\";\n\n  constructor() {\n    super();\n    this.runPromise = new Promise<Run>((extract) => {\n      this.runPromiseResolver = extract;\n    });\n  }\n\n  async persistRun(run: Run) {\n    this.runPromiseResolver(run);\n  }\n\n  async extract(): Promise<Run> {\n    return this.runPromise;\n  }\n}\n"],"mappings":";;AAEA,IAAa,qBAAb,cAAwCA,qBAAAA,WAAW;CACjD;CAEA;;CAGA,OAAO;CAEP,cAAc;AACZ,SAAO;AACP,OAAK,aAAa,IAAI,SAAc,YAAY;AAC9C,QAAK,qBAAqB;IAC1B;;CAGJ,MAAM,WAAW,KAAU;AACzB,OAAK,mBAAmB,IAAI;;CAG9B,MAAM,UAAwB;AAC5B,SAAO,KAAK"}

package/dist/utils/testing/tracers.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"tracers.d.cts","names":["BaseTracer","Run","SingleRunExtractor","Promise"],"sources":["../../../src/utils/testing/tracers.d.ts"],"sourcesContent":["import { BaseTracer, Run } from \"../../tracers/base.js\";\nexport declare class SingleRunExtractor extends BaseTracer {\n    runPromiseResolver: (run: Run) => void;\n    runPromise: Promise<Run>;\n    /** The name of the callback handler. */\n    name: string;\n    constructor();\n    persistRun(run: Run): Promise<void>;\n    extract(): Promise<Run>;\n}\n//# sourceMappingURL=tracers.d.ts.map"],"mappings":";;;cACqBE,kBAAAA,SAA2BF,UAAAA;4BAClBC;EADTC,UAAAA,EAELC,OAFKD,CAEGD,GAFe,CAAA;EACTA;EACNA,IAAAA,EAAAA,MAAAA;EAARE,WAAAA,CAAAA;EAIIF,UAAAA,CAAAA,GAAAA,EAAAA,GAAAA,CAAAA,EAAME,OAANF,CAAAA,IAAAA,CAAAA;EAAME,OAAAA,CAAAA,CAAAA,EACXA,OADWA,CACHF,GADGE,CAAAA"}
+{"version":3,"file":"tracers.d.cts","names":[],"sources":["../../../src/utils/testing/tracers.ts"],"mappings":";;;cAEa,kBAAA,SAA2B,UAAA;EACtC,kBAAA,GAAqB,GAAA,EAAK,GAAA;EAE1B,UAAA,EAAY,OAAA,CAAQ,GAAA;EAHU;EAM9B,IAAA;EAEA,WAAA,CAAA;EAOM,UAAA,CAAW,GAAA,EAAK,GAAA,GAAG,OAAA;EAInB,OAAA,CAAA,GAAW,OAAA,CAAQ,GAAA;AAAA"}

package/dist/utils/testing/tracers.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tracers.d.ts","names":["BaseTracer","Run","SingleRunExtractor","Promise"],"sources":["../../../src/utils/testing/tracers.d.ts"],"sourcesContent":["import { BaseTracer, Run } from \"../../tracers/base.js\";\nexport declare class SingleRunExtractor extends BaseTracer {\n    runPromiseResolver: (run: Run) => void;\n    runPromise: Promise<Run>;\n    /** The name of the callback handler. */\n    name: string;\n    constructor();\n    persistRun(run: Run): Promise<void>;\n    extract(): Promise<Run>;\n}\n//# sourceMappingURL=tracers.d.ts.map"],"mappings":";;;cACqBE,kBAAAA,SAA2BF,UAAAA;4BAClBC;EADTC,UAAAA,EAELC,OAFKD,CAEGD,GAFe,CAAA;EACTA;EACNA,IAAAA,EAAAA,MAAAA;EAARE,WAAAA,CAAAA;EAIIF,UAAAA,CAAAA,GAAAA,EAAAA,GAAAA,CAAAA,EAAME,OAANF,CAAAA,IAAAA,CAAAA;EAAME,OAAAA,CAAAA,CAAAA,EACXA,OADWA,CACHF,GADGE,CAAAA"}
+{"version":3,"file":"tracers.d.ts","names":[],"sources":["../../../src/utils/testing/tracers.ts"],"mappings":";;;cAEa,kBAAA,SAA2B,UAAA;EACtC,kBAAA,GAAqB,GAAA,EAAK,GAAA;EAE1B,UAAA,EAAY,OAAA,CAAQ,GAAA;EAHU;EAM9B,IAAA;EAEA,WAAA,CAAA;EAOM,UAAA,CAAW,GAAA,EAAK,GAAA,GAAG,OAAA;EAInB,OAAA,CAAA,GAAW,OAAA,CAAQ,GAAA;AAAA"}

package/dist/utils/testing/tracers.js

@@ -1,5 +1,4 @@
 import { BaseTracer } from "../../tracers/base.js";
-
 //#region src/utils/testing/tracers.ts
 var SingleRunExtractor = class extends BaseTracer {
 	runPromiseResolver;
@@ -19,7 +18,7 @@ var SingleRunExtractor = class extends BaseTracer {
 		return this.runPromise;
 	}
 };
-
 //#endregion
 export { SingleRunExtractor };
+
 //# sourceMappingURL=tracers.js.map

package/dist/utils/testing/tracers.js.map

@@ -1 +1 @@
-{"version":3,"file":"tracers.js","names":["run: Run"],"sources":["../../../src/utils/testing/tracers.ts"],"sourcesContent":["import { BaseTracer, Run } from \"../../tracers/base.js\";\n\nexport class SingleRunExtractor extends BaseTracer {\n  runPromiseResolver: (run: Run) => void;\n\n  runPromise: Promise<Run>;\n\n  /** The name of the callback handler. */\n  name = \"single_run_extractor\";\n\n  constructor() {\n    super();\n    this.runPromise = new Promise<Run>((extract) => {\n      this.runPromiseResolver = extract;\n    });\n  }\n\n  async persistRun(run: Run) {\n    this.runPromiseResolver(run);\n  }\n\n  async extract(): Promise<Run> {\n    return this.runPromise;\n  }\n}\n"],"mappings":";;;AAEA,IAAa,qBAAb,cAAwC,WAAW;CACjD;CAEA;;CAGA,OAAO;CAEP,cAAc;EACZ,OAAO;EACP,KAAK,aAAa,IAAI,QAAa,CAAC,YAAY;GAC9C,KAAK,qBAAqB;EAC3B;CACF;CAED,MAAM,WAAWA,KAAU;EACzB,KAAK,mBAAmB,IAAI;CAC7B;CAED,MAAM,UAAwB;AAC5B,SAAO,KAAK;CACb;AACF"}
+{"version":3,"file":"tracers.js","names":[],"sources":["../../../src/utils/testing/tracers.ts"],"sourcesContent":["import { BaseTracer, Run } from \"../../tracers/base.js\";\n\nexport class SingleRunExtractor extends BaseTracer {\n  runPromiseResolver: (run: Run) => void;\n\n  runPromise: Promise<Run>;\n\n  /** The name of the callback handler. */\n  name = \"single_run_extractor\";\n\n  constructor() {\n    super();\n    this.runPromise = new Promise<Run>((extract) => {\n      this.runPromiseResolver = extract;\n    });\n  }\n\n  async persistRun(run: Run) {\n    this.runPromiseResolver(run);\n  }\n\n  async extract(): Promise<Run> {\n    return this.runPromise;\n  }\n}\n"],"mappings":";;AAEA,IAAa,qBAAb,cAAwC,WAAW;CACjD;CAEA;;CAGA,OAAO;CAEP,cAAc;AACZ,SAAO;AACP,OAAK,aAAa,IAAI,SAAc,YAAY;AAC9C,QAAK,qBAAqB;IAC1B;;CAGJ,MAAM,WAAW,KAAU;AACzB,OAAK,mBAAmB,IAAI;;CAG9B,MAAM,UAAwB;AAC5B,SAAO,KAAK"}

package/dist/utils/testing/vectorstores.cjs

@@ -1,7 +1,6 @@
-const require_vectorstores = require('../../vectorstores.cjs');
-const require_document = require('../../documents/document.cjs');
-const require_similarities = require('../ml-distance/similarities.cjs');
-
+const require_document = require("../../documents/document.cjs");
+const require_similarities = require("../ml-distance/similarities.cjs");
+const require_vectorstores = require("../../vectorstores.cjs");
 //#region src/utils/testing/vectorstores.ts
 /**
 * Class that extends `VectorStore` to store vectors in memory. Provides
@@ -14,7 +13,7 @@ var FakeVectorStore = class FakeVectorStore extends require_vectorstores.VectorS
 	_vectorstoreType() {
 		return "memory";
 	}
-	constructor(embeddings, { similarity,...rest } = {}) {
+	constructor(embeddings, { similarity, ...rest } = {}) {
 		super(embeddings, rest);
 		this.similarity = similarity ?? require_similarities.cosine;
 	}
@@ -58,22 +57,19 @@ var FakeVectorStore = class FakeVectorStore extends require_vectorstores.VectorS
 	async similaritySearchVectorWithScore(query, k, filter) {
 		const filterFunction = (memoryVector) => {
 			if (!filter) return true;
-			const doc = new require_document.Document({
+			return filter(new require_document.Document({
 				metadata: memoryVector.metadata,
 				pageContent: memoryVector.content
-			});
-			return filter(doc);
+			}));
 		};
 		const filteredMemoryVectors = this.memoryVectors.filter(filterFunction);
-		const searches = filteredMemoryVectors.map((vector, index) => ({
+		return filteredMemoryVectors.map((vector, index) => ({
 			similarity: this.similarity(query, vector.embedding),
 			index
-		})).sort((a, b) => a.similarity > b.similarity ? -1 : 0).slice(0, k);
-		const result = searches.map((search) => [new require_document.Document({
+		})).sort((a, b) => a.similarity > b.similarity ? -1 : 0).slice(0, k).map((search) => [new require_document.Document({
 			metadata: filteredMemoryVectors[search.index].metadata,
 			pageContent: filteredMemoryVectors[search.index].content
 		}), search.similarity]);
-		return result;
 	}
 	/**
 	* Static method to create a `FakeVectorStore` instance from an array of
@@ -119,11 +115,10 @@ var FakeVectorStore = class FakeVectorStore extends require_vectorstores.VectorS
 	* @returns Promise that resolves with a new `FakeVectorStore` instance.
 	*/
 	static async fromExistingIndex(embeddings, dbConfig) {
-		const instance = new this(embeddings, dbConfig);
-		return instance;
+		return new this(embeddings, dbConfig);
 	}
 };
-
 //#endregion
 exports.FakeVectorStore = FakeVectorStore;
+
 //# sourceMappingURL=vectorstores.cjs.map

package/dist/utils/testing/vectorstores.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"vectorstores.cjs","names":["VectorStore","embeddings: EmbeddingsInterface","cosine","documents: Document[]","vectors: number[][]","query: number[]","k: number","filter?: this[\"FilterType\"]","memoryVector: MemoryVector","Document","result: [Document, number][]","texts: string[]","metadatas: object[] | object","dbConfig?: FakeVectorStoreArgs","docs: Document[]"],"sources":["../../../src/utils/testing/vectorstores.ts"],"sourcesContent":["import { Document } from \"../../documents/document.js\";\nimport { EmbeddingsInterface } from \"../../embeddings.js\";\nimport { VectorStore } from \"../../vectorstores.js\";\nimport { cosine } from \"../ml-distance/similarities.js\";\n\n/**\n * Interface representing a vector in memory. It includes the content\n * (text), the corresponding embedding (vector), and any associated\n * metadata.\n */\ninterface MemoryVector {\n  content: string;\n  embedding: number[];\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  metadata: Record<string, any>;\n}\n\n/**\n * Interface for the arguments that can be passed to the\n * `FakeVectorStore` constructor. It includes an optional `similarity`\n * function.\n */\nexport interface FakeVectorStoreArgs {\n  similarity?: typeof cosine;\n}\n\n/**\n * Class that extends `VectorStore` to store vectors in memory. Provides\n * methods for adding documents, performing similarity searches, and\n * creating instances from texts, documents, or an existing index.\n */\nexport class FakeVectorStore extends VectorStore {\n  declare FilterType: (doc: Document) => boolean;\n\n  memoryVectors: MemoryVector[] = [];\n\n  similarity: typeof cosine;\n\n  _vectorstoreType(): string {\n    return \"memory\";\n  }\n\n  constructor(\n    embeddings: EmbeddingsInterface,\n    { similarity, ...rest }: FakeVectorStoreArgs = {}\n  ) {\n    super(embeddings, rest);\n\n    this.similarity = similarity ?? cosine;\n  }\n\n  /**\n   * Method to add documents to the memory vector store. It extracts the\n   * text from each document, generates embeddings for them, and adds the\n   * resulting vectors to the store.\n   * @param documents Array of `Document` instances to be added to the store.\n   * @returns Promise that resolves when all documents have been added.\n   */\n  async addDocuments(documents: Document[]): Promise<void> {\n    const texts = documents.map(({ pageContent }) => pageContent);\n    return this.addVectors(\n      await this.embeddings.embedDocuments(texts),\n      documents\n    );\n  }\n\n  /**\n   * Method to add vectors to the memory vector store. It creates\n   * `MemoryVector` instances for each vector and document pair and adds\n   * them to the store.\n   * @param vectors Array of vectors to be added to the store.\n   * @param documents Array of `Document` instances corresponding to the vectors.\n   * @returns Promise that resolves when all vectors have been added.\n   */\n  async addVectors(vectors: number[][], documents: Document[]): Promise<void> {\n    const memoryVectors = vectors.map((embedding, idx) => ({\n      content: documents[idx].pageContent,\n      embedding,\n      metadata: documents[idx].metadata,\n    }));\n\n    this.memoryVectors = this.memoryVectors.concat(memoryVectors);\n  }\n\n  /**\n   * Method to perform a similarity search in the memory vector store. It\n   * calculates the similarity between the query vector and each vector in\n   * the store, sorts the results by similarity, and returns the top `k`\n   * results along with their scores.\n   * @param query Query vector to compare against the vectors in the store.\n   * @param k Number of top results to return.\n   * @param filter Optional filter function to apply to the vectors before performing the search.\n   * @returns Promise that resolves with an array of tuples, each containing a `Document` and its similarity score.\n   */\n  async similaritySearchVectorWithScore(\n    query: number[],\n    k: number,\n    filter?: this[\"FilterType\"]\n  ): Promise<[Document, number][]> {\n    const filterFunction = (memoryVector: MemoryVector) => {\n      if (!filter) {\n        return true;\n      }\n\n      const doc = new Document({\n        metadata: memoryVector.metadata,\n        pageContent: memoryVector.content,\n      });\n      return filter(doc);\n    };\n    const filteredMemoryVectors = this.memoryVectors.filter(filterFunction);\n    const searches = filteredMemoryVectors\n      .map((vector, index) => ({\n        similarity: this.similarity(query, vector.embedding),\n        index,\n      }))\n      .sort((a, b) => (a.similarity > b.similarity ? -1 : 0))\n      .slice(0, k);\n\n    const result: [Document, number][] = searches.map((search) => [\n      new Document({\n        metadata: filteredMemoryVectors[search.index].metadata,\n        pageContent: filteredMemoryVectors[search.index].content,\n      }),\n      search.similarity,\n    ]);\n\n    return result;\n  }\n\n  /**\n   * Static method to create a `FakeVectorStore` instance from an array of\n   * texts. It creates a `Document` for each text and metadata pair, and\n   * adds them to the store.\n   * @param texts Array of texts to be added to the store.\n   * @param metadatas Array or single object of metadata corresponding to the texts.\n   * @param embeddings `Embeddings` instance used to generate embeddings for the texts.\n   * @param dbConfig Optional `FakeVectorStoreArgs` to configure the `FakeVectorStore` instance.\n   * @returns Promise that resolves with a new `FakeVectorStore` instance.\n   */\n  static async fromTexts(\n    texts: string[],\n    metadatas: object[] | object,\n    embeddings: EmbeddingsInterface,\n    dbConfig?: FakeVectorStoreArgs\n  ): Promise<FakeVectorStore> {\n    const docs: Document[] = [];\n    for (let i = 0; i < texts.length; i += 1) {\n      const metadata = Array.isArray(metadatas) ? metadatas[i] : metadatas;\n      const newDoc = new Document({\n        pageContent: texts[i],\n        metadata,\n      });\n      docs.push(newDoc);\n    }\n    return FakeVectorStore.fromDocuments(docs, embeddings, dbConfig);\n  }\n\n  /**\n   * Static method to create a `FakeVectorStore` instance from an array of\n   * `Document` instances. It adds the documents to the store.\n   * @param docs Array of `Document` instances to be added to the store.\n   * @param embeddings `Embeddings` instance used to generate embeddings for the documents.\n   * @param dbConfig Optional `FakeVectorStoreArgs` to configure the `FakeVectorStore` instance.\n   * @returns Promise that resolves with a new `FakeVectorStore` instance.\n   */\n  static async fromDocuments(\n    docs: Document[],\n    embeddings: EmbeddingsInterface,\n    dbConfig?: FakeVectorStoreArgs\n  ): Promise<FakeVectorStore> {\n    const instance = new this(embeddings, dbConfig);\n    await instance.addDocuments(docs);\n    return instance;\n  }\n\n  /**\n   * Static method to create a `FakeVectorStore` instance from an existing\n   * index. It creates a new `FakeVectorStore` instance without adding any\n   * documents or vectors.\n   * @param embeddings `Embeddings` instance used to generate embeddings for the documents.\n   * @param dbConfig Optional `FakeVectorStoreArgs` to configure the `FakeVectorStore` instance.\n   * @returns Promise that resolves with a new `FakeVectorStore` instance.\n   */\n  static async fromExistingIndex(\n    embeddings: EmbeddingsInterface,\n    dbConfig?: FakeVectorStoreArgs\n  ): Promise<FakeVectorStore> {\n    const instance = new this(embeddings, dbConfig);\n    return instance;\n  }\n}\n"],"mappings":";;;;;;;;;;AA+BA,IAAa,kBAAb,MAAa,wBAAwBA,iCAAY;CAG/C,gBAAgC,CAAE;CAElC;CAEA,mBAA2B;AACzB,SAAO;CACR;CAED,YACEC,YACA,EAAE,WAAY,GAAG,MAA2B,GAAG,CAAE,GACjD;EACA,MAAM,YAAY,KAAK;EAEvB,KAAK,aAAa,cAAcC;CACjC;;;;;;;;CASD,MAAM,aAAaC,WAAsC;EACvD,MAAM,QAAQ,UAAU,IAAI,CAAC,EAAE,aAAa,KAAK,YAAY;AAC7D,SAAO,KAAK,WACV,MAAM,KAAK,WAAW,eAAe,MAAM,EAC3C,UACD;CACF;;;;;;;;;CAUD,MAAM,WAAWC,SAAqBD,WAAsC;EAC1E,MAAM,gBAAgB,QAAQ,IAAI,CAAC,WAAW,SAAS;GACrD,SAAS,UAAU,KAAK;GACxB;GACA,UAAU,UAAU,KAAK;EAC1B,GAAE;EAEH,KAAK,gBAAgB,KAAK,cAAc,OAAO,cAAc;CAC9D;;;;;;;;;;;CAYD,MAAM,gCACJE,OACAC,GACAC,QAC+B;EAC/B,MAAM,iBAAiB,CAACC,iBAA+B;AACrD,OAAI,CAAC,OACH,QAAO;GAGT,MAAM,MAAM,IAAIC,0BAAS;IACvB,UAAU,aAAa;IACvB,aAAa,aAAa;GAC3B;AACD,UAAO,OAAO,IAAI;EACnB;EACD,MAAM,wBAAwB,KAAK,cAAc,OAAO,eAAe;EACvE,MAAM,WAAW,sBACd,IAAI,CAAC,QAAQ,WAAW;GACvB,YAAY,KAAK,WAAW,OAAO,OAAO,UAAU;GACpD;EACD,GAAE,CACF,KAAK,CAAC,GAAG,MAAO,EAAE,aAAa,EAAE,aAAa,KAAK,EAAG,CACtD,MAAM,GAAG,EAAE;EAEd,MAAMC,SAA+B,SAAS,IAAI,CAAC,WAAW,CAC5D,IAAID,0BAAS;GACX,UAAU,sBAAsB,OAAO,OAAO;GAC9C,aAAa,sBAAsB,OAAO,OAAO;EAClD,IACD,OAAO,UACR,EAAC;AAEF,SAAO;CACR;;;;;;;;;;;CAYD,aAAa,UACXE,OACAC,WACAX,YACAY,UAC0B;EAC1B,MAAMC,OAAmB,CAAE;AAC3B,OAAK,IAAI,IAAI,GAAG,IAAI,MAAM,QAAQ,KAAK,GAAG;GACxC,MAAM,WAAW,MAAM,QAAQ,UAAU,GAAG,UAAU,KAAK;GAC3D,MAAM,SAAS,IAAIL,0BAAS;IAC1B,aAAa,MAAM;IACnB;GACD;GACD,KAAK,KAAK,OAAO;EAClB;AACD,SAAO,gBAAgB,cAAc,MAAM,YAAY,SAAS;CACjE;;;;;;;;;CAUD,aAAa,cACXK,MACAb,YACAY,UAC0B;EAC1B,MAAM,WAAW,IAAI,KAAK,YAAY;EACtC,MAAM,SAAS,aAAa,KAAK;AACjC,SAAO;CACR;;;;;;;;;CAUD,aAAa,kBACXZ,YACAY,UAC0B;EAC1B,MAAM,WAAW,IAAI,KAAK,YAAY;AACtC,SAAO;CACR;AACF"}
+{"version":3,"file":"vectorstores.cjs","names":["VectorStore","cosine","Document"],"sources":["../../../src/utils/testing/vectorstores.ts"],"sourcesContent":["import { Document } from \"../../documents/document.js\";\nimport { EmbeddingsInterface } from \"../../embeddings.js\";\nimport { VectorStore } from \"../../vectorstores.js\";\nimport { cosine } from \"../ml-distance/similarities.js\";\n\n/**\n * Interface representing a vector in memory. It includes the content\n * (text), the corresponding embedding (vector), and any associated\n * metadata.\n */\ninterface MemoryVector {\n  content: string;\n  embedding: number[];\n  // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n  metadata: Record<string, any>;\n}\n\n/**\n * Interface for the arguments that can be passed to the\n * `FakeVectorStore` constructor. It includes an optional `similarity`\n * function.\n */\nexport interface FakeVectorStoreArgs {\n  similarity?: typeof cosine;\n}\n\n/**\n * Class that extends `VectorStore` to store vectors in memory. Provides\n * methods for adding documents, performing similarity searches, and\n * creating instances from texts, documents, or an existing index.\n */\nexport class FakeVectorStore extends VectorStore {\n  declare FilterType: (doc: Document) => boolean;\n\n  memoryVectors: MemoryVector[] = [];\n\n  similarity: typeof cosine;\n\n  _vectorstoreType(): string {\n    return \"memory\";\n  }\n\n  constructor(\n    embeddings: EmbeddingsInterface,\n    { similarity, ...rest }: FakeVectorStoreArgs = {}\n  ) {\n    super(embeddings, rest);\n\n    this.similarity = similarity ?? cosine;\n  }\n\n  /**\n   * Method to add documents to the memory vector store. It extracts the\n   * text from each document, generates embeddings for them, and adds the\n   * resulting vectors to the store.\n   * @param documents Array of `Document` instances to be added to the store.\n   * @returns Promise that resolves when all documents have been added.\n   */\n  async addDocuments(documents: Document[]): Promise<void> {\n    const texts = documents.map(({ pageContent }) => pageContent);\n    return this.addVectors(\n      await this.embeddings.embedDocuments(texts),\n      documents\n    );\n  }\n\n  /**\n   * Method to add vectors to the memory vector store. It creates\n   * `MemoryVector` instances for each vector and document pair and adds\n   * them to the store.\n   * @param vectors Array of vectors to be added to the store.\n   * @param documents Array of `Document` instances corresponding to the vectors.\n   * @returns Promise that resolves when all vectors have been added.\n   */\n  async addVectors(vectors: number[][], documents: Document[]): Promise<void> {\n    const memoryVectors = vectors.map((embedding, idx) => ({\n      content: documents[idx].pageContent,\n      embedding,\n      metadata: documents[idx].metadata,\n    }));\n\n    this.memoryVectors = this.memoryVectors.concat(memoryVectors);\n  }\n\n  /**\n   * Method to perform a similarity search in the memory vector store. It\n   * calculates the similarity between the query vector and each vector in\n   * the store, sorts the results by similarity, and returns the top `k`\n   * results along with their scores.\n   * @param query Query vector to compare against the vectors in the store.\n   * @param k Number of top results to return.\n   * @param filter Optional filter function to apply to the vectors before performing the search.\n   * @returns Promise that resolves with an array of tuples, each containing a `Document` and its similarity score.\n   */\n  async similaritySearchVectorWithScore(\n    query: number[],\n    k: number,\n    filter?: this[\"FilterType\"]\n  ): Promise<[Document, number][]> {\n    const filterFunction = (memoryVector: MemoryVector) => {\n      if (!filter) {\n        return true;\n      }\n\n      const doc = new Document({\n        metadata: memoryVector.metadata,\n        pageContent: memoryVector.content,\n      });\n      return filter(doc);\n    };\n    const filteredMemoryVectors = this.memoryVectors.filter(filterFunction);\n    const searches = filteredMemoryVectors\n      .map((vector, index) => ({\n        similarity: this.similarity(query, vector.embedding),\n        index,\n      }))\n      .sort((a, b) => (a.similarity > b.similarity ? -1 : 0))\n      .slice(0, k);\n\n    const result: [Document, number][] = searches.map((search) => [\n      new Document({\n        metadata: filteredMemoryVectors[search.index].metadata,\n        pageContent: filteredMemoryVectors[search.index].content,\n      }),\n      search.similarity,\n    ]);\n\n    return result;\n  }\n\n  /**\n   * Static method to create a `FakeVectorStore` instance from an array of\n   * texts. It creates a `Document` for each text and metadata pair, and\n   * adds them to the store.\n   * @param texts Array of texts to be added to the store.\n   * @param metadatas Array or single object of metadata corresponding to the texts.\n   * @param embeddings `Embeddings` instance used to generate embeddings for the texts.\n   * @param dbConfig Optional `FakeVectorStoreArgs` to configure the `FakeVectorStore` instance.\n   * @returns Promise that resolves with a new `FakeVectorStore` instance.\n   */\n  static async fromTexts(\n    texts: string[],\n    metadatas: object[] | object,\n    embeddings: EmbeddingsInterface,\n    dbConfig?: FakeVectorStoreArgs\n  ): Promise<FakeVectorStore> {\n    const docs: Document[] = [];\n    for (let i = 0; i < texts.length; i += 1) {\n      const metadata = Array.isArray(metadatas) ? metadatas[i] : metadatas;\n      const newDoc = new Document({\n        pageContent: texts[i],\n        metadata,\n      });\n      docs.push(newDoc);\n    }\n    return FakeVectorStore.fromDocuments(docs, embeddings, dbConfig);\n  }\n\n  /**\n   * Static method to create a `FakeVectorStore` instance from an array of\n   * `Document` instances. It adds the documents to the store.\n   * @param docs Array of `Document` instances to be added to the store.\n   * @param embeddings `Embeddings` instance used to generate embeddings for the documents.\n   * @param dbConfig Optional `FakeVectorStoreArgs` to configure the `FakeVectorStore` instance.\n   * @returns Promise that resolves with a new `FakeVectorStore` instance.\n   */\n  static async fromDocuments(\n    docs: Document[],\n    embeddings: EmbeddingsInterface,\n    dbConfig?: FakeVectorStoreArgs\n  ): Promise<FakeVectorStore> {\n    const instance = new this(embeddings, dbConfig);\n    await instance.addDocuments(docs);\n    return instance;\n  }\n\n  /**\n   * Static method to create a `FakeVectorStore` instance from an existing\n   * index. It creates a new `FakeVectorStore` instance without adding any\n   * documents or vectors.\n   * @param embeddings `Embeddings` instance used to generate embeddings for the documents.\n   * @param dbConfig Optional `FakeVectorStoreArgs` to configure the `FakeVectorStore` instance.\n   * @returns Promise that resolves with a new `FakeVectorStore` instance.\n   */\n  static async fromExistingIndex(\n    embeddings: EmbeddingsInterface,\n    dbConfig?: FakeVectorStoreArgs\n  ): Promise<FakeVectorStore> {\n    const instance = new this(embeddings, dbConfig);\n    return instance;\n  }\n}\n"],"mappings":";;;;;;;;;AA+BA,IAAa,kBAAb,MAAa,wBAAwBA,qBAAAA,YAAY;CAG/C,gBAAgC,EAAE;CAElC;CAEA,mBAA2B;AACzB,SAAO;;CAGT,YACE,YACA,EAAE,YAAY,GAAG,SAA8B,EAAE,EACjD;AACA,QAAM,YAAY,KAAK;AAEvB,OAAK,aAAa,cAAcC,qBAAAA;;;;;;;;;CAUlC,MAAM,aAAa,WAAsC;EACvD,MAAM,QAAQ,UAAU,KAAK,EAAE,kBAAkB,YAAY;AAC7D,SAAO,KAAK,WACV,MAAM,KAAK,WAAW,eAAe,MAAM,EAC3C,UACD;;;;;;;;;;CAWH,MAAM,WAAW,SAAqB,WAAsC;EAC1E,MAAM,gBAAgB,QAAQ,KAAK,WAAW,SAAS;GACrD,SAAS,UAAU,KAAK;GACxB;GACA,UAAU,UAAU,KAAK;GAC1B,EAAE;AAEH,OAAK,gBAAgB,KAAK,cAAc,OAAO,cAAc;;;;;;;;;;;;CAa/D,MAAM,gCACJ,OACA,GACA,QAC+B;EAC/B,MAAM,kBAAkB,iBAA+B;AACrD,OAAI,CAAC,OACH,QAAO;AAOT,UAAO,OAAO,IAJEC,iBAAAA,SAAS;IACvB,UAAU,aAAa;IACvB,aAAa,aAAa;IAC3B,CACgB,CAAC;;EAEpB,MAAM,wBAAwB,KAAK,cAAc,OAAO,eAAe;AAiBvE,SAhBiB,sBACd,KAAK,QAAQ,WAAW;GACvB,YAAY,KAAK,WAAW,OAAO,OAAO,UAAU;GACpD;GACD,EAAE,CACF,MAAM,GAAG,MAAO,EAAE,aAAa,EAAE,aAAa,KAAK,EAAG,CACtD,MAAM,GAAG,EAEiC,CAAC,KAAK,WAAW,CAC5D,IAAIA,iBAAAA,SAAS;GACX,UAAU,sBAAsB,OAAO,OAAO;GAC9C,aAAa,sBAAsB,OAAO,OAAO;GAClD,CAAC,EACF,OAAO,WACR,CAEY;;;;;;;;;;;;CAaf,aAAa,UACX,OACA,WACA,YACA,UAC0B;EAC1B,MAAM,OAAmB,EAAE;AAC3B,OAAK,IAAI,IAAI,GAAG,IAAI,MAAM,QAAQ,KAAK,GAAG;GACxC,MAAM,WAAW,MAAM,QAAQ,UAAU,GAAG,UAAU,KAAK;GAC3D,MAAM,SAAS,IAAIA,iBAAAA,SAAS;IAC1B,aAAa,MAAM;IACnB;IACD,CAAC;AACF,QAAK,KAAK,OAAO;;AAEnB,SAAO,gBAAgB,cAAc,MAAM,YAAY,SAAS;;;;;;;;;;CAWlE,aAAa,cACX,MACA,YACA,UAC0B;EAC1B,MAAM,WAAW,IAAI,KAAK,YAAY,SAAS;AAC/C,QAAM,SAAS,aAAa,KAAK;AACjC,SAAO;;;;;;;;;;CAWT,aAAa,kBACX,YACA,UAC0B;AAE1B,SAAO,IADc,KAAK,YAAY,SACvB"}