@lobehub/chat 1.84.4 → 1.84.6

package/src/services/mcp.ts

@@ -1,6 +1,7 @@
 import { isDesktop } from '@/const/version';
 import { desktopClient, toolsClient } from '@/libs/trpc/client';
 import { ChatToolPayload } from '@/types/message';
+import { CustomPluginMetadata } from '@/types/tool/plugin';
 
 class MCPService {
   async invokeMcpToolCall(payload: ChatToolPayload, { signal }: { signal?: AbortSignal }) {
@@ -30,14 +31,24 @@ class MCPService {
     return toolsClient.mcp.callTool.mutate(data, { signal });
   }
 
-  async getStreamableMcpServerManifest(identifier: string, url: string) {
-    return toolsClient.mcp.getStreamableMcpServerManifest.query({ identifier, url });
+  async getStreamableMcpServerManifest(
+    identifier: string,
+    url: string,
+    metadata?: CustomPluginMetadata,
+  ) {
+    return toolsClient.mcp.getStreamableMcpServerManifest.query({ identifier, metadata, url });
   }
 
-  async getStdioMcpServerManifest(identifier: string, command: string, args?: string[]) {
+  async getStdioMcpServerManifest(
+    identifier: string,
+    command: string,
+    args?: string[],
+    metadata?: CustomPluginMetadata,
+  ) {
     return desktopClient.mcp.getStdioMcpServerManifest.query({
       args: args,
       command,
+      metadata,
       name: identifier,
     });
   }

package/src/types/tool/plugin.ts

@@ -4,13 +4,21 @@ import { LobeToolType } from './tool';
 
 export type PluginManifestMap = Record<string, LobeChatPluginManifest>;
 
+export interface CustomPluginMetadata {
+  avatar?: string;
+  description?: string;
+}
+
 export interface CustomPluginParams {
   apiMode?: 'openapi' | 'simple';
   enableSettings?: boolean;
   manifestMode?: 'local' | 'url';
   manifestUrl?: string;
+  useProxy?: boolean;
+
+  /* eslint-disable sort-keys-fix/sort-keys-fix , typescript-sort-keys/interface */
   /**
-   * 临时方案，后续需要做一次大重构
+   * TODO: 临时方案，后续需要做一次大重构
    */
   mcp?: {
     args?: string[];
@@ -18,7 +26,9 @@ export interface CustomPluginParams {
     type: 'http' | 'stdio';
     url?: string;
   };
-  useProxy?: boolean;
+  avatar?: string;
+  description?: string;
+  /* eslint-enable */
 }
 
 export interface LobeToolCustomPlugin {