@opentiny/next-sdk 0.1.0 → 0.1.1

package/dist/agent/AgentModelProvider.js

@@ -0,0 +1,87 @@
+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 { getMcpClients, getMcpTools } from './utils';
+import { AIProviderFactories } from './utils/aiProviderFactories';
+export class AgentModelProvider {
+    constructor({ llmConfig, mcpServers, llm }) {
+        this.isGetMcpClients = false;
+        this.mcpClients = [];
+        // 1、保存 mcpServer
+        this.mcpServers = mcpServers || [];
+        // 2、保存 llm
+        if (llm) {
+            this.llm = llm;
+        }
+        else if (llmConfig) {
+            let providerFn;
+            if (typeof llmConfig.providerType === 'string') {
+                providerFn = AIProviderFactories[llmConfig.providerType];
+            }
+            else {
+                providerFn = llmConfig.providerType;
+            }
+            this.llm = providerFn({
+                apiKey: llmConfig.apiKey,
+                baseURL: llmConfig.baseURL
+            });
+        }
+        else {
+            throw new Error('Either llmConfig or llm must be provided');
+        }
+    }
+    initClients() {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (!this.isGetMcpClients) {
+                this.mcpClients = yield getMcpClients(this.mcpServers);
+                this.isGetMcpClients = true;
+            }
+        });
+    }
+    chat(_a) {
+        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');
+            }
+            // 每次会话需要获取最新的工具列表，因为工具是会发生变化的
+            yield this.initClients();
+            const tools = yield getMcpTools(this.mcpClients);
+            return generateText(Object.assign({ 
+                // @ts-ignore  ProviderV2 是所有llm的父类， 在每一个具体的llm 类都有一个选择model的函数用法
+                model: this.llm(model), tools: tools, stopWhen: stepCountIs(maxSteps) }, options));
+        });
+    }
+    chatStream(_a) {
+        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');
+            }
+            // 每次会话需要获取最新的工具列表，因为工具是会发生变化的
+            yield this.initClients();
+            const tools = yield getMcpTools(this.mcpClients);
+            return streamText(Object.assign({ 
+                // @ts-ignore 同上
+                model: this.llm(model), tools: tools, stopWhen: stepCountIs(maxSteps) }, options));
+        });
+    }
+}

package/dist/agent/type.d.ts

@@ -0,0 +1,32 @@
+export type { experimental_MCPClient as MCPClient } from 'ai';
+import type { ProviderV2 } from '@ai-sdk/provider';
+import type { MCPTransport } from 'ai';
+/** 代理模型提供器的大语言配置对象 */
+export interface IAgentModelProviderLlmConfig {
+    apiKey: string;
+    baseURL: string;
+    /** 支持内置的常用模型，或者传入一个ai-sdk官方的Provider工厂函数
+     * @example
+     * import { createOpenAI } from '@ai-sdk/openai'
+     */
+    providerType: 'openai' | 'deepseek' | ((options: any) => ProviderV2);
+}
+/** Mcp Server的配置对象 */
+export type McpServerConfig = {
+    type: 'streamableHttp' | 'sse';
+    url: string;
+} | {
+    transport: MCPTransport;
+};
+/** */
+export interface IAgentModelProviderOption {
+    /** ai-sdk官方的Provider实例，不能与 llmConfig 同时传入
+     * @example
+     * import { openai } from '@ai-sdk/openai'
+     */
+    llm?: ProviderV2;
+    /** 代理模型提供器的大语言配置对象, 不能与 llm 同时传入 */
+    llmConfig?: IAgentModelProviderLlmConfig;
+    /** Mcp Server的配置对象的集合 */
+    mcpServers?: McpServerConfig[];
+}

package/dist/agent/type.js

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

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

@@ -0,0 +1,6 @@
+import { createOpenAI } from '@ai-sdk/openai';
+import { createDeepSeek } from '@ai-sdk/deepseek';
+export declare const AIProviderFactories: {
+    openai: typeof createOpenAI;
+    deepseek: typeof createDeepSeek;
+};

package/dist/agent/utils/aiProviderFactories.js

@@ -0,0 +1,6 @@
+import { createOpenAI } from '@ai-sdk/openai';
+import { createDeepSeek } from '@ai-sdk/deepseek';
+export const AIProviderFactories = {
+    ['openai']: createOpenAI,
+    ['deepseek']: createDeepSeek
+};

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

@@ -0,0 +1,8 @@
+import { ToolSet } from 'ai';
+import { McpServerConfig, MCPClient } from '../type';
+/** 创建 McpClients, 其中 mcpServers 允许为配置为 McpServerConfig, 或者任意的 MCPTransport
+ * 参考: https://ai-sdk.dev/docs/ai-sdk-core/tools-and-tool-calling#initializing-an-mcp-client
+ */
+export declare const getMcpClients: (mcpServers: McpServerConfig[]) => Promise<(MCPClient | never[])[]>;
+/** 合并所有的Mcp Tools */
+export declare const getMcpTools: (mcpClients: MCPClient[]) => Promise<ToolSet>;

package/dist/agent/utils/index.js

@@ -0,0 +1,43 @@
+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 { experimental_createMCPClient as createMCPClient } from 'ai';
+import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
+/** 创建 McpClients, 其中 mcpServers 允许为配置为 McpServerConfig, 或者任意的 MCPTransport
+ * 参考: https://ai-sdk.dev/docs/ai-sdk-core/tools-and-tool-calling#initializing-an-mcp-client
+ */
+export const getMcpClients = (mcpServers) => __awaiter(void 0, void 0, void 0, function* () {
+    if (!mcpServers || (mcpServers === null || mcpServers === void 0 ? void 0 : mcpServers.length) === 0) {
+        return [];
+    }
+    // 使用 Promise.all 并行处理所有 mcpServer 项
+    const allMcpClients = yield Promise.all(mcpServers.map((item) => __awaiter(void 0, void 0, void 0, function* () {
+        try {
+            let transport;
+            // FIXME
+            if ('type' in item && item.type === 'streamableHttp') {
+                transport = new StreamableHTTPClientTransport(new URL(item.url));
+            }
+            else {
+                transport = item; // sse 或 自定义的 MCPTranport
+            }
+            return createMCPClient({ transport: transport });
+        }
+        catch (error) {
+            console.error(`Failed to create MCP client`, item, error);
+            return [];
+        }
+    })));
+    return allMcpClients;
+});
+/** 合并所有的Mcp Tools */
+export const getMcpTools = (mcpClients) => __awaiter(void 0, void 0, void 0, function* () {
+    const tools = yield Promise.all(mcpClients.map((client) => { var _a; return (_a = client === null || client === void 0 ? void 0 : client.tools) === null || _a === void 0 ? void 0 : _a.call(client); }));
+    return tools.reduce((acc, curr) => (Object.assign(Object.assign({}, acc), curr)), {});
+});