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

package/dist/es/DialogSession.mjs

@@ -1,57 +1,57 @@
-var T = Object.defineProperty, A = Object.defineProperties;
-var I = Object.getOwnPropertyDescriptors;
-var M = Object.getOwnPropertySymbols;
-var P = Object.prototype.hasOwnProperty, k = Object.prototype.propertyIsEnumerable;
-var R = (m, n, a) => n in m ? T(m, n, { enumerable: !0, configurable: !0, writable: !0, value: a }) : m[n] = a, u = (m, n) => {
-  for (var a in n || (n = {}))
-    P.call(n, a) && R(m, a, n[a]);
-  if (M)
-    for (var a of M(n))
-      k.call(n, a) && R(m, a, n[a]);
-  return m;
-}, y = (m, n) => A(m, I(n));
-var w = (m, n, a) => (R(m, typeof n != "symbol" ? n + "" : n, a), a);
-var f = (m, n, a) => new Promise((l, d) => {
+var I = Object.defineProperty, A = Object.defineProperties;
+var P = Object.getOwnPropertyDescriptors;
+var w = Object.getOwnPropertySymbols;
+var b = Object.prototype.hasOwnProperty, D = Object.prototype.propertyIsEnumerable;
+var R = (l, c, s) => c in l ? I(l, c, { enumerable: !0, configurable: !0, writable: !0, value: s }) : l[c] = s, u = (l, c) => {
+  for (var s in c || (c = {}))
+    b.call(c, s) && R(l, s, c[s]);
+  if (w)
+    for (var s of w(c))
+      D.call(c, s) && R(l, s, c[s]);
+  return l;
+}, y = (l, c) => A(l, P(c));
+var M = (l, c, s) => (R(l, typeof c != "symbol" ? c + "" : c, s), s);
+var f = (l, c, s) => new Promise((a, d) => {
   var h = (r) => {
     try {
-      i(a.next(r));
-    } catch (o) {
-      d(o);
+      i(s.next(r));
+    } catch (m) {
+      d(m);
     }
   }, e = (r) => {
     try {
-      i(a.throw(r));
-    } catch (o) {
-      d(o);
+      i(s.throw(r));
+    } catch (m) {
+      d(m);
     }
-  }, i = (r) => r.done ? l(r.value) : Promise.resolve(r.value).then(h, e);
-  i((a = a.apply(m, n)).next());
+  }, i = (r) => r.done ? a(r.value) : Promise.resolve(r.value).then(h, e);
+  i((s = s.apply(l, c)).next());
 });
-import { IHistoryRole as b, PromptVariableType as C, ChatMessageRole as D } from "@glodon-aiot/apis";
-import O from "./Logger.mjs";
-import V from "./Session.mjs";
-import $ from "dayjs";
+import { IHistoryRole as C, PromptVariableType as v, ChatMessageRole as O } from "@glodon-aiot/apis";
+import V from "./Logger.mjs";
+import $ from "./Session.mjs";
+import _ from "dayjs";
 import { configuration as S } from "./Config.mjs";
-import { fetchEventSource as p } from "./api/fetchEventSource/fetch.mjs";
-const v = new O("[DialogSession] ");
-class B extends V {
-  constructor(a, l = "", d, h, e, i) {
-    super(a, l, d, h, e, i);
-    w(this, "ctrl", new AbortController());
+import { fetchEventSource as T } from "./api/fetchEventSource/fetch.mjs";
+const p = new V("[DialogSession] ");
+class k extends $ {
+  constructor(s, a = "", d, h, e, i) {
+    super(s, a, d, h, e, i);
+    M(this, "ctrl", new AbortController());
   }
   prepareConnection() {
-    return new Promise((a) => f(this, null, function* () {
-      this._ready = !0, this.event.emit("ready"), this.frozen = !1, a(!0);
+    return new Promise((s) => f(this, null, function* () {
+      this._ready = !0, this.event.emit("ready"), this.frozen = !1, s(!0);
     }));
   }
-  postChatV2EventSource(a, l, d, h, e, i) {
+  postChatV2EventSource(s, a, d, h, e, i) {
     return f(this, null, function* () {
       const r = {
         code: 0,
         // messageId: uuidv4(),
         content: "",
-        role: b.Robot,
-        sessionId: l,
+        role: C.Robot,
+        sessionId: a,
         reference: [],
         searchReference: [],
         messageTime: "",
@@ -60,16 +60,16 @@ class B extends V {
           isAgree: 0
         },
         isReplying: !1
-      }, o = y(u({}, d), {
+      }, m = y(u({}, d), {
         stream: S.stream
       });
-      p(`${S.apiRoot}/chat/v2/chat?applicationId=${a}&sessionId=${l}`, {
+      T(`${S.apiRoot}/chat/v2/chat?applicationId=${s}&sessionId=${a}`, {
         method: "POST",
         headers: {
           Authorization: `Bearer ${this.token}`,
           "Content-Type": "application/json"
         },
-        body: JSON.stringify(o),
+        body: JSON.stringify(m),
         openWhenHidden: !0,
         signal: h,
         onopen(g) {
@@ -77,41 +77,41 @@ class B extends V {
         },
         onmessage(g) {
           if (!g.data.includes('{"code": ')) {
-            let s = {};
+            let t = {};
             try {
-              s = JSON.parse(g.data);
-            } catch (_) {
+              t = JSON.parse(g.data);
+            } catch (J) {
             }
-            (s == null ? void 0 : s.mtype) === "search" ? r.searchReference.push(s) : r.content = r.content + g.data;
-            const c = {
+            (t == null ? void 0 : t.mtype) === "search" ? r.searchReference.push(t) : r.content = r.content + g.data;
+            const o = {
               content: r.content,
               role: 1,
               reference: r.reference,
               searchReference: r.searchReference,
               messageId: r.messageId,
-              sessionId: l,
-              messageTime: $(/* @__PURE__ */ new Date()).format("YYYY-MM-DD HH:mm:ss"),
+              sessionId: a,
+              messageTime: _(/* @__PURE__ */ new Date()).format("YYYY-MM-DD HH:mm:ss"),
               comment: {
                 isAgree: 0
               },
               isReplying: !0
             };
-            return (e == null ? void 0 : e.onmessage) && e.onmessage(c);
+            return (e == null ? void 0 : e.onmessage) && e.onmessage(o);
           }
-          const t = JSON.parse(g.data);
-          if (r.reference = t.data.reference, r.searchReference = t.data.searchReference, r.messageTime = t.data.messageTime, r.messageId = t.data.messageId, r.llmModel = t.data.llmModel, r.content = t.data.content, r.code = t.code, t.code === 200) {
-            const s = {
+          const n = JSON.parse(g.data);
+          if (r.reference = n.data.reference, r.searchReference = n.data.searchReference, r.messageTime = n.data.messageTime, r.messageId = n.data.messageId, r.llmModel = n.data.llmModel, r.content = n.data.content, r.code = n.code, n.code === 200) {
+            const t = {
               code: r.code,
               content: r.content,
               role: 1,
               reference: r.reference,
               searchReference: r.searchReference,
               messageId: r.messageId,
-              sessionId: l,
+              sessionId: a,
               messageTime: r.messageTime,
               isReplying: !0
             };
-            e != null && e.onmessage && e.onmessage(s);
+            e != null && e.onmessage && e.onmessage(t);
           } else
             return e != null && e.onclose && e.onclose(r), i && i({
               code: r.code,
@@ -129,7 +129,7 @@ class B extends V {
             reference: r.reference,
             searchReference: r.searchReference,
             messageId: r.messageId,
-            sessionId: l,
+            sessionId: a,
             messageTime: r.messageTime,
             comment: {
               isAgree: 0
@@ -142,14 +142,14 @@ class B extends V {
       });
     });
   }
-  postChatV3EventSource(a, l, d, h, e, i) {
+  postChatV3EventSource(s, a, d, h, e, i) {
     return f(this, null, function* () {
-      const r = this, o = {
+      const r = this, m = {
         code: 0,
         // messageId: uuidv4(),
         content: "",
-        role: b.Robot,
-        sessionId: l,
+        role: C.Robot,
+        sessionId: a,
         reference: [],
         searchReference: [],
         messageTime: "",
@@ -165,7 +165,7 @@ class B extends V {
       }, g = y(u({}, d), {
         stream: S.stream
       });
-      p(`${S.apiRoot}/chat/v3/chat?applicationId=${a}&sessionId=${l}`, {
+      T(`${S.apiRoot}/chat/v3/chat?applicationId=${s}&sessionId=${a}`, {
         method: "POST",
         headers: {
           Authorization: `Bearer ${this.token}`,
@@ -174,78 +174,78 @@ class B extends V {
         body: JSON.stringify(g),
         openWhenHidden: !0,
         signal: h,
-        onopen(t) {
-          return t.ok && e != null && e.onopen && e.onopen(), Promise.resolve();
+        onopen(n) {
+          return n.ok && e != null && e.onopen && e.onopen(), Promise.resolve();
         },
-        onmessage(t) {
-          if (!t.data && !t.event && !t.id && !t.retry)
+        onmessage(n) {
+          if (!n.data && !n.event && !n.id && !n.retry)
             return;
-          let s = {};
+          let t = {};
           try {
-            s = JSON.parse(t.data);
-          } catch (c) {
-            console.error("parse eventData to JSON failed. ", t.data);
+            t = JSON.parse(n.data);
+          } catch (o) {
+            console.error("parse eventData to JSON failed. ", n.data);
           }
-          if (s.code !== 200)
-            return o.index = -1, o.code = s.code, o.message = s.message, e != null && e.onmessage && e.onmessage(o), i && i({
-              code: s.code,
-              message: s.message
+          if (t.code !== 200)
+            return m.index = -1, m.code = t.code, m.message = t.message, e != null && e.onmessage && e.onmessage(m), i && i({
+              code: t.code,
+              message: t.message
             });
-          r.handleStreamAnswer(s, o), e != null && e.onmessage && e.onmessage(o);
+          r.handleStreamAnswer(t, m), e != null && e.onmessage && e.onmessage(m);
         },
         onerror() {
           e != null && e.onerror && e.onerror();
         },
         onclose() {
-          const t = y(u({}, o), {
+          const n = y(u({}, m), {
             isReplying: !1
           });
-          e != null && e.onclose && e.onclose(t);
+          e != null && e.onclose && e.onclose(n);
         }
       });
     });
   }
-  sendQuestion(a, l) {
+  sendQuestion(s, a) {
     return f(this, null, function* () {
-      var g;
       const {
         content: d,
         fileIds: h,
         oldMessageId: e,
         imageContent: i
-      } = a, {
+      } = s, {
         stream: r,
-        referenceFirst: o
-      } = l;
+        referenceFirst: m
+      } = a;
       if (this.active || (yield this.activate()), this._id) {
-        let t = {
+        const g = this.convertKnowledgesToChatFormat(this.knowledges);
+        let n = {
           content: d,
           fileIds: h,
-          referenceFirst: o,
+          referenceFirst: m,
           oldMessageId: e,
-          promptVariables: this.promptVariables.filter((s) => s.type === C.User).reduce((s, c) => y(u({}, s), {
-            [c.key]: c.value
+          promptVariables: this.promptVariables.filter((t) => t.type === v.User).reduce((t, o) => y(u({}, t), {
+            [o.key]: o.value
           }), {}),
-          systemPromptVariables: this.promptVariables.filter((s) => s.type === C.System).reduce((s, c) => y(u({}, s), {
-            [c.key]: c.value
+          systemPromptVariables: this.promptVariables.filter((t) => t.type === v.System).reduce((t, o) => y(u({}, t), {
+            [o.key]: o.value
           }), {}),
-          knowledges: (g = this.knowledges) == null ? void 0 : g.map((s) => s.id),
+          knowledges: g,
           //开启内容摘要
           enableTitleSummary: !0,
           imageContent: i
         };
-        this.ctrl = new AbortController(), this.postChatV3EventSource(this._applicationId, this._id, t, this.ctrl.signal, {
+        this.ctrl = new AbortController(), this.postChatV3EventSource(this._applicationId, this._id, n, this.ctrl.signal, {
           onopen: () => {
-            v.log("open", this.ctrl);
+            p.log("open", this.ctrl);
           },
-          onmessage: (s) => {
+          onmessage: (t) => {
             if (this._stopAnswer)
               return;
-            let c = this.messages[this.messages.length - 1];
-            c && c.role === D.Robot ? (c = u(u({}, c), s), this.messages[this.messages.length - 1] = c, this.event.emit("message:content", c), this.event.emit("message:updated", c)) : (this.pushMessage(s), this.event.emit("message:content", s));
+            let o = this.messages[this.messages.length - 1];
+            o && o.role === O.Robot ? (o = u(u({}, o), t), this.messages[this.messages.length - 1] = o, this.event.emit("message:content", o), this.event.emit("message:updated", o)) : (this.pushMessage(t), this.event.emit("message:content", t));
           },
-          onclose: (s) => f(this, null, function* () {
-            this._stopAnswer || (v.log(s, "onclose"), this.event.emit("answer:finish", s));
+          onclose: (t) => f(this, null, function* () {
+            this._stopAnswer || (p.log(t, "onclose"), this.event.emit("answer:finish", t));
           }),
           onerror: () => {
             this._stopAnswer || (this.event.emit("message:error", ""), this.frozen = !1);
@@ -262,7 +262,7 @@ class B extends V {
       throw new Error("MCP registration is not supported by DialogSession");
     });
   }
-  addMcpServers(a) {
+  addMcpServers(s) {
     return f(this, null, function* () {
       throw new Error("MCP registration is not supported by DialogSession");
     });
@@ -272,7 +272,13 @@ class B extends V {
       throw new Error("MCP registration is not supported by DialogSession");
     });
   }
+  convertKnowledgesToChatFormat(s) {
+    return s ? s.map((a) => a != null && a.fileIds ? {
+      id: a.id,
+      fileIds: a.fileIds
+    } : a.id) : [];
+  }
 }
 export {
-  B as default
+  k as default
 };