@lobehub/chat 1.79.4 → 1.79.6

package/CHANGELOG.md

@@ -2,6 +2,56 @@
 
 # Changelog
 
+### [Version 1.79.6](https://github.com/lobehub/lobe-chat/compare/v1.79.5...v1.79.6)
+
+<sup>Released on **2025-04-11**</sup>
+
+#### 🐛 Bug Fixes
+
+- **misc**: Fix `grok-3-mini` series calling.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### What's fixed
+
+- **misc**: Fix `grok-3-mini` series calling, closes [#7371](https://github.com/lobehub/lobe-chat/issues/7371) ([523c605](https://github.com/lobehub/lobe-chat/commit/523c605))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
+### [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>

package/changelog/v1.json

@@ -1,4 +1,22 @@
 [
+  {
+    "children": {
+      "fixes": [
+        "Fix grok-3-mini series calling."
+      ]
+    },
+    "date": "2025-04-11",
+    "version": "1.79.6"
+  },
+  {
+    "children": {
+      "fixes": [
+        "Fix langfuse intergation."
+      ]
+    },
+    "date": "2025-04-10",
+    "version": "1.79.5"
+  },
   {
     "children": {
       "improvements": [

package/docs/self-hosting/advanced/online-search.zh-CN.mdx

@@ -0,0 +1,59 @@
+---
+title: 配置联网搜索功能 - 增强 AI 的网络信息获取能力
+description: 了解如何为 LobeChat 配置 SearXNG 联网搜索功能，使 AI 能够获取最新的网络信息。
+tags:
+  - 联网搜索
+  - SearXNG
+  - 网络信息
+  - AI 增强
+---
+
+# 配置联网搜索功能
+
+LobeChat 支持为 AI 配置联网搜索功能，这使得 AI 能够获取最新的网络信息，从而提供更准确、更及时的回答。联网搜索功能基于 [SearXNG](https://github.com/searxng/searxng) 搜索引擎，它是一个尊重隐私的元搜索引擎，可以聚合多个搜索引擎的结果。
+
+<Callout type={'info'}>
+  SearXNG 是一个开源的元搜索引擎，可以自行部署，也可以使用公共实例。通过配置 SearXNG，LobeChat 可以让 AI 获取最新的互联网信息，从而回答时效性问题、提供最新资讯。
+</Callout>
+
+# 核心环境变量
+
+## `SEARXNG_URL`
+
+SearXNG 实例的 URL 地址，这是启用联网搜索功能的必要配置。例如：
+
+```shell
+SEARXNG_URL=https://searxng-instance.com
+```
+
+这个 URL 应该指向一个可用的 SearXNG 实例。您可以选择自行部署 SearXNG，或使用公共可用的 SearXNG 实例。
+
+您可以在 [SearXNG 实例列表](https://searx.space/) 中找到公开可用的 SearXNG 实例。选择一个响应速度快、可靠性高的实例，然后将其 URL 配置到 LobeChat 中。
+
+> 注意，使用的 `searxng` 必须开启 `json` 输出，否则 `lobe-chat` 调用会报错。
+> 如果是自托管，类似下面这样，找到 `searxng` 的配置文件，追加 `json` 即可。
+
+```bash
+$ vi searxng/settings.yml
+...
+search:
+  formats:
+    - html
+    - json
+```
+
+# 验证联网搜索功能
+
+配置完成后，您可以通过以下步骤验证联网搜索功能是否正常工作：
+
+1. 重启 LobeChat 服务
+2. 启动一个新的聊天，启动智能联网，之后向 AI 提问一个需要最新信息的问题，例如："今天的实时金价是多少？"或"最近的重大新闻有哪些？"
+3. 观察 AI 是否能够返回基于互联网搜索的最新信息
+
+如果 AI 能够回答这些时效性问题，说明联网搜索功能已经成功配置。
+
+## 参考资料
+
+- [LobeChat 联网搜索 RFC 讨论](https://github.com/lobehub/lobe-chat/discussions/6447)
+- [SearXNG GitHub 仓库](https://github.com/searxng/searxng)
+- [SearXNG 开启 json 输出的讨论](https://github.com/searxng/searxng/discussions/3542)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "1.79.4",
+  "version": "1.79.6",
   "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/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);
   });

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

@@ -22,13 +22,11 @@ describe('OllamaStream', () => {
 
       const onStartMock = vi.fn();
       const onTextMock = vi.fn();
-      const onTokenMock = vi.fn();
       const onCompletionMock = vi.fn();
 
       const protocolStream = OllamaStream(mockOllamaStream, {
         onStart: onStartMock,
         onText: onTextMock,
-        onToken: onTokenMock,
         onCompletion: onCompletionMock,
       });
 
@@ -53,9 +51,8 @@ describe('OllamaStream', () => {
       ]);
 
       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);
     });
 
@@ -100,16 +97,14 @@ describe('OllamaStream', () => {
       });
       const onStartMock = vi.fn();
       const onTextMock = vi.fn();
-      const onTokenMock = vi.fn();
       const onToolCall = vi.fn();
       const onCompletionMock = vi.fn();
 
       const protocolStream = OllamaStream(mockOllamaStream, {
         onStart: onStartMock,
         onText: onTextMock,
-        onToken: onTokenMock,
         onCompletion: onCompletionMock,
-        onToolCall,
+        onToolsCalling: onToolCall,
       });
 
       const decoder = new TextDecoder();
@@ -134,7 +129,6 @@ describe('OllamaStream', () => {
       expect(onTextMock).toHaveBeenCalledTimes(0);
       expect(onStartMock).toHaveBeenCalledTimes(1);
       expect(onToolCall).toHaveBeenCalledTimes(1);
-      expect(onTokenMock).toHaveBeenCalledTimes(0);
       expect(onCompletionMock).toHaveBeenCalledTimes(1);
     });
   });