@librechat/agents 2.4.41 → 2.4.43

package/src/scripts/image.ts

@@ -5,14 +5,17 @@ import { HumanMessage, AIMessage, BaseMessage } from '@langchain/core/messages';
 import type { RunnableConfig } from '@langchain/core/runnables';
 import type * as t from '@/types';
 import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
-import { ToolEndHandler, ModelEndHandler, createMetadataAggregator } from '@/events';
+import {
+  ToolEndHandler,
+  ModelEndHandler,
+  createMetadataAggregator,
+} from '@/events';
 import { fetchRandomImageTool, fetchRandomImageURL } from '@/tools/example';
 import { getLLMConfig } from '@/utils/llmConfig';
 import { getArgs } from '@/scripts/args';
 import { GraphEvents } from '@/common';
 import { Run } from '@/run';
 
-
 const conversationHistory: BaseMessage[] = [];
 
 async function testCodeExecution(): Promise<void> {
@@ -23,38 +26,57 @@ async function testCodeExecution(): Promise<void> {
     [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(),
     [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
     [GraphEvents.ON_RUN_STEP_COMPLETED]: {
-      handle: (event: GraphEvents.ON_RUN_STEP_COMPLETED, data: t.StreamEventData): void => {
+      handle: (
+        event: GraphEvents.ON_RUN_STEP_COMPLETED,
+        data: t.StreamEventData
+      ): void => {
         console.log('====== ON_RUN_STEP_COMPLETED ======');
         console.dir(data, { depth: null });
-        aggregateContent({ event, data: data as unknown as { result: t.ToolEndEvent } });
-      }
+        aggregateContent({
+          event,
+          data: data as unknown as { result: t.ToolEndEvent },
+        });
+      },
     },
     [GraphEvents.ON_RUN_STEP]: {
-      handle: (event: GraphEvents.ON_RUN_STEP, data: t.StreamEventData): void => {
+      handle: (
+        event: GraphEvents.ON_RUN_STEP,
+        data: t.StreamEventData
+      ): void => {
         console.log('====== ON_RUN_STEP ======');
         console.dir(data, { depth: null });
         aggregateContent({ event, data: data as t.RunStep });
-      }
+      },
     },
     [GraphEvents.ON_RUN_STEP_DELTA]: {
-      handle: (event: GraphEvents.ON_RUN_STEP_DELTA, data: t.StreamEventData): void => {
+      handle: (
+        event: GraphEvents.ON_RUN_STEP_DELTA,
+        data: t.StreamEventData
+      ): void => {
         console.log('====== ON_RUN_STEP_DELTA ======');
         console.dir(data, { depth: null });
         aggregateContent({ event, data: data as t.RunStepDeltaEvent });
-      }
+      },
     },
     [GraphEvents.ON_MESSAGE_DELTA]: {
-      handle: (event: GraphEvents.ON_MESSAGE_DELTA, data: t.StreamEventData): void => {
+      handle: (
+        event: GraphEvents.ON_MESSAGE_DELTA,
+        data: t.StreamEventData
+      ): void => {
         console.log('====== ON_MESSAGE_DELTA ======');
         console.dir(data, { depth: null });
         aggregateContent({ event, data: data as t.MessageDeltaEvent });
-      }
+      },
     },
     [GraphEvents.TOOL_START]: {
-      handle: (_event: string, data: t.StreamEventData, metadata?: Record<string, unknown>): void => {
+      handle: (
+        _event: string,
+        data: t.StreamEventData,
+        metadata?: Record<string, unknown>
+      ): void => {
         console.log('====== TOOL_START ======');
         console.dir(data, { depth: null });
-      }
+      },
     },
   };
 
@@ -67,14 +89,19 @@ async function testCodeExecution(): Promise<void> {
       llmConfig,
       tools: [fetchRandomImageTool],
       // tools: [fetchRandomImageURL],
-      instructions: 'You are a friendly AI assistant with internet capabilities. Always address the user by their name.',
+      instructions:
+        'You are a friendly AI assistant with internet capabilities. Always address the user by their name.',
       additional_instructions: `The user's name is ${userName} and they are located in ${location}.`,
     },
     returnContent: true,
     customHandlers,
   });
 
-  const config: Partial<RunnableConfig> & { version: 'v1' | 'v2'; run_id?: string; streamMode: string } = {
+  const config: Partial<RunnableConfig> & {
+    version: 'v1' | 'v2';
+    run_id?: string;
+    streamMode: string;
+  } = {
     configurable: {
       provider,
       thread_id: 'conversation-num-1',
@@ -109,7 +136,9 @@ async function testCodeExecution(): Promise<void> {
   inputs = {
     messages: conversationHistory,
   };
-  const finalContentParts2 = await run.processStream(inputs, config, { keepContent: true });
+  const finalContentParts2 = await run.processStream(inputs, config, {
+    keepContent: true,
+  });
   const finalMessages2 = run.getRunMessages();
   if (finalMessages2) {
     conversationHistory.push(...finalMessages2);
@@ -119,12 +148,15 @@ async function testCodeExecution(): Promise<void> {
 
   const { handleLLMEnd, collected } = createMetadataAggregator();
   const titleResult = await run.generateTitle({
+    provider,
     inputText: userMessage2,
     contentParts,
     chainOptions: {
-      callbacks: [{
-        handleLLMEnd,
-      }],
+      callbacks: [
+        {
+          handleLLMEnd,
+        },
+      ],
     },
   });
   console.log('Generated Title:', titleResult);
@@ -143,4 +175,4 @@ testCodeExecution().catch((err) => {
   console.log('Conversation history:');
   console.dir(conversationHistory, { depth: null });
   process.exit(1);
-});
+});

package/src/scripts/simple.ts

@@ -160,6 +160,7 @@ async function testStandardStreaming(): Promise<void> {
   console.dir(contentParts, { depth: null });
   const { handleLLMEnd, collected } = createMetadataAggregator();
   const titleOptions: t.RunTitleOptions = {
+    provider,
     inputText: userMessage,
     contentParts,
     chainOptions: {

package/src/specs/anthropic.simple.test.ts

@@ -4,10 +4,18 @@
 import { config } from 'dotenv';
 config();
 import { Calculator } from '@langchain/community/tools/calculator';
-import { HumanMessage, BaseMessage, UsageMetadata } from '@langchain/core/messages';
+import {
+  HumanMessage,
+  BaseMessage,
+  UsageMetadata,
+} from '@langchain/core/messages';
 import type { StandardGraph } from '@/graphs';
 import type * as t from '@/types';
-import { ToolEndHandler, ModelEndHandler, createMetadataAggregator } from '@/events';
+import {
+  ToolEndHandler,
+  ModelEndHandler,
+  createMetadataAggregator,
+} from '@/events';
 import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
 import { ContentTypes, GraphEvents, Providers } from '@/common';
 import { capitalizeFirstLetter } from './spec.utils';
@@ -36,7 +44,8 @@ describe(`${capitalizeFirstLetter(provider)} Streaming Tests`, () => {
   beforeEach(async () => {
     conversationHistory = [];
     collectedUsage = [];
-    const { contentParts: cp, aggregateContent: ac } = createContentAggregator();
+    const { contentParts: cp, aggregateContent: ac } =
+      createContentAggregator();
     contentParts = cp as t.MessageContentComplex[];
     aggregateContent = ac;
   });
@@ -49,36 +58,62 @@ describe(`${capitalizeFirstLetter(provider)} Streaming Tests`, () => {
     onRunStepSpy.mockReset();
   });
 
-  const setupCustomHandlers = (): Record<string | GraphEvents, t.EventHandler> => ({
+  const setupCustomHandlers = (): Record<
+    string | GraphEvents,
+    t.EventHandler
+  > => ({
     [GraphEvents.TOOL_END]: new ToolEndHandler(),
     [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(collectedUsage),
     [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
     [GraphEvents.ON_RUN_STEP_COMPLETED]: {
-      handle: (event: GraphEvents.ON_RUN_STEP_COMPLETED, data: t.StreamEventData): void => {
-        aggregateContent({ event, data: data as unknown as { result: t.ToolEndEvent; } });
-      }
+      handle: (
+        event: GraphEvents.ON_RUN_STEP_COMPLETED,
+        data: t.StreamEventData
+      ): void => {
+        aggregateContent({
+          event,
+          data: data as unknown as { result: t.ToolEndEvent },
+        });
+      },
     },
     [GraphEvents.ON_RUN_STEP]: {
-      handle: (event: GraphEvents.ON_RUN_STEP, data: t.StreamEventData, metadata, graph): void => {
+      handle: (
+        event: GraphEvents.ON_RUN_STEP,
+        data: t.StreamEventData,
+        metadata,
+        graph
+      ): void => {
         onRunStepSpy(event, data, metadata, graph);
         aggregateContent({ event, data: data as t.RunStep });
-      }
+      },
     },
     [GraphEvents.ON_RUN_STEP_DELTA]: {
-      handle: (event: GraphEvents.ON_RUN_STEP_DELTA, data: t.StreamEventData): void => {
+      handle: (
+        event: GraphEvents.ON_RUN_STEP_DELTA,
+        data: t.StreamEventData
+      ): void => {
         aggregateContent({ event, data: data as t.RunStepDeltaEvent });
-      }
+      },
     },
     [GraphEvents.ON_MESSAGE_DELTA]: {
-      handle: (event: GraphEvents.ON_MESSAGE_DELTA, data: t.StreamEventData, metadata, graph): void => {
+      handle: (
+        event: GraphEvents.ON_MESSAGE_DELTA,
+        data: t.StreamEventData,
+        metadata,
+        graph
+      ): void => {
         onMessageDeltaSpy(event, data, metadata, graph);
         aggregateContent({ event, data: data as t.MessageDeltaEvent });
-      }
+      },
     },
     [GraphEvents.TOOL_START]: {
-      handle: (_event: string, _data: t.StreamEventData, _metadata?: Record<string, unknown>): void => {
+      handle: (
+        _event: string,
+        _data: t.StreamEventData,
+        _metadata?: Record<string, unknown>
+      ): void => {
         // Handle tool start
-      }
+      },
     },
   });
 
@@ -93,7 +128,8 @@ describe(`${capitalizeFirstLetter(provider)} Streaming Tests`, () => {
         type: 'standard',
         llmConfig,
         tools: [new Calculator()],
-        instructions: 'You are a friendly AI assistant. Always address the user by their name.',
+        instructions:
+          'You are a friendly AI assistant. Always address the user by their name.',
         additional_instructions: `The user's name is ${userName} and they are located in ${location}.`,
       },
       returnContent: true,
@@ -109,7 +145,9 @@ describe(`${capitalizeFirstLetter(provider)} Streaming Tests`, () => {
 
     const finalContentParts = await run.processStream(inputs, config);
     expect(finalContentParts).toBeDefined();
-    const allTextParts = finalContentParts?.every((part) => part.type === ContentTypes.TEXT);
+    const allTextParts = finalContentParts?.every(
+      (part) => part.type === ContentTypes.TEXT
+    );
     expect(allTextParts).toBe(true);
     expect(collectedUsage.length).toBeGreaterThan(0);
     expect(collectedUsage[0].input_tokens).toBeGreaterThan(0);
@@ -117,26 +155,33 @@ describe(`${capitalizeFirstLetter(provider)} Streaming Tests`, () => {
 
     const finalMessages = run.getRunMessages();
     expect(finalMessages).toBeDefined();
-    conversationHistory.push(...finalMessages ?? []);
+    conversationHistory.push(...(finalMessages ?? []));
     expect(conversationHistory.length).toBeGreaterThan(1);
     runningHistory = conversationHistory.slice();
 
     expect(onMessageDeltaSpy).toHaveBeenCalled();
     expect(onMessageDeltaSpy.mock.calls.length).toBeGreaterThan(1);
-    expect((onMessageDeltaSpy.mock.calls[0][3] as StandardGraph).provider).toBeDefined();
+    expect(
+      (onMessageDeltaSpy.mock.calls[0][3] as StandardGraph).provider
+    ).toBeDefined();
 
     expect(onRunStepSpy).toHaveBeenCalled();
     expect(onRunStepSpy.mock.calls.length).toBeGreaterThan(0);
-    expect((onRunStepSpy.mock.calls[0][3] as StandardGraph).provider).toBeDefined();
+    expect(
+      (onRunStepSpy.mock.calls[0][3] as StandardGraph).provider
+    ).toBeDefined();
 
     const { handleLLMEnd, collected } = createMetadataAggregator();
     const titleResult = await run.generateTitle({
+      provider,
       inputText: userMessage,
       contentParts,
       chainOptions: {
-        callbacks: [{
-          handleLLMEnd,
-        }],
+        callbacks: [
+          {
+            handleLLMEnd,
+          },
+        ],
       },
     });
 
@@ -148,7 +193,10 @@ describe(`${capitalizeFirstLetter(provider)} Streaming Tests`, () => {
 
   test(`${capitalizeFirstLetter(provider)}: should follow-up`, async () => {
     console.log('Previous conversation length:', runningHistory.length);
-    console.log('Last message:', runningHistory[runningHistory.length - 1].content);
+    console.log(
+      'Last message:',
+      runningHistory[runningHistory.length - 1].content
+    );
     const { userName, location } = await getArgs();
     const llmConfig = getLLMConfig(provider);
     const customHandlers = setupCustomHandlers();
@@ -159,7 +207,8 @@ describe(`${capitalizeFirstLetter(provider)} Streaming Tests`, () => {
         type: 'standard',
         llmConfig,
         tools: [new Calculator()],
-        instructions: 'You are a friendly AI assistant. Always address the user by their name.',
+        instructions:
+          'You are a friendly AI assistant. Always address the user by their name.',
         additional_instructions: `The user's name is ${userName} and they are located in ${location}.`,
       },
       returnContent: true,
@@ -175,7 +224,9 @@ describe(`${capitalizeFirstLetter(provider)} Streaming Tests`, () => {
 
     const finalContentParts = await run.processStream(inputs, config);
     expect(finalContentParts).toBeDefined();
-    const allTextParts = finalContentParts?.every((part) => part.type === ContentTypes.TEXT);
+    const allTextParts = finalContentParts?.every(
+      (part) => part.type === ContentTypes.TEXT
+    );
     expect(allTextParts).toBe(true);
     expect(collectedUsage.length).toBeGreaterThan(0);
     expect(collectedUsage[0].input_tokens).toBeGreaterThan(0);
@@ -184,7 +235,10 @@ describe(`${capitalizeFirstLetter(provider)} Streaming Tests`, () => {
     const finalMessages = run.getRunMessages();
     expect(finalMessages).toBeDefined();
     expect(finalMessages?.length).toBeGreaterThan(0);
-    console.log(`${capitalizeFirstLetter(provider)} follow-up message:`, finalMessages?.[finalMessages.length - 1]?.content);
+    console.log(
+      `${capitalizeFirstLetter(provider)} follow-up message:`,
+      finalMessages?.[finalMessages.length - 1]?.content
+    );
 
     expect(onMessageDeltaSpy).toHaveBeenCalled();
     expect(onMessageDeltaSpy.mock.calls.length).toBeGreaterThan(1);
@@ -196,9 +250,12 @@ describe(`${capitalizeFirstLetter(provider)} Streaming Tests`, () => {
   test('should handle errors appropriately', async () => {
     // Test error scenarios
     await expect(async () => {
-      await run.processStream({
-        messages: [],
-      }, {} as any);
+      await run.processStream(
+        {
+          messages: [],
+        },
+        {} as any
+      );
     }).rejects.toThrow();
   });
-});
+});

package/src/specs/openai.simple.test.ts

@@ -4,10 +4,18 @@
 import { config } from 'dotenv';
 config();
 import { Calculator } from '@langchain/community/tools/calculator';
-import { HumanMessage, BaseMessage, UsageMetadata } from '@langchain/core/messages';
+import {
+  HumanMessage,
+  BaseMessage,
+  UsageMetadata,
+} from '@langchain/core/messages';
 import type { StandardGraph } from '@/graphs';
 import type * as t from '@/types';
-import { ToolEndHandler, ModelEndHandler, createMetadataAggregator } from '@/events';
+import {
+  ToolEndHandler,
+  ModelEndHandler,
+  createMetadataAggregator,
+} from '@/events';
 import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
 import { ContentTypes, GraphEvents, Providers } from '@/common';
 import { capitalizeFirstLetter } from './spec.utils';
@@ -36,7 +44,8 @@ describe(`${capitalizeFirstLetter(provider)} Streaming Tests`, () => {
   beforeEach(async () => {
     conversationHistory = [];
     collectedUsage = [];
-    const { contentParts: cp, aggregateContent: ac } = createContentAggregator();
+    const { contentParts: cp, aggregateContent: ac } =
+      createContentAggregator();
     contentParts = cp as t.MessageContentComplex[];
     aggregateContent = ac;
   });
@@ -49,36 +58,62 @@ describe(`${capitalizeFirstLetter(provider)} Streaming Tests`, () => {
     onRunStepSpy.mockReset();
   });
 
-  const setupCustomHandlers = (): Record<string | GraphEvents, t.EventHandler> => ({
+  const setupCustomHandlers = (): Record<
+    string | GraphEvents,
+    t.EventHandler
+  > => ({
     [GraphEvents.TOOL_END]: new ToolEndHandler(),
     [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(collectedUsage),
     [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
     [GraphEvents.ON_RUN_STEP_COMPLETED]: {
-      handle: (event: GraphEvents.ON_RUN_STEP_COMPLETED, data: t.StreamEventData): void => {
-        aggregateContent({ event, data: data as unknown as { result: t.ToolEndEvent; } });
-      }
+      handle: (
+        event: GraphEvents.ON_RUN_STEP_COMPLETED,
+        data: t.StreamEventData
+      ): void => {
+        aggregateContent({
+          event,
+          data: data as unknown as { result: t.ToolEndEvent },
+        });
+      },
     },
     [GraphEvents.ON_RUN_STEP]: {
-      handle: (event: GraphEvents.ON_RUN_STEP, data: t.StreamEventData, metadata, graph): void => {
+      handle: (
+        event: GraphEvents.ON_RUN_STEP,
+        data: t.StreamEventData,
+        metadata,
+        graph
+      ): void => {
         onRunStepSpy(event, data, metadata, graph);
         aggregateContent({ event, data: data as t.RunStep });
-      }
+      },
     },
     [GraphEvents.ON_RUN_STEP_DELTA]: {
-      handle: (event: GraphEvents.ON_RUN_STEP_DELTA, data: t.StreamEventData): void => {
+      handle: (
+        event: GraphEvents.ON_RUN_STEP_DELTA,
+        data: t.StreamEventData
+      ): void => {
         aggregateContent({ event, data: data as t.RunStepDeltaEvent });
-      }
+      },
     },
     [GraphEvents.ON_MESSAGE_DELTA]: {
-      handle: (event: GraphEvents.ON_MESSAGE_DELTA, data: t.StreamEventData, metadata, graph): void => {
+      handle: (
+        event: GraphEvents.ON_MESSAGE_DELTA,
+        data: t.StreamEventData,
+        metadata,
+        graph
+      ): void => {
         onMessageDeltaSpy(event, data, metadata, graph);
         aggregateContent({ event, data: data as t.MessageDeltaEvent });
-      }
+      },
     },
     [GraphEvents.TOOL_START]: {
-      handle: (_event: string, _data: t.StreamEventData, _metadata?: Record<string, unknown>): void => {
+      handle: (
+        _event: string,
+        _data: t.StreamEventData,
+        _metadata?: Record<string, unknown>
+      ): void => {
         // Handle tool start
-      }
+      },
     },
   });
 
@@ -93,7 +128,8 @@ describe(`${capitalizeFirstLetter(provider)} Streaming Tests`, () => {
         type: 'standard',
         llmConfig,
         tools: [new Calculator()],
-        instructions: 'You are a friendly AI assistant. Always address the user by their name.',
+        instructions:
+          'You are a friendly AI assistant. Always address the user by their name.',
         additional_instructions: `The user's name is ${userName} and they are located in ${location}.`,
       },
       returnContent: true,
@@ -109,7 +145,9 @@ describe(`${capitalizeFirstLetter(provider)} Streaming Tests`, () => {
 
     const finalContentParts = await run.processStream(inputs, config);
     expect(finalContentParts).toBeDefined();
-    const allTextParts = finalContentParts?.every((part) => part.type === ContentTypes.TEXT);
+    const allTextParts = finalContentParts?.every(
+      (part) => part.type === ContentTypes.TEXT
+    );
     expect(allTextParts).toBe(true);
     expect(collectedUsage.length).toBeGreaterThan(0);
     expect(collectedUsage[0].input_tokens).toBeGreaterThan(0);
@@ -117,26 +155,33 @@ describe(`${capitalizeFirstLetter(provider)} Streaming Tests`, () => {
 
     const finalMessages = run.getRunMessages();
     expect(finalMessages).toBeDefined();
-    conversationHistory.push(...finalMessages ?? []);
+    conversationHistory.push(...(finalMessages ?? []));
     expect(conversationHistory.length).toBeGreaterThan(1);
     runningHistory = conversationHistory.slice();
 
     expect(onMessageDeltaSpy).toHaveBeenCalled();
     expect(onMessageDeltaSpy.mock.calls.length).toBeGreaterThan(1);
-    expect((onMessageDeltaSpy.mock.calls[0][3] as StandardGraph).provider).toBeDefined();
+    expect(
+      (onMessageDeltaSpy.mock.calls[0][3] as StandardGraph).provider
+    ).toBeDefined();
 
     expect(onRunStepSpy).toHaveBeenCalled();
     expect(onRunStepSpy.mock.calls.length).toBeGreaterThan(0);
-    expect((onRunStepSpy.mock.calls[0][3] as StandardGraph).provider).toBeDefined();
+    expect(
+      (onRunStepSpy.mock.calls[0][3] as StandardGraph).provider
+    ).toBeDefined();
 
     const { handleLLMEnd, collected } = createMetadataAggregator();
     const titleResult = await run.generateTitle({
+      provider,
       inputText: userMessage,
       contentParts,
       chainOptions: {
-        callbacks: [{
-          handleLLMEnd,
-        }],
+        callbacks: [
+          {
+            handleLLMEnd,
+          },
+        ],
       },
     });
 
@@ -148,7 +193,10 @@ describe(`${capitalizeFirstLetter(provider)} Streaming Tests`, () => {
 
   test(`${capitalizeFirstLetter(provider)}: should follow-up`, async () => {
     console.log('Previous conversation length:', runningHistory.length);
-    console.log('Last message:', runningHistory[runningHistory.length - 1].content);
+    console.log(
+      'Last message:',
+      runningHistory[runningHistory.length - 1].content
+    );
     const { userName, location } = await getArgs();
     const llmConfig = getLLMConfig(provider);
     const customHandlers = setupCustomHandlers();
@@ -159,7 +207,8 @@ describe(`${capitalizeFirstLetter(provider)} Streaming Tests`, () => {
         type: 'standard',
         llmConfig,
         tools: [new Calculator()],
-        instructions: 'You are a friendly AI assistant. Always address the user by their name.',
+        instructions:
+          'You are a friendly AI assistant. Always address the user by their name.',
         additional_instructions: `The user's name is ${userName} and they are located in ${location}.`,
       },
       returnContent: true,
@@ -175,7 +224,9 @@ describe(`${capitalizeFirstLetter(provider)} Streaming Tests`, () => {
 
     const finalContentParts = await run.processStream(inputs, config);
     expect(finalContentParts).toBeDefined();
-    const allTextParts = finalContentParts?.every((part) => part.type === ContentTypes.TEXT);
+    const allTextParts = finalContentParts?.every(
+      (part) => part.type === ContentTypes.TEXT
+    );
     expect(allTextParts).toBe(true);
     expect(collectedUsage.length).toBeGreaterThan(0);
     expect(collectedUsage[0].input_tokens).toBeGreaterThan(0);
@@ -184,7 +235,10 @@ describe(`${capitalizeFirstLetter(provider)} Streaming Tests`, () => {
     const finalMessages = run.getRunMessages();
     expect(finalMessages).toBeDefined();
     expect(finalMessages?.length).toBeGreaterThan(0);
-    console.log(`${capitalizeFirstLetter(provider)} follow-up message:`, finalMessages?.[finalMessages.length - 1]?.content);
+    console.log(
+      `${capitalizeFirstLetter(provider)} follow-up message:`,
+      finalMessages?.[finalMessages.length - 1]?.content
+    );
 
     expect(onMessageDeltaSpy).toHaveBeenCalled();
     expect(onMessageDeltaSpy.mock.calls.length).toBeGreaterThan(1);
@@ -196,9 +250,12 @@ describe(`${capitalizeFirstLetter(provider)} Streaming Tests`, () => {
   test('should handle errors appropriately', async () => {
     // Test error scenarios
     await expect(async () => {
-      await run.processStream({
-        messages: [],
-      }, {} as any);
+      await run.processStream(
+        {
+          messages: [],
+        },
+        {} as any
+      );
     }).rejects.toThrow();
   });
-});
+});

package/src/stream.ts

@@ -244,6 +244,7 @@ hasToolCallChunks: ${hasToolCallChunks}
       content.every(
         (c) =>
           (c.type?.startsWith(ContentTypes.THINKING) ?? false) ||
+          (c.type?.startsWith(ContentTypes.REASONING) ?? false) ||
           (c.type?.startsWith(ContentTypes.REASONING_CONTENT) ?? false)
       )
     ) {
@@ -252,6 +253,7 @@ hasToolCallChunks: ${hasToolCallChunks}
           type: ContentTypes.THINK,
           think:
             (c as t.ThinkingContentText).thinking ??
+            (c as Partial<t.GoogleReasoningContentText>).reasoning ??
             (c as Partial<t.BedrockReasoningContentText>).reasoningText?.text ??
             '',
         })),
@@ -264,8 +266,9 @@ hasToolCallChunks: ${hasToolCallChunks}
       | undefined;
     if (
       Array.isArray(chunk.content) &&
-      (chunk.content[0]?.type === 'thinking' ||
-        chunk.content[0]?.type === 'reasoning_content')
+      (chunk.content[0]?.type === ContentTypes.THINKING ||
+        chunk.content[0]?.type === ContentTypes.REASONING ||
+        chunk.content[0]?.type === ContentTypes.REASONING_CONTENT)
     ) {
       reasoning_content = 'valid';
     }

package/src/types/llm.ts

@@ -17,6 +17,7 @@ import type {
 } from '@langchain/openai';
 import type { BedrockChatFields } from '@langchain/community/chat_models/bedrock/web';
 import type { GoogleGenerativeAIChatInput } from '@langchain/google-genai';
+import type { GeminiGenerationConfig } from '@langchain/google-common';
 import type { ChatVertexAIInput } from '@langchain/google-vertexai';
 import type { ChatDeepSeekCallOptions } from '@langchain/deepseek';
 import type { ChatOpenRouterCallOptions } from '@/llm/openrouter';
@@ -70,6 +71,7 @@ export type BedrockAnthropicInput = ChatBedrockConverseInput & {
 export type BedrockConverseClientOptions = ChatBedrockConverseInput;
 export type GoogleClientOptions = GoogleGenerativeAIChatInput & {
   customHeaders?: RequestOptions['customHeaders'];
+  thinkingConfig?: GeminiGenerationConfig['thinkingConfig'];
 };
 export type DeepSeekClientOptions = ChatDeepSeekCallOptions;
 export type XAIClientOptions = ChatXAIInput;

package/src/types/run.ts

@@ -26,6 +26,7 @@ export type StandardGraphConfig = BaseGraphConfig &
 
 export type RunTitleOptions = {
   inputText: string;
+  provider: e.Providers;
   contentParts: (s.MessageContentComplex | undefined)[];
   titlePrompt?: string;
   skipLanguage?: boolean;

package/src/types/stream.ts

@@ -251,6 +251,12 @@ export type ReasoningContentText = {
   think: string;
 };
 
+/** Vertex AI / Google Common - Reasoning Content Block Format */
+export type GoogleReasoningContentText = {
+  type: ContentTypes.REASONING;
+  reasoning: string;
+};
+
 /** Anthropic's Reasoning Content Block Format */
 export type ThinkingContentText = {
   type: ContentTypes.THINKING;