♻️ refactor: extract agent-stream into @lobechat/agent-gateway-client package (#13866)

* ♻️ refactor: extract agent-stream into @lobechat/agent-gateway-client package

Move the Agent Gateway WebSocket client from src/libs/agent-stream/ into
a standalone workspace package at packages/agent-gateway-client/. This
eliminates the duplicate AgentStreamEvent type in apps/cli and provides
a single source of truth for the Gateway WS protocol types shared by
SPA, server, and CLI consumers.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

* add agent-gateway-client

---------

Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

apps/cli/package.json

@@ -28,6 +28,7 @@
     "type-check": "tsc --noEmit"
   },
   "devDependencies": {
+    "@lobechat/agent-gateway-client": "workspace:*",
     "@lobechat/device-gateway-client": "workspace:*",
     "@lobechat/local-file-shell": "workspace:*",
     "@trpc/client": "^11.8.1",

apps/cli/pnpm-workspace.yaml

@@ -1,4 +1,5 @@
 packages:
+  - '../../packages/agent-gateway-client'
   - '../../packages/device-gateway-client'
   - '../../packages/local-file-shell'
   - '../../packages/file-loaders'

apps/cli/src/utils/agentStream.ts

@@ -1,16 +1,10 @@
+import type { AgentStreamEvent } from '@lobechat/agent-gateway-client';
 import pc from 'picocolors';
 import urlJoin from 'url-join';
 
 import { log } from './logger';
 
-export interface AgentStreamEvent {
-  data: any;
-  id?: string;
-  operationId: string;
-  stepIndex: number;
-  timestamp: number;
-  type: string;
-}
+export type { AgentStreamEvent } from '@lobechat/agent-gateway-client';
 
 interface StreamOptions {
   json?: boolean;

package.json

@@ -200,6 +200,7 @@
     "@khmyznikov/pwa-install": "0.3.9",
     "@larksuiteoapi/node-sdk": "^1.60.0",
     "@lexical/utils": "^0.42.0",
+    "@lobechat/agent-gateway-client": "workspace:*",
     "@lobechat/agent-runtime": "workspace:*",
     "@lobechat/agent-templates": "workspace:*",
     "@lobechat/builtin-agents": "workspace:*",

packages/agent-gateway-client/package.json

@@ -0,0 +1,18 @@
+{
+  "name": "@lobechat/agent-gateway-client",
+  "version": "1.0.0",
+  "private": true,
+  "description": "Browser-compatible WebSocket client for the LobeHub Agent Gateway",
+  "exports": {
+    ".": "./src/index.ts"
+  },
+  "main": "./src/index.ts",
+  "types": "./src/index.ts",
+  "scripts": {
+    "test": "bunx vitest run --silent='passed-only'",
+    "test:coverage": "bunx vitest run --coverage"
+  },
+  "devDependencies": {
+    "vitest": "^3.0.0"
+  }
+}

packages/agent-gateway-client/vitest.config.mts

@@ -0,0 +1,18 @@
+import { defineConfig } from 'vitest/config';
+
+export default defineConfig({
+  test: {
+    coverage: {
+      exclude: [
+        'coverage/**',
+        'dist/**',
+        '**/node_modules/**',
+        '**/*.d.ts',
+        '**/*.config.*',
+        '**/index.ts',
+      ],
+      reporter: ['text', 'json', 'lcov', 'text-summary'],
+    },
+    environment: 'happy-dom',
+  },
+});

src/server/modules/AgentRuntime/GatewayStreamNotifier.ts

@@ -1,8 +1,7 @@
+import type { ToolExecuteData } from '@lobechat/agent-gateway-client';
 import debug from 'debug';
 import urlJoin from 'url-join';
 
-import type { ToolExecuteData } from '@/libs/agent-stream/types';
-
 import {
   getDefaultReasonDetail,
   type StreamChunkData,

src/server/modules/AgentRuntime/types.ts

@@ -1,7 +1,6 @@
+import type { ToolExecuteData } from '@lobechat/agent-gateway-client';
 import { type AgentState } from '@lobechat/agent-runtime';
 
-import type { ToolExecuteData } from '@/libs/agent-stream/types';
-
 import { type AgentOperationMetadata, type StepResult } from './AgentStateManager';
 import { type StreamChunkData, type StreamEvent } from './StreamEventManager';
 

src/store/chat/slices/aiChat/actions/__tests__/clientToolExecution.test.ts

@@ -1,7 +1,6 @@
+import type { ToolExecuteData } from '@lobechat/agent-gateway-client';
 import { beforeEach, describe, expect, it, vi } from 'vitest';
 
-import type { ToolExecuteData } from '@/libs/agent-stream';
-
 import { ClientToolExecutionActionImpl } from '../clientToolExecution';
 
 // ─── Hoisted mocks ───

src/store/chat/slices/aiChat/actions/__tests__/gateway.test.ts

@@ -1,6 +1,6 @@
+import type { AgentStreamEvent } from '@lobechat/agent-gateway-client';
 import { afterEach, describe, expect, it, vi } from 'vitest';
 
-import type { AgentStreamEvent } from '@/libs/agent-stream';
 import { aiAgentService } from '@/services/aiAgent';
 
 import type { GatewayConnection } from '../gateway';

src/store/chat/slices/aiChat/actions/__tests__/gatewayEventHandler.test.ts

@@ -1,7 +1,6 @@
+import type { AgentStreamEvent } from '@lobechat/agent-gateway-client';
 import { describe, expect, it, vi } from 'vitest';
 
-import type { AgentStreamEvent } from '@/libs/agent-stream';
-
 import { createGatewayEventHandler } from '../gatewayEventHandler';
 
 vi.mock('@/services/message', () => ({

src/store/chat/slices/aiChat/actions/clientToolExecution.ts

@@ -1,8 +1,8 @@
+import { type ToolExecuteData, type ToolResultMessage } from '@lobechat/agent-gateway-client';
 import { type BuiltinToolContext } from '@lobechat/types';
 import debug from 'debug';
 import { produce } from 'immer';
 
-import { type ToolExecuteData, type ToolResultMessage } from '@/libs/agent-stream';
 import { mcpService } from '@/services/mcp';
 import { type ChatStore } from '@/store/chat/store';
 import { hasExecutor, invokeExecutor } from '@/store/tool/slices/builtin/executors';

src/store/chat/slices/aiChat/actions/gateway.ts

@@ -1,12 +1,12 @@
+import {
+  AgentStreamClient,
+  type AgentStreamClientOptions,
+  type AgentStreamEvent,
+  type ConnectionStatus,
+} from '@lobechat/agent-gateway-client';
 import type { ConversationContext, ExecAgentResult } from '@lobechat/types';
 
 import { isDesktop } from '@/const/version';
-import type {
-  AgentStreamClientOptions,
-  AgentStreamEvent,
-  ConnectionStatus,
-} from '@/libs/agent-stream';
-import { AgentStreamClient } from '@/libs/agent-stream/client';
 import { aiAgentService, type ResumeApprovalParam } from '@/services/aiAgent';
 import { messageService } from '@/services/message';
 import { topicService } from '@/services/topic';

src/store/chat/slices/aiChat/actions/gatewayEventHandler.ts

@@ -1,12 +1,12 @@
-import type { ConversationContext } from '@lobechat/types';
-
 import type {
   AgentStreamEvent,
   StepCompleteData,
   StreamChunkData,
   StreamStartData,
   ToolExecuteData,
-} from '@/libs/agent-stream';
+} from '@lobechat/agent-gateway-client';
+import type { ConversationContext } from '@lobechat/types';
+
 import { messageService } from '@/services/message';
 import type { ChatStore } from '@/store/chat/store';