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

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"}

package/dist/messages/ai.d.ts

@@ -1,148 +1,55 @@
 import { ContentBlock } from "./content/index.js";
-import { BaseMessage, BaseMessageChunk, BaseMessageFields, MessageType } from "./base.js";
+import { BaseMessage, BaseMessageChunk, BaseMessageFields } from "./base.js";
 import { InvalidToolCall, ToolCall, ToolCallChunk } from "./tool.js";
+import { $InferMessageContent, $InferMessageProperty, MessageStructure } from "./message.js";
 
 //#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.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"ai.d.ts","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.ts","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"}

package/dist/messages/ai.js

@@ -1,18 +1,14 @@
 import { parsePartialJson } from "../utils/json.js";
 import { BaseMessage, BaseMessageChunk, _mergeDicts, _mergeLists, mergeContent } from "./base.js";
 import { getTranslator } from "./block_translators/index.js";
+import { mergeResponseMetadata, mergeUsageMetadata } from "./metadata.js";
 import { defaultToolCallParser } from "./tool.js";
 
 //#region src/messages/ai.ts
-/**
-* Represents an AI message in a conversation.
-*/
 var AIMessage = class extends 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 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;
@@ -61,6 +57,7 @@ var AIMessage = class extends 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 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 = 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 = getTranslator(this.response_metadata.model_provider);
 			if (translator) return translator.translateContent(this);
 		}
 		const blocks = super.contentBlocks;
@@ -108,10 +101,19 @@ var AIMessage = class extends 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 BaseMessageChunk {
+var AIMessageChunk = class extends 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 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 : "{}";
@@ -200,14 +200,10 @@ var AIMessageChunk = class AIMessageChunk extends 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 = 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 = 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 BaseMessageChunk {
 		const combinedFields = {
 			content: mergeContent(this.content, chunk.content),
 			additional_kwargs: _mergeDicts(this.additional_kwargs, chunk.additional_kwargs),
-			response_metadata: _mergeDicts(this.response_metadata, chunk.response_metadata),
+			response_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 BaseMessageChunk {
 			const rawToolCalls = _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 = 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";
 	}
 };