@lobehub/chat 1.81.4 → 1.81.6

package/src/libs/agent-runtime/utils/sensenovaHelpers.ts

@@ -1,5 +1,4 @@
 export const convertSenseNovaMessage = (content: any) => {
-
   // 如果为单条 string 类 content，则格式转换为 text 类
   if (typeof content === 'string') {
     return [{ text: content, type: 'text' }];
@@ -16,8 +15,8 @@ export const convertSenseNovaMessage = (content: any) => {
         const url = item.image_url.url;
 
         // 如果 image_url 为 base64 格式，则返回 image_base64 类，否则返回 image_url 类
-        return url.startsWith('data:image/jpeg;base64') 
-          ? { 
+        return url.startsWith('data:image/jpeg;base64')
+          ? {
               image_base64: url.split(',')[1],
               type: 'image_base64',
             }

package/src/locales/default/hotkey.ts

@@ -5,6 +5,10 @@ const hotkey: HotkeyI18nTranslations = {
     desc: '将当前输入内容添加为用户消息，但不触发生成',
     title: '添加一条用户消息',
   },
+  clearCurrentMessages: {
+    desc: '清空当前会话的消息和上传的文件',
+    title: '清空会话消息',
+  },
   editMessage: {
     desc: '通过按住 Alt 并双击消息进入编辑模式',
     title: '编辑消息',

package/src/server/modules/MCPClient/__tests__/__snapshots__/index.test.ts.snap

@@ -0,0 +1,113 @@
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`MCPClient > Stdio Transport (using SDK Mock Server) > should list tools via stdio 1`] = `
+[
+  {
+    "description": "Echoes back a message with 'Hello' prefix",
+    "inputSchema": {
+      "$schema": "http://json-schema.org/draft-07/schema#",
+      "additionalProperties": false,
+      "properties": {
+        "message": {
+          "description": "The message to echo",
+          "type": "string",
+        },
+      },
+      "required": [
+        "message",
+      ],
+      "type": "object",
+    },
+    "name": "echo",
+  },
+  {
+    "description": "Lists all available tools and methods",
+    "inputSchema": {
+      "$schema": "http://json-schema.org/draft-07/schema#",
+      "additionalProperties": false,
+      "properties": {},
+      "type": "object",
+    },
+    "name": "debug",
+  },
+  {
+    "description": "Adds two numbers",
+    "inputSchema": {
+      "$schema": "http://json-schema.org/draft-07/schema#",
+      "additionalProperties": false,
+      "properties": {
+        "a": {
+          "description": "The first number",
+          "type": "number",
+        },
+        "b": {
+          "description": "The second number",
+          "type": "number",
+        },
+      },
+      "required": [
+        "a",
+        "b",
+      ],
+      "type": "object",
+    },
+    "name": "add",
+  },
+]
+`;
+
+exports[`MCPClient > Stdio Transport > should list tools via stdio 1`] = `
+[
+  {
+    "description": "Echoes back a message with 'Hello' prefix",
+    "inputSchema": {
+      "$schema": "http://json-schema.org/draft-07/schema#",
+      "additionalProperties": false,
+      "properties": {
+        "message": {
+          "description": "The message to echo",
+          "type": "string",
+        },
+      },
+      "required": [
+        "message",
+      ],
+      "type": "object",
+    },
+    "name": "echo",
+  },
+  {
+    "description": "Lists all available tools and methods",
+    "inputSchema": {
+      "$schema": "http://json-schema.org/draft-07/schema#",
+      "additionalProperties": false,
+      "properties": {},
+      "type": "object",
+    },
+    "name": "debug",
+  },
+  {
+    "description": "Adds two numbers",
+    "inputSchema": {
+      "$schema": "http://json-schema.org/draft-07/schema#",
+      "additionalProperties": false,
+      "properties": {
+        "a": {
+          "description": "The first number",
+          "type": "number",
+        },
+        "b": {
+          "description": "The second number",
+          "type": "number",
+        },
+      },
+      "required": [
+        "a",
+        "b",
+      ],
+      "type": "object",
+    },
+    "name": "add",
+  },
+]
+`;

package/src/server/modules/MCPClient/__tests__/index.test.ts

@@ -0,0 +1,81 @@
+import { afterEach, beforeEach, describe, expect, it } from 'vitest';
+
+import { MCPClient } from '../index';
+
+describe('MCPClient', () => {
+  // --- Updated Stdio Transport tests ---
+  describe('Stdio Transport', () => {
+    let mcpClient: MCPClient;
+    const stdioConnection = {
+      id: 'mcp-hello-world',
+      name: 'Stdio SDK Test Connection',
+      type: 'stdio' as const,
+      command: 'npx', // Use node to run the compiled mock server
+      args: ['mcp-hello-world@1.1.2'], // Use the path to the compiled JS file
+    };
+
+    beforeEach(async () => {
+      // args are now set directly in the connection object
+      mcpClient = new MCPClient(stdioConnection);
+      // Initialize the client - this starts the stdio process
+      await mcpClient.initialize();
+      // Add a small delay to allow the server process to fully start (optional, but can help)
+      await new Promise((resolve) => setTimeout(resolve, 100));
+    });
+
+    afterEach(async () => {
+      // Assume SDK client/transport handles process termination gracefully
+      // If processes leak, more explicit cleanup might be needed here
+    });
+
+    it('should create and initialize an instance with stdio transport', () => {
+      expect(mcpClient).toBeInstanceOf(MCPClient);
+    });
+
+    it('should list tools via stdio', async () => {
+      const result = await mcpClient.listTools();
+
+      // Check exact length if no other tools are expected
+      expect(result.tools).toHaveLength(3);
+
+      // Expect the tools defined in mock-sdk-server.ts
+      expect(result.tools).toMatchSnapshot();
+    });
+
+    it('should call the "echo" tool via stdio', async () => {
+      const toolName = 'echo';
+      const toolArgs = { message: 'hello stdio' };
+      // Expect the result format defined in mock-sdk-server.ts
+      const expectedResult = {
+        content: [{ type: 'text', text: 'You said: hello stdio' }],
+      };
+
+      const result = await mcpClient.callTool(toolName, toolArgs);
+      expect(result).toEqual(expectedResult);
+    });
+
+    it('should call the "add" tool via stdio', async () => {
+      const toolName = 'add';
+      const toolArgs = { a: 5, b: 7 };
+
+      const result = await mcpClient.callTool(toolName, toolArgs);
+      expect(result).toEqual({
+        content: [{ type: 'text', text: 'The sum is: 12' }],
+      });
+    });
+  });
+
+  // Error Handling tests remain the same...
+  describe('Error Handling', () => {
+    it('should throw error for unsupported connection type', () => {
+      const connection = {
+        id: 'invalid-test',
+        name: 'Invalid Test Connection',
+        type: 'invalid' as any,
+      };
+      expect(() => new MCPClient(connection as any)).toThrow(
+        'Unsupported MCP connection type: invalid',
+      );
+    });
+  });
+});

package/src/server/modules/MCPClient/index.ts

@@ -0,0 +1,80 @@
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js';
+import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
+import type { Transport } from '@modelcontextprotocol/sdk/shared/transport.d.ts';
+import debug from 'debug';
+
+const log = debug('lobe-mcp:client');
+
+interface MCPConnectionBase {
+  id: string;
+  name: string;
+  type: 'http' | 'stdio';
+}
+
+interface HttpMCPConnection extends MCPConnectionBase {
+  type: 'http';
+  url: string;
+}
+
+interface StdioMCPConnection extends MCPConnectionBase {
+  args: string[];
+  command: string;
+  type: 'stdio';
+}
+type MCPConnection = HttpMCPConnection | StdioMCPConnection;
+
+export class MCPClient {
+  private mcp: Client;
+  private transport: Transport;
+
+  constructor(connection: MCPConnection) {
+    log('Creating MCPClient with connection: %O', connection);
+    this.mcp = new Client({ name: 'lobehub-mcp-client', version: '1.0.0' });
+
+    switch (connection.type) {
+      case 'http': {
+        log('Using HTTP transport with url: %s', connection.url);
+        this.transport = new StreamableHTTPClientTransport(new URL(connection.url));
+        break;
+      }
+      case 'stdio': {
+        log(
+          'Using Stdio transport with command: %s and args: %O',
+          connection.command,
+          connection.args,
+        );
+        this.transport = new StdioClientTransport({
+          args: connection.args,
+          command: connection.command,
+        });
+        break;
+      }
+      default: {
+        const err = new Error(`Unsupported MCP connection type: ${(connection as any).type}`);
+        log('Error creating client: %O', err);
+        throw err;
+      }
+    }
+  }
+
+  async initialize() {
+    log('Initializing MCP connection...');
+    await this.mcp.connect(this.transport);
+    log('MCP connection initialized.');
+  }
+
+  async listTools() {
+    log('Listing tools...');
+    const tools = await this.mcp.listTools();
+    log('Listed tools: %O', tools);
+    return tools;
+  }
+
+  async callTool(toolName: string, args: any) {
+    log('Calling tool: %s with args: %O', toolName, args);
+    const result = await this.mcp.callTool({ arguments: args, name: toolName });
+    log('Tool call result: %O', result);
+    return result;
+  }
+}

package/src/types/hotkey.ts

@@ -58,6 +58,7 @@ export const KeyEnum = {
 
 export const HotkeyEnum = {
   AddUserMessage: 'addUserMessage',
+  ClearCurrentMessages: 'clearCurrentMessages',
   EditMessage: 'editMessage',
   OpenChatSettings: 'openChatSettings',
   OpenHotkeyHelper: 'openHotkeyHelper',