@opentiny/next-sdk 0.1.14 → 0.1.15-beta.2

package/dist/agent/AgentModelProvider.js

@@ -1,293 +0,0 @@
-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());
-    });
-};
-var __rest = (this && this.__rest) || function (s, e) {
-    var t = {};
-    for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
-        t[p] = s[p];
-    if (s != null && typeof Object.getOwnPropertySymbols === "function")
-        for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
-            if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
-                t[p[i]] = s[p[i]];
-        }
-    return t;
-};
-import { streamText, stepCountIs, generateText } from 'ai';
-import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
-import { InMemoryTransport } from '@modelcontextprotocol/sdk/inMemory.js';
-import { createOpenAI } from '@ai-sdk/openai';
-import { createDeepSeek } from '@ai-sdk/deepseek';
-import { ExtensionClientTransport } from '../transport/ExtensionClientTransport';
-import { MessageChannelTransport } from '@opentiny/next';
-import { WebMcpClient } from '../WebMcpClient';
-import { getAISDKTools } from './utils/getAISDKTools';
-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 }) {
-        /**  当前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 = {};
-        /** 当前 mcpClients 所对应的tools，键为服务器名称 */
-        this.mcpTools = {};
-        /** 需要实时过滤掉的tools name*/
-        this.ignoreToolnames = [];
-        /** 缓存 ai-sdk response 中的 多轮会话的上下文 */
-        this.messages = [];
-        if (!llmConfig) {
-            throw new Error('llmConfig is required to initialize AgentModelProvider');
-        }
-        this.mcpServers = mcpServers || {};
-        this.mcpClients = {};
-        this.mcpTools = {};
-        if (llmConfig.llm) {
-            this.llm = llmConfig.llm;
-        }
-        else if (llmConfig.providerType) {
-            const providerType = llmConfig.providerType;
-            let providerFn;
-            if (typeof providerType === 'string') {
-                providerFn = AIProviderFactories[providerType];
-            }
-            else {
-                providerFn = providerType;
-            }
-            this.llm = providerFn({
-                apiKey: llmConfig.apiKey,
-                baseURL: llmConfig.baseURL
-            });
-        }
-        else {
-            throw new Error('Either llmConfig.llm or llmConfig.providerType must be provided');
-        }
-    }
-    /** 创建一个 ai-sdk的 mcpClient, 创建失败则返回 null */
-    _createOneClient(serverConfig) {
-        return __awaiter(this, void 0, void 0, function* () {
-            try {
-                let transport;
-                // transport 一定是 streamableHttp 或者就是： ai-sdk允许的 transport
-                if ('type' in serverConfig && serverConfig.type.toLocaleLowerCase() === 'streamablehttp') {
-                    transport = new StreamableHTTPClientTransport(new URL(serverConfig.url));
-                }
-                else if ('type' in serverConfig && serverConfig.type === 'extension') {
-                    transport = new ExtensionClientTransport(serverConfig.sessionId);
-                }
-                else {
-                    transport = serverConfig;
-                }
-                const client = new WebMcpClient({ name: 'mcp-web-client', version: '1.0.0' }, { capabilities: { roots: { listChanged: true }, sampling: {}, elicitation: {} } });
-                yield client.connect(transport);
-                //@ts-ignore
-                client['__transport__'] = transport;
-                return client;
-            }
-            catch (error) {
-                if (this.onError) {
-                    this.onError((error === null || error === void 0 ? void 0 : error.message) || `Failed to create MCP client`, error);
-                }
-                console.error(`Failed to create MCP client`, serverConfig, error);
-                return null;
-            }
-        });
-    }
-    /** 关闭一个 mcpClient */
-    _closeOneClient(client) {
-        return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b, _c;
-            try {
-                const transport = client['__transport__'];
-                // 如果是 InMemoryTransport，不关闭传输层 因为它是配对的，关闭一端会影响另一端（服务端）
-                if ((transport && transport instanceof InMemoryTransport) ||
-                    (transport && transport instanceof MessageChannelTransport)) {
-                    return;
-                }
-                // 其他类型的传输正常关闭
-                yield ((_a = transport === null || transport === void 0 ? void 0 : transport.terminateSession) === null || _a === void 0 ? void 0 : _a.call(transport));
-                yield ((_b = transport === null || transport === void 0 ? void 0 : transport.close) === null || _b === void 0 ? void 0 : _b.call(transport));
-                yield ((_c = client === null || client === void 0 ? void 0 : client.close) === null || _c === void 0 ? void 0 : _c.call(client));
-            }
-            catch (error) { }
-        });
-    }
-    /** 创建所有 mcpClients */
-    _createMpcClients() {
-        return __awaiter(this, void 0, void 0, function* () {
-            // 使用 Promise.all 并行处理所有 mcpServer 项
-            const serverEntries = Object.entries(this.mcpServers);
-            const clients = yield Promise.all(serverEntries.map((_a) => __awaiter(this, [_a], void 0, function* ([serverName, server]) {
-                const client = yield this._createOneClient(server);
-                return { serverName, client };
-            })));
-            // 将结果存储到对象中，使用 serverName 作为键
-            this.mcpClients = {};
-            clients.forEach(({ serverName, client }) => {
-                this.mcpClients[serverName] = client;
-            });
-        });
-    }
-    /** 查询所有 mcpClients 的 tools, 失败则保存为null */
-    _createMpcTools() {
-        return __awaiter(this, void 0, void 0, function* () {
-            const clientEntries = Object.entries(this.mcpClients);
-            const tools = yield Promise.all(clientEntries.map((_a) => __awaiter(this, [_a], void 0, function* ([serverName, client]) {
-                try {
-                    const result = client ? yield getAISDKTools(client) : null;
-                    return { serverName, tools: result };
-                }
-                catch (error) {
-                    if (this.onError) {
-                        this.onError((error === null || error === void 0 ? void 0 : error.message) || `Failed to query tools`, error);
-                    }
-                    console.error(`Failed to query tools`, error);
-                    return { serverName, tools: null };
-                }
-            })));
-            // 将结果存储到对象中，使用 serverName 作为键
-            this.mcpTools = {};
-            tools.forEach(({ serverName, tools: toolsData }) => {
-                const normalizedTools = toolsData && typeof toolsData === 'object' ? toolsData : {};
-                this.mcpTools[serverName] = normalizedTools;
-            });
-        });
-    }
-    /** 关闭所有的 clients */
-    closeAll() {
-        return __awaiter(this, void 0, void 0, function* () {
-            yield Promise.all(Object.values(this.mcpClients).map((client) => __awaiter(this, void 0, void 0, function* () {
-                try {
-                    yield this._closeOneClient(client);
-                }
-                catch (error) {
-                    if (this.onError) {
-                        this.onError((error === null || error === void 0 ? void 0 : error.message) || `Failed to close client`, error);
-                    }
-                    console.error(`Failed to close client`, error);
-                }
-            })));
-        });
-    }
-    /** 创建所有的 mcpClients，并更新它们的tools */
-    initClientsAndTools() {
-        return __awaiter(this, void 0, void 0, function* () {
-            var _a;
-            yield this._createMpcClients();
-            yield this._createMpcTools();
-            (_a = this.onUpdatedTools) === null || _a === void 0 ? void 0 : _a.call(this);
-        });
-    }
-    /** 全量更新所有的 mcpServers */
-    updateMcpServers(mcpServers) {
-        return __awaiter(this, void 0, void 0, function* () {
-            yield this.closeAll();
-            this.mcpServers = mcpServers || this.mcpServers;
-            yield this.initClientsAndTools();
-        });
-    }
-    /** 插入一个新的mcpServer，如果已经存在则返回false */
-    insertMcpServer(serverName, mcpServer) {
-        return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b;
-            // 检查是否已存在相同名称的服务器
-            if (this.mcpServers[serverName]) {
-                return false;
-            }
-            const client = yield this._createOneClient(mcpServer);
-            if (!client) {
-                // 创建客户端失败时直接返回，避免后续出现空指针问题
-                (_a = this.onError) === null || _a === void 0 ? void 0 : _a.call(this, `Failed to create MCP client: ${serverName}`);
-                return null;
-            }
-            this.mcpClients[serverName] = client;
-            const tools = yield getAISDKTools(client);
-            // 工具列表可能为 null，统一兜底为空对象，确保类型安全
-            this.mcpTools[serverName] = tools && typeof tools === 'object' ? tools : {};
-            this.mcpServers[serverName] = mcpServer;
-            (_b = this.onUpdatedTools) === null || _b === void 0 ? void 0 : _b.call(this);
-            return client;
-        });
-    }
-    /** 通过服务器名称删除mcpServer： mcpServers mcpClients  mcpTools ignoreToolnames  */
-    removeMcpServer(serverName) {
-        return __awaiter(this, void 0, void 0, function* () {
-            var _a;
-            if (!this.mcpServers[serverName]) {
-                return;
-            }
-            // 删除 mcpServer
-            delete this.mcpServers[serverName];
-            // 关闭并删除 client
-            const delClient = this.mcpClients[serverName];
-            delete this.mcpClients[serverName];
-            try {
-                yield this._closeOneClient(delClient);
-            }
-            catch (error) { }
-            // 删除 tools 并清理 ignoreToolnames
-            const delTool = this.mcpTools[serverName];
-            delete this.mcpTools[serverName];
-            if (delTool) {
-                Object.keys(delTool).forEach((toolName) => {
-                    this.ignoreToolnames = this.ignoreToolnames.filter((name) => name !== toolName);
-                });
-            }
-            (_a = this.onUpdatedTools) === null || _a === void 0 ? void 0 : _a.call(this);
-        });
-    }
-    /** 创建临时允许调用的tools集合 */
-    _tempMergeTools(extraTool = {}) {
-        // 将对象的值转换为数组后再 reduce
-        const toolsResult = Object.values(this.mcpTools).reduce((acc, curr) => (Object.assign(Object.assign({}, acc), curr)), {});
-        Object.assign(toolsResult, extraTool);
-        this.ignoreToolnames.forEach((name) => {
-            delete toolsResult[name];
-        });
-        return toolsResult;
-    }
-    _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.initClientsAndTools();
-            const chatOptions = Object.assign(Object.assign({ 
-                // @ts-ignore  ProviderV2 是所有llm的父类， 在每一个具体的llm 类都有一个选择model的函数用法
-                model: this.llm(model), stopWhen: stepCountIs(maxSteps) }, options), { tools: this._tempMergeTools(options.tools) });
-            if (options.message && !options.messages) {
-                this.messages.push({ role: 'user', content: options.message });
-                chatOptions.messages = [...this.messages];
-            }
-            const result = chatMethod(chatOptions);
-            (_b = result === null || result === void 0 ? void 0 : result.response) === null || _b === void 0 ? void 0 : _b.then((res) => {
-                this.messages.push(...res.messages);
-            });
-            return result;
-        });
-    }
-    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.js

@@ -1 +0,0 @@
-export {};

package/dist/agent/utils/getAISDKTools.js

@@ -1,36 +0,0 @@
-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中读取 tools 数组，并转换成 ai-sdk 的tool的对象格式。
- * @params client  一个已连接好的 WebMcpClient
- * @returns ai-sdk的dynamicTool对象。
- */
-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;
-    }
-});