@opentiny/next-sdk 0.0.1-alpha.2 → 0.0.1-alpha.4

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-export * from './client';
-export * from './server';
+export * from './next-client';
+export * from './next-server';
 export * from './mcp-host';
 export * from './plugins/createMessageChannelTransport';
 export * from './plugins/createProxy';

package/dist/index.mjs

@@ -1,357 +1,20 @@
-var b = Object.defineProperty;
-var x = (s, e, t) => e in s ? b(s, e, { enumerable: !0, configurable: !0, writable: !0, value: t }) : s[e] = t;
-var i = (s, e, t) => x(s, typeof e != "symbol" ? e + "" : e, t);
-import { Client as v } from "@modelcontextprotocol/sdk/client/index.js";
-import { LoggingMessageNotificationSchema as O, PromptListChangedNotificationSchema as P, ResourceUpdatedNotificationSchema as D, ToolListChangedNotificationSchema as q, ElicitRequestSchema as U, ListRootsRequestSchema as w, CreateMessageRequestSchema as y, PingRequestSchema as H, SetLevelRequestSchema as k, ListResourcesRequestSchema as _, UnsubscribeRequestSchema as $, SubscribeRequestSchema as j } from "@modelcontextprotocol/sdk/types.js";
-import { McpServer as Y } from "@modelcontextprotocol/sdk/server/mcp.js";
-import F from "openai";
-import { MessageChannelClientTransport as K, MessageChannelServerTransport as B, createTransportPair as J, createStreamProxy as V, createSseProxy as W, streamOptions as X, sseOptions as z } from "@opentiny/next";
-import { StreamableHTTPClientTransport as R } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
-import { SSEClientTransport as A } from "@modelcontextprotocol/sdk/client/sse.js";
-import { InMemoryTransport as G } from "@modelcontextprotocol/sdk/inMemory.js";
-const S = {
-  toolListChanged: q,
-  resourceUpdated: D,
-  promptListChanged: P,
-  loggingMessage: O
-}, Q = {
-  createMessage: y,
-  listRoots: w,
-  elicit: U
-};
-class L extends v {
-  constructor(t, o) {
-    super(t, o);
-    i(this, "nextTransport", null);
-    i(this, "isMessageChannel", !1);
-  }
-  // 注册插件
-  use(t) {
-    return t(this);
-  }
-  // 注册事件,包括notification以及requestHandle
-  on(t, o) {
-    t in S ? this.setNotificationHandler(S[t], o) : this.setRequestHandler(Q[t], o);
-  }
-  async connectTransport() {
-    return await this.connect(this.nextTransport);
-  }
-}
-const de = (s, e) => new L(s, e), Z = {
-  subscribe: j,
-  unsubscribe: $,
-  listResources: _,
-  createMessage: y,
-  listRoots: w,
-  setLevel: k,
-  ping: H
-};
-class ee extends Y {
-  constructor(t, o) {
-    super(t, o);
-    i(this, "nextTransport");
-    i(this, "isMessageChannel", !1);
-  }
-  use(t) {
-    return t(this);
-  }
-  async connectTransport() {
-    var t, o;
-    return this.isMessageChannel && await ((o = (t = this.nextTransport) == null ? void 0 : t.listen) == null ? void 0 : o.call(t)), await this.connect(this.nextTransport);
-  }
-  on(t, o) {
-    this.server.setRequestHandler(Z[t], o);
-  }
-}
-const Se = (s, e) => new ee(s, e);
-var c = /* @__PURE__ */ ((s) => (s.FUNCTION = "function", s.USER = "user", s.ASSISTANT = "assistant", s.DEVELOPER = "developer", s.SYSTEM = "system", s.TOOL = "tool", s))(c || {});
-const C = 3;
-class te {
-  // 最大迭代次数
-  constructor({ llmOption: e, mcpClients: t }) {
-    i(this, "llmOption");
-    i(this, "mcpClients");
-    i(this, "llm");
-    i(this, "mcpClientMap", /* @__PURE__ */ new Map());
-    i(this, "messages", []);
-    i(this, "toolClientMap", /* @__PURE__ */ new Map());
-    i(this, "iteration", C);
-    this.llmOption = e, this.mcpClients = t, this.messages.push({ role: c.SYSTEM, content: "You are a helpful assistant with access to tools." });
-  }
-  async generateMcpClient() {
-    for (let e = 0; e < this.mcpClients.length; e++) {
-      const t = this.mcpClients[e], { tools: o } = await t.listTools();
-      this.mcpClientMap.set(t, o), o.forEach((a) => {
-        this.toolClientMap.set(a.name, t);
-      });
-    }
-  }
-  init() {
-    this.llm = new F({
-      baseURL: this.llmOption.url,
-      apiKey: this.llmOption.apiKey,
-      dangerouslyAllowBrowser: !0
-    });
-  }
-  validateRequest(e) {
-    if (!e.messages || !Array.isArray(e.messages) || e.messages.length === 0)
-      throw new Error("请求必须包含至少一条消息");
-    for (const t of e.messages)
-      if (!t.role || !t.content)
-        throw new Error("每条消息必须包含角色和内容");
-  }
-  async getMcpTools() {
-    const e = [];
-    for (const [, o] of this.mcpClientMap.entries())
-      e.push(...o);
-    return e.map((o) => ({
-      type: "function",
-      function: {
-        name: o.name,
-        description: o.description,
-        parameters: o.inputSchema
-      }
-    }));
-  }
-  async doLLMChart() {
-    const e = await this.getMcpTools();
-    return await this.llm.chat.completions.create({
-      messages: this.messages,
-      model: this.llmOption.model,
-      tools: e,
-      stream: !0
-    });
-  }
-  /**
-   * 解析 LLM 回复，提取工具调用和最终回复内容
-   * @param response LLM 回复对象
-   * @returns [工具调用数组, 回复内容]
-   */
-  async parseToolCalls(e, t) {
-    var h, p, f, n;
-    const o = /* @__PURE__ */ new Map();
-    let a = "", r = "";
-    for await (const T of e) {
-      const u = T.choices[0].delta, g = u == null ? void 0 : u.tool_calls, d = u == null ? void 0 : u.content;
-      if (d && (a += d, t.onData({
-        delta: {
-          role: "assistant",
-          content: d
-        }
-      })), g && g.length > 0) {
-        const l = g[0];
-        if (l.id) {
-          r = l.id;
-          const m = {
-            id: l.id,
-            type: l.type,
-            function: {
-              name: ((h = l.function) == null ? void 0 : h.name) || "",
-              arguments: ((p = l.function) == null ? void 0 : p.arguments) || ""
-            }
-          };
-          o.set(r, m), t.onData({
-            delta: {
-              role: c.ASSISTANT,
-              content: `
-
-正在调用工具：${m.function.name}
-
-参数：`
-            }
-          });
-        }
-        if (o.has(r)) {
-          const m = o.get(r);
-          m && (m.function.arguments += ((f = l.function) == null ? void 0 : f.arguments) || "", t.onData({
-            delta: {
-              role: c.ASSISTANT,
-              content: `${(n = l.function) == null ? void 0 : n.arguments}`
-            }
-          }));
-        }
-      }
-    }
-    return t.onData({
-      delta: {
-        role: "assistant",
-        content: `
-
-`
-      }
-    }), [Array.from(o.values()), a];
-  }
-  /**
-   * 执行工具调用
-   * @param toolCalls 工具调用请求列表
-   * @returns 工具调用结果和消息
-   */
-  async callTools(e, t) {
-    var o, a, r, h;
-    try {
-      const p = [], f = [];
-      for (const n of e) {
-        const T = n.function.name, u = this.toolClientMap.get(T);
-        if (!u)
-          continue;
-        let g = {};
-        try {
-          g = typeof n.function.arguments == "string" ? JSON.parse(n.function.arguments) : n.function.arguments;
-        } catch (N) {
-          console.error(`Failed to parse tool arguments for ${T}:`, N), g = {};
-        }
-        const d = {
-          id: n.id,
-          type: n.type,
-          function: {
-            name: ((o = n.function) == null ? void 0 : o.name) || "",
-            arguments: ((a = n.function) == null ? void 0 : a.arguments) || ""
-          }
-        };
-        t.onData({
-          delta: {
-            role: c.TOOL,
-            toolCall: d
-          }
-        });
-        const l = await u.callTool({
-          name: T,
-          arguments: g
-        }), m = this.getToolCallMessage(l), I = {
-          role: c.TOOL,
-          tool_call_id: n.id,
-          content: m
-        }, E = {
-          id: n.id,
-          type: n.type,
-          callToolContent: m,
-          function: {
-            name: ((r = n.function) == null ? void 0 : r.name) || "",
-            arguments: ((h = n.function) == null ? void 0 : h.arguments) || ""
-          }
-        };
-        t.onData({
-          delta: {
-            role: c.TOOL,
-            toolCall: E
-          }
-        }), f.push(I), p.push({
-          call: T,
-          result: l
-        });
-      }
-      return { toolResults: p, toolCallMessages: f };
-    } catch (p) {
-      return console.error("Error calling tools:", p), { toolResults: [], toolCallMessages: [{ role: c.ASSISTANT, content: "call tools failed!" }] };
-    }
-  }
-  /**
-   * 解析工具调用结果，拼接为字符串
-   * @param toolCallResult 工具调用结果
-   * @returns 拼接后的字符串
-   */
-  getToolCallMessage(e) {
-    var o;
-    let t = "";
-    return (o = e.content) != null && o.length && e.content.forEach((a) => {
-      switch (a.type) {
-        case "text":
-          t += a.text;
-          break;
-        case "image":
-        case "audio":
-        case "resource":
-          t += a.data;
-          break;
-      }
-    }), t;
-  }
-  async chatStream(e, t) {
-    return await this.generateMcpClient(), typeof e == "string" ? this.messages.push({ role: c.USER, content: e }) : this.messages.push(e.messages[e.messages.length - 1]), this.iteration = C, await this.processSteamToolCallsAndResponses(t).catch((o) => {
-      console.error("Chat failed:", o);
-    }), "ok";
-  }
-  /**
-   * 聊天迭代主流程，支持多轮工具调用和最终回复
-   */
-  async processSteamToolCallsAndResponses(e) {
-    try {
-      const t = [];
-      for (; this.iteration > 0; ) {
-        const o = await this.doLLMChart(), [a, r] = await this.parseToolCalls(o, e);
-        if (a.length) {
-          const h = {
-            role: c.ASSISTANT,
-            content: `调用工具:${a.map((n) => n.function.name).join(",")}`,
-            tool_calls: a.map((n) => ({
-              id: n.id,
-              type: n.type,
-              function: n.function
-            }))
-          };
-          this.messages.push(h);
-          const { toolResults: p, toolCallMessages: f } = await this.callTools(a, e);
-          t.push(...p), f.forEach((n) => this.messages.push(n)), this.iteration--;
-        } else
-          this.messages.push({ role: c.ASSISTANT, content: r }), this.iteration = 0;
-      }
-      e.onDone();
-    } catch (t) {
-      throw console.error("Chat iteration failed:", t), t;
-    }
-  }
-}
-class se extends te {
-  constructor({ llmOption: e, mcpClients: t }) {
-    super({ llmOption: e, mcpClients: t });
-  }
-}
-const Ce = ({ llmOption: s, mcpClients: e }) => {
-  let t;
-  return s.llm === "deepseek" && (t = new se({ llmOption: s, mcpClients: e }), t.init()), t;
-}, Me = (s) => (e) => {
-  e.nextTransport ? e.nextTransport = new K(s.endpoint, s.globalObject) : e.nextTransport = new B(s.endpoint, s.globalObject), e.isMessageChannel = !0;
-}, oe = () => typeof window < "u", ne = (s, e) => {
-  const t = async () => {
-    e instanceof R ? await (e == null ? void 0 : e.terminateSession()) : e instanceof A && await (e == null ? void 0 : e.close()), s.close();
-  };
-  oe() && (window.addEventListener("beforeunload", t), window.addEventListener("pagehide", t));
-}, [ae, M] = J(), we = () => (s) => (s.nextTransport = M, M), ye = (s) => async (e) => {
-  let t = "", o = null;
-  if (s.type === "stream") {
-    const { sessionId: a, transport: r } = await V({
-      client: e,
-      ...s
-    });
-    t = a, o = r;
-  } else {
-    const { sessionId: a, transport: r } = await W({
-      client: e,
-      ...s
-    });
-    t = a, o = r;
-  }
-  return e.nextTransport = ae, ne(e, o), {
-    sessionId: t,
-    transport: o
-  };
-}, Re = (s) => (e) => {
-  const { type: t, url: o, token: a } = s;
-  let r = null;
-  t === "stream" ? r = new R(new URL(o), X(a)) : t === "sse" && (r = new A(new URL(o), z(a))), e.nextTransport = r;
-}, [re, ie] = G.createLinkedPair(), Ae = () => (s) => {
-  s instanceof L ? s.nextTransport = re : s.nextTransport = ie;
-};
+import { NextClient as t, createClient as o } from "./next-client/index.mjs";
+import { NextServer as n, createServer as p } from "./next-server/index.mjs";
+import { createMCPHost as c } from "./mcp-host/index.mjs";
+import { createMessageChannelTransport as f } from "./plugins/createMessageChannelTransport.mjs";
+import { createClientProxy as v, createServerProxy as C, serverTransport as l } from "./plugins/createProxy.mjs";
+import { connectMcpServer as S } from "./plugins/connectMcpServer.mjs";
+import { createInMemoryTransport as y } from "./plugins/createInMemoryTransport.mjs";
 export {
-  L as NextClient,
-  ee as NextServer,
-  Re as connectMcpServer,
-  de as createClient,
-  ye as createClientProxy,
-  Ae as createInMemoryTransport,
-  Ce as createMCPHost,
-  Me as createMessageChannelTransport,
-  Se as createServer,
-  we as createServerProxy,
-  M as serverTransport
+  t as NextClient,
+  n as NextServer,
+  S as connectMcpServer,
+  o as createClient,
+  v as createClientProxy,
+  y as createInMemoryTransport,
+  c as createMCPHost,
+  f as createMessageChannelTransport,
+  p as createServer,
+  C as createServerProxy,
+  l as serverTransport
 };

package/dist/mcp-host/core/index.d.ts

@@ -9,6 +9,7 @@ export declare class MCPHost {
     messages: any[];
     protected toolClientMap: Map<string, Client>;
     protected iteration: number;
+    protected resourcesMap: Map<string, any>;
     constructor({ llmOption, mcpClients }: {
         llmOption: any;
         mcpClients: Client[];
@@ -50,6 +51,7 @@ export declare class MCPHost {
         messages: Message[];
         options: any;
     }, handler: StreamHandler): Promise<string>;
+    clearMessages(): void;
     /**
      * 聊天迭代主流程，支持多轮工具调用和最终回复
      */

package/dist/mcp-host/core/index.mjs

@@ -0,0 +1,254 @@
+var S = Object.defineProperty;
+var A = (g, t, o) => t in g ? S(g, t, { enumerable: !0, configurable: !0, writable: !0, value: o }) : g[t] = o;
+var u = (g, t, o) => A(g, typeof t != "symbol" ? t + "" : t, o);
+import O from "openai";
+import { Role as r } from "../../type.mjs";
+const M = 3;
+class D {
+  // 缓存资源防止资源读取频繁触发
+  constructor({ llmOption: t, mcpClients: o }) {
+    u(this, "llmOption");
+    u(this, "mcpClients");
+    u(this, "llm");
+    u(this, "mcpClientMap", /* @__PURE__ */ new Map());
+    u(this, "messages", []);
+    u(this, "toolClientMap", /* @__PURE__ */ new Map());
+    u(this, "iteration", M);
+    // 最大迭代次数
+    u(this, "resourcesMap", /* @__PURE__ */ new Map());
+    this.llmOption = t, this.mcpClients = o, this.messages.push({ role: r.SYSTEM, content: "You are a helpful assistant with access to tools." });
+  }
+  async generateMcpClient() {
+    for (let t = 0; t < this.mcpClients.length; t++) {
+      const o = this.mcpClients[t], { tools: s } = await o.listTools();
+      this.mcpClientMap.set(o, s), s.forEach((a) => {
+        this.toolClientMap.set(a.name, o);
+      });
+      const { resources: n } = await o.listResources();
+      for (const a of n)
+        if (!this.resourcesMap.has(a.uri)) {
+          const c = await o.readResource({ uri: a.uri }), l = { ...a, content: c.contents };
+          this.resourcesMap.set(a.uri, l), this.messages.push({ role: r.SYSTEM, content: JSON.stringify(l) });
+        }
+    }
+  }
+  init() {
+    this.llm = new O({
+      baseURL: this.llmOption.url,
+      apiKey: this.llmOption.apiKey,
+      dangerouslyAllowBrowser: !0
+    });
+  }
+  validateRequest(t) {
+    if (!t.messages || !Array.isArray(t.messages) || t.messages.length === 0)
+      throw new Error("请求必须包含至少一条消息");
+    for (const o of t.messages)
+      if (!o.role || !o.content)
+        throw new Error("每条消息必须包含角色和内容");
+  }
+  async getMcpTools() {
+    const t = [];
+    for (const [, s] of this.mcpClientMap.entries())
+      t.push(...s);
+    return t.map((s) => ({
+      type: "function",
+      function: {
+        name: s.name,
+        description: s.description,
+        parameters: s.inputSchema
+      }
+    }));
+  }
+  async doLLMChart() {
+    const t = await this.getMcpTools(), o = {
+      messages: this.messages,
+      model: this.llmOption.model,
+      stream: !0
+    };
+    return t.length > 0 && (o.tools = t), await this.llm.chat.completions.create(o);
+  }
+  /**
+   * 解析 LLM 回复，提取工具调用和最终回复内容
+   * @param response LLM 回复对象
+   * @returns [工具调用数组, 回复内容]
+   */
+  async parseToolCalls(t, o) {
+    var c, l, h, e;
+    const s = /* @__PURE__ */ new Map();
+    let n = "", a = "";
+    for await (const C of t) {
+      const p = C.choices[0].delta, f = p == null ? void 0 : p.tool_calls, T = p == null ? void 0 : p.content;
+      if (T && (n += T, o.onData({
+        delta: {
+          role: "assistant",
+          content: T
+        }
+      })), f && f.length > 0) {
+        const i = f[0];
+        if (i.id) {
+          a = i.id;
+          const m = {
+            id: i.id,
+            type: i.type,
+            function: {
+              name: ((c = i.function) == null ? void 0 : c.name) || "",
+              arguments: ((l = i.function) == null ? void 0 : l.arguments) || ""
+            }
+          };
+          s.set(a, m), o.onData({
+            delta: {
+              role: r.ASSISTANT,
+              content: `
+
+正在调用工具：${m.function.name}
+
+参数：`
+            }
+          });
+        }
+        if (s.has(a)) {
+          const m = s.get(a);
+          m && (m.function.arguments += ((h = i.function) == null ? void 0 : h.arguments) || "", o.onData({
+            delta: {
+              role: r.ASSISTANT,
+              content: `${(e = i.function) == null ? void 0 : e.arguments}`
+            }
+          }));
+        }
+      }
+    }
+    return o.onData({
+      delta: {
+        role: "assistant",
+        content: `
+
+`
+      }
+    }), [Array.from(s.values()), n];
+  }
+  /**
+   * 执行工具调用
+   * @param toolCalls 工具调用请求列表
+   * @returns 工具调用结果和消息
+   */
+  async callTools(t, o) {
+    var s, n, a, c;
+    try {
+      const l = [], h = [];
+      for (const e of t) {
+        const C = e.function.name, p = this.toolClientMap.get(C);
+        if (!p)
+          continue;
+        let f = {};
+        try {
+          f = typeof e.function.arguments == "string" ? JSON.parse(e.function.arguments) : e.function.arguments;
+        } catch (w) {
+          console.error(`Failed to parse tool arguments for ${C}:`, w), f = {};
+        }
+        const T = {
+          id: e.id,
+          type: e.type,
+          function: {
+            name: ((s = e.function) == null ? void 0 : s.name) || "",
+            arguments: ((n = e.function) == null ? void 0 : n.arguments) || ""
+          }
+        };
+        o.onData({
+          delta: {
+            role: r.TOOL,
+            toolCall: T
+          }
+        });
+        const i = await p.callTool({
+          name: C,
+          arguments: f
+        }), m = this.getToolCallMessage(i), y = {
+          role: r.TOOL,
+          tool_call_id: e.id,
+          content: m
+        }, d = {
+          id: e.id,
+          type: e.type,
+          callToolContent: m,
+          function: {
+            name: ((a = e.function) == null ? void 0 : a.name) || "",
+            arguments: ((c = e.function) == null ? void 0 : c.arguments) || ""
+          }
+        };
+        o.onData({
+          delta: {
+            role: r.TOOL,
+            toolCall: d
+          }
+        }), h.push(y), l.push({
+          call: C,
+          result: i
+        });
+      }
+      return { toolResults: l, toolCallMessages: h };
+    } catch (l) {
+      return console.error("Error calling tools:", l), { toolResults: [], toolCallMessages: [{ role: r.ASSISTANT, content: "call tools failed!" }] };
+    }
+  }
+  /**
+   * 解析工具调用结果，拼接为字符串
+   * @param toolCallResult 工具调用结果
+   * @returns 拼接后的字符串
+   */
+  getToolCallMessage(t) {
+    var s;
+    let o = "";
+    return (s = t.content) != null && s.length && t.content.forEach((n) => {
+      switch (n.type) {
+        case "text":
+          o += n.text;
+          break;
+        case "image":
+        case "audio":
+        case "resource":
+          o += n.data;
+          break;
+      }
+    }), o;
+  }
+  async chatStream(t, o) {
+    return await this.generateMcpClient(), typeof t == "string" ? this.messages.push({ role: r.USER, content: t }) : this.messages.push(t.messages[t.messages.length - 1]), this.iteration = M, await this.processSteamToolCallsAndResponses(o).catch((s) => {
+      console.error("Chat failed:", s);
+    }), "ok";
+  }
+  clearMessages() {
+    this.messages = [];
+  }
+  /**
+   * 聊天迭代主流程，支持多轮工具调用和最终回复
+   */
+  async processSteamToolCallsAndResponses(t) {
+    try {
+      const o = [];
+      for (; this.iteration > 0; ) {
+        const s = await this.doLLMChart(), [n, a] = await this.parseToolCalls(s, t);
+        if (n.length) {
+          const c = {
+            role: r.ASSISTANT,
+            content: `调用工具:${n.map((e) => e.function.name).join(",")}`,
+            tool_calls: n.map((e) => ({
+              id: e.id,
+              type: e.type,
+              function: e.function
+            }))
+          };
+          this.messages.push(c);
+          const { toolResults: l, toolCallMessages: h } = await this.callTools(n, t);
+          o.push(...l), h.forEach((e) => this.messages.push(e)), this.iteration--;
+        } else
+          this.messages.push({ role: r.ASSISTANT, content: a }), this.iteration = 0;
+      }
+      t.onDone();
+    } catch (o) {
+      throw console.error("Chat iteration failed:", o), o;
+    }
+  }
+}
+export {
+  D as MCPHost
+};

package/dist/mcp-host/deepseek/index.mjs

@@ -0,0 +1,9 @@
+import { MCPHost as s } from "../core/index.mjs";
+class p extends s {
+  constructor({ llmOption: e, mcpClients: o }) {
+    super({ llmOption: e, mcpClients: o });
+  }
+}
+export {
+  p as DeepSeekMcpHost
+};

package/dist/mcp-host/index.mjs

@@ -0,0 +1,8 @@
+import { DeepSeekMcpHost as r } from "./deepseek/index.mjs";
+const s = ({ llmOption: t, mcpClients: o }) => {
+  let e;
+  return t.llm === "deepseek" && (e = new r({ llmOption: t, mcpClients: o }), e.init()), e;
+};
+export {
+  s as createMCPHost
+};