@glodon-aiot/bot-client-sdk 3.7.0-alpha.2 → 3.8.0-alpha.4

package/dist/es/AgentSession.mjs

@@ -1,88 +1,88 @@
-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 = (d, i, r) => i in d ? A(d, i, { enumerable: !0, configurable: !0, writable: !0, value: r }) : d[i] = r, m = (d, i) => {
-  for (var r in i || (i = {}))
-    x.call(i, r) && S(d, r, i[r]);
-  if (b)
-    for (var r of b(i))
-      E.call(i, r) && S(d, r, i[r]);
-  return d;
-}, w = (d, i) => C(d, P(i));
-var v = (d, i, r) => new Promise((o, n) => {
+var A = Object.defineProperty, P = Object.defineProperties;
+var b = Object.getOwnPropertyDescriptors;
+var S = Object.getOwnPropertySymbols;
+var T = Object.prototype.hasOwnProperty, x = Object.prototype.propertyIsEnumerable;
+var C = (m, t, i) => t in m ? A(m, t, { enumerable: !0, configurable: !0, writable: !0, value: i }) : m[t] = i, d = (m, t) => {
+  for (var i in t || (t = {}))
+    T.call(t, i) && C(m, i, t[i]);
+  if (S)
+    for (var i of S(t))
+      x.call(t, i) && C(m, i, t[i]);
+  return m;
+}, w = (m, t) => P(m, b(t));
+var v = (m, t, i) => new Promise((o, r) => {
   var s = (a) => {
     try {
-      t(r.next(a));
-    } catch (h) {
-      n(h);
+      n(i.next(a));
+    } catch (g) {
+      r(g);
     }
   }, e = (a) => {
     try {
-      t(r.throw(a));
-    } catch (h) {
-      n(h);
+      n(i.throw(a));
+    } catch (g) {
+      r(g);
     }
-  }, t = (a) => a.done ? o(a.value) : Promise.resolve(a.value).then(s, e);
-  t((r = r.apply(d, i)).next());
+  }, n = (a) => a.done ? o(a.value) : Promise.resolve(a.value).then(s, e);
+  n((i = i.apply(m, t)).next());
 });
 import { ChatMessageRole as y, PromptVariableType as _ } from "@glodon-aiot/apis";
-import L from "./Logger.mjs";
+import E from "./Logger.mjs";
 import { configuration as u } from "./Config.mjs";
-import { io as T } from "socket.io-client";
-import { UserMode as k } from "./PayloadTypes.mjs";
-import q, { SessionErrors as I } from "./Session.mjs";
-import N from "dayjs";
-const p = new L("[AgentSession] ");
+import { io as L } from "socket.io-client";
+import { UserMode as I } from "./PayloadTypes.mjs";
+import q, { SessionErrors as M } from "./Session.mjs";
+import F from "dayjs";
+const p = new E("[AgentSession] ");
 class H extends q {
-  constructor(i, r = "", o, n, s, e) {
-    super(i, r, o, n, s, e);
+  constructor(t, i = "", o, r, s, e) {
+    super(t, i, o, r, s, e);
   }
   prepareConnection() {
-    return new Promise((i) => v(this, null, function* () {
-      this.socket && i(!0);
-      const r = {
+    return new Promise((t) => v(this, null, function* () {
+      this.socket && t(!0);
+      const i = {
         version: "1.0",
         stream: JSON.stringify(u.stream),
         appId: this._applicationId,
         sessionId: this.id,
         Authorization: `${this.token}`,
-        userMode: u.debug ? k.Debug : k.Normal
-      }, o = new URL(u.apiRoot), n = yield T(o.origin, {
+        userMode: u.debug ? I.Debug : I.Normal
+      }, o = new URL(u.apiRoot), r = yield L(o.origin, {
         path: o.pathname.replace("api", "socketio") + "/copilot",
-        query: r,
+        query: i,
         transports: ["websocket"]
       });
-      n.on("connect", () => {
-        p.log("socket:connect"), this.event.emit("socket:connect"), n.emit("init");
-      }), n.on("disconnect", () => {
+      r.on("connect", () => {
+        p.log("socket:connect"), this.event.emit("socket:connect"), r.emit("init");
+      }), r.on("disconnect", () => {
         p.log("socket:disconnect"), this.event.emit("socket:disconnect"), this._ready = !1;
-      }), n.on("ready", (s) => {
+      }), r.on("ready", (s) => {
         if (s.code === 200) {
           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);
+          const e = u.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);
         } else {
           this._ready && (this._ready = !1);
           const e = new Error(
-            I.SocketConnectionFailed
+            M.SocketConnectionFailed
             // , {
             //   cause: data,
             // }
           );
-          p.error("error", e), this.event.emit("error", e), i(!1);
+          p.error("error", e), this.event.emit("error", e), t(!1);
         }
-      }), n.on("action", (s, e) => v(this, null, function* () {
+      }), r.on("action", (s, e) => v(this, null, function* () {
         if (this._stopAnswer || !this.id)
           return;
         const {
-          type: t,
+          type: n,
           data: {
             runnerIsClient: a,
-            pluginCode: h,
-            funcCode: g,
+            pluginCode: g,
+            funcCode: h,
             parameters: c,
-            requestBody: U
+            requestBody: N
           }
         } = s, l = this.messages[this.messages.length - 1];
         if (!l.messageId)
@@ -113,7 +113,7 @@ class H extends q {
         }
         if (this.inactivating) {
           const f = new Error(
-            I.Inactive
+            M.Inactive
             // {
             //   cause: {
             //     code: -1,
@@ -123,57 +123,57 @@ class H extends q {
             // }
           );
           p.warn("error", f), this.event.emit("error", f);
-        } else if (t === "FUNCTION_CALL" && a) {
-          if (p.log(`calling function ${g} of plugin  ${h} with argument:
- ${JSON.stringify(c)}`), !u.plugins[h]) {
+        } else if (n === "FUNCTION_CALL" && a) {
+          if (p.log(`calling function ${h} of plugin  ${g} with argument:
+ ${JSON.stringify(c)}`), !u.plugins[g]) {
             e({
               code: 404,
               message: "对不起，没有找到插件。"
             });
             return;
           }
-          const f = u.plugins[h][g];
+          const f = u.plugins[g][h];
           typeof f != "function" && e({
             code: 404,
             message: "对不起，没有找到插件方法。"
           });
-          let M = c == null ? void 0 : c.query;
+          let R = c == null ? void 0 : c.query;
           try {
-            M = JSON.parse(c == null ? void 0 : c.query);
-          } catch (R) {
+            R = JSON.parse(c == null ? void 0 : c.query);
+          } catch (k) {
           }
-          f(M, (...R) => {
-            this._stopAnswer || e(...R);
+          f(R, (...k) => {
+            this._stopAnswer || e(...k);
           });
         }
-      })), n.on("answer", (s) => {
+      })), r.on("answer", (s) => {
         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 === y.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));
             else {
-              const t = {
+              const n = {
                 content: "",
                 role: y.Robot,
                 reference: [],
                 searchReference: [],
                 messageId: "",
                 sessionId: this.id,
-                messageTime: N(/* @__PURE__ */ new Date()).format("YYYY-MM-DD HH:mm:ss"),
+                messageTime: F(/* @__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 && (t.isReplying = !1, this.event.emit("answer:finish", t));
+              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));
             }
-          } else if (s.data.updateSession && this.data && (this.data = m(m({}, this.data), s.data.updateSession), this.event.emit("data:updated", this.data)), s.data.index === -1) {
+          } 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 = m(m({}, 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 t = m({
+              const n = d({
                 messageId: "",
                 content: "",
                 role: y.Robot,
@@ -188,11 +188,11 @@ class H extends q {
                 },
                 isReplying: !0
               }, s.data);
-              this.event.emit("message:content", t), this.pushMessage(t), this.event.emit("answer:finish", t);
+              this.event.emit("message:content", n), this.pushMessage(n), this.event.emit("answer:finish", n);
             }
           } else {
             let e = this.messages[this.messages.length - 1];
-            e && !e.messageId ? (e = m(m({}, e), s.data), this.messages[this.messages.length - 1] = e, this.event.emit("message:content", e), this.event.emit("message:updated", e)) : (this.pushMessage(m({
+            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: y.Robot,
@@ -208,12 +208,12 @@ class H extends q {
               isReplying: !0
             }, s.data)), this.event.emit("message:content", s.data));
           }
-      }), n.on("reference", (s) => v(this, null, function* () {
+      }), r.on("reference", (s) => v(this, null, function* () {
         if (this._stopAnswer)
           return;
         if (!this.id) {
           const c = new Error(
-            I.IdUndefined
+            M.IdUndefined
             // {
             //   cause: SESSION_ERROR_CAUSE[SessionErrors.IdUndefined],
             // }
@@ -223,13 +223,13 @@ class H extends q {
         }
         const {
           type: e,
-          data: t,
+          data: n,
           requestId: a
-        } = s, h = this.messages.findIndex((c) => c.messageId === a && c.role === y.User);
-        if (h < 0)
+        } = s, g = this.messages.findIndex((c) => c.messageId === a && c.role === y.User);
+        if (g < 0)
           return;
-        const g = h + 1;
-        if (g >= this.messages.length) {
+        const h = g + 1;
+        if (h >= this.messages.length) {
           const c = {
             messageId: "",
             content: "",
@@ -247,70 +247,73 @@ class H extends q {
           };
           this.pushMessage(c);
         }
-        e === "KG" ? (this.messages[g] = w(m({}, 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] = w(m({}, 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();
+        e === "KG" ? (this.messages[h] = w(d({}, this.messages[h]), {
+          reference: n.knowledgeReference
+        }), yield this.fillMessageReference(this.messages[h]), this.event.emit("message:reference", n.knowledgeReference), this.event.emit("message:updated", this.messages[h])) : e === "WEB_SEARCH" && (this.messages[h] = w(d({}, this.messages[h]), {
+          searchReference: n.knowledgeReference
+        }), this.event.emit("message:searchreference", n.searchReference), this.event.emit("message:updated", this.messages[h]));
+      })), this.socket = r, r.open();
     }));
   }
   registerMcpFromClientConfig() {
-    return u.mcp ? new Promise((i, r) => {
+    return u.mcp ? new Promise((t, i) => {
       if (!this.socket) {
-        r("socket is undefined");
+        i("socket is undefined");
         return;
       }
       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));
+        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));
       });
     }) : Promise.resolve(!0);
   }
-  addMcpServers(i) {
+  addMcpServers(t) {
     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));
+      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));
         });
       }) : (console.error(" socket is undefined."), Promise.reject("socket is undefined")) : (console.error("session is not avalible."), Promise.reject("session is not avalible"));
     });
   }
   listMcpServers() {
     return v(this, null, function* () {
-      return this.ready ? this.socket ? new Promise((i, r) => {
+      return this.ready ? this.socket ? new Promise((t, i) => {
         this.socket && this.socket.emit("list_mcp_servers", (o) => {
-          o.code === 200 ? i(o.data) : r(o);
+          o.code === 200 ? t(o.data) : i(o);
         });
       }) : (console.error(" socket is undefined."), Promise.reject("socket is undefined")) : (console.error("session is not avalible."), Promise.reject("session is not avalible"));
     });
   }
-  sendQuestion(i, r) {
+  sendQuestion(t, i) {
     return v(this, null, function* () {
-      var n, s, e;
-      let o = w(m({}, i), {
-        promptVariables: this.promptVariables.filter((t) => t.type === _.User).reduce((t, a) => w(m({}, t), {
+      var r, s, e;
+      this.convertKnowledgesToChatFormat(this.knowledges);
+      let o = w(d({}, t), {
+        promptVariables: this.promptVariables.filter((n) => n.type === _.User).reduce((n, a) => w(d({}, n), {
           [a.key]: a.value
         }), {}),
-        systemPromptVariables: this.promptVariables.filter((t) => t.type === _.System).reduce((t, a) => w(m({}, t), {
+        systemPromptVariables: this.promptVariables.filter((n) => n.type === _.System).reduce((n, a) => w(d({}, n), {
           [a.key]: a.value
         }), {}),
-        knowledges: (n = this.knowledges) == null ? void 0 : n.map((t) => t.id),
+        knowledges: (r = this.knowledges) == null ? void 0 : r.map((n) => n.id),
+        //agent支持指定文件时再放开
+        // knowledges: knowledgesForChat,
         //开启内容摘要
         enableTitleSummary: !0
       });
       if (!this.active) {
         if (yield this.activate(), !this.socket) {
-          const t = new Error(
-            I.SocketConnectionFailed
+          const n = new Error(
+            M.SocketConnectionFailed
             // {
             //   cause: { code: -1, message: 'socket is null' },
             // }
           );
-          return p.warn("error", t), this.event.emit("error", t), !1;
+          return p.warn("error", n), this.event.emit("error", n), !1;
         }
         (s = this.socket) == null || s.emit("question", o);
         return;
@@ -319,8 +322,14 @@ class H extends q {
     });
   }
   stopAnswer() {
-    var i;
-    (i = this.socket) == null || i.emit("stop_answer");
+    var t;
+    (t = this.socket) == null || t.emit("stop_answer");
+  }
+  convertKnowledgesToChatFormat(t) {
+    return t ? t.map((i) => i != null && i.fileIds ? {
+      id: i.id,
+      fileIds: i.fileIds
+    } : i.id) : [];
   }
 }
 export {