@langchain/core 1.2.9 → 1.2.10

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 (167) hide show
  1. package/CHANGELOG.md +6 -0
  2. package/dist/_virtual/_rolldown/runtime.cjs +1 -1
  3. package/dist/callbacks/dispatch/index.cjs +2 -1
  4. package/dist/callbacks/dispatch/index.cjs.map +1 -1
  5. package/dist/callbacks/dispatch/index.js +2 -1
  6. package/dist/callbacks/dispatch/index.js.map +1 -1
  7. package/dist/indexing/base.cjs +10 -6
  8. package/dist/indexing/base.cjs.map +1 -1
  9. package/dist/indexing/base.js +10 -6
  10. package/dist/indexing/base.js.map +1 -1
  11. package/dist/language_models/base.cjs +1 -1
  12. package/dist/language_models/base.cjs.map +1 -1
  13. package/dist/language_models/base.js +1 -1
  14. package/dist/language_models/base.js.map +1 -1
  15. package/dist/language_models/chat_models.cjs +10 -5
  16. package/dist/language_models/chat_models.cjs.map +1 -1
  17. package/dist/language_models/chat_models.js +10 -5
  18. package/dist/language_models/chat_models.js.map +1 -1
  19. package/dist/language_models/stream.cjs +0 -2
  20. package/dist/language_models/stream.cjs.map +1 -1
  21. package/dist/language_models/stream.js +0 -2
  22. package/dist/language_models/stream.js.map +1 -1
  23. package/dist/load/index.cjs +7 -5
  24. package/dist/load/index.cjs.map +1 -1
  25. package/dist/load/index.js +7 -5
  26. package/dist/load/index.js.map +1 -1
  27. package/dist/load/serializable.cjs +2 -1
  28. package/dist/load/serializable.cjs.map +1 -1
  29. package/dist/load/serializable.js +2 -1
  30. package/dist/load/serializable.js.map +1 -1
  31. package/dist/load/validation.cjs +4 -3
  32. package/dist/load/validation.cjs.map +1 -1
  33. package/dist/load/validation.js +4 -3
  34. package/dist/load/validation.js.map +1 -1
  35. package/dist/messages/base.cjs +13 -18
  36. package/dist/messages/base.cjs.map +1 -1
  37. package/dist/messages/base.d.cts.map +1 -1
  38. package/dist/messages/base.d.ts.map +1 -1
  39. package/dist/messages/base.js +13 -18
  40. package/dist/messages/base.js.map +1 -1
  41. package/dist/messages/block_translators/anthropic.cjs.map +1 -1
  42. package/dist/messages/block_translators/anthropic.js.map +1 -1
  43. package/dist/messages/block_translators/google.cjs +10 -9
  44. package/dist/messages/block_translators/google.cjs.map +1 -1
  45. package/dist/messages/block_translators/google.js +10 -9
  46. package/dist/messages/block_translators/google.js.map +1 -1
  47. package/dist/messages/block_translators/google_vertexai.cjs +14 -12
  48. package/dist/messages/block_translators/google_vertexai.cjs.map +1 -1
  49. package/dist/messages/block_translators/google_vertexai.js +14 -12
  50. package/dist/messages/block_translators/google_vertexai.js.map +1 -1
  51. package/dist/messages/message.cjs.map +1 -1
  52. package/dist/messages/message.d.cts.map +1 -1
  53. package/dist/messages/message.d.ts.map +1 -1
  54. package/dist/messages/message.js.map +1 -1
  55. package/dist/messages/transformers.cjs +12 -8
  56. package/dist/messages/transformers.cjs.map +1 -1
  57. package/dist/messages/transformers.js +12 -8
  58. package/dist/messages/transformers.js.map +1 -1
  59. package/dist/output_parsers/transform.cjs.map +1 -1
  60. package/dist/prompts/chat.cjs +2 -1
  61. package/dist/prompts/chat.cjs.map +1 -1
  62. package/dist/prompts/chat.js +2 -1
  63. package/dist/prompts/chat.js.map +1 -1
  64. package/dist/prompts/few_shot.cjs +6 -4
  65. package/dist/prompts/few_shot.cjs.map +1 -1
  66. package/dist/prompts/few_shot.js +6 -4
  67. package/dist/prompts/few_shot.js.map +1 -1
  68. package/dist/prompts/image.cjs +2 -1
  69. package/dist/prompts/image.cjs.map +1 -1
  70. package/dist/prompts/image.js +2 -1
  71. package/dist/prompts/image.js.map +1 -1
  72. package/dist/prompts/prompt.cjs.map +1 -1
  73. package/dist/prompts/prompt.d.cts.map +1 -1
  74. package/dist/prompts/prompt.d.ts.map +1 -1
  75. package/dist/prompts/prompt.js.map +1 -1
  76. package/dist/prompts/string.cjs +2 -1
  77. package/dist/prompts/string.cjs.map +1 -1
  78. package/dist/prompts/string.js +2 -1
  79. package/dist/prompts/string.js.map +1 -1
  80. package/dist/prompts/template.cjs.map +1 -1
  81. package/dist/prompts/template.js.map +1 -1
  82. package/dist/runnables/base.cjs +60 -40
  83. package/dist/runnables/base.cjs.map +1 -1
  84. package/dist/runnables/base.js +60 -40
  85. package/dist/runnables/base.js.map +1 -1
  86. package/dist/runnables/branch.cjs +16 -12
  87. package/dist/runnables/branch.cjs.map +1 -1
  88. package/dist/runnables/branch.js +16 -12
  89. package/dist/runnables/branch.js.map +1 -1
  90. package/dist/runnables/config.cjs +28 -23
  91. package/dist/runnables/config.cjs.map +1 -1
  92. package/dist/runnables/config.js +28 -23
  93. package/dist/runnables/config.js.map +1 -1
  94. package/dist/runnables/graph.cjs +2 -1
  95. package/dist/runnables/graph.cjs.map +1 -1
  96. package/dist/runnables/graph.js +2 -1
  97. package/dist/runnables/graph.js.map +1 -1
  98. package/dist/runnables/passthrough.cjs +8 -6
  99. package/dist/runnables/passthrough.cjs.map +1 -1
  100. package/dist/runnables/passthrough.js +8 -6
  101. package/dist/runnables/passthrough.js.map +1 -1
  102. package/dist/singletons/async_local_storage/context.cjs.map +1 -1
  103. package/dist/singletons/tracer.cjs +4 -1
  104. package/dist/singletons/tracer.cjs.map +1 -1
  105. package/dist/singletons/tracer.js +4 -1
  106. package/dist/singletons/tracer.js.map +1 -1
  107. package/dist/tools/index.cjs +22 -20
  108. package/dist/tools/index.cjs.map +1 -1
  109. package/dist/tools/index.js +22 -20
  110. package/dist/tools/index.js.map +1 -1
  111. package/dist/tracers/base.cjs +11 -10
  112. package/dist/tracers/base.cjs.map +1 -1
  113. package/dist/tracers/base.js +11 -10
  114. package/dist/tracers/base.js.map +1 -1
  115. package/dist/tracers/log_stream.cjs +7 -6
  116. package/dist/tracers/log_stream.cjs.map +1 -1
  117. package/dist/tracers/log_stream.js +7 -6
  118. package/dist/tracers/log_stream.js.map +1 -1
  119. package/dist/tracers/tracer_langchain.cjs.map +1 -1
  120. package/dist/utils/env.cjs.map +1 -1
  121. package/dist/utils/env.d.cts.map +1 -1
  122. package/dist/utils/env.d.ts.map +1 -1
  123. package/dist/utils/env.js.map +1 -1
  124. package/dist/utils/event_source_parse.cjs +1 -3
  125. package/dist/utils/event_source_parse.cjs.map +1 -1
  126. package/dist/utils/event_source_parse.js +1 -3
  127. package/dist/utils/event_source_parse.js.map +1 -1
  128. package/dist/utils/fast-json-patch/src/core.cjs +4 -2
  129. package/dist/utils/fast-json-patch/src/core.cjs.map +1 -1
  130. package/dist/utils/fast-json-patch/src/core.js +4 -2
  131. package/dist/utils/fast-json-patch/src/core.js.map +1 -1
  132. package/dist/utils/fast-json-patch/src/helpers.cjs.map +1 -1
  133. package/dist/utils/fast-json-patch/src/helpers.js.map +1 -1
  134. package/dist/utils/js-sha256/hash.cjs.map +1 -1
  135. package/dist/utils/js-sha256/hash.js.map +1 -1
  136. package/dist/utils/json_schema.cjs +4 -2
  137. package/dist/utils/json_schema.cjs.map +1 -1
  138. package/dist/utils/json_schema.js +4 -2
  139. package/dist/utils/json_schema.js.map +1 -1
  140. package/dist/utils/sax-js/sax.cjs +41 -35
  141. package/dist/utils/sax-js/sax.cjs.map +1 -1
  142. package/dist/utils/sax-js/sax.js +41 -35
  143. package/dist/utils/sax-js/sax.js.map +1 -1
  144. package/dist/utils/testing/chat_models.cjs +2 -1
  145. package/dist/utils/testing/chat_models.cjs.map +1 -1
  146. package/dist/utils/testing/chat_models.js +2 -1
  147. package/dist/utils/testing/chat_models.js.map +1 -1
  148. package/dist/utils/tiktoken.cjs.map +1 -1
  149. package/dist/utils/types/zod.cjs.map +1 -1
  150. package/dist/utils/types/zod.js.map +1 -1
  151. package/dist/utils/zod-to-json-schema/parsers/bigint.cjs +9 -9
  152. package/dist/utils/zod-to-json-schema/parsers/bigint.cjs.map +1 -1
  153. package/dist/utils/zod-to-json-schema/parsers/bigint.js +9 -9
  154. package/dist/utils/zod-to-json-schema/parsers/bigint.js.map +1 -1
  155. package/dist/utils/zod-to-json-schema/parsers/date.cjs +1 -3
  156. package/dist/utils/zod-to-json-schema/parsers/date.cjs.map +1 -1
  157. package/dist/utils/zod-to-json-schema/parsers/date.js +1 -3
  158. package/dist/utils/zod-to-json-schema/parsers/date.js.map +1 -1
  159. package/dist/utils/zod-to-json-schema/parsers/number.cjs +9 -9
  160. package/dist/utils/zod-to-json-schema/parsers/number.cjs.map +1 -1
  161. package/dist/utils/zod-to-json-schema/parsers/number.js +9 -9
  162. package/dist/utils/zod-to-json-schema/parsers/number.js.map +1 -1
  163. package/dist/utils/zod-to-json-schema/parsers/string.cjs +3 -13
  164. package/dist/utils/zod-to-json-schema/parsers/string.cjs.map +1 -1
  165. package/dist/utils/zod-to-json-schema/parsers/string.js +3 -13
  166. package/dist/utils/zod-to-json-schema/parsers/string.js.map +1 -1
  167. package/package.json +5 -5
@@ -1 +1 @@
1
- {"version":3,"file":"transformers.cjs","names":["RunnableLambda","convertToChunk","isBaseMessageChunk","HumanMessage","HumanMessageChunk","AIMessage","AIMessageChunk","SystemMessage","SystemMessageChunk","ToolMessage","ToolMessageChunk","FunctionMessage","FunctionMessageChunk","ChatMessage","ChatMessageChunk","RemoveMessage"],"sources":["../../src/messages/transformers.ts"],"sourcesContent":["import { BaseLanguageModel } from \"../language_models/base.js\";\nimport { Runnable, RunnableLambda } from \"../runnables/base.js\";\nimport { AIMessage, AIMessageChunk, AIMessageChunkFields } from \"./ai.js\";\nimport {\n BaseMessage,\n BaseMessageChunk,\n BaseMessageFields,\n isBaseMessageChunk,\n} from \"./base.js\";\nimport { ChatMessage, ChatMessageChunk, ChatMessageFields } from \"./chat.js\";\nimport {\n FunctionMessage,\n FunctionMessageChunk,\n FunctionMessageFields,\n} from \"./function.js\";\nimport { HumanMessage, HumanMessageChunk } from \"./human.js\";\nimport { MessageType } from \"./message.js\";\nimport { RemoveMessage } from \"./modifier.js\";\nimport { SystemMessage, SystemMessageChunk } from \"./system.js\";\nimport { ToolMessage, ToolMessageChunk, ToolMessageFields } from \"./tool.js\";\nimport { convertToChunk } from \"./utils.js\";\n\nexport type MessageUnion =\n | typeof HumanMessage\n | typeof AIMessage\n | typeof SystemMessage\n | typeof ChatMessage\n | typeof FunctionMessage\n | typeof ToolMessage\n | typeof RemoveMessage;\nexport type MessageChunkUnion =\n | typeof HumanMessageChunk\n | typeof AIMessageChunk\n | typeof SystemMessageChunk\n | typeof FunctionMessageChunk\n | typeof ToolMessageChunk\n | typeof ChatMessageChunk\n | typeof RemoveMessage; // RemoveMessage does not have a chunk class.\nexport type MessageTypeOrClass = MessageType | MessageUnion | MessageChunkUnion;\n\nconst _isMessageType = (msg: BaseMessage, types: MessageTypeOrClass[]) => {\n const typesAsStrings = [\n ...new Set<string>(\n types?.map((t) => {\n if (typeof t === \"string\") {\n return t;\n }\n // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n const instantiatedMsgClass = new (t as any)({});\n if (\n !(\"getType\" in instantiatedMsgClass) ||\n typeof instantiatedMsgClass.getType !== \"function\"\n ) {\n throw new Error(\"Invalid type provided.\");\n }\n return instantiatedMsgClass.getType();\n })\n ),\n ];\n const msgType = msg.getType();\n return typesAsStrings.some((t) => t === msgType);\n};\n\nexport interface FilterMessagesFields {\n /**\n * @param {string[] | undefined} includeNames Message names to include.\n */\n includeNames?: string[];\n /**\n * @param {string[] | undefined} excludeNames Messages names to exclude.\n */\n excludeNames?: string[];\n /**\n * @param {(MessageType | BaseMessage)[] | undefined} includeTypes Message types to include. Can be specified as string names (e.g.\n * \"system\", \"human\", \"ai\", ...) or as BaseMessage classes (e.g.\n * SystemMessage, HumanMessage, AIMessage, ...).\n */\n includeTypes?: MessageTypeOrClass[];\n /**\n * @param {(MessageType | BaseMessage)[] | undefined} excludeTypes Message types to exclude. Can be specified as string names (e.g.\n * \"system\", \"human\", \"ai\", ...) or as BaseMessage classes (e.g.\n * SystemMessage, HumanMessage, AIMessage, ...).\n */\n excludeTypes?: MessageTypeOrClass[];\n /**\n * @param {string[] | undefined} includeIds Message IDs to include.\n */\n includeIds?: string[];\n /**\n * @param {string[] | undefined} excludeIds Message IDs to exclude.\n */\n excludeIds?: string[];\n}\n\n/**\n * Filter messages based on name, type or id.\n *\n * @param {BaseMessage[] | FilterMessagesFields} messagesOrOptions - Either an array of BaseMessage objects to filter or the filtering options. If an array is provided, the `options` parameter should also be supplied. If filtering options are provided, a RunnableLambda is returned.\n * @param {FilterMessagesFields} [options] - Optional filtering options. Should only be provided if `messagesOrOptions` is an array of BaseMessage objects.\n * @returns A list of Messages that meets at least one of the include conditions and none\n * of the exclude conditions, or a RunnableLambda which does the same. If no include conditions are specified then\n * anything that is not explicitly excluded will be included.\n * @throws {Error} If two incompatible arguments are provided.\n *\n * @example\n * ```typescript\n * import { filterMessages, AIMessage, HumanMessage, SystemMessage } from \"@langchain/core/messages\";\n *\n * const messages = [\n * new SystemMessage(\"you're a good assistant.\"),\n * new HumanMessage({ content: \"what's your name\", id: \"foo\", name: \"example_user\" }),\n * new AIMessage({ content: \"steve-o\", id: \"bar\", name: \"example_assistant\" }),\n * new HumanMessage({ content: \"what's your favorite color\", id: \"baz\" }),\n * new AIMessage({ content: \"silicon blue\" , id: \"blah\" }),\n * ];\n *\n * filterMessages(messages, {\n * includeNames: [\"example_user\", \"example_assistant\"],\n * includeTypes: [\"system\"],\n * excludeIds: [\"bar\"],\n * });\n * ```\n *\n * The above example would return:\n * ```typescript\n * [\n * new SystemMessage(\"you're a good assistant.\"),\n * new HumanMessage({ content: \"what's your name\", id: \"foo\", name: \"example_user\" }),\n * ]\n * ```\n */\nexport function filterMessages(\n options?: FilterMessagesFields\n): Runnable<BaseMessage[], BaseMessage[]>;\nexport function filterMessages(\n messages: BaseMessage[],\n options?: FilterMessagesFields\n): BaseMessage[];\nexport function filterMessages(\n messagesOrOptions?: BaseMessage[] | FilterMessagesFields,\n options?: FilterMessagesFields\n): BaseMessage[] | Runnable<BaseMessage[], BaseMessage[]> {\n if (Array.isArray(messagesOrOptions)) {\n return _filterMessages(messagesOrOptions, options);\n }\n return RunnableLambda.from((input: BaseMessage[]): BaseMessage[] => {\n return _filterMessages(input, messagesOrOptions);\n });\n}\n\nfunction _filterMessages(\n messages: BaseMessage[],\n options: FilterMessagesFields = {}\n): BaseMessage[] {\n const {\n includeNames,\n excludeNames,\n includeTypes,\n excludeTypes,\n includeIds,\n excludeIds,\n } = options;\n\n const filtered: BaseMessage[] = [];\n\n for (const msg of messages) {\n if (excludeNames && msg.name && excludeNames.includes(msg.name)) {\n continue;\n } else if (excludeTypes && _isMessageType(msg, excludeTypes)) {\n continue;\n } else if (excludeIds && msg.id && excludeIds.includes(msg.id)) {\n continue;\n }\n\n // default to inclusion when no inclusion criteria given.\n if (!(includeTypes || includeIds || includeNames)) {\n filtered.push(msg);\n } else if (\n includeNames &&\n msg.name &&\n includeNames.some((iName) => iName === msg.name)\n ) {\n filtered.push(msg);\n } else if (includeTypes && _isMessageType(msg, includeTypes)) {\n filtered.push(msg);\n } else if (includeIds && msg.id && includeIds.some((id) => id === msg.id)) {\n filtered.push(msg);\n }\n }\n\n return filtered;\n}\n\n/**\n * Merge consecutive Messages of the same type.\n *\n * **NOTE**: ToolMessages are not merged, as each has a distinct tool call id that\n * can't be merged.\n *\n * @param {BaseMessage[] | undefined} messages Sequence of Message-like objects to merge. Optional. If not provided, a RunnableLambda is returned.\n * @returns List of BaseMessages with consecutive runs of message types merged into single\n * messages, or a RunnableLambda which returns a list of BaseMessages If two messages being merged both have string contents, the merged\n * content is a concatenation of the two strings with a new-line separator. If at\n * least one of the messages has a list of content blocks, the merged content is a\n * list of content blocks.\n *\n * @example\n * ```typescript\n * import { mergeMessageRuns, AIMessage, HumanMessage, SystemMessage, ToolCall } from \"@langchain/core/messages\";\n *\n * const messages = [\n * new SystemMessage(\"you're a good assistant.\"),\n * new HumanMessage({ content: \"what's your favorite color\", id: \"foo\" }),\n * new HumanMessage({ content: \"wait your favorite food\", id: \"bar\" }),\n * new AIMessage({\n * content: \"my favorite colo\",\n * tool_calls: [{ name: \"blah_tool\", args: { x: 2 }, id: \"123\" }],\n * id: \"baz\",\n * }),\n * new AIMessage({\n * content: [{ type: \"text\", text: \"my favorite dish is lasagna\" }],\n * tool_calls: [{ name: \"blah_tool\", args: { x: -10 }, id: \"456\" }],\n * id: \"blur\",\n * }),\n * ];\n *\n * mergeMessageRuns(messages);\n * ```\n *\n * The above example would return:\n * ```typescript\n * [\n * new SystemMessage(\"you're a good assistant.\"),\n * new HumanMessage({\n * content: \"what's your favorite colorwait your favorite food\",\n * id: \"foo\",\n * }),\n * new AIMessage({\n * content: [\n * { type: \"text\", text: \"my favorite colo\" },\n * { type: \"text\", text: \"my favorite dish is lasagna\" },\n * ],\n * tool_calls: [\n * { name: \"blah_tool\", args: { x: 2 }, id: \"123\" },\n * { name: \"blah_tool\", args: { x: -10 }, id: \"456\" },\n * ],\n * id: \"baz\",\n * }),\n * ]\n * ```\n */\nexport function mergeMessageRuns(): Runnable<BaseMessage[], BaseMessage[]>;\nexport function mergeMessageRuns(messages: BaseMessage[]): BaseMessage[];\nexport function mergeMessageRuns(\n messages?: BaseMessage[]\n): BaseMessage[] | Runnable<BaseMessage[], BaseMessage[]> {\n if (Array.isArray(messages)) {\n return _mergeMessageRuns(messages);\n }\n return RunnableLambda.from(_mergeMessageRuns);\n}\n\nfunction _mergeMessageRuns(messages: BaseMessage[]): BaseMessage[] {\n if (!messages.length) {\n return [];\n }\n const merged: BaseMessage[] = [];\n for (const msg of messages) {\n const curr = msg;\n const last = merged.pop();\n if (!last) {\n merged.push(curr);\n } else if (\n curr.getType() === \"tool\" ||\n !(curr.getType() === last.getType())\n ) {\n merged.push(last, curr);\n } else {\n const lastChunk = convertToChunk(last) as BaseMessageChunk;\n const currChunk = convertToChunk(curr) as BaseMessageChunk;\n const mergedChunks = lastChunk.concat(currChunk);\n if (\n typeof lastChunk.content === \"string\" &&\n typeof currChunk.content === \"string\"\n ) {\n mergedChunks.content = `${lastChunk.content}\\n${currChunk.content}`;\n }\n merged.push(_chunkToMsg(mergedChunks));\n }\n }\n return merged;\n}\n\n// Since we can not import from `@langchain/textsplitters` we need\n// to reconstruct the interface here. Only `splitText` is used by\n// `trimMessages`, so we avoid coupling to the full `BaseDocumentTransformer`\n// surface and just require the method we call.\ninterface _TextSplitterInterface {\n splitText(text: string): Promise<string[]>;\n}\n\nexport interface TrimMessagesFields {\n /**\n * @param {number} maxTokens Max token count of trimmed messages.\n */\n maxTokens: number;\n /**\n * @param {((messages: BaseMessage[]) => number) | ((messages: BaseMessage[]) => Promise<number>) | BaseLanguageModel} tokenCounter\n * Function or LLM for counting tokens in an array of `BaseMessage`s.\n * If a `BaseLanguageModel` is passed in then `BaseLanguageModel.getNumTokens()` will be used.\n */\n tokenCounter:\n | ((messages: BaseMessage[]) => number)\n | ((messages: BaseMessage[]) => Promise<number>)\n | BaseLanguageModel;\n /**\n * @param {\"first\" | \"last\"} [strategy=\"last\"] Strategy for trimming.\n * - \"first\": Keep the first <= n_count tokens of the messages.\n * - \"last\": Keep the last <= n_count tokens of the messages.\n * @default \"last\"\n */\n strategy?: \"first\" | \"last\";\n /**\n * @param {boolean} [allowPartial=false] Whether to split a message if only part of the message can be included.\n * If `strategy: \"last\"` then the last partial contents of a message are included.\n * If `strategy: \"first\"` then the first partial contents of a message are included.\n * @default false\n */\n allowPartial?: boolean;\n /**\n * @param {MessageTypeOrClass | MessageTypeOrClass[]} [endOn] The message type to end on.\n * If specified then every message after the last occurrence of this type is ignored.\n * If `strategy === \"last\"` then this is done before we attempt to get the last `maxTokens`.\n * If `strategy === \"first\"` then this is done after we get the first `maxTokens`.\n * Can be specified as string names (e.g. \"system\", \"human\", \"ai\", ...) or as `BaseMessage` classes\n * (e.g. `SystemMessage`, `HumanMessage`, `AIMessage`, ...). Can be a single type or an array of types.\n */\n endOn?: MessageTypeOrClass | MessageTypeOrClass[];\n /**\n * @param {MessageTypeOrClass | MessageTypeOrClass[]} [startOn] The message type to start on.\n * Should only be specified if `strategy: \"last\"`. If specified then every message before the first occurrence\n * of this type is ignored. This is done after we trim the initial messages to the last `maxTokens`.\n * Does not apply to a `SystemMessage` at index 0 if `includeSystem: true`.\n * Can be specified as string names (e.g. \"system\", \"human\", \"ai\", ...) or as `BaseMessage` classes\n * (e.g. `SystemMessage`, `HumanMessage`, `AIMessage`, ...). Can be a single type or an array of types.\n */\n startOn?: MessageTypeOrClass | MessageTypeOrClass[];\n /**\n * @param {boolean} [includeSystem=false] Whether to keep the `SystemMessage` if there is one at index 0.\n * Should only be specified if `strategy: \"last\"`.\n * @default false\n */\n includeSystem?: boolean;\n /**\n * @param {((text: string) => string[]) | BaseDocumentTransformer} [textSplitter] Function or `BaseDocumentTransformer` for\n * splitting the string contents of a message. Only used if `allowPartial: true`.\n * If `strategy: \"last\"` then the last split tokens from a partial message will be included.\n * If `strategy: \"first\"` then the first split tokens from a partial message will be included.\n * Token splitter assumes that separators are kept, so that split contents can be directly concatenated\n * to recreate the original text. Defaults to splitting on newlines.\n */\n textSplitter?:\n | ((text: string) => string[])\n | ((text: string) => Promise<string[]>)\n | _TextSplitterInterface;\n}\n\n/**\n * Trim messages to be below a token count.\n *\n * @param {BaseMessage[]} messages Array of `BaseMessage` instances to trim.\n * @param {TrimMessagesFields} options Trimming options.\n * @returns An array of trimmed `BaseMessage`s or a `Runnable` that takes a sequence of `BaseMessage`-like objects and returns\n * an array of trimmed `BaseMessage`s.\n * @throws {Error} If two incompatible arguments are specified or an unrecognized `strategy` is specified.\n *\n * @example\n * ```typescript\n * import { trimMessages, AIMessage, BaseMessage, HumanMessage, SystemMessage } from \"@langchain/core/messages\";\n *\n * const messages = [\n * new SystemMessage(\"This is a 4 token text. The full message is 10 tokens.\"),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"first\",\n * }),\n * new AIMessage({\n * content: [\n * { type: \"text\", text: \"This is the FIRST 4 token block.\" },\n * { type: \"text\", text: \"This is the SECOND 4 token block.\" },\n * ],\n * id: \"second\",\n * }),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"third\",\n * }),\n * new AIMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"fourth\",\n * }),\n * ];\n *\n * function dummyTokenCounter(messages: BaseMessage[]): number {\n * // treat each message like it adds 3 default tokens at the beginning\n * // of the message and at the end of the message. 3 + 4 + 3 = 10 tokens\n * // per message.\n *\n * const defaultContentLen = 4;\n * const defaultMsgPrefixLen = 3;\n * const defaultMsgSuffixLen = 3;\n *\n * let count = 0;\n * for (const msg of messages) {\n * if (typeof msg.content === \"string\") {\n * count += defaultMsgPrefixLen + defaultContentLen + defaultMsgSuffixLen;\n * }\n * if (Array.isArray(msg.content)) {\n * count +=\n * defaultMsgPrefixLen +\n * msg.content.length * defaultContentLen +\n * defaultMsgSuffixLen;\n * }\n * }\n * return count;\n * }\n * ```\n *\n * First 30 tokens, not allowing partial messages:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 30,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"first\",\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"first\",\n * }),\n * ]\n * ```\n *\n * First 30 tokens, allowing partial messages:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 30,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"first\",\n * allowPartial: true,\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"first\",\n * }),\n * new AIMessage({\n * content: [{ type: \"text\", text: \"This is the FIRST 4 token block.\" }],\n * id: \"second\",\n * }),\n * ]\n * ```\n *\n * First 30 tokens, allowing partial messages, have to end on HumanMessage:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 30,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"first\",\n * allowPartial: true,\n * endOn: \"human\",\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"first\",\n * }),\n * ]\n * ```\n *\n * Last 30 tokens, including system message, not allowing partial messages:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 30,\n * includeSystem: true,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"last\",\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"third\",\n * }),\n * new AIMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"fourth\",\n * }),\n * ]\n * ```\n *\n * Last 40 tokens, including system message, allowing partial messages:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 40,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"last\",\n * allowPartial: true,\n * includeSystem: true,\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new AIMessage({\n * content: [{ type: \"text\", text: \"This is the FIRST 4 token block.\" }],\n * id: \"second\",\n * }),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"third\",\n * }),\n * new AIMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"fourth\",\n * }),\n * ]\n * ```\n *\n * Last 30 tokens, including system message, allowing partial messages, end on HumanMessage:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 30,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"last\",\n * endOn: \"human\",\n * includeSystem: true,\n * allowPartial: true,\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new AIMessage({\n * content: [{ type: \"text\", text: \"This is the FIRST 4 token block.\" }],\n * id: \"second\",\n * }),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"third\",\n * }),\n * ]\n * ```\n *\n * Last 40 tokens, including system message, allowing partial messages, start on HumanMessage:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 40,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"last\",\n * includeSystem: true,\n * allowPartial: true,\n * startOn: \"human\",\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"third\",\n * }),\n * new AIMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"fourth\",\n * }),\n * ]\n * ```\n */\nexport function trimMessages(\n options: TrimMessagesFields\n): Runnable<BaseMessage[], BaseMessage[]>;\nexport function trimMessages(\n messages: BaseMessage[],\n options: TrimMessagesFields\n): Promise<BaseMessage[]>;\nexport function trimMessages(\n messagesOrOptions: BaseMessage[] | TrimMessagesFields,\n options?: TrimMessagesFields\n): Promise<BaseMessage[]> | Runnable<BaseMessage[], BaseMessage[]> {\n if (Array.isArray(messagesOrOptions)) {\n const messages = messagesOrOptions;\n if (!options) {\n throw new Error(\"Options parameter is required when providing messages.\");\n }\n return _trimMessagesHelper(messages, options);\n } else {\n const trimmerOptions = messagesOrOptions;\n return RunnableLambda.from((input: BaseMessage[]) =>\n _trimMessagesHelper(input, trimmerOptions)\n ).withConfig({\n runName: \"trim_messages\",\n });\n }\n}\n\nasync function _trimMessagesHelper(\n messages: BaseMessage[],\n options: TrimMessagesFields\n): Promise<Array<BaseMessage>> {\n const {\n maxTokens,\n tokenCounter,\n strategy = \"last\",\n allowPartial = false,\n endOn,\n startOn,\n includeSystem = false,\n textSplitter,\n } = options;\n if (startOn && strategy === \"first\") {\n throw new Error(\n \"`startOn` should only be specified if `strategy` is 'last'.\"\n );\n }\n if (includeSystem && strategy === \"first\") {\n throw new Error(\n \"`includeSystem` should only be specified if `strategy` is 'last'.\"\n );\n }\n\n let listTokenCounter: (msgs: BaseMessage[]) => Promise<number>;\n if (\"getNumTokens\" in tokenCounter) {\n listTokenCounter = async (msgs: BaseMessage[]): Promise<number> => {\n const tokenCounts = await Promise.all(\n msgs.map((msg) => tokenCounter.getNumTokens(msg.content))\n );\n return tokenCounts.reduce((sum, count) => sum + count, 0);\n };\n } else {\n listTokenCounter = async (msgs: BaseMessage[]): Promise<number> =>\n tokenCounter(msgs);\n }\n\n let textSplitterFunc: (text: string) => Promise<string[]> =\n defaultTextSplitter;\n if (textSplitter) {\n if (\"splitText\" in textSplitter) {\n textSplitterFunc = textSplitter.splitText.bind(textSplitter);\n } else {\n textSplitterFunc = async (text: string): Promise<string[]> =>\n textSplitter(text);\n }\n }\n\n if (strategy === \"first\") {\n return _firstMaxTokens(messages, {\n maxTokens,\n tokenCounter: listTokenCounter,\n textSplitter: textSplitterFunc,\n partialStrategy: allowPartial ? \"first\" : undefined,\n endOn,\n });\n } else if (strategy === \"last\") {\n return _lastMaxTokens(messages, {\n maxTokens,\n tokenCounter: listTokenCounter,\n textSplitter: textSplitterFunc,\n allowPartial,\n includeSystem,\n startOn,\n endOn,\n });\n } else {\n throw new Error(\n `Unrecognized strategy: '${strategy}'. Must be one of 'first' or 'last'.`\n );\n }\n}\n\nasync function _firstMaxTokens(\n messages: BaseMessage[],\n options: {\n maxTokens: number;\n tokenCounter: (messages: BaseMessage[]) => Promise<number>;\n textSplitter: (text: string) => Promise<string[]>;\n partialStrategy?: \"first\" | \"last\";\n endOn?: MessageTypeOrClass | MessageTypeOrClass[];\n }\n): Promise<BaseMessage[]> {\n const { maxTokens, tokenCounter, textSplitter, partialStrategy, endOn } =\n options;\n let messagesCopy = [...messages];\n let idx = 0;\n for (let i = 0; i < messagesCopy.length; i += 1) {\n const remainingMessages = i > 0 ? messagesCopy.slice(0, -i) : messagesCopy;\n if ((await tokenCounter(remainingMessages)) <= maxTokens) {\n idx = messagesCopy.length - i;\n break;\n }\n }\n if (idx < messagesCopy.length && partialStrategy) {\n let includedPartial = false;\n if (Array.isArray(messagesCopy[idx].content)) {\n const excluded = messagesCopy[idx];\n if (typeof excluded.content === \"string\") {\n throw new Error(\"Expected content to be an array.\");\n }\n\n const numBlock = excluded.content.length;\n const reversedContent =\n partialStrategy === \"last\"\n ? [...excluded.content].reverse()\n : excluded.content;\n for (let i = 1; i <= numBlock; i += 1) {\n const partialContent =\n partialStrategy === \"first\"\n ? reversedContent.slice(0, i)\n : reversedContent.slice(-i);\n const fields = Object.fromEntries(\n Object.entries(excluded).filter(\n ([k]) => k !== \"type\" && !k.startsWith(\"lc_\")\n )\n ) as BaseMessageFields;\n const updatedMessage = _switchTypeToMessage(excluded.getType(), {\n ...fields,\n content: partialContent,\n });\n const slicedMessages = [...messagesCopy.slice(0, idx), updatedMessage];\n if ((await tokenCounter(slicedMessages)) <= maxTokens) {\n messagesCopy = slicedMessages;\n idx += 1;\n includedPartial = true;\n } else {\n break;\n }\n }\n if (includedPartial && partialStrategy === \"last\") {\n excluded.content = [...reversedContent].reverse();\n }\n }\n if (!includedPartial) {\n const excluded = messagesCopy[idx];\n let text: string | undefined;\n if (\n Array.isArray(excluded.content) &&\n excluded.content.some(\n (block) => typeof block === \"string\" || block.type === \"text\"\n )\n ) {\n const textBlock = excluded.content.find(\n (block) => block.type === \"text\" && block.text\n ) as { type: \"text\"; text: string } | undefined;\n text = textBlock?.text;\n } else if (typeof excluded.content === \"string\") {\n text = excluded.content;\n }\n if (text) {\n const splitTexts = await textSplitter(text);\n const numSplits = splitTexts.length;\n if (partialStrategy === \"last\") {\n splitTexts.reverse();\n }\n for (let _ = 0; _ < numSplits - 1; _ += 1) {\n splitTexts.pop();\n excluded.content = splitTexts.join(\"\");\n if (\n (await tokenCounter([...messagesCopy.slice(0, idx), excluded])) <=\n maxTokens\n ) {\n if (partialStrategy === \"last\") {\n excluded.content = [...splitTexts].reverse().join(\"\");\n }\n messagesCopy = [...messagesCopy.slice(0, idx), excluded];\n idx += 1;\n break;\n }\n }\n }\n }\n }\n\n if (endOn) {\n const endOnArr = Array.isArray(endOn) ? endOn : [endOn];\n while (idx > 0 && !_isMessageType(messagesCopy[idx - 1], endOnArr)) {\n idx -= 1;\n }\n }\n\n return messagesCopy.slice(0, idx);\n}\n\nasync function _lastMaxTokens(\n messages: BaseMessage[],\n options: {\n maxTokens: number;\n tokenCounter: (messages: BaseMessage[]) => Promise<number>;\n textSplitter: (text: string) => Promise<string[]>;\n /**\n * @default {false}\n */\n allowPartial?: boolean;\n /**\n * @default {false}\n */\n includeSystem?: boolean;\n startOn?: MessageTypeOrClass | MessageTypeOrClass[];\n endOn?: MessageTypeOrClass | MessageTypeOrClass[];\n }\n): Promise<BaseMessage[]> {\n const {\n allowPartial = false,\n includeSystem = false,\n endOn,\n startOn,\n ...rest\n } = options;\n\n // Create a copy of messages to avoid mutation\n let messagesCopy = messages.map((message) => {\n const fields = Object.fromEntries(\n Object.entries(message).filter(\n ([k]) => k !== \"type\" && !k.startsWith(\"lc_\")\n )\n ) as BaseMessageFields;\n return _switchTypeToMessage(\n message.getType(),\n fields,\n isBaseMessageChunk(message)\n );\n });\n\n if (endOn) {\n const endOnArr = Array.isArray(endOn) ? endOn : [endOn];\n while (\n messagesCopy.length > 0 &&\n !_isMessageType(messagesCopy[messagesCopy.length - 1], endOnArr)\n ) {\n messagesCopy = messagesCopy.slice(0, -1);\n }\n }\n\n const swappedSystem =\n includeSystem && messagesCopy[0]?.getType() === \"system\";\n let reversed_ = swappedSystem\n ? messagesCopy.slice(0, 1).concat(messagesCopy.slice(1).reverse())\n : messagesCopy.reverse();\n\n reversed_ = await _firstMaxTokens(reversed_, {\n ...rest,\n partialStrategy: allowPartial ? \"last\" : undefined,\n endOn: startOn,\n });\n\n if (swappedSystem) {\n return [reversed_[0], ...reversed_.slice(1).reverse()];\n } else {\n return reversed_.reverse();\n }\n}\n\nconst _MSG_CHUNK_MAP: Record<\n MessageType,\n {\n message: MessageUnion;\n messageChunk: MessageChunkUnion;\n }\n> = {\n human: {\n message: HumanMessage,\n messageChunk: HumanMessageChunk,\n },\n ai: {\n message: AIMessage,\n messageChunk: AIMessageChunk,\n },\n system: {\n message: SystemMessage,\n messageChunk: SystemMessageChunk,\n },\n developer: {\n message: SystemMessage,\n messageChunk: SystemMessageChunk,\n },\n tool: {\n message: ToolMessage,\n messageChunk: ToolMessageChunk,\n },\n function: {\n message: FunctionMessage,\n messageChunk: FunctionMessageChunk,\n },\n generic: {\n message: ChatMessage,\n messageChunk: ChatMessageChunk,\n },\n remove: {\n message: RemoveMessage,\n messageChunk: RemoveMessage, // RemoveMessage does not have a chunk class.\n },\n};\n\nfunction _switchTypeToMessage(\n messageType: MessageType,\n fields: BaseMessageFields\n): BaseMessage;\nfunction _switchTypeToMessage(\n messageType: MessageType,\n fields: BaseMessageFields,\n returnChunk: true\n): BaseMessageChunk;\nfunction _switchTypeToMessage(\n messageType: MessageType,\n fields: BaseMessageFields,\n returnChunk?: boolean\n): BaseMessageChunk | BaseMessage;\nfunction _switchTypeToMessage(\n messageType: MessageType,\n fields: BaseMessageFields,\n returnChunk?: boolean\n): BaseMessageChunk | BaseMessage {\n let chunk: BaseMessageChunk | undefined;\n let msg: BaseMessage | undefined;\n\n switch (messageType) {\n case \"human\":\n if (returnChunk) {\n chunk = new HumanMessageChunk(fields);\n } else {\n msg = new HumanMessage(fields);\n }\n break;\n case \"ai\":\n if (returnChunk) {\n let aiChunkFields: AIMessageChunkFields = {\n ...fields,\n };\n if (\"tool_calls\" in aiChunkFields) {\n aiChunkFields = {\n ...aiChunkFields,\n tool_call_chunks: aiChunkFields.tool_calls?.map((tc) => ({\n ...tc,\n type: \"tool_call_chunk\",\n index: undefined,\n args: JSON.stringify(tc.args),\n })),\n };\n }\n chunk = new AIMessageChunk(aiChunkFields);\n } else {\n msg = new AIMessage(fields);\n }\n break;\n case \"system\":\n if (returnChunk) {\n chunk = new SystemMessageChunk(fields);\n } else {\n msg = new SystemMessage(fields);\n }\n break;\n case \"developer\":\n if (returnChunk) {\n chunk = new SystemMessageChunk({\n ...fields,\n additional_kwargs: {\n ...fields.additional_kwargs,\n __openai_role__: \"developer\",\n },\n });\n } else {\n msg = new SystemMessage({\n ...fields,\n additional_kwargs: {\n ...fields.additional_kwargs,\n __openai_role__: \"developer\",\n },\n });\n }\n break;\n case \"tool\":\n if (\"tool_call_id\" in fields) {\n if (returnChunk) {\n chunk = new ToolMessageChunk(fields as ToolMessageFields);\n } else {\n msg = new ToolMessage(fields as ToolMessageFields);\n }\n } else {\n throw new Error(\n \"Can not convert ToolMessage to ToolMessageChunk if 'tool_call_id' field is not defined.\"\n );\n }\n break;\n case \"function\":\n if (returnChunk) {\n chunk = new FunctionMessageChunk(fields as FunctionMessageFields);\n } else {\n if (!fields.name) {\n throw new Error(\"FunctionMessage must have a 'name' field\");\n }\n msg = new FunctionMessage(fields as FunctionMessageFields);\n }\n break;\n case \"generic\":\n if (\"role\" in fields) {\n if (returnChunk) {\n chunk = new ChatMessageChunk(fields as ChatMessageFields);\n } else {\n msg = new ChatMessage(fields as ChatMessageFields);\n }\n } else {\n throw new Error(\n \"Can not convert ChatMessage to ChatMessageChunk if 'role' field is not defined.\"\n );\n }\n break;\n default:\n throw new Error(`Unrecognized message type ${messageType}`);\n }\n\n if (returnChunk && chunk) {\n return chunk;\n }\n if (msg) {\n return msg;\n }\n throw new Error(`Unrecognized message type ${messageType}`);\n}\n\nfunction _chunkToMsg(chunk: BaseMessageChunk): BaseMessage {\n const chunkType = chunk.getType();\n let msg: BaseMessage | undefined;\n const fields = Object.fromEntries(\n Object.entries(chunk).filter(\n ([k]) => ![\"type\", \"tool_call_chunks\"].includes(k) && !k.startsWith(\"lc_\")\n )\n ) as BaseMessageFields;\n\n if (chunkType in _MSG_CHUNK_MAP) {\n msg = _switchTypeToMessage(chunkType, fields);\n }\n\n if (!msg) {\n throw new Error(\n `Unrecognized message chunk class ${chunkType}. Supported classes are ${Object.keys(\n _MSG_CHUNK_MAP\n )}`\n );\n }\n\n return msg;\n}\n\n/**\n * The default text splitter function that splits text by newlines.\n *\n * @param {string} text\n * @returns A promise that resolves to an array of strings split by newlines.\n */\nexport function defaultTextSplitter(text: string): Promise<string[]> {\n const splits = text.split(\"\\n\");\n return Promise.resolve([\n ...splits.slice(0, -1).map((s) => `${s}\\n`),\n splits[splits.length - 1],\n ]);\n}\n"],"mappings":";;;;;;;;;;;AAwCA,MAAM,kBAAkB,KAAkB,UAAgC;CACxE,MAAM,iBAAiB,CACrB,GAAG,IAAI,IACL,OAAO,KAAK,MAAM;EAChB,IAAI,OAAO,MAAM,UACf,OAAO;EAGT,MAAM,uBAAuB,IAAK,EAAU,CAAC,CAAC;EAC9C,IACE,EAAE,aAAa,yBACf,OAAO,qBAAqB,YAAY,YAExC,MAAM,IAAI,MAAM,wBAAwB;EAE1C,OAAO,qBAAqB,QAAQ;CACtC,CAAC,CACH,CACF;CACA,MAAM,UAAU,IAAI,QAAQ;CAC5B,OAAO,eAAe,MAAM,MAAM,MAAM,OAAO;AACjD;AA6EA,SAAgB,eACd,mBACA,SACwD;CACxD,IAAI,MAAM,QAAQ,iBAAiB,GACjC,OAAO,gBAAgB,mBAAmB,OAAO;CAEnD,OAAOA,eAAAA,eAAe,MAAM,UAAwC;EAClE,OAAO,gBAAgB,OAAO,iBAAiB;CACjD,CAAC;AACH;AAEA,SAAS,gBACP,UACA,UAAgC,CAAC,GAClB;CACf,MAAM,EACJ,cACA,cACA,cACA,cACA,YACA,eACE;CAEJ,MAAM,WAA0B,CAAC;CAEjC,KAAK,MAAM,OAAO,UAAU;EAC1B,IAAI,gBAAgB,IAAI,QAAQ,aAAa,SAAS,IAAI,IAAI,GAC5D;OACK,IAAI,gBAAgB,eAAe,KAAK,YAAY,GACzD;OACK,IAAI,cAAc,IAAI,MAAM,WAAW,SAAS,IAAI,EAAE,GAC3D;EAIF,IAAI,EAAE,gBAAgB,cAAc,eAClC,SAAS,KAAK,GAAG;OACZ,IACL,gBACA,IAAI,QACJ,aAAa,MAAM,UAAU,UAAU,IAAI,IAAI,GAE/C,SAAS,KAAK,GAAG;OACZ,IAAI,gBAAgB,eAAe,KAAK,YAAY,GACzD,SAAS,KAAK,GAAG;OACZ,IAAI,cAAc,IAAI,MAAM,WAAW,MAAM,OAAO,OAAO,IAAI,EAAE,GACtE,SAAS,KAAK,GAAG;CAErB;CAEA,OAAO;AACT;AA8DA,SAAgB,iBACd,UACwD;CACxD,IAAI,MAAM,QAAQ,QAAQ,GACxB,OAAO,kBAAkB,QAAQ;CAEnC,OAAOA,eAAAA,eAAe,KAAK,iBAAiB;AAC9C;AAEA,SAAS,kBAAkB,UAAwC;CACjE,IAAI,CAAC,SAAS,QACZ,OAAO,CAAC;CAEV,MAAM,SAAwB,CAAC;CAC/B,KAAK,MAAM,OAAO,UAAU;EAC1B,MAAM,OAAO;EACb,MAAM,OAAO,OAAO,IAAI;EACxB,IAAI,CAAC,MACH,OAAO,KAAK,IAAI;OACX,IACL,KAAK,QAAQ,MAAM,UACnB,EAAE,KAAK,QAAQ,MAAM,KAAK,QAAQ,IAElC,OAAO,KAAK,MAAM,IAAI;OACjB;GACL,MAAM,YAAYC,cAAAA,eAAe,IAAI;GACrC,MAAM,YAAYA,cAAAA,eAAe,IAAI;GACrC,MAAM,eAAe,UAAU,OAAO,SAAS;GAC/C,IACE,OAAO,UAAU,YAAY,YAC7B,OAAO,UAAU,YAAY,UAE7B,aAAa,UAAU,GAAG,UAAU,QAAQ,IAAI,UAAU;GAE5D,OAAO,KAAK,YAAY,YAAY,CAAC;EACvC;CACF;CACA,OAAO;AACT;AA8UA,SAAgB,aACd,mBACA,SACiE;CACjE,IAAI,MAAM,QAAQ,iBAAiB,GAAG;EACpC,MAAM,WAAW;EACjB,IAAI,CAAC,SACH,MAAM,IAAI,MAAM,wDAAwD;EAE1E,OAAO,oBAAoB,UAAU,OAAO;CAC9C,OAAO;EACL,MAAM,iBAAiB;EACvB,OAAOD,eAAAA,eAAe,MAAM,UAC1B,oBAAoB,OAAO,cAAc,CAC3C,CAAC,CAAC,WAAW,EACX,SAAS,gBACX,CAAC;CACH;AACF;AAEA,eAAe,oBACb,UACA,SAC6B;CAC7B,MAAM,EACJ,WACA,cACA,WAAW,QACX,eAAe,OACf,OACA,SACA,gBAAgB,OAChB,iBACE;CACJ,IAAI,WAAW,aAAa,SAC1B,MAAM,IAAI,MACR,6DACF;CAEF,IAAI,iBAAiB,aAAa,SAChC,MAAM,IAAI,MACR,mEACF;CAGF,IAAI;CACJ,IAAI,kBAAkB,cACpB,mBAAmB,OAAO,SAAyC;EAIjE,QAAO,MAHmB,QAAQ,IAChC,KAAK,KAAK,QAAQ,aAAa,aAAa,IAAI,OAAO,CAAC,CAC1D,EAAA,CACmB,QAAQ,KAAK,UAAU,MAAM,OAAO,CAAC;CAC1D;MAEA,mBAAmB,OAAO,SACxB,aAAa,IAAI;CAGrB,IAAI,mBACF;CACF,IAAI,cACF,IAAI,eAAe,cACjB,mBAAmB,aAAa,UAAU,KAAK,YAAY;MAE3D,mBAAmB,OAAO,SACxB,aAAa,IAAI;CAIvB,IAAI,aAAa,SACf,OAAO,gBAAgB,UAAU;EAC/B;EACA,cAAc;EACd,cAAc;EACd,iBAAiB,eAAe,UAAU,KAAA;EAC1C;CACF,CAAC;MACI,IAAI,aAAa,QACtB,OAAO,eAAe,UAAU;EAC9B;EACA,cAAc;EACd,cAAc;EACd;EACA;EACA;EACA;CACF,CAAC;MAED,MAAM,IAAI,MACR,2BAA2B,SAAS,qCACtC;AAEJ;AAEA,eAAe,gBACb,UACA,SAOwB;CACxB,MAAM,EAAE,WAAW,cAAc,cAAc,iBAAiB,UAC9D;CACF,IAAI,eAAe,CAAC,GAAG,QAAQ;CAC/B,IAAI,MAAM;CACV,KAAK,IAAI,IAAI,GAAG,IAAI,aAAa,QAAQ,KAAK,GAE5C,IAAK,MAAM,aADe,IAAI,IAAI,aAAa,MAAM,GAAG,CAAC,CAAC,IAAI,YACrB,KAAM,WAAW;EACxD,MAAM,aAAa,SAAS;EAC5B;CACF;CAEF,IAAI,MAAM,aAAa,UAAU,iBAAiB;EAChD,IAAI,kBAAkB;EACtB,IAAI,MAAM,QAAQ,aAAa,IAAI,CAAC,OAAO,GAAG;GAC5C,MAAM,WAAW,aAAa;GAC9B,IAAI,OAAO,SAAS,YAAY,UAC9B,MAAM,IAAI,MAAM,kCAAkC;GAGpD,MAAM,WAAW,SAAS,QAAQ;GAClC,MAAM,kBACJ,oBAAoB,SAChB,CAAC,GAAG,SAAS,OAAO,CAAC,CAAC,QAAQ,IAC9B,SAAS;GACf,KAAK,IAAI,IAAI,GAAG,KAAK,UAAU,KAAK,GAAG;IACrC,MAAM,iBACJ,oBAAoB,UAChB,gBAAgB,MAAM,GAAG,CAAC,IAC1B,gBAAgB,MAAM,CAAC,CAAC;IAC9B,MAAM,SAAS,OAAO,YACpB,OAAO,QAAQ,QAAQ,CAAC,CAAC,QACtB,CAAC,OAAO,MAAM,UAAU,CAAC,EAAE,WAAW,KAAK,CAC9C,CACF;IACA,MAAM,iBAAiB,qBAAqB,SAAS,QAAQ,GAAG;KAC9D,GAAG;KACH,SAAS;IACX,CAAC;IACD,MAAM,iBAAiB,CAAC,GAAG,aAAa,MAAM,GAAG,GAAG,GAAG,cAAc;IACrE,IAAK,MAAM,aAAa,cAAc,KAAM,WAAW;KACrD,eAAe;KACf,OAAO;KACP,kBAAkB;IACpB,OACE;GAEJ;GACA,IAAI,mBAAmB,oBAAoB,QACzC,SAAS,UAAU,CAAC,GAAG,eAAe,CAAC,CAAC,QAAQ;EAEpD;EACA,IAAI,CAAC,iBAAiB;GACpB,MAAM,WAAW,aAAa;GAC9B,IAAI;GACJ,IACE,MAAM,QAAQ,SAAS,OAAO,KAC9B,SAAS,QAAQ,MACd,UAAU,OAAO,UAAU,YAAY,MAAM,SAAS,MACzD,GAKA,OAHkB,SAAS,QAAQ,MAChC,UAAU,MAAM,SAAS,UAAU,MAAM,IAE7B,CAAC,EAAE;QACb,IAAI,OAAO,SAAS,YAAY,UACrC,OAAO,SAAS;GAElB,IAAI,MAAM;IACR,MAAM,aAAa,MAAM,aAAa,IAAI;IAC1C,MAAM,YAAY,WAAW;IAC7B,IAAI,oBAAoB,QACtB,WAAW,QAAQ;IAErB,KAAK,IAAI,IAAI,GAAG,IAAI,YAAY,GAAG,KAAK,GAAG;KACzC,WAAW,IAAI;KACf,SAAS,UAAU,WAAW,KAAK,EAAE;KACrC,IACG,MAAM,aAAa,CAAC,GAAG,aAAa,MAAM,GAAG,GAAG,GAAG,QAAQ,CAAC,KAC7D,WACA;MACA,IAAI,oBAAoB,QACtB,SAAS,UAAU,CAAC,GAAG,UAAU,CAAC,CAAC,QAAQ,CAAC,CAAC,KAAK,EAAE;MAEtD,eAAe,CAAC,GAAG,aAAa,MAAM,GAAG,GAAG,GAAG,QAAQ;MACvD,OAAO;MACP;KACF;IACF;GACF;EACF;CACF;CAEA,IAAI,OAAO;EACT,MAAM,WAAW,MAAM,QAAQ,KAAK,IAAI,QAAQ,CAAC,KAAK;EACtD,OAAO,MAAM,KAAK,CAAC,eAAe,aAAa,MAAM,IAAI,QAAQ,GAC/D,OAAO;CAEX;CAEA,OAAO,aAAa,MAAM,GAAG,GAAG;AAClC;AAEA,eAAe,eACb,UACA,SAewB;CACxB,MAAM,EACJ,eAAe,OACf,gBAAgB,OAChB,OACA,SACA,GAAG,SACD;CAGJ,IAAI,eAAe,SAAS,KAAK,YAAY;EAC3C,MAAM,SAAS,OAAO,YACpB,OAAO,QAAQ,OAAO,CAAC,CAAC,QACrB,CAAC,OAAO,MAAM,UAAU,CAAC,EAAE,WAAW,KAAK,CAC9C,CACF;EACA,OAAO,qBACL,QAAQ,QAAQ,GAChB,QACAE,aAAAA,mBAAmB,OAAO,CAC5B;CACF,CAAC;CAED,IAAI,OAAO;EACT,MAAM,WAAW,MAAM,QAAQ,KAAK,IAAI,QAAQ,CAAC,KAAK;EACtD,OACE,aAAa,SAAS,KACtB,CAAC,eAAe,aAAa,aAAa,SAAS,IAAI,QAAQ,GAE/D,eAAe,aAAa,MAAM,GAAG,EAAE;CAE3C;CAEA,MAAM,gBACJ,iBAAiB,aAAa,EAAE,EAAE,QAAQ,MAAM;CAClD,IAAI,YAAY,gBACZ,aAAa,MAAM,GAAG,CAAC,CAAC,CAAC,OAAO,aAAa,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,IAC/D,aAAa,QAAQ;CAEzB,YAAY,MAAM,gBAAgB,WAAW;EAC3C,GAAG;EACH,iBAAiB,eAAe,SAAS,KAAA;EACzC,OAAO;CACT,CAAC;CAED,IAAI,eACF,OAAO,CAAC,UAAU,IAAI,GAAG,UAAU,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC;MAErD,OAAO,UAAU,QAAQ;AAE7B;AAEA,MAAM,iBAMF;CACF,OAAO;EACL,SAASC,cAAAA;EACT,cAAcC,cAAAA;CAChB;CACA,IAAI;EACF,SAASC,WAAAA;EACT,cAAcC,WAAAA;CAChB;CACA,QAAQ;EACN,SAASC,eAAAA;EACT,cAAcC,eAAAA;CAChB;CACA,WAAW;EACT,SAASD,eAAAA;EACT,cAAcC,eAAAA;CAChB;CACA,MAAM;EACJ,SAASC,sBAAAA;EACT,cAAcC,sBAAAA;CAChB;CACA,UAAU;EACR,SAASC,iBAAAA;EACT,cAAcC,iBAAAA;CAChB;CACA,SAAS;EACP,SAASC,aAAAA;EACT,cAAcC,aAAAA;CAChB;CACA,QAAQ;EACN,SAASC,iBAAAA;EACT,cAAcA,iBAAAA;CAChB;AACF;AAgBA,SAAS,qBACP,aACA,QACA,aACgC;CAChC,IAAI;CACJ,IAAI;CAEJ,QAAQ,aAAR;EACE,KAAK;GACH,IAAI,aACF,QAAQ,IAAIX,cAAAA,kBAAkB,MAAM;QAEpC,MAAM,IAAID,cAAAA,aAAa,MAAM;GAE/B;EACF,KAAK;GACH,IAAI,aAAa;IACf,IAAI,gBAAsC,EACxC,GAAG,OACL;IACA,IAAI,gBAAgB,eAClB,gBAAgB;KACd,GAAG;KACH,kBAAkB,cAAc,YAAY,KAAK,QAAQ;MACvD,GAAG;MACH,MAAM;MACN,OAAO,KAAA;MACP,MAAM,KAAK,UAAU,GAAG,IAAI;KAC9B,EAAE;IACJ;IAEF,QAAQ,IAAIG,WAAAA,eAAe,aAAa;GAC1C,OACE,MAAM,IAAID,WAAAA,UAAU,MAAM;GAE5B;EACF,KAAK;GACH,IAAI,aACF,QAAQ,IAAIG,eAAAA,mBAAmB,MAAM;QAErC,MAAM,IAAID,eAAAA,cAAc,MAAM;GAEhC;EACF,KAAK;GACH,IAAI,aACF,QAAQ,IAAIC,eAAAA,mBAAmB;IAC7B,GAAG;IACH,mBAAmB;KACjB,GAAG,OAAO;KACV,iBAAiB;IACnB;GACF,CAAC;QAED,MAAM,IAAID,eAAAA,cAAc;IACtB,GAAG;IACH,mBAAmB;KACjB,GAAG,OAAO;KACV,iBAAiB;IACnB;GACF,CAAC;GAEH;EACF,KAAK;GACH,IAAI,kBAAkB,QACpB,IAAI,aACF,QAAQ,IAAIG,sBAAAA,iBAAiB,MAA2B;QAExD,MAAM,IAAID,sBAAAA,YAAY,MAA2B;QAGnD,MAAM,IAAI,MACR,yFACF;GAEF;EACF,KAAK;GACH,IAAI,aACF,QAAQ,IAAIG,iBAAAA,qBAAqB,MAA+B;QAC3D;IACL,IAAI,CAAC,OAAO,MACV,MAAM,IAAI,MAAM,0CAA0C;IAE5D,MAAM,IAAID,iBAAAA,gBAAgB,MAA+B;GAC3D;GACA;EACF,KAAK;GACH,IAAI,UAAU,QACZ,IAAI,aACF,QAAQ,IAAIG,aAAAA,iBAAiB,MAA2B;QAExD,MAAM,IAAID,aAAAA,YAAY,MAA2B;QAGnD,MAAM,IAAI,MACR,iFACF;GAEF;EACF,SACE,MAAM,IAAI,MAAM,6BAA6B,aAAa;CAC9D;CAEA,IAAI,eAAe,OACjB,OAAO;CAET,IAAI,KACF,OAAO;CAET,MAAM,IAAI,MAAM,6BAA6B,aAAa;AAC5D;AAEA,SAAS,YAAY,OAAsC;CACzD,MAAM,YAAY,MAAM,QAAQ;CAChC,IAAI;CACJ,MAAM,SAAS,OAAO,YACpB,OAAO,QAAQ,KAAK,CAAC,CAAC,QACnB,CAAC,OAAO,CAAC,CAAC,QAAQ,kBAAkB,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,WAAW,KAAK,CAC3E,CACF;CAEA,IAAI,aAAa,gBACf,MAAM,qBAAqB,WAAW,MAAM;CAG9C,IAAI,CAAC,KACH,MAAM,IAAI,MACR,oCAAoC,UAAU,0BAA0B,OAAO,KAC7E,cACF,GACF;CAGF,OAAO;AACT;;;;;;;AAQA,SAAgB,oBAAoB,MAAiC;CACnE,MAAM,SAAS,KAAK,MAAM,IAAI;CAC9B,OAAO,QAAQ,QAAQ,CACrB,GAAG,OAAO,MAAM,GAAG,EAAE,CAAC,CAAC,KAAK,MAAM,GAAG,EAAE,GAAG,GAC1C,OAAO,OAAO,SAAS,EACzB,CAAC;AACH"}
1
+ {"version":3,"file":"transformers.cjs","names":["RunnableLambda","convertToChunk","isBaseMessageChunk","HumanMessage","HumanMessageChunk","AIMessage","AIMessageChunk","SystemMessage","SystemMessageChunk","ToolMessage","ToolMessageChunk","FunctionMessage","FunctionMessageChunk","ChatMessage","ChatMessageChunk","RemoveMessage"],"sources":["../../src/messages/transformers.ts"],"sourcesContent":["import { BaseLanguageModel } from \"../language_models/base.js\";\nimport { Runnable, RunnableLambda } from \"../runnables/base.js\";\nimport { AIMessage, AIMessageChunk, AIMessageChunkFields } from \"./ai.js\";\nimport {\n BaseMessage,\n BaseMessageChunk,\n BaseMessageFields,\n isBaseMessageChunk,\n} from \"./base.js\";\nimport { ChatMessage, ChatMessageChunk, ChatMessageFields } from \"./chat.js\";\nimport {\n FunctionMessage,\n FunctionMessageChunk,\n FunctionMessageFields,\n} from \"./function.js\";\nimport { HumanMessage, HumanMessageChunk } from \"./human.js\";\nimport { MessageType } from \"./message.js\";\nimport { RemoveMessage } from \"./modifier.js\";\nimport { SystemMessage, SystemMessageChunk } from \"./system.js\";\nimport { ToolMessage, ToolMessageChunk, ToolMessageFields } from \"./tool.js\";\nimport { convertToChunk } from \"./utils.js\";\n\nexport type MessageUnion =\n | typeof HumanMessage\n | typeof AIMessage\n | typeof SystemMessage\n | typeof ChatMessage\n | typeof FunctionMessage\n | typeof ToolMessage\n | typeof RemoveMessage;\nexport type MessageChunkUnion =\n | typeof HumanMessageChunk\n | typeof AIMessageChunk\n | typeof SystemMessageChunk\n | typeof FunctionMessageChunk\n | typeof ToolMessageChunk\n | typeof ChatMessageChunk\n | typeof RemoveMessage; // RemoveMessage does not have a chunk class.\nexport type MessageTypeOrClass = MessageType | MessageUnion | MessageChunkUnion;\n\nconst _isMessageType = (msg: BaseMessage, types: MessageTypeOrClass[]) => {\n const typesAsStrings = [\n ...new Set<string>(\n types?.map((t) => {\n if (typeof t === \"string\") {\n return t;\n }\n // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n const instantiatedMsgClass = new (t as any)({});\n if (\n !(\"getType\" in instantiatedMsgClass) ||\n typeof instantiatedMsgClass.getType !== \"function\"\n ) {\n throw new Error(\"Invalid type provided.\");\n }\n return instantiatedMsgClass.getType();\n })\n ),\n ];\n const msgType = msg.getType();\n return typesAsStrings.some((t) => t === msgType);\n};\n\nexport interface FilterMessagesFields {\n /**\n * @param {string[] | undefined} includeNames Message names to include.\n */\n includeNames?: string[];\n /**\n * @param {string[] | undefined} excludeNames Messages names to exclude.\n */\n excludeNames?: string[];\n /**\n * @param {(MessageType | BaseMessage)[] | undefined} includeTypes Message types to include. Can be specified as string names (e.g.\n * \"system\", \"human\", \"ai\", ...) or as BaseMessage classes (e.g.\n * SystemMessage, HumanMessage, AIMessage, ...).\n */\n includeTypes?: MessageTypeOrClass[];\n /**\n * @param {(MessageType | BaseMessage)[] | undefined} excludeTypes Message types to exclude. Can be specified as string names (e.g.\n * \"system\", \"human\", \"ai\", ...) or as BaseMessage classes (e.g.\n * SystemMessage, HumanMessage, AIMessage, ...).\n */\n excludeTypes?: MessageTypeOrClass[];\n /**\n * @param {string[] | undefined} includeIds Message IDs to include.\n */\n includeIds?: string[];\n /**\n * @param {string[] | undefined} excludeIds Message IDs to exclude.\n */\n excludeIds?: string[];\n}\n\n/**\n * Filter messages based on name, type or id.\n *\n * @param {BaseMessage[] | FilterMessagesFields} messagesOrOptions - Either an array of BaseMessage objects to filter or the filtering options. If an array is provided, the `options` parameter should also be supplied. If filtering options are provided, a RunnableLambda is returned.\n * @param {FilterMessagesFields} [options] - Optional filtering options. Should only be provided if `messagesOrOptions` is an array of BaseMessage objects.\n * @returns A list of Messages that meets at least one of the include conditions and none\n * of the exclude conditions, or a RunnableLambda which does the same. If no include conditions are specified then\n * anything that is not explicitly excluded will be included.\n * @throws {Error} If two incompatible arguments are provided.\n *\n * @example\n * ```typescript\n * import { filterMessages, AIMessage, HumanMessage, SystemMessage } from \"@langchain/core/messages\";\n *\n * const messages = [\n * new SystemMessage(\"you're a good assistant.\"),\n * new HumanMessage({ content: \"what's your name\", id: \"foo\", name: \"example_user\" }),\n * new AIMessage({ content: \"steve-o\", id: \"bar\", name: \"example_assistant\" }),\n * new HumanMessage({ content: \"what's your favorite color\", id: \"baz\" }),\n * new AIMessage({ content: \"silicon blue\" , id: \"blah\" }),\n * ];\n *\n * filterMessages(messages, {\n * includeNames: [\"example_user\", \"example_assistant\"],\n * includeTypes: [\"system\"],\n * excludeIds: [\"bar\"],\n * });\n * ```\n *\n * The above example would return:\n * ```typescript\n * [\n * new SystemMessage(\"you're a good assistant.\"),\n * new HumanMessage({ content: \"what's your name\", id: \"foo\", name: \"example_user\" }),\n * ]\n * ```\n */\nexport function filterMessages(\n options?: FilterMessagesFields\n): Runnable<BaseMessage[], BaseMessage[]>;\nexport function filterMessages(\n messages: BaseMessage[],\n options?: FilterMessagesFields\n): BaseMessage[];\nexport function filterMessages(\n messagesOrOptions?: BaseMessage[] | FilterMessagesFields,\n options?: FilterMessagesFields\n): BaseMessage[] | Runnable<BaseMessage[], BaseMessage[]> {\n if (Array.isArray(messagesOrOptions)) {\n return _filterMessages(messagesOrOptions, options);\n }\n return RunnableLambda.from((input: BaseMessage[]): BaseMessage[] => {\n return _filterMessages(input, messagesOrOptions);\n });\n}\n\nfunction _filterMessages(\n messages: BaseMessage[],\n options: FilterMessagesFields = {}\n): BaseMessage[] {\n const {\n includeNames,\n excludeNames,\n includeTypes,\n excludeTypes,\n includeIds,\n excludeIds,\n } = options;\n\n const filtered: BaseMessage[] = [];\n\n for (const msg of messages) {\n if (excludeNames && msg.name && excludeNames.includes(msg.name)) {\n continue;\n } else if (excludeTypes && _isMessageType(msg, excludeTypes)) {\n continue;\n } else if (excludeIds && msg.id && excludeIds.includes(msg.id)) {\n continue;\n }\n\n // default to inclusion when no inclusion criteria given.\n if (!(includeTypes || includeIds || includeNames)) {\n filtered.push(msg);\n } else if (\n includeNames &&\n msg.name &&\n includeNames.some((iName) => iName === msg.name)\n ) {\n filtered.push(msg);\n } else if (includeTypes && _isMessageType(msg, includeTypes)) {\n filtered.push(msg);\n } else if (includeIds && msg.id && includeIds.some((id) => id === msg.id)) {\n filtered.push(msg);\n }\n }\n\n return filtered;\n}\n\n/**\n * Merge consecutive Messages of the same type.\n *\n * **NOTE**: ToolMessages are not merged, as each has a distinct tool call id that\n * can't be merged.\n *\n * @param {BaseMessage[] | undefined} messages Sequence of Message-like objects to merge. Optional. If not provided, a RunnableLambda is returned.\n * @returns List of BaseMessages with consecutive runs of message types merged into single\n * messages, or a RunnableLambda which returns a list of BaseMessages If two messages being merged both have string contents, the merged\n * content is a concatenation of the two strings with a new-line separator. If at\n * least one of the messages has a list of content blocks, the merged content is a\n * list of content blocks.\n *\n * @example\n * ```typescript\n * import { mergeMessageRuns, AIMessage, HumanMessage, SystemMessage, ToolCall } from \"@langchain/core/messages\";\n *\n * const messages = [\n * new SystemMessage(\"you're a good assistant.\"),\n * new HumanMessage({ content: \"what's your favorite color\", id: \"foo\" }),\n * new HumanMessage({ content: \"wait your favorite food\", id: \"bar\" }),\n * new AIMessage({\n * content: \"my favorite colo\",\n * tool_calls: [{ name: \"blah_tool\", args: { x: 2 }, id: \"123\" }],\n * id: \"baz\",\n * }),\n * new AIMessage({\n * content: [{ type: \"text\", text: \"my favorite dish is lasagna\" }],\n * tool_calls: [{ name: \"blah_tool\", args: { x: -10 }, id: \"456\" }],\n * id: \"blur\",\n * }),\n * ];\n *\n * mergeMessageRuns(messages);\n * ```\n *\n * The above example would return:\n * ```typescript\n * [\n * new SystemMessage(\"you're a good assistant.\"),\n * new HumanMessage({\n * content: \"what's your favorite colorwait your favorite food\",\n * id: \"foo\",\n * }),\n * new AIMessage({\n * content: [\n * { type: \"text\", text: \"my favorite colo\" },\n * { type: \"text\", text: \"my favorite dish is lasagna\" },\n * ],\n * tool_calls: [\n * { name: \"blah_tool\", args: { x: 2 }, id: \"123\" },\n * { name: \"blah_tool\", args: { x: -10 }, id: \"456\" },\n * ],\n * id: \"baz\",\n * }),\n * ]\n * ```\n */\nexport function mergeMessageRuns(): Runnable<BaseMessage[], BaseMessage[]>;\nexport function mergeMessageRuns(messages: BaseMessage[]): BaseMessage[];\nexport function mergeMessageRuns(\n messages?: BaseMessage[]\n): BaseMessage[] | Runnable<BaseMessage[], BaseMessage[]> {\n if (Array.isArray(messages)) {\n return _mergeMessageRuns(messages);\n }\n return RunnableLambda.from(_mergeMessageRuns);\n}\n\nfunction _mergeMessageRuns(messages: BaseMessage[]): BaseMessage[] {\n if (!messages.length) {\n return [];\n }\n const merged: BaseMessage[] = [];\n for (const msg of messages) {\n const curr = msg;\n const last = merged.pop();\n if (!last) {\n merged.push(curr);\n } else if (\n curr.getType() === \"tool\" ||\n !(curr.getType() === last.getType())\n ) {\n merged.push(last, curr);\n } else {\n const lastChunk = convertToChunk(last) as BaseMessageChunk;\n const currChunk = convertToChunk(curr) as BaseMessageChunk;\n const mergedChunks = lastChunk.concat(currChunk);\n if (\n typeof lastChunk.content === \"string\" &&\n typeof currChunk.content === \"string\"\n ) {\n mergedChunks.content = `${lastChunk.content}\\n${currChunk.content}`;\n }\n merged.push(_chunkToMsg(mergedChunks));\n }\n }\n return merged;\n}\n\n// Since we can not import from `@langchain/textsplitters` we need\n// to reconstruct the interface here. Only `splitText` is used by\n// `trimMessages`, so we avoid coupling to the full `BaseDocumentTransformer`\n// surface and just require the method we call.\ninterface _TextSplitterInterface {\n splitText(text: string): Promise<string[]>;\n}\n\nexport interface TrimMessagesFields {\n /**\n * @param {number} maxTokens Max token count of trimmed messages.\n */\n maxTokens: number;\n /**\n * @param {((messages: BaseMessage[]) => number) | ((messages: BaseMessage[]) => Promise<number>) | BaseLanguageModel} tokenCounter\n * Function or LLM for counting tokens in an array of `BaseMessage`s.\n * If a `BaseLanguageModel` is passed in then `BaseLanguageModel.getNumTokens()` will be used.\n */\n tokenCounter:\n | ((messages: BaseMessage[]) => number)\n | ((messages: BaseMessage[]) => Promise<number>)\n | BaseLanguageModel;\n /**\n * @param {\"first\" | \"last\"} [strategy=\"last\"] Strategy for trimming.\n * - \"first\": Keep the first <= n_count tokens of the messages.\n * - \"last\": Keep the last <= n_count tokens of the messages.\n * @default \"last\"\n */\n strategy?: \"first\" | \"last\";\n /**\n * @param {boolean} [allowPartial=false] Whether to split a message if only part of the message can be included.\n * If `strategy: \"last\"` then the last partial contents of a message are included.\n * If `strategy: \"first\"` then the first partial contents of a message are included.\n * @default false\n */\n allowPartial?: boolean;\n /**\n * @param {MessageTypeOrClass | MessageTypeOrClass[]} [endOn] The message type to end on.\n * If specified then every message after the last occurrence of this type is ignored.\n * If `strategy === \"last\"` then this is done before we attempt to get the last `maxTokens`.\n * If `strategy === \"first\"` then this is done after we get the first `maxTokens`.\n * Can be specified as string names (e.g. \"system\", \"human\", \"ai\", ...) or as `BaseMessage` classes\n * (e.g. `SystemMessage`, `HumanMessage`, `AIMessage`, ...). Can be a single type or an array of types.\n */\n endOn?: MessageTypeOrClass | MessageTypeOrClass[];\n /**\n * @param {MessageTypeOrClass | MessageTypeOrClass[]} [startOn] The message type to start on.\n * Should only be specified if `strategy: \"last\"`. If specified then every message before the first occurrence\n * of this type is ignored. This is done after we trim the initial messages to the last `maxTokens`.\n * Does not apply to a `SystemMessage` at index 0 if `includeSystem: true`.\n * Can be specified as string names (e.g. \"system\", \"human\", \"ai\", ...) or as `BaseMessage` classes\n * (e.g. `SystemMessage`, `HumanMessage`, `AIMessage`, ...). Can be a single type or an array of types.\n */\n startOn?: MessageTypeOrClass | MessageTypeOrClass[];\n /**\n * @param {boolean} [includeSystem=false] Whether to keep the `SystemMessage` if there is one at index 0.\n * Should only be specified if `strategy: \"last\"`.\n * @default false\n */\n includeSystem?: boolean;\n /**\n * @param {((text: string) => string[]) | BaseDocumentTransformer} [textSplitter] Function or `BaseDocumentTransformer` for\n * splitting the string contents of a message. Only used if `allowPartial: true`.\n * If `strategy: \"last\"` then the last split tokens from a partial message will be included.\n * If `strategy: \"first\"` then the first split tokens from a partial message will be included.\n * Token splitter assumes that separators are kept, so that split contents can be directly concatenated\n * to recreate the original text. Defaults to splitting on newlines.\n */\n textSplitter?:\n | ((text: string) => string[])\n | ((text: string) => Promise<string[]>)\n | _TextSplitterInterface;\n}\n\n/**\n * Trim messages to be below a token count.\n *\n * @param {BaseMessage[]} messages Array of `BaseMessage` instances to trim.\n * @param {TrimMessagesFields} options Trimming options.\n * @returns An array of trimmed `BaseMessage`s or a `Runnable` that takes a sequence of `BaseMessage`-like objects and returns\n * an array of trimmed `BaseMessage`s.\n * @throws {Error} If two incompatible arguments are specified or an unrecognized `strategy` is specified.\n *\n * @example\n * ```typescript\n * import { trimMessages, AIMessage, BaseMessage, HumanMessage, SystemMessage } from \"@langchain/core/messages\";\n *\n * const messages = [\n * new SystemMessage(\"This is a 4 token text. The full message is 10 tokens.\"),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"first\",\n * }),\n * new AIMessage({\n * content: [\n * { type: \"text\", text: \"This is the FIRST 4 token block.\" },\n * { type: \"text\", text: \"This is the SECOND 4 token block.\" },\n * ],\n * id: \"second\",\n * }),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"third\",\n * }),\n * new AIMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"fourth\",\n * }),\n * ];\n *\n * function dummyTokenCounter(messages: BaseMessage[]): number {\n * // treat each message like it adds 3 default tokens at the beginning\n * // of the message and at the end of the message. 3 + 4 + 3 = 10 tokens\n * // per message.\n *\n * const defaultContentLen = 4;\n * const defaultMsgPrefixLen = 3;\n * const defaultMsgSuffixLen = 3;\n *\n * let count = 0;\n * for (const msg of messages) {\n * if (typeof msg.content === \"string\") {\n * count += defaultMsgPrefixLen + defaultContentLen + defaultMsgSuffixLen;\n * }\n * if (Array.isArray(msg.content)) {\n * count +=\n * defaultMsgPrefixLen +\n * msg.content.length * defaultContentLen +\n * defaultMsgSuffixLen;\n * }\n * }\n * return count;\n * }\n * ```\n *\n * First 30 tokens, not allowing partial messages:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 30,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"first\",\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"first\",\n * }),\n * ]\n * ```\n *\n * First 30 tokens, allowing partial messages:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 30,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"first\",\n * allowPartial: true,\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"first\",\n * }),\n * new AIMessage({\n * content: [{ type: \"text\", text: \"This is the FIRST 4 token block.\" }],\n * id: \"second\",\n * }),\n * ]\n * ```\n *\n * First 30 tokens, allowing partial messages, have to end on HumanMessage:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 30,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"first\",\n * allowPartial: true,\n * endOn: \"human\",\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"first\",\n * }),\n * ]\n * ```\n *\n * Last 30 tokens, including system message, not allowing partial messages:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 30,\n * includeSystem: true,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"last\",\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"third\",\n * }),\n * new AIMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"fourth\",\n * }),\n * ]\n * ```\n *\n * Last 40 tokens, including system message, allowing partial messages:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 40,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"last\",\n * allowPartial: true,\n * includeSystem: true,\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new AIMessage({\n * content: [{ type: \"text\", text: \"This is the FIRST 4 token block.\" }],\n * id: \"second\",\n * }),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"third\",\n * }),\n * new AIMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"fourth\",\n * }),\n * ]\n * ```\n *\n * Last 30 tokens, including system message, allowing partial messages, end on HumanMessage:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 30,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"last\",\n * endOn: \"human\",\n * includeSystem: true,\n * allowPartial: true,\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new AIMessage({\n * content: [{ type: \"text\", text: \"This is the FIRST 4 token block.\" }],\n * id: \"second\",\n * }),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"third\",\n * }),\n * ]\n * ```\n *\n * Last 40 tokens, including system message, allowing partial messages, start on HumanMessage:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 40,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"last\",\n * includeSystem: true,\n * allowPartial: true,\n * startOn: \"human\",\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"third\",\n * }),\n * new AIMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"fourth\",\n * }),\n * ]\n * ```\n */\nexport function trimMessages(\n options: TrimMessagesFields\n): Runnable<BaseMessage[], BaseMessage[]>;\nexport function trimMessages(\n messages: BaseMessage[],\n options: TrimMessagesFields\n): Promise<BaseMessage[]>;\nexport function trimMessages(\n messagesOrOptions: BaseMessage[] | TrimMessagesFields,\n options?: TrimMessagesFields\n): Promise<BaseMessage[]> | Runnable<BaseMessage[], BaseMessage[]> {\n if (Array.isArray(messagesOrOptions)) {\n const messages = messagesOrOptions;\n if (!options) {\n throw new Error(\"Options parameter is required when providing messages.\");\n }\n return _trimMessagesHelper(messages, options);\n } else {\n const trimmerOptions = messagesOrOptions;\n return RunnableLambda.from((input: BaseMessage[]) =>\n _trimMessagesHelper(input, trimmerOptions)\n ).withConfig({\n runName: \"trim_messages\",\n });\n }\n}\n\nasync function _trimMessagesHelper(\n messages: BaseMessage[],\n options: TrimMessagesFields\n): Promise<Array<BaseMessage>> {\n const {\n maxTokens,\n tokenCounter,\n strategy = \"last\",\n allowPartial = false,\n endOn,\n startOn,\n includeSystem = false,\n textSplitter,\n } = options;\n if (startOn && strategy === \"first\") {\n throw new Error(\n \"`startOn` should only be specified if `strategy` is 'last'.\"\n );\n }\n if (includeSystem && strategy === \"first\") {\n throw new Error(\n \"`includeSystem` should only be specified if `strategy` is 'last'.\"\n );\n }\n\n let listTokenCounter: (msgs: BaseMessage[]) => Promise<number>;\n if (\"getNumTokens\" in tokenCounter) {\n listTokenCounter = async (msgs: BaseMessage[]): Promise<number> => {\n const tokenCounts = await Promise.all(\n msgs.map((msg) => tokenCounter.getNumTokens(msg.content))\n );\n return tokenCounts.reduce((sum, count) => sum + count, 0);\n };\n } else {\n listTokenCounter = async (msgs: BaseMessage[]): Promise<number> =>\n tokenCounter(msgs);\n }\n\n let textSplitterFunc: (text: string) => Promise<string[]> =\n defaultTextSplitter;\n if (textSplitter) {\n if (\"splitText\" in textSplitter) {\n textSplitterFunc = textSplitter.splitText.bind(textSplitter);\n } else {\n textSplitterFunc = async (text: string): Promise<string[]> =>\n textSplitter(text);\n }\n }\n\n if (strategy === \"first\") {\n return _firstMaxTokens(messages, {\n maxTokens,\n tokenCounter: listTokenCounter,\n textSplitter: textSplitterFunc,\n partialStrategy: allowPartial ? \"first\" : undefined,\n endOn,\n });\n } else if (strategy === \"last\") {\n return _lastMaxTokens(messages, {\n maxTokens,\n tokenCounter: listTokenCounter,\n textSplitter: textSplitterFunc,\n allowPartial,\n includeSystem,\n startOn,\n endOn,\n });\n } else {\n throw new Error(\n `Unrecognized strategy: '${strategy}'. Must be one of 'first' or 'last'.`\n );\n }\n}\n\nasync function _firstMaxTokens(\n messages: BaseMessage[],\n options: {\n maxTokens: number;\n tokenCounter: (messages: BaseMessage[]) => Promise<number>;\n textSplitter: (text: string) => Promise<string[]>;\n partialStrategy?: \"first\" | \"last\";\n endOn?: MessageTypeOrClass | MessageTypeOrClass[];\n }\n): Promise<BaseMessage[]> {\n const { maxTokens, tokenCounter, textSplitter, partialStrategy, endOn } =\n options;\n let messagesCopy = [...messages];\n let idx = 0;\n for (let i = 0; i < messagesCopy.length; i += 1) {\n const remainingMessages = i > 0 ? messagesCopy.slice(0, -i) : messagesCopy;\n if ((await tokenCounter(remainingMessages)) <= maxTokens) {\n idx = messagesCopy.length - i;\n break;\n }\n }\n if (idx < messagesCopy.length && partialStrategy) {\n let includedPartial = false;\n if (Array.isArray(messagesCopy[idx].content)) {\n const excluded = messagesCopy[idx];\n if (typeof excluded.content === \"string\") {\n throw new Error(\"Expected content to be an array.\");\n }\n\n const numBlock = excluded.content.length;\n const reversedContent =\n partialStrategy === \"last\"\n ? [...excluded.content].reverse()\n : excluded.content;\n for (let i = 1; i <= numBlock; i += 1) {\n const partialContent =\n partialStrategy === \"first\"\n ? reversedContent.slice(0, i)\n : reversedContent.slice(-i);\n const fields = Object.fromEntries(\n Object.entries(excluded).filter(\n ([k]) => k !== \"type\" && !k.startsWith(\"lc_\")\n )\n ) as BaseMessageFields;\n const updatedMessage = _switchTypeToMessage(excluded.getType(), {\n ...fields,\n content: partialContent,\n });\n const slicedMessages = [...messagesCopy.slice(0, idx), updatedMessage];\n if ((await tokenCounter(slicedMessages)) <= maxTokens) {\n messagesCopy = slicedMessages;\n idx += 1;\n includedPartial = true;\n } else {\n break;\n }\n }\n if (includedPartial && partialStrategy === \"last\") {\n excluded.content = [...reversedContent].reverse();\n }\n }\n if (!includedPartial) {\n const excluded = messagesCopy[idx];\n let text: string | undefined;\n if (\n Array.isArray(excluded.content) &&\n excluded.content.some(\n (block) => typeof block === \"string\" || block.type === \"text\"\n )\n ) {\n const textBlock = excluded.content.find(\n (block) => block.type === \"text\" && block.text\n ) as { type: \"text\"; text: string } | undefined;\n text = textBlock?.text;\n } else if (typeof excluded.content === \"string\") {\n text = excluded.content;\n }\n if (text) {\n const splitTexts = await textSplitter(text);\n const numSplits = splitTexts.length;\n if (partialStrategy === \"last\") {\n splitTexts.reverse();\n }\n for (let _ = 0; _ < numSplits - 1; _ += 1) {\n splitTexts.pop();\n excluded.content = splitTexts.join(\"\");\n if (\n (await tokenCounter([...messagesCopy.slice(0, idx), excluded])) <=\n maxTokens\n ) {\n if (partialStrategy === \"last\") {\n excluded.content = [...splitTexts].reverse().join(\"\");\n }\n messagesCopy = [...messagesCopy.slice(0, idx), excluded];\n idx += 1;\n break;\n }\n }\n }\n }\n }\n\n if (endOn) {\n const endOnArr = Array.isArray(endOn) ? endOn : [endOn];\n while (idx > 0 && !_isMessageType(messagesCopy[idx - 1], endOnArr)) {\n idx -= 1;\n }\n }\n\n return messagesCopy.slice(0, idx);\n}\n\nasync function _lastMaxTokens(\n messages: BaseMessage[],\n options: {\n maxTokens: number;\n tokenCounter: (messages: BaseMessage[]) => Promise<number>;\n textSplitter: (text: string) => Promise<string[]>;\n /**\n * @default {false}\n */\n allowPartial?: boolean;\n /**\n * @default {false}\n */\n includeSystem?: boolean;\n startOn?: MessageTypeOrClass | MessageTypeOrClass[];\n endOn?: MessageTypeOrClass | MessageTypeOrClass[];\n }\n): Promise<BaseMessage[]> {\n const {\n allowPartial = false,\n includeSystem = false,\n endOn,\n startOn,\n ...rest\n } = options;\n\n // Create a copy of messages to avoid mutation\n let messagesCopy = messages.map((message) => {\n const fields = Object.fromEntries(\n Object.entries(message).filter(\n ([k]) => k !== \"type\" && !k.startsWith(\"lc_\")\n )\n ) as BaseMessageFields;\n return _switchTypeToMessage(\n message.getType(),\n fields,\n isBaseMessageChunk(message)\n );\n });\n\n if (endOn) {\n const endOnArr = Array.isArray(endOn) ? endOn : [endOn];\n while (\n messagesCopy.length > 0 &&\n !_isMessageType(messagesCopy[messagesCopy.length - 1], endOnArr)\n ) {\n messagesCopy = messagesCopy.slice(0, -1);\n }\n }\n\n const swappedSystem =\n includeSystem && messagesCopy[0]?.getType() === \"system\";\n let reversed_ = swappedSystem\n ? messagesCopy.slice(0, 1).concat(messagesCopy.slice(1).reverse())\n : messagesCopy.reverse();\n\n reversed_ = await _firstMaxTokens(reversed_, {\n ...rest,\n partialStrategy: allowPartial ? \"last\" : undefined,\n endOn: startOn,\n });\n\n if (swappedSystem) {\n return [reversed_[0], ...reversed_.slice(1).reverse()];\n } else {\n return reversed_.reverse();\n }\n}\n\nconst _MSG_CHUNK_MAP: Record<\n MessageType,\n {\n message: MessageUnion;\n messageChunk: MessageChunkUnion;\n }\n> = {\n human: {\n message: HumanMessage,\n messageChunk: HumanMessageChunk,\n },\n ai: {\n message: AIMessage,\n messageChunk: AIMessageChunk,\n },\n system: {\n message: SystemMessage,\n messageChunk: SystemMessageChunk,\n },\n developer: {\n message: SystemMessage,\n messageChunk: SystemMessageChunk,\n },\n tool: {\n message: ToolMessage,\n messageChunk: ToolMessageChunk,\n },\n function: {\n message: FunctionMessage,\n messageChunk: FunctionMessageChunk,\n },\n generic: {\n message: ChatMessage,\n messageChunk: ChatMessageChunk,\n },\n remove: {\n message: RemoveMessage,\n messageChunk: RemoveMessage, // RemoveMessage does not have a chunk class.\n },\n};\n\nfunction _switchTypeToMessage(\n messageType: MessageType,\n fields: BaseMessageFields\n): BaseMessage;\nfunction _switchTypeToMessage(\n messageType: MessageType,\n fields: BaseMessageFields,\n returnChunk: true\n): BaseMessageChunk;\nfunction _switchTypeToMessage(\n messageType: MessageType,\n fields: BaseMessageFields,\n returnChunk?: boolean\n): BaseMessageChunk | BaseMessage;\nfunction _switchTypeToMessage(\n messageType: MessageType,\n fields: BaseMessageFields,\n returnChunk?: boolean\n): BaseMessageChunk | BaseMessage {\n let chunk: BaseMessageChunk | undefined;\n let msg: BaseMessage | undefined;\n\n switch (messageType) {\n case \"human\":\n if (returnChunk) {\n chunk = new HumanMessageChunk(fields);\n } else {\n msg = new HumanMessage(fields);\n }\n break;\n case \"ai\":\n if (returnChunk) {\n let aiChunkFields: AIMessageChunkFields = {\n ...fields,\n };\n if (\"tool_calls\" in aiChunkFields) {\n aiChunkFields = {\n ...aiChunkFields,\n tool_call_chunks: aiChunkFields.tool_calls?.map((tc) => ({\n ...tc,\n type: \"tool_call_chunk\",\n index: undefined,\n args: JSON.stringify(tc.args),\n })),\n };\n }\n chunk = new AIMessageChunk(aiChunkFields);\n } else {\n msg = new AIMessage(fields);\n }\n break;\n case \"system\":\n if (returnChunk) {\n chunk = new SystemMessageChunk(fields);\n } else {\n msg = new SystemMessage(fields);\n }\n break;\n case \"developer\":\n if (returnChunk) {\n chunk = new SystemMessageChunk({\n ...fields,\n additional_kwargs: {\n ...fields.additional_kwargs,\n __openai_role__: \"developer\",\n },\n });\n } else {\n msg = new SystemMessage({\n ...fields,\n additional_kwargs: {\n ...fields.additional_kwargs,\n __openai_role__: \"developer\",\n },\n });\n }\n break;\n case \"tool\":\n if (\"tool_call_id\" in fields) {\n if (returnChunk) {\n chunk = new ToolMessageChunk(fields as ToolMessageFields);\n } else {\n msg = new ToolMessage(fields as ToolMessageFields);\n }\n } else {\n throw new Error(\n \"Can not convert ToolMessage to ToolMessageChunk if 'tool_call_id' field is not defined.\"\n );\n }\n break;\n case \"function\":\n if (returnChunk) {\n chunk = new FunctionMessageChunk(fields as FunctionMessageFields);\n } else {\n if (!fields.name) {\n throw new Error(\"FunctionMessage must have a 'name' field\");\n }\n msg = new FunctionMessage(fields as FunctionMessageFields);\n }\n break;\n case \"generic\":\n if (\"role\" in fields) {\n if (returnChunk) {\n chunk = new ChatMessageChunk(fields as ChatMessageFields);\n } else {\n msg = new ChatMessage(fields as ChatMessageFields);\n }\n } else {\n throw new Error(\n \"Can not convert ChatMessage to ChatMessageChunk if 'role' field is not defined.\"\n );\n }\n break;\n default:\n throw new Error(`Unrecognized message type ${messageType}`);\n }\n\n if (returnChunk && chunk) {\n return chunk;\n }\n if (msg) {\n return msg;\n }\n throw new Error(`Unrecognized message type ${messageType}`);\n}\n\nfunction _chunkToMsg(chunk: BaseMessageChunk): BaseMessage {\n const chunkType = chunk.getType();\n let msg: BaseMessage | undefined;\n const fields = Object.fromEntries(\n Object.entries(chunk).filter(\n ([k]) => ![\"type\", \"tool_call_chunks\"].includes(k) && !k.startsWith(\"lc_\")\n )\n ) as BaseMessageFields;\n\n if (chunkType in _MSG_CHUNK_MAP) {\n msg = _switchTypeToMessage(chunkType, fields);\n }\n\n if (!msg) {\n throw new Error(\n `Unrecognized message chunk class ${chunkType}. Supported classes are ${Object.keys(\n _MSG_CHUNK_MAP\n )}`\n );\n }\n\n return msg;\n}\n\n/**\n * The default text splitter function that splits text by newlines.\n *\n * @param {string} text\n * @returns A promise that resolves to an array of strings split by newlines.\n */\nexport function defaultTextSplitter(text: string): Promise<string[]> {\n const splits = text.split(\"\\n\");\n return Promise.resolve([\n ...splits.slice(0, -1).map((s) => `${s}\\n`),\n splits[splits.length - 1],\n ]);\n}\n"],"mappings":";;;;;;;;;;;AAwCA,MAAM,kBAAkB,KAAkB,UAAgC;CACxE,MAAM,iBAAiB,CACrB,GAAG,IAAI,IACL,OAAO,KAAK,MAAM;EAChB,IAAI,OAAO,MAAM,UACf,OAAO;EAGT,MAAM,uBAAuB,IAAK,EAAU,CAAC,CAAC;EAC9C,IACE,EAAE,aAAa,yBACf,OAAO,qBAAqB,YAAY,YAExC,MAAM,IAAI,MAAM,wBAAwB;EAE1C,OAAO,qBAAqB,QAAQ;CACtC,CAAC,CACH,CACF;CACA,MAAM,UAAU,IAAI,QAAQ;CAC5B,OAAO,eAAe,MAAM,MAAM,MAAM,OAAO;AACjD;AA6EA,SAAgB,eACd,mBACA,SACwD;CACxD,IAAI,MAAM,QAAQ,iBAAiB,GACjC,OAAO,gBAAgB,mBAAmB,OAAO;CAEnD,OAAOA,eAAAA,eAAe,MAAM,UAAwC;EAClE,OAAO,gBAAgB,OAAO,iBAAiB;CACjD,CAAC;AACH;AAEA,SAAS,gBACP,UACA,UAAgC,CAAC,GAClB;CACf,MAAM,EACJ,cACA,cACA,cACA,cACA,YACA,eACE;CAEJ,MAAM,WAA0B,CAAC;CAEjC,KAAK,MAAM,OAAO,UAAU;EAC1B,IAAI,gBAAgB,IAAI,QAAQ,aAAa,SAAS,IAAI,IAAI,GAC5D;OACK,IAAI,gBAAgB,eAAe,KAAK,YAAY,GACzD;OACK,IAAI,cAAc,IAAI,MAAM,WAAW,SAAS,IAAI,EAAE,GAC3D;EAIF,IAAI,EAAE,gBAAgB,cAAc,eAClC,SAAS,KAAK,GAAG;OACZ,IACL,gBACA,IAAI,QACJ,aAAa,MAAM,UAAU,UAAU,IAAI,IAAI,GAE/C,SAAS,KAAK,GAAG;OACZ,IAAI,gBAAgB,eAAe,KAAK,YAAY,GACzD,SAAS,KAAK,GAAG;OACZ,IAAI,cAAc,IAAI,MAAM,WAAW,MAAM,OAAO,OAAO,IAAI,EAAE,GACtE,SAAS,KAAK,GAAG;CAErB;CAEA,OAAO;AACT;AA8DA,SAAgB,iBACd,UACwD;CACxD,IAAI,MAAM,QAAQ,QAAQ,GACxB,OAAO,kBAAkB,QAAQ;CAEnC,OAAOA,eAAAA,eAAe,KAAK,iBAAiB;AAC9C;AAEA,SAAS,kBAAkB,UAAwC;CACjE,IAAI,CAAC,SAAS,QACZ,OAAO,CAAC;CAEV,MAAM,SAAwB,CAAC;CAC/B,KAAK,MAAM,OAAO,UAAU;EAC1B,MAAM,OAAO;EACb,MAAM,OAAO,OAAO,IAAI;EACxB,IAAI,CAAC,MACH,OAAO,KAAK,IAAI;OACX,IACL,KAAK,QAAQ,MAAM,UACnB,EAAE,KAAK,QAAQ,MAAM,KAAK,QAAQ,IAElC,OAAO,KAAK,MAAM,IAAI;OACjB;GACL,MAAM,YAAYC,cAAAA,eAAe,IAAI;GACrC,MAAM,YAAYA,cAAAA,eAAe,IAAI;GACrC,MAAM,eAAe,UAAU,OAAO,SAAS;GAC/C,IACE,OAAO,UAAU,YAAY,YAC7B,OAAO,UAAU,YAAY,UAE7B,aAAa,UAAU,GAAG,UAAU,QAAQ,IAAI,UAAU;GAE5D,OAAO,KAAK,YAAY,YAAY,CAAC;EACvC;CACF;CACA,OAAO;AACT;AA8UA,SAAgB,aACd,mBACA,SACiE;CACjE,IAAI,MAAM,QAAQ,iBAAiB,GAAG;EACpC,MAAM,WAAW;EACjB,IAAI,CAAC,SACH,MAAM,IAAI,MAAM,wDAAwD;EAE1E,OAAO,oBAAoB,UAAU,OAAO;CAC9C,OAAO;EACL,MAAM,iBAAiB;EACvB,OAAOD,eAAAA,eAAe,MAAM,UAC1B,oBAAoB,OAAO,cAAc,CAC3C,CAAC,CAAC,WAAW,EACX,SAAS,gBACX,CAAC;CACH;AACF;AAEA,eAAe,oBACb,UACA,SAC6B;CAC7B,MAAM,EACJ,WACA,cACA,WAAW,QACX,eAAe,OACf,OACA,SACA,gBAAgB,OAChB,iBACE;CACJ,IAAI,WAAW,aAAa,SAC1B,MAAM,IAAI,MACR,6DACF;CAEF,IAAI,iBAAiB,aAAa,SAChC,MAAM,IAAI,MACR,mEACF;CAGF,IAAI;CACJ,IAAI,kBAAkB,cACpB,mBAAmB,OAAO,SAAyC;EAIjE,QAAO,MAHmB,QAAQ,IAChC,KAAK,KAAK,QAAQ,aAAa,aAAa,IAAI,OAAO,CAAC,CAC1D,EAAA,CACmB,QAAQ,KAAK,UAAU,MAAM,OAAO,CAAC;CAC1D;MAEA,mBAAmB,OAAO,SACxB,aAAa,IAAI;CAGrB,IAAI,mBACF;CACF,IAAI,cAAc;EAChB,IAAI,eAAe,cACjB,mBAAmB,aAAa,UAAU,KAAK,YAAY;OAE3D,mBAAmB,OAAO,SACxB,aAAa,IAAI;CAEvB;CAEA,IAAI,aAAa,SACf,OAAO,gBAAgB,UAAU;EAC/B;EACA,cAAc;EACd,cAAc;EACd,iBAAiB,eAAe,UAAU,KAAA;EAC1C;CACF,CAAC;MACI,IAAI,aAAa,QACtB,OAAO,eAAe,UAAU;EAC9B;EACA,cAAc;EACd,cAAc;EACd;EACA;EACA;EACA;CACF,CAAC;MAED,MAAM,IAAI,MACR,2BAA2B,SAAS,qCACtC;AAEJ;AAEA,eAAe,gBACb,UACA,SAOwB;CACxB,MAAM,EAAE,WAAW,cAAc,cAAc,iBAAiB,UAC9D;CACF,IAAI,eAAe,CAAC,GAAG,QAAQ;CAC/B,IAAI,MAAM;CACV,KAAK,IAAI,IAAI,GAAG,IAAI,aAAa,QAAQ,KAAK,GAE5C,IAAK,MAAM,aADe,IAAI,IAAI,aAAa,MAAM,GAAG,CAAC,CAAC,IAAI,YACrB,KAAM,WAAW;EACxD,MAAM,aAAa,SAAS;EAC5B;CACF;CAEF,IAAI,MAAM,aAAa,UAAU,iBAAiB;EAChD,IAAI,kBAAkB;EACtB,IAAI,MAAM,QAAQ,aAAa,IAAI,CAAC,OAAO,GAAG;GAC5C,MAAM,WAAW,aAAa;GAC9B,IAAI,OAAO,SAAS,YAAY,UAC9B,MAAM,IAAI,MAAM,kCAAkC;GAGpD,MAAM,WAAW,SAAS,QAAQ;GAClC,MAAM,kBACJ,oBAAoB,SAChB,CAAC,GAAG,SAAS,OAAO,CAAC,CAAC,QAAQ,IAC9B,SAAS;GACf,KAAK,IAAI,IAAI,GAAG,KAAK,UAAU,KAAK,GAAG;IACrC,MAAM,iBACJ,oBAAoB,UAChB,gBAAgB,MAAM,GAAG,CAAC,IAC1B,gBAAgB,MAAM,CAAC,CAAC;IAC9B,MAAM,SAAS,OAAO,YACpB,OAAO,QAAQ,QAAQ,CAAC,CAAC,QACtB,CAAC,OAAO,MAAM,UAAU,CAAC,EAAE,WAAW,KAAK,CAC9C,CACF;IACA,MAAM,iBAAiB,qBAAqB,SAAS,QAAQ,GAAG;KAC9D,GAAG;KACH,SAAS;IACX,CAAC;IACD,MAAM,iBAAiB,CAAC,GAAG,aAAa,MAAM,GAAG,GAAG,GAAG,cAAc;IACrE,IAAK,MAAM,aAAa,cAAc,KAAM,WAAW;KACrD,eAAe;KACf,OAAO;KACP,kBAAkB;IACpB,OACE;GAEJ;GACA,IAAI,mBAAmB,oBAAoB,QACzC,SAAS,UAAU,CAAC,GAAG,eAAe,CAAC,CAAC,QAAQ;EAEpD;EACA,IAAI,CAAC,iBAAiB;GACpB,MAAM,WAAW,aAAa;GAC9B,IAAI;GACJ,IACE,MAAM,QAAQ,SAAS,OAAO,KAC9B,SAAS,QAAQ,MACd,UAAU,OAAO,UAAU,YAAY,MAAM,SAAS,MACzD,GAKA,OAHkB,SAAS,QAAQ,MAChC,UAAU,MAAM,SAAS,UAAU,MAAM,IAE7B,CAAC,EAAE;QACb,IAAI,OAAO,SAAS,YAAY,UACrC,OAAO,SAAS;GAElB,IAAI,MAAM;IACR,MAAM,aAAa,MAAM,aAAa,IAAI;IAC1C,MAAM,YAAY,WAAW;IAC7B,IAAI,oBAAoB,QACtB,WAAW,QAAQ;IAErB,KAAK,IAAI,IAAI,GAAG,IAAI,YAAY,GAAG,KAAK,GAAG;KACzC,WAAW,IAAI;KACf,SAAS,UAAU,WAAW,KAAK,EAAE;KACrC,IACG,MAAM,aAAa,CAAC,GAAG,aAAa,MAAM,GAAG,GAAG,GAAG,QAAQ,CAAC,KAC7D,WACA;MACA,IAAI,oBAAoB,QACtB,SAAS,UAAU,CAAC,GAAG,UAAU,CAAC,CAAC,QAAQ,CAAC,CAAC,KAAK,EAAE;MAEtD,eAAe,CAAC,GAAG,aAAa,MAAM,GAAG,GAAG,GAAG,QAAQ;MACvD,OAAO;MACP;KACF;IACF;GACF;EACF;CACF;CAEA,IAAI,OAAO;EACT,MAAM,WAAW,MAAM,QAAQ,KAAK,IAAI,QAAQ,CAAC,KAAK;EACtD,OAAO,MAAM,KAAK,CAAC,eAAe,aAAa,MAAM,IAAI,QAAQ,GAC/D,OAAO;CAEX;CAEA,OAAO,aAAa,MAAM,GAAG,GAAG;AAClC;AAEA,eAAe,eACb,UACA,SAewB;CACxB,MAAM,EACJ,eAAe,OACf,gBAAgB,OAChB,OACA,SACA,GAAG,SACD;CAGJ,IAAI,eAAe,SAAS,KAAK,YAAY;EAC3C,MAAM,SAAS,OAAO,YACpB,OAAO,QAAQ,OAAO,CAAC,CAAC,QACrB,CAAC,OAAO,MAAM,UAAU,CAAC,EAAE,WAAW,KAAK,CAC9C,CACF;EACA,OAAO,qBACL,QAAQ,QAAQ,GAChB,QACAE,aAAAA,mBAAmB,OAAO,CAC5B;CACF,CAAC;CAED,IAAI,OAAO;EACT,MAAM,WAAW,MAAM,QAAQ,KAAK,IAAI,QAAQ,CAAC,KAAK;EACtD,OACE,aAAa,SAAS,KACtB,CAAC,eAAe,aAAa,aAAa,SAAS,IAAI,QAAQ,GAE/D,eAAe,aAAa,MAAM,GAAG,EAAE;CAE3C;CAEA,MAAM,gBACJ,iBAAiB,aAAa,EAAE,EAAE,QAAQ,MAAM;CAClD,IAAI,YAAY,gBACZ,aAAa,MAAM,GAAG,CAAC,CAAC,CAAC,OAAO,aAAa,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,IAC/D,aAAa,QAAQ;CAEzB,YAAY,MAAM,gBAAgB,WAAW;EAC3C,GAAG;EACH,iBAAiB,eAAe,SAAS,KAAA;EACzC,OAAO;CACT,CAAC;CAED,IAAI,eACF,OAAO,CAAC,UAAU,IAAI,GAAG,UAAU,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC;MAErD,OAAO,UAAU,QAAQ;AAE7B;AAEA,MAAM,iBAMF;CACF,OAAO;EACL,SAASC,cAAAA;EACT,cAAcC,cAAAA;CAChB;CACA,IAAI;EACF,SAASC,WAAAA;EACT,cAAcC,WAAAA;CAChB;CACA,QAAQ;EACN,SAASC,eAAAA;EACT,cAAcC,eAAAA;CAChB;CACA,WAAW;EACT,SAASD,eAAAA;EACT,cAAcC,eAAAA;CAChB;CACA,MAAM;EACJ,SAASC,sBAAAA;EACT,cAAcC,sBAAAA;CAChB;CACA,UAAU;EACR,SAASC,iBAAAA;EACT,cAAcC,iBAAAA;CAChB;CACA,SAAS;EACP,SAASC,aAAAA;EACT,cAAcC,aAAAA;CAChB;CACA,QAAQ;EACN,SAASC,iBAAAA;EACT,cAAcA,iBAAAA;CAChB;AACF;AAgBA,SAAS,qBACP,aACA,QACA,aACgC;CAChC,IAAI;CACJ,IAAI;CAEJ,QAAQ,aAAR;EACE,KAAK;GACH,IAAI,aACF,QAAQ,IAAIX,cAAAA,kBAAkB,MAAM;QAEpC,MAAM,IAAID,cAAAA,aAAa,MAAM;GAE/B;EACF,KAAK;GACH,IAAI,aAAa;IACf,IAAI,gBAAsC,EACxC,GAAG,OACL;IACA,IAAI,gBAAgB,eAClB,gBAAgB;KACd,GAAG;KACH,kBAAkB,cAAc,YAAY,KAAK,QAAQ;MACvD,GAAG;MACH,MAAM;MACN,OAAO,KAAA;MACP,MAAM,KAAK,UAAU,GAAG,IAAI;KAC9B,EAAE;IACJ;IAEF,QAAQ,IAAIG,WAAAA,eAAe,aAAa;GAC1C,OACE,MAAM,IAAID,WAAAA,UAAU,MAAM;GAE5B;EACF,KAAK;GACH,IAAI,aACF,QAAQ,IAAIG,eAAAA,mBAAmB,MAAM;QAErC,MAAM,IAAID,eAAAA,cAAc,MAAM;GAEhC;EACF,KAAK;GACH,IAAI,aACF,QAAQ,IAAIC,eAAAA,mBAAmB;IAC7B,GAAG;IACH,mBAAmB;KACjB,GAAG,OAAO;KACV,iBAAiB;IACnB;GACF,CAAC;QAED,MAAM,IAAID,eAAAA,cAAc;IACtB,GAAG;IACH,mBAAmB;KACjB,GAAG,OAAO;KACV,iBAAiB;IACnB;GACF,CAAC;GAEH;EACF,KAAK;GACH,IAAI,kBAAkB,QAAQ;IAC5B,IAAI,aACF,QAAQ,IAAIG,sBAAAA,iBAAiB,MAA2B;SAExD,MAAM,IAAID,sBAAAA,YAAY,MAA2B;GAErD,OACE,MAAM,IAAI,MACR,yFACF;GAEF;EACF,KAAK;GACH,IAAI,aACF,QAAQ,IAAIG,iBAAAA,qBAAqB,MAA+B;QAC3D;IACL,IAAI,CAAC,OAAO,MACV,MAAM,IAAI,MAAM,0CAA0C;IAE5D,MAAM,IAAID,iBAAAA,gBAAgB,MAA+B;GAC3D;GACA;EACF,KAAK;GACH,IAAI,UAAU,QAAQ;IACpB,IAAI,aACF,QAAQ,IAAIG,aAAAA,iBAAiB,MAA2B;SAExD,MAAM,IAAID,aAAAA,YAAY,MAA2B;GAErD,OACE,MAAM,IAAI,MACR,iFACF;GAEF;EACF,SACE,MAAM,IAAI,MAAM,6BAA6B,aAAa;CAC9D;CAEA,IAAI,eAAe,OACjB,OAAO;CAET,IAAI,KACF,OAAO;CAET,MAAM,IAAI,MAAM,6BAA6B,aAAa;AAC5D;AAEA,SAAS,YAAY,OAAsC;CACzD,MAAM,YAAY,MAAM,QAAQ;CAChC,IAAI;CACJ,MAAM,SAAS,OAAO,YACpB,OAAO,QAAQ,KAAK,CAAC,CAAC,QACnB,CAAC,OAAO,CAAC,CAAC,QAAQ,kBAAkB,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,WAAW,KAAK,CAC3E,CACF;CAEA,IAAI,aAAa,gBACf,MAAM,qBAAqB,WAAW,MAAM;CAG9C,IAAI,CAAC,KACH,MAAM,IAAI,MACR,oCAAoC,UAAU,0BAA0B,OAAO,KAC7E,cACF,GACF;CAGF,OAAO;AACT;;;;;;;AAQA,SAAgB,oBAAoB,MAAiC;CACnE,MAAM,SAAS,KAAK,MAAM,IAAI;CAC9B,OAAO,QAAQ,QAAQ,CACrB,GAAG,OAAO,MAAM,GAAG,EAAE,CAAC,CAAC,KAAK,MAAM,GAAG,EAAE,GAAG,GAC1C,OAAO,OAAO,SAAS,EACzB,CAAC;AACH"}
@@ -81,8 +81,10 @@ async function _trimMessagesHelper(messages, options) {
81
81
  };
82
82
  else listTokenCounter = async (msgs) => tokenCounter(msgs);
83
83
  let textSplitterFunc = defaultTextSplitter;
84
- if (textSplitter) if ("splitText" in textSplitter) textSplitterFunc = textSplitter.splitText.bind(textSplitter);
85
- else textSplitterFunc = async (text) => textSplitter(text);
84
+ if (textSplitter) {
85
+ if ("splitText" in textSplitter) textSplitterFunc = textSplitter.splitText.bind(textSplitter);
86
+ else textSplitterFunc = async (text) => textSplitter(text);
87
+ }
86
88
  if (strategy === "first") return _firstMaxTokens(messages, {
87
89
  maxTokens,
88
90
  tokenCounter: listTokenCounter,
@@ -258,9 +260,10 @@ function _switchTypeToMessage(messageType, fields, returnChunk) {
258
260
  });
259
261
  break;
260
262
  case "tool":
261
- if ("tool_call_id" in fields) if (returnChunk) chunk = new ToolMessageChunk(fields);
262
- else msg = new ToolMessage(fields);
263
- else throw new Error("Can not convert ToolMessage to ToolMessageChunk if 'tool_call_id' field is not defined.");
263
+ if ("tool_call_id" in fields) {
264
+ if (returnChunk) chunk = new ToolMessageChunk(fields);
265
+ else msg = new ToolMessage(fields);
266
+ } else throw new Error("Can not convert ToolMessage to ToolMessageChunk if 'tool_call_id' field is not defined.");
264
267
  break;
265
268
  case "function":
266
269
  if (returnChunk) chunk = new FunctionMessageChunk(fields);
@@ -270,9 +273,10 @@ function _switchTypeToMessage(messageType, fields, returnChunk) {
270
273
  }
271
274
  break;
272
275
  case "generic":
273
- if ("role" in fields) if (returnChunk) chunk = new ChatMessageChunk(fields);
274
- else msg = new ChatMessage(fields);
275
- else throw new Error("Can not convert ChatMessage to ChatMessageChunk if 'role' field is not defined.");
276
+ if ("role" in fields) {
277
+ if (returnChunk) chunk = new ChatMessageChunk(fields);
278
+ else msg = new ChatMessage(fields);
279
+ } else throw new Error("Can not convert ChatMessage to ChatMessageChunk if 'role' field is not defined.");
276
280
  break;
277
281
  default: throw new Error(`Unrecognized message type ${messageType}`);
278
282
  }
@@ -1 +1 @@
1
- {"version":3,"file":"transformers.js","names":[],"sources":["../../src/messages/transformers.ts"],"sourcesContent":["import { BaseLanguageModel } from \"../language_models/base.js\";\nimport { Runnable, RunnableLambda } from \"../runnables/base.js\";\nimport { AIMessage, AIMessageChunk, AIMessageChunkFields } from \"./ai.js\";\nimport {\n BaseMessage,\n BaseMessageChunk,\n BaseMessageFields,\n isBaseMessageChunk,\n} from \"./base.js\";\nimport { ChatMessage, ChatMessageChunk, ChatMessageFields } from \"./chat.js\";\nimport {\n FunctionMessage,\n FunctionMessageChunk,\n FunctionMessageFields,\n} from \"./function.js\";\nimport { HumanMessage, HumanMessageChunk } from \"./human.js\";\nimport { MessageType } from \"./message.js\";\nimport { RemoveMessage } from \"./modifier.js\";\nimport { SystemMessage, SystemMessageChunk } from \"./system.js\";\nimport { ToolMessage, ToolMessageChunk, ToolMessageFields } from \"./tool.js\";\nimport { convertToChunk } from \"./utils.js\";\n\nexport type MessageUnion =\n | typeof HumanMessage\n | typeof AIMessage\n | typeof SystemMessage\n | typeof ChatMessage\n | typeof FunctionMessage\n | typeof ToolMessage\n | typeof RemoveMessage;\nexport type MessageChunkUnion =\n | typeof HumanMessageChunk\n | typeof AIMessageChunk\n | typeof SystemMessageChunk\n | typeof FunctionMessageChunk\n | typeof ToolMessageChunk\n | typeof ChatMessageChunk\n | typeof RemoveMessage; // RemoveMessage does not have a chunk class.\nexport type MessageTypeOrClass = MessageType | MessageUnion | MessageChunkUnion;\n\nconst _isMessageType = (msg: BaseMessage, types: MessageTypeOrClass[]) => {\n const typesAsStrings = [\n ...new Set<string>(\n types?.map((t) => {\n if (typeof t === \"string\") {\n return t;\n }\n // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n const instantiatedMsgClass = new (t as any)({});\n if (\n !(\"getType\" in instantiatedMsgClass) ||\n typeof instantiatedMsgClass.getType !== \"function\"\n ) {\n throw new Error(\"Invalid type provided.\");\n }\n return instantiatedMsgClass.getType();\n })\n ),\n ];\n const msgType = msg.getType();\n return typesAsStrings.some((t) => t === msgType);\n};\n\nexport interface FilterMessagesFields {\n /**\n * @param {string[] | undefined} includeNames Message names to include.\n */\n includeNames?: string[];\n /**\n * @param {string[] | undefined} excludeNames Messages names to exclude.\n */\n excludeNames?: string[];\n /**\n * @param {(MessageType | BaseMessage)[] | undefined} includeTypes Message types to include. Can be specified as string names (e.g.\n * \"system\", \"human\", \"ai\", ...) or as BaseMessage classes (e.g.\n * SystemMessage, HumanMessage, AIMessage, ...).\n */\n includeTypes?: MessageTypeOrClass[];\n /**\n * @param {(MessageType | BaseMessage)[] | undefined} excludeTypes Message types to exclude. Can be specified as string names (e.g.\n * \"system\", \"human\", \"ai\", ...) or as BaseMessage classes (e.g.\n * SystemMessage, HumanMessage, AIMessage, ...).\n */\n excludeTypes?: MessageTypeOrClass[];\n /**\n * @param {string[] | undefined} includeIds Message IDs to include.\n */\n includeIds?: string[];\n /**\n * @param {string[] | undefined} excludeIds Message IDs to exclude.\n */\n excludeIds?: string[];\n}\n\n/**\n * Filter messages based on name, type or id.\n *\n * @param {BaseMessage[] | FilterMessagesFields} messagesOrOptions - Either an array of BaseMessage objects to filter or the filtering options. If an array is provided, the `options` parameter should also be supplied. If filtering options are provided, a RunnableLambda is returned.\n * @param {FilterMessagesFields} [options] - Optional filtering options. Should only be provided if `messagesOrOptions` is an array of BaseMessage objects.\n * @returns A list of Messages that meets at least one of the include conditions and none\n * of the exclude conditions, or a RunnableLambda which does the same. If no include conditions are specified then\n * anything that is not explicitly excluded will be included.\n * @throws {Error} If two incompatible arguments are provided.\n *\n * @example\n * ```typescript\n * import { filterMessages, AIMessage, HumanMessage, SystemMessage } from \"@langchain/core/messages\";\n *\n * const messages = [\n * new SystemMessage(\"you're a good assistant.\"),\n * new HumanMessage({ content: \"what's your name\", id: \"foo\", name: \"example_user\" }),\n * new AIMessage({ content: \"steve-o\", id: \"bar\", name: \"example_assistant\" }),\n * new HumanMessage({ content: \"what's your favorite color\", id: \"baz\" }),\n * new AIMessage({ content: \"silicon blue\" , id: \"blah\" }),\n * ];\n *\n * filterMessages(messages, {\n * includeNames: [\"example_user\", \"example_assistant\"],\n * includeTypes: [\"system\"],\n * excludeIds: [\"bar\"],\n * });\n * ```\n *\n * The above example would return:\n * ```typescript\n * [\n * new SystemMessage(\"you're a good assistant.\"),\n * new HumanMessage({ content: \"what's your name\", id: \"foo\", name: \"example_user\" }),\n * ]\n * ```\n */\nexport function filterMessages(\n options?: FilterMessagesFields\n): Runnable<BaseMessage[], BaseMessage[]>;\nexport function filterMessages(\n messages: BaseMessage[],\n options?: FilterMessagesFields\n): BaseMessage[];\nexport function filterMessages(\n messagesOrOptions?: BaseMessage[] | FilterMessagesFields,\n options?: FilterMessagesFields\n): BaseMessage[] | Runnable<BaseMessage[], BaseMessage[]> {\n if (Array.isArray(messagesOrOptions)) {\n return _filterMessages(messagesOrOptions, options);\n }\n return RunnableLambda.from((input: BaseMessage[]): BaseMessage[] => {\n return _filterMessages(input, messagesOrOptions);\n });\n}\n\nfunction _filterMessages(\n messages: BaseMessage[],\n options: FilterMessagesFields = {}\n): BaseMessage[] {\n const {\n includeNames,\n excludeNames,\n includeTypes,\n excludeTypes,\n includeIds,\n excludeIds,\n } = options;\n\n const filtered: BaseMessage[] = [];\n\n for (const msg of messages) {\n if (excludeNames && msg.name && excludeNames.includes(msg.name)) {\n continue;\n } else if (excludeTypes && _isMessageType(msg, excludeTypes)) {\n continue;\n } else if (excludeIds && msg.id && excludeIds.includes(msg.id)) {\n continue;\n }\n\n // default to inclusion when no inclusion criteria given.\n if (!(includeTypes || includeIds || includeNames)) {\n filtered.push(msg);\n } else if (\n includeNames &&\n msg.name &&\n includeNames.some((iName) => iName === msg.name)\n ) {\n filtered.push(msg);\n } else if (includeTypes && _isMessageType(msg, includeTypes)) {\n filtered.push(msg);\n } else if (includeIds && msg.id && includeIds.some((id) => id === msg.id)) {\n filtered.push(msg);\n }\n }\n\n return filtered;\n}\n\n/**\n * Merge consecutive Messages of the same type.\n *\n * **NOTE**: ToolMessages are not merged, as each has a distinct tool call id that\n * can't be merged.\n *\n * @param {BaseMessage[] | undefined} messages Sequence of Message-like objects to merge. Optional. If not provided, a RunnableLambda is returned.\n * @returns List of BaseMessages with consecutive runs of message types merged into single\n * messages, or a RunnableLambda which returns a list of BaseMessages If two messages being merged both have string contents, the merged\n * content is a concatenation of the two strings with a new-line separator. If at\n * least one of the messages has a list of content blocks, the merged content is a\n * list of content blocks.\n *\n * @example\n * ```typescript\n * import { mergeMessageRuns, AIMessage, HumanMessage, SystemMessage, ToolCall } from \"@langchain/core/messages\";\n *\n * const messages = [\n * new SystemMessage(\"you're a good assistant.\"),\n * new HumanMessage({ content: \"what's your favorite color\", id: \"foo\" }),\n * new HumanMessage({ content: \"wait your favorite food\", id: \"bar\" }),\n * new AIMessage({\n * content: \"my favorite colo\",\n * tool_calls: [{ name: \"blah_tool\", args: { x: 2 }, id: \"123\" }],\n * id: \"baz\",\n * }),\n * new AIMessage({\n * content: [{ type: \"text\", text: \"my favorite dish is lasagna\" }],\n * tool_calls: [{ name: \"blah_tool\", args: { x: -10 }, id: \"456\" }],\n * id: \"blur\",\n * }),\n * ];\n *\n * mergeMessageRuns(messages);\n * ```\n *\n * The above example would return:\n * ```typescript\n * [\n * new SystemMessage(\"you're a good assistant.\"),\n * new HumanMessage({\n * content: \"what's your favorite colorwait your favorite food\",\n * id: \"foo\",\n * }),\n * new AIMessage({\n * content: [\n * { type: \"text\", text: \"my favorite colo\" },\n * { type: \"text\", text: \"my favorite dish is lasagna\" },\n * ],\n * tool_calls: [\n * { name: \"blah_tool\", args: { x: 2 }, id: \"123\" },\n * { name: \"blah_tool\", args: { x: -10 }, id: \"456\" },\n * ],\n * id: \"baz\",\n * }),\n * ]\n * ```\n */\nexport function mergeMessageRuns(): Runnable<BaseMessage[], BaseMessage[]>;\nexport function mergeMessageRuns(messages: BaseMessage[]): BaseMessage[];\nexport function mergeMessageRuns(\n messages?: BaseMessage[]\n): BaseMessage[] | Runnable<BaseMessage[], BaseMessage[]> {\n if (Array.isArray(messages)) {\n return _mergeMessageRuns(messages);\n }\n return RunnableLambda.from(_mergeMessageRuns);\n}\n\nfunction _mergeMessageRuns(messages: BaseMessage[]): BaseMessage[] {\n if (!messages.length) {\n return [];\n }\n const merged: BaseMessage[] = [];\n for (const msg of messages) {\n const curr = msg;\n const last = merged.pop();\n if (!last) {\n merged.push(curr);\n } else if (\n curr.getType() === \"tool\" ||\n !(curr.getType() === last.getType())\n ) {\n merged.push(last, curr);\n } else {\n const lastChunk = convertToChunk(last) as BaseMessageChunk;\n const currChunk = convertToChunk(curr) as BaseMessageChunk;\n const mergedChunks = lastChunk.concat(currChunk);\n if (\n typeof lastChunk.content === \"string\" &&\n typeof currChunk.content === \"string\"\n ) {\n mergedChunks.content = `${lastChunk.content}\\n${currChunk.content}`;\n }\n merged.push(_chunkToMsg(mergedChunks));\n }\n }\n return merged;\n}\n\n// Since we can not import from `@langchain/textsplitters` we need\n// to reconstruct the interface here. Only `splitText` is used by\n// `trimMessages`, so we avoid coupling to the full `BaseDocumentTransformer`\n// surface and just require the method we call.\ninterface _TextSplitterInterface {\n splitText(text: string): Promise<string[]>;\n}\n\nexport interface TrimMessagesFields {\n /**\n * @param {number} maxTokens Max token count of trimmed messages.\n */\n maxTokens: number;\n /**\n * @param {((messages: BaseMessage[]) => number) | ((messages: BaseMessage[]) => Promise<number>) | BaseLanguageModel} tokenCounter\n * Function or LLM for counting tokens in an array of `BaseMessage`s.\n * If a `BaseLanguageModel` is passed in then `BaseLanguageModel.getNumTokens()` will be used.\n */\n tokenCounter:\n | ((messages: BaseMessage[]) => number)\n | ((messages: BaseMessage[]) => Promise<number>)\n | BaseLanguageModel;\n /**\n * @param {\"first\" | \"last\"} [strategy=\"last\"] Strategy for trimming.\n * - \"first\": Keep the first <= n_count tokens of the messages.\n * - \"last\": Keep the last <= n_count tokens of the messages.\n * @default \"last\"\n */\n strategy?: \"first\" | \"last\";\n /**\n * @param {boolean} [allowPartial=false] Whether to split a message if only part of the message can be included.\n * If `strategy: \"last\"` then the last partial contents of a message are included.\n * If `strategy: \"first\"` then the first partial contents of a message are included.\n * @default false\n */\n allowPartial?: boolean;\n /**\n * @param {MessageTypeOrClass | MessageTypeOrClass[]} [endOn] The message type to end on.\n * If specified then every message after the last occurrence of this type is ignored.\n * If `strategy === \"last\"` then this is done before we attempt to get the last `maxTokens`.\n * If `strategy === \"first\"` then this is done after we get the first `maxTokens`.\n * Can be specified as string names (e.g. \"system\", \"human\", \"ai\", ...) or as `BaseMessage` classes\n * (e.g. `SystemMessage`, `HumanMessage`, `AIMessage`, ...). Can be a single type or an array of types.\n */\n endOn?: MessageTypeOrClass | MessageTypeOrClass[];\n /**\n * @param {MessageTypeOrClass | MessageTypeOrClass[]} [startOn] The message type to start on.\n * Should only be specified if `strategy: \"last\"`. If specified then every message before the first occurrence\n * of this type is ignored. This is done after we trim the initial messages to the last `maxTokens`.\n * Does not apply to a `SystemMessage` at index 0 if `includeSystem: true`.\n * Can be specified as string names (e.g. \"system\", \"human\", \"ai\", ...) or as `BaseMessage` classes\n * (e.g. `SystemMessage`, `HumanMessage`, `AIMessage`, ...). Can be a single type or an array of types.\n */\n startOn?: MessageTypeOrClass | MessageTypeOrClass[];\n /**\n * @param {boolean} [includeSystem=false] Whether to keep the `SystemMessage` if there is one at index 0.\n * Should only be specified if `strategy: \"last\"`.\n * @default false\n */\n includeSystem?: boolean;\n /**\n * @param {((text: string) => string[]) | BaseDocumentTransformer} [textSplitter] Function or `BaseDocumentTransformer` for\n * splitting the string contents of a message. Only used if `allowPartial: true`.\n * If `strategy: \"last\"` then the last split tokens from a partial message will be included.\n * If `strategy: \"first\"` then the first split tokens from a partial message will be included.\n * Token splitter assumes that separators are kept, so that split contents can be directly concatenated\n * to recreate the original text. Defaults to splitting on newlines.\n */\n textSplitter?:\n | ((text: string) => string[])\n | ((text: string) => Promise<string[]>)\n | _TextSplitterInterface;\n}\n\n/**\n * Trim messages to be below a token count.\n *\n * @param {BaseMessage[]} messages Array of `BaseMessage` instances to trim.\n * @param {TrimMessagesFields} options Trimming options.\n * @returns An array of trimmed `BaseMessage`s or a `Runnable` that takes a sequence of `BaseMessage`-like objects and returns\n * an array of trimmed `BaseMessage`s.\n * @throws {Error} If two incompatible arguments are specified or an unrecognized `strategy` is specified.\n *\n * @example\n * ```typescript\n * import { trimMessages, AIMessage, BaseMessage, HumanMessage, SystemMessage } from \"@langchain/core/messages\";\n *\n * const messages = [\n * new SystemMessage(\"This is a 4 token text. The full message is 10 tokens.\"),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"first\",\n * }),\n * new AIMessage({\n * content: [\n * { type: \"text\", text: \"This is the FIRST 4 token block.\" },\n * { type: \"text\", text: \"This is the SECOND 4 token block.\" },\n * ],\n * id: \"second\",\n * }),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"third\",\n * }),\n * new AIMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"fourth\",\n * }),\n * ];\n *\n * function dummyTokenCounter(messages: BaseMessage[]): number {\n * // treat each message like it adds 3 default tokens at the beginning\n * // of the message and at the end of the message. 3 + 4 + 3 = 10 tokens\n * // per message.\n *\n * const defaultContentLen = 4;\n * const defaultMsgPrefixLen = 3;\n * const defaultMsgSuffixLen = 3;\n *\n * let count = 0;\n * for (const msg of messages) {\n * if (typeof msg.content === \"string\") {\n * count += defaultMsgPrefixLen + defaultContentLen + defaultMsgSuffixLen;\n * }\n * if (Array.isArray(msg.content)) {\n * count +=\n * defaultMsgPrefixLen +\n * msg.content.length * defaultContentLen +\n * defaultMsgSuffixLen;\n * }\n * }\n * return count;\n * }\n * ```\n *\n * First 30 tokens, not allowing partial messages:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 30,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"first\",\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"first\",\n * }),\n * ]\n * ```\n *\n * First 30 tokens, allowing partial messages:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 30,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"first\",\n * allowPartial: true,\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"first\",\n * }),\n * new AIMessage({\n * content: [{ type: \"text\", text: \"This is the FIRST 4 token block.\" }],\n * id: \"second\",\n * }),\n * ]\n * ```\n *\n * First 30 tokens, allowing partial messages, have to end on HumanMessage:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 30,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"first\",\n * allowPartial: true,\n * endOn: \"human\",\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"first\",\n * }),\n * ]\n * ```\n *\n * Last 30 tokens, including system message, not allowing partial messages:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 30,\n * includeSystem: true,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"last\",\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"third\",\n * }),\n * new AIMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"fourth\",\n * }),\n * ]\n * ```\n *\n * Last 40 tokens, including system message, allowing partial messages:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 40,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"last\",\n * allowPartial: true,\n * includeSystem: true,\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new AIMessage({\n * content: [{ type: \"text\", text: \"This is the FIRST 4 token block.\" }],\n * id: \"second\",\n * }),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"third\",\n * }),\n * new AIMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"fourth\",\n * }),\n * ]\n * ```\n *\n * Last 30 tokens, including system message, allowing partial messages, end on HumanMessage:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 30,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"last\",\n * endOn: \"human\",\n * includeSystem: true,\n * allowPartial: true,\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new AIMessage({\n * content: [{ type: \"text\", text: \"This is the FIRST 4 token block.\" }],\n * id: \"second\",\n * }),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"third\",\n * }),\n * ]\n * ```\n *\n * Last 40 tokens, including system message, allowing partial messages, start on HumanMessage:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 40,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"last\",\n * includeSystem: true,\n * allowPartial: true,\n * startOn: \"human\",\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"third\",\n * }),\n * new AIMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"fourth\",\n * }),\n * ]\n * ```\n */\nexport function trimMessages(\n options: TrimMessagesFields\n): Runnable<BaseMessage[], BaseMessage[]>;\nexport function trimMessages(\n messages: BaseMessage[],\n options: TrimMessagesFields\n): Promise<BaseMessage[]>;\nexport function trimMessages(\n messagesOrOptions: BaseMessage[] | TrimMessagesFields,\n options?: TrimMessagesFields\n): Promise<BaseMessage[]> | Runnable<BaseMessage[], BaseMessage[]> {\n if (Array.isArray(messagesOrOptions)) {\n const messages = messagesOrOptions;\n if (!options) {\n throw new Error(\"Options parameter is required when providing messages.\");\n }\n return _trimMessagesHelper(messages, options);\n } else {\n const trimmerOptions = messagesOrOptions;\n return RunnableLambda.from((input: BaseMessage[]) =>\n _trimMessagesHelper(input, trimmerOptions)\n ).withConfig({\n runName: \"trim_messages\",\n });\n }\n}\n\nasync function _trimMessagesHelper(\n messages: BaseMessage[],\n options: TrimMessagesFields\n): Promise<Array<BaseMessage>> {\n const {\n maxTokens,\n tokenCounter,\n strategy = \"last\",\n allowPartial = false,\n endOn,\n startOn,\n includeSystem = false,\n textSplitter,\n } = options;\n if (startOn && strategy === \"first\") {\n throw new Error(\n \"`startOn` should only be specified if `strategy` is 'last'.\"\n );\n }\n if (includeSystem && strategy === \"first\") {\n throw new Error(\n \"`includeSystem` should only be specified if `strategy` is 'last'.\"\n );\n }\n\n let listTokenCounter: (msgs: BaseMessage[]) => Promise<number>;\n if (\"getNumTokens\" in tokenCounter) {\n listTokenCounter = async (msgs: BaseMessage[]): Promise<number> => {\n const tokenCounts = await Promise.all(\n msgs.map((msg) => tokenCounter.getNumTokens(msg.content))\n );\n return tokenCounts.reduce((sum, count) => sum + count, 0);\n };\n } else {\n listTokenCounter = async (msgs: BaseMessage[]): Promise<number> =>\n tokenCounter(msgs);\n }\n\n let textSplitterFunc: (text: string) => Promise<string[]> =\n defaultTextSplitter;\n if (textSplitter) {\n if (\"splitText\" in textSplitter) {\n textSplitterFunc = textSplitter.splitText.bind(textSplitter);\n } else {\n textSplitterFunc = async (text: string): Promise<string[]> =>\n textSplitter(text);\n }\n }\n\n if (strategy === \"first\") {\n return _firstMaxTokens(messages, {\n maxTokens,\n tokenCounter: listTokenCounter,\n textSplitter: textSplitterFunc,\n partialStrategy: allowPartial ? \"first\" : undefined,\n endOn,\n });\n } else if (strategy === \"last\") {\n return _lastMaxTokens(messages, {\n maxTokens,\n tokenCounter: listTokenCounter,\n textSplitter: textSplitterFunc,\n allowPartial,\n includeSystem,\n startOn,\n endOn,\n });\n } else {\n throw new Error(\n `Unrecognized strategy: '${strategy}'. Must be one of 'first' or 'last'.`\n );\n }\n}\n\nasync function _firstMaxTokens(\n messages: BaseMessage[],\n options: {\n maxTokens: number;\n tokenCounter: (messages: BaseMessage[]) => Promise<number>;\n textSplitter: (text: string) => Promise<string[]>;\n partialStrategy?: \"first\" | \"last\";\n endOn?: MessageTypeOrClass | MessageTypeOrClass[];\n }\n): Promise<BaseMessage[]> {\n const { maxTokens, tokenCounter, textSplitter, partialStrategy, endOn } =\n options;\n let messagesCopy = [...messages];\n let idx = 0;\n for (let i = 0; i < messagesCopy.length; i += 1) {\n const remainingMessages = i > 0 ? messagesCopy.slice(0, -i) : messagesCopy;\n if ((await tokenCounter(remainingMessages)) <= maxTokens) {\n idx = messagesCopy.length - i;\n break;\n }\n }\n if (idx < messagesCopy.length && partialStrategy) {\n let includedPartial = false;\n if (Array.isArray(messagesCopy[idx].content)) {\n const excluded = messagesCopy[idx];\n if (typeof excluded.content === \"string\") {\n throw new Error(\"Expected content to be an array.\");\n }\n\n const numBlock = excluded.content.length;\n const reversedContent =\n partialStrategy === \"last\"\n ? [...excluded.content].reverse()\n : excluded.content;\n for (let i = 1; i <= numBlock; i += 1) {\n const partialContent =\n partialStrategy === \"first\"\n ? reversedContent.slice(0, i)\n : reversedContent.slice(-i);\n const fields = Object.fromEntries(\n Object.entries(excluded).filter(\n ([k]) => k !== \"type\" && !k.startsWith(\"lc_\")\n )\n ) as BaseMessageFields;\n const updatedMessage = _switchTypeToMessage(excluded.getType(), {\n ...fields,\n content: partialContent,\n });\n const slicedMessages = [...messagesCopy.slice(0, idx), updatedMessage];\n if ((await tokenCounter(slicedMessages)) <= maxTokens) {\n messagesCopy = slicedMessages;\n idx += 1;\n includedPartial = true;\n } else {\n break;\n }\n }\n if (includedPartial && partialStrategy === \"last\") {\n excluded.content = [...reversedContent].reverse();\n }\n }\n if (!includedPartial) {\n const excluded = messagesCopy[idx];\n let text: string | undefined;\n if (\n Array.isArray(excluded.content) &&\n excluded.content.some(\n (block) => typeof block === \"string\" || block.type === \"text\"\n )\n ) {\n const textBlock = excluded.content.find(\n (block) => block.type === \"text\" && block.text\n ) as { type: \"text\"; text: string } | undefined;\n text = textBlock?.text;\n } else if (typeof excluded.content === \"string\") {\n text = excluded.content;\n }\n if (text) {\n const splitTexts = await textSplitter(text);\n const numSplits = splitTexts.length;\n if (partialStrategy === \"last\") {\n splitTexts.reverse();\n }\n for (let _ = 0; _ < numSplits - 1; _ += 1) {\n splitTexts.pop();\n excluded.content = splitTexts.join(\"\");\n if (\n (await tokenCounter([...messagesCopy.slice(0, idx), excluded])) <=\n maxTokens\n ) {\n if (partialStrategy === \"last\") {\n excluded.content = [...splitTexts].reverse().join(\"\");\n }\n messagesCopy = [...messagesCopy.slice(0, idx), excluded];\n idx += 1;\n break;\n }\n }\n }\n }\n }\n\n if (endOn) {\n const endOnArr = Array.isArray(endOn) ? endOn : [endOn];\n while (idx > 0 && !_isMessageType(messagesCopy[idx - 1], endOnArr)) {\n idx -= 1;\n }\n }\n\n return messagesCopy.slice(0, idx);\n}\n\nasync function _lastMaxTokens(\n messages: BaseMessage[],\n options: {\n maxTokens: number;\n tokenCounter: (messages: BaseMessage[]) => Promise<number>;\n textSplitter: (text: string) => Promise<string[]>;\n /**\n * @default {false}\n */\n allowPartial?: boolean;\n /**\n * @default {false}\n */\n includeSystem?: boolean;\n startOn?: MessageTypeOrClass | MessageTypeOrClass[];\n endOn?: MessageTypeOrClass | MessageTypeOrClass[];\n }\n): Promise<BaseMessage[]> {\n const {\n allowPartial = false,\n includeSystem = false,\n endOn,\n startOn,\n ...rest\n } = options;\n\n // Create a copy of messages to avoid mutation\n let messagesCopy = messages.map((message) => {\n const fields = Object.fromEntries(\n Object.entries(message).filter(\n ([k]) => k !== \"type\" && !k.startsWith(\"lc_\")\n )\n ) as BaseMessageFields;\n return _switchTypeToMessage(\n message.getType(),\n fields,\n isBaseMessageChunk(message)\n );\n });\n\n if (endOn) {\n const endOnArr = Array.isArray(endOn) ? endOn : [endOn];\n while (\n messagesCopy.length > 0 &&\n !_isMessageType(messagesCopy[messagesCopy.length - 1], endOnArr)\n ) {\n messagesCopy = messagesCopy.slice(0, -1);\n }\n }\n\n const swappedSystem =\n includeSystem && messagesCopy[0]?.getType() === \"system\";\n let reversed_ = swappedSystem\n ? messagesCopy.slice(0, 1).concat(messagesCopy.slice(1).reverse())\n : messagesCopy.reverse();\n\n reversed_ = await _firstMaxTokens(reversed_, {\n ...rest,\n partialStrategy: allowPartial ? \"last\" : undefined,\n endOn: startOn,\n });\n\n if (swappedSystem) {\n return [reversed_[0], ...reversed_.slice(1).reverse()];\n } else {\n return reversed_.reverse();\n }\n}\n\nconst _MSG_CHUNK_MAP: Record<\n MessageType,\n {\n message: MessageUnion;\n messageChunk: MessageChunkUnion;\n }\n> = {\n human: {\n message: HumanMessage,\n messageChunk: HumanMessageChunk,\n },\n ai: {\n message: AIMessage,\n messageChunk: AIMessageChunk,\n },\n system: {\n message: SystemMessage,\n messageChunk: SystemMessageChunk,\n },\n developer: {\n message: SystemMessage,\n messageChunk: SystemMessageChunk,\n },\n tool: {\n message: ToolMessage,\n messageChunk: ToolMessageChunk,\n },\n function: {\n message: FunctionMessage,\n messageChunk: FunctionMessageChunk,\n },\n generic: {\n message: ChatMessage,\n messageChunk: ChatMessageChunk,\n },\n remove: {\n message: RemoveMessage,\n messageChunk: RemoveMessage, // RemoveMessage does not have a chunk class.\n },\n};\n\nfunction _switchTypeToMessage(\n messageType: MessageType,\n fields: BaseMessageFields\n): BaseMessage;\nfunction _switchTypeToMessage(\n messageType: MessageType,\n fields: BaseMessageFields,\n returnChunk: true\n): BaseMessageChunk;\nfunction _switchTypeToMessage(\n messageType: MessageType,\n fields: BaseMessageFields,\n returnChunk?: boolean\n): BaseMessageChunk | BaseMessage;\nfunction _switchTypeToMessage(\n messageType: MessageType,\n fields: BaseMessageFields,\n returnChunk?: boolean\n): BaseMessageChunk | BaseMessage {\n let chunk: BaseMessageChunk | undefined;\n let msg: BaseMessage | undefined;\n\n switch (messageType) {\n case \"human\":\n if (returnChunk) {\n chunk = new HumanMessageChunk(fields);\n } else {\n msg = new HumanMessage(fields);\n }\n break;\n case \"ai\":\n if (returnChunk) {\n let aiChunkFields: AIMessageChunkFields = {\n ...fields,\n };\n if (\"tool_calls\" in aiChunkFields) {\n aiChunkFields = {\n ...aiChunkFields,\n tool_call_chunks: aiChunkFields.tool_calls?.map((tc) => ({\n ...tc,\n type: \"tool_call_chunk\",\n index: undefined,\n args: JSON.stringify(tc.args),\n })),\n };\n }\n chunk = new AIMessageChunk(aiChunkFields);\n } else {\n msg = new AIMessage(fields);\n }\n break;\n case \"system\":\n if (returnChunk) {\n chunk = new SystemMessageChunk(fields);\n } else {\n msg = new SystemMessage(fields);\n }\n break;\n case \"developer\":\n if (returnChunk) {\n chunk = new SystemMessageChunk({\n ...fields,\n additional_kwargs: {\n ...fields.additional_kwargs,\n __openai_role__: \"developer\",\n },\n });\n } else {\n msg = new SystemMessage({\n ...fields,\n additional_kwargs: {\n ...fields.additional_kwargs,\n __openai_role__: \"developer\",\n },\n });\n }\n break;\n case \"tool\":\n if (\"tool_call_id\" in fields) {\n if (returnChunk) {\n chunk = new ToolMessageChunk(fields as ToolMessageFields);\n } else {\n msg = new ToolMessage(fields as ToolMessageFields);\n }\n } else {\n throw new Error(\n \"Can not convert ToolMessage to ToolMessageChunk if 'tool_call_id' field is not defined.\"\n );\n }\n break;\n case \"function\":\n if (returnChunk) {\n chunk = new FunctionMessageChunk(fields as FunctionMessageFields);\n } else {\n if (!fields.name) {\n throw new Error(\"FunctionMessage must have a 'name' field\");\n }\n msg = new FunctionMessage(fields as FunctionMessageFields);\n }\n break;\n case \"generic\":\n if (\"role\" in fields) {\n if (returnChunk) {\n chunk = new ChatMessageChunk(fields as ChatMessageFields);\n } else {\n msg = new ChatMessage(fields as ChatMessageFields);\n }\n } else {\n throw new Error(\n \"Can not convert ChatMessage to ChatMessageChunk if 'role' field is not defined.\"\n );\n }\n break;\n default:\n throw new Error(`Unrecognized message type ${messageType}`);\n }\n\n if (returnChunk && chunk) {\n return chunk;\n }\n if (msg) {\n return msg;\n }\n throw new Error(`Unrecognized message type ${messageType}`);\n}\n\nfunction _chunkToMsg(chunk: BaseMessageChunk): BaseMessage {\n const chunkType = chunk.getType();\n let msg: BaseMessage | undefined;\n const fields = Object.fromEntries(\n Object.entries(chunk).filter(\n ([k]) => ![\"type\", \"tool_call_chunks\"].includes(k) && !k.startsWith(\"lc_\")\n )\n ) as BaseMessageFields;\n\n if (chunkType in _MSG_CHUNK_MAP) {\n msg = _switchTypeToMessage(chunkType, fields);\n }\n\n if (!msg) {\n throw new Error(\n `Unrecognized message chunk class ${chunkType}. Supported classes are ${Object.keys(\n _MSG_CHUNK_MAP\n )}`\n );\n }\n\n return msg;\n}\n\n/**\n * The default text splitter function that splits text by newlines.\n *\n * @param {string} text\n * @returns A promise that resolves to an array of strings split by newlines.\n */\nexport function defaultTextSplitter(text: string): Promise<string[]> {\n const splits = text.split(\"\\n\");\n return Promise.resolve([\n ...splits.slice(0, -1).map((s) => `${s}\\n`),\n splits[splits.length - 1],\n ]);\n}\n"],"mappings":";;;;;;;;;;;AAwCA,MAAM,kBAAkB,KAAkB,UAAgC;CACxE,MAAM,iBAAiB,CACrB,GAAG,IAAI,IACL,OAAO,KAAK,MAAM;EAChB,IAAI,OAAO,MAAM,UACf,OAAO;EAGT,MAAM,uBAAuB,IAAK,EAAU,CAAC,CAAC;EAC9C,IACE,EAAE,aAAa,yBACf,OAAO,qBAAqB,YAAY,YAExC,MAAM,IAAI,MAAM,wBAAwB;EAE1C,OAAO,qBAAqB,QAAQ;CACtC,CAAC,CACH,CACF;CACA,MAAM,UAAU,IAAI,QAAQ;CAC5B,OAAO,eAAe,MAAM,MAAM,MAAM,OAAO;AACjD;AA6EA,SAAgB,eACd,mBACA,SACwD;CACxD,IAAI,MAAM,QAAQ,iBAAiB,GACjC,OAAO,gBAAgB,mBAAmB,OAAO;CAEnD,OAAO,eAAe,MAAM,UAAwC;EAClE,OAAO,gBAAgB,OAAO,iBAAiB;CACjD,CAAC;AACH;AAEA,SAAS,gBACP,UACA,UAAgC,CAAC,GAClB;CACf,MAAM,EACJ,cACA,cACA,cACA,cACA,YACA,eACE;CAEJ,MAAM,WAA0B,CAAC;CAEjC,KAAK,MAAM,OAAO,UAAU;EAC1B,IAAI,gBAAgB,IAAI,QAAQ,aAAa,SAAS,IAAI,IAAI,GAC5D;OACK,IAAI,gBAAgB,eAAe,KAAK,YAAY,GACzD;OACK,IAAI,cAAc,IAAI,MAAM,WAAW,SAAS,IAAI,EAAE,GAC3D;EAIF,IAAI,EAAE,gBAAgB,cAAc,eAClC,SAAS,KAAK,GAAG;OACZ,IACL,gBACA,IAAI,QACJ,aAAa,MAAM,UAAU,UAAU,IAAI,IAAI,GAE/C,SAAS,KAAK,GAAG;OACZ,IAAI,gBAAgB,eAAe,KAAK,YAAY,GACzD,SAAS,KAAK,GAAG;OACZ,IAAI,cAAc,IAAI,MAAM,WAAW,MAAM,OAAO,OAAO,IAAI,EAAE,GACtE,SAAS,KAAK,GAAG;CAErB;CAEA,OAAO;AACT;AA8DA,SAAgB,iBACd,UACwD;CACxD,IAAI,MAAM,QAAQ,QAAQ,GACxB,OAAO,kBAAkB,QAAQ;CAEnC,OAAO,eAAe,KAAK,iBAAiB;AAC9C;AAEA,SAAS,kBAAkB,UAAwC;CACjE,IAAI,CAAC,SAAS,QACZ,OAAO,CAAC;CAEV,MAAM,SAAwB,CAAC;CAC/B,KAAK,MAAM,OAAO,UAAU;EAC1B,MAAM,OAAO;EACb,MAAM,OAAO,OAAO,IAAI;EACxB,IAAI,CAAC,MACH,OAAO,KAAK,IAAI;OACX,IACL,KAAK,QAAQ,MAAM,UACnB,EAAE,KAAK,QAAQ,MAAM,KAAK,QAAQ,IAElC,OAAO,KAAK,MAAM,IAAI;OACjB;GACL,MAAM,YAAY,eAAe,IAAI;GACrC,MAAM,YAAY,eAAe,IAAI;GACrC,MAAM,eAAe,UAAU,OAAO,SAAS;GAC/C,IACE,OAAO,UAAU,YAAY,YAC7B,OAAO,UAAU,YAAY,UAE7B,aAAa,UAAU,GAAG,UAAU,QAAQ,IAAI,UAAU;GAE5D,OAAO,KAAK,YAAY,YAAY,CAAC;EACvC;CACF;CACA,OAAO;AACT;AA8UA,SAAgB,aACd,mBACA,SACiE;CACjE,IAAI,MAAM,QAAQ,iBAAiB,GAAG;EACpC,MAAM,WAAW;EACjB,IAAI,CAAC,SACH,MAAM,IAAI,MAAM,wDAAwD;EAE1E,OAAO,oBAAoB,UAAU,OAAO;CAC9C,OAAO;EACL,MAAM,iBAAiB;EACvB,OAAO,eAAe,MAAM,UAC1B,oBAAoB,OAAO,cAAc,CAC3C,CAAC,CAAC,WAAW,EACX,SAAS,gBACX,CAAC;CACH;AACF;AAEA,eAAe,oBACb,UACA,SAC6B;CAC7B,MAAM,EACJ,WACA,cACA,WAAW,QACX,eAAe,OACf,OACA,SACA,gBAAgB,OAChB,iBACE;CACJ,IAAI,WAAW,aAAa,SAC1B,MAAM,IAAI,MACR,6DACF;CAEF,IAAI,iBAAiB,aAAa,SAChC,MAAM,IAAI,MACR,mEACF;CAGF,IAAI;CACJ,IAAI,kBAAkB,cACpB,mBAAmB,OAAO,SAAyC;EAIjE,QAAO,MAHmB,QAAQ,IAChC,KAAK,KAAK,QAAQ,aAAa,aAAa,IAAI,OAAO,CAAC,CAC1D,EAAA,CACmB,QAAQ,KAAK,UAAU,MAAM,OAAO,CAAC;CAC1D;MAEA,mBAAmB,OAAO,SACxB,aAAa,IAAI;CAGrB,IAAI,mBACF;CACF,IAAI,cACF,IAAI,eAAe,cACjB,mBAAmB,aAAa,UAAU,KAAK,YAAY;MAE3D,mBAAmB,OAAO,SACxB,aAAa,IAAI;CAIvB,IAAI,aAAa,SACf,OAAO,gBAAgB,UAAU;EAC/B;EACA,cAAc;EACd,cAAc;EACd,iBAAiB,eAAe,UAAU,KAAA;EAC1C;CACF,CAAC;MACI,IAAI,aAAa,QACtB,OAAO,eAAe,UAAU;EAC9B;EACA,cAAc;EACd,cAAc;EACd;EACA;EACA;EACA;CACF,CAAC;MAED,MAAM,IAAI,MACR,2BAA2B,SAAS,qCACtC;AAEJ;AAEA,eAAe,gBACb,UACA,SAOwB;CACxB,MAAM,EAAE,WAAW,cAAc,cAAc,iBAAiB,UAC9D;CACF,IAAI,eAAe,CAAC,GAAG,QAAQ;CAC/B,IAAI,MAAM;CACV,KAAK,IAAI,IAAI,GAAG,IAAI,aAAa,QAAQ,KAAK,GAE5C,IAAK,MAAM,aADe,IAAI,IAAI,aAAa,MAAM,GAAG,CAAC,CAAC,IAAI,YACrB,KAAM,WAAW;EACxD,MAAM,aAAa,SAAS;EAC5B;CACF;CAEF,IAAI,MAAM,aAAa,UAAU,iBAAiB;EAChD,IAAI,kBAAkB;EACtB,IAAI,MAAM,QAAQ,aAAa,IAAI,CAAC,OAAO,GAAG;GAC5C,MAAM,WAAW,aAAa;GAC9B,IAAI,OAAO,SAAS,YAAY,UAC9B,MAAM,IAAI,MAAM,kCAAkC;GAGpD,MAAM,WAAW,SAAS,QAAQ;GAClC,MAAM,kBACJ,oBAAoB,SAChB,CAAC,GAAG,SAAS,OAAO,CAAC,CAAC,QAAQ,IAC9B,SAAS;GACf,KAAK,IAAI,IAAI,GAAG,KAAK,UAAU,KAAK,GAAG;IACrC,MAAM,iBACJ,oBAAoB,UAChB,gBAAgB,MAAM,GAAG,CAAC,IAC1B,gBAAgB,MAAM,CAAC,CAAC;IAC9B,MAAM,SAAS,OAAO,YACpB,OAAO,QAAQ,QAAQ,CAAC,CAAC,QACtB,CAAC,OAAO,MAAM,UAAU,CAAC,EAAE,WAAW,KAAK,CAC9C,CACF;IACA,MAAM,iBAAiB,qBAAqB,SAAS,QAAQ,GAAG;KAC9D,GAAG;KACH,SAAS;IACX,CAAC;IACD,MAAM,iBAAiB,CAAC,GAAG,aAAa,MAAM,GAAG,GAAG,GAAG,cAAc;IACrE,IAAK,MAAM,aAAa,cAAc,KAAM,WAAW;KACrD,eAAe;KACf,OAAO;KACP,kBAAkB;IACpB,OACE;GAEJ;GACA,IAAI,mBAAmB,oBAAoB,QACzC,SAAS,UAAU,CAAC,GAAG,eAAe,CAAC,CAAC,QAAQ;EAEpD;EACA,IAAI,CAAC,iBAAiB;GACpB,MAAM,WAAW,aAAa;GAC9B,IAAI;GACJ,IACE,MAAM,QAAQ,SAAS,OAAO,KAC9B,SAAS,QAAQ,MACd,UAAU,OAAO,UAAU,YAAY,MAAM,SAAS,MACzD,GAKA,OAHkB,SAAS,QAAQ,MAChC,UAAU,MAAM,SAAS,UAAU,MAAM,IAE7B,CAAC,EAAE;QACb,IAAI,OAAO,SAAS,YAAY,UACrC,OAAO,SAAS;GAElB,IAAI,MAAM;IACR,MAAM,aAAa,MAAM,aAAa,IAAI;IAC1C,MAAM,YAAY,WAAW;IAC7B,IAAI,oBAAoB,QACtB,WAAW,QAAQ;IAErB,KAAK,IAAI,IAAI,GAAG,IAAI,YAAY,GAAG,KAAK,GAAG;KACzC,WAAW,IAAI;KACf,SAAS,UAAU,WAAW,KAAK,EAAE;KACrC,IACG,MAAM,aAAa,CAAC,GAAG,aAAa,MAAM,GAAG,GAAG,GAAG,QAAQ,CAAC,KAC7D,WACA;MACA,IAAI,oBAAoB,QACtB,SAAS,UAAU,CAAC,GAAG,UAAU,CAAC,CAAC,QAAQ,CAAC,CAAC,KAAK,EAAE;MAEtD,eAAe,CAAC,GAAG,aAAa,MAAM,GAAG,GAAG,GAAG,QAAQ;MACvD,OAAO;MACP;KACF;IACF;GACF;EACF;CACF;CAEA,IAAI,OAAO;EACT,MAAM,WAAW,MAAM,QAAQ,KAAK,IAAI,QAAQ,CAAC,KAAK;EACtD,OAAO,MAAM,KAAK,CAAC,eAAe,aAAa,MAAM,IAAI,QAAQ,GAC/D,OAAO;CAEX;CAEA,OAAO,aAAa,MAAM,GAAG,GAAG;AAClC;AAEA,eAAe,eACb,UACA,SAewB;CACxB,MAAM,EACJ,eAAe,OACf,gBAAgB,OAChB,OACA,SACA,GAAG,SACD;CAGJ,IAAI,eAAe,SAAS,KAAK,YAAY;EAC3C,MAAM,SAAS,OAAO,YACpB,OAAO,QAAQ,OAAO,CAAC,CAAC,QACrB,CAAC,OAAO,MAAM,UAAU,CAAC,EAAE,WAAW,KAAK,CAC9C,CACF;EACA,OAAO,qBACL,QAAQ,QAAQ,GAChB,QACA,mBAAmB,OAAO,CAC5B;CACF,CAAC;CAED,IAAI,OAAO;EACT,MAAM,WAAW,MAAM,QAAQ,KAAK,IAAI,QAAQ,CAAC,KAAK;EACtD,OACE,aAAa,SAAS,KACtB,CAAC,eAAe,aAAa,aAAa,SAAS,IAAI,QAAQ,GAE/D,eAAe,aAAa,MAAM,GAAG,EAAE;CAE3C;CAEA,MAAM,gBACJ,iBAAiB,aAAa,EAAE,EAAE,QAAQ,MAAM;CAClD,IAAI,YAAY,gBACZ,aAAa,MAAM,GAAG,CAAC,CAAC,CAAC,OAAO,aAAa,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,IAC/D,aAAa,QAAQ;CAEzB,YAAY,MAAM,gBAAgB,WAAW;EAC3C,GAAG;EACH,iBAAiB,eAAe,SAAS,KAAA;EACzC,OAAO;CACT,CAAC;CAED,IAAI,eACF,OAAO,CAAC,UAAU,IAAI,GAAG,UAAU,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC;MAErD,OAAO,UAAU,QAAQ;AAE7B;AAEA,MAAM,iBAMF;CACF,OAAO;EACL,SAAS;EACT,cAAc;CAChB;CACA,IAAI;EACF,SAAS;EACT,cAAc;CAChB;CACA,QAAQ;EACN,SAAS;EACT,cAAc;CAChB;CACA,WAAW;EACT,SAAS;EACT,cAAc;CAChB;CACA,MAAM;EACJ,SAAS;EACT,cAAc;CAChB;CACA,UAAU;EACR,SAAS;EACT,cAAc;CAChB;CACA,SAAS;EACP,SAAS;EACT,cAAc;CAChB;CACA,QAAQ;EACN,SAAS;EACT,cAAc;CAChB;AACF;AAgBA,SAAS,qBACP,aACA,QACA,aACgC;CAChC,IAAI;CACJ,IAAI;CAEJ,QAAQ,aAAR;EACE,KAAK;GACH,IAAI,aACF,QAAQ,IAAI,kBAAkB,MAAM;QAEpC,MAAM,IAAI,aAAa,MAAM;GAE/B;EACF,KAAK;GACH,IAAI,aAAa;IACf,IAAI,gBAAsC,EACxC,GAAG,OACL;IACA,IAAI,gBAAgB,eAClB,gBAAgB;KACd,GAAG;KACH,kBAAkB,cAAc,YAAY,KAAK,QAAQ;MACvD,GAAG;MACH,MAAM;MACN,OAAO,KAAA;MACP,MAAM,KAAK,UAAU,GAAG,IAAI;KAC9B,EAAE;IACJ;IAEF,QAAQ,IAAI,eAAe,aAAa;GAC1C,OACE,MAAM,IAAI,UAAU,MAAM;GAE5B;EACF,KAAK;GACH,IAAI,aACF,QAAQ,IAAI,mBAAmB,MAAM;QAErC,MAAM,IAAI,cAAc,MAAM;GAEhC;EACF,KAAK;GACH,IAAI,aACF,QAAQ,IAAI,mBAAmB;IAC7B,GAAG;IACH,mBAAmB;KACjB,GAAG,OAAO;KACV,iBAAiB;IACnB;GACF,CAAC;QAED,MAAM,IAAI,cAAc;IACtB,GAAG;IACH,mBAAmB;KACjB,GAAG,OAAO;KACV,iBAAiB;IACnB;GACF,CAAC;GAEH;EACF,KAAK;GACH,IAAI,kBAAkB,QACpB,IAAI,aACF,QAAQ,IAAI,iBAAiB,MAA2B;QAExD,MAAM,IAAI,YAAY,MAA2B;QAGnD,MAAM,IAAI,MACR,yFACF;GAEF;EACF,KAAK;GACH,IAAI,aACF,QAAQ,IAAI,qBAAqB,MAA+B;QAC3D;IACL,IAAI,CAAC,OAAO,MACV,MAAM,IAAI,MAAM,0CAA0C;IAE5D,MAAM,IAAI,gBAAgB,MAA+B;GAC3D;GACA;EACF,KAAK;GACH,IAAI,UAAU,QACZ,IAAI,aACF,QAAQ,IAAI,iBAAiB,MAA2B;QAExD,MAAM,IAAI,YAAY,MAA2B;QAGnD,MAAM,IAAI,MACR,iFACF;GAEF;EACF,SACE,MAAM,IAAI,MAAM,6BAA6B,aAAa;CAC9D;CAEA,IAAI,eAAe,OACjB,OAAO;CAET,IAAI,KACF,OAAO;CAET,MAAM,IAAI,MAAM,6BAA6B,aAAa;AAC5D;AAEA,SAAS,YAAY,OAAsC;CACzD,MAAM,YAAY,MAAM,QAAQ;CAChC,IAAI;CACJ,MAAM,SAAS,OAAO,YACpB,OAAO,QAAQ,KAAK,CAAC,CAAC,QACnB,CAAC,OAAO,CAAC,CAAC,QAAQ,kBAAkB,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,WAAW,KAAK,CAC3E,CACF;CAEA,IAAI,aAAa,gBACf,MAAM,qBAAqB,WAAW,MAAM;CAG9C,IAAI,CAAC,KACH,MAAM,IAAI,MACR,oCAAoC,UAAU,0BAA0B,OAAO,KAC7E,cACF,GACF;CAGF,OAAO;AACT;;;;;;;AAQA,SAAgB,oBAAoB,MAAiC;CACnE,MAAM,SAAS,KAAK,MAAM,IAAI;CAC9B,OAAO,QAAQ,QAAQ,CACrB,GAAG,OAAO,MAAM,GAAG,EAAE,CAAC,CAAC,KAAK,MAAM,GAAG,EAAE,GAAG,GAC1C,OAAO,OAAO,SAAS,EACzB,CAAC;AACH"}
1
+ {"version":3,"file":"transformers.js","names":[],"sources":["../../src/messages/transformers.ts"],"sourcesContent":["import { BaseLanguageModel } from \"../language_models/base.js\";\nimport { Runnable, RunnableLambda } from \"../runnables/base.js\";\nimport { AIMessage, AIMessageChunk, AIMessageChunkFields } from \"./ai.js\";\nimport {\n BaseMessage,\n BaseMessageChunk,\n BaseMessageFields,\n isBaseMessageChunk,\n} from \"./base.js\";\nimport { ChatMessage, ChatMessageChunk, ChatMessageFields } from \"./chat.js\";\nimport {\n FunctionMessage,\n FunctionMessageChunk,\n FunctionMessageFields,\n} from \"./function.js\";\nimport { HumanMessage, HumanMessageChunk } from \"./human.js\";\nimport { MessageType } from \"./message.js\";\nimport { RemoveMessage } from \"./modifier.js\";\nimport { SystemMessage, SystemMessageChunk } from \"./system.js\";\nimport { ToolMessage, ToolMessageChunk, ToolMessageFields } from \"./tool.js\";\nimport { convertToChunk } from \"./utils.js\";\n\nexport type MessageUnion =\n | typeof HumanMessage\n | typeof AIMessage\n | typeof SystemMessage\n | typeof ChatMessage\n | typeof FunctionMessage\n | typeof ToolMessage\n | typeof RemoveMessage;\nexport type MessageChunkUnion =\n | typeof HumanMessageChunk\n | typeof AIMessageChunk\n | typeof SystemMessageChunk\n | typeof FunctionMessageChunk\n | typeof ToolMessageChunk\n | typeof ChatMessageChunk\n | typeof RemoveMessage; // RemoveMessage does not have a chunk class.\nexport type MessageTypeOrClass = MessageType | MessageUnion | MessageChunkUnion;\n\nconst _isMessageType = (msg: BaseMessage, types: MessageTypeOrClass[]) => {\n const typesAsStrings = [\n ...new Set<string>(\n types?.map((t) => {\n if (typeof t === \"string\") {\n return t;\n }\n // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n const instantiatedMsgClass = new (t as any)({});\n if (\n !(\"getType\" in instantiatedMsgClass) ||\n typeof instantiatedMsgClass.getType !== \"function\"\n ) {\n throw new Error(\"Invalid type provided.\");\n }\n return instantiatedMsgClass.getType();\n })\n ),\n ];\n const msgType = msg.getType();\n return typesAsStrings.some((t) => t === msgType);\n};\n\nexport interface FilterMessagesFields {\n /**\n * @param {string[] | undefined} includeNames Message names to include.\n */\n includeNames?: string[];\n /**\n * @param {string[] | undefined} excludeNames Messages names to exclude.\n */\n excludeNames?: string[];\n /**\n * @param {(MessageType | BaseMessage)[] | undefined} includeTypes Message types to include. Can be specified as string names (e.g.\n * \"system\", \"human\", \"ai\", ...) or as BaseMessage classes (e.g.\n * SystemMessage, HumanMessage, AIMessage, ...).\n */\n includeTypes?: MessageTypeOrClass[];\n /**\n * @param {(MessageType | BaseMessage)[] | undefined} excludeTypes Message types to exclude. Can be specified as string names (e.g.\n * \"system\", \"human\", \"ai\", ...) or as BaseMessage classes (e.g.\n * SystemMessage, HumanMessage, AIMessage, ...).\n */\n excludeTypes?: MessageTypeOrClass[];\n /**\n * @param {string[] | undefined} includeIds Message IDs to include.\n */\n includeIds?: string[];\n /**\n * @param {string[] | undefined} excludeIds Message IDs to exclude.\n */\n excludeIds?: string[];\n}\n\n/**\n * Filter messages based on name, type or id.\n *\n * @param {BaseMessage[] | FilterMessagesFields} messagesOrOptions - Either an array of BaseMessage objects to filter or the filtering options. If an array is provided, the `options` parameter should also be supplied. If filtering options are provided, a RunnableLambda is returned.\n * @param {FilterMessagesFields} [options] - Optional filtering options. Should only be provided if `messagesOrOptions` is an array of BaseMessage objects.\n * @returns A list of Messages that meets at least one of the include conditions and none\n * of the exclude conditions, or a RunnableLambda which does the same. If no include conditions are specified then\n * anything that is not explicitly excluded will be included.\n * @throws {Error} If two incompatible arguments are provided.\n *\n * @example\n * ```typescript\n * import { filterMessages, AIMessage, HumanMessage, SystemMessage } from \"@langchain/core/messages\";\n *\n * const messages = [\n * new SystemMessage(\"you're a good assistant.\"),\n * new HumanMessage({ content: \"what's your name\", id: \"foo\", name: \"example_user\" }),\n * new AIMessage({ content: \"steve-o\", id: \"bar\", name: \"example_assistant\" }),\n * new HumanMessage({ content: \"what's your favorite color\", id: \"baz\" }),\n * new AIMessage({ content: \"silicon blue\" , id: \"blah\" }),\n * ];\n *\n * filterMessages(messages, {\n * includeNames: [\"example_user\", \"example_assistant\"],\n * includeTypes: [\"system\"],\n * excludeIds: [\"bar\"],\n * });\n * ```\n *\n * The above example would return:\n * ```typescript\n * [\n * new SystemMessage(\"you're a good assistant.\"),\n * new HumanMessage({ content: \"what's your name\", id: \"foo\", name: \"example_user\" }),\n * ]\n * ```\n */\nexport function filterMessages(\n options?: FilterMessagesFields\n): Runnable<BaseMessage[], BaseMessage[]>;\nexport function filterMessages(\n messages: BaseMessage[],\n options?: FilterMessagesFields\n): BaseMessage[];\nexport function filterMessages(\n messagesOrOptions?: BaseMessage[] | FilterMessagesFields,\n options?: FilterMessagesFields\n): BaseMessage[] | Runnable<BaseMessage[], BaseMessage[]> {\n if (Array.isArray(messagesOrOptions)) {\n return _filterMessages(messagesOrOptions, options);\n }\n return RunnableLambda.from((input: BaseMessage[]): BaseMessage[] => {\n return _filterMessages(input, messagesOrOptions);\n });\n}\n\nfunction _filterMessages(\n messages: BaseMessage[],\n options: FilterMessagesFields = {}\n): BaseMessage[] {\n const {\n includeNames,\n excludeNames,\n includeTypes,\n excludeTypes,\n includeIds,\n excludeIds,\n } = options;\n\n const filtered: BaseMessage[] = [];\n\n for (const msg of messages) {\n if (excludeNames && msg.name && excludeNames.includes(msg.name)) {\n continue;\n } else if (excludeTypes && _isMessageType(msg, excludeTypes)) {\n continue;\n } else if (excludeIds && msg.id && excludeIds.includes(msg.id)) {\n continue;\n }\n\n // default to inclusion when no inclusion criteria given.\n if (!(includeTypes || includeIds || includeNames)) {\n filtered.push(msg);\n } else if (\n includeNames &&\n msg.name &&\n includeNames.some((iName) => iName === msg.name)\n ) {\n filtered.push(msg);\n } else if (includeTypes && _isMessageType(msg, includeTypes)) {\n filtered.push(msg);\n } else if (includeIds && msg.id && includeIds.some((id) => id === msg.id)) {\n filtered.push(msg);\n }\n }\n\n return filtered;\n}\n\n/**\n * Merge consecutive Messages of the same type.\n *\n * **NOTE**: ToolMessages are not merged, as each has a distinct tool call id that\n * can't be merged.\n *\n * @param {BaseMessage[] | undefined} messages Sequence of Message-like objects to merge. Optional. If not provided, a RunnableLambda is returned.\n * @returns List of BaseMessages with consecutive runs of message types merged into single\n * messages, or a RunnableLambda which returns a list of BaseMessages If two messages being merged both have string contents, the merged\n * content is a concatenation of the two strings with a new-line separator. If at\n * least one of the messages has a list of content blocks, the merged content is a\n * list of content blocks.\n *\n * @example\n * ```typescript\n * import { mergeMessageRuns, AIMessage, HumanMessage, SystemMessage, ToolCall } from \"@langchain/core/messages\";\n *\n * const messages = [\n * new SystemMessage(\"you're a good assistant.\"),\n * new HumanMessage({ content: \"what's your favorite color\", id: \"foo\" }),\n * new HumanMessage({ content: \"wait your favorite food\", id: \"bar\" }),\n * new AIMessage({\n * content: \"my favorite colo\",\n * tool_calls: [{ name: \"blah_tool\", args: { x: 2 }, id: \"123\" }],\n * id: \"baz\",\n * }),\n * new AIMessage({\n * content: [{ type: \"text\", text: \"my favorite dish is lasagna\" }],\n * tool_calls: [{ name: \"blah_tool\", args: { x: -10 }, id: \"456\" }],\n * id: \"blur\",\n * }),\n * ];\n *\n * mergeMessageRuns(messages);\n * ```\n *\n * The above example would return:\n * ```typescript\n * [\n * new SystemMessage(\"you're a good assistant.\"),\n * new HumanMessage({\n * content: \"what's your favorite colorwait your favorite food\",\n * id: \"foo\",\n * }),\n * new AIMessage({\n * content: [\n * { type: \"text\", text: \"my favorite colo\" },\n * { type: \"text\", text: \"my favorite dish is lasagna\" },\n * ],\n * tool_calls: [\n * { name: \"blah_tool\", args: { x: 2 }, id: \"123\" },\n * { name: \"blah_tool\", args: { x: -10 }, id: \"456\" },\n * ],\n * id: \"baz\",\n * }),\n * ]\n * ```\n */\nexport function mergeMessageRuns(): Runnable<BaseMessage[], BaseMessage[]>;\nexport function mergeMessageRuns(messages: BaseMessage[]): BaseMessage[];\nexport function mergeMessageRuns(\n messages?: BaseMessage[]\n): BaseMessage[] | Runnable<BaseMessage[], BaseMessage[]> {\n if (Array.isArray(messages)) {\n return _mergeMessageRuns(messages);\n }\n return RunnableLambda.from(_mergeMessageRuns);\n}\n\nfunction _mergeMessageRuns(messages: BaseMessage[]): BaseMessage[] {\n if (!messages.length) {\n return [];\n }\n const merged: BaseMessage[] = [];\n for (const msg of messages) {\n const curr = msg;\n const last = merged.pop();\n if (!last) {\n merged.push(curr);\n } else if (\n curr.getType() === \"tool\" ||\n !(curr.getType() === last.getType())\n ) {\n merged.push(last, curr);\n } else {\n const lastChunk = convertToChunk(last) as BaseMessageChunk;\n const currChunk = convertToChunk(curr) as BaseMessageChunk;\n const mergedChunks = lastChunk.concat(currChunk);\n if (\n typeof lastChunk.content === \"string\" &&\n typeof currChunk.content === \"string\"\n ) {\n mergedChunks.content = `${lastChunk.content}\\n${currChunk.content}`;\n }\n merged.push(_chunkToMsg(mergedChunks));\n }\n }\n return merged;\n}\n\n// Since we can not import from `@langchain/textsplitters` we need\n// to reconstruct the interface here. Only `splitText` is used by\n// `trimMessages`, so we avoid coupling to the full `BaseDocumentTransformer`\n// surface and just require the method we call.\ninterface _TextSplitterInterface {\n splitText(text: string): Promise<string[]>;\n}\n\nexport interface TrimMessagesFields {\n /**\n * @param {number} maxTokens Max token count of trimmed messages.\n */\n maxTokens: number;\n /**\n * @param {((messages: BaseMessage[]) => number) | ((messages: BaseMessage[]) => Promise<number>) | BaseLanguageModel} tokenCounter\n * Function or LLM for counting tokens in an array of `BaseMessage`s.\n * If a `BaseLanguageModel` is passed in then `BaseLanguageModel.getNumTokens()` will be used.\n */\n tokenCounter:\n | ((messages: BaseMessage[]) => number)\n | ((messages: BaseMessage[]) => Promise<number>)\n | BaseLanguageModel;\n /**\n * @param {\"first\" | \"last\"} [strategy=\"last\"] Strategy for trimming.\n * - \"first\": Keep the first <= n_count tokens of the messages.\n * - \"last\": Keep the last <= n_count tokens of the messages.\n * @default \"last\"\n */\n strategy?: \"first\" | \"last\";\n /**\n * @param {boolean} [allowPartial=false] Whether to split a message if only part of the message can be included.\n * If `strategy: \"last\"` then the last partial contents of a message are included.\n * If `strategy: \"first\"` then the first partial contents of a message are included.\n * @default false\n */\n allowPartial?: boolean;\n /**\n * @param {MessageTypeOrClass | MessageTypeOrClass[]} [endOn] The message type to end on.\n * If specified then every message after the last occurrence of this type is ignored.\n * If `strategy === \"last\"` then this is done before we attempt to get the last `maxTokens`.\n * If `strategy === \"first\"` then this is done after we get the first `maxTokens`.\n * Can be specified as string names (e.g. \"system\", \"human\", \"ai\", ...) or as `BaseMessage` classes\n * (e.g. `SystemMessage`, `HumanMessage`, `AIMessage`, ...). Can be a single type or an array of types.\n */\n endOn?: MessageTypeOrClass | MessageTypeOrClass[];\n /**\n * @param {MessageTypeOrClass | MessageTypeOrClass[]} [startOn] The message type to start on.\n * Should only be specified if `strategy: \"last\"`. If specified then every message before the first occurrence\n * of this type is ignored. This is done after we trim the initial messages to the last `maxTokens`.\n * Does not apply to a `SystemMessage` at index 0 if `includeSystem: true`.\n * Can be specified as string names (e.g. \"system\", \"human\", \"ai\", ...) or as `BaseMessage` classes\n * (e.g. `SystemMessage`, `HumanMessage`, `AIMessage`, ...). Can be a single type or an array of types.\n */\n startOn?: MessageTypeOrClass | MessageTypeOrClass[];\n /**\n * @param {boolean} [includeSystem=false] Whether to keep the `SystemMessage` if there is one at index 0.\n * Should only be specified if `strategy: \"last\"`.\n * @default false\n */\n includeSystem?: boolean;\n /**\n * @param {((text: string) => string[]) | BaseDocumentTransformer} [textSplitter] Function or `BaseDocumentTransformer` for\n * splitting the string contents of a message. Only used if `allowPartial: true`.\n * If `strategy: \"last\"` then the last split tokens from a partial message will be included.\n * If `strategy: \"first\"` then the first split tokens from a partial message will be included.\n * Token splitter assumes that separators are kept, so that split contents can be directly concatenated\n * to recreate the original text. Defaults to splitting on newlines.\n */\n textSplitter?:\n | ((text: string) => string[])\n | ((text: string) => Promise<string[]>)\n | _TextSplitterInterface;\n}\n\n/**\n * Trim messages to be below a token count.\n *\n * @param {BaseMessage[]} messages Array of `BaseMessage` instances to trim.\n * @param {TrimMessagesFields} options Trimming options.\n * @returns An array of trimmed `BaseMessage`s or a `Runnable` that takes a sequence of `BaseMessage`-like objects and returns\n * an array of trimmed `BaseMessage`s.\n * @throws {Error} If two incompatible arguments are specified or an unrecognized `strategy` is specified.\n *\n * @example\n * ```typescript\n * import { trimMessages, AIMessage, BaseMessage, HumanMessage, SystemMessage } from \"@langchain/core/messages\";\n *\n * const messages = [\n * new SystemMessage(\"This is a 4 token text. The full message is 10 tokens.\"),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"first\",\n * }),\n * new AIMessage({\n * content: [\n * { type: \"text\", text: \"This is the FIRST 4 token block.\" },\n * { type: \"text\", text: \"This is the SECOND 4 token block.\" },\n * ],\n * id: \"second\",\n * }),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"third\",\n * }),\n * new AIMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"fourth\",\n * }),\n * ];\n *\n * function dummyTokenCounter(messages: BaseMessage[]): number {\n * // treat each message like it adds 3 default tokens at the beginning\n * // of the message and at the end of the message. 3 + 4 + 3 = 10 tokens\n * // per message.\n *\n * const defaultContentLen = 4;\n * const defaultMsgPrefixLen = 3;\n * const defaultMsgSuffixLen = 3;\n *\n * let count = 0;\n * for (const msg of messages) {\n * if (typeof msg.content === \"string\") {\n * count += defaultMsgPrefixLen + defaultContentLen + defaultMsgSuffixLen;\n * }\n * if (Array.isArray(msg.content)) {\n * count +=\n * defaultMsgPrefixLen +\n * msg.content.length * defaultContentLen +\n * defaultMsgSuffixLen;\n * }\n * }\n * return count;\n * }\n * ```\n *\n * First 30 tokens, not allowing partial messages:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 30,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"first\",\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"first\",\n * }),\n * ]\n * ```\n *\n * First 30 tokens, allowing partial messages:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 30,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"first\",\n * allowPartial: true,\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"first\",\n * }),\n * new AIMessage({\n * content: [{ type: \"text\", text: \"This is the FIRST 4 token block.\" }],\n * id: \"second\",\n * }),\n * ]\n * ```\n *\n * First 30 tokens, allowing partial messages, have to end on HumanMessage:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 30,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"first\",\n * allowPartial: true,\n * endOn: \"human\",\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"first\",\n * }),\n * ]\n * ```\n *\n * Last 30 tokens, including system message, not allowing partial messages:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 30,\n * includeSystem: true,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"last\",\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"third\",\n * }),\n * new AIMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"fourth\",\n * }),\n * ]\n * ```\n *\n * Last 40 tokens, including system message, allowing partial messages:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 40,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"last\",\n * allowPartial: true,\n * includeSystem: true,\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new AIMessage({\n * content: [{ type: \"text\", text: \"This is the FIRST 4 token block.\" }],\n * id: \"second\",\n * }),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"third\",\n * }),\n * new AIMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"fourth\",\n * }),\n * ]\n * ```\n *\n * Last 30 tokens, including system message, allowing partial messages, end on HumanMessage:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 30,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"last\",\n * endOn: \"human\",\n * includeSystem: true,\n * allowPartial: true,\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new AIMessage({\n * content: [{ type: \"text\", text: \"This is the FIRST 4 token block.\" }],\n * id: \"second\",\n * }),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"third\",\n * }),\n * ]\n * ```\n *\n * Last 40 tokens, including system message, allowing partial messages, start on HumanMessage:\n * ```typescript\n * await trimMessages(messages, {\n * maxTokens: 40,\n * tokenCounter: dummyTokenCounter,\n * strategy: \"last\",\n * includeSystem: true,\n * allowPartial: true,\n * startOn: \"human\",\n * });\n * ```\n *\n * Output:\n * ```typescript\n * [\n * new SystemMessage(\n * \"This is a 4 token text. The full message is 10 tokens.\"\n * ),\n * new HumanMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"third\",\n * }),\n * new AIMessage({\n * content: \"This is a 4 token text. The full message is 10 tokens.\",\n * id: \"fourth\",\n * }),\n * ]\n * ```\n */\nexport function trimMessages(\n options: TrimMessagesFields\n): Runnable<BaseMessage[], BaseMessage[]>;\nexport function trimMessages(\n messages: BaseMessage[],\n options: TrimMessagesFields\n): Promise<BaseMessage[]>;\nexport function trimMessages(\n messagesOrOptions: BaseMessage[] | TrimMessagesFields,\n options?: TrimMessagesFields\n): Promise<BaseMessage[]> | Runnable<BaseMessage[], BaseMessage[]> {\n if (Array.isArray(messagesOrOptions)) {\n const messages = messagesOrOptions;\n if (!options) {\n throw new Error(\"Options parameter is required when providing messages.\");\n }\n return _trimMessagesHelper(messages, options);\n } else {\n const trimmerOptions = messagesOrOptions;\n return RunnableLambda.from((input: BaseMessage[]) =>\n _trimMessagesHelper(input, trimmerOptions)\n ).withConfig({\n runName: \"trim_messages\",\n });\n }\n}\n\nasync function _trimMessagesHelper(\n messages: BaseMessage[],\n options: TrimMessagesFields\n): Promise<Array<BaseMessage>> {\n const {\n maxTokens,\n tokenCounter,\n strategy = \"last\",\n allowPartial = false,\n endOn,\n startOn,\n includeSystem = false,\n textSplitter,\n } = options;\n if (startOn && strategy === \"first\") {\n throw new Error(\n \"`startOn` should only be specified if `strategy` is 'last'.\"\n );\n }\n if (includeSystem && strategy === \"first\") {\n throw new Error(\n \"`includeSystem` should only be specified if `strategy` is 'last'.\"\n );\n }\n\n let listTokenCounter: (msgs: BaseMessage[]) => Promise<number>;\n if (\"getNumTokens\" in tokenCounter) {\n listTokenCounter = async (msgs: BaseMessage[]): Promise<number> => {\n const tokenCounts = await Promise.all(\n msgs.map((msg) => tokenCounter.getNumTokens(msg.content))\n );\n return tokenCounts.reduce((sum, count) => sum + count, 0);\n };\n } else {\n listTokenCounter = async (msgs: BaseMessage[]): Promise<number> =>\n tokenCounter(msgs);\n }\n\n let textSplitterFunc: (text: string) => Promise<string[]> =\n defaultTextSplitter;\n if (textSplitter) {\n if (\"splitText\" in textSplitter) {\n textSplitterFunc = textSplitter.splitText.bind(textSplitter);\n } else {\n textSplitterFunc = async (text: string): Promise<string[]> =>\n textSplitter(text);\n }\n }\n\n if (strategy === \"first\") {\n return _firstMaxTokens(messages, {\n maxTokens,\n tokenCounter: listTokenCounter,\n textSplitter: textSplitterFunc,\n partialStrategy: allowPartial ? \"first\" : undefined,\n endOn,\n });\n } else if (strategy === \"last\") {\n return _lastMaxTokens(messages, {\n maxTokens,\n tokenCounter: listTokenCounter,\n textSplitter: textSplitterFunc,\n allowPartial,\n includeSystem,\n startOn,\n endOn,\n });\n } else {\n throw new Error(\n `Unrecognized strategy: '${strategy}'. Must be one of 'first' or 'last'.`\n );\n }\n}\n\nasync function _firstMaxTokens(\n messages: BaseMessage[],\n options: {\n maxTokens: number;\n tokenCounter: (messages: BaseMessage[]) => Promise<number>;\n textSplitter: (text: string) => Promise<string[]>;\n partialStrategy?: \"first\" | \"last\";\n endOn?: MessageTypeOrClass | MessageTypeOrClass[];\n }\n): Promise<BaseMessage[]> {\n const { maxTokens, tokenCounter, textSplitter, partialStrategy, endOn } =\n options;\n let messagesCopy = [...messages];\n let idx = 0;\n for (let i = 0; i < messagesCopy.length; i += 1) {\n const remainingMessages = i > 0 ? messagesCopy.slice(0, -i) : messagesCopy;\n if ((await tokenCounter(remainingMessages)) <= maxTokens) {\n idx = messagesCopy.length - i;\n break;\n }\n }\n if (idx < messagesCopy.length && partialStrategy) {\n let includedPartial = false;\n if (Array.isArray(messagesCopy[idx].content)) {\n const excluded = messagesCopy[idx];\n if (typeof excluded.content === \"string\") {\n throw new Error(\"Expected content to be an array.\");\n }\n\n const numBlock = excluded.content.length;\n const reversedContent =\n partialStrategy === \"last\"\n ? [...excluded.content].reverse()\n : excluded.content;\n for (let i = 1; i <= numBlock; i += 1) {\n const partialContent =\n partialStrategy === \"first\"\n ? reversedContent.slice(0, i)\n : reversedContent.slice(-i);\n const fields = Object.fromEntries(\n Object.entries(excluded).filter(\n ([k]) => k !== \"type\" && !k.startsWith(\"lc_\")\n )\n ) as BaseMessageFields;\n const updatedMessage = _switchTypeToMessage(excluded.getType(), {\n ...fields,\n content: partialContent,\n });\n const slicedMessages = [...messagesCopy.slice(0, idx), updatedMessage];\n if ((await tokenCounter(slicedMessages)) <= maxTokens) {\n messagesCopy = slicedMessages;\n idx += 1;\n includedPartial = true;\n } else {\n break;\n }\n }\n if (includedPartial && partialStrategy === \"last\") {\n excluded.content = [...reversedContent].reverse();\n }\n }\n if (!includedPartial) {\n const excluded = messagesCopy[idx];\n let text: string | undefined;\n if (\n Array.isArray(excluded.content) &&\n excluded.content.some(\n (block) => typeof block === \"string\" || block.type === \"text\"\n )\n ) {\n const textBlock = excluded.content.find(\n (block) => block.type === \"text\" && block.text\n ) as { type: \"text\"; text: string } | undefined;\n text = textBlock?.text;\n } else if (typeof excluded.content === \"string\") {\n text = excluded.content;\n }\n if (text) {\n const splitTexts = await textSplitter(text);\n const numSplits = splitTexts.length;\n if (partialStrategy === \"last\") {\n splitTexts.reverse();\n }\n for (let _ = 0; _ < numSplits - 1; _ += 1) {\n splitTexts.pop();\n excluded.content = splitTexts.join(\"\");\n if (\n (await tokenCounter([...messagesCopy.slice(0, idx), excluded])) <=\n maxTokens\n ) {\n if (partialStrategy === \"last\") {\n excluded.content = [...splitTexts].reverse().join(\"\");\n }\n messagesCopy = [...messagesCopy.slice(0, idx), excluded];\n idx += 1;\n break;\n }\n }\n }\n }\n }\n\n if (endOn) {\n const endOnArr = Array.isArray(endOn) ? endOn : [endOn];\n while (idx > 0 && !_isMessageType(messagesCopy[idx - 1], endOnArr)) {\n idx -= 1;\n }\n }\n\n return messagesCopy.slice(0, idx);\n}\n\nasync function _lastMaxTokens(\n messages: BaseMessage[],\n options: {\n maxTokens: number;\n tokenCounter: (messages: BaseMessage[]) => Promise<number>;\n textSplitter: (text: string) => Promise<string[]>;\n /**\n * @default {false}\n */\n allowPartial?: boolean;\n /**\n * @default {false}\n */\n includeSystem?: boolean;\n startOn?: MessageTypeOrClass | MessageTypeOrClass[];\n endOn?: MessageTypeOrClass | MessageTypeOrClass[];\n }\n): Promise<BaseMessage[]> {\n const {\n allowPartial = false,\n includeSystem = false,\n endOn,\n startOn,\n ...rest\n } = options;\n\n // Create a copy of messages to avoid mutation\n let messagesCopy = messages.map((message) => {\n const fields = Object.fromEntries(\n Object.entries(message).filter(\n ([k]) => k !== \"type\" && !k.startsWith(\"lc_\")\n )\n ) as BaseMessageFields;\n return _switchTypeToMessage(\n message.getType(),\n fields,\n isBaseMessageChunk(message)\n );\n });\n\n if (endOn) {\n const endOnArr = Array.isArray(endOn) ? endOn : [endOn];\n while (\n messagesCopy.length > 0 &&\n !_isMessageType(messagesCopy[messagesCopy.length - 1], endOnArr)\n ) {\n messagesCopy = messagesCopy.slice(0, -1);\n }\n }\n\n const swappedSystem =\n includeSystem && messagesCopy[0]?.getType() === \"system\";\n let reversed_ = swappedSystem\n ? messagesCopy.slice(0, 1).concat(messagesCopy.slice(1).reverse())\n : messagesCopy.reverse();\n\n reversed_ = await _firstMaxTokens(reversed_, {\n ...rest,\n partialStrategy: allowPartial ? \"last\" : undefined,\n endOn: startOn,\n });\n\n if (swappedSystem) {\n return [reversed_[0], ...reversed_.slice(1).reverse()];\n } else {\n return reversed_.reverse();\n }\n}\n\nconst _MSG_CHUNK_MAP: Record<\n MessageType,\n {\n message: MessageUnion;\n messageChunk: MessageChunkUnion;\n }\n> = {\n human: {\n message: HumanMessage,\n messageChunk: HumanMessageChunk,\n },\n ai: {\n message: AIMessage,\n messageChunk: AIMessageChunk,\n },\n system: {\n message: SystemMessage,\n messageChunk: SystemMessageChunk,\n },\n developer: {\n message: SystemMessage,\n messageChunk: SystemMessageChunk,\n },\n tool: {\n message: ToolMessage,\n messageChunk: ToolMessageChunk,\n },\n function: {\n message: FunctionMessage,\n messageChunk: FunctionMessageChunk,\n },\n generic: {\n message: ChatMessage,\n messageChunk: ChatMessageChunk,\n },\n remove: {\n message: RemoveMessage,\n messageChunk: RemoveMessage, // RemoveMessage does not have a chunk class.\n },\n};\n\nfunction _switchTypeToMessage(\n messageType: MessageType,\n fields: BaseMessageFields\n): BaseMessage;\nfunction _switchTypeToMessage(\n messageType: MessageType,\n fields: BaseMessageFields,\n returnChunk: true\n): BaseMessageChunk;\nfunction _switchTypeToMessage(\n messageType: MessageType,\n fields: BaseMessageFields,\n returnChunk?: boolean\n): BaseMessageChunk | BaseMessage;\nfunction _switchTypeToMessage(\n messageType: MessageType,\n fields: BaseMessageFields,\n returnChunk?: boolean\n): BaseMessageChunk | BaseMessage {\n let chunk: BaseMessageChunk | undefined;\n let msg: BaseMessage | undefined;\n\n switch (messageType) {\n case \"human\":\n if (returnChunk) {\n chunk = new HumanMessageChunk(fields);\n } else {\n msg = new HumanMessage(fields);\n }\n break;\n case \"ai\":\n if (returnChunk) {\n let aiChunkFields: AIMessageChunkFields = {\n ...fields,\n };\n if (\"tool_calls\" in aiChunkFields) {\n aiChunkFields = {\n ...aiChunkFields,\n tool_call_chunks: aiChunkFields.tool_calls?.map((tc) => ({\n ...tc,\n type: \"tool_call_chunk\",\n index: undefined,\n args: JSON.stringify(tc.args),\n })),\n };\n }\n chunk = new AIMessageChunk(aiChunkFields);\n } else {\n msg = new AIMessage(fields);\n }\n break;\n case \"system\":\n if (returnChunk) {\n chunk = new SystemMessageChunk(fields);\n } else {\n msg = new SystemMessage(fields);\n }\n break;\n case \"developer\":\n if (returnChunk) {\n chunk = new SystemMessageChunk({\n ...fields,\n additional_kwargs: {\n ...fields.additional_kwargs,\n __openai_role__: \"developer\",\n },\n });\n } else {\n msg = new SystemMessage({\n ...fields,\n additional_kwargs: {\n ...fields.additional_kwargs,\n __openai_role__: \"developer\",\n },\n });\n }\n break;\n case \"tool\":\n if (\"tool_call_id\" in fields) {\n if (returnChunk) {\n chunk = new ToolMessageChunk(fields as ToolMessageFields);\n } else {\n msg = new ToolMessage(fields as ToolMessageFields);\n }\n } else {\n throw new Error(\n \"Can not convert ToolMessage to ToolMessageChunk if 'tool_call_id' field is not defined.\"\n );\n }\n break;\n case \"function\":\n if (returnChunk) {\n chunk = new FunctionMessageChunk(fields as FunctionMessageFields);\n } else {\n if (!fields.name) {\n throw new Error(\"FunctionMessage must have a 'name' field\");\n }\n msg = new FunctionMessage(fields as FunctionMessageFields);\n }\n break;\n case \"generic\":\n if (\"role\" in fields) {\n if (returnChunk) {\n chunk = new ChatMessageChunk(fields as ChatMessageFields);\n } else {\n msg = new ChatMessage(fields as ChatMessageFields);\n }\n } else {\n throw new Error(\n \"Can not convert ChatMessage to ChatMessageChunk if 'role' field is not defined.\"\n );\n }\n break;\n default:\n throw new Error(`Unrecognized message type ${messageType}`);\n }\n\n if (returnChunk && chunk) {\n return chunk;\n }\n if (msg) {\n return msg;\n }\n throw new Error(`Unrecognized message type ${messageType}`);\n}\n\nfunction _chunkToMsg(chunk: BaseMessageChunk): BaseMessage {\n const chunkType = chunk.getType();\n let msg: BaseMessage | undefined;\n const fields = Object.fromEntries(\n Object.entries(chunk).filter(\n ([k]) => ![\"type\", \"tool_call_chunks\"].includes(k) && !k.startsWith(\"lc_\")\n )\n ) as BaseMessageFields;\n\n if (chunkType in _MSG_CHUNK_MAP) {\n msg = _switchTypeToMessage(chunkType, fields);\n }\n\n if (!msg) {\n throw new Error(\n `Unrecognized message chunk class ${chunkType}. Supported classes are ${Object.keys(\n _MSG_CHUNK_MAP\n )}`\n );\n }\n\n return msg;\n}\n\n/**\n * The default text splitter function that splits text by newlines.\n *\n * @param {string} text\n * @returns A promise that resolves to an array of strings split by newlines.\n */\nexport function defaultTextSplitter(text: string): Promise<string[]> {\n const splits = text.split(\"\\n\");\n return Promise.resolve([\n ...splits.slice(0, -1).map((s) => `${s}\\n`),\n splits[splits.length - 1],\n ]);\n}\n"],"mappings":";;;;;;;;;;;AAwCA,MAAM,kBAAkB,KAAkB,UAAgC;CACxE,MAAM,iBAAiB,CACrB,GAAG,IAAI,IACL,OAAO,KAAK,MAAM;EAChB,IAAI,OAAO,MAAM,UACf,OAAO;EAGT,MAAM,uBAAuB,IAAK,EAAU,CAAC,CAAC;EAC9C,IACE,EAAE,aAAa,yBACf,OAAO,qBAAqB,YAAY,YAExC,MAAM,IAAI,MAAM,wBAAwB;EAE1C,OAAO,qBAAqB,QAAQ;CACtC,CAAC,CACH,CACF;CACA,MAAM,UAAU,IAAI,QAAQ;CAC5B,OAAO,eAAe,MAAM,MAAM,MAAM,OAAO;AACjD;AA6EA,SAAgB,eACd,mBACA,SACwD;CACxD,IAAI,MAAM,QAAQ,iBAAiB,GACjC,OAAO,gBAAgB,mBAAmB,OAAO;CAEnD,OAAO,eAAe,MAAM,UAAwC;EAClE,OAAO,gBAAgB,OAAO,iBAAiB;CACjD,CAAC;AACH;AAEA,SAAS,gBACP,UACA,UAAgC,CAAC,GAClB;CACf,MAAM,EACJ,cACA,cACA,cACA,cACA,YACA,eACE;CAEJ,MAAM,WAA0B,CAAC;CAEjC,KAAK,MAAM,OAAO,UAAU;EAC1B,IAAI,gBAAgB,IAAI,QAAQ,aAAa,SAAS,IAAI,IAAI,GAC5D;OACK,IAAI,gBAAgB,eAAe,KAAK,YAAY,GACzD;OACK,IAAI,cAAc,IAAI,MAAM,WAAW,SAAS,IAAI,EAAE,GAC3D;EAIF,IAAI,EAAE,gBAAgB,cAAc,eAClC,SAAS,KAAK,GAAG;OACZ,IACL,gBACA,IAAI,QACJ,aAAa,MAAM,UAAU,UAAU,IAAI,IAAI,GAE/C,SAAS,KAAK,GAAG;OACZ,IAAI,gBAAgB,eAAe,KAAK,YAAY,GACzD,SAAS,KAAK,GAAG;OACZ,IAAI,cAAc,IAAI,MAAM,WAAW,MAAM,OAAO,OAAO,IAAI,EAAE,GACtE,SAAS,KAAK,GAAG;CAErB;CAEA,OAAO;AACT;AA8DA,SAAgB,iBACd,UACwD;CACxD,IAAI,MAAM,QAAQ,QAAQ,GACxB,OAAO,kBAAkB,QAAQ;CAEnC,OAAO,eAAe,KAAK,iBAAiB;AAC9C;AAEA,SAAS,kBAAkB,UAAwC;CACjE,IAAI,CAAC,SAAS,QACZ,OAAO,CAAC;CAEV,MAAM,SAAwB,CAAC;CAC/B,KAAK,MAAM,OAAO,UAAU;EAC1B,MAAM,OAAO;EACb,MAAM,OAAO,OAAO,IAAI;EACxB,IAAI,CAAC,MACH,OAAO,KAAK,IAAI;OACX,IACL,KAAK,QAAQ,MAAM,UACnB,EAAE,KAAK,QAAQ,MAAM,KAAK,QAAQ,IAElC,OAAO,KAAK,MAAM,IAAI;OACjB;GACL,MAAM,YAAY,eAAe,IAAI;GACrC,MAAM,YAAY,eAAe,IAAI;GACrC,MAAM,eAAe,UAAU,OAAO,SAAS;GAC/C,IACE,OAAO,UAAU,YAAY,YAC7B,OAAO,UAAU,YAAY,UAE7B,aAAa,UAAU,GAAG,UAAU,QAAQ,IAAI,UAAU;GAE5D,OAAO,KAAK,YAAY,YAAY,CAAC;EACvC;CACF;CACA,OAAO;AACT;AA8UA,SAAgB,aACd,mBACA,SACiE;CACjE,IAAI,MAAM,QAAQ,iBAAiB,GAAG;EACpC,MAAM,WAAW;EACjB,IAAI,CAAC,SACH,MAAM,IAAI,MAAM,wDAAwD;EAE1E,OAAO,oBAAoB,UAAU,OAAO;CAC9C,OAAO;EACL,MAAM,iBAAiB;EACvB,OAAO,eAAe,MAAM,UAC1B,oBAAoB,OAAO,cAAc,CAC3C,CAAC,CAAC,WAAW,EACX,SAAS,gBACX,CAAC;CACH;AACF;AAEA,eAAe,oBACb,UACA,SAC6B;CAC7B,MAAM,EACJ,WACA,cACA,WAAW,QACX,eAAe,OACf,OACA,SACA,gBAAgB,OAChB,iBACE;CACJ,IAAI,WAAW,aAAa,SAC1B,MAAM,IAAI,MACR,6DACF;CAEF,IAAI,iBAAiB,aAAa,SAChC,MAAM,IAAI,MACR,mEACF;CAGF,IAAI;CACJ,IAAI,kBAAkB,cACpB,mBAAmB,OAAO,SAAyC;EAIjE,QAAO,MAHmB,QAAQ,IAChC,KAAK,KAAK,QAAQ,aAAa,aAAa,IAAI,OAAO,CAAC,CAC1D,EAAA,CACmB,QAAQ,KAAK,UAAU,MAAM,OAAO,CAAC;CAC1D;MAEA,mBAAmB,OAAO,SACxB,aAAa,IAAI;CAGrB,IAAI,mBACF;CACF,IAAI,cAAc;EAChB,IAAI,eAAe,cACjB,mBAAmB,aAAa,UAAU,KAAK,YAAY;OAE3D,mBAAmB,OAAO,SACxB,aAAa,IAAI;CAEvB;CAEA,IAAI,aAAa,SACf,OAAO,gBAAgB,UAAU;EAC/B;EACA,cAAc;EACd,cAAc;EACd,iBAAiB,eAAe,UAAU,KAAA;EAC1C;CACF,CAAC;MACI,IAAI,aAAa,QACtB,OAAO,eAAe,UAAU;EAC9B;EACA,cAAc;EACd,cAAc;EACd;EACA;EACA;EACA;CACF,CAAC;MAED,MAAM,IAAI,MACR,2BAA2B,SAAS,qCACtC;AAEJ;AAEA,eAAe,gBACb,UACA,SAOwB;CACxB,MAAM,EAAE,WAAW,cAAc,cAAc,iBAAiB,UAC9D;CACF,IAAI,eAAe,CAAC,GAAG,QAAQ;CAC/B,IAAI,MAAM;CACV,KAAK,IAAI,IAAI,GAAG,IAAI,aAAa,QAAQ,KAAK,GAE5C,IAAK,MAAM,aADe,IAAI,IAAI,aAAa,MAAM,GAAG,CAAC,CAAC,IAAI,YACrB,KAAM,WAAW;EACxD,MAAM,aAAa,SAAS;EAC5B;CACF;CAEF,IAAI,MAAM,aAAa,UAAU,iBAAiB;EAChD,IAAI,kBAAkB;EACtB,IAAI,MAAM,QAAQ,aAAa,IAAI,CAAC,OAAO,GAAG;GAC5C,MAAM,WAAW,aAAa;GAC9B,IAAI,OAAO,SAAS,YAAY,UAC9B,MAAM,IAAI,MAAM,kCAAkC;GAGpD,MAAM,WAAW,SAAS,QAAQ;GAClC,MAAM,kBACJ,oBAAoB,SAChB,CAAC,GAAG,SAAS,OAAO,CAAC,CAAC,QAAQ,IAC9B,SAAS;GACf,KAAK,IAAI,IAAI,GAAG,KAAK,UAAU,KAAK,GAAG;IACrC,MAAM,iBACJ,oBAAoB,UAChB,gBAAgB,MAAM,GAAG,CAAC,IAC1B,gBAAgB,MAAM,CAAC,CAAC;IAC9B,MAAM,SAAS,OAAO,YACpB,OAAO,QAAQ,QAAQ,CAAC,CAAC,QACtB,CAAC,OAAO,MAAM,UAAU,CAAC,EAAE,WAAW,KAAK,CAC9C,CACF;IACA,MAAM,iBAAiB,qBAAqB,SAAS,QAAQ,GAAG;KAC9D,GAAG;KACH,SAAS;IACX,CAAC;IACD,MAAM,iBAAiB,CAAC,GAAG,aAAa,MAAM,GAAG,GAAG,GAAG,cAAc;IACrE,IAAK,MAAM,aAAa,cAAc,KAAM,WAAW;KACrD,eAAe;KACf,OAAO;KACP,kBAAkB;IACpB,OACE;GAEJ;GACA,IAAI,mBAAmB,oBAAoB,QACzC,SAAS,UAAU,CAAC,GAAG,eAAe,CAAC,CAAC,QAAQ;EAEpD;EACA,IAAI,CAAC,iBAAiB;GACpB,MAAM,WAAW,aAAa;GAC9B,IAAI;GACJ,IACE,MAAM,QAAQ,SAAS,OAAO,KAC9B,SAAS,QAAQ,MACd,UAAU,OAAO,UAAU,YAAY,MAAM,SAAS,MACzD,GAKA,OAHkB,SAAS,QAAQ,MAChC,UAAU,MAAM,SAAS,UAAU,MAAM,IAE7B,CAAC,EAAE;QACb,IAAI,OAAO,SAAS,YAAY,UACrC,OAAO,SAAS;GAElB,IAAI,MAAM;IACR,MAAM,aAAa,MAAM,aAAa,IAAI;IAC1C,MAAM,YAAY,WAAW;IAC7B,IAAI,oBAAoB,QACtB,WAAW,QAAQ;IAErB,KAAK,IAAI,IAAI,GAAG,IAAI,YAAY,GAAG,KAAK,GAAG;KACzC,WAAW,IAAI;KACf,SAAS,UAAU,WAAW,KAAK,EAAE;KACrC,IACG,MAAM,aAAa,CAAC,GAAG,aAAa,MAAM,GAAG,GAAG,GAAG,QAAQ,CAAC,KAC7D,WACA;MACA,IAAI,oBAAoB,QACtB,SAAS,UAAU,CAAC,GAAG,UAAU,CAAC,CAAC,QAAQ,CAAC,CAAC,KAAK,EAAE;MAEtD,eAAe,CAAC,GAAG,aAAa,MAAM,GAAG,GAAG,GAAG,QAAQ;MACvD,OAAO;MACP;KACF;IACF;GACF;EACF;CACF;CAEA,IAAI,OAAO;EACT,MAAM,WAAW,MAAM,QAAQ,KAAK,IAAI,QAAQ,CAAC,KAAK;EACtD,OAAO,MAAM,KAAK,CAAC,eAAe,aAAa,MAAM,IAAI,QAAQ,GAC/D,OAAO;CAEX;CAEA,OAAO,aAAa,MAAM,GAAG,GAAG;AAClC;AAEA,eAAe,eACb,UACA,SAewB;CACxB,MAAM,EACJ,eAAe,OACf,gBAAgB,OAChB,OACA,SACA,GAAG,SACD;CAGJ,IAAI,eAAe,SAAS,KAAK,YAAY;EAC3C,MAAM,SAAS,OAAO,YACpB,OAAO,QAAQ,OAAO,CAAC,CAAC,QACrB,CAAC,OAAO,MAAM,UAAU,CAAC,EAAE,WAAW,KAAK,CAC9C,CACF;EACA,OAAO,qBACL,QAAQ,QAAQ,GAChB,QACA,mBAAmB,OAAO,CAC5B;CACF,CAAC;CAED,IAAI,OAAO;EACT,MAAM,WAAW,MAAM,QAAQ,KAAK,IAAI,QAAQ,CAAC,KAAK;EACtD,OACE,aAAa,SAAS,KACtB,CAAC,eAAe,aAAa,aAAa,SAAS,IAAI,QAAQ,GAE/D,eAAe,aAAa,MAAM,GAAG,EAAE;CAE3C;CAEA,MAAM,gBACJ,iBAAiB,aAAa,EAAE,EAAE,QAAQ,MAAM;CAClD,IAAI,YAAY,gBACZ,aAAa,MAAM,GAAG,CAAC,CAAC,CAAC,OAAO,aAAa,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,IAC/D,aAAa,QAAQ;CAEzB,YAAY,MAAM,gBAAgB,WAAW;EAC3C,GAAG;EACH,iBAAiB,eAAe,SAAS,KAAA;EACzC,OAAO;CACT,CAAC;CAED,IAAI,eACF,OAAO,CAAC,UAAU,IAAI,GAAG,UAAU,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC;MAErD,OAAO,UAAU,QAAQ;AAE7B;AAEA,MAAM,iBAMF;CACF,OAAO;EACL,SAAS;EACT,cAAc;CAChB;CACA,IAAI;EACF,SAAS;EACT,cAAc;CAChB;CACA,QAAQ;EACN,SAAS;EACT,cAAc;CAChB;CACA,WAAW;EACT,SAAS;EACT,cAAc;CAChB;CACA,MAAM;EACJ,SAAS;EACT,cAAc;CAChB;CACA,UAAU;EACR,SAAS;EACT,cAAc;CAChB;CACA,SAAS;EACP,SAAS;EACT,cAAc;CAChB;CACA,QAAQ;EACN,SAAS;EACT,cAAc;CAChB;AACF;AAgBA,SAAS,qBACP,aACA,QACA,aACgC;CAChC,IAAI;CACJ,IAAI;CAEJ,QAAQ,aAAR;EACE,KAAK;GACH,IAAI,aACF,QAAQ,IAAI,kBAAkB,MAAM;QAEpC,MAAM,IAAI,aAAa,MAAM;GAE/B;EACF,KAAK;GACH,IAAI,aAAa;IACf,IAAI,gBAAsC,EACxC,GAAG,OACL;IACA,IAAI,gBAAgB,eAClB,gBAAgB;KACd,GAAG;KACH,kBAAkB,cAAc,YAAY,KAAK,QAAQ;MACvD,GAAG;MACH,MAAM;MACN,OAAO,KAAA;MACP,MAAM,KAAK,UAAU,GAAG,IAAI;KAC9B,EAAE;IACJ;IAEF,QAAQ,IAAI,eAAe,aAAa;GAC1C,OACE,MAAM,IAAI,UAAU,MAAM;GAE5B;EACF,KAAK;GACH,IAAI,aACF,QAAQ,IAAI,mBAAmB,MAAM;QAErC,MAAM,IAAI,cAAc,MAAM;GAEhC;EACF,KAAK;GACH,IAAI,aACF,QAAQ,IAAI,mBAAmB;IAC7B,GAAG;IACH,mBAAmB;KACjB,GAAG,OAAO;KACV,iBAAiB;IACnB;GACF,CAAC;QAED,MAAM,IAAI,cAAc;IACtB,GAAG;IACH,mBAAmB;KACjB,GAAG,OAAO;KACV,iBAAiB;IACnB;GACF,CAAC;GAEH;EACF,KAAK;GACH,IAAI,kBAAkB,QAAQ;IAC5B,IAAI,aACF,QAAQ,IAAI,iBAAiB,MAA2B;SAExD,MAAM,IAAI,YAAY,MAA2B;GAErD,OACE,MAAM,IAAI,MACR,yFACF;GAEF;EACF,KAAK;GACH,IAAI,aACF,QAAQ,IAAI,qBAAqB,MAA+B;QAC3D;IACL,IAAI,CAAC,OAAO,MACV,MAAM,IAAI,MAAM,0CAA0C;IAE5D,MAAM,IAAI,gBAAgB,MAA+B;GAC3D;GACA;EACF,KAAK;GACH,IAAI,UAAU,QAAQ;IACpB,IAAI,aACF,QAAQ,IAAI,iBAAiB,MAA2B;SAExD,MAAM,IAAI,YAAY,MAA2B;GAErD,OACE,MAAM,IAAI,MACR,iFACF;GAEF;EACF,SACE,MAAM,IAAI,MAAM,6BAA6B,aAAa;CAC9D;CAEA,IAAI,eAAe,OACjB,OAAO;CAET,IAAI,KACF,OAAO;CAET,MAAM,IAAI,MAAM,6BAA6B,aAAa;AAC5D;AAEA,SAAS,YAAY,OAAsC;CACzD,MAAM,YAAY,MAAM,QAAQ;CAChC,IAAI;CACJ,MAAM,SAAS,OAAO,YACpB,OAAO,QAAQ,KAAK,CAAC,CAAC,QACnB,CAAC,OAAO,CAAC,CAAC,QAAQ,kBAAkB,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,WAAW,KAAK,CAC3E,CACF;CAEA,IAAI,aAAa,gBACf,MAAM,qBAAqB,WAAW,MAAM;CAG9C,IAAI,CAAC,KACH,MAAM,IAAI,MACR,oCAAoC,UAAU,0BAA0B,OAAO,KAC7E,cACF,GACF;CAGF,OAAO;AACT;;;;;;;AAQA,SAAgB,oBAAoB,MAAiC;CACnE,MAAM,SAAS,KAAK,MAAM,IAAI;CAC9B,OAAO,QAAQ,QAAQ,CACrB,GAAG,OAAO,MAAM,GAAG,EAAE,CAAC,CAAC,KAAK,MAAM,GAAG,EAAE,GAAG,GAC1C,OAAO,OAAO,SAAS,EACzB,CAAC;AACH"}
@@ -1 +1 @@
1
- {"version":3,"file":"transform.cjs","names":["BaseOutputParser","isBaseMessageChunk","ChatGenerationChunk","isBaseMessage","convertToChunk","GenerationChunk"],"sources":["../../src/output_parsers/transform.ts"],"sourcesContent":["import { deepCompareStrict } from \"@cfworker/json-schema\";\nimport { BaseOutputParser } from \"./base.js\";\nimport {\n type BaseMessage,\n isBaseMessage,\n isBaseMessageChunk,\n} from \"../messages/base.js\";\nimport { convertToChunk } from \"../messages/utils.js\";\nimport type { BaseCallbackConfig } from \"../callbacks/manager.js\";\nimport {\n type Generation,\n type ChatGeneration,\n GenerationChunk,\n ChatGenerationChunk,\n} from \"../outputs.js\";\n\n/**\n * Class to parse the output of an LLM call that also allows streaming inputs.\n */\nexport abstract class BaseTransformOutputParser<\n T = unknown,\n> extends BaseOutputParser<T> {\n async *_transform(\n inputGenerator: AsyncGenerator<string | BaseMessage>\n ): AsyncGenerator<T> {\n for await (const chunk of inputGenerator) {\n if (typeof chunk === \"string\") {\n yield this.parseResult([{ text: chunk }]);\n } else {\n yield this.parseResult([\n {\n message: chunk,\n text: this._baseMessageToString(chunk),\n },\n ]);\n }\n }\n }\n\n /**\n * Transforms an asynchronous generator of input into an asynchronous\n * generator of parsed output.\n * @param inputGenerator An asynchronous generator of input.\n * @param options A configuration object.\n * @returns An asynchronous generator of parsed output.\n */\n async *transform(\n inputGenerator: AsyncGenerator<string | BaseMessage>,\n options: BaseCallbackConfig\n ): AsyncGenerator<T> {\n yield* this._transformStreamWithConfig(\n inputGenerator,\n this._transform.bind(this),\n {\n ...options,\n runType: \"parser\",\n }\n );\n }\n}\n\nexport type BaseCumulativeTransformOutputParserInput = { diff?: boolean };\n\n/**\n * A base class for output parsers that can handle streaming input. It\n * extends the `BaseTransformOutputParser` class and provides a method for\n * converting parsed outputs into a diff format.\n */\nexport abstract class BaseCumulativeTransformOutputParser<\n T = unknown,\n> extends BaseTransformOutputParser<T> {\n protected diff = false;\n\n constructor(fields?: BaseCumulativeTransformOutputParserInput) {\n super(fields);\n this.diff = fields?.diff ?? this.diff;\n }\n\n // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n protected abstract _diff(prev: any | undefined, next: any): any;\n\n abstract parsePartialResult(\n generations: Generation[] | ChatGeneration[]\n ): Promise<T | undefined>;\n\n async *_transform(\n inputGenerator: AsyncGenerator<string | BaseMessage>\n ): AsyncGenerator<T> {\n let prevParsed: T | undefined;\n let accGen: GenerationChunk | undefined;\n for await (const chunk of inputGenerator) {\n if (typeof chunk !== \"string\" && typeof chunk.content !== \"string\") {\n throw new Error(\"Cannot handle non-string output.\");\n }\n let chunkGen: GenerationChunk;\n if (isBaseMessageChunk(chunk)) {\n if (typeof chunk.content !== \"string\") {\n throw new Error(\"Cannot handle non-string message output.\");\n }\n chunkGen = new ChatGenerationChunk({\n message: chunk,\n text: chunk.content,\n });\n } else if (isBaseMessage(chunk)) {\n if (typeof chunk.content !== \"string\") {\n throw new Error(\"Cannot handle non-string message output.\");\n }\n chunkGen = new ChatGenerationChunk({\n message: convertToChunk(chunk),\n text: chunk.content,\n });\n } else {\n chunkGen = new GenerationChunk({ text: chunk });\n }\n\n if (accGen === undefined) {\n accGen = chunkGen;\n } else {\n accGen = accGen.concat(chunkGen);\n }\n\n const parsed = await this.parsePartialResult([accGen]);\n if (\n parsed !== undefined &&\n parsed !== null &&\n !deepCompareStrict(parsed, prevParsed)\n ) {\n if (this.diff) {\n yield this._diff(prevParsed, parsed);\n } else {\n yield parsed;\n }\n prevParsed = parsed;\n }\n }\n }\n\n getFormatInstructions(): string {\n return \"\";\n }\n}\n"],"mappings":";;;;;;;;;AAmBA,IAAsB,4BAAtB,cAEUA,eAAAA,iBAAoB;CAC5B,OAAO,WACL,gBACmB;EACnB,WAAW,MAAM,SAAS,gBACxB,IAAI,OAAO,UAAU,UACnB,MAAM,KAAK,YAAY,CAAC,EAAE,MAAM,MAAM,CAAC,CAAC;OAExC,MAAM,KAAK,YAAY,CACrB;GACE,SAAS;GACT,MAAM,KAAK,qBAAqB,KAAK;EACvC,CACF,CAAC;CAGP;;;;;;;;CASA,OAAO,UACL,gBACA,SACmB;EACnB,OAAO,KAAK,2BACV,gBACA,KAAK,WAAW,KAAK,IAAI,GACzB;GACE,GAAG;GACH,SAAS;EACX,CACF;CACF;AACF;;;;;;AASA,IAAsB,sCAAtB,cAEU,0BAA6B;CACrC,OAAiB;CAEjB,YAAY,QAAmD;EAC7D,MAAM,MAAM;EACZ,KAAK,OAAO,QAAQ,QAAQ,KAAK;CACnC;CASA,OAAO,WACL,gBACmB;EACnB,IAAI;EACJ,IAAI;EACJ,WAAW,MAAM,SAAS,gBAAgB;GACxC,IAAI,OAAO,UAAU,YAAY,OAAO,MAAM,YAAY,UACxD,MAAM,IAAI,MAAM,kCAAkC;GAEpD,IAAI;GACJ,IAAIC,aAAAA,mBAAmB,KAAK,GAAG;IAC7B,IAAI,OAAO,MAAM,YAAY,UAC3B,MAAM,IAAI,MAAM,0CAA0C;IAE5D,WAAW,IAAIC,gBAAAA,oBAAoB;KACjC,SAAS;KACT,MAAM,MAAM;IACd,CAAC;GACH,OAAO,IAAIC,aAAAA,cAAc,KAAK,GAAG;IAC/B,IAAI,OAAO,MAAM,YAAY,UAC3B,MAAM,IAAI,MAAM,0CAA0C;IAE5D,WAAW,IAAID,gBAAAA,oBAAoB;KACjC,SAASE,cAAAA,eAAe,KAAK;KAC7B,MAAM,MAAM;IACd,CAAC;GACH,OACE,WAAW,IAAIC,gBAAAA,gBAAgB,EAAE,MAAM,MAAM,CAAC;GAGhD,IAAI,WAAW,KAAA,GACb,SAAS;QAET,SAAS,OAAO,OAAO,QAAQ;GAGjC,MAAM,SAAS,MAAM,KAAK,mBAAmB,CAAC,MAAM,CAAC;GACrD,IACE,WAAW,KAAA,KACX,WAAW,QACX,EAAA,GAAA,sBAAA,kBAAA,CAAmB,QAAQ,UAAU,GACrC;IACA,IAAI,KAAK,MACP,MAAM,KAAK,MAAM,YAAY,MAAM;SAEnC,MAAM;IAER,aAAa;GACf;EACF;CACF;CAEA,wBAAgC;EAC9B,OAAO;CACT;AACF"}
1
+ {"version":3,"file":"transform.cjs","names":["BaseOutputParser","isBaseMessageChunk","ChatGenerationChunk","isBaseMessage","convertToChunk","GenerationChunk","deepCompareStrict"],"sources":["../../src/output_parsers/transform.ts"],"sourcesContent":["import { deepCompareStrict } from \"@cfworker/json-schema\";\nimport { BaseOutputParser } from \"./base.js\";\nimport {\n type BaseMessage,\n isBaseMessage,\n isBaseMessageChunk,\n} from \"../messages/base.js\";\nimport { convertToChunk } from \"../messages/utils.js\";\nimport type { BaseCallbackConfig } from \"../callbacks/manager.js\";\nimport {\n type Generation,\n type ChatGeneration,\n GenerationChunk,\n ChatGenerationChunk,\n} from \"../outputs.js\";\n\n/**\n * Class to parse the output of an LLM call that also allows streaming inputs.\n */\nexport abstract class BaseTransformOutputParser<\n T = unknown,\n> extends BaseOutputParser<T> {\n async *_transform(\n inputGenerator: AsyncGenerator<string | BaseMessage>\n ): AsyncGenerator<T> {\n for await (const chunk of inputGenerator) {\n if (typeof chunk === \"string\") {\n yield this.parseResult([{ text: chunk }]);\n } else {\n yield this.parseResult([\n {\n message: chunk,\n text: this._baseMessageToString(chunk),\n },\n ]);\n }\n }\n }\n\n /**\n * Transforms an asynchronous generator of input into an asynchronous\n * generator of parsed output.\n * @param inputGenerator An asynchronous generator of input.\n * @param options A configuration object.\n * @returns An asynchronous generator of parsed output.\n */\n async *transform(\n inputGenerator: AsyncGenerator<string | BaseMessage>,\n options: BaseCallbackConfig\n ): AsyncGenerator<T> {\n yield* this._transformStreamWithConfig(\n inputGenerator,\n this._transform.bind(this),\n {\n ...options,\n runType: \"parser\",\n }\n );\n }\n}\n\nexport type BaseCumulativeTransformOutputParserInput = { diff?: boolean };\n\n/**\n * A base class for output parsers that can handle streaming input. It\n * extends the `BaseTransformOutputParser` class and provides a method for\n * converting parsed outputs into a diff format.\n */\nexport abstract class BaseCumulativeTransformOutputParser<\n T = unknown,\n> extends BaseTransformOutputParser<T> {\n protected diff = false;\n\n constructor(fields?: BaseCumulativeTransformOutputParserInput) {\n super(fields);\n this.diff = fields?.diff ?? this.diff;\n }\n\n // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n protected abstract _diff(prev: any | undefined, next: any): any;\n\n abstract parsePartialResult(\n generations: Generation[] | ChatGeneration[]\n ): Promise<T | undefined>;\n\n async *_transform(\n inputGenerator: AsyncGenerator<string | BaseMessage>\n ): AsyncGenerator<T> {\n let prevParsed: T | undefined;\n let accGen: GenerationChunk | undefined;\n for await (const chunk of inputGenerator) {\n if (typeof chunk !== \"string\" && typeof chunk.content !== \"string\") {\n throw new Error(\"Cannot handle non-string output.\");\n }\n let chunkGen: GenerationChunk;\n if (isBaseMessageChunk(chunk)) {\n if (typeof chunk.content !== \"string\") {\n throw new Error(\"Cannot handle non-string message output.\");\n }\n chunkGen = new ChatGenerationChunk({\n message: chunk,\n text: chunk.content,\n });\n } else if (isBaseMessage(chunk)) {\n if (typeof chunk.content !== \"string\") {\n throw new Error(\"Cannot handle non-string message output.\");\n }\n chunkGen = new ChatGenerationChunk({\n message: convertToChunk(chunk),\n text: chunk.content,\n });\n } else {\n chunkGen = new GenerationChunk({ text: chunk });\n }\n\n if (accGen === undefined) {\n accGen = chunkGen;\n } else {\n accGen = accGen.concat(chunkGen);\n }\n\n const parsed = await this.parsePartialResult([accGen]);\n if (\n parsed !== undefined &&\n parsed !== null &&\n !deepCompareStrict(parsed, prevParsed)\n ) {\n if (this.diff) {\n yield this._diff(prevParsed, parsed);\n } else {\n yield parsed;\n }\n prevParsed = parsed;\n }\n }\n }\n\n getFormatInstructions(): string {\n return \"\";\n }\n}\n"],"mappings":";;;;;;;;;AAmBA,IAAsB,4BAAtB,cAEUA,eAAAA,iBAAoB;CAC5B,OAAO,WACL,gBACmB;EACnB,WAAW,MAAM,SAAS,gBACxB,IAAI,OAAO,UAAU,UACnB,MAAM,KAAK,YAAY,CAAC,EAAE,MAAM,MAAM,CAAC,CAAC;OAExC,MAAM,KAAK,YAAY,CACrB;GACE,SAAS;GACT,MAAM,KAAK,qBAAqB,KAAK;EACvC,CACF,CAAC;CAGP;;;;;;;;CASA,OAAO,UACL,gBACA,SACmB;EACnB,OAAO,KAAK,2BACV,gBACA,KAAK,WAAW,KAAK,IAAI,GACzB;GACE,GAAG;GACH,SAAS;EACX,CACF;CACF;AACF;;;;;;AASA,IAAsB,sCAAtB,cAEU,0BAA6B;CACrC,OAAiB;CAEjB,YAAY,QAAmD;EAC7D,MAAM,MAAM;EACZ,KAAK,OAAO,QAAQ,QAAQ,KAAK;CACnC;CASA,OAAO,WACL,gBACmB;EACnB,IAAI;EACJ,IAAI;EACJ,WAAW,MAAM,SAAS,gBAAgB;GACxC,IAAI,OAAO,UAAU,YAAY,OAAO,MAAM,YAAY,UACxD,MAAM,IAAI,MAAM,kCAAkC;GAEpD,IAAI;GACJ,IAAIC,aAAAA,mBAAmB,KAAK,GAAG;IAC7B,IAAI,OAAO,MAAM,YAAY,UAC3B,MAAM,IAAI,MAAM,0CAA0C;IAE5D,WAAW,IAAIC,gBAAAA,oBAAoB;KACjC,SAAS;KACT,MAAM,MAAM;IACd,CAAC;GACH,OAAO,IAAIC,aAAAA,cAAc,KAAK,GAAG;IAC/B,IAAI,OAAO,MAAM,YAAY,UAC3B,MAAM,IAAI,MAAM,0CAA0C;IAE5D,WAAW,IAAID,gBAAAA,oBAAoB;KACjC,SAASE,cAAAA,eAAe,KAAK;KAC7B,MAAM,MAAM;IACd,CAAC;GACH,OACE,WAAW,IAAIC,gBAAAA,gBAAgB,EAAE,MAAM,MAAM,CAAC;GAGhD,IAAI,WAAW,KAAA,GACb,SAAS;QAET,SAAS,OAAO,OAAO,QAAQ;GAGjC,MAAM,SAAS,MAAM,KAAK,mBAAmB,CAAC,MAAM,CAAC;GACrD,IACE,WAAW,KAAA,KACX,WAAW,QACX,EAAA,GAACC,sBAAAA,kBAAAA,CAAkB,QAAQ,UAAU,GACrC;IACA,IAAI,KAAK,MACP,MAAM,KAAK,MAAM,YAAY,MAAM;SAEnC,MAAM;IAER,aAAa;GACf;EACF;CACF;CAEA,wBAAgC;EAC9B,OAAO;CACT;AACF"}
@@ -115,7 +115,8 @@ var BaseChatPromptTemplate = class extends require_base$2.BasePromptTemplate {
115
115
  return (await this.formatPromptValue(values)).toString();
116
116
  }
117
117
  async formatPromptValue(values) {
118
- return new require_prompt_values.ChatPromptValue(await this.formatMessages(values));
118
+ const resultMessages = await this.formatMessages(values);
119
+ return new require_prompt_values.ChatPromptValue(resultMessages);
119
120
  }
120
121
  };
121
122
  /**