@glodon-aiot/bot-client-sdk 3.3.3 → 3.3.4-beta.2

package/dist/es/AgentSession.mjs

@@ -1,87 +1,88 @@
-var R = Object.defineProperty, b = Object.defineProperties;
-var S = Object.getOwnPropertyDescriptors;
-var C = Object.getOwnPropertySymbols;
-var A = Object.prototype.hasOwnProperty, E = Object.prototype.propertyIsEnumerable;
-var P = (d, t, n) => t in d ? R(d, t, { enumerable: !0, configurable: !0, writable: !0, value: n }) : d[t] = n, h = (d, t) => {
-  for (var n in t || (t = {}))
-    A.call(t, n) && P(d, n, t[n]);
-  if (C)
-    for (var n of C(t))
-      E.call(t, n) && P(d, n, t[n]);
-  return d;
-}, k = (d, t) => b(d, S(t));
-var p = (d, t, n) => new Promise((r, i) => {
-  var s = (c) => {
+var A = Object.defineProperty, C = Object.defineProperties;
+var P = Object.getOwnPropertyDescriptors;
+var b = Object.getOwnPropertySymbols;
+var x = Object.prototype.hasOwnProperty, E = Object.prototype.propertyIsEnumerable;
+var S = (m, i, r) => i in m ? A(m, i, { enumerable: !0, configurable: !0, writable: !0, value: r }) : m[i] = r, d = (m, i) => {
+  for (var r in i || (i = {}))
+    x.call(i, r) && S(m, r, i[r]);
+  if (b)
+    for (var r of b(i))
+      E.call(i, r) && S(m, r, i[r]);
+  return m;
+}, y = (m, i) => C(m, P(i));
+var v = (m, i, r) => new Promise((o, n) => {
+  var s = (a) => {
     try {
-      o(n.next(c));
-    } catch (g) {
-      i(g);
+      t(r.next(a));
+    } catch (h) {
+      n(h);
     }
-  }, e = (c) => {
+  }, e = (a) => {
     try {
-      o(n.throw(c));
-    } catch (g) {
-      i(g);
+      t(r.throw(a));
+    } catch (h) {
+      n(h);
     }
-  }, o = (c) => c.done ? r(c.value) : Promise.resolve(c.value).then(s, e);
-  o((n = n.apply(d, t)).next());
+  }, t = (a) => a.done ? o(a.value) : Promise.resolve(a.value).then(s, e);
+  t((r = r.apply(m, i)).next());
 });
-import { ChatMessageRole as I } from "@glodon-aiot/apis";
-import q from "./Logger.mjs";
-import { configuration as v } from "./Config.mjs";
-import { io as x } from "socket.io-client";
-import { UserMode as w } from "./PayloadTypes.mjs";
-import L, { SessionErrors as y } from "./Session.mjs";
-const u = new q("[AgentSession] ");
-class D extends L {
-  constructor(t, n = "", r, i, s, e) {
-    super(t, n, r, i, s, e);
+import { ChatMessageRole as w, PromptVariableType as _ } from "@glodon-aiot/apis";
+import L from "./Logger.mjs";
+import { configuration as u } from "./Config.mjs";
+import { io as q } from "socket.io-client";
+import { UserMode as k } from "./PayloadTypes.mjs";
+import N, { SessionErrors as I } from "./Session.mjs";
+import T from "dayjs";
+const p = new L("[AgentSession] ");
+class H extends N {
+  constructor(i, r = "", o, n, s, e) {
+    super(i, r, o, n, s, e);
   }
   prepareConnection() {
-    return new Promise((t) => p(this, null, function* () {
-      this.socket && t(!0);
-      const n = {
+    return new Promise((i) => v(this, null, function* () {
+      this.socket && i(!0);
+      const r = {
         version: "1.0",
-        stream: "false",
+        stream: JSON.stringify(u.stream),
         appId: this._applicationId,
         sessionId: this.id,
         Authorization: `${this.token}`,
-        userMode: v.debug ? w.Debug : w.Normal
-      }, r = new URL(v.apiRoot), i = yield x(r.origin, {
-        path: r.pathname.replace("api", "socketio") + "/copilot",
-        query: n,
+        userMode: u.debug ? k.Debug : k.Normal
+      }, o = new URL(u.apiRoot), n = yield q(o.origin, {
+        path: o.pathname.replace("api", "socketio") + "/copilot",
+        query: r,
         transports: ["websocket"]
       });
-      i.on("connect", () => {
-        u.log("socket:connect"), this.event.emit("socket:connect"), i.emit("init");
-      }), i.on("disconnect", () => {
-        u.log("socket:disconnect"), this.event.emit("socket:disconnect"), this._ready = !1;
-      }), i.on("ready", (s) => {
+      n.on("connect", () => {
+        p.log("socket:connect"), this.event.emit("socket:connect"), n.emit("init");
+      }), n.on("disconnect", () => {
+        p.log("socket:disconnect"), this.event.emit("socket:disconnect"), this._ready = !1;
+      }), n.on("ready", (s) => {
         if (s.code === 200) {
-          u.log("ready");
-          const e = v.debug ? w.Debug : w.Normal;
-          e === w.Debug && s.userMode, this._ready || (this._ready = !0, this.event.emit("ready")), this.frozen = !1, this.registerMcpFromClientConfig(), t(!0);
+          p.log("ready");
+          const e = u.debug ? k.Debug : k.Normal;
+          e === k.Debug && s.userMode, this._ready || (this._ready = !0, this.event.emit("ready")), this.frozen = !1, this.registerMcpFromClientConfig(), i(!0);
         } else {
           this._ready && (this._ready = !1);
           const e = new Error(
-            y.SocketConnectionFailed
+            I.SocketConnectionFailed
             // , {
             //   cause: data,
             // }
           );
-          u.error("error", e), this.event.emit("error", e), t(!1);
+          p.error("error", e), this.event.emit("error", e), i(!1);
         }
-      }), i.on("action", (s, e) => p(this, null, function* () {
+      }), n.on("action", (s, e) => v(this, null, function* () {
         if (this._stopAnswer || !this.id)
           return;
         const {
-          type: o,
+          type: t,
           data: {
-            runnerIsClient: c,
-            pluginCode: g,
-            funcCode: m,
-            parameters: a,
-            requestBody: N
+            runnerIsClient: a,
+            pluginCode: h,
+            funcCode: g,
+            parameters: c,
+            requestBody: U
           }
         } = s, f = this.messages[this.messages.length - 1];
         if (!f.messageId)
@@ -93,7 +94,7 @@ class D extends L {
           const l = {
             messageId: "",
             content: "",
-            role: I.Robot,
+            role: w.Robot,
             sessionId: this.id,
             reference: [],
             searchReference: [],
@@ -102,7 +103,8 @@ class D extends L {
             actions: [[s]],
             comment: {
               isAgree: 0
-            }
+            },
+            isReplying: !0
           };
           this.pushMessage(l), this.event.emit("message:action", {
             messageId: l.messageId,
@@ -111,7 +113,7 @@ class D extends L {
         }
         if (this.inactivating) {
           const l = new Error(
-            y.Inactive
+            I.Inactive
             // {
             //   cause: {
             //     code: -1,
@@ -120,41 +122,61 @@ class D extends L {
             //   },
             // }
           );
-          u.warn("error", l), this.event.emit("error", l);
-        } else if (o === "FUNCTION_CALL" && c) {
-          if (u.log(`calling function ${m} of plugin  ${g} with argument:
- ${JSON.stringify(a)}`), !v.plugins[g]) {
+          p.warn("error", l), this.event.emit("error", l);
+        } else if (t === "FUNCTION_CALL" && a) {
+          if (p.log(`calling function ${g} of plugin  ${h} with argument:
+ ${JSON.stringify(c)}`), !u.plugins[h]) {
             e({
               code: 404,
               message: "对不起，没有找到插件。"
             });
             return;
           }
-          const l = v.plugins[g][m];
+          const l = u.plugins[h][g];
           typeof l != "function" && e({
             code: 404,
             message: "对不起，没有找到插件方法。"
           });
-          let M = a == null ? void 0 : a.query;
+          let M = c == null ? void 0 : c.query;
           try {
-            M = JSON.parse(a == null ? void 0 : a.query);
-          } catch (_) {
+            M = JSON.parse(c == null ? void 0 : c.query);
+          } catch (R) {
           }
-          l(M, (..._) => {
-            this._stopAnswer || e(..._);
+          l(M, (...R) => {
+            this._stopAnswer || e(...R);
           });
         }
-      })), i.on("answer", (s) => {
+      })), n.on("answer", (s) => {
         if (!this._stopAnswer)
-          if (s.data.code !== 200 && (s.data.index = -1, this.event.emit("error", s.data)), s.data.index === -1) {
+          if (s.data.code !== 200 && (s.data.index = -1, this.event.emit("error", s.data)), s.data.stream || s.data.stream === void 0) {
+            let e = this.messages[this.messages.length - 1];
+            if (e && e.role === w.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 && this.event.emit("answer:finish", e);
+            else {
+              const t = {
+                content: "",
+                role: w.Robot,
+                reference: [],
+                searchReference: [],
+                messageId: "",
+                sessionId: this.id,
+                messageTime: T(/* @__PURE__ */ new Date()).format("YYYY-MM-DD HH:mm:ss"),
+                comment: {
+                  isAgree: 0
+                },
+                isReplying: !0
+              };
+              this.handleStreamAnswer(s.data, t), this.pushMessage(t), this.event.emit("message:content", t), t.index === -1 && this.event.emit("answer:finish", t);
+            }
+          } else if (s.data.index === -1) {
             let e = this.messages[this.messages.length - 1];
             if (e && !e.messageId)
-              e = h(h({}, e), s.data), this.event.emit("message:content", e), this.event.emit("message:updated", e), this.event.emit("answer:finish", e);
+              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 {
-              const o = h({
+              const t = d({
                 messageId: "",
                 content: "",
-                role: I.Robot,
+                role: w.Robot,
                 sessionId: this.id,
                 reference: [],
                 searchReference: [],
@@ -163,40 +185,55 @@ class D extends L {
                 actions: [],
                 comment: {
                   isAgree: 0
-                }
+                },
+                isReplying: !0
               }, s.data);
-              this.event.emit("message:content", o), this.pushMessage(o), this.event.emit("answer:finish", o);
+              this.event.emit("message:content", t), this.pushMessage(t), this.event.emit("answer:finish", t);
             }
           } else {
             let e = this.messages[this.messages.length - 1];
-            e && !e.messageId ? (e = h(h({}, e), s.data), this.messages[this.messages.length - 1] = e, this.event.emit("message:content", e), this.event.emit("message:updated", e)) : (this.pushMessage(s.data), this.event.emit("message:content", s.data));
+            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({
+              messageId: "",
+              content: "",
+              role: w.Robot,
+              sessionId: this.id,
+              reference: [],
+              searchReference: [],
+              isLoading: !1,
+              messageTime: "",
+              actions: [],
+              comment: {
+                isAgree: 0
+              },
+              isReplying: !0
+            }, s.data)), this.event.emit("message:content", s.data));
           }
-      }), i.on("reference", (s) => p(this, null, function* () {
+      }), n.on("reference", (s) => v(this, null, function* () {
         if (this._stopAnswer)
           return;
         if (!this.id) {
-          const a = new Error(
-            y.IdUndefined
+          const c = new Error(
+            I.IdUndefined
             // {
             //   cause: SESSION_ERROR_CAUSE[SessionErrors.IdUndefined],
             // }
           );
-          u.error(a), this.event.emit("error", a), this.frozen = !1;
+          p.error(c), this.event.emit("error", c), this.frozen = !1;
           return;
         }
         const {
           type: e,
-          data: o,
-          requestId: c
-        } = s, g = this.messages.findIndex((a) => a.messageId === c && a.role === I.User);
-        if (g < 0)
+          data: t,
+          requestId: a
+        } = s, h = this.messages.findIndex((c) => c.messageId === a && c.role === w.User);
+        if (h < 0)
           return;
-        const m = g + 1;
-        if (m >= this.messages.length) {
-          const a = {
+        const g = h + 1;
+        if (g >= this.messages.length) {
+          const c = {
             messageId: "",
             content: "",
-            role: I.Robot,
+            role: w.Robot,
             sessionId: this.id,
             reference: [],
             searchReference: [],
@@ -205,81 +242,85 @@ class D extends L {
             actions: [],
             comment: {
               isAgree: 0
-            }
+            },
+            isReplying: !0
           };
-          this.pushMessage(a);
+          this.pushMessage(c);
         }
-        e === "KG" ? (this.messages[m] = k(h({}, this.messages[m]), {
-          reference: o.knowledgeReference
-        }), yield this.fillMessageReference(this.messages[m]), this.event.emit("message:reference", o.knowledgeReference), this.event.emit("message:updated", this.messages[m])) : e === "WEB_SEARCH" && (this.messages[m] = k(h({}, this.messages[m]), {
-          searchReference: o.knowledgeReference
-        }), this.event.emit("message:searchreference", o.searchReference), this.event.emit("message:updated", this.messages[m]));
-      })), this.socket = i, i.open();
+        e === "KG" ? (this.messages[g] = y(d({}, this.messages[g]), {
+          reference: t.knowledgeReference
+        }), yield this.fillMessageReference(this.messages[g]), this.event.emit("message:reference", t.knowledgeReference), this.event.emit("message:updated", this.messages[g])) : e === "WEB_SEARCH" && (this.messages[g] = y(d({}, this.messages[g]), {
+          searchReference: t.knowledgeReference
+        }), this.event.emit("message:searchreference", t.searchReference), this.event.emit("message:updated", this.messages[g]));
+      })), this.socket = n, n.open();
     }));
   }
   registerMcpFromClientConfig() {
-    return v.mcp ? new Promise((t, n) => {
+    return u.mcp ? new Promise((i, r) => {
       if (!this.socket) {
-        n("socket is undefined");
+        r("socket is undefined");
         return;
       }
-      this.socket.emit("add_mcp_servers", v.mcp, (r) => {
-        r.code === 200 ? (Object.keys(r.data).forEach((i) => {
-          r.data[i].code !== 200 && console.error(i, "MCP Server 注册失败。", r.data[i]);
-        }), t(!0)) : (console.error("MCP Server 注册失败。", r), n(r));
+      this.socket.emit("add_mcp_servers", u.mcp, (o) => {
+        o.code === 200 ? (Object.keys(o.data).forEach((n) => {
+          o.data[n].code !== 200 && console.error(n, "MCP Server 注册失败。", o.data[n]);
+        }), i(!0)) : (console.error("MCP Server 注册失败。", o), r(o));
       });
     }) : Promise.resolve(!0);
   }
-  addMcpServers(t) {
-    return p(this, null, function* () {
-      return this.ready ? this.socket ? new Promise((n, r) => {
-        this.socket && this.socket.emit("add_mcp_servers", t, (i) => {
-          i.code === 200 ? (Object.keys(i.data).forEach((s) => {
-            i.data[s].code !== 200 && console.error(s, "MCP Server 注册失败。", i.data[s]);
-          }), n(!0)) : (console.error("MCP Server 注册失败。", i), r(i));
+  addMcpServers(i) {
+    return v(this, null, function* () {
+      return this.ready ? this.socket ? new Promise((r, o) => {
+        this.socket && this.socket.emit("add_mcp_servers", i, (n) => {
+          n.code === 200 ? (Object.keys(n.data).forEach((s) => {
+            n.data[s].code !== 200 && console.error(s, "MCP Server 注册失败。", n.data[s]);
+          }), r(!0)) : (console.error("MCP Server 注册失败。", n), o(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, n) => {
-        this.socket && this.socket.emit("list_mcp_servers", (r) => {
-          r.code === 200 ? t(r.data) : n(r);
+    return v(this, null, function* () {
+      return this.ready ? this.socket ? new Promise((i, r) => {
+        this.socket && this.socket.emit("list_mcp_servers", (o) => {
+          o.code === 200 ? i(o.data) : r(o);
         });
       }) : (console.error(" socket is undefined."), Promise.reject("socket is undefined")) : (console.error("session is not avalible."), Promise.reject("session is not avalible"));
     });
   }
-  sendQuestion(t, n) {
-    return p(this, null, function* () {
-      var i, s, e;
-      let r = k(h({}, t), {
-        promptVariables: this.promptVariables.reduce((o, c) => k(h({}, o), {
-          [c.key]: c.value
+  sendQuestion(i, r) {
+    return v(this, null, function* () {
+      var n, s, e;
+      let o = y(d({}, i), {
+        promptVariables: this.promptVariables.filter((t) => t.type === _.User).reduce((t, a) => y(d({}, t), {
+          [a.key]: a.value
+        }), {}),
+        systemPromptVariables: this.promptVariables.filter((t) => t.type === _.System).reduce((t, a) => y(d({}, t), {
+          [a.key]: a.value
         }), {}),
-        knowledges: (i = this.knowledges) == null ? void 0 : i.map((o) => o.id)
+        knowledges: (n = this.knowledges) == null ? void 0 : n.map((t) => t.id)
       });
       if (!this.active) {
         if (yield this.activate(), !this.socket) {
-          const o = new Error(
-            y.SocketConnectionFailed
+          const t = new Error(
+            I.SocketConnectionFailed
             // {
             //   cause: { code: -1, message: 'socket is null' },
             // }
           );
-          return u.warn("error", o), this.event.emit("error", o), !1;
+          return p.warn("error", t), this.event.emit("error", t), !1;
         }
-        (s = this.socket) == null || s.emit("question", r);
+        (s = this.socket) == null || s.emit("question", o);
         return;
       }
-      (e = this.socket) == null || e.emit("question", r);
+      (e = this.socket) == null || e.emit("question", o);
     });
   }
   stopAnswer() {
-    var t;
-    (t = this.socket) == null || t.emit("stop_answer");
+    var i;
+    (i = this.socket) == null || i.emit("stop_answer");
   }
 }
 export {
-  D as default
+  H as default
 };