@langchain/core 1.2.0-dev-1775763803878 → 1.2.0

package/dist/utils/testing/openai_stream_fixtures.js

@@ -0,0 +1,155 @@
+//#region src/utils/testing/openai_stream_fixtures.ts
+function asAsyncIterable(items) {
+	return { async *[Symbol.asyncIterator]() {
+		for (const item of items) yield item;
+	} };
+}
+function openAITextOnlyChunksWithUsage(model = "test-model") {
+	const chunks = openAITextOnlyChunks(model);
+	const last = chunks[chunks.length - 1];
+	chunks[chunks.length - 1] = {
+		...last,
+		usage: {
+			prompt_tokens: 10,
+			completion_tokens: 2,
+			total_tokens: 12
+		}
+	};
+	return chunks;
+}
+function openAITextOnlyChunks(model = "test-model") {
+	return [
+		{
+			id: "chatcmpl-text",
+			model,
+			choices: [{
+				index: 0,
+				delta: {
+					role: "assistant",
+					content: "Hello"
+				},
+				finish_reason: null
+			}]
+		},
+		{
+			id: "chatcmpl-text",
+			model,
+			choices: [{
+				index: 0,
+				delta: { content: " world" },
+				finish_reason: null
+			}]
+		},
+		{
+			id: "chatcmpl-text",
+			model,
+			choices: [{
+				index: 0,
+				delta: {},
+				finish_reason: "stop"
+			}]
+		}
+	];
+}
+function openAIReasoningTextChunks(model = "test-model") {
+	return [
+		{
+			id: "chatcmpl-reason",
+			model,
+			choices: [{
+				index: 0,
+				delta: {
+					role: "assistant",
+					reasoning_content: "Let me reason..."
+				},
+				finish_reason: null
+			}]
+		},
+		{
+			id: "chatcmpl-reason",
+			model,
+			choices: [{
+				index: 0,
+				delta: { content: "Answer." },
+				finish_reason: null
+			}]
+		},
+		{
+			id: "chatcmpl-reason",
+			model,
+			choices: [{
+				index: 0,
+				delta: {},
+				finish_reason: "stop"
+			}]
+		}
+	];
+}
+function openAIToolCallChunks(model = "test-model") {
+	return [
+		{
+			id: "chatcmpl-tools",
+			model,
+			choices: [{
+				index: 0,
+				delta: {
+					role: "assistant",
+					content: "Let me search."
+				},
+				finish_reason: null
+			}]
+		},
+		{
+			id: "chatcmpl-tools",
+			model,
+			choices: [{
+				index: 0,
+				delta: { tool_calls: [{
+					index: 0,
+					id: "call_abc",
+					type: "function",
+					function: {
+						name: "web_search",
+						arguments: "{\"query\""
+					}
+				}] },
+				finish_reason: null
+			}]
+		},
+		{
+			id: "chatcmpl-tools",
+			model,
+			choices: [{
+				index: 0,
+				delta: { tool_calls: [{
+					index: 0,
+					function: { arguments: ":\"weather\"}" }
+				}] },
+				finish_reason: null
+			}]
+		},
+		{
+			id: "chatcmpl-tools",
+			model,
+			choices: [{
+				index: 0,
+				delta: {},
+				finish_reason: "tool_calls"
+			}]
+		}
+	];
+}
+function sseResponseFromOpenAIChunks(chunks) {
+	const encoder = new TextEncoder();
+	return new Response(new ReadableStream({ start(controller) {
+		for (const chunk of chunks) controller.enqueue(encoder.encode(`data: ${JSON.stringify(chunk)}\n\n`));
+		controller.close();
+	} }), {
+		status: 200,
+		headers: { "Content-Type": "text/event-stream" }
+	});
+}
+//#endregion
+export { asAsyncIterable, openAIReasoningTextChunks, openAITextOnlyChunks, openAITextOnlyChunksWithUsage, openAIToolCallChunks, sseResponseFromOpenAIChunks };
+
+//# sourceMappingURL=openai_stream_fixtures.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"openai_stream_fixtures.js","names":[],"sources":["../../../src/utils/testing/openai_stream_fixtures.ts"],"sourcesContent":["import type { OpenAICompletionsStreamChunk } from \"../../language_models/openai_completions_stream.js\";\n\nexport function asAsyncIterable<T>(items: T[]): AsyncIterable<T> {\n  return {\n    async *[Symbol.asyncIterator]() {\n      for (const item of items) {\n        yield item;\n      }\n    },\n  };\n}\n\nexport function openAITextOnlyChunksWithUsage(\n  model = \"test-model\"\n): OpenAICompletionsStreamChunk[] {\n  const chunks = openAITextOnlyChunks(model);\n  const last = chunks[chunks.length - 1]!;\n  chunks[chunks.length - 1] = {\n    ...last,\n    usage: {\n      prompt_tokens: 10,\n      completion_tokens: 2,\n      total_tokens: 12,\n    },\n  };\n  return chunks;\n}\n\nexport function openAITextOnlyChunks(\n  model = \"test-model\"\n): OpenAICompletionsStreamChunk[] {\n  return [\n    {\n      id: \"chatcmpl-text\",\n      model,\n      choices: [\n        {\n          index: 0,\n          delta: { role: \"assistant\", content: \"Hello\" },\n          finish_reason: null,\n        },\n      ],\n    },\n    {\n      id: \"chatcmpl-text\",\n      model,\n      choices: [\n        {\n          index: 0,\n          delta: { content: \" world\" },\n          finish_reason: null,\n        },\n      ],\n    },\n    {\n      id: \"chatcmpl-text\",\n      model,\n      choices: [{ index: 0, delta: {}, finish_reason: \"stop\" }],\n    },\n  ];\n}\n\nexport function openAIReasoningTextChunks(\n  model = \"test-model\"\n): OpenAICompletionsStreamChunk[] {\n  return [\n    {\n      id: \"chatcmpl-reason\",\n      model,\n      choices: [\n        {\n          index: 0,\n          delta: { role: \"assistant\", reasoning_content: \"Let me reason...\" },\n          finish_reason: null,\n        },\n      ],\n    },\n    {\n      id: \"chatcmpl-reason\",\n      model,\n      choices: [\n        {\n          index: 0,\n          delta: { content: \"Answer.\" },\n          finish_reason: null,\n        },\n      ],\n    },\n    {\n      id: \"chatcmpl-reason\",\n      model,\n      choices: [{ index: 0, delta: {}, finish_reason: \"stop\" }],\n    },\n  ];\n}\n\nexport function openAIToolCallChunks(\n  model = \"test-model\"\n): OpenAICompletionsStreamChunk[] {\n  return [\n    {\n      id: \"chatcmpl-tools\",\n      model,\n      choices: [\n        {\n          index: 0,\n          delta: { role: \"assistant\", content: \"Let me search.\" },\n          finish_reason: null,\n        },\n      ],\n    },\n    {\n      id: \"chatcmpl-tools\",\n      model,\n      choices: [\n        {\n          index: 0,\n          delta: {\n            tool_calls: [\n              {\n                index: 0,\n                id: \"call_abc\",\n                type: \"function\",\n                function: { name: \"web_search\", arguments: '{\"query\"' },\n              },\n            ],\n          },\n          finish_reason: null,\n        },\n      ],\n    },\n    {\n      id: \"chatcmpl-tools\",\n      model,\n      choices: [\n        {\n          index: 0,\n          delta: {\n            tool_calls: [\n              {\n                index: 0,\n                function: { arguments: ':\"weather\"}' },\n              },\n            ],\n          },\n          finish_reason: null,\n        },\n      ],\n    },\n    {\n      id: \"chatcmpl-tools\",\n      model,\n      choices: [{ index: 0, delta: {}, finish_reason: \"tool_calls\" }],\n    },\n  ];\n}\n\nexport function sseResponseFromOpenAIChunks(\n  chunks: OpenAICompletionsStreamChunk[]\n): Response {\n  const encoder = new TextEncoder();\n  return new Response(\n    new ReadableStream({\n      start(controller) {\n        for (const chunk of chunks) {\n          controller.enqueue(\n            encoder.encode(`data: ${JSON.stringify(chunk)}\\n\\n`)\n          );\n        }\n        controller.close();\n      },\n    }),\n    { status: 200, headers: { \"Content-Type\": \"text/event-stream\" } }\n  );\n}\n"],"mappings":";AAEA,SAAgB,gBAAmB,OAA8B;AAC/D,QAAO,EACL,QAAQ,OAAO,iBAAiB;AAC9B,OAAK,MAAM,QAAQ,MACjB,OAAM;IAGX;;AAGH,SAAgB,8BACd,QAAQ,cACwB;CAChC,MAAM,SAAS,qBAAqB,MAAM;CAC1C,MAAM,OAAO,OAAO,OAAO,SAAS;AACpC,QAAO,OAAO,SAAS,KAAK;EAC1B,GAAG;EACH,OAAO;GACL,eAAe;GACf,mBAAmB;GACnB,cAAc;GACf;EACF;AACD,QAAO;;AAGT,SAAgB,qBACd,QAAQ,cACwB;AAChC,QAAO;EACL;GACE,IAAI;GACJ;GACA,SAAS,CACP;IACE,OAAO;IACP,OAAO;KAAE,MAAM;KAAa,SAAS;KAAS;IAC9C,eAAe;IAChB,CACF;GACF;EACD;GACE,IAAI;GACJ;GACA,SAAS,CACP;IACE,OAAO;IACP,OAAO,EAAE,SAAS,UAAU;IAC5B,eAAe;IAChB,CACF;GACF;EACD;GACE,IAAI;GACJ;GACA,SAAS,CAAC;IAAE,OAAO;IAAG,OAAO,EAAE;IAAE,eAAe;IAAQ,CAAC;GAC1D;EACF;;AAGH,SAAgB,0BACd,QAAQ,cACwB;AAChC,QAAO;EACL;GACE,IAAI;GACJ;GACA,SAAS,CACP;IACE,OAAO;IACP,OAAO;KAAE,MAAM;KAAa,mBAAmB;KAAoB;IACnE,eAAe;IAChB,CACF;GACF;EACD;GACE,IAAI;GACJ;GACA,SAAS,CACP;IACE,OAAO;IACP,OAAO,EAAE,SAAS,WAAW;IAC7B,eAAe;IAChB,CACF;GACF;EACD;GACE,IAAI;GACJ;GACA,SAAS,CAAC;IAAE,OAAO;IAAG,OAAO,EAAE;IAAE,eAAe;IAAQ,CAAC;GAC1D;EACF;;AAGH,SAAgB,qBACd,QAAQ,cACwB;AAChC,QAAO;EACL;GACE,IAAI;GACJ;GACA,SAAS,CACP;IACE,OAAO;IACP,OAAO;KAAE,MAAM;KAAa,SAAS;KAAkB;IACvD,eAAe;IAChB,CACF;GACF;EACD;GACE,IAAI;GACJ;GACA,SAAS,CACP;IACE,OAAO;IACP,OAAO,EACL,YAAY,CACV;KACE,OAAO;KACP,IAAI;KACJ,MAAM;KACN,UAAU;MAAE,MAAM;MAAc,WAAW;MAAY;KACxD,CACF,EACF;IACD,eAAe;IAChB,CACF;GACF;EACD;GACE,IAAI;GACJ;GACA,SAAS,CACP;IACE,OAAO;IACP,OAAO,EACL,YAAY,CACV;KACE,OAAO;KACP,UAAU,EAAE,WAAW,iBAAe;KACvC,CACF,EACF;IACD,eAAe;IAChB,CACF;GACF;EACD;GACE,IAAI;GACJ;GACA,SAAS,CAAC;IAAE,OAAO;IAAG,OAAO,EAAE;IAAE,eAAe;IAAc,CAAC;GAChE;EACF;;AAGH,SAAgB,4BACd,QACU;CACV,MAAM,UAAU,IAAI,aAAa;AACjC,QAAO,IAAI,SACT,IAAI,eAAe,EACjB,MAAM,YAAY;AAChB,OAAK,MAAM,SAAS,OAClB,YAAW,QACT,QAAQ,OAAO,SAAS,KAAK,UAAU,MAAM,CAAC,MAAM,CACrD;AAEH,aAAW,OAAO;IAErB,CAAC,EACF;EAAE,QAAQ;EAAK,SAAS,EAAE,gBAAgB,qBAAqB;EAAE,CAClE"}

package/dist/utils/testing/stream.cjs

@@ -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
+exports.streamMatchers = streamMatchers;
+
+//# sourceMappingURL=stream.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"stream.cjs","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/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