@langchain/core 1.1.29 → 1.1.31

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (122) hide show
  1. package/CHANGELOG.md +29 -0
  2. package/dist/language_models/base.cjs +1 -1
  3. package/dist/language_models/base.cjs.map +1 -1
  4. package/dist/language_models/base.d.cts +6 -0
  5. package/dist/language_models/base.d.cts.map +1 -1
  6. package/dist/language_models/base.d.ts +6 -0
  7. package/dist/language_models/base.d.ts.map +1 -1
  8. package/dist/language_models/base.js +1 -1
  9. package/dist/language_models/base.js.map +1 -1
  10. package/dist/language_models/chat_models.cjs +8 -23
  11. package/dist/language_models/chat_models.cjs.map +1 -1
  12. package/dist/language_models/chat_models.d.cts +6 -0
  13. package/dist/language_models/chat_models.d.cts.map +1 -1
  14. package/dist/language_models/chat_models.d.ts +6 -0
  15. package/dist/language_models/chat_models.d.ts.map +1 -1
  16. package/dist/language_models/chat_models.js +9 -24
  17. package/dist/language_models/chat_models.js.map +1 -1
  18. package/dist/language_models/structured_output.cjs +88 -0
  19. package/dist/language_models/structured_output.cjs.map +1 -0
  20. package/dist/language_models/structured_output.d.cts +46 -0
  21. package/dist/language_models/structured_output.d.cts.map +1 -0
  22. package/dist/language_models/structured_output.d.ts +50 -0
  23. package/dist/language_models/structured_output.d.ts.map +1 -0
  24. package/dist/language_models/structured_output.js +79 -0
  25. package/dist/language_models/structured_output.js.map +1 -0
  26. package/dist/load/import_map.cjs +10 -4
  27. package/dist/load/import_map.cjs.map +1 -1
  28. package/dist/load/import_map.js +12 -6
  29. package/dist/load/import_map.js.map +1 -1
  30. package/dist/messages/base.cjs.map +1 -1
  31. package/dist/messages/base.d.cts +2 -2
  32. package/dist/messages/base.d.cts.map +1 -1
  33. package/dist/messages/base.d.ts +2 -2
  34. package/dist/messages/base.d.ts.map +1 -1
  35. package/dist/messages/base.js.map +1 -1
  36. package/dist/output_parsers/index.cjs +4 -1
  37. package/dist/output_parsers/index.cjs.map +1 -1
  38. package/dist/output_parsers/index.d.cts +2 -1
  39. package/dist/output_parsers/index.d.ts +2 -1
  40. package/dist/output_parsers/index.js +4 -2
  41. package/dist/output_parsers/index.js.map +1 -1
  42. package/dist/output_parsers/openai_tools/index.d.cts +2 -2
  43. package/dist/output_parsers/openai_tools/index.d.ts +2 -2
  44. package/dist/output_parsers/openai_tools/json_output_tools_parsers.cjs +8 -1
  45. package/dist/output_parsers/openai_tools/json_output_tools_parsers.cjs.map +1 -1
  46. package/dist/output_parsers/openai_tools/json_output_tools_parsers.d.cts +6 -1
  47. package/dist/output_parsers/openai_tools/json_output_tools_parsers.d.cts.map +1 -1
  48. package/dist/output_parsers/openai_tools/json_output_tools_parsers.d.ts +6 -1
  49. package/dist/output_parsers/openai_tools/json_output_tools_parsers.d.ts.map +1 -1
  50. package/dist/output_parsers/openai_tools/json_output_tools_parsers.js +8 -1
  51. package/dist/output_parsers/openai_tools/json_output_tools_parsers.js.map +1 -1
  52. package/dist/output_parsers/standard_schema.cjs +40 -0
  53. package/dist/output_parsers/standard_schema.cjs.map +1 -0
  54. package/dist/output_parsers/standard_schema.d.cts +16 -0
  55. package/dist/output_parsers/standard_schema.d.cts.map +1 -0
  56. package/dist/output_parsers/standard_schema.d.ts +16 -0
  57. package/dist/output_parsers/standard_schema.d.ts.map +1 -0
  58. package/dist/output_parsers/standard_schema.js +40 -0
  59. package/dist/output_parsers/standard_schema.js.map +1 -0
  60. package/dist/runnables/base.cjs +1 -1
  61. package/dist/runnables/base.cjs.map +1 -1
  62. package/dist/runnables/base.js +2 -2
  63. package/dist/runnables/base.js.map +1 -1
  64. package/dist/testing/fake_model_builder.cjs +174 -0
  65. package/dist/testing/fake_model_builder.cjs.map +1 -0
  66. package/dist/testing/fake_model_builder.d.cts +87 -0
  67. package/dist/testing/fake_model_builder.d.cts.map +1 -0
  68. package/dist/testing/fake_model_builder.d.ts +88 -0
  69. package/dist/testing/fake_model_builder.d.ts.map +1 -0
  70. package/dist/testing/fake_model_builder.js +174 -0
  71. package/dist/testing/fake_model_builder.js.map +1 -0
  72. package/dist/testing/index.cjs +41 -0
  73. package/dist/testing/index.cjs.map +1 -0
  74. package/dist/testing/index.d.cts +3 -0
  75. package/dist/testing/index.d.ts +3 -0
  76. package/dist/testing/index.js +23 -0
  77. package/dist/testing/index.js.map +1 -0
  78. package/dist/testing/matchers.cjs +213 -0
  79. package/dist/testing/matchers.cjs.map +1 -0
  80. package/dist/testing/matchers.d.cts +94 -0
  81. package/dist/testing/matchers.d.cts.map +1 -0
  82. package/dist/testing/matchers.d.ts +94 -0
  83. package/dist/testing/matchers.d.ts.map +1 -0
  84. package/dist/testing/matchers.js +203 -0
  85. package/dist/testing/matchers.js.map +1 -0
  86. package/dist/utils/json_schema.cjs +3 -1
  87. package/dist/utils/json_schema.cjs.map +1 -1
  88. package/dist/utils/json_schema.d.cts +3 -2
  89. package/dist/utils/json_schema.d.cts.map +1 -1
  90. package/dist/utils/json_schema.d.ts +3 -2
  91. package/dist/utils/json_schema.d.ts.map +1 -1
  92. package/dist/utils/json_schema.js +3 -1
  93. package/dist/utils/json_schema.js.map +1 -1
  94. package/dist/utils/standard_schema.cjs +43 -0
  95. package/dist/utils/standard_schema.cjs.map +1 -0
  96. package/dist/utils/standard_schema.d.cts +27 -0
  97. package/dist/utils/standard_schema.d.cts.map +1 -0
  98. package/dist/utils/standard_schema.d.ts +27 -0
  99. package/dist/utils/standard_schema.d.ts.map +1 -0
  100. package/dist/utils/standard_schema.js +34 -0
  101. package/dist/utils/standard_schema.js.map +1 -0
  102. package/dist/utils/types/zod.cjs +1 -1
  103. package/dist/utils/types/zod.cjs.map +1 -1
  104. package/dist/utils/types/zod.d.cts +3 -2
  105. package/dist/utils/types/zod.d.cts.map +1 -1
  106. package/dist/utils/types/zod.d.ts +3 -2
  107. package/dist/utils/types/zod.d.ts.map +1 -1
  108. package/dist/utils/types/zod.js +1 -1
  109. package/dist/utils/types/zod.js.map +1 -1
  110. package/errors.cjs +1 -0
  111. package/errors.d.cts +1 -0
  112. package/errors.d.ts +1 -0
  113. package/errors.js +1 -0
  114. package/language_models/structured_output.cjs +1 -0
  115. package/language_models/structured_output.d.cts +1 -0
  116. package/language_models/structured_output.d.ts +1 -0
  117. package/language_models/structured_output.js +1 -0
  118. package/package.json +42 -3
  119. package/utils/standard_schema.cjs +1 -0
  120. package/utils/standard_schema.d.cts +1 -0
  121. package/utils/standard_schema.d.ts +1 -0
  122. package/utils/standard_schema.js +1 -0
@@ -0,0 +1,213 @@
1
+ const require_base = require('../messages/base.cjs');
2
+ const require_messages_tool = require('../messages/tool.cjs');
3
+ const require_ai = require('../messages/ai.cjs');
4
+ const require_human = require('../messages/human.cjs');
5
+ const require_system = require('../messages/system.cjs');
6
+ require('../messages/index.cjs');
7
+
8
+ //#region src/testing/matchers.ts
9
+ function getMessageTypeName(msg) {
10
+ if (!require_base.BaseMessage.isInstance(msg)) return typeof msg;
11
+ return msg.constructor.name || msg.type;
12
+ }
13
+ function makeMessageTypeMatcher(typeName, isInstance) {
14
+ return function(received, expected) {
15
+ const { isNot, utils } = this;
16
+ if (!isInstance(received)) return {
17
+ pass: false,
18
+ message: () => `${utils.matcherHint(`toBe${typeName}`, void 0, void 0)}\n\nExpected: ${isNot ? "not " : ""}${typeName}\nReceived: ${getMessageTypeName(received)}`,
19
+ actual: getMessageTypeName(received),
20
+ expected: typeName
21
+ };
22
+ if (expected === void 0) return {
23
+ pass: true,
24
+ message: () => `${utils.matcherHint(`toBe${typeName}`, void 0, void 0)}\n\nExpected: not ${typeName}\nReceived: ${typeName}`
25
+ };
26
+ const msg = received;
27
+ if (typeof expected === "string") return {
28
+ pass: msg.content === expected,
29
+ message: () => `${utils.matcherHint(`toBe${typeName}`, void 0, void 0)}\n\nExpected: ${typeName} with content ${utils.printExpected(expected)}\nReceived: ${typeName} with content ${utils.printReceived(msg.content)}`,
30
+ actual: msg.content,
31
+ expected
32
+ };
33
+ return {
34
+ pass: Object.entries(expected).every(([key, value]) => this.equals(msg[key], value)),
35
+ message: () => {
36
+ const receivedFields = {};
37
+ for (const key of Object.keys(expected)) receivedFields[key] = msg[key];
38
+ return `${utils.matcherHint(`toBe${typeName}`, void 0, void 0)}\n\nExpected: ${typeName} matching ${utils.printExpected(expected)}\nReceived: ${typeName} with ${utils.printReceived(receivedFields)}`;
39
+ },
40
+ actual: (() => {
41
+ const receivedFields = {};
42
+ for (const key of Object.keys(expected)) receivedFields[key] = msg[key];
43
+ return receivedFields;
44
+ })(),
45
+ expected
46
+ };
47
+ };
48
+ }
49
+ const toBeHumanMessage = makeMessageTypeMatcher("HumanMessage", require_human.HumanMessage.isInstance);
50
+ const toBeAIMessage = makeMessageTypeMatcher("AIMessage", require_ai.AIMessage.isInstance);
51
+ const toBeSystemMessage = makeMessageTypeMatcher("SystemMessage", require_system.SystemMessage.isInstance);
52
+ const toBeToolMessage = makeMessageTypeMatcher("ToolMessage", require_messages_tool.ToolMessage.isInstance);
53
+ function toHaveToolCalls(received, expected) {
54
+ const { isNot, utils } = this;
55
+ if (!require_ai.AIMessage.isInstance(received)) return {
56
+ pass: false,
57
+ message: () => `${utils.matcherHint("toHaveToolCalls")}\n\nExpected: AIMessage\nReceived: ${getMessageTypeName(received)}`
58
+ };
59
+ const actual = received.tool_calls ?? [];
60
+ if (actual.length !== expected.length) return {
61
+ pass: false,
62
+ message: () => `${utils.matcherHint("toHaveToolCalls")}\n\nExpected ${isNot ? "not " : ""}${expected.length} tool call(s), received ${actual.length}`,
63
+ actual: actual.length,
64
+ expected: expected.length
65
+ };
66
+ const unmatched = expected.filter((exp) => !actual.some((tc) => Object.entries(exp).every(([key, value]) => this.equals(tc[key], value))));
67
+ if (unmatched.length > 0) return {
68
+ pass: false,
69
+ message: () => `${utils.matcherHint("toHaveToolCalls")}\n\nCould not find matching tool call(s) for:\n${utils.printExpected(unmatched)}\nReceived tool calls: ${utils.printReceived(actual.map((tc) => ({
70
+ name: tc.name,
71
+ id: tc.id,
72
+ args: tc.args
73
+ })))}`,
74
+ actual: actual.map((tc) => ({
75
+ name: tc.name,
76
+ id: tc.id,
77
+ args: tc.args
78
+ })),
79
+ expected
80
+ };
81
+ return {
82
+ pass: true,
83
+ message: () => `${utils.matcherHint("toHaveToolCalls")}\n\nExpected AIMessage not to have matching tool calls`
84
+ };
85
+ }
86
+ function toHaveToolCallCount(received, expected) {
87
+ const { isNot, utils } = this;
88
+ if (!require_ai.AIMessage.isInstance(received)) return {
89
+ pass: false,
90
+ message: () => `${utils.matcherHint("toHaveToolCallCount")}\n\nExpected: AIMessage\nReceived: ${getMessageTypeName(received)}`
91
+ };
92
+ const actual = received.tool_calls?.length ?? 0;
93
+ return {
94
+ pass: actual === expected,
95
+ message: () => `${utils.matcherHint("toHaveToolCallCount")}\n\nExpected ${isNot ? "not " : ""}${expected} tool call(s)\nReceived: ${actual}`,
96
+ actual,
97
+ expected
98
+ };
99
+ }
100
+ function toContainToolCall(received, expected) {
101
+ const { isNot, utils } = this;
102
+ if (!require_ai.AIMessage.isInstance(received)) return {
103
+ pass: false,
104
+ message: () => `${utils.matcherHint("toContainToolCall")}\n\nExpected: AIMessage\nReceived: ${getMessageTypeName(received)}`
105
+ };
106
+ const actual = received.tool_calls ?? [];
107
+ return {
108
+ pass: actual.some((tc) => Object.entries(expected).every(([key, value]) => this.equals(tc[key], value))),
109
+ message: () => `${utils.matcherHint("toContainToolCall")}\n\nExpected AIMessage ${isNot ? "not " : ""}to contain a tool call matching ${utils.printExpected(expected)}\nReceived tool calls: ${utils.printReceived(actual.map((tc) => ({
110
+ name: tc.name,
111
+ id: tc.id
112
+ })))}`,
113
+ actual: actual.map((tc) => ({
114
+ name: tc.name,
115
+ id: tc.id
116
+ })),
117
+ expected
118
+ };
119
+ }
120
+ function toHaveToolMessages(received, expected) {
121
+ const { isNot, utils } = this;
122
+ if (!Array.isArray(received)) return {
123
+ pass: false,
124
+ message: () => `${utils.matcherHint("toHaveToolMessages")}\n\nExpected an array of messages\nReceived: ${typeof received}`
125
+ };
126
+ const toolMessages = received.filter(require_messages_tool.ToolMessage.isInstance);
127
+ if (toolMessages.length !== expected.length) return {
128
+ pass: false,
129
+ message: () => `${utils.matcherHint("toHaveToolMessages")}\n\nExpected ${isNot ? "not " : ""}${expected.length} tool message(s), found ${toolMessages.length}`,
130
+ actual: toolMessages.length,
131
+ expected: expected.length
132
+ };
133
+ for (let i = 0; i < expected.length; i++) if (!Object.entries(expected[i]).every(([key, value]) => this.equals(toolMessages[i][key], value))) return {
134
+ pass: false,
135
+ message: () => {
136
+ const receivedFields = {};
137
+ for (const key of Object.keys(expected[i])) receivedFields[key] = toolMessages[i][key];
138
+ return `${utils.matcherHint("toHaveToolMessages")}\n\nTool message at index ${i} did not match:\nExpected: ${utils.printExpected(expected[i])}\nReceived: ${utils.printReceived(receivedFields)}`;
139
+ },
140
+ actual: toolMessages[i],
141
+ expected: expected[i]
142
+ };
143
+ return {
144
+ pass: true,
145
+ message: () => `${utils.matcherHint("toHaveToolMessages")}\n\nExpected messages not to contain matching tool messages`
146
+ };
147
+ }
148
+ function toHaveBeenInterrupted(received, expectedValue) {
149
+ const { isNot, utils } = this;
150
+ const interrupts = received?.__interrupt__;
151
+ if (!(Array.isArray(interrupts) && interrupts.length > 0)) return {
152
+ pass: false,
153
+ message: () => `${utils.matcherHint("toHaveBeenInterrupted")}\n\nExpected result ${isNot ? "not " : ""}to have been interrupted\nReceived __interrupt__: ${utils.printReceived(interrupts)}`
154
+ };
155
+ if (expectedValue === void 0) return {
156
+ pass: true,
157
+ message: () => `${utils.matcherHint("toHaveBeenInterrupted")}\n\nExpected result not to have been interrupted\nReceived ${interrupts.length} interrupt(s)`
158
+ };
159
+ const actualValue = interrupts[0]?.value;
160
+ return {
161
+ pass: this.equals(actualValue, expectedValue),
162
+ message: () => `${utils.matcherHint("toHaveBeenInterrupted")}\n\nExpected interrupt value: ${utils.printExpected(expectedValue)}\nReceived interrupt value: ${utils.printReceived(actualValue)}`,
163
+ actual: actualValue,
164
+ expected: expectedValue
165
+ };
166
+ }
167
+ function toHaveStructuredResponse(received, expected) {
168
+ const { isNot, utils } = this;
169
+ const structuredResponse = received?.structuredResponse;
170
+ if (!(structuredResponse !== void 0)) return {
171
+ pass: false,
172
+ message: () => `${utils.matcherHint("toHaveStructuredResponse")}\n\nExpected result ${isNot ? "not " : ""}to have a structured response\nReceived structuredResponse: undefined`
173
+ };
174
+ if (expected === void 0) return {
175
+ pass: true,
176
+ message: () => `${utils.matcherHint("toHaveStructuredResponse")}\n\nExpected result not to have a structured response`
177
+ };
178
+ return {
179
+ pass: Object.entries(expected).every(([key, value]) => this.equals(structuredResponse[key], value)),
180
+ message: () => `${utils.matcherHint("toHaveStructuredResponse")}\n\nExpected structured response: ${utils.printExpected(expected)}\nReceived structured response: ${utils.printReceived(structuredResponse)}`,
181
+ actual: structuredResponse,
182
+ expected
183
+ };
184
+ }
185
+ /**
186
+ * All matcher functions bundled for convenient use with `expect.extend()`.
187
+ */
188
+ const langchainMatchers = {
189
+ toBeHumanMessage,
190
+ toBeAIMessage,
191
+ toBeSystemMessage,
192
+ toBeToolMessage,
193
+ toHaveToolCalls,
194
+ toHaveToolCallCount,
195
+ toContainToolCall,
196
+ toHaveToolMessages,
197
+ toHaveBeenInterrupted,
198
+ toHaveStructuredResponse
199
+ };
200
+
201
+ //#endregion
202
+ exports.langchainMatchers = langchainMatchers;
203
+ exports.toBeAIMessage = toBeAIMessage;
204
+ exports.toBeHumanMessage = toBeHumanMessage;
205
+ exports.toBeSystemMessage = toBeSystemMessage;
206
+ exports.toBeToolMessage = toBeToolMessage;
207
+ exports.toContainToolCall = toContainToolCall;
208
+ exports.toHaveBeenInterrupted = toHaveBeenInterrupted;
209
+ exports.toHaveStructuredResponse = toHaveStructuredResponse;
210
+ exports.toHaveToolCallCount = toHaveToolCallCount;
211
+ exports.toHaveToolCalls = toHaveToolCalls;
212
+ exports.toHaveToolMessages = toHaveToolMessages;
213
+ //# sourceMappingURL=matchers.cjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"matchers.cjs","names":["BaseMessage","HumanMessage","AIMessage","SystemMessage","ToolMessage"],"sources":["../../src/testing/matchers.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { BaseMessage } from \"../messages/index.js\";\nimport { HumanMessage } from \"../messages/index.js\";\nimport { AIMessage } from \"../messages/index.js\";\nimport { SystemMessage } from \"../messages/index.js\";\nimport { ToolMessage } from \"../messages/index.js\";\n\n/**\n * The `this` context that Vitest and Jest provide to custom matchers\n * via `expect.extend`.\n *\n * Compatible with both frameworks:\n * - https://vitest.dev/guide/extending-matchers.html\n * - https://jestjs.io/docs/expect#expectextendmatchers\n */\ninterface ExpectExtendThis {\n isNot?: boolean;\n equals(a: unknown, b: unknown): boolean;\n utils: {\n matcherHint(name: string, received?: string, expected?: string): 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 getMessageTypeName(msg: unknown): string {\n if (!BaseMessage.isInstance(msg)) return typeof msg;\n return msg.constructor.name || msg.type;\n}\n\nfunction makeMessageTypeMatcher(\n typeName: string,\n isInstance: (obj: unknown) => boolean\n) {\n return function (\n this: ExpectExtendThis,\n received: unknown,\n expected?: string | Record<string, unknown>\n ): ExpectationResult {\n const { isNot, utils } = this;\n\n const instancePass = isInstance(received);\n if (!instancePass) {\n return {\n pass: false,\n message: () =>\n `${utils.matcherHint(`toBe${typeName}`, undefined, undefined)}\\n\\n` +\n `Expected: ${isNot ? \"not \" : \"\"}${typeName}\\n` +\n `Received: ${getMessageTypeName(received)}`,\n actual: getMessageTypeName(received),\n expected: typeName,\n };\n }\n\n if (expected === undefined) {\n return {\n pass: true,\n message: () =>\n `${utils.matcherHint(`toBe${typeName}`, undefined, undefined)}\\n\\n` +\n `Expected: not ${typeName}\\n` +\n `Received: ${typeName}`,\n };\n }\n\n const msg = received as BaseMessage;\n if (typeof expected === \"string\") {\n const contentPass = msg.content === expected;\n return {\n pass: contentPass,\n message: () =>\n `${utils.matcherHint(`toBe${typeName}`, undefined, undefined)}\\n\\n` +\n `Expected: ${typeName} with content ${utils.printExpected(expected)}\\n` +\n `Received: ${typeName} with content ${utils.printReceived(msg.content)}`,\n actual: msg.content,\n expected,\n };\n }\n\n const fieldsPass = Object.entries(expected).every(([key, value]) =>\n this.equals((msg as any)[key], value)\n );\n return {\n pass: fieldsPass,\n message: () => {\n const receivedFields: Record<string, unknown> = {};\n for (const key of Object.keys(expected)) {\n receivedFields[key] = (msg as any)[key];\n }\n return (\n `${utils.matcherHint(`toBe${typeName}`, undefined, undefined)}\\n\\n` +\n `Expected: ${typeName} matching ${utils.printExpected(expected)}\\n` +\n `Received: ${typeName} with ${utils.printReceived(receivedFields)}`\n );\n },\n actual: (() => {\n const receivedFields: Record<string, unknown> = {};\n for (const key of Object.keys(expected)) {\n receivedFields[key] = (msg as any)[key];\n }\n return receivedFields;\n })(),\n expected,\n };\n };\n}\n\nexport const toBeHumanMessage = makeMessageTypeMatcher(\n \"HumanMessage\",\n HumanMessage.isInstance\n);\n\nexport const toBeAIMessage = makeMessageTypeMatcher(\n \"AIMessage\",\n AIMessage.isInstance\n);\n\nexport const toBeSystemMessage = makeMessageTypeMatcher(\n \"SystemMessage\",\n SystemMessage.isInstance\n);\n\nexport const toBeToolMessage = makeMessageTypeMatcher(\n \"ToolMessage\",\n ToolMessage.isInstance\n);\n\nexport function toHaveToolCalls(\n this: ExpectExtendThis,\n received: unknown,\n expected: Array<Record<string, unknown>>\n): ExpectationResult {\n const { isNot, utils } = this;\n\n if (!AIMessage.isInstance(received)) {\n return {\n pass: false,\n message: () =>\n `${utils.matcherHint(\"toHaveToolCalls\")}\\n\\n` +\n `Expected: AIMessage\\n` +\n `Received: ${getMessageTypeName(received)}`,\n };\n }\n\n const actual = received.tool_calls ?? [];\n\n if (actual.length !== expected.length) {\n return {\n pass: false,\n message: () =>\n `${utils.matcherHint(\"toHaveToolCalls\")}\\n\\n` +\n `Expected ${isNot ? \"not \" : \"\"}${expected.length} tool call(s), received ${actual.length}`,\n actual: actual.length,\n expected: expected.length,\n };\n }\n\n const unmatched = expected.filter(\n (exp) =>\n !actual.some((tc) =>\n Object.entries(exp).every(([key, value]) =>\n this.equals((tc as any)[key], value)\n )\n )\n );\n\n if (unmatched.length > 0) {\n return {\n pass: false,\n message: () =>\n `${utils.matcherHint(\"toHaveToolCalls\")}\\n\\n` +\n `Could not find matching tool call(s) for:\\n` +\n `${utils.printExpected(unmatched)}\\n` +\n `Received tool calls: ${utils.printReceived(actual.map((tc) => ({ name: tc.name, id: tc.id, args: tc.args })))}`,\n actual: actual.map((tc) => ({ name: tc.name, id: tc.id, args: tc.args })),\n expected,\n };\n }\n\n return {\n pass: true,\n message: () =>\n `${utils.matcherHint(\"toHaveToolCalls\")}\\n\\n` +\n `Expected AIMessage not to have matching tool calls`,\n };\n}\n\nexport function toHaveToolCallCount(\n this: ExpectExtendThis,\n received: unknown,\n expected: number\n): ExpectationResult {\n const { isNot, utils } = this;\n\n if (!AIMessage.isInstance(received)) {\n return {\n pass: false,\n message: () =>\n `${utils.matcherHint(\"toHaveToolCallCount\")}\\n\\n` +\n `Expected: AIMessage\\n` +\n `Received: ${getMessageTypeName(received)}`,\n };\n }\n\n const actual = received.tool_calls?.length ?? 0;\n const pass = actual === expected;\n\n return {\n pass,\n message: () =>\n `${utils.matcherHint(\"toHaveToolCallCount\")}\\n\\n` +\n `Expected ${isNot ? \"not \" : \"\"}${expected} tool call(s)\\n` +\n `Received: ${actual}`,\n actual,\n expected,\n };\n}\n\nexport function toContainToolCall(\n this: ExpectExtendThis,\n received: unknown,\n expected: Record<string, unknown>\n): ExpectationResult {\n const { isNot, utils } = this;\n\n if (!AIMessage.isInstance(received)) {\n return {\n pass: false,\n message: () =>\n `${utils.matcherHint(\"toContainToolCall\")}\\n\\n` +\n `Expected: AIMessage\\n` +\n `Received: ${getMessageTypeName(received)}`,\n };\n }\n\n const actual = received.tool_calls ?? [];\n const found = actual.some((tc) =>\n Object.entries(expected).every(([key, value]) =>\n this.equals((tc as any)[key], value)\n )\n );\n\n return {\n pass: found,\n message: () =>\n `${utils.matcherHint(\"toContainToolCall\")}\\n\\n` +\n `Expected AIMessage ${isNot ? \"not \" : \"\"}to contain a tool call matching ${utils.printExpected(expected)}\\n` +\n `Received tool calls: ${utils.printReceived(actual.map((tc) => ({ name: tc.name, id: tc.id })))}`,\n actual: actual.map((tc) => ({ name: tc.name, id: tc.id })),\n expected,\n };\n}\n\nexport function toHaveToolMessages(\n this: ExpectExtendThis,\n received: unknown,\n expected: Array<Record<string, unknown>>\n): ExpectationResult {\n const { isNot, utils } = this;\n\n if (!Array.isArray(received)) {\n return {\n pass: false,\n message: () =>\n `${utils.matcherHint(\"toHaveToolMessages\")}\\n\\n` +\n `Expected an array of messages\\n` +\n `Received: ${typeof received}`,\n };\n }\n\n const toolMessages = (received as BaseMessage[]).filter(\n ToolMessage.isInstance\n );\n\n if (toolMessages.length !== expected.length) {\n return {\n pass: false,\n message: () =>\n `${utils.matcherHint(\"toHaveToolMessages\")}\\n\\n` +\n `Expected ${isNot ? \"not \" : \"\"}${expected.length} tool message(s), found ${toolMessages.length}`,\n actual: toolMessages.length,\n expected: expected.length,\n };\n }\n\n for (let i = 0; i < expected.length; i++) {\n const match = Object.entries(expected[i]).every(([key, value]) =>\n this.equals((toolMessages[i] as any)[key], value)\n );\n if (!match) {\n return {\n pass: false,\n message: () => {\n const receivedFields: Record<string, unknown> = {};\n for (const key of Object.keys(expected[i])) {\n receivedFields[key] = (toolMessages[i] as any)[key];\n }\n return (\n `${utils.matcherHint(\"toHaveToolMessages\")}\\n\\n` +\n `Tool message at index ${i} did not match:\\n` +\n `Expected: ${utils.printExpected(expected[i])}\\n` +\n `Received: ${utils.printReceived(receivedFields)}`\n );\n },\n actual: toolMessages[i],\n expected: expected[i],\n };\n }\n }\n\n return {\n pass: true,\n message: () =>\n `${utils.matcherHint(\"toHaveToolMessages\")}\\n\\n` +\n `Expected messages not to contain matching tool messages`,\n };\n}\n\nexport function toHaveBeenInterrupted(\n this: ExpectExtendThis,\n received: unknown,\n expectedValue?: unknown\n): ExpectationResult {\n const { isNot, utils } = this;\n\n const result = received as Record<string, any>;\n const interrupts = result?.__interrupt__;\n const hasInterrupt = Array.isArray(interrupts) && interrupts.length > 0;\n\n if (!hasInterrupt) {\n return {\n pass: false,\n message: () =>\n `${utils.matcherHint(\"toHaveBeenInterrupted\")}\\n\\n` +\n `Expected result ${isNot ? \"not \" : \"\"}to have been interrupted\\n` +\n `Received __interrupt__: ${utils.printReceived(interrupts)}`,\n };\n }\n\n if (expectedValue === undefined) {\n return {\n pass: true,\n message: () =>\n `${utils.matcherHint(\"toHaveBeenInterrupted\")}\\n\\n` +\n `Expected result not to have been interrupted\\n` +\n `Received ${interrupts.length} interrupt(s)`,\n };\n }\n\n const actualValue = interrupts[0]?.value;\n const valuePass = this.equals(actualValue, expectedValue);\n\n return {\n pass: valuePass,\n message: () =>\n `${utils.matcherHint(\"toHaveBeenInterrupted\")}\\n\\n` +\n `Expected interrupt value: ${utils.printExpected(expectedValue)}\\n` +\n `Received interrupt value: ${utils.printReceived(actualValue)}`,\n actual: actualValue,\n expected: expectedValue,\n };\n}\n\nexport function toHaveStructuredResponse(\n this: ExpectExtendThis,\n received: unknown,\n expected?: Record<string, unknown>\n): ExpectationResult {\n const { isNot, utils } = this;\n\n const result = received as Record<string, any>;\n const structuredResponse = result?.structuredResponse;\n const isDefined = structuredResponse !== undefined;\n\n if (!isDefined) {\n return {\n pass: false,\n message: () =>\n `${utils.matcherHint(\"toHaveStructuredResponse\")}\\n\\n` +\n `Expected result ${isNot ? \"not \" : \"\"}to have a structured response\\n` +\n `Received structuredResponse: undefined`,\n };\n }\n\n if (expected === undefined) {\n return {\n pass: true,\n message: () =>\n `${utils.matcherHint(\"toHaveStructuredResponse\")}\\n\\n` +\n `Expected result not to have a structured response`,\n };\n }\n\n const fieldsPass = Object.entries(expected).every(([key, value]) =>\n this.equals(structuredResponse[key], value)\n );\n\n return {\n pass: fieldsPass,\n message: () =>\n `${utils.matcherHint(\"toHaveStructuredResponse\")}\\n\\n` +\n `Expected structured response: ${utils.printExpected(expected)}\\n` +\n `Received structured response: ${utils.printReceived(structuredResponse)}`,\n actual: structuredResponse,\n expected,\n };\n}\n\n/**\n * All matcher functions bundled for convenient use with `expect.extend()`.\n */\nexport const langchainMatchers = {\n toBeHumanMessage,\n toBeAIMessage,\n toBeSystemMessage,\n toBeToolMessage,\n toHaveToolCalls,\n toHaveToolCallCount,\n toContainToolCall,\n toHaveToolMessages,\n toHaveBeenInterrupted,\n toHaveStructuredResponse,\n};\n\nexport interface LangChainMatchers<R = unknown> {\n toBeHumanMessage(expected?: string | { content?: string; id?: string }): R;\n toBeAIMessage(expected?: string | { content?: string; name?: string }): R;\n toBeSystemMessage(\n expected?: string | { content?: string; additional_kwargs?: object }\n ): R;\n toBeToolMessage(\n expected?:\n | string\n | {\n content?: string;\n name?: string;\n status?: string;\n tool_call_id?: string;\n }\n ): R;\n toHaveToolCalls(\n expected: Array<{\n name?: string;\n id?: string;\n args?: Record<string, unknown>;\n }>\n ): R;\n toHaveToolCallCount(expected: number): R;\n toContainToolCall(expected: {\n name?: string;\n id?: string;\n args?: Record<string, unknown>;\n }): R;\n toHaveToolMessages(\n expected: Array<{\n content?: string;\n name?: string;\n status?: string;\n tool_call_id?: string;\n }>\n ): R;\n toHaveBeenInterrupted(expectedValue?: unknown): R;\n toHaveStructuredResponse(expected?: Record<string, unknown>): R;\n}\n\ndeclare module \"vitest\" {\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n interface Matchers<T = any> extends LangChainMatchers<T> {}\n}\n"],"mappings":";;;;;;;;AAgCA,SAAS,mBAAmB,KAAsB;AAChD,KAAI,CAACA,yBAAY,WAAW,IAAI,CAAE,QAAO,OAAO;AAChD,QAAO,IAAI,YAAY,QAAQ,IAAI;;AAGrC,SAAS,uBACP,UACA,YACA;AACA,QAAO,SAEL,UACA,UACmB;EACnB,MAAM,EAAE,OAAO,UAAU;AAGzB,MAAI,CADiB,WAAW,SAAS,CAEvC,QAAO;GACL,MAAM;GACN,eACE,GAAG,MAAM,YAAY,OAAO,YAAY,QAAW,OAAU,CAAC,gBACjD,QAAQ,SAAS,KAAK,SAAS,cAC/B,mBAAmB,SAAS;GAC3C,QAAQ,mBAAmB,SAAS;GACpC,UAAU;GACX;AAGH,MAAI,aAAa,OACf,QAAO;GACL,MAAM;GACN,eACE,GAAG,MAAM,YAAY,OAAO,YAAY,QAAW,OAAU,CAAC,oBAC7C,SAAS,cACb;GAChB;EAGH,MAAM,MAAM;AACZ,MAAI,OAAO,aAAa,SAEtB,QAAO;GACL,MAFkB,IAAI,YAAY;GAGlC,eACE,GAAG,MAAM,YAAY,OAAO,YAAY,QAAW,OAAU,CAAC,gBACjD,SAAS,gBAAgB,MAAM,cAAc,SAAS,CAAC,cACvD,SAAS,gBAAgB,MAAM,cAAc,IAAI,QAAQ;GACxE,QAAQ,IAAI;GACZ;GACD;AAMH,SAAO;GACL,MAJiB,OAAO,QAAQ,SAAS,CAAC,OAAO,CAAC,KAAK,WACvD,KAAK,OAAQ,IAAY,MAAM,MAAM,CACtC;GAGC,eAAe;IACb,MAAM,iBAA0C,EAAE;AAClD,SAAK,MAAM,OAAO,OAAO,KAAK,SAAS,CACrC,gBAAe,OAAQ,IAAY;AAErC,WACE,GAAG,MAAM,YAAY,OAAO,YAAY,QAAW,OAAU,CAAC,gBACjD,SAAS,YAAY,MAAM,cAAc,SAAS,CAAC,cACnD,SAAS,QAAQ,MAAM,cAAc,eAAe;;GAGrE,eAAe;IACb,MAAM,iBAA0C,EAAE;AAClD,SAAK,MAAM,OAAO,OAAO,KAAK,SAAS,CACrC,gBAAe,OAAQ,IAAY;AAErC,WAAO;OACL;GACJ;GACD;;;AAIL,MAAa,mBAAmB,uBAC9B,gBACAC,2BAAa,WACd;AAED,MAAa,gBAAgB,uBAC3B,aACAC,qBAAU,WACX;AAED,MAAa,oBAAoB,uBAC/B,iBACAC,6BAAc,WACf;AAED,MAAa,kBAAkB,uBAC7B,eACAC,kCAAY,WACb;AAED,SAAgB,gBAEd,UACA,UACmB;CACnB,MAAM,EAAE,OAAO,UAAU;AAEzB,KAAI,CAACF,qBAAU,WAAW,SAAS,CACjC,QAAO;EACL,MAAM;EACN,eACE,GAAG,MAAM,YAAY,kBAAkB,CAAC,qCAE3B,mBAAmB,SAAS;EAC5C;CAGH,MAAM,SAAS,SAAS,cAAc,EAAE;AAExC,KAAI,OAAO,WAAW,SAAS,OAC7B,QAAO;EACL,MAAM;EACN,eACE,GAAG,MAAM,YAAY,kBAAkB,CAAC,eAC5B,QAAQ,SAAS,KAAK,SAAS,OAAO,0BAA0B,OAAO;EACrF,QAAQ,OAAO;EACf,UAAU,SAAS;EACpB;CAGH,MAAM,YAAY,SAAS,QACxB,QACC,CAAC,OAAO,MAAM,OACZ,OAAO,QAAQ,IAAI,CAAC,OAAO,CAAC,KAAK,WAC/B,KAAK,OAAQ,GAAW,MAAM,MAAM,CACrC,CACF,CACJ;AAED,KAAI,UAAU,SAAS,EACrB,QAAO;EACL,MAAM;EACN,eACE,GAAG,MAAM,YAAY,kBAAkB,CAAC,iDAErC,MAAM,cAAc,UAAU,CAAC,yBACV,MAAM,cAAc,OAAO,KAAK,QAAQ;GAAE,MAAM,GAAG;GAAM,IAAI,GAAG;GAAI,MAAM,GAAG;GAAM,EAAE,CAAC;EAChH,QAAQ,OAAO,KAAK,QAAQ;GAAE,MAAM,GAAG;GAAM,IAAI,GAAG;GAAI,MAAM,GAAG;GAAM,EAAE;EACzE;EACD;AAGH,QAAO;EACL,MAAM;EACN,eACE,GAAG,MAAM,YAAY,kBAAkB,CAAC;EAE3C;;AAGH,SAAgB,oBAEd,UACA,UACmB;CACnB,MAAM,EAAE,OAAO,UAAU;AAEzB,KAAI,CAACA,qBAAU,WAAW,SAAS,CACjC,QAAO;EACL,MAAM;EACN,eACE,GAAG,MAAM,YAAY,sBAAsB,CAAC,qCAE/B,mBAAmB,SAAS;EAC5C;CAGH,MAAM,SAAS,SAAS,YAAY,UAAU;AAG9C,QAAO;EACL,MAHW,WAAW;EAItB,eACE,GAAG,MAAM,YAAY,sBAAsB,CAAC,eAChC,QAAQ,SAAS,KAAK,SAAS,2BAC9B;EACf;EACA;EACD;;AAGH,SAAgB,kBAEd,UACA,UACmB;CACnB,MAAM,EAAE,OAAO,UAAU;AAEzB,KAAI,CAACA,qBAAU,WAAW,SAAS,CACjC,QAAO;EACL,MAAM;EACN,eACE,GAAG,MAAM,YAAY,oBAAoB,CAAC,qCAE7B,mBAAmB,SAAS;EAC5C;CAGH,MAAM,SAAS,SAAS,cAAc,EAAE;AAOxC,QAAO;EACL,MAPY,OAAO,MAAM,OACzB,OAAO,QAAQ,SAAS,CAAC,OAAO,CAAC,KAAK,WACpC,KAAK,OAAQ,GAAW,MAAM,MAAM,CACrC,CACF;EAIC,eACE,GAAG,MAAM,YAAY,oBAAoB,CAAC,yBACpB,QAAQ,SAAS,GAAG,kCAAkC,MAAM,cAAc,SAAS,CAAC,yBAClF,MAAM,cAAc,OAAO,KAAK,QAAQ;GAAE,MAAM,GAAG;GAAM,IAAI,GAAG;GAAI,EAAE,CAAC;EACjG,QAAQ,OAAO,KAAK,QAAQ;GAAE,MAAM,GAAG;GAAM,IAAI,GAAG;GAAI,EAAE;EAC1D;EACD;;AAGH,SAAgB,mBAEd,UACA,UACmB;CACnB,MAAM,EAAE,OAAO,UAAU;AAEzB,KAAI,CAAC,MAAM,QAAQ,SAAS,CAC1B,QAAO;EACL,MAAM;EACN,eACE,GAAG,MAAM,YAAY,qBAAqB,CAAC,+CAE9B,OAAO;EACvB;CAGH,MAAM,eAAgB,SAA2B,OAC/CE,kCAAY,WACb;AAED,KAAI,aAAa,WAAW,SAAS,OACnC,QAAO;EACL,MAAM;EACN,eACE,GAAG,MAAM,YAAY,qBAAqB,CAAC,eAC/B,QAAQ,SAAS,KAAK,SAAS,OAAO,0BAA0B,aAAa;EAC3F,QAAQ,aAAa;EACrB,UAAU,SAAS;EACpB;AAGH,MAAK,IAAI,IAAI,GAAG,IAAI,SAAS,QAAQ,IAInC,KAAI,CAHU,OAAO,QAAQ,SAAS,GAAG,CAAC,OAAO,CAAC,KAAK,WACrD,KAAK,OAAQ,aAAa,GAAW,MAAM,MAAM,CAClD,CAEC,QAAO;EACL,MAAM;EACN,eAAe;GACb,MAAM,iBAA0C,EAAE;AAClD,QAAK,MAAM,OAAO,OAAO,KAAK,SAAS,GAAG,CACxC,gBAAe,OAAQ,aAAa,GAAW;AAEjD,UACE,GAAG,MAAM,YAAY,qBAAqB,CAAC,4BAClB,EAAE,6BACd,MAAM,cAAc,SAAS,GAAG,CAAC,cACjC,MAAM,cAAc,eAAe;;EAGpD,QAAQ,aAAa;EACrB,UAAU,SAAS;EACpB;AAIL,QAAO;EACL,MAAM;EACN,eACE,GAAG,MAAM,YAAY,qBAAqB,CAAC;EAE9C;;AAGH,SAAgB,sBAEd,UACA,eACmB;CACnB,MAAM,EAAE,OAAO,UAAU;CAGzB,MAAM,aADS,UACY;AAG3B,KAAI,EAFiB,MAAM,QAAQ,WAAW,IAAI,WAAW,SAAS,GAGpE,QAAO;EACL,MAAM;EACN,eACE,GAAG,MAAM,YAAY,wBAAwB,CAAC,sBAC3B,QAAQ,SAAS,GAAG,oDACZ,MAAM,cAAc,WAAW;EAC7D;AAGH,KAAI,kBAAkB,OACpB,QAAO;EACL,MAAM;EACN,eACE,GAAG,MAAM,YAAY,wBAAwB,CAAC,6DAElC,WAAW,OAAO;EACjC;CAGH,MAAM,cAAc,WAAW,IAAI;AAGnC,QAAO;EACL,MAHgB,KAAK,OAAO,aAAa,cAAc;EAIvD,eACE,GAAG,MAAM,YAAY,wBAAwB,CAAC,gCACjB,MAAM,cAAc,cAAc,CAAC,8BACnC,MAAM,cAAc,YAAY;EAC/D,QAAQ;EACR,UAAU;EACX;;AAGH,SAAgB,yBAEd,UACA,UACmB;CACnB,MAAM,EAAE,OAAO,UAAU;CAGzB,MAAM,qBADS,UACoB;AAGnC,KAAI,EAFc,uBAAuB,QAGvC,QAAO;EACL,MAAM;EACN,eACE,GAAG,MAAM,YAAY,2BAA2B,CAAC,sBAC9B,QAAQ,SAAS,GAAG;EAE1C;AAGH,KAAI,aAAa,OACf,QAAO;EACL,MAAM;EACN,eACE,GAAG,MAAM,YAAY,2BAA2B,CAAC;EAEpD;AAOH,QAAO;EACL,MALiB,OAAO,QAAQ,SAAS,CAAC,OAAO,CAAC,KAAK,WACvD,KAAK,OAAO,mBAAmB,MAAM,MAAM,CAC5C;EAIC,eACE,GAAG,MAAM,YAAY,2BAA2B,CAAC,oCAChB,MAAM,cAAc,SAAS,CAAC,kCAC9B,MAAM,cAAc,mBAAmB;EAC1E,QAAQ;EACR;EACD;;;;;AAMH,MAAa,oBAAoB;CAC/B;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACD"}
@@ -0,0 +1,94 @@
1
+ //#region src/testing/matchers.d.ts
2
+ /**
3
+ * The `this` context that Vitest and Jest provide to custom matchers
4
+ * via `expect.extend`.
5
+ *
6
+ * Compatible with both frameworks:
7
+ * - https://vitest.dev/guide/extending-matchers.html
8
+ * - https://jestjs.io/docs/expect#expectextendmatchers
9
+ */
10
+ interface ExpectExtendThis {
11
+ isNot?: boolean;
12
+ equals(a: unknown, b: unknown): boolean;
13
+ utils: {
14
+ matcherHint(name: string, received?: string, expected?: string): string;
15
+ printReceived(value: unknown): string;
16
+ printExpected(value: unknown): string;
17
+ };
18
+ }
19
+ interface ExpectationResult {
20
+ pass: boolean;
21
+ message: () => string;
22
+ actual?: unknown;
23
+ expected?: unknown;
24
+ }
25
+ declare const toBeHumanMessage: (this: ExpectExtendThis, received: unknown, expected?: string | Record<string, unknown> | undefined) => ExpectationResult;
26
+ declare const toBeAIMessage: (this: ExpectExtendThis, received: unknown, expected?: string | Record<string, unknown> | undefined) => ExpectationResult;
27
+ declare const toBeSystemMessage: (this: ExpectExtendThis, received: unknown, expected?: string | Record<string, unknown> | undefined) => ExpectationResult;
28
+ declare const toBeToolMessage: (this: ExpectExtendThis, received: unknown, expected?: string | Record<string, unknown> | undefined) => ExpectationResult;
29
+ declare function toHaveToolCalls(this: ExpectExtendThis, received: unknown, expected: Array<Record<string, unknown>>): ExpectationResult;
30
+ declare function toHaveToolCallCount(this: ExpectExtendThis, received: unknown, expected: number): ExpectationResult;
31
+ declare function toContainToolCall(this: ExpectExtendThis, received: unknown, expected: Record<string, unknown>): ExpectationResult;
32
+ declare function toHaveToolMessages(this: ExpectExtendThis, received: unknown, expected: Array<Record<string, unknown>>): ExpectationResult;
33
+ declare function toHaveBeenInterrupted(this: ExpectExtendThis, received: unknown, expectedValue?: unknown): ExpectationResult;
34
+ declare function toHaveStructuredResponse(this: ExpectExtendThis, received: unknown, expected?: Record<string, unknown>): ExpectationResult;
35
+ /**
36
+ * All matcher functions bundled for convenient use with `expect.extend()`.
37
+ */
38
+ declare const langchainMatchers: {
39
+ toBeHumanMessage: (this: ExpectExtendThis, received: unknown, expected?: string | Record<string, unknown> | undefined) => ExpectationResult;
40
+ toBeAIMessage: (this: ExpectExtendThis, received: unknown, expected?: string | Record<string, unknown> | undefined) => ExpectationResult;
41
+ toBeSystemMessage: (this: ExpectExtendThis, received: unknown, expected?: string | Record<string, unknown> | undefined) => ExpectationResult;
42
+ toBeToolMessage: (this: ExpectExtendThis, received: unknown, expected?: string | Record<string, unknown> | undefined) => ExpectationResult;
43
+ toHaveToolCalls: typeof toHaveToolCalls;
44
+ toHaveToolCallCount: typeof toHaveToolCallCount;
45
+ toContainToolCall: typeof toContainToolCall;
46
+ toHaveToolMessages: typeof toHaveToolMessages;
47
+ toHaveBeenInterrupted: typeof toHaveBeenInterrupted;
48
+ toHaveStructuredResponse: typeof toHaveStructuredResponse;
49
+ };
50
+ interface LangChainMatchers<R = unknown> {
51
+ toBeHumanMessage(expected?: string | {
52
+ content?: string;
53
+ id?: string;
54
+ }): R;
55
+ toBeAIMessage(expected?: string | {
56
+ content?: string;
57
+ name?: string;
58
+ }): R;
59
+ toBeSystemMessage(expected?: string | {
60
+ content?: string;
61
+ additional_kwargs?: object;
62
+ }): R;
63
+ toBeToolMessage(expected?: string | {
64
+ content?: string;
65
+ name?: string;
66
+ status?: string;
67
+ tool_call_id?: string;
68
+ }): R;
69
+ toHaveToolCalls(expected: Array<{
70
+ name?: string;
71
+ id?: string;
72
+ args?: Record<string, unknown>;
73
+ }>): R;
74
+ toHaveToolCallCount(expected: number): R;
75
+ toContainToolCall(expected: {
76
+ name?: string;
77
+ id?: string;
78
+ args?: Record<string, unknown>;
79
+ }): R;
80
+ toHaveToolMessages(expected: Array<{
81
+ content?: string;
82
+ name?: string;
83
+ status?: string;
84
+ tool_call_id?: string;
85
+ }>): R;
86
+ toHaveBeenInterrupted(expectedValue?: unknown): R;
87
+ toHaveStructuredResponse(expected?: Record<string, unknown>): R;
88
+ }
89
+ declare module "vitest" {
90
+ interface Matchers<T = any> extends LangChainMatchers<T> {}
91
+ }
92
+ //#endregion
93
+ export { LangChainMatchers, langchainMatchers, toBeAIMessage, toBeHumanMessage, toBeSystemMessage, toBeToolMessage, toContainToolCall, toHaveBeenInterrupted, toHaveStructuredResponse, toHaveToolCallCount, toHaveToolCalls, toHaveToolMessages };
94
+ //# sourceMappingURL=matchers.d.cts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"matchers.d.cts","names":[],"sources":["../../src/testing/matchers.ts"],"mappings":";;;;;;;;;UAeU,gBAAA;EACR,KAAA;EACA,MAAA,CAAO,CAAA,WAAY,CAAA;EACnB,KAAA;IACE,WAAA,CAAY,IAAA,UAAc,QAAA,WAAmB,QAAA;IAC7C,aAAA,CAAc,KAAA;IACd,aAAA,CAAc,KAAA;EAAA;AAAA;AAAA,UAIR,iBAAA;EACR,IAAA;EACA,OAAA;EACA,MAAA;EACA,QAAA;AAAA;AAAA,cAoFW,gBAAA,GAAgB,IAAA,EAAA,gBAAA,EAAA,QAAA,WAAA,QAAA,YAAA,MAAA,kCAAA,iBAAA;AAAA,cAKhB,aAAA,GAAa,IAAA,EAAA,gBAAA,EAAA,QAAA,WAAA,QAAA,YAAA,MAAA,kCAAA,iBAAA;AAAA,cAKb,iBAAA,GAAiB,IAAA,EAAA,gBAAA,EAAA,QAAA,WAAA,QAAA,YAAA,MAAA,kCAAA,iBAAA;AAAA,cAKjB,eAAA,GAAe,IAAA,EAAA,gBAAA,EAAA,QAAA,WAAA,QAAA,YAAA,MAAA,kCAAA,iBAAA;AAAA,iBAKZ,eAAA,CACd,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,QAAA,EAAU,KAAA,CAAM,MAAA,qBACf,iBAAA;AAAA,iBAwDa,mBAAA,CACd,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,QAAA,WACC,iBAAA;AAAA,iBA2Ba,iBAAA,CACd,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,QAAA,EAAU,MAAA,oBACT,iBAAA;AAAA,iBA+Ba,kBAAA,CACd,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,QAAA,EAAU,KAAA,CAAM,MAAA,qBACf,iBAAA;AAAA,iBA6Da,qBAAA,CACd,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,aAAA,aACC,iBAAA;AAAA,iBAyCa,wBAAA,CACd,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,QAAA,GAAW,MAAA,oBACV,iBAAA;;;;cA4CU,iBAAA;;;;;;;;;;;;UAaI,iBAAA;EACf,gBAAA,CAAiB,QAAA;IAAsB,OAAA;IAAkB,EAAA;EAAA,IAAgB,CAAA;EACzE,aAAA,CAAc,QAAA;IAAsB,OAAA;IAAkB,IAAA;EAAA,IAAkB,CAAA;EACxE,iBAAA,CACE,QAAA;IAAsB,OAAA;IAAkB,iBAAA;EAAA,IACvC,CAAA;EACH,eAAA,CACE,QAAA;IAGM,OAAA;IACA,IAAA;IACA,MAAA;IACA,YAAA;EAAA,IAEL,CAAA;EACH,eAAA,CACE,QAAA,EAAU,KAAA;IACR,IAAA;IACA,EAAA;IACA,IAAA,GAAO,MAAA;EAAA,KAER,CAAA;EACH,mBAAA,CAAoB,QAAA,WAAmB,CAAA;EACvC,iBAAA,CAAkB,QAAA;IAChB,IAAA;IACA,EAAA;IACA,IAAA,GAAO,MAAA;EAAA,IACL,CAAA;EACJ,kBAAA,CACE,QAAA,EAAU,KAAA;IACR,OAAA;IACA,IAAA;IACA,MAAA;IACA,YAAA;EAAA,KAED,CAAA;EACH,qBAAA,CAAsB,aAAA,aAA0B,CAAA;EAChD,wBAAA,CAAyB,QAAA,GAAW,MAAA,oBAA0B,CAAA;AAAA;AAAA;EAAA,UAKpD,QAAA,kBAA0B,iBAAA,CAAkB,CAAA;AAAA"}
@@ -0,0 +1,94 @@
1
+ //#region src/testing/matchers.d.ts
2
+ /**
3
+ * The `this` context that Vitest and Jest provide to custom matchers
4
+ * via `expect.extend`.
5
+ *
6
+ * Compatible with both frameworks:
7
+ * - https://vitest.dev/guide/extending-matchers.html
8
+ * - https://jestjs.io/docs/expect#expectextendmatchers
9
+ */
10
+ interface ExpectExtendThis {
11
+ isNot?: boolean;
12
+ equals(a: unknown, b: unknown): boolean;
13
+ utils: {
14
+ matcherHint(name: string, received?: string, expected?: string): string;
15
+ printReceived(value: unknown): string;
16
+ printExpected(value: unknown): string;
17
+ };
18
+ }
19
+ interface ExpectationResult {
20
+ pass: boolean;
21
+ message: () => string;
22
+ actual?: unknown;
23
+ expected?: unknown;
24
+ }
25
+ declare const toBeHumanMessage: (this: ExpectExtendThis, received: unknown, expected?: string | Record<string, unknown> | undefined) => ExpectationResult;
26
+ declare const toBeAIMessage: (this: ExpectExtendThis, received: unknown, expected?: string | Record<string, unknown> | undefined) => ExpectationResult;
27
+ declare const toBeSystemMessage: (this: ExpectExtendThis, received: unknown, expected?: string | Record<string, unknown> | undefined) => ExpectationResult;
28
+ declare const toBeToolMessage: (this: ExpectExtendThis, received: unknown, expected?: string | Record<string, unknown> | undefined) => ExpectationResult;
29
+ declare function toHaveToolCalls(this: ExpectExtendThis, received: unknown, expected: Array<Record<string, unknown>>): ExpectationResult;
30
+ declare function toHaveToolCallCount(this: ExpectExtendThis, received: unknown, expected: number): ExpectationResult;
31
+ declare function toContainToolCall(this: ExpectExtendThis, received: unknown, expected: Record<string, unknown>): ExpectationResult;
32
+ declare function toHaveToolMessages(this: ExpectExtendThis, received: unknown, expected: Array<Record<string, unknown>>): ExpectationResult;
33
+ declare function toHaveBeenInterrupted(this: ExpectExtendThis, received: unknown, expectedValue?: unknown): ExpectationResult;
34
+ declare function toHaveStructuredResponse(this: ExpectExtendThis, received: unknown, expected?: Record<string, unknown>): ExpectationResult;
35
+ /**
36
+ * All matcher functions bundled for convenient use with `expect.extend()`.
37
+ */
38
+ declare const langchainMatchers: {
39
+ toBeHumanMessage: (this: ExpectExtendThis, received: unknown, expected?: string | Record<string, unknown> | undefined) => ExpectationResult;
40
+ toBeAIMessage: (this: ExpectExtendThis, received: unknown, expected?: string | Record<string, unknown> | undefined) => ExpectationResult;
41
+ toBeSystemMessage: (this: ExpectExtendThis, received: unknown, expected?: string | Record<string, unknown> | undefined) => ExpectationResult;
42
+ toBeToolMessage: (this: ExpectExtendThis, received: unknown, expected?: string | Record<string, unknown> | undefined) => ExpectationResult;
43
+ toHaveToolCalls: typeof toHaveToolCalls;
44
+ toHaveToolCallCount: typeof toHaveToolCallCount;
45
+ toContainToolCall: typeof toContainToolCall;
46
+ toHaveToolMessages: typeof toHaveToolMessages;
47
+ toHaveBeenInterrupted: typeof toHaveBeenInterrupted;
48
+ toHaveStructuredResponse: typeof toHaveStructuredResponse;
49
+ };
50
+ interface LangChainMatchers<R = unknown> {
51
+ toBeHumanMessage(expected?: string | {
52
+ content?: string;
53
+ id?: string;
54
+ }): R;
55
+ toBeAIMessage(expected?: string | {
56
+ content?: string;
57
+ name?: string;
58
+ }): R;
59
+ toBeSystemMessage(expected?: string | {
60
+ content?: string;
61
+ additional_kwargs?: object;
62
+ }): R;
63
+ toBeToolMessage(expected?: string | {
64
+ content?: string;
65
+ name?: string;
66
+ status?: string;
67
+ tool_call_id?: string;
68
+ }): R;
69
+ toHaveToolCalls(expected: Array<{
70
+ name?: string;
71
+ id?: string;
72
+ args?: Record<string, unknown>;
73
+ }>): R;
74
+ toHaveToolCallCount(expected: number): R;
75
+ toContainToolCall(expected: {
76
+ name?: string;
77
+ id?: string;
78
+ args?: Record<string, unknown>;
79
+ }): R;
80
+ toHaveToolMessages(expected: Array<{
81
+ content?: string;
82
+ name?: string;
83
+ status?: string;
84
+ tool_call_id?: string;
85
+ }>): R;
86
+ toHaveBeenInterrupted(expectedValue?: unknown): R;
87
+ toHaveStructuredResponse(expected?: Record<string, unknown>): R;
88
+ }
89
+ declare module "vitest" {
90
+ interface Matchers<T = any> extends LangChainMatchers<T> {}
91
+ }
92
+ //#endregion
93
+ export { LangChainMatchers, langchainMatchers, toBeAIMessage, toBeHumanMessage, toBeSystemMessage, toBeToolMessage, toContainToolCall, toHaveBeenInterrupted, toHaveStructuredResponse, toHaveToolCallCount, toHaveToolCalls, toHaveToolMessages };
94
+ //# sourceMappingURL=matchers.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"matchers.d.ts","names":[],"sources":["../../src/testing/matchers.ts"],"mappings":";;;;;;;;;UAeU,gBAAA;EACR,KAAA;EACA,MAAA,CAAO,CAAA,WAAY,CAAA;EACnB,KAAA;IACE,WAAA,CAAY,IAAA,UAAc,QAAA,WAAmB,QAAA;IAC7C,aAAA,CAAc,KAAA;IACd,aAAA,CAAc,KAAA;EAAA;AAAA;AAAA,UAIR,iBAAA;EACR,IAAA;EACA,OAAA;EACA,MAAA;EACA,QAAA;AAAA;AAAA,cAoFW,gBAAA,GAAgB,IAAA,EAAA,gBAAA,EAAA,QAAA,WAAA,QAAA,YAAA,MAAA,kCAAA,iBAAA;AAAA,cAKhB,aAAA,GAAa,IAAA,EAAA,gBAAA,EAAA,QAAA,WAAA,QAAA,YAAA,MAAA,kCAAA,iBAAA;AAAA,cAKb,iBAAA,GAAiB,IAAA,EAAA,gBAAA,EAAA,QAAA,WAAA,QAAA,YAAA,MAAA,kCAAA,iBAAA;AAAA,cAKjB,eAAA,GAAe,IAAA,EAAA,gBAAA,EAAA,QAAA,WAAA,QAAA,YAAA,MAAA,kCAAA,iBAAA;AAAA,iBAKZ,eAAA,CACd,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,QAAA,EAAU,KAAA,CAAM,MAAA,qBACf,iBAAA;AAAA,iBAwDa,mBAAA,CACd,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,QAAA,WACC,iBAAA;AAAA,iBA2Ba,iBAAA,CACd,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,QAAA,EAAU,MAAA,oBACT,iBAAA;AAAA,iBA+Ba,kBAAA,CACd,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,QAAA,EAAU,KAAA,CAAM,MAAA,qBACf,iBAAA;AAAA,iBA6Da,qBAAA,CACd,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,aAAA,aACC,iBAAA;AAAA,iBAyCa,wBAAA,CACd,IAAA,EAAM,gBAAA,EACN,QAAA,WACA,QAAA,GAAW,MAAA,oBACV,iBAAA;;;;cA4CU,iBAAA;;;;;;;;;;;;UAaI,iBAAA;EACf,gBAAA,CAAiB,QAAA;IAAsB,OAAA;IAAkB,EAAA;EAAA,IAAgB,CAAA;EACzE,aAAA,CAAc,QAAA;IAAsB,OAAA;IAAkB,IAAA;EAAA,IAAkB,CAAA;EACxE,iBAAA,CACE,QAAA;IAAsB,OAAA;IAAkB,iBAAA;EAAA,IACvC,CAAA;EACH,eAAA,CACE,QAAA;IAGM,OAAA;IACA,IAAA;IACA,MAAA;IACA,YAAA;EAAA,IAEL,CAAA;EACH,eAAA,CACE,QAAA,EAAU,KAAA;IACR,IAAA;IACA,EAAA;IACA,IAAA,GAAO,MAAA;EAAA,KAER,CAAA;EACH,mBAAA,CAAoB,QAAA,WAAmB,CAAA;EACvC,iBAAA,CAAkB,QAAA;IAChB,IAAA;IACA,EAAA;IACA,IAAA,GAAO,MAAA;EAAA,IACL,CAAA;EACJ,kBAAA,CACE,QAAA,EAAU,KAAA;IACR,OAAA;IACA,IAAA;IACA,MAAA;IACA,YAAA;EAAA,KAED,CAAA;EACH,qBAAA,CAAsB,aAAA,aAA0B,CAAA;EAChD,wBAAA,CAAyB,QAAA,GAAW,MAAA,oBAA0B,CAAA;AAAA;AAAA;EAAA,UAKpD,QAAA,kBAA0B,iBAAA,CAAkB,CAAA;AAAA"}
@@ -0,0 +1,203 @@
1
+ import { BaseMessage } from "../messages/base.js";
2
+ import { ToolMessage } from "../messages/tool.js";
3
+ import { AIMessage } from "../messages/ai.js";
4
+ import { HumanMessage } from "../messages/human.js";
5
+ import { SystemMessage } from "../messages/system.js";
6
+ import "../messages/index.js";
7
+
8
+ //#region src/testing/matchers.ts
9
+ function getMessageTypeName(msg) {
10
+ if (!BaseMessage.isInstance(msg)) return typeof msg;
11
+ return msg.constructor.name || msg.type;
12
+ }
13
+ function makeMessageTypeMatcher(typeName, isInstance) {
14
+ return function(received, expected) {
15
+ const { isNot, utils } = this;
16
+ if (!isInstance(received)) return {
17
+ pass: false,
18
+ message: () => `${utils.matcherHint(`toBe${typeName}`, void 0, void 0)}\n\nExpected: ${isNot ? "not " : ""}${typeName}\nReceived: ${getMessageTypeName(received)}`,
19
+ actual: getMessageTypeName(received),
20
+ expected: typeName
21
+ };
22
+ if (expected === void 0) return {
23
+ pass: true,
24
+ message: () => `${utils.matcherHint(`toBe${typeName}`, void 0, void 0)}\n\nExpected: not ${typeName}\nReceived: ${typeName}`
25
+ };
26
+ const msg = received;
27
+ if (typeof expected === "string") return {
28
+ pass: msg.content === expected,
29
+ message: () => `${utils.matcherHint(`toBe${typeName}`, void 0, void 0)}\n\nExpected: ${typeName} with content ${utils.printExpected(expected)}\nReceived: ${typeName} with content ${utils.printReceived(msg.content)}`,
30
+ actual: msg.content,
31
+ expected
32
+ };
33
+ return {
34
+ pass: Object.entries(expected).every(([key, value]) => this.equals(msg[key], value)),
35
+ message: () => {
36
+ const receivedFields = {};
37
+ for (const key of Object.keys(expected)) receivedFields[key] = msg[key];
38
+ return `${utils.matcherHint(`toBe${typeName}`, void 0, void 0)}\n\nExpected: ${typeName} matching ${utils.printExpected(expected)}\nReceived: ${typeName} with ${utils.printReceived(receivedFields)}`;
39
+ },
40
+ actual: (() => {
41
+ const receivedFields = {};
42
+ for (const key of Object.keys(expected)) receivedFields[key] = msg[key];
43
+ return receivedFields;
44
+ })(),
45
+ expected
46
+ };
47
+ };
48
+ }
49
+ const toBeHumanMessage = makeMessageTypeMatcher("HumanMessage", HumanMessage.isInstance);
50
+ const toBeAIMessage = makeMessageTypeMatcher("AIMessage", AIMessage.isInstance);
51
+ const toBeSystemMessage = makeMessageTypeMatcher("SystemMessage", SystemMessage.isInstance);
52
+ const toBeToolMessage = makeMessageTypeMatcher("ToolMessage", ToolMessage.isInstance);
53
+ function toHaveToolCalls(received, expected) {
54
+ const { isNot, utils } = this;
55
+ if (!AIMessage.isInstance(received)) return {
56
+ pass: false,
57
+ message: () => `${utils.matcherHint("toHaveToolCalls")}\n\nExpected: AIMessage\nReceived: ${getMessageTypeName(received)}`
58
+ };
59
+ const actual = received.tool_calls ?? [];
60
+ if (actual.length !== expected.length) return {
61
+ pass: false,
62
+ message: () => `${utils.matcherHint("toHaveToolCalls")}\n\nExpected ${isNot ? "not " : ""}${expected.length} tool call(s), received ${actual.length}`,
63
+ actual: actual.length,
64
+ expected: expected.length
65
+ };
66
+ const unmatched = expected.filter((exp) => !actual.some((tc) => Object.entries(exp).every(([key, value]) => this.equals(tc[key], value))));
67
+ if (unmatched.length > 0) return {
68
+ pass: false,
69
+ message: () => `${utils.matcherHint("toHaveToolCalls")}\n\nCould not find matching tool call(s) for:\n${utils.printExpected(unmatched)}\nReceived tool calls: ${utils.printReceived(actual.map((tc) => ({
70
+ name: tc.name,
71
+ id: tc.id,
72
+ args: tc.args
73
+ })))}`,
74
+ actual: actual.map((tc) => ({
75
+ name: tc.name,
76
+ id: tc.id,
77
+ args: tc.args
78
+ })),
79
+ expected
80
+ };
81
+ return {
82
+ pass: true,
83
+ message: () => `${utils.matcherHint("toHaveToolCalls")}\n\nExpected AIMessage not to have matching tool calls`
84
+ };
85
+ }
86
+ function toHaveToolCallCount(received, expected) {
87
+ const { isNot, utils } = this;
88
+ if (!AIMessage.isInstance(received)) return {
89
+ pass: false,
90
+ message: () => `${utils.matcherHint("toHaveToolCallCount")}\n\nExpected: AIMessage\nReceived: ${getMessageTypeName(received)}`
91
+ };
92
+ const actual = received.tool_calls?.length ?? 0;
93
+ return {
94
+ pass: actual === expected,
95
+ message: () => `${utils.matcherHint("toHaveToolCallCount")}\n\nExpected ${isNot ? "not " : ""}${expected} tool call(s)\nReceived: ${actual}`,
96
+ actual,
97
+ expected
98
+ };
99
+ }
100
+ function toContainToolCall(received, expected) {
101
+ const { isNot, utils } = this;
102
+ if (!AIMessage.isInstance(received)) return {
103
+ pass: false,
104
+ message: () => `${utils.matcherHint("toContainToolCall")}\n\nExpected: AIMessage\nReceived: ${getMessageTypeName(received)}`
105
+ };
106
+ const actual = received.tool_calls ?? [];
107
+ return {
108
+ pass: actual.some((tc) => Object.entries(expected).every(([key, value]) => this.equals(tc[key], value))),
109
+ message: () => `${utils.matcherHint("toContainToolCall")}\n\nExpected AIMessage ${isNot ? "not " : ""}to contain a tool call matching ${utils.printExpected(expected)}\nReceived tool calls: ${utils.printReceived(actual.map((tc) => ({
110
+ name: tc.name,
111
+ id: tc.id
112
+ })))}`,
113
+ actual: actual.map((tc) => ({
114
+ name: tc.name,
115
+ id: tc.id
116
+ })),
117
+ expected
118
+ };
119
+ }
120
+ function toHaveToolMessages(received, expected) {
121
+ const { isNot, utils } = this;
122
+ if (!Array.isArray(received)) return {
123
+ pass: false,
124
+ message: () => `${utils.matcherHint("toHaveToolMessages")}\n\nExpected an array of messages\nReceived: ${typeof received}`
125
+ };
126
+ const toolMessages = received.filter(ToolMessage.isInstance);
127
+ if (toolMessages.length !== expected.length) return {
128
+ pass: false,
129
+ message: () => `${utils.matcherHint("toHaveToolMessages")}\n\nExpected ${isNot ? "not " : ""}${expected.length} tool message(s), found ${toolMessages.length}`,
130
+ actual: toolMessages.length,
131
+ expected: expected.length
132
+ };
133
+ for (let i = 0; i < expected.length; i++) if (!Object.entries(expected[i]).every(([key, value]) => this.equals(toolMessages[i][key], value))) return {
134
+ pass: false,
135
+ message: () => {
136
+ const receivedFields = {};
137
+ for (const key of Object.keys(expected[i])) receivedFields[key] = toolMessages[i][key];
138
+ return `${utils.matcherHint("toHaveToolMessages")}\n\nTool message at index ${i} did not match:\nExpected: ${utils.printExpected(expected[i])}\nReceived: ${utils.printReceived(receivedFields)}`;
139
+ },
140
+ actual: toolMessages[i],
141
+ expected: expected[i]
142
+ };
143
+ return {
144
+ pass: true,
145
+ message: () => `${utils.matcherHint("toHaveToolMessages")}\n\nExpected messages not to contain matching tool messages`
146
+ };
147
+ }
148
+ function toHaveBeenInterrupted(received, expectedValue) {
149
+ const { isNot, utils } = this;
150
+ const interrupts = received?.__interrupt__;
151
+ if (!(Array.isArray(interrupts) && interrupts.length > 0)) return {
152
+ pass: false,
153
+ message: () => `${utils.matcherHint("toHaveBeenInterrupted")}\n\nExpected result ${isNot ? "not " : ""}to have been interrupted\nReceived __interrupt__: ${utils.printReceived(interrupts)}`
154
+ };
155
+ if (expectedValue === void 0) return {
156
+ pass: true,
157
+ message: () => `${utils.matcherHint("toHaveBeenInterrupted")}\n\nExpected result not to have been interrupted\nReceived ${interrupts.length} interrupt(s)`
158
+ };
159
+ const actualValue = interrupts[0]?.value;
160
+ return {
161
+ pass: this.equals(actualValue, expectedValue),
162
+ message: () => `${utils.matcherHint("toHaveBeenInterrupted")}\n\nExpected interrupt value: ${utils.printExpected(expectedValue)}\nReceived interrupt value: ${utils.printReceived(actualValue)}`,
163
+ actual: actualValue,
164
+ expected: expectedValue
165
+ };
166
+ }
167
+ function toHaveStructuredResponse(received, expected) {
168
+ const { isNot, utils } = this;
169
+ const structuredResponse = received?.structuredResponse;
170
+ if (!(structuredResponse !== void 0)) return {
171
+ pass: false,
172
+ message: () => `${utils.matcherHint("toHaveStructuredResponse")}\n\nExpected result ${isNot ? "not " : ""}to have a structured response\nReceived structuredResponse: undefined`
173
+ };
174
+ if (expected === void 0) return {
175
+ pass: true,
176
+ message: () => `${utils.matcherHint("toHaveStructuredResponse")}\n\nExpected result not to have a structured response`
177
+ };
178
+ return {
179
+ pass: Object.entries(expected).every(([key, value]) => this.equals(structuredResponse[key], value)),
180
+ message: () => `${utils.matcherHint("toHaveStructuredResponse")}\n\nExpected structured response: ${utils.printExpected(expected)}\nReceived structured response: ${utils.printReceived(structuredResponse)}`,
181
+ actual: structuredResponse,
182
+ expected
183
+ };
184
+ }
185
+ /**
186
+ * All matcher functions bundled for convenient use with `expect.extend()`.
187
+ */
188
+ const langchainMatchers = {
189
+ toBeHumanMessage,
190
+ toBeAIMessage,
191
+ toBeSystemMessage,
192
+ toBeToolMessage,
193
+ toHaveToolCalls,
194
+ toHaveToolCallCount,
195
+ toContainToolCall,
196
+ toHaveToolMessages,
197
+ toHaveBeenInterrupted,
198
+ toHaveStructuredResponse
199
+ };
200
+
201
+ //#endregion
202
+ export { langchainMatchers, toBeAIMessage, toBeHumanMessage, toBeSystemMessage, toBeToolMessage, toContainToolCall, toHaveBeenInterrupted, toHaveStructuredResponse, toHaveToolCallCount, toHaveToolCalls, toHaveToolMessages };
203
+ //# sourceMappingURL=matchers.js.map