@librechat/agents 3.0.78 → 3.0.79

package/dist/types/llm/bedrock/index.d.ts

@@ -1,5 +1,9 @@
 /**
  * Optimized ChatBedrockConverse wrapper that fixes contentBlockIndex conflicts
+ * and adds support for latest @langchain/aws features:
+ *
+ * - Application Inference Profiles (PR #9129)
+ * - Service Tiers (Priority/Standard/Flex) (PR #9785) - requires AWS SDK 3.966.0+
  *
  * Bedrock sends the same contentBlockIndex for both text and tool_use content blocks,
  * causing LangChain's merge logic to fail with "field[contentBlockIndex] already exists"
@@ -11,19 +15,91 @@
  * the purpose of tracking tool call ordering.
  */
 import { ChatBedrockConverse } from '@langchain/aws';
+import { ChatGenerationChunk, ChatResult } from '@langchain/core/outputs';
+import type { CallbackManagerForLLMRun } from '@langchain/core/callbacks/manager';
 import type { ChatBedrockConverseInput } from '@langchain/aws';
 import type { BaseMessage } from '@langchain/core/messages';
-import { ChatGenerationChunk } from '@langchain/core/outputs';
-import type { CallbackManagerForLLMRun } from '@langchain/core/callbacks/manager';
+/**
+ * Service tier type for Bedrock invocations.
+ * Requires AWS SDK >= 3.966.0 to actually work.
+ * @see https://docs.aws.amazon.com/bedrock/latest/userguide/service-tiers-inference.html
+ */
+export type ServiceTierType = 'priority' | 'default' | 'flex' | 'reserved';
+/**
+ * Extended input interface with additional features:
+ * - applicationInferenceProfile: Use an inference profile ARN instead of model ID
+ * - serviceTier: Specify service tier (Priority, Standard, Flex, Reserved)
+ */
+export interface CustomChatBedrockConverseInput extends ChatBedrockConverseInput {
+    /**
+     * Application Inference Profile ARN to use for the model.
+     * For example, "arn:aws:bedrock:eu-west-1:123456789102:application-inference-profile/fm16bt65tzgx"
+     * When provided, this ARN will be used for the actual inference calls instead of the model ID.
+     * Must still provide `model` as normal modelId to benefit from all the metadata.
+     * @see https://docs.aws.amazon.com/bedrock/latest/userguide/inference-profiles-create.html
+     */
+    applicationInferenceProfile?: string;
+    /**
+     * Service tier for model invocation.
+     * Specifies the processing tier type used for serving the request.
+     * Supported values are 'priority', 'default', 'flex', and 'reserved'.
+     *
+     * - 'priority': Prioritized processing for lower latency
+     * - 'default': Standard processing tier
+     * - 'flex': Flexible processing tier with lower cost
+     * - 'reserved': Reserved capacity for consistent performance
+     *
+     * If not provided, AWS uses the default tier.
+     * Note: Requires AWS SDK >= 3.966.0 to work.
+     * @see https://docs.aws.amazon.com/bedrock/latest/userguide/service-tiers-inference.html
+     */
+    serviceTier?: ServiceTierType;
+}
+/**
+ * Extended call options with serviceTier override support.
+ */
+export interface CustomChatBedrockConverseCallOptions {
+    serviceTier?: ServiceTierType;
+}
 export declare class CustomChatBedrockConverse extends ChatBedrockConverse {
-    constructor(fields?: ChatBedrockConverseInput);
+    /**
+     * Application Inference Profile ARN to use instead of model ID.
+     */
+    applicationInferenceProfile?: string;
+    /**
+     * Service tier for model invocation.
+     */
+    serviceTier?: ServiceTierType;
+    constructor(fields?: CustomChatBedrockConverseInput);
     static lc_name(): string;
     /**
-     * Override _streamResponseChunks to strip contentBlockIndex from response_metadata
-     * This prevents LangChain's merge conflicts when the same index is used for
-     * different content types (text vs tool calls)
+     * Get the model ID to use for API calls.
+     * Returns applicationInferenceProfile if set, otherwise returns this.model.
+     */
+    protected getModelId(): string;
+    /**
+     * Override invocationParams to add serviceTier support.
+     */
+    invocationParams(options?: this['ParsedCallOptions'] & CustomChatBedrockConverseCallOptions): ReturnType<ChatBedrockConverse['invocationParams']> & {
+        serviceTier?: {
+            type: ServiceTierType;
+        };
+    };
+    /**
+     * Override _generateNonStreaming to use applicationInferenceProfile as modelId.
+     */
+    _generateNonStreaming(messages: BaseMessage[], options: this['ParsedCallOptions'] & CustomChatBedrockConverseCallOptions, _runManager?: CallbackManagerForLLMRun): Promise<ChatResult>;
+    /**
+     * Override _streamResponseChunks to:
+     * 1. Use applicationInferenceProfile as modelId
+     * 2. Include serviceTier in request
+     * 3. Strip contentBlockIndex from response_metadata to prevent merge conflicts
+     */
+    _streamResponseChunks(messages: BaseMessage[], options: this['ParsedCallOptions'] & CustomChatBedrockConverseCallOptions, runManager?: CallbackManagerForLLMRun): AsyncGenerator<ChatGenerationChunk>;
+    /**
+     * Clean a chunk by removing contentBlockIndex from response_metadata.
      */
-    _streamResponseChunks(messages: BaseMessage[], options: this['ParsedCallOptions'], runManager?: CallbackManagerForLLMRun): AsyncGenerator<ChatGenerationChunk>;
+    private cleanChunk;
     /**
      * Check if contentBlockIndex exists at any level in the object
      */

package/dist/types/llm/bedrock/types.d.ts

@@ -0,0 +1,27 @@
+/**
+ * Type definitions for Bedrock Converse utilities.
+ */
+import type { Message as BedrockMessage, SystemContentBlock as BedrockSystemContentBlock, ContentBlock as BedrockContentBlock, ConverseResponse, ContentBlockDeltaEvent, ConverseStreamMetadataEvent, ContentBlockStartEvent, ReasoningContentBlock, ReasoningContentBlockDelta } from '@aws-sdk/client-bedrock-runtime';
+/**
+ * Reasoning content block type for LangChain messages.
+ */
+export interface MessageContentReasoningBlock {
+    type: 'reasoning_content';
+    reasoningText?: {
+        text?: string;
+        signature?: string;
+    };
+    redactedContent?: string;
+}
+export interface MessageContentReasoningBlockReasoningTextPartial {
+    type: 'reasoning_content';
+    reasoningText: {
+        text?: string;
+        signature?: string;
+    };
+}
+export interface MessageContentReasoningBlockRedacted {
+    type: 'reasoning_content';
+    redactedContent: string;
+}
+export type { BedrockMessage, BedrockSystemContentBlock, BedrockContentBlock, ConverseResponse, ContentBlockDeltaEvent, ConverseStreamMetadataEvent, ContentBlockStartEvent, ReasoningContentBlock, ReasoningContentBlockDelta, };

package/dist/types/llm/bedrock/utils/index.d.ts

@@ -0,0 +1,5 @@
+/**
+ * Bedrock Converse utility exports.
+ */
+export { convertToConverseMessages, extractImageInfo, langchainReasoningBlockToBedrockReasoningBlock, concatenateLangchainReasoningBlocks, } from './message_inputs';
+export { convertConverseMessageToLangChainMessage, handleConverseStreamContentBlockStart, handleConverseStreamContentBlockDelta, handleConverseStreamMetadata, bedrockReasoningBlockToLangchainReasoningBlock, bedrockReasoningDeltaToLangchainPartialReasoningBlock, } from './message_outputs';

package/dist/types/llm/bedrock/utils/message_inputs.d.ts

@@ -0,0 +1,31 @@
+/**
+ * Utility functions for converting LangChain messages to Bedrock Converse messages.
+ * Ported from @langchain/aws common.js
+ */
+import { type BaseMessage, MessageContentComplex } from '@langchain/core/messages';
+import type { BedrockMessage, BedrockSystemContentBlock, BedrockContentBlock, MessageContentReasoningBlock } from '../types';
+/**
+ * Convert a LangChain reasoning block to a Bedrock reasoning block.
+ */
+export declare function langchainReasoningBlockToBedrockReasoningBlock(content: MessageContentReasoningBlock): {
+    reasoningText?: {
+        text?: string;
+        signature?: string;
+    };
+    redactedContent?: Uint8Array;
+};
+/**
+ * Concatenate consecutive reasoning blocks in content array.
+ */
+export declare function concatenateLangchainReasoningBlocks(content: Array<MessageContentComplex | MessageContentReasoningBlock>): Array<MessageContentComplex | MessageContentReasoningBlock>;
+/**
+ * Extract image info from a base64 string or URL.
+ */
+export declare function extractImageInfo(base64: string): BedrockContentBlock;
+/**
+ * Convert LangChain messages to Bedrock Converse messages.
+ */
+export declare function convertToConverseMessages(messages: BaseMessage[]): {
+    converseMessages: BedrockMessage[];
+    converseSystem: BedrockSystemContentBlock[];
+};

package/dist/types/llm/bedrock/utils/message_outputs.d.ts

@@ -0,0 +1,33 @@
+/**
+ * Utility functions for converting Bedrock Converse responses to LangChain messages.
+ * Ported from @langchain/aws common.js
+ */
+import { AIMessage } from '@langchain/core/messages';
+import { ChatGenerationChunk } from '@langchain/core/outputs';
+import type { BedrockMessage, ConverseResponse, ContentBlockDeltaEvent, ConverseStreamMetadataEvent, ContentBlockStartEvent, ReasoningContentBlock, ReasoningContentBlockDelta, MessageContentReasoningBlock, MessageContentReasoningBlockReasoningTextPartial, MessageContentReasoningBlockRedacted } from '../types';
+/**
+ * Convert a Bedrock reasoning block delta to a LangChain partial reasoning block.
+ */
+export declare function bedrockReasoningDeltaToLangchainPartialReasoningBlock(reasoningContent: ReasoningContentBlockDelta): MessageContentReasoningBlockReasoningTextPartial | MessageContentReasoningBlockRedacted;
+/**
+ * Convert a Bedrock reasoning block to a LangChain reasoning block.
+ */
+export declare function bedrockReasoningBlockToLangchainReasoningBlock(reasoningContent: ReasoningContentBlock): MessageContentReasoningBlock;
+/**
+ * Convert a Bedrock Converse message to a LangChain message.
+ */
+export declare function convertConverseMessageToLangChainMessage(message: BedrockMessage, responseMetadata: Omit<ConverseResponse, 'output'>): AIMessage;
+/**
+ * Handle a content block delta event from Bedrock Converse stream.
+ */
+export declare function handleConverseStreamContentBlockDelta(contentBlockDelta: ContentBlockDeltaEvent): ChatGenerationChunk;
+/**
+ * Handle a content block start event from Bedrock Converse stream.
+ */
+export declare function handleConverseStreamContentBlockStart(contentBlockStart: ContentBlockStartEvent): ChatGenerationChunk | null;
+/**
+ * Handle a metadata event from Bedrock Converse stream.
+ */
+export declare function handleConverseStreamMetadata(metadata: ConverseStreamMetadataEvent, extra: {
+    streamUsage: boolean;
+}): ChatGenerationChunk;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@librechat/agents",
-  "version": "3.0.78",
+  "version": "3.0.79",
   "main": "./dist/cjs/main.cjs",
   "module": "./dist/esm/main.mjs",
   "types": "./dist/types/index.d.ts",
@@ -53,11 +53,11 @@
     "code_exec_session": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/code_exec_session.ts --provider 'openAI' --name 'Jo' --location 'New York, NY'",
     "code_exec_multi_session": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/code_exec_multi_session.ts --provider 'openAI' --name 'Jo' --location 'New York, NY'",
     "code_exec_simple": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/code_exec_simple.ts --provider 'openAI' --name 'Jo' --location 'New York, NY'",
-    "simple": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/simple.ts --provider 'openrouter' --name 'Jo' --location 'New York, NY'",
+    "simple": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/simple.ts --provider 'bedrock' --name 'Jo' --location 'New York, NY'",
     "caching": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/caching.ts --name 'Jo' --location 'New York, NY'",
     "thinking": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/thinking.ts --name 'Jo' --location 'New York, NY'",
     "memory": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/memory.ts --provider 'openAI' --name 'Jo' --location 'New York, NY'",
-    "tool": "node --trace-warnings -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/tools.ts --provider 'openrouter' --name 'Jo' --location 'New York, NY'",
+    "tool": "node --trace-warnings -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/tools.ts --provider 'bedrock' --name 'Jo' --location 'New York, NY'",
     "search": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/search.ts --provider 'bedrock' --name 'Jo' --location 'New York, NY'",
     "tool_search": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/tool_search.ts",
     "programmatic_exec": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/programmatic_exec.ts",
@@ -111,6 +111,7 @@
     }
   },
   "dependencies": {
+    "@aws-sdk/client-bedrock-runtime": "^3.970.0",
     "@langchain/anthropic": "^0.3.26",
     "@langchain/aws": "^0.1.15",
     "@langchain/core": "^0.3.80",

package/src/llm/bedrock/index.ts

@@ -1,5 +1,9 @@
 /**
  * Optimized ChatBedrockConverse wrapper that fixes contentBlockIndex conflicts
+ * and adds support for latest @langchain/aws features:
+ *
+ * - Application Inference Profiles (PR #9129)
+ * - Service Tiers (Priority/Standard/Flex) (PR #9785) - requires AWS SDK 3.966.0+
  *
  * Bedrock sends the same contentBlockIndex for both text and tool_use content blocks,
  * causing LangChain's merge logic to fail with "field[contentBlockIndex] already exists"
@@ -12,15 +16,85 @@
  */
 
 import { ChatBedrockConverse } from '@langchain/aws';
-import type { ChatBedrockConverseInput } from '@langchain/aws';
 import { AIMessageChunk } from '@langchain/core/messages';
-import type { BaseMessage } from '@langchain/core/messages';
-import { ChatGenerationChunk } from '@langchain/core/outputs';
+import { ChatGenerationChunk, ChatResult } from '@langchain/core/outputs';
 import type { CallbackManagerForLLMRun } from '@langchain/core/callbacks/manager';
+import type { ChatBedrockConverseInput } from '@langchain/aws';
+import type { BaseMessage } from '@langchain/core/messages';
+import {
+  ConverseCommand,
+  ConverseStreamCommand,
+} from '@aws-sdk/client-bedrock-runtime';
+import {
+  convertToConverseMessages,
+  convertConverseMessageToLangChainMessage,
+  handleConverseStreamContentBlockStart,
+  handleConverseStreamContentBlockDelta,
+  handleConverseStreamMetadata,
+} from './utils';
+
+/**
+ * Service tier type for Bedrock invocations.
+ * Requires AWS SDK >= 3.966.0 to actually work.
+ * @see https://docs.aws.amazon.com/bedrock/latest/userguide/service-tiers-inference.html
+ */
+export type ServiceTierType = 'priority' | 'default' | 'flex' | 'reserved';
+
+/**
+ * Extended input interface with additional features:
+ * - applicationInferenceProfile: Use an inference profile ARN instead of model ID
+ * - serviceTier: Specify service tier (Priority, Standard, Flex, Reserved)
+ */
+export interface CustomChatBedrockConverseInput
+  extends ChatBedrockConverseInput {
+  /**
+   * Application Inference Profile ARN to use for the model.
+   * For example, "arn:aws:bedrock:eu-west-1:123456789102:application-inference-profile/fm16bt65tzgx"
+   * When provided, this ARN will be used for the actual inference calls instead of the model ID.
+   * Must still provide `model` as normal modelId to benefit from all the metadata.
+   * @see https://docs.aws.amazon.com/bedrock/latest/userguide/inference-profiles-create.html
+   */
+  applicationInferenceProfile?: string;
+
+  /**
+   * Service tier for model invocation.
+   * Specifies the processing tier type used for serving the request.
+   * Supported values are 'priority', 'default', 'flex', and 'reserved'.
+   *
+   * - 'priority': Prioritized processing for lower latency
+   * - 'default': Standard processing tier
+   * - 'flex': Flexible processing tier with lower cost
+   * - 'reserved': Reserved capacity for consistent performance
+   *
+   * If not provided, AWS uses the default tier.
+   * Note: Requires AWS SDK >= 3.966.0 to work.
+   * @see https://docs.aws.amazon.com/bedrock/latest/userguide/service-tiers-inference.html
+   */
+  serviceTier?: ServiceTierType;
+}
+
+/**
+ * Extended call options with serviceTier override support.
+ */
+export interface CustomChatBedrockConverseCallOptions {
+  serviceTier?: ServiceTierType;
+}
 
 export class CustomChatBedrockConverse extends ChatBedrockConverse {
-  constructor(fields?: ChatBedrockConverseInput) {
+  /**
+   * Application Inference Profile ARN to use instead of model ID.
+   */
+  applicationInferenceProfile?: string;
+
+  /**
+   * Service tier for model invocation.
+   */
+  serviceTier?: ServiceTierType;
+
+  constructor(fields?: CustomChatBedrockConverseInput) {
     super(fields);
+    this.applicationInferenceProfile = fields?.applicationInferenceProfile;
+    this.serviceTier = fields?.serviceTier;
   }
 
   static lc_name(): string {
@@ -28,52 +102,169 @@ export class CustomChatBedrockConverse extends ChatBedrockConverse {
   }
 
   /**
-   * Override _streamResponseChunks to strip contentBlockIndex from response_metadata
-   * This prevents LangChain's merge conflicts when the same index is used for
-   * different content types (text vs tool calls)
+   * Get the model ID to use for API calls.
+   * Returns applicationInferenceProfile if set, otherwise returns this.model.
+   */
+  protected getModelId(): string {
+    return this.applicationInferenceProfile ?? this.model;
+  }
+
+  /**
+   * Override invocationParams to add serviceTier support.
    */
-  async *_streamResponseChunks(
+  override invocationParams(
+    options?: this['ParsedCallOptions'] & CustomChatBedrockConverseCallOptions
+  ): ReturnType<ChatBedrockConverse['invocationParams']> & {
+    serviceTier?: { type: ServiceTierType };
+  } {
+    const baseParams = super.invocationParams(options);
+
+    // Get serviceTier from options or fall back to class-level setting
+    const serviceTierType = options?.serviceTier ?? this.serviceTier;
+
+    return {
+      ...baseParams,
+      serviceTier: serviceTierType ? { type: serviceTierType } : undefined,
+    };
+  }
+
+  /**
+   * Override _generateNonStreaming to use applicationInferenceProfile as modelId.
+   */
+  override async _generateNonStreaming(
     messages: BaseMessage[],
-    options: this['ParsedCallOptions'],
+    options: this['ParsedCallOptions'] & CustomChatBedrockConverseCallOptions,
+    _runManager?: CallbackManagerForLLMRun
+  ): Promise<ChatResult> {
+    const { converseMessages, converseSystem } =
+      convertToConverseMessages(messages);
+    const params = this.invocationParams(options);
+
+    const command = new ConverseCommand({
+      modelId: this.getModelId(),
+      messages: converseMessages,
+      system: converseSystem,
+      requestMetadata: options.requestMetadata,
+      ...params,
+    });
+
+    const response = await this.client.send(command, {
+      abortSignal: options.signal,
+    });
+
+    const { output, ...responseMetadata } = response;
+    if (!output?.message) {
+      throw new Error('No message found in Bedrock response.');
+    }
+
+    const message = convertConverseMessageToLangChainMessage(
+      output.message,
+      responseMetadata
+    );
+
+    return {
+      generations: [
+        {
+          text: typeof message.content === 'string' ? message.content : '',
+          message,
+        },
+      ],
+    };
+  }
+
+  /**
+   * Override _streamResponseChunks to:
+   * 1. Use applicationInferenceProfile as modelId
+   * 2. Include serviceTier in request
+   * 3. Strip contentBlockIndex from response_metadata to prevent merge conflicts
+   */
+  override async *_streamResponseChunks(
+    messages: BaseMessage[],
+    options: this['ParsedCallOptions'] & CustomChatBedrockConverseCallOptions,
     runManager?: CallbackManagerForLLMRun
   ): AsyncGenerator<ChatGenerationChunk> {
-    const baseStream = super._streamResponseChunks(
-      messages,
-      options,
-      runManager
-    );
+    const { converseMessages, converseSystem } =
+      convertToConverseMessages(messages);
+    const params = this.invocationParams(options);
+
+    let { streamUsage } = this;
+    if (options.streamUsage !== undefined) {
+      streamUsage = options.streamUsage;
+    }
+
+    const command = new ConverseStreamCommand({
+      modelId: this.getModelId(),
+      messages: converseMessages,
+      system: converseSystem,
+      requestMetadata: options.requestMetadata,
+      ...params,
+    });
 
-    for await (const chunk of baseStream) {
-      // Only process if we have response_metadata
-      if (
-        chunk.message instanceof AIMessageChunk &&
-        (chunk.message as Partial<AIMessageChunk>).response_metadata &&
-        typeof chunk.message.response_metadata === 'object'
-      ) {
-        // Check if contentBlockIndex exists anywhere in response_metadata (top level or nested)
-        const hasContentBlockIndex = this.hasContentBlockIndex(
-          chunk.message.response_metadata
-        );
-
-        if (hasContentBlockIndex) {
-          const cleanedMetadata = this.removeContentBlockIndex(
-            chunk.message.response_metadata
-          ) as Record<string, unknown>;
-
-          yield new ChatGenerationChunk({
-            text: chunk.text,
-            message: new AIMessageChunk({
-              ...chunk.message,
-              response_metadata: cleanedMetadata,
-            }),
-            generationInfo: chunk.generationInfo,
-          });
-          continue;
+    const response = await this.client.send(command, {
+      abortSignal: options.signal,
+    });
+
+    if (response.stream) {
+      for await (const event of response.stream) {
+        if (event.contentBlockStart != null) {
+          const chunk = handleConverseStreamContentBlockStart(
+            event.contentBlockStart
+          ) as ChatGenerationChunk | undefined;
+          if (chunk !== undefined) {
+            const cleanedChunk = this.cleanChunk(chunk);
+            yield cleanedChunk;
+            await runManager?.handleLLMNewToken(cleanedChunk.text || '');
+          }
+        } else if (event.contentBlockDelta != null) {
+          const chunk = handleConverseStreamContentBlockDelta(
+            event.contentBlockDelta
+          ) as ChatGenerationChunk | undefined;
+          if (chunk !== undefined) {
+            const cleanedChunk = this.cleanChunk(chunk);
+            yield cleanedChunk;
+            await runManager?.handleLLMNewToken(cleanedChunk.text || '');
+          }
+        } else if (event.metadata != null) {
+          const chunk = handleConverseStreamMetadata(event.metadata, {
+            streamUsage,
+          }) as ChatGenerationChunk | undefined;
+          if (chunk !== undefined) {
+            const cleanedChunk = this.cleanChunk(chunk);
+            yield cleanedChunk;
+          }
         }
       }
+    }
+  }
 
-      yield chunk;
+  /**
+   * Clean a chunk by removing contentBlockIndex from response_metadata.
+   */
+  private cleanChunk(chunk: ChatGenerationChunk): ChatGenerationChunk {
+    const message = chunk.message;
+    if (!(message instanceof AIMessageChunk)) {
+      return chunk;
     }
+
+    const metadata = message.response_metadata as Record<string, unknown>;
+    const hasContentBlockIndex = this.hasContentBlockIndex(metadata);
+    if (!hasContentBlockIndex) {
+      return chunk;
+    }
+
+    const cleanedMetadata = this.removeContentBlockIndex(metadata) as Record<
+      string,
+      unknown
+    >;
+
+    return new ChatGenerationChunk({
+      text: chunk.text,
+      message: new AIMessageChunk({
+        ...message,
+        response_metadata: cleanedMetadata,
+      }),
+      generationInfo: chunk.generationInfo,
+    });
   }
 
   /**