@langchain/core 1.0.0-alpha.2 → 1.0.0-alpha.4

package/dist/messages/ai.cjs

@@ -1,18 +1,14 @@
 const require_json = require('../utils/json.cjs');
 const require_base = require('./base.cjs');
 const require_index = require('./block_translators/index.cjs');
+const require_metadata = require('./metadata.cjs');
 const require_messages_tool = require('./tool.cjs');
 
 //#region src/messages/ai.ts
-/**
-* Represents an AI message in a conversation.
-*/
 var AIMessage = class extends require_base.BaseMessage {
+	type = "ai";
 	tool_calls = [];
 	invalid_tool_calls = [];
-	/**
-	* If provided, token usage information associated with the message.
-	*/
 	usage_metadata;
 	get lc_aliases() {
 		return {
@@ -21,13 +17,13 @@ var AIMessage = class extends require_base.BaseMessage {
 			invalid_tool_calls: "invalid_tool_calls"
 		};
 	}
-	constructor(fields, kwargs) {
+	constructor(fields) {
 		let initParams;
-		if (typeof fields === "string") initParams = {
+		if (typeof fields === "string" || Array.isArray(fields)) initParams = {
 			content: fields,
 			tool_calls: [],
 			invalid_tool_calls: [],
-			additional_kwargs: kwargs ?? {}
+			additional_kwargs: {}
 		};
 		else {
 			initParams = fields;
@@ -48,7 +44,7 @@ var AIMessage = class extends require_base.BaseMessage {
 					initParams.tool_calls = initParams.tool_calls ?? [];
 					initParams.invalid_tool_calls = initParams.invalid_tool_calls ?? [];
 				}
-			} catch (e) {
+			} catch {
 				initParams.tool_calls = [];
 				initParams.invalid_tool_calls = [];
 			}
@@ -61,6 +57,7 @@ var AIMessage = class extends require_base.BaseMessage {
 				})));
 				const missingToolCalls = initParams.contentBlocks.filter((block) => block.type === "tool_call").filter((block) => !initParams.tool_calls?.some((toolCall) => toolCall.id === block.id && toolCall.name === block.name));
 				if (missingToolCalls.length > 0) initParams.tool_calls = missingToolCalls.map((block) => ({
+					type: "tool_call",
 					id: block.id,
 					name: block.name,
 					args: block.args
@@ -77,14 +74,10 @@ var AIMessage = class extends require_base.BaseMessage {
 	static lc_name() {
 		return "AIMessage";
 	}
-	_getType() {
-		return "ai";
-	}
 	get contentBlocks() {
-		if (this.response_metadata?.output_version === "v1") return this.content;
-		const modelProvider = this.response_metadata?.model_provider;
-		if (modelProvider) {
-			const translator = require_index.getTranslator(modelProvider);
+		if ("output_version" in this.response_metadata && this.response_metadata.output_version === "v1") return this.content;
+		if ("model_provider" in this.response_metadata && typeof this.response_metadata.model_provider === "string") {
+			const translator = require_index.getTranslator(this.response_metadata.model_provider);
 			if (translator) return translator.translateContent(this);
 		}
 		const blocks = super.contentBlocks;
@@ -108,10 +101,19 @@ var AIMessage = class extends require_base.BaseMessage {
 			usage_metadata: this.usage_metadata
 		};
 	}
+	static isInstance(obj) {
+		return super.isInstance(obj) && obj.type === "ai";
+	}
 };
+/**
+* @deprecated Use {@link AIMessage.isInstance} instead
+*/
 function isAIMessage(x) {
 	return x._getType() === "ai";
 }
+/**
+* @deprecated Use {@link AIMessageChunk.isInstance} instead
+*/
 function isAIMessageChunk(x) {
 	return x._getType() === "ai";
 }
@@ -119,17 +121,15 @@ function isAIMessageChunk(x) {
 * Represents a chunk of an AI message, which can be concatenated with
 * other AI message chunks.
 */
-var AIMessageChunk = class AIMessageChunk extends require_base.BaseMessageChunk {
+var AIMessageChunk = class extends require_base.BaseMessageChunk {
+	type = "ai";
 	tool_calls = [];
 	invalid_tool_calls = [];
 	tool_call_chunks = [];
-	/**
-	* If provided, token usage information associated with the message.
-	*/
 	usage_metadata;
 	constructor(fields) {
 		let initParams;
-		if (typeof fields === "string") initParams = {
+		if (typeof fields === "string" || Array.isArray(fields)) initParams = {
 			content: fields,
 			tool_calls: [],
 			invalid_tool_calls: [],
@@ -152,7 +152,7 @@ var AIMessageChunk = class AIMessageChunk extends require_base.BaseMessageChunk
 			const toolCalls = [];
 			const invalidToolCalls = [];
 			for (const [id, chunks] of Object.entries(groupedToolCallChunk)) {
-				let parsedArgs = {};
+				let parsedArgs = null;
 				const name = chunks[0]?.name ?? "";
 				const joinedArgs = chunks.map((c) => c.args || "").join("");
 				const argsStr = joinedArgs.length ? joinedArgs : "{}";
@@ -166,7 +166,7 @@ var AIMessageChunk = class AIMessageChunk extends require_base.BaseMessageChunk
 						id: originalId,
 						type: "tool_call"
 					});
-				} catch (e) {
+				} catch {
 					invalidToolCalls.push({
 						name,
 						args: argsStr,
@@ -200,14 +200,10 @@ var AIMessageChunk = class AIMessageChunk extends require_base.BaseMessageChunk
 	static lc_name() {
 		return "AIMessageChunk";
 	}
-	_getType() {
-		return "ai";
-	}
 	get contentBlocks() {
-		if (this.response_metadata?.output_version === "v1") return this.content;
-		const modelProvider = this.response_metadata?.model_provider;
-		if (modelProvider) {
-			const translator = require_index.getTranslator(modelProvider);
+		if ("output_version" in this.response_metadata && this.response_metadata.output_version === "v1") return this.content;
+		if ("model_provider" in this.response_metadata && typeof this.response_metadata.model_provider === "string") {
+			const translator = require_index.getTranslator(this.response_metadata.model_provider);
 			if (translator) return translator.translateContent(this);
 		}
 		const blocks = super.contentBlocks;
@@ -238,7 +234,7 @@ var AIMessageChunk = class AIMessageChunk extends require_base.BaseMessageChunk
 		const combinedFields = {
 			content: require_base.mergeContent(this.content, chunk.content),
 			additional_kwargs: require_base._mergeDicts(this.additional_kwargs, chunk.additional_kwargs),
-			response_metadata: require_base._mergeDicts(this.response_metadata, chunk.response_metadata),
+			response_metadata: require_metadata.mergeResponseMetadata(this.response_metadata, chunk.response_metadata),
 			tool_call_chunks: [],
 			id: this.id ?? chunk.id
 		};
@@ -246,36 +242,12 @@ var AIMessageChunk = class AIMessageChunk extends require_base.BaseMessageChunk
 			const rawToolCalls = require_base._mergeLists(this.tool_call_chunks, chunk.tool_call_chunks);
 			if (rawToolCalls !== void 0 && rawToolCalls.length > 0) combinedFields.tool_call_chunks = rawToolCalls;
 		}
-		if (this.usage_metadata !== void 0 || chunk.usage_metadata !== void 0) {
-			const inputTokenDetails = {
-				...(this.usage_metadata?.input_token_details?.audio !== void 0 || chunk.usage_metadata?.input_token_details?.audio !== void 0) && { audio: (this.usage_metadata?.input_token_details?.audio ?? 0) + (chunk.usage_metadata?.input_token_details?.audio ?? 0) },
-				...(this.usage_metadata?.input_token_details?.cache_read !== void 0 || chunk.usage_metadata?.input_token_details?.cache_read !== void 0) && { cache_read: (this.usage_metadata?.input_token_details?.cache_read ?? 0) + (chunk.usage_metadata?.input_token_details?.cache_read ?? 0) },
-				...(this.usage_metadata?.input_token_details?.cache_creation !== void 0 || chunk.usage_metadata?.input_token_details?.cache_creation !== void 0) && { cache_creation: (this.usage_metadata?.input_token_details?.cache_creation ?? 0) + (chunk.usage_metadata?.input_token_details?.cache_creation ?? 0) }
-			};
-			const outputTokenDetails = {
-				...(this.usage_metadata?.output_token_details?.audio !== void 0 || chunk.usage_metadata?.output_token_details?.audio !== void 0) && { audio: (this.usage_metadata?.output_token_details?.audio ?? 0) + (chunk.usage_metadata?.output_token_details?.audio ?? 0) },
-				...(this.usage_metadata?.output_token_details?.reasoning !== void 0 || chunk.usage_metadata?.output_token_details?.reasoning !== void 0) && { reasoning: (this.usage_metadata?.output_token_details?.reasoning ?? 0) + (chunk.usage_metadata?.output_token_details?.reasoning ?? 0) }
-			};
-			const left = this.usage_metadata ?? {
-				input_tokens: 0,
-				output_tokens: 0,
-				total_tokens: 0
-			};
-			const right = chunk.usage_metadata ?? {
-				input_tokens: 0,
-				output_tokens: 0,
-				total_tokens: 0
-			};
-			const usage_metadata = {
-				input_tokens: left.input_tokens + right.input_tokens,
-				output_tokens: left.output_tokens + right.output_tokens,
-				total_tokens: left.total_tokens + right.total_tokens,
-				...Object.keys(inputTokenDetails).length > 0 && { input_token_details: inputTokenDetails },
-				...Object.keys(outputTokenDetails).length > 0 && { output_token_details: outputTokenDetails }
-			};
-			combinedFields.usage_metadata = usage_metadata;
-		}
-		return new AIMessageChunk(combinedFields);
+		if (this.usage_metadata !== void 0 || chunk.usage_metadata !== void 0) combinedFields.usage_metadata = require_metadata.mergeUsageMetadata(this.usage_metadata, chunk.usage_metadata);
+		const Cls = this.constructor;
+		return new Cls(combinedFields);
+	}
+	static isInstance(obj) {
+		return super.isInstance(obj) && obj.type === "ai";
 	}
 };
 

package/dist/messages/ai.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"ai.cjs","names":["BaseMessage","fields: string | AIMessageFields","kwargs?: Record<string, unknown>","initParams: AIMessageFields","toolCalls","defaultToolCallParser","getTranslator","x: BaseMessage","x: BaseMessageChunk","BaseMessageChunk","fields: string | AIMessageChunkFields","initParams: AIMessageChunkFields","toolCalls: ToolCall[]","invalidToolCalls: InvalidToolCall[]","parsePartialJson","chunk: AIMessageChunk","combinedFields: AIMessageChunkFields","mergeContent","_mergeDicts","_mergeLists","inputTokenDetails: InputTokenDetails","outputTokenDetails: OutputTokenDetails","left: UsageMetadata","right: UsageMetadata","usage_metadata: UsageMetadata"],"sources":["../../src/messages/ai.ts"],"sourcesContent":["import { parsePartialJson } from \"../utils/json.js\";\nimport {\n  BaseMessage,\n  BaseMessageChunk,\n  mergeContent,\n  _mergeDicts,\n  type MessageType,\n  BaseMessageFields,\n  _mergeLists,\n} from \"./base.js\";\nimport { getTranslator } from \"./block_translators/index.js\";\nimport { ContentBlock } from \"./content/index.js\";\nimport {\n  InvalidToolCall,\n  ToolCall,\n  ToolCallChunk,\n  defaultToolCallParser,\n} from \"./tool.js\";\n\nexport type AIMessageFields = BaseMessageFields & {\n  tool_calls?: ToolCall[];\n  invalid_tool_calls?: InvalidToolCall[];\n  usage_metadata?: UsageMetadata;\n};\n\nexport type ModalitiesTokenDetails = {\n  /**\n   * Text tokens.\n   * Does not need to be reported, but some models will do so.\n   */\n  text?: number;\n\n  /**\n   * Image (non-video) tokens.\n   */\n  image?: number;\n\n  /**\n   * Audio tokens.\n   */\n  audio?: number;\n\n  /**\n   * Video tokens.\n   */\n  video?: number;\n\n  /**\n   * Document tokens.\n   * e.g. PDF\n   */\n  document?: number;\n};\n\n/**\n * Breakdown of input token counts.\n *\n * Does not *need* to sum to full input token count. Does *not* need to have all keys.\n */\nexport type InputTokenDetails = ModalitiesTokenDetails & {\n  /**\n   * Input tokens that were cached and there was a cache hit.\n   *\n   * Since there was a cache hit, the tokens were read from the cache.\n   * More precisely, the model state given these tokens was read from the cache.\n   */\n  cache_read?: number;\n\n  /**\n   * Input tokens that were cached and there was a cache miss.\n   *\n   * Since there was a cache miss, the cache was created from these tokens.\n   */\n  cache_creation?: number;\n};\n\n/**\n * Breakdown of output token counts.\n *\n * Does *not* need to sum to full output token count. Does *not* need to have all keys.\n */\nexport type OutputTokenDetails = ModalitiesTokenDetails & {\n  /**\n   * Reasoning output tokens.\n   *\n   * Tokens generated by the model in a chain of thought process (i.e. by\n   * OpenAI's o1 models) that are not returned as part of model output.\n   */\n  reasoning?: number;\n};\n\n/**\n * Usage metadata for a message, such as token counts.\n */\nexport type UsageMetadata = {\n  /**\n   * Count of input (or prompt) tokens. Sum of all input token types.\n   */\n  input_tokens: number;\n  /**\n   * Count of output (or completion) tokens. Sum of all output token types.\n   */\n  output_tokens: number;\n  /**\n   * Total token count. Sum of input_tokens + output_tokens.\n   */\n  total_tokens: number;\n\n  /**\n   * Breakdown of input token counts.\n   *\n   * Does *not* need to sum to full input token count. Does *not* need to have all keys.\n   */\n  input_token_details?: InputTokenDetails;\n\n  /**\n   * Breakdown of output token counts.\n   *\n   * Does *not* need to sum to full output token count. Does *not* need to have all keys.\n   */\n  output_token_details?: OutputTokenDetails;\n};\n\n/**\n * Represents an AI message in a conversation.\n */\nexport class AIMessage extends BaseMessage implements AIMessageFields {\n  // These are typed as optional to avoid breaking changes and allow for casting\n  // from BaseMessage.\n  tool_calls?: ToolCall[] = [];\n\n  invalid_tool_calls?: InvalidToolCall[] = [];\n\n  /**\n   * If provided, token usage information associated with the message.\n   */\n  usage_metadata?: UsageMetadata;\n\n  get lc_aliases(): Record<string, string> {\n    // exclude snake case conversion to pascal case\n    return {\n      ...super.lc_aliases,\n      tool_calls: \"tool_calls\",\n      invalid_tool_calls: \"invalid_tool_calls\",\n    };\n  }\n\n  constructor(\n    fields: string | AIMessageFields,\n    /** @deprecated */\n    kwargs?: Record<string, unknown>\n  ) {\n    let initParams: AIMessageFields;\n    if (typeof fields === \"string\") {\n      initParams = {\n        content: fields,\n        tool_calls: [],\n        invalid_tool_calls: [],\n        additional_kwargs: kwargs ?? {},\n      };\n    } else {\n      initParams = fields;\n      const rawToolCalls = initParams.additional_kwargs?.tool_calls;\n      const toolCalls = initParams.tool_calls;\n      if (\n        !(rawToolCalls == null) &&\n        rawToolCalls.length > 0 &&\n        (toolCalls === undefined || toolCalls.length === 0)\n      ) {\n        console.warn(\n          [\n            \"New LangChain packages are available that more efficiently handle\",\n            \"tool calling.\\n\\nPlease upgrade your packages to versions that set\",\n            \"message tool calls. e.g., `pnpm install @langchain/anthropic`,\",\n            \"pnpm install @langchain/openai`, etc.\",\n          ].join(\" \")\n        );\n      }\n      try {\n        if (!(rawToolCalls == null) && toolCalls === undefined) {\n          const [toolCalls, invalidToolCalls] =\n            defaultToolCallParser(rawToolCalls);\n          initParams.tool_calls = toolCalls ?? [];\n          initParams.invalid_tool_calls = invalidToolCalls ?? [];\n        } else {\n          initParams.tool_calls = initParams.tool_calls ?? [];\n          initParams.invalid_tool_calls = initParams.invalid_tool_calls ?? [];\n        }\n      } catch (e) {\n        // Do nothing if parsing fails\n        initParams.tool_calls = [];\n        initParams.invalid_tool_calls = [];\n      }\n      if (initParams.contentBlocks !== undefined) {\n        // Add constructor tool calls as content blocks\n        initParams.contentBlocks.push(\n          ...initParams.tool_calls.map((toolCall) => ({\n            type: \"tool_call\" as const,\n            id: toolCall.id,\n            name: toolCall.name,\n            args: toolCall.args,\n          }))\n        );\n        // Add content block tool calls that aren't in the constructor tool calls\n        const missingToolCalls = initParams.contentBlocks\n          .filter<ContentBlock.Tools.ToolCall>(\n            (block) => block.type === \"tool_call\"\n          )\n          .filter(\n            (block) =>\n              !initParams.tool_calls?.some(\n                (toolCall) =>\n                  toolCall.id === block.id && toolCall.name === block.name\n              )\n          );\n        if (missingToolCalls.length > 0) {\n          initParams.tool_calls = missingToolCalls.map((block) => ({\n            id: block.id!,\n            name: block.name,\n            args: block.args as Record<string, unknown>,\n          }));\n        }\n      }\n    }\n    // Sadly, TypeScript only allows super() calls at root if the class has\n    // properties with initializers, so we have to check types twice.\n    super(initParams);\n    if (typeof initParams !== \"string\") {\n      this.tool_calls = initParams.tool_calls ?? this.tool_calls;\n      this.invalid_tool_calls =\n        initParams.invalid_tool_calls ?? this.invalid_tool_calls;\n    }\n    this.usage_metadata = initParams.usage_metadata;\n  }\n\n  static lc_name() {\n    return \"AIMessage\";\n  }\n\n  _getType(): MessageType {\n    return \"ai\";\n  }\n\n  get contentBlocks(): Array<ContentBlock.Standard> {\n    if (this.response_metadata?.output_version === \"v1\") {\n      return this.content as Array<ContentBlock.Standard>;\n    }\n\n    const modelProvider = this.response_metadata?.model_provider;\n    if (modelProvider) {\n      const translator = getTranslator(modelProvider);\n      if (translator) {\n        return translator.translateContent(this);\n      }\n    }\n\n    const blocks = super.contentBlocks;\n\n    if (this.tool_calls) {\n      const missingToolCalls = this.tool_calls.filter(\n        (block) =>\n          !blocks.some((b) => b.id === block.id && b.name === block.name)\n      );\n      blocks.push(\n        ...missingToolCalls.map((block) => ({\n          ...block,\n          type: \"tool_call\" as const,\n          id: block.id,\n          name: block.name,\n          args: block.args,\n        }))\n      );\n    }\n\n    return blocks;\n  }\n\n  override get _printableFields(): Record<string, unknown> {\n    return {\n      ...super._printableFields,\n      tool_calls: this.tool_calls,\n      invalid_tool_calls: this.invalid_tool_calls,\n      usage_metadata: this.usage_metadata,\n    };\n  }\n}\n\nexport function isAIMessage(x: BaseMessage): x is AIMessage {\n  return x._getType() === \"ai\";\n}\n\nexport function isAIMessageChunk(x: BaseMessageChunk): x is AIMessageChunk {\n  return x._getType() === \"ai\";\n}\n\nexport type AIMessageChunkFields = AIMessageFields & {\n  tool_call_chunks?: ToolCallChunk[];\n};\n\n/**\n * Represents a chunk of an AI message, which can be concatenated with\n * other AI message chunks.\n */\nexport class AIMessageChunk extends BaseMessageChunk {\n  // Must redeclare tool call fields since there is no multiple inheritance in JS.\n  // These are typed as optional to avoid breaking changes and allow for casting\n  // from BaseMessage.\n  tool_calls?: ToolCall[] = [];\n\n  invalid_tool_calls?: InvalidToolCall[] = [];\n\n  tool_call_chunks?: ToolCallChunk[] = [];\n\n  /**\n   * If provided, token usage information associated with the message.\n   */\n  usage_metadata?: UsageMetadata;\n\n  constructor(fields: string | AIMessageChunkFields) {\n    let initParams: AIMessageChunkFields;\n    if (typeof fields === \"string\") {\n      initParams = {\n        content: fields,\n        tool_calls: [],\n        invalid_tool_calls: [],\n        tool_call_chunks: [],\n      };\n    } else if (fields.tool_call_chunks === undefined) {\n      initParams = {\n        ...fields,\n        tool_calls: fields.tool_calls ?? [],\n        invalid_tool_calls: [],\n        tool_call_chunks: [],\n        usage_metadata:\n          fields.usage_metadata !== undefined\n            ? fields.usage_metadata\n            : undefined,\n      };\n    } else {\n      const groupedToolCallChunk = fields.tool_call_chunks.reduce(\n        (acc, chunk) => {\n          // Assign a fallback ID if the chunk doesn't have one\n          // This can happen with tools that have empty schemas\n          const chunkId = chunk.id || `fallback-${chunk.index || 0}`;\n          acc[chunkId] = acc[chunkId] ?? [];\n          acc[chunkId].push(chunk);\n          return acc;\n        },\n        {} as Record<string, ToolCallChunk[]>\n      );\n\n      const toolCalls: ToolCall[] = [];\n      const invalidToolCalls: InvalidToolCall[] = [];\n      for (const [id, chunks] of Object.entries(groupedToolCallChunk)) {\n        let parsedArgs = {};\n        const name = chunks[0]?.name ?? \"\";\n        const joinedArgs = chunks.map((c) => c.args || \"\").join(\"\");\n        const argsStr = joinedArgs.length ? joinedArgs : \"{}\";\n        // Use the original ID from the first chunk if it exists, otherwise use the grouped ID\n        const originalId = chunks[0]?.id || id;\n        try {\n          parsedArgs = parsePartialJson(argsStr);\n          if (\n            parsedArgs === null ||\n            typeof parsedArgs !== \"object\" ||\n            Array.isArray(parsedArgs)\n          ) {\n            throw new Error(\"Malformed tool call chunk args.\");\n          }\n          toolCalls.push({\n            name,\n            args: parsedArgs,\n            id: originalId,\n            type: \"tool_call\",\n          });\n        } catch (e) {\n          invalidToolCalls.push({\n            name,\n            args: argsStr,\n            id: originalId,\n            error: \"Malformed args.\",\n            type: \"invalid_tool_call\",\n          });\n        }\n      }\n      initParams = {\n        ...fields,\n        tool_calls: toolCalls,\n        invalid_tool_calls: invalidToolCalls,\n        usage_metadata:\n          fields.usage_metadata !== undefined\n            ? fields.usage_metadata\n            : undefined,\n      };\n    }\n    // Sadly, TypeScript only allows super() calls at root if the class has\n    // properties with initializers, so we have to check types twice.\n    super(initParams);\n    this.tool_call_chunks =\n      initParams.tool_call_chunks ?? this.tool_call_chunks;\n    this.tool_calls = initParams.tool_calls ?? this.tool_calls;\n    this.invalid_tool_calls =\n      initParams.invalid_tool_calls ?? this.invalid_tool_calls;\n    this.usage_metadata = initParams.usage_metadata;\n  }\n\n  get lc_aliases(): Record<string, string> {\n    // exclude snake case conversion to pascal case\n    return {\n      ...super.lc_aliases,\n      tool_calls: \"tool_calls\",\n      invalid_tool_calls: \"invalid_tool_calls\",\n      tool_call_chunks: \"tool_call_chunks\",\n    };\n  }\n\n  static lc_name() {\n    return \"AIMessageChunk\";\n  }\n\n  _getType(): MessageType {\n    return \"ai\";\n  }\n\n  get contentBlocks(): Array<ContentBlock.Standard> {\n    if (this.response_metadata?.output_version === \"v1\") {\n      return this.content as Array<ContentBlock.Standard>;\n    }\n\n    const modelProvider = this.response_metadata?.model_provider;\n    if (modelProvider) {\n      const translator = getTranslator(modelProvider);\n      if (translator) {\n        return translator.translateContent(this);\n      }\n    }\n\n    const blocks = super.contentBlocks;\n\n    if (this.tool_calls) {\n      if (typeof this.content !== \"string\") {\n        const contentToolCalls = this.content\n          .filter((block) => block.type === \"tool_call\")\n          .map((block) => block.id);\n        for (const toolCall of this.tool_calls) {\n          if (toolCall.id && !contentToolCalls.includes(toolCall.id)) {\n            blocks.push({\n              ...toolCall,\n              type: \"tool_call\",\n              id: toolCall.id,\n              name: toolCall.name,\n              args: toolCall.args,\n            });\n          }\n        }\n      }\n    }\n\n    return blocks;\n  }\n\n  override get _printableFields(): Record<string, unknown> {\n    return {\n      ...super._printableFields,\n      tool_calls: this.tool_calls,\n      tool_call_chunks: this.tool_call_chunks,\n      invalid_tool_calls: this.invalid_tool_calls,\n      usage_metadata: this.usage_metadata,\n    };\n  }\n\n  concat(chunk: AIMessageChunk) {\n    const combinedFields: AIMessageChunkFields = {\n      content: mergeContent(this.content, chunk.content),\n      additional_kwargs: _mergeDicts(\n        this.additional_kwargs,\n        chunk.additional_kwargs\n      ),\n      response_metadata: _mergeDicts(\n        this.response_metadata,\n        chunk.response_metadata\n      ),\n      tool_call_chunks: [],\n      id: this.id ?? chunk.id,\n    };\n    if (\n      this.tool_call_chunks !== undefined ||\n      chunk.tool_call_chunks !== undefined\n    ) {\n      const rawToolCalls = _mergeLists(\n        this.tool_call_chunks,\n        chunk.tool_call_chunks\n      );\n      if (rawToolCalls !== undefined && rawToolCalls.length > 0) {\n        combinedFields.tool_call_chunks = rawToolCalls;\n      }\n    }\n    if (\n      this.usage_metadata !== undefined ||\n      chunk.usage_metadata !== undefined\n    ) {\n      const inputTokenDetails: InputTokenDetails = {\n        ...((this.usage_metadata?.input_token_details?.audio !== undefined ||\n          chunk.usage_metadata?.input_token_details?.audio !== undefined) && {\n          audio:\n            (this.usage_metadata?.input_token_details?.audio ?? 0) +\n            (chunk.usage_metadata?.input_token_details?.audio ?? 0),\n        }),\n        ...((this.usage_metadata?.input_token_details?.cache_read !==\n          undefined ||\n          chunk.usage_metadata?.input_token_details?.cache_read !==\n            undefined) && {\n          cache_read:\n            (this.usage_metadata?.input_token_details?.cache_read ?? 0) +\n            (chunk.usage_metadata?.input_token_details?.cache_read ?? 0),\n        }),\n        ...((this.usage_metadata?.input_token_details?.cache_creation !==\n          undefined ||\n          chunk.usage_metadata?.input_token_details?.cache_creation !==\n            undefined) && {\n          cache_creation:\n            (this.usage_metadata?.input_token_details?.cache_creation ?? 0) +\n            (chunk.usage_metadata?.input_token_details?.cache_creation ?? 0),\n        }),\n      };\n\n      const outputTokenDetails: OutputTokenDetails = {\n        ...((this.usage_metadata?.output_token_details?.audio !== undefined ||\n          chunk.usage_metadata?.output_token_details?.audio !== undefined) && {\n          audio:\n            (this.usage_metadata?.output_token_details?.audio ?? 0) +\n            (chunk.usage_metadata?.output_token_details?.audio ?? 0),\n        }),\n        ...((this.usage_metadata?.output_token_details?.reasoning !==\n          undefined ||\n          chunk.usage_metadata?.output_token_details?.reasoning !==\n            undefined) && {\n          reasoning:\n            (this.usage_metadata?.output_token_details?.reasoning ?? 0) +\n            (chunk.usage_metadata?.output_token_details?.reasoning ?? 0),\n        }),\n      };\n\n      const left: UsageMetadata = this.usage_metadata ?? {\n        input_tokens: 0,\n        output_tokens: 0,\n        total_tokens: 0,\n      };\n      const right: UsageMetadata = chunk.usage_metadata ?? {\n        input_tokens: 0,\n        output_tokens: 0,\n        total_tokens: 0,\n      };\n      const usage_metadata: UsageMetadata = {\n        input_tokens: left.input_tokens + right.input_tokens,\n        output_tokens: left.output_tokens + right.output_tokens,\n        total_tokens: left.total_tokens + right.total_tokens,\n        // Do not include `input_token_details` / `output_token_details` keys in combined fields\n        // unless their values are defined.\n        ...(Object.keys(inputTokenDetails).length > 0 && {\n          input_token_details: inputTokenDetails,\n        }),\n        ...(Object.keys(outputTokenDetails).length > 0 && {\n          output_token_details: outputTokenDetails,\n        }),\n      };\n      combinedFields.usage_metadata = usage_metadata;\n    }\n    return new AIMessageChunk(combinedFields);\n  }\n}\n"],"mappings":";;;;;;;;;AA8HA,IAAa,YAAb,cAA+BA,yBAAuC;CAGpE,aAA0B,CAAE;CAE5B,qBAAyC,CAAE;;;;CAK3C;CAEA,IAAI,aAAqC;AAEvC,SAAO;GACL,GAAG,MAAM;GACT,YAAY;GACZ,oBAAoB;EACrB;CACF;CAED,YACEC,QAEAC,QACA;EACA,IAAIC;AACJ,MAAI,OAAO,WAAW,UACpB,aAAa;GACX,SAAS;GACT,YAAY,CAAE;GACd,oBAAoB,CAAE;GACtB,mBAAmB,UAAU,CAAE;EAChC;OACI;GACL,aAAa;GACb,MAAM,eAAe,WAAW,mBAAmB;GACnD,MAAM,YAAY,WAAW;AAC7B,OACE,EAAE,gBAAgB,SAClB,aAAa,SAAS,MACrB,cAAc,UAAa,UAAU,WAAW,IAEjD,QAAQ,KACN;IACE;IACA;IACA;IACA;GACD,EAAC,KAAK,IAAI,CACZ;AAEH,OAAI;AACF,QAAI,EAAE,gBAAgB,SAAS,cAAc,QAAW;KACtD,MAAM,CAACC,aAAW,iBAAiB,GACjCC,4CAAsB,aAAa;KACrC,WAAW,aAAaD,eAAa,CAAE;KACvC,WAAW,qBAAqB,oBAAoB,CAAE;IACvD,OAAM;KACL,WAAW,aAAa,WAAW,cAAc,CAAE;KACnD,WAAW,qBAAqB,WAAW,sBAAsB,CAAE;IACpE;GACF,SAAQ,GAAG;IAEV,WAAW,aAAa,CAAE;IAC1B,WAAW,qBAAqB,CAAE;GACnC;AACD,OAAI,WAAW,kBAAkB,QAAW;IAE1C,WAAW,cAAc,KACvB,GAAG,WAAW,WAAW,IAAI,CAAC,cAAc;KAC1C,MAAM;KACN,IAAI,SAAS;KACb,MAAM,SAAS;KACf,MAAM,SAAS;IAChB,GAAE,CACJ;IAED,MAAM,mBAAmB,WAAW,cACjC,OACC,CAAC,UAAU,MAAM,SAAS,YAC3B,CACA,OACC,CAAC,UACC,CAAC,WAAW,YAAY,KACtB,CAAC,aACC,SAAS,OAAO,MAAM,MAAM,SAAS,SAAS,MAAM,KACvD,CACJ;AACH,QAAI,iBAAiB,SAAS,GAC5B,WAAW,aAAa,iBAAiB,IAAI,CAAC,WAAW;KACvD,IAAI,MAAM;KACV,MAAM,MAAM;KACZ,MAAM,MAAM;IACb,GAAE;GAEN;EACF;EAGD,MAAM,WAAW;AACjB,MAAI,OAAO,eAAe,UAAU;GAClC,KAAK,aAAa,WAAW,cAAc,KAAK;GAChD,KAAK,qBACH,WAAW,sBAAsB,KAAK;EACzC;EACD,KAAK,iBAAiB,WAAW;CAClC;CAED,OAAO,UAAU;AACf,SAAO;CACR;CAED,WAAwB;AACtB,SAAO;CACR;CAED,IAAI,gBAA8C;AAChD,MAAI,KAAK,mBAAmB,mBAAmB,KAC7C,QAAO,KAAK;EAGd,MAAM,gBAAgB,KAAK,mBAAmB;AAC9C,MAAI,eAAe;GACjB,MAAM,aAAaE,4BAAc,cAAc;AAC/C,OAAI,WACF,QAAO,WAAW,iBAAiB,KAAK;EAE3C;EAED,MAAM,SAAS,MAAM;AAErB,MAAI,KAAK,YAAY;GACnB,MAAM,mBAAmB,KAAK,WAAW,OACvC,CAAC,UACC,CAAC,OAAO,KAAK,CAAC,MAAM,EAAE,OAAO,MAAM,MAAM,EAAE,SAAS,MAAM,KAAK,CAClE;GACD,OAAO,KACL,GAAG,iBAAiB,IAAI,CAAC,WAAW;IAClC,GAAG;IACH,MAAM;IACN,IAAI,MAAM;IACV,MAAM,MAAM;IACZ,MAAM,MAAM;GACb,GAAE,CACJ;EACF;AAED,SAAO;CACR;CAED,IAAa,mBAA4C;AACvD,SAAO;GACL,GAAG,MAAM;GACT,YAAY,KAAK;GACjB,oBAAoB,KAAK;GACzB,gBAAgB,KAAK;EACtB;CACF;AACF;AAED,SAAgB,YAAYC,GAAgC;AAC1D,QAAO,EAAE,UAAU,KAAK;AACzB;AAED,SAAgB,iBAAiBC,GAA0C;AACzE,QAAO,EAAE,UAAU,KAAK;AACzB;;;;;AAUD,IAAa,iBAAb,MAAa,uBAAuBC,8BAAiB;CAInD,aAA0B,CAAE;CAE5B,qBAAyC,CAAE;CAE3C,mBAAqC,CAAE;;;;CAKvC;CAEA,YAAYC,QAAuC;EACjD,IAAIC;AACJ,MAAI,OAAO,WAAW,UACpB,aAAa;GACX,SAAS;GACT,YAAY,CAAE;GACd,oBAAoB,CAAE;GACtB,kBAAkB,CAAE;EACrB;WACQ,OAAO,qBAAqB,QACrC,aAAa;GACX,GAAG;GACH,YAAY,OAAO,cAAc,CAAE;GACnC,oBAAoB,CAAE;GACtB,kBAAkB,CAAE;GACpB,gBACE,OAAO,mBAAmB,SACtB,OAAO,iBACP;EACP;OACI;GACL,MAAM,uBAAuB,OAAO,iBAAiB,OACnD,CAAC,KAAK,UAAU;IAGd,MAAM,UAAU,MAAM,MAAM,CAAC,SAAS,EAAE,MAAM,SAAS,GAAG;IAC1D,IAAI,WAAW,IAAI,YAAY,CAAE;IACjC,IAAI,SAAS,KAAK,MAAM;AACxB,WAAO;GACR,GACD,CAAE,EACH;GAED,MAAMC,YAAwB,CAAE;GAChC,MAAMC,mBAAsC,CAAE;AAC9C,QAAK,MAAM,CAAC,IAAI,OAAO,IAAI,OAAO,QAAQ,qBAAqB,EAAE;IAC/D,IAAI,aAAa,CAAE;IACnB,MAAM,OAAO,OAAO,IAAI,QAAQ;IAChC,MAAM,aAAa,OAAO,IAAI,CAAC,MAAM,EAAE,QAAQ,GAAG,CAAC,KAAK,GAAG;IAC3D,MAAM,UAAU,WAAW,SAAS,aAAa;IAEjD,MAAM,aAAa,OAAO,IAAI,MAAM;AACpC,QAAI;KACF,aAAaC,8BAAiB,QAAQ;AACtC,SACE,eAAe,QACf,OAAO,eAAe,YACtB,MAAM,QAAQ,WAAW,CAEzB,OAAM,IAAI,MAAM;KAElB,UAAU,KAAK;MACb;MACA,MAAM;MACN,IAAI;MACJ,MAAM;KACP,EAAC;IACH,SAAQ,GAAG;KACV,iBAAiB,KAAK;MACpB;MACA,MAAM;MACN,IAAI;MACJ,OAAO;MACP,MAAM;KACP,EAAC;IACH;GACF;GACD,aAAa;IACX,GAAG;IACH,YAAY;IACZ,oBAAoB;IACpB,gBACE,OAAO,mBAAmB,SACtB,OAAO,iBACP;GACP;EACF;EAGD,MAAM,WAAW;EACjB,KAAK,mBACH,WAAW,oBAAoB,KAAK;EACtC,KAAK,aAAa,WAAW,cAAc,KAAK;EAChD,KAAK,qBACH,WAAW,sBAAsB,KAAK;EACxC,KAAK,iBAAiB,WAAW;CAClC;CAED,IAAI,aAAqC;AAEvC,SAAO;GACL,GAAG,MAAM;GACT,YAAY;GACZ,oBAAoB;GACpB,kBAAkB;EACnB;CACF;CAED,OAAO,UAAU;AACf,SAAO;CACR;CAED,WAAwB;AACtB,SAAO;CACR;CAED,IAAI,gBAA8C;AAChD,MAAI,KAAK,mBAAmB,mBAAmB,KAC7C,QAAO,KAAK;EAGd,MAAM,gBAAgB,KAAK,mBAAmB;AAC9C,MAAI,eAAe;GACjB,MAAM,aAAaR,4BAAc,cAAc;AAC/C,OAAI,WACF,QAAO,WAAW,iBAAiB,KAAK;EAE3C;EAED,MAAM,SAAS,MAAM;AAErB,MAAI,KAAK,YACP;OAAI,OAAO,KAAK,YAAY,UAAU;IACpC,MAAM,mBAAmB,KAAK,QAC3B,OAAO,CAAC,UAAU,MAAM,SAAS,YAAY,CAC7C,IAAI,CAAC,UAAU,MAAM,GAAG;AAC3B,SAAK,MAAM,YAAY,KAAK,WAC1B,KAAI,SAAS,MAAM,CAAC,iBAAiB,SAAS,SAAS,GAAG,EACxD,OAAO,KAAK;KACV,GAAG;KACH,MAAM;KACN,IAAI,SAAS;KACb,MAAM,SAAS;KACf,MAAM,SAAS;IAChB,EAAC;GAGP;;AAGH,SAAO;CACR;CAED,IAAa,mBAA4C;AACvD,SAAO;GACL,GAAG,MAAM;GACT,YAAY,KAAK;GACjB,kBAAkB,KAAK;GACvB,oBAAoB,KAAK;GACzB,gBAAgB,KAAK;EACtB;CACF;CAED,OAAOS,OAAuB;EAC5B,MAAMC,iBAAuC;GAC3C,SAASC,0BAAa,KAAK,SAAS,MAAM,QAAQ;GAClD,mBAAmBC,yBACjB,KAAK,mBACL,MAAM,kBACP;GACD,mBAAmBA,yBACjB,KAAK,mBACL,MAAM,kBACP;GACD,kBAAkB,CAAE;GACpB,IAAI,KAAK,MAAM,MAAM;EACtB;AACD,MACE,KAAK,qBAAqB,UAC1B,MAAM,qBAAqB,QAC3B;GACA,MAAM,eAAeC,yBACnB,KAAK,kBACL,MAAM,iBACP;AACD,OAAI,iBAAiB,UAAa,aAAa,SAAS,GACtD,eAAe,mBAAmB;EAErC;AACD,MACE,KAAK,mBAAmB,UACxB,MAAM,mBAAmB,QACzB;GACA,MAAMC,oBAAuC;IAC3C,IAAK,KAAK,gBAAgB,qBAAqB,UAAU,UACvD,MAAM,gBAAgB,qBAAqB,UAAU,WAAc,EACnE,QACG,KAAK,gBAAgB,qBAAqB,SAAS,MACnD,MAAM,gBAAgB,qBAAqB,SAAS,GACxD;IACD,IAAK,KAAK,gBAAgB,qBAAqB,eAC7C,UACA,MAAM,gBAAgB,qBAAqB,eACzC,WAAc,EAChB,aACG,KAAK,gBAAgB,qBAAqB,cAAc,MACxD,MAAM,gBAAgB,qBAAqB,cAAc,GAC7D;IACD,IAAK,KAAK,gBAAgB,qBAAqB,mBAC7C,UACA,MAAM,gBAAgB,qBAAqB,mBACzC,WAAc,EAChB,iBACG,KAAK,gBAAgB,qBAAqB,kBAAkB,MAC5D,MAAM,gBAAgB,qBAAqB,kBAAkB,GACjE;GACF;GAED,MAAMC,qBAAyC;IAC7C,IAAK,KAAK,gBAAgB,sBAAsB,UAAU,UACxD,MAAM,gBAAgB,sBAAsB,UAAU,WAAc,EACpE,QACG,KAAK,gBAAgB,sBAAsB,SAAS,MACpD,MAAM,gBAAgB,sBAAsB,SAAS,GACzD;IACD,IAAK,KAAK,gBAAgB,sBAAsB,cAC9C,UACA,MAAM,gBAAgB,sBAAsB,cAC1C,WAAc,EAChB,YACG,KAAK,gBAAgB,sBAAsB,aAAa,MACxD,MAAM,gBAAgB,sBAAsB,aAAa,GAC7D;GACF;GAED,MAAMC,OAAsB,KAAK,kBAAkB;IACjD,cAAc;IACd,eAAe;IACf,cAAc;GACf;GACD,MAAMC,QAAuB,MAAM,kBAAkB;IACnD,cAAc;IACd,eAAe;IACf,cAAc;GACf;GACD,MAAMC,iBAAgC;IACpC,cAAc,KAAK,eAAe,MAAM;IACxC,eAAe,KAAK,gBAAgB,MAAM;IAC1C,cAAc,KAAK,eAAe,MAAM;IAGxC,GAAI,OAAO,KAAK,kBAAkB,CAAC,SAAS,KAAK,EAC/C,qBAAqB,kBACtB;IACD,GAAI,OAAO,KAAK,mBAAmB,CAAC,SAAS,KAAK,EAChD,sBAAsB,mBACvB;GACF;GACD,eAAe,iBAAiB;EACjC;AACD,SAAO,IAAI,eAAe;CAC3B;AACF"}
+{"version":3,"file":"ai.cjs","names":["BaseMessage","fields: $InferMessageContent<TStructure, \"ai\"> | AIMessageFields<TStructure>","initParams: AIMessageFields<TStructure>","toolCalls","defaultToolCallParser","getTranslator","obj: unknown","x: BaseMessage","x: BaseMessageChunk","BaseMessageChunk","fields:\n      | $InferMessageContent<TStructure, \"ai\">\n      | AIMessageChunkFields<TStructure>","initParams: AIMessageChunkFields<TStructure>","toolCalls: ToolCall[]","invalidToolCalls: InvalidToolCall[]","parsedArgs: Record<string, unknown> | null","parsePartialJson","chunk: AIMessageChunk<TStructure>","combinedFields: AIMessageChunkFields","mergeContent","_mergeDicts","mergeResponseMetadata","_mergeLists","mergeUsageMetadata"],"sources":["../../src/messages/ai.ts"],"sourcesContent":["import { parsePartialJson } from \"../utils/json.js\";\nimport {\n  BaseMessage,\n  BaseMessageChunk,\n  mergeContent,\n  _mergeDicts,\n  _mergeLists,\n  BaseMessageFields,\n} from \"./base.js\";\nimport { getTranslator } from \"./block_translators/index.js\";\nimport { ContentBlock } from \"./content/index.js\";\nimport {\n  $InferMessageContent,\n  $InferMessageProperty,\n  MessageStructure,\n} from \"./message.js\";\nimport { mergeResponseMetadata, mergeUsageMetadata } from \"./metadata.js\";\nimport {\n  InvalidToolCall,\n  ToolCall,\n  ToolCallChunk,\n  defaultToolCallParser,\n} from \"./tool.js\";\nimport { Constructor } from \"./utils.js\";\n\nexport interface AIMessageFields<\n  TStructure extends MessageStructure = MessageStructure\n> extends BaseMessageFields<TStructure, \"ai\"> {\n  tool_calls?: ToolCall[];\n  invalid_tool_calls?: InvalidToolCall[];\n  usage_metadata?: $InferMessageProperty<TStructure, \"ai\", \"usage_metadata\">;\n}\n\nexport class AIMessage<TStructure extends MessageStructure = MessageStructure>\n  extends BaseMessage<TStructure, \"ai\">\n  implements AIMessageFields<TStructure>\n{\n  readonly type = \"ai\" as const;\n\n  tool_calls?: ToolCall[] = [];\n\n  invalid_tool_calls?: InvalidToolCall[] = [];\n\n  usage_metadata?: AIMessageFields<TStructure>[\"usage_metadata\"];\n\n  get lc_aliases(): Record<string, string> {\n    // exclude snake case conversion to pascal case\n    return {\n      ...super.lc_aliases,\n      tool_calls: \"tool_calls\",\n      invalid_tool_calls: \"invalid_tool_calls\",\n    };\n  }\n\n  constructor(\n    fields: $InferMessageContent<TStructure, \"ai\"> | AIMessageFields<TStructure>\n  ) {\n    let initParams: AIMessageFields<TStructure>;\n    if (typeof fields === \"string\" || Array.isArray(fields)) {\n      initParams = {\n        content: fields,\n        tool_calls: [],\n        invalid_tool_calls: [],\n        additional_kwargs: {},\n      };\n    } else {\n      initParams = fields;\n      const rawToolCalls = initParams.additional_kwargs?.tool_calls;\n      const toolCalls = initParams.tool_calls;\n      if (\n        !(rawToolCalls == null) &&\n        rawToolCalls.length > 0 &&\n        (toolCalls === undefined || toolCalls.length === 0)\n      ) {\n        console.warn(\n          [\n            \"New LangChain packages are available that more efficiently handle\",\n            \"tool calling.\\n\\nPlease upgrade your packages to versions that set\",\n            \"message tool calls. e.g., `pnpm install @langchain/anthropic`,\",\n            \"pnpm install @langchain/openai`, etc.\",\n          ].join(\" \")\n        );\n      }\n      try {\n        if (!(rawToolCalls == null) && toolCalls === undefined) {\n          const [toolCalls, invalidToolCalls] =\n            defaultToolCallParser(rawToolCalls);\n          initParams.tool_calls = toolCalls ?? [];\n          initParams.invalid_tool_calls = invalidToolCalls ?? [];\n        } else {\n          initParams.tool_calls = initParams.tool_calls ?? [];\n          initParams.invalid_tool_calls = initParams.invalid_tool_calls ?? [];\n        }\n      } catch {\n        // Do nothing if parsing fails\n        initParams.tool_calls = [];\n        initParams.invalid_tool_calls = [];\n      }\n      if (initParams.contentBlocks !== undefined) {\n        // Add constructor tool calls as content blocks\n        initParams.contentBlocks.push(\n          ...initParams.tool_calls.map((toolCall) => ({\n            type: \"tool_call\" as const,\n            id: toolCall.id,\n            name: toolCall.name,\n            args: toolCall.args,\n          }))\n        );\n        // Add content block tool calls that aren't in the constructor tool calls\n        const missingToolCalls = initParams.contentBlocks\n          .filter<ContentBlock.Tools.ToolCall>(\n            (block): block is ContentBlock.Tools.ToolCall =>\n              block.type === \"tool_call\"\n          )\n          .filter(\n            (block) =>\n              !initParams.tool_calls?.some(\n                (toolCall) =>\n                  toolCall.id === block.id && toolCall.name === block.name\n              )\n          );\n        if (missingToolCalls.length > 0) {\n          initParams.tool_calls = missingToolCalls.map((block) => ({\n            type: \"tool_call\" as const,\n            id: block.id!,\n            name: block.name,\n            args: block.args as Record<string, unknown>,\n          }));\n        }\n      }\n    }\n    // Sadly, TypeScript only allows super() calls at root if the class has\n    // properties with initializers, so we have to check types twice.\n    super(initParams);\n    if (typeof initParams !== \"string\") {\n      this.tool_calls = initParams.tool_calls ?? this.tool_calls;\n      this.invalid_tool_calls =\n        initParams.invalid_tool_calls ?? this.invalid_tool_calls;\n    }\n    this.usage_metadata = initParams.usage_metadata;\n  }\n\n  static lc_name() {\n    return \"AIMessage\";\n  }\n\n  get contentBlocks(): Array<ContentBlock.Standard> {\n    if (\n      \"output_version\" in this.response_metadata &&\n      this.response_metadata.output_version === \"v1\"\n    ) {\n      return this.content as Array<ContentBlock.Standard>;\n    }\n\n    if (\n      \"model_provider\" in this.response_metadata &&\n      typeof this.response_metadata.model_provider === \"string\"\n    ) {\n      const translator = getTranslator(this.response_metadata.model_provider);\n      if (translator) {\n        return translator.translateContent(this);\n      }\n    }\n\n    const blocks = super.contentBlocks;\n\n    if (this.tool_calls) {\n      const missingToolCalls = this.tool_calls.filter(\n        (block) =>\n          !blocks.some((b) => b.id === block.id && b.name === block.name)\n      );\n      blocks.push(\n        ...missingToolCalls.map((block) => ({\n          ...block,\n          type: \"tool_call\" as const,\n          id: block.id,\n          name: block.name,\n          args: block.args,\n        }))\n      );\n    }\n\n    return blocks;\n  }\n\n  override get _printableFields(): Record<string, unknown> {\n    return {\n      ...super._printableFields,\n      tool_calls: this.tool_calls,\n      invalid_tool_calls: this.invalid_tool_calls,\n      usage_metadata: this.usage_metadata,\n    };\n  }\n\n  static isInstance(obj: unknown): obj is AIMessage {\n    return super.isInstance(obj) && obj.type === \"ai\";\n  }\n}\n\n/**\n * @deprecated Use {@link AIMessage.isInstance} instead\n */\nexport function isAIMessage<TStructure extends MessageStructure>(\n  x: BaseMessage\n): x is AIMessage<TStructure> {\n  return x._getType() === \"ai\";\n}\n\n/**\n * @deprecated Use {@link AIMessageChunk.isInstance} instead\n */\nexport function isAIMessageChunk<TStructure extends MessageStructure>(\n  x: BaseMessageChunk\n): x is AIMessageChunk<TStructure> {\n  return x._getType() === \"ai\";\n}\n\nexport type AIMessageChunkFields<\n  TStructure extends MessageStructure = MessageStructure\n> = AIMessageFields<TStructure> & {\n  tool_call_chunks?: ToolCallChunk[];\n};\n\n/**\n * Represents a chunk of an AI message, which can be concatenated with\n * other AI message chunks.\n */\nexport class AIMessageChunk<\n    TStructure extends MessageStructure = MessageStructure\n  >\n  extends BaseMessageChunk<TStructure, \"ai\">\n  implements AIMessage<TStructure>, AIMessageChunkFields<TStructure>\n{\n  readonly type = \"ai\" as const;\n\n  tool_calls?: ToolCall[] = [];\n\n  invalid_tool_calls?: InvalidToolCall[] = [];\n\n  tool_call_chunks?: ToolCallChunk[] = [];\n\n  usage_metadata?: AIMessageChunkFields<TStructure>[\"usage_metadata\"];\n\n  constructor(\n    fields:\n      | $InferMessageContent<TStructure, \"ai\">\n      | AIMessageChunkFields<TStructure>\n  ) {\n    let initParams: AIMessageChunkFields<TStructure>;\n    if (typeof fields === \"string\" || Array.isArray(fields)) {\n      initParams = {\n        content: fields,\n        tool_calls: [],\n        invalid_tool_calls: [],\n        tool_call_chunks: [],\n      };\n    } else if (fields.tool_call_chunks === undefined) {\n      initParams = {\n        ...fields,\n        tool_calls: fields.tool_calls ?? [],\n        invalid_tool_calls: [],\n        tool_call_chunks: [],\n        usage_metadata:\n          fields.usage_metadata !== undefined\n            ? fields.usage_metadata\n            : undefined,\n      };\n    } else {\n      const groupedToolCallChunk = fields.tool_call_chunks.reduce(\n        (acc, chunk) => {\n          // Assign a fallback ID if the chunk doesn't have one\n          // This can happen with tools that have empty schemas\n          const chunkId = chunk.id || `fallback-${chunk.index || 0}`;\n          acc[chunkId] = acc[chunkId] ?? [];\n          acc[chunkId].push(chunk);\n          return acc;\n        },\n        {} as Record<string, ToolCallChunk[]>\n      );\n\n      const toolCalls: ToolCall[] = [];\n      const invalidToolCalls: InvalidToolCall[] = [];\n      for (const [id, chunks] of Object.entries(groupedToolCallChunk)) {\n        let parsedArgs: Record<string, unknown> | null = null;\n        const name = chunks[0]?.name ?? \"\";\n        const joinedArgs = chunks.map((c) => c.args || \"\").join(\"\");\n        const argsStr = joinedArgs.length ? joinedArgs : \"{}\";\n        // Use the original ID from the first chunk if it exists, otherwise use the grouped ID\n        const originalId = chunks[0]?.id || id;\n        try {\n          parsedArgs = parsePartialJson(argsStr);\n          if (\n            parsedArgs === null ||\n            typeof parsedArgs !== \"object\" ||\n            Array.isArray(parsedArgs)\n          ) {\n            throw new Error(\"Malformed tool call chunk args.\");\n          }\n          toolCalls.push({\n            name,\n            args: parsedArgs,\n            id: originalId,\n            type: \"tool_call\",\n          });\n        } catch {\n          invalidToolCalls.push({\n            name,\n            args: argsStr,\n            id: originalId,\n            error: \"Malformed args.\",\n            type: \"invalid_tool_call\",\n          });\n        }\n      }\n      initParams = {\n        ...fields,\n        tool_calls: toolCalls,\n        invalid_tool_calls: invalidToolCalls,\n        usage_metadata:\n          fields.usage_metadata !== undefined\n            ? fields.usage_metadata\n            : undefined,\n      };\n    }\n    // Sadly, TypeScript only allows super() calls at root if the class has\n    // properties with initializers, so we have to check types twice.\n    super(initParams);\n    this.tool_call_chunks =\n      initParams.tool_call_chunks ?? this.tool_call_chunks;\n    this.tool_calls = initParams.tool_calls ?? this.tool_calls;\n    this.invalid_tool_calls =\n      initParams.invalid_tool_calls ?? this.invalid_tool_calls;\n    this.usage_metadata = initParams.usage_metadata;\n  }\n\n  get lc_aliases(): Record<string, string> {\n    // exclude snake case conversion to pascal case\n    return {\n      ...super.lc_aliases,\n      tool_calls: \"tool_calls\",\n      invalid_tool_calls: \"invalid_tool_calls\",\n      tool_call_chunks: \"tool_call_chunks\",\n    };\n  }\n\n  static lc_name() {\n    return \"AIMessageChunk\";\n  }\n\n  get contentBlocks(): Array<ContentBlock.Standard> {\n    if (\n      \"output_version\" in this.response_metadata &&\n      this.response_metadata.output_version === \"v1\"\n    ) {\n      return this.content as Array<ContentBlock.Standard>;\n    }\n\n    if (\n      \"model_provider\" in this.response_metadata &&\n      typeof this.response_metadata.model_provider === \"string\"\n    ) {\n      const translator = getTranslator(this.response_metadata.model_provider);\n      if (translator) {\n        return translator.translateContent(this);\n      }\n    }\n\n    const blocks = super.contentBlocks;\n\n    if (this.tool_calls) {\n      if (typeof this.content !== \"string\") {\n        const contentToolCalls = this.content\n          .filter((block) => block.type === \"tool_call\")\n          .map((block) => block.id);\n        for (const toolCall of this.tool_calls) {\n          if (toolCall.id && !contentToolCalls.includes(toolCall.id)) {\n            blocks.push({\n              ...toolCall,\n              type: \"tool_call\",\n              id: toolCall.id,\n              name: toolCall.name,\n              args: toolCall.args,\n            });\n          }\n        }\n      }\n    }\n\n    return blocks;\n  }\n\n  override get _printableFields(): Record<string, unknown> {\n    return {\n      ...super._printableFields,\n      tool_calls: this.tool_calls,\n      tool_call_chunks: this.tool_call_chunks,\n      invalid_tool_calls: this.invalid_tool_calls,\n      usage_metadata: this.usage_metadata,\n    };\n  }\n\n  concat(chunk: AIMessageChunk<TStructure>) {\n    const combinedFields: AIMessageChunkFields = {\n      content: mergeContent(this.content, chunk.content),\n      additional_kwargs: _mergeDicts(\n        this.additional_kwargs,\n        chunk.additional_kwargs\n      ),\n      response_metadata: mergeResponseMetadata(\n        this.response_metadata,\n        chunk.response_metadata\n      ),\n      tool_call_chunks: [],\n      id: this.id ?? chunk.id,\n    };\n    if (\n      this.tool_call_chunks !== undefined ||\n      chunk.tool_call_chunks !== undefined\n    ) {\n      const rawToolCalls = _mergeLists(\n        this.tool_call_chunks as ContentBlock.Tools.ToolCallChunk[],\n        chunk.tool_call_chunks as ContentBlock.Tools.ToolCallChunk[]\n      );\n      if (rawToolCalls !== undefined && rawToolCalls.length > 0) {\n        combinedFields.tool_call_chunks = rawToolCalls;\n      }\n    }\n    if (\n      this.usage_metadata !== undefined ||\n      chunk.usage_metadata !== undefined\n    ) {\n      combinedFields.usage_metadata = mergeUsageMetadata(\n        this.usage_metadata,\n        chunk.usage_metadata\n      );\n    }\n    const Cls = this.constructor as Constructor<this>;\n    return new Cls(combinedFields);\n  }\n\n  static isInstance(obj: unknown): obj is AIMessageChunk {\n    return super.isInstance(obj) && obj.type === \"ai\";\n  }\n}\n"],"mappings":";;;;;;;AAiCA,IAAa,YAAb,cACUA,yBAEV;CACE,AAAS,OAAO;CAEhB,aAA0B,CAAE;CAE5B,qBAAyC,CAAE;CAE3C;CAEA,IAAI,aAAqC;AAEvC,SAAO;GACL,GAAG,MAAM;GACT,YAAY;GACZ,oBAAoB;EACrB;CACF;CAED,YACEC,QACA;EACA,IAAIC;AACJ,MAAI,OAAO,WAAW,YAAY,MAAM,QAAQ,OAAO,EACrD,aAAa;GACX,SAAS;GACT,YAAY,CAAE;GACd,oBAAoB,CAAE;GACtB,mBAAmB,CAAE;EACtB;OACI;GACL,aAAa;GACb,MAAM,eAAe,WAAW,mBAAmB;GACnD,MAAM,YAAY,WAAW;AAC7B,OACE,EAAE,gBAAgB,SAClB,aAAa,SAAS,MACrB,cAAc,UAAa,UAAU,WAAW,IAEjD,QAAQ,KACN;IACE;IACA;IACA;IACA;GACD,EAAC,KAAK,IAAI,CACZ;AAEH,OAAI;AACF,QAAI,EAAE,gBAAgB,SAAS,cAAc,QAAW;KACtD,MAAM,CAACC,aAAW,iBAAiB,GACjCC,4CAAsB,aAAa;KACrC,WAAW,aAAaD,eAAa,CAAE;KACvC,WAAW,qBAAqB,oBAAoB,CAAE;IACvD,OAAM;KACL,WAAW,aAAa,WAAW,cAAc,CAAE;KACnD,WAAW,qBAAqB,WAAW,sBAAsB,CAAE;IACpE;GACF,QAAO;IAEN,WAAW,aAAa,CAAE;IAC1B,WAAW,qBAAqB,CAAE;GACnC;AACD,OAAI,WAAW,kBAAkB,QAAW;IAE1C,WAAW,cAAc,KACvB,GAAG,WAAW,WAAW,IAAI,CAAC,cAAc;KAC1C,MAAM;KACN,IAAI,SAAS;KACb,MAAM,SAAS;KACf,MAAM,SAAS;IAChB,GAAE,CACJ;IAED,MAAM,mBAAmB,WAAW,cACjC,OACC,CAAC,UACC,MAAM,SAAS,YAClB,CACA,OACC,CAAC,UACC,CAAC,WAAW,YAAY,KACtB,CAAC,aACC,SAAS,OAAO,MAAM,MAAM,SAAS,SAAS,MAAM,KACvD,CACJ;AACH,QAAI,iBAAiB,SAAS,GAC5B,WAAW,aAAa,iBAAiB,IAAI,CAAC,WAAW;KACvD,MAAM;KACN,IAAI,MAAM;KACV,MAAM,MAAM;KACZ,MAAM,MAAM;IACb,GAAE;GAEN;EACF;EAGD,MAAM,WAAW;AACjB,MAAI,OAAO,eAAe,UAAU;GAClC,KAAK,aAAa,WAAW,cAAc,KAAK;GAChD,KAAK,qBACH,WAAW,sBAAsB,KAAK;EACzC;EACD,KAAK,iBAAiB,WAAW;CAClC;CAED,OAAO,UAAU;AACf,SAAO;CACR;CAED,IAAI,gBAA8C;AAChD,MACE,oBAAoB,KAAK,qBACzB,KAAK,kBAAkB,mBAAmB,KAE1C,QAAO,KAAK;AAGd,MACE,oBAAoB,KAAK,qBACzB,OAAO,KAAK,kBAAkB,mBAAmB,UACjD;GACA,MAAM,aAAaE,4BAAc,KAAK,kBAAkB,eAAe;AACvE,OAAI,WACF,QAAO,WAAW,iBAAiB,KAAK;EAE3C;EAED,MAAM,SAAS,MAAM;AAErB,MAAI,KAAK,YAAY;GACnB,MAAM,mBAAmB,KAAK,WAAW,OACvC,CAAC,UACC,CAAC,OAAO,KAAK,CAAC,MAAM,EAAE,OAAO,MAAM,MAAM,EAAE,SAAS,MAAM,KAAK,CAClE;GACD,OAAO,KACL,GAAG,iBAAiB,IAAI,CAAC,WAAW;IAClC,GAAG;IACH,MAAM;IACN,IAAI,MAAM;IACV,MAAM,MAAM;IACZ,MAAM,MAAM;GACb,GAAE,CACJ;EACF;AAED,SAAO;CACR;CAED,IAAa,mBAA4C;AACvD,SAAO;GACL,GAAG,MAAM;GACT,YAAY,KAAK;GACjB,oBAAoB,KAAK;GACzB,gBAAgB,KAAK;EACtB;CACF;CAED,OAAO,WAAWC,KAAgC;AAChD,SAAO,MAAM,WAAW,IAAI,IAAI,IAAI,SAAS;CAC9C;AACF;;;;AAKD,SAAgB,YACdC,GAC4B;AAC5B,QAAO,EAAE,UAAU,KAAK;AACzB;;;;AAKD,SAAgB,iBACdC,GACiC;AACjC,QAAO,EAAE,UAAU,KAAK;AACzB;;;;;AAYD,IAAa,iBAAb,cAGUC,8BAEV;CACE,AAAS,OAAO;CAEhB,aAA0B,CAAE;CAE5B,qBAAyC,CAAE;CAE3C,mBAAqC,CAAE;CAEvC;CAEA,YACEC,QAGA;EACA,IAAIC;AACJ,MAAI,OAAO,WAAW,YAAY,MAAM,QAAQ,OAAO,EACrD,aAAa;GACX,SAAS;GACT,YAAY,CAAE;GACd,oBAAoB,CAAE;GACtB,kBAAkB,CAAE;EACrB;WACQ,OAAO,qBAAqB,QACrC,aAAa;GACX,GAAG;GACH,YAAY,OAAO,cAAc,CAAE;GACnC,oBAAoB,CAAE;GACtB,kBAAkB,CAAE;GACpB,gBACE,OAAO,mBAAmB,SACtB,OAAO,iBACP;EACP;OACI;GACL,MAAM,uBAAuB,OAAO,iBAAiB,OACnD,CAAC,KAAK,UAAU;IAGd,MAAM,UAAU,MAAM,MAAM,CAAC,SAAS,EAAE,MAAM,SAAS,GAAG;IAC1D,IAAI,WAAW,IAAI,YAAY,CAAE;IACjC,IAAI,SAAS,KAAK,MAAM;AACxB,WAAO;GACR,GACD,CAAE,EACH;GAED,MAAMC,YAAwB,CAAE;GAChC,MAAMC,mBAAsC,CAAE;AAC9C,QAAK,MAAM,CAAC,IAAI,OAAO,IAAI,OAAO,QAAQ,qBAAqB,EAAE;IAC/D,IAAIC,aAA6C;IACjD,MAAM,OAAO,OAAO,IAAI,QAAQ;IAChC,MAAM,aAAa,OAAO,IAAI,CAAC,MAAM,EAAE,QAAQ,GAAG,CAAC,KAAK,GAAG;IAC3D,MAAM,UAAU,WAAW,SAAS,aAAa;IAEjD,MAAM,aAAa,OAAO,IAAI,MAAM;AACpC,QAAI;KACF,aAAaC,8BAAiB,QAAQ;AACtC,SACE,eAAe,QACf,OAAO,eAAe,YACtB,MAAM,QAAQ,WAAW,CAEzB,OAAM,IAAI,MAAM;KAElB,UAAU,KAAK;MACb;MACA,MAAM;MACN,IAAI;MACJ,MAAM;KACP,EAAC;IACH,QAAO;KACN,iBAAiB,KAAK;MACpB;MACA,MAAM;MACN,IAAI;MACJ,OAAO;MACP,MAAM;KACP,EAAC;IACH;GACF;GACD,aAAa;IACX,GAAG;IACH,YAAY;IACZ,oBAAoB;IACpB,gBACE,OAAO,mBAAmB,SACtB,OAAO,iBACP;GACP;EACF;EAGD,MAAM,WAAW;EACjB,KAAK,mBACH,WAAW,oBAAoB,KAAK;EACtC,KAAK,aAAa,WAAW,cAAc,KAAK;EAChD,KAAK,qBACH,WAAW,sBAAsB,KAAK;EACxC,KAAK,iBAAiB,WAAW;CAClC;CAED,IAAI,aAAqC;AAEvC,SAAO;GACL,GAAG,MAAM;GACT,YAAY;GACZ,oBAAoB;GACpB,kBAAkB;EACnB;CACF;CAED,OAAO,UAAU;AACf,SAAO;CACR;CAED,IAAI,gBAA8C;AAChD,MACE,oBAAoB,KAAK,qBACzB,KAAK,kBAAkB,mBAAmB,KAE1C,QAAO,KAAK;AAGd,MACE,oBAAoB,KAAK,qBACzB,OAAO,KAAK,kBAAkB,mBAAmB,UACjD;GACA,MAAM,aAAaV,4BAAc,KAAK,kBAAkB,eAAe;AACvE,OAAI,WACF,QAAO,WAAW,iBAAiB,KAAK;EAE3C;EAED,MAAM,SAAS,MAAM;AAErB,MAAI,KAAK,YACP;OAAI,OAAO,KAAK,YAAY,UAAU;IACpC,MAAM,mBAAmB,KAAK,QAC3B,OAAO,CAAC,UAAU,MAAM,SAAS,YAAY,CAC7C,IAAI,CAAC,UAAU,MAAM,GAAG;AAC3B,SAAK,MAAM,YAAY,KAAK,WAC1B,KAAI,SAAS,MAAM,CAAC,iBAAiB,SAAS,SAAS,GAAG,EACxD,OAAO,KAAK;KACV,GAAG;KACH,MAAM;KACN,IAAI,SAAS;KACb,MAAM,SAAS;KACf,MAAM,SAAS;IAChB,EAAC;GAGP;;AAGH,SAAO;CACR;CAED,IAAa,mBAA4C;AACvD,SAAO;GACL,GAAG,MAAM;GACT,YAAY,KAAK;GACjB,kBAAkB,KAAK;GACvB,oBAAoB,KAAK;GACzB,gBAAgB,KAAK;EACtB;CACF;CAED,OAAOW,OAAmC;EACxC,MAAMC,iBAAuC;GAC3C,SAASC,0BAAa,KAAK,SAAS,MAAM,QAAQ;GAClD,mBAAmBC,yBACjB,KAAK,mBACL,MAAM,kBACP;GACD,mBAAmBC,uCACjB,KAAK,mBACL,MAAM,kBACP;GACD,kBAAkB,CAAE;GACpB,IAAI,KAAK,MAAM,MAAM;EACtB;AACD,MACE,KAAK,qBAAqB,UAC1B,MAAM,qBAAqB,QAC3B;GACA,MAAM,eAAeC,yBACnB,KAAK,kBACL,MAAM,iBACP;AACD,OAAI,iBAAiB,UAAa,aAAa,SAAS,GACtD,eAAe,mBAAmB;EAErC;AACD,MACE,KAAK,mBAAmB,UACxB,MAAM,mBAAmB,QAEzB,eAAe,iBAAiBC,oCAC9B,KAAK,gBACL,MAAM,eACP;EAEH,MAAM,MAAM,KAAK;AACjB,SAAO,IAAI,IAAI;CAChB;CAED,OAAO,WAAWhB,KAAqC;AACrD,SAAO,MAAM,WAAW,IAAI,IAAI,IAAI,SAAS;CAC9C;AACF"}

package/dist/messages/ai.d.cts

@@ -1,148 +1,55 @@
 import { ContentBlock } from "./content/index.cjs";
-import { BaseMessage, BaseMessageChunk, BaseMessageFields, MessageType } from "./base.cjs";
+import { BaseMessage, BaseMessageChunk, BaseMessageFields } from "./base.cjs";
 import { InvalidToolCall, ToolCall, ToolCallChunk } from "./tool.cjs";
+import { $InferMessageContent, $InferMessageProperty, MessageStructure } from "./message.cjs";
 
 //#region src/messages/ai.d.ts
-type AIMessageFields = BaseMessageFields & {
+interface AIMessageFields<TStructure extends MessageStructure = MessageStructure> extends BaseMessageFields<TStructure, "ai"> {
   tool_calls?: ToolCall[];
   invalid_tool_calls?: InvalidToolCall[];
-  usage_metadata?: UsageMetadata;
-};
-type ModalitiesTokenDetails = {
-  /**
-   * Text tokens.
-   * Does not need to be reported, but some models will do so.
-   */
-  text?: number;
-  /**
-   * Image (non-video) tokens.
-   */
-  image?: number;
-  /**
-   * Audio tokens.
-   */
-  audio?: number;
-  /**
-   * Video tokens.
-   */
-  video?: number;
-  /**
-   * Document tokens.
-   * e.g. PDF
-   */
-  document?: number;
-};
-/**
- * Breakdown of input token counts.
- *
- * Does not *need* to sum to full input token count. Does *not* need to have all keys.
- */
-type InputTokenDetails = ModalitiesTokenDetails & {
-  /**
-   * Input tokens that were cached and there was a cache hit.
-   *
-   * Since there was a cache hit, the tokens were read from the cache.
-   * More precisely, the model state given these tokens was read from the cache.
-   */
-  cache_read?: number;
-  /**
-   * Input tokens that were cached and there was a cache miss.
-   *
-   * Since there was a cache miss, the cache was created from these tokens.
-   */
-  cache_creation?: number;
-};
-/**
- * Breakdown of output token counts.
- *
- * Does *not* need to sum to full output token count. Does *not* need to have all keys.
- */
-type OutputTokenDetails = ModalitiesTokenDetails & {
-  /**
-   * Reasoning output tokens.
-   *
-   * Tokens generated by the model in a chain of thought process (i.e. by
-   * OpenAI's o1 models) that are not returned as part of model output.
-   */
-  reasoning?: number;
-};
-/**
- * Usage metadata for a message, such as token counts.
- */
-type UsageMetadata = {
-  /**
-   * Count of input (or prompt) tokens. Sum of all input token types.
-   */
-  input_tokens: number;
-  /**
-   * Count of output (or completion) tokens. Sum of all output token types.
-   */
-  output_tokens: number;
-  /**
-   * Total token count. Sum of input_tokens + output_tokens.
-   */
-  total_tokens: number;
-  /**
-   * Breakdown of input token counts.
-   *
-   * Does *not* need to sum to full input token count. Does *not* need to have all keys.
-   */
-  input_token_details?: InputTokenDetails;
-  /**
-   * Breakdown of output token counts.
-   *
-   * Does *not* need to sum to full output token count. Does *not* need to have all keys.
-   */
-  output_token_details?: OutputTokenDetails;
-};
-/**
- * Represents an AI message in a conversation.
- */
-declare class AIMessage extends BaseMessage implements AIMessageFields {
-  // These are typed as optional to avoid breaking changes and allow for casting
-  // from BaseMessage.
+  usage_metadata?: $InferMessageProperty<TStructure, "ai", "usage_metadata">;
+}
+declare class AIMessage<TStructure extends MessageStructure = MessageStructure> extends BaseMessage<TStructure, "ai"> implements AIMessageFields<TStructure> {
+  readonly type: "ai";
   tool_calls?: ToolCall[];
   invalid_tool_calls?: InvalidToolCall[];
-  /**
-   * If provided, token usage information associated with the message.
-   */
-  usage_metadata?: UsageMetadata;
+  usage_metadata?: AIMessageFields<TStructure>["usage_metadata"];
   get lc_aliases(): Record<string, string>;
-  constructor(fields: string | AIMessageFields, /** @deprecated */
-  kwargs?: Record<string, unknown>);
+  constructor(fields: $InferMessageContent<TStructure, "ai"> | AIMessageFields<TStructure>);
   static lc_name(): string;
-  _getType(): MessageType;
   get contentBlocks(): Array<ContentBlock.Standard>;
   get _printableFields(): Record<string, unknown>;
+  static isInstance(obj: unknown): obj is AIMessage;
 }
-declare function isAIMessage(x: BaseMessage): x is AIMessage;
-declare function isAIMessageChunk(x: BaseMessageChunk): x is AIMessageChunk;
-type AIMessageChunkFields = AIMessageFields & {
+/**
+ * @deprecated Use {@link AIMessage.isInstance} instead
+ */
+declare function isAIMessage<TStructure extends MessageStructure>(x: BaseMessage): x is AIMessage<TStructure>;
+/**
+ * @deprecated Use {@link AIMessageChunk.isInstance} instead
+ */
+declare function isAIMessageChunk<TStructure extends MessageStructure>(x: BaseMessageChunk): x is AIMessageChunk<TStructure>;
+type AIMessageChunkFields<TStructure extends MessageStructure = MessageStructure> = AIMessageFields<TStructure> & {
   tool_call_chunks?: ToolCallChunk[];
 };
 /**
  * Represents a chunk of an AI message, which can be concatenated with
  * other AI message chunks.
  */
-declare class AIMessageChunk extends BaseMessageChunk {
-  // Must redeclare tool call fields since there is no multiple inheritance in JS.
-  // These are typed as optional to avoid breaking changes and allow for casting
-  // from BaseMessage.
+declare class AIMessageChunk<TStructure extends MessageStructure = MessageStructure> extends BaseMessageChunk<TStructure, "ai"> implements AIMessage<TStructure>, AIMessageChunkFields<TStructure> {
+  readonly type: "ai";
   tool_calls?: ToolCall[];
   invalid_tool_calls?: InvalidToolCall[];
   tool_call_chunks?: ToolCallChunk[];
-  /**
-   * If provided, token usage information associated with the message.
-   */
-  usage_metadata?: UsageMetadata;
-  constructor(fields: string | AIMessageChunkFields);
+  usage_metadata?: AIMessageChunkFields<TStructure>["usage_metadata"];
+  constructor(fields: $InferMessageContent<TStructure, "ai"> | AIMessageChunkFields<TStructure>);
   get lc_aliases(): Record<string, string>;
   static lc_name(): string;
-  _getType(): MessageType;
   get contentBlocks(): Array<ContentBlock.Standard>;
   get _printableFields(): Record<string, unknown>;
-  concat(chunk: AIMessageChunk): AIMessageChunk;
+  concat(chunk: AIMessageChunk<TStructure>): this;
+  static isInstance(obj: unknown): obj is AIMessageChunk;
 }
 //#endregion
-export { AIMessage, AIMessageChunk, AIMessageChunkFields, AIMessageFields, InputTokenDetails, ModalitiesTokenDetails, OutputTokenDetails, UsageMetadata, isAIMessage, isAIMessageChunk };
+export { AIMessage, AIMessageChunk, AIMessageChunkFields, AIMessageFields, isAIMessage, isAIMessageChunk };
 //# sourceMappingURL=ai.d.cts.map

package/dist/messages/ai.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"ai.d.cts","names":["BaseMessage","BaseMessageChunk","MessageType","BaseMessageFields","ContentBlock","InvalidToolCall","ToolCall","ToolCallChunk","AIMessageFields","UsageMetadata","ModalitiesTokenDetails","InputTokenDetails","OutputTokenDetails","AIMessage","Record","Standard","Array","isAIMessage","isAIMessageChunk","AIMessageChunk","AIMessageChunkFields"],"sources":["../../src/messages/ai.d.ts"],"sourcesContent":["import { BaseMessage, BaseMessageChunk, type MessageType, BaseMessageFields } from \"./base.js\";\nimport { ContentBlock } from \"./content/index.js\";\nimport { InvalidToolCall, ToolCall, ToolCallChunk } from \"./tool.js\";\nexport type AIMessageFields = BaseMessageFields & {\n    tool_calls?: ToolCall[];\n    invalid_tool_calls?: InvalidToolCall[];\n    usage_metadata?: UsageMetadata;\n};\nexport type ModalitiesTokenDetails = {\n    /**\n     * Text tokens.\n     * Does not need to be reported, but some models will do so.\n     */\n    text?: number;\n    /**\n     * Image (non-video) tokens.\n     */\n    image?: number;\n    /**\n     * Audio tokens.\n     */\n    audio?: number;\n    /**\n     * Video tokens.\n     */\n    video?: number;\n    /**\n     * Document tokens.\n     * e.g. PDF\n     */\n    document?: number;\n};\n/**\n * Breakdown of input token counts.\n *\n * Does not *need* to sum to full input token count. Does *not* need to have all keys.\n */\nexport type InputTokenDetails = ModalitiesTokenDetails & {\n    /**\n     * Input tokens that were cached and there was a cache hit.\n     *\n     * Since there was a cache hit, the tokens were read from the cache.\n     * More precisely, the model state given these tokens was read from the cache.\n     */\n    cache_read?: number;\n    /**\n     * Input tokens that were cached and there was a cache miss.\n     *\n     * Since there was a cache miss, the cache was created from these tokens.\n     */\n    cache_creation?: number;\n};\n/**\n * Breakdown of output token counts.\n *\n * Does *not* need to sum to full output token count. Does *not* need to have all keys.\n */\nexport type OutputTokenDetails = ModalitiesTokenDetails & {\n    /**\n     * Reasoning output tokens.\n     *\n     * Tokens generated by the model in a chain of thought process (i.e. by\n     * OpenAI's o1 models) that are not returned as part of model output.\n     */\n    reasoning?: number;\n};\n/**\n * Usage metadata for a message, such as token counts.\n */\nexport type UsageMetadata = {\n    /**\n     * Count of input (or prompt) tokens. Sum of all input token types.\n     */\n    input_tokens: number;\n    /**\n     * Count of output (or completion) tokens. Sum of all output token types.\n     */\n    output_tokens: number;\n    /**\n     * Total token count. Sum of input_tokens + output_tokens.\n     */\n    total_tokens: number;\n    /**\n     * Breakdown of input token counts.\n     *\n     * Does *not* need to sum to full input token count. Does *not* need to have all keys.\n     */\n    input_token_details?: InputTokenDetails;\n    /**\n     * Breakdown of output token counts.\n     *\n     * Does *not* need to sum to full output token count. Does *not* need to have all keys.\n     */\n    output_token_details?: OutputTokenDetails;\n};\n/**\n * Represents an AI message in a conversation.\n */\nexport declare class AIMessage extends BaseMessage implements AIMessageFields {\n    // These are typed as optional to avoid breaking changes and allow for casting\n    // from BaseMessage.\n    tool_calls?: ToolCall[];\n    invalid_tool_calls?: InvalidToolCall[];\n    /**\n     * If provided, token usage information associated with the message.\n     */\n    usage_metadata?: UsageMetadata;\n    get lc_aliases(): Record<string, string>;\n    constructor(fields: string | AIMessageFields, \n    /** @deprecated */\n    kwargs?: Record<string, unknown>);\n    static lc_name(): string;\n    _getType(): MessageType;\n    get contentBlocks(): Array<ContentBlock.Standard>;\n    get _printableFields(): Record<string, unknown>;\n}\nexport declare function isAIMessage(x: BaseMessage): x is AIMessage;\nexport declare function isAIMessageChunk(x: BaseMessageChunk): x is AIMessageChunk;\nexport type AIMessageChunkFields = AIMessageFields & {\n    tool_call_chunks?: ToolCallChunk[];\n};\n/**\n * Represents a chunk of an AI message, which can be concatenated with\n * other AI message chunks.\n */\nexport declare class AIMessageChunk extends BaseMessageChunk {\n    // Must redeclare tool call fields since there is no multiple inheritance in JS.\n    // These are typed as optional to avoid breaking changes and allow for casting\n    // from BaseMessage.\n    tool_calls?: ToolCall[];\n    invalid_tool_calls?: InvalidToolCall[];\n    tool_call_chunks?: ToolCallChunk[];\n    /**\n     * If provided, token usage information associated with the message.\n     */\n    usage_metadata?: UsageMetadata;\n    constructor(fields: string | AIMessageChunkFields);\n    get lc_aliases(): Record<string, string>;\n    static lc_name(): string;\n    _getType(): MessageType;\n    get contentBlocks(): Array<ContentBlock.Standard>;\n    get _printableFields(): Record<string, unknown>;\n    concat(chunk: AIMessageChunk): AIMessageChunk;\n}\n"],"mappings":";;;;;KAGYQ,eAAAA,GAAkBL;eACbG;EADLE,kBAAe,CAAA,EAEFH,eAFE,EAAA;EAAA,cAAA,CAAA,EAGNI,aAHM;CAAA;AACVH,KAILI,sBAAAA,GAJKJ;EAAQ;;AAES;AAElC;EA6BYK,IAAAA,CAAAA,EAAAA,MAAAA;EAoBAC;AAYZ;;EAAyB,KAkBCD,CAAAA,EAAAA,MAAAA;EAAiB;AAME;AAK7C;EAA8B,KAAA,CAAA,EAAA,MAAA;EAAA;;;EAQI,KACZG,CAAAA,EAAAA,MAAAA;EAAM;;;;EAMwB,QAA3BE,CAAAA,EAAAA,MAAAA;CAAK;;;AAf+C;AAkB7E;;AAAuChB,KA/E3BW,iBAAAA,GAAoBD,sBA+EOV,GAAAA;EAAW;AAAiB;AACnE;;;;EAAkF,UAAA,CAAA,EAAA,MAAA;EACtEoB;;;;AACwB;EAMfD,cAAAA,CAAAA,EAAc,MAAA;CAAA;;;;;;AAYbL,KAhFVF,kBAAAA,GAAqBF,sBAgFXI,GAAAA;EAAM;;;;;;EAKqB,SAjBLb,CAAAA,EAAAA,MAAAA;AAAgB,CAAA;;;;KAxDhDQ,aAAAA;;;;;;;;;;;;;;;;;;wBAkBcE;;;;;;yBAMCC;;;;;cAKNC,SAAAA,SAAkBb,WAAAA,YAAuBQ;;;eAG7CF;uBACQD;;;;mBAIJI;oBACCK;+BACWN;WAEpBM;;cAEGZ;uBACSc,MAAMZ,YAAAA,CAAaW;0BAChBD;;iBAEJG,WAAAA,IAAejB,mBAAmBa;iBAClCK,gBAAAA,IAAoBjB,wBAAwBkB;KACxDC,oBAAAA,GAAuBZ;qBACZD;;;;;;cAMFY,cAAAA,SAAuBlB,gBAAAA;;;;eAI3BK;uBACQD;qBACFE;;;;mBAIFE;+BACYW;oBACXN;;cAENZ;uBACSc,MAAMZ,YAAAA,CAAaW;0BAChBD;gBACVK,iBAAiBA"}
+{"version":3,"file":"ai.d.cts","names":["BaseMessage","BaseMessageChunk","BaseMessageFields","ContentBlock","$InferMessageContent","$InferMessageProperty","MessageStructure","InvalidToolCall","ToolCall","ToolCallChunk","AIMessageFields","TStructure","AIMessage","Record","Standard","Array","isAIMessage","isAIMessageChunk","AIMessageChunk","AIMessageChunkFields"],"sources":["../../src/messages/ai.d.ts"],"sourcesContent":["import { BaseMessage, BaseMessageChunk, BaseMessageFields } from \"./base.js\";\nimport { ContentBlock } from \"./content/index.js\";\nimport { $InferMessageContent, $InferMessageProperty, MessageStructure } from \"./message.js\";\nimport { InvalidToolCall, ToolCall, ToolCallChunk } from \"./tool.js\";\nexport interface AIMessageFields<TStructure extends MessageStructure = MessageStructure> extends BaseMessageFields<TStructure, \"ai\"> {\n    tool_calls?: ToolCall[];\n    invalid_tool_calls?: InvalidToolCall[];\n    usage_metadata?: $InferMessageProperty<TStructure, \"ai\", \"usage_metadata\">;\n}\nexport declare class AIMessage<TStructure extends MessageStructure = MessageStructure> extends BaseMessage<TStructure, \"ai\"> implements AIMessageFields<TStructure> {\n    readonly type: \"ai\";\n    tool_calls?: ToolCall[];\n    invalid_tool_calls?: InvalidToolCall[];\n    usage_metadata?: AIMessageFields<TStructure>[\"usage_metadata\"];\n    get lc_aliases(): Record<string, string>;\n    constructor(fields: $InferMessageContent<TStructure, \"ai\"> | AIMessageFields<TStructure>);\n    static lc_name(): string;\n    get contentBlocks(): Array<ContentBlock.Standard>;\n    get _printableFields(): Record<string, unknown>;\n    static isInstance(obj: unknown): obj is AIMessage;\n}\n/**\n * @deprecated Use {@link AIMessage.isInstance} instead\n */\nexport declare function isAIMessage<TStructure extends MessageStructure>(x: BaseMessage): x is AIMessage<TStructure>;\n/**\n * @deprecated Use {@link AIMessageChunk.isInstance} instead\n */\nexport declare function isAIMessageChunk<TStructure extends MessageStructure>(x: BaseMessageChunk): x is AIMessageChunk<TStructure>;\nexport type AIMessageChunkFields<TStructure extends MessageStructure = MessageStructure> = AIMessageFields<TStructure> & {\n    tool_call_chunks?: ToolCallChunk[];\n};\n/**\n * Represents a chunk of an AI message, which can be concatenated with\n * other AI message chunks.\n */\nexport declare class AIMessageChunk<TStructure extends MessageStructure = MessageStructure> extends BaseMessageChunk<TStructure, \"ai\"> implements AIMessage<TStructure>, AIMessageChunkFields<TStructure> {\n    readonly type: \"ai\";\n    tool_calls?: ToolCall[];\n    invalid_tool_calls?: InvalidToolCall[];\n    tool_call_chunks?: ToolCallChunk[];\n    usage_metadata?: AIMessageChunkFields<TStructure>[\"usage_metadata\"];\n    constructor(fields: $InferMessageContent<TStructure, \"ai\"> | AIMessageChunkFields<TStructure>);\n    get lc_aliases(): Record<string, string>;\n    static lc_name(): string;\n    get contentBlocks(): Array<ContentBlock.Standard>;\n    get _printableFields(): Record<string, unknown>;\n    concat(chunk: AIMessageChunk<TStructure>): this;\n    static isInstance(obj: unknown): obj is AIMessageChunk;\n}\n"],"mappings":";;;;;;UAIiBU,mCAAmCJ,mBAAmBA,0BAA0BJ,kBAAkBS;eAClGH;EADAE,kBAAe,CAAA,EAEPH,eAFO,EAAA;EAAA,cAAA,CAAA,EAGXF,qBAHW,CAGWM,UAHX,EAAA,IAAA,EAAA,gBAAA,CAAA;;AAAuCL,cAKlDM,SALkDN,CAAAA,mBAKrBA,gBALqBA,GAKFA,gBALEA,CAAAA,SAKwBN,WALxBM,CAKoCK,UALpCL,EAAAA,IAAAA,CAAAA,YAKiEI,eALjEJ,CAKiFK,UALjFL,CAAAA,CAAAA;EAAgB,SAA4BK,IAAAA,EAAAA,IAAAA;EAAU,UAC5GH,CAAAA,EAMAA,QANAA,EAAAA;EAAQ,kBACAD,CAAAA,EAMAA,eANAA,EAAAA;EAAe,cACGI,CAAAA,EAMtBD,eANsBC,CAMNA,UANMA,CAAAA,CAAAA,gBAAAA,CAAAA;EAAU,IAAhCN,UAAAA,CAAAA,CAAAA,EAOCQ,MAPDR,CAAAA,MAAAA,EAAAA,MAAAA,CAAAA;EAAqB,WAHuDH,CAAAA,MAAAA,EAWzEE,oBAXyEF,CAWpDS,UAXoDT,EAAAA,IAAAA,CAAAA,GAWhCQ,eAXgCR,CAWhBS,UAXgBT,CAAAA;EAAiB,OAAA,OAAA,CAAA,CAAA,EAAA,MAAA;EAK7FU,IAAAA,aAAS,CAAA,CAAA,EAQLG,KARK,CAQCZ,YAAAA,CAAaW,QARd,CAAA;EAAA,IAAA,gBAAA,CAAA,CAAA,EASFD,MATE,CAAA,MAAA,EAAA,OAAA,CAAA;EAAA,OAAoBP,UAAAA,CAAAA,GAAAA,EAAAA,OAAAA,CAAAA,EAAAA,GAAAA,IAUNM,SAVMN;;;;;AAGzBC,iBAYDS,WAZCT,CAAAA,mBAY8BD,gBAZ9BC,CAAAA,CAAAA,CAAAA,EAYmDP,WAZnDO,CAAAA,EAAAA,CAAAA,IAYsEK,SAZtEL,CAYgFI,UAZhFJ,CAAAA;;;;AAGoBI,iBAarBM,gBAbqBN,CAAAA,mBAaeL,gBAbfK,CAAAA,CAAAA,CAAAA,EAaoCV,gBAbpCU,CAAAA,EAAAA,CAAAA,IAa4DO,cAb5DP,CAa2EA,UAb3EA,CAAAA;AAArBP,KAcZe,oBAdYf,CAAAA,mBAc4BE,gBAd5BF,GAc+CE,gBAd/CF,CAAAA,GAcmEM,eAdnEN,CAcmFO,UAdnFP,CAAAA,GAAAA;EAAoB,gBAAqCO,CAAAA,EAe1DF,aAf0DE,EAAAA;CAAU;;;;;AANIX,cA2B1EkB,cA3B0ElB,CAAAA,mBA2BxCM,gBA3BwCN,GA2BrBM,gBA3BqBN,CAAAA,SA2BKC,gBA3BLD,CA2BsBW,UA3BtBX,EAAAA,IAAAA,CAAAA,YA2BmDY,SA3BnDZ,CA2B6DW,UA3B7DX,CAAAA,EA2B0EmB,oBA3B1EnB,CA2B+FW,UA3B/FX,CAAAA,CAAAA;EAAW,SAA8BU,IAAAA,EAAAA,IAAAA;EAAe,UAAA,CAAA,EA6BtIF,QA7BsI,EAAA;EAe/HQ,kBAAW,CAAA,EAeVT,eAfU,EAAA;EAAA,gBAAA,CAAA,EAgBZE,aAhBY,EAAA;EAAA,cAAoBH,CAAAA,EAiBlCa,oBAjBkCb,CAiBbK,UAjBaL,CAAAA,CAAAA,gBAAAA,CAAAA;EAAgB,WAAKN,CAAAA,MAAAA,EAkBpDI,oBAlBoDJ,CAkB/BW,UAlB+BX,EAAAA,IAAAA,CAAAA,GAkBXmB,oBAlBWnB,CAkBUW,UAlBVX,CAAAA;EAAW,IAAkBW,UAAAA,CAAAA,CAAAA,EAmBnFE,MAnBmFF,CAAAA,MAAAA,EAAAA,MAAAA,CAAAA;EAAU,OAApBC,OAAAA,CAAAA,CAAAA,EAAAA,MAAAA;EAAS,IAAA,aAAA,CAAA,CAAA,EAqB/EG,KArB+E,CAqBzEZ,YAAAA,CAAaW,QArB4D,CAAA;EAIhFG,IAAAA,gBAAgB,CAAA,CAAA,EAkBZJ,MAlBY,CAAA,MAAA,EAAA,OAAA,CAAA;EAAA,MAAA,CAAA,KAAA,EAmBtBK,cAnBsB,CAmBPP,UAnBO,CAAA,CAAA,EAAA,IAAA;EAAA,OAAoBL,UAAAA,CAAAA,GAAAA,EAAAA,OAAAA,CAAAA,EAAAA,GAAAA,IAoBhBY,cApBgBZ"}