@lobehub/chat 1.79.3 → 1.79.5

package/CHANGELOG.md

@@ -2,6 +2,56 @@
 
 # Changelog
 
+### [Version 1.79.5](https://github.com/lobehub/lobe-chat/compare/v1.79.4...v1.79.5)
+
+<sup>Released on **2025-04-10**</sup>
+
+#### 🐛 Bug Fixes
+
+- **misc**: Fix langfuse intergation.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### What's fixed
+
+- **misc**: Fix langfuse intergation, closes [#7367](https://github.com/lobehub/lobe-chat/issues/7367) ([22b5236](https://github.com/lobehub/lobe-chat/commit/22b5236))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
+### [Version 1.79.4](https://github.com/lobehub/lobe-chat/compare/v1.79.3...v1.79.4)
+
+<sup>Released on **2025-04-10**</sup>
+
+#### 💄 Styles
+
+- **misc**: Update Grok 3 models.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### Styles
+
+- **misc**: Update Grok 3 models, closes [#7360](https://github.com/lobehub/lobe-chat/issues/7360) ([d2b9120](https://github.com/lobehub/lobe-chat/commit/d2b9120))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
 ### [Version 1.79.3](https://github.com/lobehub/lobe-chat/compare/v1.79.2...v1.79.3)
 
 <sup>Released on **2025-04-10**</sup>

package/changelog/v1.json

@@ -1,4 +1,22 @@
 [
+  {
+    "children": {
+      "fixes": [
+        "Fix langfuse intergation."
+      ]
+    },
+    "date": "2025-04-10",
+    "version": "1.79.5"
+  },
+  {
+    "children": {
+      "improvements": [
+        "Update Grok 3 models."
+      ]
+    },
+    "date": "2025-04-10",
+    "version": "1.79.4"
+  },
   {
     "children": {
       "fixes": [

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "1.79.3",
+  "version": "1.79.5",
   "description": "Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.",
   "keywords": [
     "framework",
@@ -189,8 +189,8 @@
     "js-sha256": "^0.11.0",
     "jsonl-parse-stringify": "^1.0.3",
     "langchain": "^0.3.19",
-    "langfuse": "3.29.1",
-    "langfuse-core": "3.29.1",
+    "langfuse": "^3.37.1",
+    "langfuse-core": "^3.37.1",
     "lodash-es": "^4.17.21",
     "lucide-react": "^0.487.0",
     "mammoth": "^1.9.0",

package/src/app/(backend)/webapi/chat/[provider]/route.ts

@@ -33,10 +33,7 @@ export const POST = checkAuth(async (req: Request, { params, jwtPayload, createR
     let traceOptions = {};
     // If user enable trace
     if (tracePayload?.enabled) {
-      traceOptions = createTraceOptions(data, {
-        provider,
-        trace: tracePayload,
-      });
+      traceOptions = createTraceOptions(data, { provider, trace: tracePayload });
     }
 
     return await agentRuntime.chat(data, {

package/src/app/(backend)/webapi/plugin/gateway/route.ts

@@ -52,6 +52,7 @@ export const POST = async (req: Request) => {
     return createErrorResponse(result.error as ErrorType);
   }
 
+  // TODO: need to be replace by better telemetry system
   // add trace
   const tracePayload = getTracePayload(req);
   const traceClient = new TraceClient();

package/src/app/(backend)/webapi/trace/route.ts

@@ -1,3 +1,5 @@
+import { after } from 'next/server';
+
 import { TraceEventType } from '@/const/trace';
 import { TraceClient } from '@/libs/traces';
 import { TraceEventBasePayload, TraceEventPayloads } from '@/types/trace';
@@ -35,6 +37,9 @@ export const POST = async (req: Request) => {
     }
   }
 
-  await traceClient.shutdownAsync();
+  after(async () => {
+    await traceClient.shutdownAsync();
+  });
+
   return new Response(undefined, { status: 201 });
 };

package/src/config/aiModels/xai.ts

@@ -1,51 +1,68 @@
 import { AIChatModelCard } from '@/types/aiModel';
-
+// https://docs.x.ai/docs/models
 const xaiChatModels: AIChatModelCard[] = [
   {
     abilities: {
       functionCall: true,
     },
     contextWindowTokens: 131_072,
-    description: '拥有与 Grok 2 相当的性能，但具有更高的效率、速度和功能。',
-    displayName: 'Grok Beta',
+    description: '旗舰级模型，擅长数据提取、编程和文本摘要等企业级应用，拥有金融、医疗、法律和科学等领域的深厚知识。',
+    displayName: 'Grok 3 Beta',
     enabled: true,
-    id: 'grok-beta',
+    id: 'grok-3-beta',
     pricing: {
-      input: 5,
+      input: 3,
       output: 15,
     },
+    releasedAt: '2025-04-03',
     type: 'chat',
   },
   {
     abilities: {
       functionCall: true,
-      vision: true,
     },
-    contextWindowTokens: 8192,
-    description: '最新的图像理解模型，可以处理各种各样的视觉信息，包括文档、图表、截图和照片等。',
-    displayName: 'Grok Vision Beta',
-    enabled: true,
-    id: 'grok-vision-beta',
+    contextWindowTokens: 131_072,
+    description: '旗舰级模型，擅长数据提取、编程和文本摘要等企业级应用，拥有金融、医疗、法律和科学等领域的深厚知识。',
+    displayName: 'Grok 3 Beta (Fast mode)',
+    id: 'grok-3-fast-beta',
     pricing: {
       input: 5,
-      output: 15,
+      output: 25,
     },
+    releasedAt: '2025-04-03',
     type: 'chat',
   },
   {
     abilities: {
       functionCall: true,
+      reasoning: true,
     },
     contextWindowTokens: 131_072,
-    description: '该模型在准确性、指令遵循和多语言能力方面有所改进。',
-    displayName: 'Grok 2 1212',
+    description: '轻量级模型，回话前会先思考。运行快速、智能，适用于不需要深层领域知识的逻辑任务，并能获取原始的思维轨迹。',
+    displayName: 'Grok 3 Mini Beta',
     enabled: true,
-    id: 'grok-2-1212',
+    id: 'grok-3-mini-beta',
     pricing: {
-      input: 2,
-      output: 10,
+      input: 0.3,
+      output: 0.5,
     },
-    releasedAt: '2024-12-12',
+    releasedAt: '2025-04-03',
+    type: 'chat',
+  },
+  {
+    abilities: {
+      functionCall: true,
+      reasoning: true,
+    },
+    contextWindowTokens: 131_072,
+    description: '轻量级模型，回话前会先思考。运行快速、智能，适用于不需要深层领域知识的逻辑任务，并能获取原始的思维轨迹。',
+    displayName: 'Grok 3 Mini Beta (Fast mode)',
+    id: 'grok-3-mini-fast-beta',
+    pricing: {
+      input: 0.6,
+      output: 4,
+    },
+    releasedAt: '2025-04-03',
     type: 'chat',
   },
   {
@@ -65,6 +82,50 @@ const xaiChatModels: AIChatModelCard[] = [
     releasedAt: '2024-12-12',
     type: 'chat',
   },
+  {
+    abilities: {
+      functionCall: true,
+    },
+    contextWindowTokens: 131_072,
+    description: '该模型在准确性、指令遵循和多语言能力方面有所改进。',
+    displayName: 'Grok 2 1212',
+    id: 'grok-2-1212', // legacy
+    pricing: {
+      input: 2,
+      output: 10,
+    },
+    releasedAt: '2024-12-12',
+    type: 'chat',
+  },
+  {
+    abilities: {
+      functionCall: true,
+    },
+    contextWindowTokens: 131_072,
+    description: '拥有与 Grok 2 相当的性能，但具有更高的效率、速度和功能。',
+    displayName: 'Grok Beta',
+    id: 'grok-beta', // legacy
+    pricing: {
+      input: 5,
+      output: 15,
+    },
+    type: 'chat',
+  },
+  {
+    abilities: {
+      functionCall: true,
+      vision: true,
+    },
+    contextWindowTokens: 8192,
+    description: '最新的图像理解模型，可以处理各种各样的视觉信息，包括文档、图表、截图和照片等。',
+    displayName: 'Grok Vision Beta',
+    id: 'grok-vision-beta', // legacy
+    pricing: {
+      input: 5,
+      output: 15,
+    },
+    type: 'chat',
+  },
 ];
 
 export const allModels = [...xaiChatModels];

package/src/const/trace.ts

@@ -7,15 +7,13 @@ export enum TraceNameMap {
   Conversation = 'Conversation',
   EmojiPicker = 'Emoji Picker',
   FetchPluginAPI = 'Fetch Plugin API',
-  InvokePlugin = 'Invoke Plugin',
   LanguageDetect = 'Language Detect',
+  SearchIntentRecognition = 'Search Intent Recognition',
   SummaryAgentDescription = 'Summary Agent Description',
   SummaryAgentTags = 'Summary Agent Tags',
   SummaryAgentTitle = 'Summary Agent Title',
   SummaryTopicTitle = 'Summary Topic Title',
   Translator = 'Translator',
-  // mean user have relative events
-  UserEvents = 'User Events',
 }
 
 export enum TraceEventType {
@@ -33,7 +31,7 @@ export enum TraceTagMap {
   Chat = 'Chat Competition',
   SystemChain = 'System Chain',
   ToolCalling = 'Tool Calling',
-  ToolsCall = 'Tools Call',
+  ToolsCalling = 'Tools Calling',
 }
 
 export interface TracePayload {

package/src/database/models/__tests__/_util.ts

@@ -1,11 +1,13 @@
 import { clientDB, initializeDB } from '@/database/client/db';
-import { getTestDBInstance } from '@/database/core/dbForTest';
 import { LobeChatDatabase } from '@/database/type';
 
 const isServerDBMode = process.env.TEST_SERVER_DB === '1';
 
 export const getTestDB = async () => {
-  if (isServerDBMode) return await getTestDBInstance();
+  if (isServerDBMode) {
+    const { getTestDBInstance } = await import('@/database/core/dbForTest');
+    return await getTestDBInstance();
+  }
 
   await initializeDB();
   return clientDB as LobeChatDatabase;

package/src/libs/agent-runtime/AgentRuntime.test.ts

@@ -66,10 +66,7 @@ const testRuntime = (providerId: string, payload?: any) => {
 let mockModelRuntime: AgentRuntime;
 beforeEach(async () => {
   const jwtPayload: JWTPayload = { apiKey: 'user-openai-key', baseURL: 'user-endpoint' };
-  mockModelRuntime = await AgentRuntime.initializeWithProvider(
-    ModelProvider.OpenAI,
-    jwtPayload,
-  );
+  mockModelRuntime = await AgentRuntime.initializeWithProvider(ModelProvider.OpenAI, jwtPayload);
 });
 
 describe('AgentRuntime', () => {
@@ -112,7 +109,7 @@ describe('AgentRuntime', () => {
         provider: 'openai',
         trace: {
           traceId: 'test-trace-id',
-          traceName: TraceNameMap.Conversation,
+          traceName: TraceNameMap.SummaryTopicTitle,
           sessionId: 'test-session-id',
           topicId: 'test-topic-id',
           tags: [],
@@ -136,7 +133,7 @@ describe('AgentRuntime', () => {
         provider: 'openai',
         trace: {
           traceId: 'test-trace-id',
-          traceName: TraceNameMap.Conversation,
+          traceName: TraceNameMap.SummaryTopicTitle,
           sessionId: 'test-session-id',
           topicId: 'test-topic-id',
           tags: [],
@@ -147,7 +144,7 @@ describe('AgentRuntime', () => {
 
       const updateMock = vi.fn();
 
-      it('should call experimental_onToolCall correctly', async () => {
+      it('should call onToolsCalling correctly', async () => {
         vi.spyOn(langfuseCfg, 'getLangfuseConfig').mockReturnValue({
           ENABLE_LANGFUSE: true,
           LANGFUSE_PUBLIC_KEY: 'abc',
@@ -157,9 +154,9 @@ describe('AgentRuntime', () => {
         // 使用 spyOn 模拟 chat 方法
         vi.spyOn(LobeOpenAI.prototype, 'chat').mockImplementation(
           async (payload, { callback }: any) => {
-            // 模拟 experimental_onToolCall 回调的触发
-            if (callback?.experimental_onToolCall) {
-              await callback.experimental_onToolCall();
+            // 模拟 onToolCall 回调的触发
+            if (callback?.onToolsCalling) {
+              await callback.onToolsCalling();
             }
             return new Response('abc');
           },
@@ -168,7 +165,7 @@ describe('AgentRuntime', () => {
 
         await mockModelRuntime.chat(payload, createTraceOptions(payload, options));
 
-        expect(updateMock).toHaveBeenCalledWith({ tags: ['Tools Call'] });
+        expect(updateMock).toHaveBeenCalledWith({ tags: ['Tools Calling'] });
       });
       it('should call onStart correctly', async () => {
         vi.spyOn(langfuseCfg, 'getLangfuseConfig').mockReturnValue({
@@ -204,7 +201,7 @@ describe('AgentRuntime', () => {
         vi.spyOn(LobeOpenAI.prototype, 'chat').mockImplementation(
           async (payload, { callback }: any) => {
             if (callback?.onCompletion) {
-              await callback.onCompletion('Test completion');
+              await callback.onCompletion({ text: 'Test completion' });
             }
             return new Response('Success');
           },
@@ -215,14 +212,11 @@ describe('AgentRuntime', () => {
         // Verify onCompletion was called with expected output
         expect(updateMock).toHaveBeenCalledWith({
           endTime: expect.any(Date),
-          metadata: {
-            provider: 'openai',
-            tools: undefined,
-          },
+          metadata: {},
           output: 'Test completion',
         });
       });
-      it('should call onFinal correctly', async () => {
+      it.skip('should call onFinal correctly', async () => {
         vi.spyOn(langfuseCfg, 'getLangfuseConfig').mockReturnValue({
           ENABLE_LANGFUSE: true,
           LANGFUSE_PUBLIC_KEY: 'abc',

package/src/libs/agent-runtime/helpers/index.ts

@@ -0,0 +1 @@
+export * from './parseToolCalls';

package/src/{utils/fetch/__tests__ → libs/agent-runtime/helpers}/parseToolCalls.test.ts

@@ -1,7 +1,6 @@
 import { describe, expect, it } from 'vitest';
-import { ZodError } from 'zod';
 
-import { parseToolCalls } from '../parseToolCalls';
+import { parseToolCalls } from './parseToolCalls';
 
 describe('parseToolCalls', () => {
   it('should create add new item', () => {

package/src/libs/agent-runtime/index.ts

@@ -8,6 +8,7 @@ export { LobeDeepSeekAI } from './deepseek';
 export * from './error';
 export { LobeGoogleAI } from './google';
 export { LobeGroq } from './groq';
+export * from './helpers';
 export { LobeMinimaxAI } from './minimax';
 export { LobeMistralAI } from './mistral';
 export { LobeMoonshotAI } from './moonshot';

package/src/libs/agent-runtime/types/chat.ts

@@ -1,4 +1,25 @@
-import { MessageToolCall } from '@/types/message';
+import { DeepPartial } from 'utility-types';
+
+import { ModelTokensUsage, ToolFunction } from '@/types/message';
+
+export interface MessageToolCall {
+  /**
+   * The function that the model called.
+   */
+  function: ToolFunction;
+
+  /**
+   * The ID of the tool call.
+   */
+  id: string;
+
+  /**
+   * The type of the tool. Currently, only `function` is supported.
+   */
+  type: 'function' | string;
+}
+
+export type MessageToolCallChunk = DeepPartial<MessageToolCall> & { index: number };
 
 export type LLMRoleType = 'user' | 'system' | 'assistant' | 'function' | 'tool';
 
@@ -165,18 +186,29 @@ export interface ChatCompletionTool {
   type: 'function';
 }
 
+interface OnFinishData {
+  grounding?: any;
+  text: string;
+  thinking?: string;
+  toolsCalling?: MessageToolCall[];
+  usage?: ModelTokensUsage;
+}
+
 export interface ChatStreamCallbacks {
+  onCompletion?: (data: OnFinishData) => Promise<void> | void;
   /**
-   * `onCompletion`: Called for each tokenized message.
+   * `onFinal`: Called once when the stream is closed with the final completion message.
    **/
-  onCompletion?: (completion: string) => Promise<void> | void;
-  /** `onFinal`: Called once when the stream is closed with the final completion message. */
-  onFinal?: (completion: string) => Promise<void> | void;
+  onFinal?: (data: OnFinishData) => Promise<void> | void;
+  onGrounding?: (grounding: any) => Promise<void> | void;
   /** `onStart`: Called once when the stream is initialized. */
   onStart?: () => Promise<void> | void;
   /** `onText`: Called for each text chunk. */
-  onText?: (text: string) => Promise<void> | void;
-  /** `onToken`: Called for each tokenized message. */
-  onToken?: (token: string) => Promise<void> | void;
-  onToolCall?: () => Promise<void> | void;
+  onText?: (content: string) => Promise<void> | void;
+  onThinking?: (content: string) => Promise<void> | void;
+  onToolsCalling?: (data: {
+    chunk: MessageToolCallChunk[];
+    toolsCalling: MessageToolCall[];
+  }) => Promise<void> | void;
+  onUsage?: (usage: ModelTokensUsage) => Promise<void> | void;
 }

package/src/libs/agent-runtime/utils/openaiCompatibleFactory/index.test.ts

@@ -776,7 +776,7 @@ describe('LobeOpenAICompatibleFactory', () => {
         // 准备 callback 和 headers
         const mockCallback: ChatStreamCallbacks = {
           onStart: vi.fn(),
-          onToken: vi.fn(),
+          onCompletion: vi.fn(),
         };
         const mockHeaders = { 'Custom-Header': 'TestValue' };
 
@@ -793,7 +793,9 @@ describe('LobeOpenAICompatibleFactory', () => {
         // 验证 callback 被调用
         await result.text(); // 确保流被消费
         expect(mockCallback.onStart).toHaveBeenCalled();
-        expect(mockCallback.onToken).toHaveBeenCalledWith('hello');
+        expect(mockCallback.onCompletion).toHaveBeenCalledWith({
+          text: 'hello',
+        });
 
         // 验证 headers 被正确传递
         expect(result.headers.get('Custom-Header')).toEqual('TestValue');

package/src/libs/agent-runtime/utils/streams/anthropic.test.ts

@@ -58,13 +58,11 @@ describe('AnthropicStream', () => {
 
     const onStartMock = vi.fn();
     const onTextMock = vi.fn();
-    const onTokenMock = vi.fn();
     const onCompletionMock = vi.fn();
 
     const protocolStream = AnthropicStream(mockAnthropicStream, {
       onStart: onStartMock,
       onText: onTextMock,
-      onToken: onTokenMock,
       onCompletion: onCompletionMock,
     });
 
@@ -92,9 +90,8 @@ describe('AnthropicStream', () => {
     ]);
 
     expect(onStartMock).toHaveBeenCalledTimes(1);
-    expect(onTextMock).toHaveBeenNthCalledWith(1, '"Hello"');
-    expect(onTextMock).toHaveBeenNthCalledWith(2, '" world!"');
-    expect(onTokenMock).toHaveBeenCalledTimes(2);
+    expect(onTextMock).toHaveBeenNthCalledWith(1, 'Hello');
+    expect(onTextMock).toHaveBeenNthCalledWith(2, ' world!');
     expect(onCompletionMock).toHaveBeenCalledTimes(1);
   });
 
@@ -168,7 +165,7 @@ describe('AnthropicStream', () => {
     const onToolCallMock = vi.fn();
 
     const protocolStream = AnthropicStream(mockReadableStream, {
-      onToolCall: onToolCallMock,
+      onToolsCalling: onToolCallMock,
     });
 
     const decoder = new TextDecoder();
@@ -320,7 +317,7 @@ describe('AnthropicStream', () => {
     const onToolCallMock = vi.fn();
 
     const protocolStream = AnthropicStream(mockReadableStream, {
-      onToolCall: onToolCallMock,
+      onToolsCalling: onToolCallMock,
     });
 
     const decoder = new TextDecoder();

package/src/libs/agent-runtime/utils/streams/bedrock/llama.test.ts

@@ -20,13 +20,11 @@ describe('AWSBedrockLlamaStream', () => {
 
     const onStartMock = vi.fn();
     const onTextMock = vi.fn();
-    const onTokenMock = vi.fn();
     const onCompletionMock = vi.fn();
 
     const protocolStream = AWSBedrockLlamaStream(mockBedrockStream, {
       onStart: onStartMock,
       onText: onTextMock,
-      onToken: onTokenMock,
       onCompletion: onCompletionMock,
     });
 
@@ -51,9 +49,8 @@ describe('AWSBedrockLlamaStream', () => {
     ]);
 
     expect(onStartMock).toHaveBeenCalledTimes(1);
-    expect(onTextMock).toHaveBeenNthCalledWith(1, '"Hello"');
-    expect(onTextMock).toHaveBeenNthCalledWith(2, '" world!"');
-    expect(onTokenMock).toHaveBeenCalledTimes(2);
+    expect(onTextMock).toHaveBeenNthCalledWith(1, 'Hello');
+    expect(onTextMock).toHaveBeenNthCalledWith(2, ' world!');
     expect(onCompletionMock).toHaveBeenCalledTimes(1);
   });
 
@@ -73,13 +70,11 @@ describe('AWSBedrockLlamaStream', () => {
 
     const onStartMock = vi.fn();
     const onTextMock = vi.fn();
-    const onTokenMock = vi.fn();
     const onCompletionMock = vi.fn();
 
     const protocolStream = AWSBedrockLlamaStream(mockBedrockStream, {
       onStart: onStartMock,
       onText: onTextMock,
-      onToken: onTokenMock,
       onCompletion: onCompletionMock,
     });
 
@@ -104,9 +99,8 @@ describe('AWSBedrockLlamaStream', () => {
     ]);
 
     expect(onStartMock).toHaveBeenCalledTimes(1);
-    expect(onTextMock).toHaveBeenNthCalledWith(1, '"Hello"');
-    expect(onTextMock).toHaveBeenNthCalledWith(2, '" world!"');
-    expect(onTokenMock).toHaveBeenCalledTimes(2);
+    expect(onTextMock).toHaveBeenNthCalledWith(1, 'Hello');
+    expect(onTextMock).toHaveBeenNthCalledWith(2, ' world!');
     expect(onCompletionMock).toHaveBeenCalledTimes(1);
   });
 
@@ -143,7 +137,6 @@ describe('AWSBedrockLlamaStream', () => {
     const protocolStream = AWSBedrockLlamaStream(mockBedrockStream, {
       onStart: onStartMock,
       onText: onTextMock,
-      onToken: onTokenMock,
       onCompletion: onCompletionMock,
     });
 
@@ -168,9 +161,8 @@ describe('AWSBedrockLlamaStream', () => {
     ]);
 
     expect(onStartMock).toHaveBeenCalledTimes(1);
-    expect(onTextMock).toHaveBeenNthCalledWith(1, '"Hello"');
-    expect(onTextMock).toHaveBeenNthCalledWith(2, '" world!"');
-    expect(onTokenMock).toHaveBeenCalledTimes(2);
+    expect(onTextMock).toHaveBeenNthCalledWith(1, 'Hello');
+    expect(onTextMock).toHaveBeenNthCalledWith(2, ' world!');
     expect(onCompletionMock).toHaveBeenCalledTimes(1);
   });
 

package/src/libs/agent-runtime/utils/streams/google-ai.test.ts

@@ -30,15 +30,13 @@ describe('GoogleGenerativeAIStream', () => {
 
     const onStartMock = vi.fn();
     const onTextMock = vi.fn();
-    const onTokenMock = vi.fn();
     const onToolCallMock = vi.fn();
     const onCompletionMock = vi.fn();
 
     const protocolStream = GoogleGenerativeAIStream(mockGoogleStream, {
       onStart: onStartMock,
       onText: onTextMock,
-      onToken: onTokenMock,
-      onToolCall: onToolCallMock,
+      onToolsCalling: onToolCallMock,
       onCompletion: onCompletionMock,
     });
 
@@ -68,9 +66,8 @@ describe('GoogleGenerativeAIStream', () => {
     ]);
 
     expect(onStartMock).toHaveBeenCalledTimes(1);
-    expect(onTextMock).toHaveBeenNthCalledWith(1, '"Hello"');
-    expect(onTextMock).toHaveBeenNthCalledWith(2, '" world!"');
-    expect(onTokenMock).toHaveBeenCalledTimes(2);
+    expect(onTextMock).toHaveBeenNthCalledWith(1, 'Hello');
+    expect(onTextMock).toHaveBeenNthCalledWith(2, ' world!');
     expect(onToolCallMock).toHaveBeenCalledTimes(1);
     expect(onCompletionMock).toHaveBeenCalledTimes(1);
   });