@glodon-aiot/bot-client-sdk 3.3.1 → 3.3.2-beta.1

package/dist/es/AgentSession.mjs

@@ -1,66 +1,67 @@
-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) => {
+var S = Object.defineProperty, A = Object.defineProperties;
+var b = Object.getOwnPropertyDescriptors;
+var _ = Object.getOwnPropertySymbols;
+var P = Object.prototype.hasOwnProperty, x = Object.prototype.propertyIsEnumerable;
+var C = (m, t, n) => t in m ? S(m, t, { enumerable: !0, configurable: !0, writable: !0, value: n }) : m[t] = n, g = (m, 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) => {
+    P.call(t, n) && C(m, n, t[n]);
+  if (_)
+    for (var n of _(t))
+      x.call(t, n) && C(m, n, t[n]);
+  return m;
+}, w = (m, t) => A(m, b(t));
+var v = (m, t, n) => new Promise((o, i) => {
   var s = (c) => {
     try {
-      o(n.next(c));
-    } catch (g) {
-      i(g);
+      r(n.next(c));
+    } catch (h) {
+      i(h);
     }
   }, e = (c) => {
     try {
-      o(n.throw(c));
-    } catch (g) {
-      i(g);
+      r(n.throw(c));
+    } catch (h) {
+      i(h);
     }
-  }, o = (c) => c.done ? r(c.value) : Promise.resolve(c.value).then(s, e);
-  o((n = n.apply(d, t)).next());
+  }, r = (c) => c.done ? o(c.value) : Promise.resolve(c.value).then(s, e);
+  r((n = n.apply(m, t)).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 { ChatMessageRole as k } from "@glodon-aiot/apis";
+import E from "./Logger.mjs";
+import { configuration as u } from "./Config.mjs";
+import { io as q } from "socket.io-client";
+import { UserMode as M } 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 N from "dayjs";
+const p = new E("[AgentSession] ");
+class H extends L {
+  constructor(t, n = "", o, i, s, e) {
+    super(t, n, o, i, s, e);
   }
   prepareConnection() {
-    return new Promise((t) => p(this, null, function* () {
+    return new Promise((t) => v(this, null, function* () {
       this.socket && t(!0);
       const n = {
         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",
+        userMode: u.debug ? M.Debug : M.Normal
+      }, o = new URL(u.apiRoot), i = yield q(o.origin, {
+        path: o.pathname.replace("api", "socketio") + "/copilot",
         query: n,
         transports: ["websocket"]
       });
       i.on("connect", () => {
-        u.log("socket:connect"), this.event.emit("socket:connect"), i.emit("init");
+        p.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;
+        p.log("socket:disconnect"), this.event.emit("socket:disconnect"), this._ready = !1;
       }), i.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 ? M.Debug : M.Normal;
+          e === M.Debug && s.userMode, this._ready || (this._ready = !0, this.event.emit("ready")), this.frozen = !1, this.registerMcpFromClientConfig(), t(!0);
         } else {
           this._ready && (this._ready = !1);
           const e = new Error(
@@ -69,19 +70,19 @@ class D extends L {
             //   cause: data,
             // }
           );
-          u.error("error", e), this.event.emit("error", e), t(!1);
+          p.error("error", e), this.event.emit("error", e), t(!1);
         }
-      }), i.on("action", (s, e) => p(this, null, function* () {
+      }), i.on("action", (s, e) => v(this, null, function* () {
         if (this._stopAnswer || !this.id)
           return;
         const {
-          type: o,
+          type: r,
           data: {
             runnerIsClient: c,
-            pluginCode: g,
-            funcCode: m,
+            pluginCode: h,
+            funcCode: d,
             parameters: a,
-            requestBody: N
+            requestBody: j
           }
         } = 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: k.Robot,
             sessionId: this.id,
             reference: [],
             searchReference: [],
@@ -120,41 +121,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 (r === "FUNCTION_CALL" && c) {
+          if (p.log(`calling function ${d} of plugin  ${h} with argument:
+ ${JSON.stringify(a)}`), !u.plugins[h]) {
             e({
               code: 404,
               message: "对不起，没有找到插件。"
             });
             return;
           }
-          const l = v.plugins[g][m];
+          const l = u.plugins[h][d];
           typeof l != "function" && e({
             code: 404,
             message: "对不起，没有找到插件方法。"
           });
-          let M = a == null ? void 0 : a.query;
+          let I = a == null ? void 0 : a.query;
           try {
-            M = JSON.parse(a == null ? void 0 : a.query);
-          } catch (_) {
+            I = JSON.parse(a == null ? void 0 : a.query);
+          } catch (R) {
           }
-          l(M, (..._) => {
-            this._stopAnswer || e(..._);
+          l(I, (...R) => {
+            this._stopAnswer || e(...R);
           });
         }
       })), i.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), s.data.stream || s.data.stream === void 0) {
+            let e = this.messages[this.messages.length - 1];
+            if (e && e.role === k.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 r = {
+                content: "",
+                role: 1,
+                reference: [],
+                searchReference: [],
+                messageId: "",
+                sessionId: this.id,
+                messageTime: N(/* @__PURE__ */ new Date()).format("YYYY-MM-DD HH:mm:ss"),
+                comment: {
+                  isAgree: 0
+                },
+                isReplying: !0
+              };
+              this.handleStreamAnswer(s.data, r), this.pushMessage(r), this.event.emit("message:content", r), e.index === -1 && this.event.emit("answer:finish", e);
+            }
+          } 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 = g(g({}, 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 r = g({
                 messageId: "",
                 content: "",
-                role: I.Robot,
+                role: k.Robot,
                 sessionId: this.id,
                 reference: [],
                 searchReference: [],
@@ -165,13 +186,13 @@ class D extends L {
                   isAgree: 0
                 }
               }, s.data);
-              this.event.emit("message:content", o), this.pushMessage(o), this.event.emit("answer:finish", o);
+              this.event.emit("message:content", r), this.pushMessage(r), this.event.emit("answer:finish", r);
             }
           } 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 = g(g({}, 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));
           }
-      }), i.on("reference", (s) => p(this, null, function* () {
+      }), i.on("reference", (s) => v(this, null, function* () {
         if (this._stopAnswer)
           return;
         if (!this.id) {
@@ -181,22 +202,22 @@ class D extends L {
             //   cause: SESSION_ERROR_CAUSE[SessionErrors.IdUndefined],
             // }
           );
-          u.error(a), this.event.emit("error", a), this.frozen = !1;
+          p.error(a), this.event.emit("error", a), this.frozen = !1;
           return;
         }
         const {
           type: e,
-          data: o,
+          data: r,
           requestId: c
-        } = s, g = this.messages.findIndex((a) => a.messageId === c && a.role === I.User);
-        if (g < 0)
+        } = s, h = this.messages.findIndex((a) => a.messageId === c && a.role === k.User);
+        if (h < 0)
           return;
-        const m = g + 1;
-        if (m >= this.messages.length) {
+        const d = h + 1;
+        if (d >= this.messages.length) {
           const a = {
             messageId: "",
             content: "",
-            role: I.Robot,
+            role: k.Robot,
             sessionId: this.id,
             reference: [],
             searchReference: [],
@@ -209,70 +230,70 @@ class D extends L {
           };
           this.pushMessage(a);
         }
-        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]));
+        e === "KG" ? (this.messages[d] = w(g({}, this.messages[d]), {
+          reference: r.knowledgeReference
+        }), yield this.fillMessageReference(this.messages[d]), this.event.emit("message:reference", r.knowledgeReference), this.event.emit("message:updated", this.messages[d])) : e === "WEB_SEARCH" && (this.messages[d] = w(g({}, this.messages[d]), {
+          searchReference: r.knowledgeReference
+        }), this.event.emit("message:searchreference", r.searchReference), this.event.emit("message:updated", this.messages[d]));
       })), this.socket = i, i.open();
     }));
   }
   registerMcpFromClientConfig() {
-    return v.mcp ? new Promise((t, n) => {
+    return u.mcp ? new Promise((t, n) => {
       if (!this.socket) {
         n("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((i) => {
+          o.data[i].code !== 200 && console.error(i, "MCP Server 注册失败。", o.data[i]);
+        }), t(!0)) : (console.error("MCP Server 注册失败。", o), n(o));
       });
     }) : Promise.resolve(!0);
   }
   addMcpServers(t) {
-    return p(this, null, function* () {
-      return this.ready ? this.socket ? new Promise((n, r) => {
+    return v(this, null, function* () {
+      return this.ready ? this.socket ? new Promise((n, o) => {
         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));
+          }), n(!0)) : (console.error("MCP Server 注册失败。", i), o(i));
         });
       }) : (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 v(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);
+        this.socket && this.socket.emit("list_mcp_servers", (o) => {
+          o.code === 200 ? t(o.data) : n(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* () {
+    return v(this, null, function* () {
       var i, s, e;
-      let r = k(h({}, t), {
-        promptVariables: this.promptVariables.reduce((o, c) => k(h({}, o), {
+      let o = w(g({}, t), {
+        promptVariables: this.promptVariables.reduce((r, c) => w(g({}, r), {
           [c.key]: c.value
         }), {}),
-        knowledges: (i = this.knowledges) == null ? void 0 : i.map((o) => o.id)
+        knowledges: (i = this.knowledges) == null ? void 0 : i.map((r) => r.id)
       });
       if (!this.active) {
         if (yield this.activate(), !this.socket) {
-          const o = new Error(
+          const r = new Error(
             y.SocketConnectionFailed
             // {
             //   cause: { code: -1, message: 'socket is null' },
             // }
           );
-          return u.warn("error", o), this.event.emit("error", o), !1;
+          return p.warn("error", r), this.event.emit("error", r), !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() {
@@ -281,5 +302,5 @@ class D extends L {
   }
 }
 export {
-  D as default
+  H as default
 };

package/dist/es/Config.mjs

@@ -1,15 +1,15 @@
-var v = Object.defineProperty, s = Object.defineProperties;
-var u = Object.getOwnPropertyDescriptors;
+var s = Object.defineProperty, u = Object.defineProperties;
+var v = Object.getOwnPropertyDescriptors;
 var n = Object.getOwnPropertySymbols;
 var p = Object.prototype.hasOwnProperty, h = Object.prototype.propertyIsEnumerable;
-var g = (o, t, e) => t in o ? v(o, t, { enumerable: !0, configurable: !0, writable: !0, value: e }) : o[t] = e, i = (o, t) => {
+var g = (o, t, e) => t in o ? s(o, t, { enumerable: !0, configurable: !0, writable: !0, value: e }) : o[t] = e, i = (o, t) => {
   for (var e in t || (t = {}))
     p.call(t, e) && g(o, e, t[e]);
   if (n)
     for (var e of n(t))
       h.call(t, e) && g(o, e, t[e]);
   return o;
-}, l = (o, t) => s(o, u(t));
+}, l = (o, t) => u(o, v(t));
 import d from "./Logger.mjs";
 const R = new d("[Config] ");
 var f = /* @__PURE__ */ ((o) => (o.dev = "https://aiot-dev.glodon.com/api/cvforcepd", o.test = "https://aiot-dev.glodon.com/api/cvforce", o.prod = "https://copilot.glodon.com/api/cvforce", o.local = "http://localhost:3000/api/cvforcepd", o))(f || {});
@@ -22,9 +22,10 @@ let c = {
   applicationId: "",
   plugins: {},
   mcp: void 0,
-  loadRelatedQuesions: !0
+  loadRelatedQuesions: !0,
+  stream: !0
 };
-function x(o) {
+function w(o) {
   d.debug = (o == null ? void 0 : o.debug) || !1, c = l(i(i({}, c), o), {
     apiRoot: (o == null ? void 0 : o.apiRoot) || c.apiRoot || (o != null && o.env ? f[o.env || "prod"] : c.apiRoot)
   }), d.debug && R.log("updated", c);
@@ -32,5 +33,5 @@ function x(o) {
 export {
   f as ApiUrl,
   c as configuration,
-  x as configure
+  w as configure
 };