@opentiny/next-sdk 0.1.3 → 0.1.4

package/McpSdk.ts

@@ -0,0 +1,14 @@
+import Ajv from 'ajv'
+export { Ajv }
+export * from '@modelcontextprotocol/sdk/types.js'
+export * from '@modelcontextprotocol/sdk/shared/uriTemplate.js'
+export * from '@modelcontextprotocol/sdk/server/completable.js'
+export * from '@modelcontextprotocol/sdk/shared/metadataUtils.js'
+export * from '@modelcontextprotocol/sdk/client/index.js'
+export * from '@modelcontextprotocol/sdk/client/auth.js'
+export * from '@modelcontextprotocol/sdk/client/sse.js'
+export * from '@modelcontextprotocol/sdk/client/streamableHttp.js'
+// @ts-ignore
+export * from '@modelcontextprotocol/sdk/server/mcp.js'
+export * from '@modelcontextprotocol/sdk/shared/transport.js'
+export * from '@modelcontextprotocol/sdk/shared/protocol.js'

package/WebAgent.ts

@@ -0,0 +1,5 @@
+export { AgentModelProvider } from './agent/AgentModelProvider'
+export { createRemoter } from './remoter/createRemoter'
+export { getAISDKTools } from './agent/utils/getAISDKTools'
+export { QrCode, type QrCodeOption } from './remoter/QrCode'
+export type * from './agent/type'

package/WebMcp.ts

@@ -0,0 +1,18 @@
+import Ajv from 'ajv'
+export * from './WebMcpServer'
+export * from './WebMcpClient'
+export { Ajv }
+export { z } from 'zod'
+export { AuthClientProvider } from '@opentiny/next'
+export { ResourceTemplate } from '@modelcontextprotocol/sdk/server/mcp.js'
+export { UriTemplate } from '@modelcontextprotocol/sdk/shared/uriTemplate.js'
+export { completable } from '@modelcontextprotocol/sdk/server/completable.js'
+export { getDisplayName } from '@modelcontextprotocol/sdk/shared/metadataUtils.js'
+export type * from 'zod'
+export type * from '@opentiny/next'
+export type * from '@modelcontextprotocol/sdk/types.js'
+export type * from '@modelcontextprotocol/sdk/shared/protocol.js'
+export type * from '@modelcontextprotocol/sdk/shared/transport.js'
+export type * from '@modelcontextprotocol/sdk/client/sse.js'
+export type * from '@modelcontextprotocol/sdk/client/streamableHttp.js'
+export type * from '@modelcontextprotocol/sdk/server/mcp.js'

package/Zod.ts

@@ -0,0 +1 @@
+export { z } from 'zod'

package/agent/AgentModelProvider.ts

@@ -1,16 +1,41 @@
 import { streamText, stepCountIs, generateText } from 'ai'
+import { experimental_createMCPClient as createMCPClient, experimental_MCPClientConfig as MCPClientConfig } from 'ai'
 import type { ToolSet } from 'ai'
-import { getMcpClients, getMcpTools } from './utils'
+import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js'
 import type { IAgentModelProviderOption, McpServerConfig } from './type'
-import { AIProviderFactories } from './utils/aiProviderFactories'
 import { ProviderV2 } from '@ai-sdk/provider'
 import { OpenAIProvider } from '@ai-sdk/openai'
+import { createOpenAI } from '@ai-sdk/openai'
+import { createDeepSeek } from '@ai-sdk/deepseek'
 
+export const AIProviderFactories = {
+  ['openai']: createOpenAI,
+  ['deepseek']: createDeepSeek
+}
+
+type ChatMethodFn = typeof streamText | typeof generateText
+
+/** 一个通用的ai-sdk的agent封装
+ * @summary 内部自动管理了 llm, mcpServer, ai-sdk的clients 和 tools
+ * @returns 暴露了 chat, chatStream方法
+ */
 export class AgentModelProvider {
   llm: ProviderV2 | OpenAIProvider
-  mcpServers: McpServerConfig[]
-  isGetMcpClients = false
+  /**  mcpServers 允许为配置为 McpServerConfig, 或者任意的 MCPTransport
+   * 参考: https://ai-sdk.dev/docs/ai-sdk-core/tools-and-tool-calling#initializing-an-mcp-client */
+  mcpServers: McpServerConfig[] = []
+  /**  ai-sdk的 mcpClient */
   mcpClients: any[] = []
+  /** 所有的tools */
+  mcpTools: Array<Record<string, any>> = []
+  /**  需要实时过滤掉的tools name*/
+  ignoreToolnames: string[] = []
+
+  /** 在 chat 之前，自动更新 所有的tools */
+  autoUpdateTools = true
+
+  /** chat 时，自动更新 所有的tools 后的事件 */
+  onUpdatedTools: (() => void) | undefined
 
   constructor({ llmConfig, mcpServers, llm }: IAgentModelProviderOption) {
     // 1、保存 mcpServer
@@ -36,53 +61,143 @@ export class AgentModelProvider {
     }
   }
 
-  async initClients() {
-    if (!this.isGetMcpClients) {
-      this.mcpClients = await getMcpClients(this.mcpServers)
-      this.isGetMcpClients = true
+  /** 创建一个 ai-sdk的 mcpClient, 创建失败则返回 Null */
+  private async _createOneClient(serverConfig: McpServerConfig) {
+    try {
+      let transport: MCPClientConfig['transport']
+      // transport 一定是 streamableHttp 或者就是： ai-sdk允许的 transport
+      if ('type' in serverConfig && serverConfig.type.toLocaleLowerCase() === 'streamablehttp') {
+        transport = new StreamableHTTPClientTransport(new URL(serverConfig.url))
+      } else {
+        transport = serverConfig as MCPClientConfig['transport']
+      }
+
+      return createMCPClient({ transport: transport as MCPClientConfig['transport'] })
+    } catch (error) {
+      console.error(`Failed to create MCP client`, serverConfig, error)
+      return null
     }
   }
-  async chat({
-    model,
-    maxSteps = 5,
-    ...options
-  }: Parameters<typeof generateText>[0] & { maxSteps?: number }): Promise<any> {
-    if (!this.llm) {
-      throw new Error('LLM is not initialized')
+  /** 创建 ai-sdk的 mcpClient, 失败则保存为null */
+  private async _createMpcClients() {
+    // 使用 Promise.all 并行处理所有 mcpServer 项
+    this.mcpClients = await Promise.all(
+      this.mcpServers.map(async (server) => {
+        return this._createOneClient(server)
+      })
+    )
+  }
+  /** 创建所有 mcpClients 的 tools, 失败则保存为null */
+  private async _createMpcTools() {
+    this.mcpTools = await Promise.all(
+      this.mcpClients.map(async (client) => {
+        try {
+          return client ? await client?.tools?.() : null
+        } catch (error) {
+          return null
+        }
+      })
+    )
+  }
+  /** 关闭所有的 clients */
+  async closeAll() {
+    await Promise.all(
+      this.mcpClients.map(async (client) => {
+        try {
+          client.close()
+        } catch (error) {}
+      })
+    )
+  }
+
+  async initClientsAndTools() {
+    await this._createMpcClients()
+    await this._createMpcTools()
+  }
+
+  async updateMcpServers(mcpServers: McpServerConfig[]) {
+    await this.closeAll()
+    this.mcpServers = mcpServers
+    await this.initClientsAndTools()
+  }
+
+  async insertMcpServer(mcpServer: McpServerConfig) {
+    const find = this.mcpServers.find((item: any) => item.url === mcpServer.url)
+
+    if (!find) {
+      this.mcpServers = [...this.mcpServers, mcpServer]
+      const client = await this._createOneClient(mcpServer)
+      this.mcpClients.push(client)
+      this.mcpTools.push((await client?.tools?.()) as Record<string, any>)
+      return true
     }
+    return false
+  }
+  /** 通过引用，删除一个 mcpServers mcpClients  mcpTools ignoreToolnames  */
+  removeMcpServer(mcpServer: McpServerConfig) {
+    const index = this.mcpServers.findIndex((server) => server === mcpServer)
 
-    // 每次会话需要获取最新的工具列表，因为工具是会发生变化的
-    await this.initClients()
-    const tools = await getMcpTools(this.mcpClients, options)
+    // 移除
+    this.mcpServers.splice(index, 1)
 
-    return generateText({
-      // @ts-ignore  ProviderV2 是所有llm的父类， 在每一个具体的llm 类都有一个选择model的函数用法
-      model: this.llm(model),
-      tools: tools as ToolSet,
-      stopWhen: stepCountIs(maxSteps),
-      ...options
+    // 移除
+    const delClient = this.mcpClients[index]
+    this.mcpClients.splice(index, 1)
+    try {
+      delClient?.close()
+    } catch (error) {}
+
+    // 移除 tools
+    const delTool = this.mcpTools[index]
+    this.mcpTools.splice(index, 1)
+
+    // 移除 ignoreToolnames
+    if (delTool) {
+      Object.keys(delTool).forEach((toolName) => {
+        this.ignoreToolnames = this.ignoreToolnames.filter((name) => name !== toolName)
+      })
+    }
+  }
+
+  /** 创建临时允许调用的tools集合 */
+  tempMergeTools(extraTool = {}) {
+    const toolsResult = this.mcpTools.reduce((acc, curr) => ({ ...acc, ...curr }), {})
+    Object.assign(toolsResult, extraTool)
+
+    this.ignoreToolnames.forEach((name) => {
+      delete toolsResult[name]
     })
+
+    return toolsResult
   }
 
-  async chatStream({
-    model,
-    maxSteps = 5,
-    ...options
-  }: Parameters<typeof streamText>[0] & { maxSteps?: number }): Promise<any> {
+  async _chat(
+    chatMethod: ChatMethodFn,
+    { model, maxSteps = 5, ...options }: Parameters<typeof generateText>[0] & { maxSteps?: number }
+  ): Promise<any> {
     if (!this.llm) {
       throw new Error('LLM is not initialized')
     }
 
-    // 每次会话需要获取最新的工具列表，因为工具是会发生变化的
-    await this.initClients()
-    const tools = await getMcpTools(this.mcpClients, options)
+    if (this.autoUpdateTools) {
+      await this._createMpcTools()
+      this.onUpdatedTools?.()
+    }
 
-    return streamText({
-      // @ts-ignore 同上
+    return chatMethod({
+      // @ts-ignore  ProviderV2 是所有llm的父类， 在每一个具体的llm 类都有一个选择model的函数用法
       model: this.llm(model),
-      tools: tools as ToolSet,
+      tools: this.tempMergeTools(options.tools) as ToolSet,
       stopWhen: stepCountIs(maxSteps),
       ...options
     })
   }
+
+  async chat(options: Parameters<typeof generateText>[0] & { maxSteps?: number }): Promise<any> {
+    return this._chat(generateText, options)
+  }
+
+  async chatStream(options: Parameters<typeof streamText>[0] & { maxSteps?: number }): Promise<any> {
+    return this._chat(streamText, options as any)
+  }
 }

package/agent/type.ts

@@ -14,7 +14,7 @@ export interface IAgentModelProviderLlmConfig {
 }
 
 /** Mcp Server的配置对象 */
-export type McpServerConfig = { type: 'streamableHttp' | 'sse'; url: string } | { transport: MCPTransport }
+export type McpServerConfig = { type: 'streamableHttp' | 'sse'; url: string } & { transport: MCPTransport }
 
 /** */
 export interface IAgentModelProviderOption {

package/agent/utils/getAISDKTools.ts

@@ -0,0 +1,35 @@
+import { dynamicTool, jsonSchema, Tool, ToolCallOptions, ToolSet } from 'ai'
+import { WebMcpClient } from '../../WebMcpClient'
+
+/**
+ * 快速 从官方 mcp 或 WebMcpClient 这2种client中， 抽取成 ai-sdk 所需要的 tool的对象
+ * @params client  一个已连接好的 WebMcpClient
+ * @returns  ai-sdk的 tool 格式的对象。
+ */
+export const getAISDKTools = async (client: WebMcpClient): Promise<ToolSet> => {
+  const tools: Record<string, Tool> = {}
+
+  try {
+    const listToolsResult = await client.listTools()
+
+    for (const { name, description, inputSchema } of listToolsResult.tools) {
+      const execute = async (args: any, options: ToolCallOptions): Promise<any> => {
+        return client.callTool({ name, arguments: args }, { signal: options?.abortSignal })
+      }
+
+      tools[name] = dynamicTool({
+        description,
+        inputSchema: jsonSchema({
+          ...inputSchema,
+          properties: (inputSchema.properties as Record<string, any>) ?? {},
+          additionalProperties: false
+        }),
+        execute
+      })
+    }
+
+    return tools
+  } catch (error) {
+    throw error
+  }
+}

package/dist/agent/AgentModelProvider.d.ts

@@ -2,17 +2,56 @@ import { streamText, generateText } from 'ai';
 import type { IAgentModelProviderOption, McpServerConfig } from './type';
 import { ProviderV2 } from '@ai-sdk/provider';
 import { OpenAIProvider } from '@ai-sdk/openai';
+import { createOpenAI } from '@ai-sdk/openai';
+import { createDeepSeek } from '@ai-sdk/deepseek';
+export declare const AIProviderFactories: {
+    openai: typeof createOpenAI;
+    deepseek: typeof createDeepSeek;
+};
+type ChatMethodFn = typeof streamText | typeof generateText;
+/** 一个通用的ai-sdk的agent封装
+ * @summary 内部自动管理了 llm, mcpServer, ai-sdk的clients 和 tools
+ * @returns 暴露了 chat, chatStream方法
+ */
 export declare class AgentModelProvider {
     llm: ProviderV2 | OpenAIProvider;
+    /**  mcpServers 允许为配置为 McpServerConfig, 或者任意的 MCPTransport
+     * 参考: https://ai-sdk.dev/docs/ai-sdk-core/tools-and-tool-calling#initializing-an-mcp-client */
     mcpServers: McpServerConfig[];
-    isGetMcpClients: boolean;
+    /**  ai-sdk的 mcpClient */
     mcpClients: any[];
+    /** 所有的tools */
+    mcpTools: Array<Record<string, any>>;
+    /**  需要实时过滤掉的tools name*/
+    ignoreToolnames: string[];
+    /** 在 chat 之前，自动更新 所有的tools */
+    autoUpdateTools: boolean;
+    /** chat 时，自动更新 所有的tools 后的事件 */
+    onUpdatedTools: (() => void) | undefined;
     constructor({ llmConfig, mcpServers, llm }: IAgentModelProviderOption);
-    initClients(): Promise<void>;
-    chat({ model, maxSteps, ...options }: Parameters<typeof generateText>[0] & {
+    /** 创建一个 ai-sdk的 mcpClient, 创建失败则返回 Null */
+    private _createOneClient;
+    /** 创建 ai-sdk的 mcpClient, 失败则保存为null */
+    private _createMpcClients;
+    /** 创建所有 mcpClients 的 tools, 失败则保存为null */
+    private _createMpcTools;
+    /** 关闭所有的 clients */
+    closeAll(): Promise<void>;
+    initClientsAndTools(): Promise<void>;
+    updateMcpServers(mcpServers: McpServerConfig[]): Promise<void>;
+    insertMcpServer(mcpServer: McpServerConfig): Promise<boolean>;
+    /** 通过引用，删除一个 mcpServers mcpClients  mcpTools ignoreToolnames  */
+    removeMcpServer(mcpServer: McpServerConfig): void;
+    /** 创建临时允许调用的tools集合 */
+    tempMergeTools(extraTool?: {}): Record<string, any>;
+    _chat(chatMethod: ChatMethodFn, { model, maxSteps, ...options }: Parameters<typeof generateText>[0] & {
         maxSteps?: number;
     }): Promise<any>;
-    chatStream({ model, maxSteps, ...options }: Parameters<typeof streamText>[0] & {
+    chat(options: Parameters<typeof generateText>[0] & {
+        maxSteps?: number;
+    }): Promise<any>;
+    chatStream(options: Parameters<typeof streamText>[0] & {
         maxSteps?: number;
     }): Promise<any>;
 }
+export {};

package/dist/agent/AgentModelProvider.js

@@ -19,12 +19,31 @@ var __rest = (this && this.__rest) || function (s, e) {
     return t;
 };
 import { streamText, stepCountIs, generateText } from 'ai';
-import { getMcpClients, getMcpTools } from './utils';
-import { AIProviderFactories } from './utils/aiProviderFactories';
+import { experimental_createMCPClient as createMCPClient } from 'ai';
+import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
+import { createOpenAI } from '@ai-sdk/openai';
+import { createDeepSeek } from '@ai-sdk/deepseek';
+export const AIProviderFactories = {
+    ['openai']: createOpenAI,
+    ['deepseek']: createDeepSeek
+};
+/** 一个通用的ai-sdk的agent封装
+ * @summary 内部自动管理了 llm, mcpServer, ai-sdk的clients 和 tools
+ * @returns 暴露了 chat, chatStream方法
+ */
 export class AgentModelProvider {
     constructor({ llmConfig, mcpServers, llm }) {
-        this.isGetMcpClients = false;
+        /**  mcpServers 允许为配置为 McpServerConfig, 或者任意的 MCPTransport
+         * 参考: https://ai-sdk.dev/docs/ai-sdk-core/tools-and-tool-calling#initializing-an-mcp-client */
+        this.mcpServers = [];
+        /**  ai-sdk的 mcpClient */
         this.mcpClients = [];
+        /** 所有的tools */
+        this.mcpTools = [];
+        /**  需要实时过滤掉的tools name*/
+        this.ignoreToolnames = [];
+        /** 在 chat 之前，自动更新 所有的tools */
+        this.autoUpdateTools = true;
         // 1、保存 mcpServer
         this.mcpServers = mcpServers || [];
         // 2、保存 llm
@@ -48,40 +67,142 @@ export class AgentModelProvider {
             throw new Error('Either llmConfig or llm must be provided');
         }
     }
-    initClients() {
+    /** 创建一个 ai-sdk的 mcpClient, 创建失败则返回 Null */
+    _createOneClient(serverConfig) {
         return __awaiter(this, void 0, void 0, function* () {
-            if (!this.isGetMcpClients) {
-                this.mcpClients = yield getMcpClients(this.mcpServers);
-                this.isGetMcpClients = true;
+            try {
+                let transport;
+                // transport 一定是 streamableHttp 或者就是： ai-sdk允许的 transport
+                if ('type' in serverConfig && serverConfig.type.toLocaleLowerCase() === 'streamablehttp') {
+                    transport = new StreamableHTTPClientTransport(new URL(serverConfig.url));
+                }
+                else {
+                    transport = serverConfig;
+                }
+                return createMCPClient({ transport: transport });
+            }
+            catch (error) {
+                console.error(`Failed to create MCP client`, serverConfig, error);
+                return null;
             }
         });
     }
-    chat(_a) {
+    /** 创建 ai-sdk的 mcpClient, 失败则保存为null */
+    _createMpcClients() {
         return __awaiter(this, void 0, void 0, function* () {
-            var { model, maxSteps = 5 } = _a, options = __rest(_a, ["model", "maxSteps"]);
-            if (!this.llm) {
-                throw new Error('LLM is not initialized');
+            // 使用 Promise.all 并行处理所有 mcpServer 项
+            this.mcpClients = yield Promise.all(this.mcpServers.map((server) => __awaiter(this, void 0, void 0, function* () {
+                return this._createOneClient(server);
+            })));
+        });
+    }
+    /** 创建所有 mcpClients 的 tools, 失败则保存为null */
+    _createMpcTools() {
+        return __awaiter(this, void 0, void 0, function* () {
+            this.mcpTools = yield Promise.all(this.mcpClients.map((client) => __awaiter(this, void 0, void 0, function* () {
+                var _a;
+                try {
+                    return client ? yield ((_a = client === null || client === void 0 ? void 0 : client.tools) === null || _a === void 0 ? void 0 : _a.call(client)) : null;
+                }
+                catch (error) {
+                    return null;
+                }
+            })));
+        });
+    }
+    /** 关闭所有的 clients */
+    closeAll() {
+        return __awaiter(this, void 0, void 0, function* () {
+            yield Promise.all(this.mcpClients.map((client) => __awaiter(this, void 0, void 0, function* () {
+                try {
+                    client.close();
+                }
+                catch (error) { }
+            })));
+        });
+    }
+    initClientsAndTools() {
+        return __awaiter(this, void 0, void 0, function* () {
+            yield this._createMpcClients();
+            yield this._createMpcTools();
+        });
+    }
+    updateMcpServers(mcpServers) {
+        return __awaiter(this, void 0, void 0, function* () {
+            yield this.closeAll();
+            this.mcpServers = mcpServers;
+            yield this.initClientsAndTools();
+        });
+    }
+    insertMcpServer(mcpServer) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a;
+            const find = this.mcpServers.find((item) => item.url === mcpServer.url);
+            if (!find) {
+                this.mcpServers = [...this.mcpServers, mcpServer];
+                const client = yield this._createOneClient(mcpServer);
+                this.mcpClients.push(client);
+                this.mcpTools.push((yield ((_a = client === null || client === void 0 ? void 0 : client.tools) === null || _a === void 0 ? void 0 : _a.call(client))));
+                return true;
             }
-            // 每次会话需要获取最新的工具列表，因为工具是会发生变化的
-            yield this.initClients();
-            const tools = yield getMcpTools(this.mcpClients, options);
-            return generateText(Object.assign({ 
-                // @ts-ignore  ProviderV2 是所有llm的父类， 在每一个具体的llm 类都有一个选择model的函数用法
-                model: this.llm(model), tools: tools, stopWhen: stepCountIs(maxSteps) }, options));
+            return false;
+        });
+    }
+    /** 通过引用，删除一个 mcpServers mcpClients  mcpTools ignoreToolnames  */
+    removeMcpServer(mcpServer) {
+        const index = this.mcpServers.findIndex((server) => server === mcpServer);
+        // 移除
+        this.mcpServers.splice(index, 1);
+        // 移除
+        const delClient = this.mcpClients[index];
+        this.mcpClients.splice(index, 1);
+        try {
+            delClient === null || delClient === void 0 ? void 0 : delClient.close();
+        }
+        catch (error) { }
+        // 移除 tools
+        const delTool = this.mcpTools[index];
+        this.mcpTools.splice(index, 1);
+        // 移除 ignoreToolnames
+        if (delTool) {
+            Object.keys(delTool).forEach((toolName) => {
+                this.ignoreToolnames = this.ignoreToolnames.filter((name) => name !== toolName);
+            });
+        }
+    }
+    /** 创建临时允许调用的tools集合 */
+    tempMergeTools(extraTool = {}) {
+        const toolsResult = this.mcpTools.reduce((acc, curr) => (Object.assign(Object.assign({}, acc), curr)), {});
+        Object.assign(toolsResult, extraTool);
+        this.ignoreToolnames.forEach((name) => {
+            delete toolsResult[name];
         });
+        return toolsResult;
     }
-    chatStream(_a) {
+    _chat(chatMethod, _a) {
         return __awaiter(this, void 0, void 0, function* () {
+            var _b;
             var { model, maxSteps = 5 } = _a, options = __rest(_a, ["model", "maxSteps"]);
             if (!this.llm) {
                 throw new Error('LLM is not initialized');
             }
-            // 每次会话需要获取最新的工具列表，因为工具是会发生变化的
-            yield this.initClients();
-            const tools = yield getMcpTools(this.mcpClients, options);
-            return streamText(Object.assign({ 
-                // @ts-ignore 同上
-                model: this.llm(model), tools: tools, stopWhen: stepCountIs(maxSteps) }, options));
+            if (this.autoUpdateTools) {
+                yield this._createMpcTools();
+                (_b = this.onUpdatedTools) === null || _b === void 0 ? void 0 : _b.call(this);
+            }
+            return chatMethod(Object.assign({ 
+                // @ts-ignore  ProviderV2 是所有llm的父类， 在每一个具体的llm 类都有一个选择model的函数用法
+                model: this.llm(model), tools: this.tempMergeTools(options.tools), stopWhen: stepCountIs(maxSteps) }, options));
+        });
+    }
+    chat(options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return this._chat(generateText, options);
+        });
+    }
+    chatStream(options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            return this._chat(streamText, options);
         });
     }
 }

package/dist/agent/type.d.ts

@@ -15,7 +15,7 @@ export interface IAgentModelProviderLlmConfig {
 export type McpServerConfig = {
     type: 'streamableHttp' | 'sse';
     url: string;
-} | {
+} & {
     transport: MCPTransport;
 };
 /** */

package/dist/agent/utils/getAISDKTools.d.ts

@@ -0,0 +1,8 @@
+import { ToolSet } from 'ai';
+import { WebMcpClient } from '../../WebMcpClient';
+/**
+ * 快速 从官方 mcp 或 WebMcpClient 这2种client中， 抽取成 ai-sdk 所需要的 tool的对象
+ * @params client  一个已连接好的 WebMcpClient
+ * @returns  ai-sdk的 tool 格式的对象。
+ */
+export declare const getAISDKTools: (client: WebMcpClient) => Promise<ToolSet>;

package/dist/agent/utils/getAISDKTools.js

@@ -0,0 +1,36 @@
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+import { dynamicTool, jsonSchema } from 'ai';
+/**
+ * 快速 从官方 mcp 或 WebMcpClient 这2种client中， 抽取成 ai-sdk 所需要的 tool的对象
+ * @params client  一个已连接好的 WebMcpClient
+ * @returns  ai-sdk的 tool 格式的对象。
+ */
+export const getAISDKTools = (client) => __awaiter(void 0, void 0, void 0, function* () {
+    var _a;
+    const tools = {};
+    try {
+        const listToolsResult = yield client.listTools();
+        for (const { name, description, inputSchema } of listToolsResult.tools) {
+            const execute = (args, options) => __awaiter(void 0, void 0, void 0, function* () {
+                return client.callTool({ name, arguments: args }, { signal: options === null || options === void 0 ? void 0 : options.abortSignal });
+            });
+            tools[name] = dynamicTool({
+                description,
+                inputSchema: jsonSchema(Object.assign(Object.assign({}, inputSchema), { properties: (_a = inputSchema.properties) !== null && _a !== void 0 ? _a : {}, additionalProperties: false })),
+                execute
+            });
+        }
+        return tools;
+    }
+    catch (error) {
+        throw error;
+    }
+});

package/dist/index.d.ts

@@ -1,6 +1,4 @@
 import Ajv from 'ajv';
-export * from './WebMcpServer';
-export * from './WebMcpClient';
 export { Ajv };
 export { z } from 'zod';
 export { AuthClientProvider } from '@opentiny/next';
@@ -16,7 +14,10 @@ export type * from '@modelcontextprotocol/sdk/shared/transport.js';
 export type * from '@modelcontextprotocol/sdk/client/sse.js';
 export type * from '@modelcontextprotocol/sdk/client/streamableHttp.js';
 export type * from '@modelcontextprotocol/sdk/server/mcp.js';
-export { AgentModelProvider } from './agent/AgentModelProvider';
+export * from './WebMcpServer';
+export * from './WebMcpClient';
 export { createRemoter } from './remoter/createRemoter';
-export { getAISDKTools } from './agent/utils';
+export { AgentModelProvider } from './agent/AgentModelProvider';
+export { getAISDKTools } from './agent/utils/getAISDKTools';
 export { QrCode, type QrCodeOption } from './remoter/QrCode';
+export type * from './agent/type';