@glodon-aiot/bot-client-sdk 3.25.1-alpha.6 → 3.26.0-alpha.4

package/dist/es/AgentSession.mjs

@@ -1,98 +1,122 @@
-var b = Object.defineProperty, T = Object.defineProperties;
-var x = Object.getOwnPropertyDescriptors;
-var _ = Object.getOwnPropertySymbols;
-var E = Object.prototype.hasOwnProperty, L = Object.prototype.propertyIsEnumerable;
-var A = (m, t, i) => t in m ? b(m, t, { enumerable: !0, configurable: !0, writable: !0, value: i }) : m[t] = i, d = (m, t) => {
-  for (var i in t || (t = {}))
-    E.call(t, i) && A(m, i, t[i]);
-  if (_)
-    for (var i of _(t))
-      L.call(t, i) && A(m, i, t[i]);
-  return m;
-}, w = (m, t) => T(m, x(t));
-var p = (m, t, i) => new Promise((o, r) => {
-  var s = (a) => {
+var T = Object.defineProperty, j = Object.defineProperties;
+var L = Object.getOwnPropertyDescriptors;
+var S = Object.getOwnPropertySymbols;
+var F = Object.prototype.hasOwnProperty, N = Object.prototype.propertyIsEnumerable;
+var M = (r, c, e) => c in r ? T(r, c, { enumerable: !0, configurable: !0, writable: !0, value: e }) : r[c] = e, h = (r, c) => {
+  for (var e in c || (c = {}))
+    F.call(c, e) && M(r, e, c[e]);
+  if (S)
+    for (var e of S(c))
+      N.call(c, e) && M(r, e, c[e]);
+  return r;
+}, w = (r, c) => j(r, L(c));
+var b = (r, c, e) => (M(r, typeof c != "symbol" ? c + "" : c, e), e);
+var y = (r, c, e) => new Promise((o, a) => {
+  var n = (i) => {
     try {
-      n(i.next(a));
-    } catch (h) {
-      r(h);
+      s(e.next(i));
+    } catch (d) {
+      a(d);
     }
-  }, e = (a) => {
+  }, t = (i) => {
     try {
-      n(i.throw(a));
-    } catch (h) {
-      r(h);
+      s(e.throw(i));
+    } catch (d) {
+      a(d);
     }
-  }, n = (a) => a.done ? o(a.value) : Promise.resolve(a.value).then(s, e);
-  n((i = i.apply(m, t)).next());
+  }, s = (i) => i.done ? o(i.value) : Promise.resolve(i.value).then(n, t);
+  s((e = e.apply(r, c)).next());
 });
-import { ChatMessageRole as v, PromptVariableType as P } from "@glodon-aiot/apis";
-import q from "./Logger.mjs";
-import { configuration as y } from "./Config.mjs";
-import { io as F } from "socket.io-client";
+import { ChatMessageRole as v, PromptVariableType as A } from "@glodon-aiot/apis";
+import O from "./Logger.mjs";
+import { configuration as C } from "./Config.mjs";
+import { io as U } from "socket.io-client";
 import { UserMode as I } from "./PayloadTypes.mjs";
-import N, { SessionErrors as M } from "./Session.mjs";
-import U from "dayjs";
-const u = new q("[AgentSession] ");
-class K extends N {
-  constructor(t, i = "", o, r, s, e) {
-    super(t, i, o, r, s, e);
+import D, { SessionErrors as k } from "./Session.mjs";
+import V from "dayjs";
+import { v4 as J } from "uuid";
+const f = new O("[AgentSession] ");
+function _(r) {
+  if (typeof r != "string")
+    return r;
+  const c = r.trim();
+  if (c)
+    try {
+      return JSON.parse(c);
+    } catch (e) {
+      return r;
+    }
+}
+function x(r) {
+  return r == null || r === "" || typeof r == "object" && !Array.isArray(r) && Object.keys(r).length === 0;
+}
+function Y(r, c) {
+  const e = _(r == null ? void 0 : r.query);
+  if (!x(e))
+    return e;
+  const o = _(c);
+  return x(o) && e != null ? e : o;
+}
+class Z extends D {
+  constructor(e, o = "", a, n, t, s) {
+    super(e, o, a, n, t, s);
+    b(this, "inflightCalls", /* @__PURE__ */ new Set());
   }
   prepareConnection() {
-    return new Promise((t) => p(this, null, function* () {
-      this.socket && t(!0);
-      const i = {
+    return new Promise((e) => y(this, null, function* () {
+      this.socket && e(!0);
+      const o = {
         version: "1.0",
-        stream: JSON.stringify(y.stream),
+        stream: JSON.stringify(C.stream),
         appId: this._applicationId,
         sessionId: this.id,
         Authorization: `${this.token}`,
-        userMode: y.debug ? I.Debug : I.Normal
-      }, o = new URL(y.apiRoot), r = yield F(o.origin, {
-        path: o.pathname.replace("api", "socketio") + "/copilot",
-        query: i,
+        userMode: C.debug ? I.Debug : I.Normal
+      }, a = new URL(C.apiRoot), n = yield U(a.origin, {
+        path: a.pathname.replace("api", "socketio") + "/copilot",
+        query: o,
         transports: ["websocket"]
       });
-      r.on("connect", () => {
-        u.log("socket:connect"), this.event.emit("socket:connect"), r.emit("init");
-      }), r.on("disconnect", () => {
-        u.log("socket:disconnect"), this.event.emit("socket:disconnect"), this._ready = !1;
-      }), r.on("ready", (s) => {
-        if (s.code === 200) {
-          u.log("ready");
-          const e = y.debug ? I.Debug : I.Normal;
-          e === I.Debug && s.userMode, this._ready || (this._ready = !0, this.event.emit("ready")), this.frozen = !1, this.registerMcpFromClientConfig(), t(!0);
+      n.on("connect", () => {
+        f.log("socket:connect"), this.event.emit("socket:connect"), n.emit("init");
+      }), n.on("disconnect", () => {
+        f.log("socket:disconnect"), this.event.emit("socket:disconnect"), this._ready = !1;
+      }), n.on("ready", (t) => {
+        if (t.code === 200) {
+          f.log("ready");
+          const s = C.debug ? I.Debug : I.Normal;
+          s === I.Debug && t.userMode, this._ready || (this._ready = !0, this.event.emit("ready")), this.frozen = !1, this.registerMcpFromClientConfig(), e(!0);
         } else {
           this._ready && (this._ready = !1);
-          const e = new Error(
-            M.SocketConnectionFailed
+          const s = new Error(
+            k.SocketConnectionFailed
             // , {
             //   cause: data,
             // }
           );
-          u.error("error", e), this.event.emit("error", e), t(!1);
+          f.error("error", s), this.event.emit("error", s), e(!1);
         }
-      }), r.on("action", (s, e) => p(this, null, function* () {
-        var k;
+      }), n.on("action", (t, s) => y(this, null, function* () {
+        var R;
         if (this._stopAnswer || !this.id)
           return;
         const {
-          type: n,
+          type: i,
           data: {
-            runnerIsClient: a,
-            pluginCode: h,
+            runnerIsClient: d,
+            pluginCode: l,
             funcCode: g,
-            parameters: c,
-            requestBody: j
+            parameters: m,
+            requestBody: E
           }
-        } = s, l = this.messages[this.messages.length - 1];
-        if (!l.messageId)
-          l.actions || (l.actions = []), l.actions[s.data.step - 1] || (l.actions[s.data.step - 1] = []), l.actions[s.data.step - 1].push(s), this.event.emit("message:action", {
-            messageId: l.messageId,
-            action: s
-          }), this.event.emit("message:updated", l);
+        } = t, u = this.messages[this.messages.length - 1];
+        if (!u.messageId)
+          u.actions || (u.actions = []), u.actions[t.data.step - 1] || (u.actions[t.data.step - 1] = []), u.actions[t.data.step - 1].push(t), this.event.emit("message:action", {
+            messageId: u.messageId,
+            action: t
+          }), this.event.emit("message:updated", u);
         else {
-          const f = {
+          const p = {
             messageId: "",
             content: "",
             role: v.Robot,
@@ -101,20 +125,20 @@ class K extends N {
             searchReference: [],
             isLoading: !0,
             messageTime: "",
-            actions: [[s]],
+            actions: [[t]],
             comment: {
               isAgree: 0
             },
             isReplying: !0
           };
-          this.pushMessage(f), this.event.emit("message:action", {
-            messageId: f.messageId,
-            action: s
+          this.pushMessage(p), this.event.emit("message:action", {
+            messageId: p.messageId,
+            action: t
           });
         }
         if (this.inactivating) {
-          const f = new Error(
-            M.Inactive
+          const p = new Error(
+            k.Inactive
             // {
             //   cause: {
             //     code: -1,
@@ -123,50 +147,48 @@ class K extends N {
             //   },
             // }
           );
-          u.warn("error", f), this.event.emit("error", f);
-        } else if (n === "FUNCTION_CALL" && a) {
-          u.log(`calling function ${g} of plugin  ${h} with argument:
- ${JSON.stringify(c)}`);
-          const f = (k = y.plugins) == null ? void 0 : k[h];
-          if (!f) {
-            e({
+          f.warn("error", p), this.event.emit("error", p);
+        } else if (i === "FUNCTION_CALL" && d) {
+          f.log(`calling function ${g} of plugin  ${l} with argument:
+ ${JSON.stringify(m)}`);
+          const p = (R = C.plugins) == null ? void 0 : R[l];
+          if (!p) {
+            s({
               code: 404,
               message: "没有找到插件。"
             });
             return;
           }
-          const R = f[g];
-          if (typeof R != "function") {
-            e({
+          const P = p[g];
+          if (typeof P != "function") {
+            s({
               code: 404,
               message: "没有找到插件方法。"
             });
             return;
           }
-          let S = c == null ? void 0 : c.query;
-          try {
-            S = JSON.parse(c == null ? void 0 : c.query);
-          } catch (C) {
-          }
-          R(S, (...C) => {
-            this._stopAnswer || e(...C);
+          const q = Y(m, E);
+          this.invokeClientPlugin(P, q, s, {
+            requestId: t.requestId,
+            pluginCode: l,
+            functionCode: g
           });
         }
-      })), r.on("answer", (s) => {
+      })), n.on("answer", (t) => {
         if (!this._stopAnswer)
-          if (s.data.code !== 200 && (s.data.index = -1), s.data.stream || s.data.stream === void 0) {
-            let e = this.messages[this.messages.length - 1];
-            if (e && e.role === v.Robot)
-              this.handleStreamAnswer(s.data, e), this.messages[this.messages.length - 1] = e, this.event.emit("message:content", e), this.event.emit("message:updated", e), e.index === -1 && (e.isReplying = !1, this.event.emit("answer:finish", e));
+          if (t.data.code !== 200 && (t.data.index = -1), t.data.stream || t.data.stream === void 0) {
+            let s = this.messages[this.messages.length - 1];
+            if (s && s.role === v.Robot)
+              this.handleStreamAnswer(t.data, s), this.messages[this.messages.length - 1] = s, this.event.emit("message:content", s), this.event.emit("message:updated", s), s.index === -1 && (s.isReplying = !1, this.event.emit("answer:finish", s));
             else {
-              const n = {
+              const i = {
                 content: "",
                 role: v.Robot,
                 reference: [],
                 searchReference: [],
                 messageId: "",
                 sessionId: this.id,
-                messageTime: U(/* @__PURE__ */ new Date()).format("YYYY-MM-DD HH:mm:ss"),
+                messageTime: V(/* @__PURE__ */ new Date()).format("YYYY-MM-DD HH:mm:ss"),
                 comment: {
                   isAgree: 0
                 },
@@ -181,14 +203,14 @@ class K extends N {
                 thinkingStatus: "",
                 searchingStatus: ""
               };
-              this.handleStreamAnswer(s.data, n), this.pushMessage(n), this.event.emit("message:content", n), n.index === -1 && (n.isReplying = !1, this.event.emit("answer:finish", n));
+              this.handleStreamAnswer(t.data, i), this.pushMessage(i), this.event.emit("message:content", i), i.index === -1 && (i.isReplying = !1, this.event.emit("answer:finish", i));
             }
-          } else if (s.data.updateSession && this.data && (this.data = d(d({}, this.data), s.data.updateSession), this.event.emit("data:updated", this.data)), s.data.index === -1) {
-            let e = this.messages[this.messages.length - 1];
-            if (e && !e.messageId)
-              e = d(d({}, e), s.data), this.event.emit("message:content", e), this.event.emit("message:updated", e), this.event.emit("answer:finish", e);
+          } else if (t.data.updateSession && this.data && (this.data = h(h({}, this.data), t.data.updateSession), this.event.emit("data:updated", this.data)), t.data.index === -1) {
+            let s = this.messages[this.messages.length - 1];
+            if (s && !s.messageId)
+              s = h(h({}, s), t.data), this.event.emit("message:content", s), this.event.emit("message:updated", s), this.event.emit("answer:finish", s);
             else {
-              const n = d({
+              const i = h({
                 messageId: "",
                 content: "",
                 role: v.Robot,
@@ -202,12 +224,12 @@ class K extends N {
                   isAgree: 0
                 },
                 isReplying: !0
-              }, s.data);
-              this.event.emit("message:content", n), this.pushMessage(n), this.event.emit("answer:finish", n);
+              }, t.data);
+              this.event.emit("message:content", i), this.pushMessage(i), this.event.emit("answer:finish", i);
             }
           } else {
-            let e = this.messages[this.messages.length - 1];
-            e && !e.messageId ? (e = d(d({}, e), s.data), this.messages[this.messages.length - 1] = e, this.event.emit("message:content", e), this.event.emit("message:updated", e)) : (this.pushMessage(d({
+            let s = this.messages[this.messages.length - 1];
+            s && !s.messageId ? (s = h(h({}, s), t.data), this.messages[this.messages.length - 1] = s, this.event.emit("message:content", s), this.event.emit("message:updated", s)) : (this.pushMessage(h({
               messageId: "",
               content: "",
               role: v.Robot,
@@ -221,31 +243,31 @@ class K extends N {
                 isAgree: 0
               },
               isReplying: !0
-            }, s.data)), this.event.emit("message:content", s.data));
+            }, t.data)), this.event.emit("message:content", t.data));
           }
-      }), r.on("reference", (s) => p(this, null, function* () {
+      }), n.on("reference", (t) => y(this, null, function* () {
         if (this._stopAnswer)
           return;
         if (!this.id) {
-          const c = new Error(
-            M.IdUndefined
+          const m = new Error(
+            k.IdUndefined
             // {
             //   cause: SESSION_ERROR_CAUSE[SessionErrors.IdUndefined],
             // }
           );
-          u.error(c), this.event.emit("error", c), this.frozen = !1;
+          f.error(m), this.event.emit("error", m), this.frozen = !1;
           return;
         }
         const {
-          type: e,
-          data: n,
-          requestId: a
-        } = s, h = this.messages.findIndex((c) => c.messageId === a && c.role === v.User);
-        if (h < 0)
+          type: s,
+          data: i,
+          requestId: d
+        } = t, l = this.messages.findIndex((m) => m.messageId === d && m.role === v.User);
+        if (l < 0)
           return;
-        const g = h + 1;
+        const g = l + 1;
         if (g >= this.messages.length) {
-          const c = {
+          const m = {
             messageId: "",
             content: "",
             role: v.Robot,
@@ -260,61 +282,61 @@ class K extends N {
             },
             isReplying: !0
           };
-          this.pushMessage(c);
+          this.pushMessage(m);
         }
-        e === "KG" ? (this.messages[g] = w(d({}, this.messages[g]), {
-          reference: n.knowledgeReference
-        }), yield this.fillMessageReference(this.messages[g]), this.event.emit("message:reference", n.knowledgeReference), this.event.emit("message:updated", this.messages[g])) : e === "WEB_SEARCH" && (this.messages[g] = w(d({}, this.messages[g]), {
-          searchReference: n.knowledgeReference
-        }), this.event.emit("message:searchreference", n.searchReference), this.event.emit("message:updated", this.messages[g]));
-      })), this.socket = r, r.open();
+        s === "KG" ? (this.messages[g] = w(h({}, this.messages[g]), {
+          reference: i.knowledgeReference
+        }), yield this.fillMessageReference(this.messages[g]), this.event.emit("message:reference", i.knowledgeReference), this.event.emit("message:updated", this.messages[g])) : s === "WEB_SEARCH" && (this.messages[g] = w(h({}, this.messages[g]), {
+          searchReference: i.knowledgeReference
+        }), this.event.emit("message:searchreference", i.searchReference), this.event.emit("message:updated", this.messages[g]));
+      })), this.socket = n, n.open();
     }));
   }
   registerMcpFromClientConfig() {
-    return y.mcp ? new Promise((t, i) => {
+    return C.mcp ? new Promise((e, o) => {
       if (!this.socket) {
-        i("socket is undefined");
+        o("socket is undefined");
         return;
       }
-      this.socket.emit("add_mcp_servers", y.mcp, (o) => {
-        o.code === 200 ? (Object.keys(o.data).forEach((r) => {
-          o.data[r].code !== 200 && console.error(r, "MCP Server 注册失败。", o.data[r]);
-        }), t(!0)) : (console.error("MCP Server 注册失败。", o), i(o));
+      this.socket.emit("add_mcp_servers", C.mcp, (a) => {
+        a.code === 200 ? (Object.keys(a.data).forEach((n) => {
+          a.data[n].code !== 200 && console.error(n, "MCP Server 注册失败。", a.data[n]);
+        }), e(!0)) : (console.error("MCP Server 注册失败。", a), o(a));
       });
     }) : Promise.resolve(!0);
   }
-  addMcpServers(t) {
-    return p(this, null, function* () {
-      return this.ready ? this.socket ? new Promise((i, o) => {
-        this.socket && this.socket.emit("add_mcp_servers", t, (r) => {
-          r.code === 200 ? (Object.keys(r.data).forEach((s) => {
-            r.data[s].code !== 200 && console.error(s, "MCP Server 注册失败。", r.data[s]);
-          }), i(!0)) : (console.error("MCP Server 注册失败。", r), o(r));
+  addMcpServers(e) {
+    return y(this, null, function* () {
+      return this.ready ? this.socket ? new Promise((o, a) => {
+        this.socket && this.socket.emit("add_mcp_servers", e, (n) => {
+          n.code === 200 ? (Object.keys(n.data).forEach((t) => {
+            n.data[t].code !== 200 && console.error(t, "MCP Server 注册失败。", n.data[t]);
+          }), o(!0)) : (console.error("MCP Server 注册失败。", n), a(n));
         });
       }) : (console.error(" socket is undefined."), Promise.reject("socket is undefined")) : (console.error("session is not avalible."), Promise.reject("session is not avalible"));
     });
   }
   listMcpServers() {
-    return p(this, null, function* () {
-      return this.ready ? this.socket ? new Promise((t, i) => {
-        this.socket && this.socket.emit("list_mcp_servers", (o) => {
-          o.code === 200 ? t(o.data) : i(o);
+    return y(this, null, function* () {
+      return this.ready ? this.socket ? new Promise((e, o) => {
+        this.socket && this.socket.emit("list_mcp_servers", (a) => {
+          a.code === 200 ? e(a.data) : o(a);
         });
       }) : (console.error(" socket is undefined."), Promise.reject("socket is undefined")) : (console.error("session is not avalible."), Promise.reject("session is not avalible"));
     });
   }
-  sendQuestion(t, i) {
-    return p(this, null, function* () {
-      var r, s, e;
+  sendQuestion(e, o) {
+    return y(this, null, function* () {
+      var n, t, s;
       this.convertKnowledgesToChatFormat(this.knowledges);
-      let o = w(d({}, t), {
-        promptVariables: this.promptVariables.filter((n) => n.type === P.User).reduce((n, a) => w(d({}, n), {
-          [a.key]: a.value
+      let a = w(h({}, e), {
+        promptVariables: this.promptVariables.filter((i) => i.type === A.User).reduce((i, d) => w(h({}, i), {
+          [d.key]: d.value
         }), {}),
-        systemPromptVariables: this.promptVariables.filter((n) => n.type === P.System).reduce((n, a) => w(d({}, n), {
-          [a.key]: a.value
+        systemPromptVariables: this.promptVariables.filter((i) => i.type === A.System).reduce((i, d) => w(h({}, i), {
+          [d.key]: d.value
         }), {}),
-        knowledges: (r = this.knowledges) == null ? void 0 : r.map((n) => n.id),
+        knowledges: (n = this.knowledges) == null ? void 0 : n.map((i) => i.id),
         //agent支持指定文件时再放开
         // knowledges: knowledgesForChat,
         //开启内容摘要
@@ -322,31 +344,74 @@ class K extends N {
       });
       if (!this.active) {
         if (yield this.activate(), !this.socket) {
-          const n = new Error(
-            M.SocketConnectionFailed
+          const i = new Error(
+            k.SocketConnectionFailed
             // {
             //   cause: { code: -1, message: 'socket is null' },
             // }
           );
-          return u.warn("error", n), this.event.emit("error", n), !1;
+          return f.warn("error", i), this.event.emit("error", i), !1;
         }
-        (s = this.socket) == null || s.emit("question", o);
+        (t = this.socket) == null || t.emit("question", a);
         return;
       }
-      (e = this.socket) == null || e.emit("question", o);
+      (s = this.socket) == null || s.emit("question", a);
     });
   }
+  /**
+   * 调用客户端插件：传入第三个参数 context（旧插件可忽略），结果最多回传一次，
+   * 停止回答或调用被取消后不再回传。
+   */
+  invokeClientPlugin(e, o, a, n) {
+    const t = new AbortController();
+    this.inflightCalls.add(t);
+    let s = !1;
+    const i = (...l) => {
+      s || (s = !0, this.inflightCalls.delete(t), !(this._stopAnswer || t.signal.aborted) && a(...l));
+    }, d = {
+      requestId: n.requestId || J(),
+      applicationId: this._applicationId,
+      sessionId: this.id,
+      pluginCode: n.pluginCode,
+      functionCode: n.functionCode,
+      signal: t.signal
+    };
+    try {
+      const l = e(o, i, d);
+      l && typeof l.catch == "function" && l.catch((g) => {
+        f.error("client plugin failed", g), i({
+          code: 500,
+          message: "客户端插件执行失败。"
+        });
+      });
+    } catch (l) {
+      f.error("client plugin failed", l), i({
+        code: 500,
+        message: "客户端插件执行失败。"
+      });
+    }
+  }
+  abortClientPluginCalls() {
+    const e = Array.from(this.inflightCalls);
+    this.inflightCalls.clear(), e.forEach((o) => o.abort());
+  }
   stopAnswer() {
-    var t;
-    (t = this.socket) == null || t.emit("stop_answer");
+    var e;
+    this.abortClientPluginCalls(), (e = this.socket) == null || e.emit("stop_answer");
+  }
+  inactivate(e = !1) {
+    return this.abortClientPluginCalls(), super.inactivate(e);
+  }
+  destroy() {
+    return this.abortClientPluginCalls(), super.destroy();
   }
-  convertKnowledgesToChatFormat(t) {
-    return t ? t.map((i) => i != null && i.fileIds ? {
-      id: i.id,
-      fileIds: i.fileIds
-    } : i.id) : [];
+  convertKnowledgesToChatFormat(e) {
+    return e ? e.map((o) => o != null && o.fileIds ? {
+      id: o.id,
+      fileIds: o.fileIds
+    } : o.id) : [];
   }
 }
 export {
-  K as default
+  Z as default
 };