@glodon-aiot/agent-cli-sdk 3.0.9 → 3.0.10-beta.1

package/dist/es/AgentSession.mjs

@@ -0,0 +1,246 @@
+var C = Object.defineProperty, S = Object.defineProperties;
+var b = Object.getOwnPropertyDescriptors;
+var A = Object.getOwnPropertySymbols;
+var x = Object.prototype.hasOwnProperty, E = Object.prototype.propertyIsEnumerable;
+var _ = (a, t, i) => t in a ? C(a, t, { enumerable: !0, configurable: !0, writable: !0, value: i }) : a[t] = i, l = (a, t) => {
+  for (var i in t || (t = {}))
+    x.call(t, i) && _(a, i, t[i]);
+  if (A)
+    for (var i of A(t))
+      E.call(t, i) && _(a, i, t[i]);
+  return a;
+}, y = (a, t) => S(a, b(t));
+var p = (a, t, i) => new Promise((m, n) => {
+  var s = (c) => {
+    try {
+      r(i.next(c));
+    } catch (h) {
+      n(h);
+    }
+  }, e = (c) => {
+    try {
+      r(i.throw(c));
+    } catch (h) {
+      n(h);
+    }
+  }, r = (c) => c.done ? m(c.value) : Promise.resolve(c.value).then(s, e);
+  r((i = i.apply(a, t)).next());
+});
+import { ChatMessageRole as I } from "@glodon-aiot/apis";
+import L from "./Logger.mjs";
+import { configuration as w } from "./Config.mjs";
+import { io as q } from "./node_modules/socket.io-client/build/esm/index.mjs";
+import { UserMode as v } from "./PayloadTypes.mjs";
+import N, { SessionErrors as k } from "./Session.mjs";
+const u = new L("[AgentSession] ");
+class J extends N {
+  constructor(t, i = "", m, n, s, e) {
+    super(t, i, m, n, s, e);
+  }
+  prepareConnection() {
+    return new Promise((t) => p(this, null, function* () {
+      this.socket && t(!0);
+      const i = {
+        version: "1.0",
+        stream: "false",
+        appId: this._applicationId,
+        sessionId: this.id,
+        Authorization: `${this.token}`,
+        userMode: w.debug ? v.Debug : v.Normal
+      }, m = new URL(w.apiRoot), n = yield q(m.origin, {
+        path: m.pathname.replace("api", "socketio") + "/copilot",
+        query: i,
+        transports: ["websocket"]
+      });
+      n.on("connect", () => {
+        u.log("socket:connect"), this.event.emit("socket:connect"), n.emit("init");
+      }), n.on("disconnect", () => {
+        u.log("socket:disconnect"), this.event.emit("socket:disconnect");
+      }), n.on("ready", (s) => {
+        if (s.code === 200) {
+          u.log("ready");
+          const e = w.debug ? v.Debug : v.Normal;
+          e === v.Debug && s.userMode, this._ready || (this._ready = !0, this.event.emit("ready")), this.frozen = !1, t(!0);
+        } else {
+          this._ready && (this._ready = !1);
+          const e = new Error(
+            k.SocketConnectionFailed
+            // , {
+            //   cause: data,
+            // }
+          );
+          u.error("error", e), this.event.emit("error", e), t(!1);
+        }
+      }), n.on("action", (s, e) => p(this, null, function* () {
+        if (this._stopAnswer || !this.id)
+          return;
+        const {
+          type: r,
+          data: {
+            runnerIsClient: c,
+            pluginCode: h,
+            funcCode: g,
+            parameters: o,
+            requestBody: U
+          }
+        } = s, d = this.messages[this.messages.length - 1];
+        if (!d.messageId)
+          d.actions || (d.actions = []), d.actions[s.data.step - 1] || (d.actions[s.data.step - 1] = []), d.actions[s.data.step - 1].push(s), this.event.emit("message:action", {
+            messageId: d.messageId,
+            action: s
+          }), this.event.emit("message:updated", d);
+        else {
+          const f = {
+            messageId: "",
+            content: "",
+            role: I.Robot,
+            sessionId: this.id,
+            reference: [],
+            searchReference: [],
+            isLoading: !0,
+            messageTime: "",
+            actions: [[s]],
+            comment: {
+              isAgree: 0
+            }
+          };
+          this.pushMessage(f), this.event.emit("message:action", {
+            messageId: f.messageId,
+            action: s
+          });
+        }
+        if (this.inactivating) {
+          const f = new Error(
+            k.Inactive
+            // {
+            //   cause: {
+            //     code: -1,
+            //     message:
+            //       'session is inactive, skipping plugin client function call',
+            //   },
+            // }
+          );
+          u.warn("error", f), this.event.emit("error", f);
+        } else if (r === "FUNCTION_CALL" && c) {
+          if (u.log(`calling function ${g} of plugin  ${h} with argument:
+ ${JSON.stringify(o)}`), !w.plugins[h]) {
+            e({
+              code: 404,
+              message: "对不起，没有找到插件。"
+            });
+            return;
+          }
+          const f = w.plugins[h][g];
+          typeof f != "function" && e({
+            code: 404,
+            message: "对不起，没有找到插件方法。"
+          });
+          let M = o == null ? void 0 : o.query;
+          try {
+            M = JSON.parse(o == null ? void 0 : o.query);
+          } catch (R) {
+          }
+          f(M, (...R) => {
+            this._stopAnswer || e(...R);
+          });
+        }
+      })), 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) {
+            let e = this.messages[this.messages.length - 1];
+            if (e && !e.messageId)
+              e = l(l({}, e), s.data), this.event.emit("message:content", e), this.event.emit("message:updated", e), this.event.emit("answer:finish", e);
+            else {
+              const r = l({
+                messageId: "",
+                content: "",
+                role: I.Robot,
+                sessionId: this.id,
+                reference: [],
+                searchReference: [],
+                isLoading: !1,
+                messageTime: "",
+                actions: [],
+                comment: {
+                  isAgree: 0
+                }
+              }, s.data);
+              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 = l(l({}, 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));
+          }
+      }), n.on("reference", (s) => p(this, null, function* () {
+        if (this._stopAnswer)
+          return;
+        if (!this.id) {
+          const o = new Error(
+            k.IdUndefined
+            // {
+            //   cause: SESSION_ERROR_CAUSE[SessionErrors.IdUndefined],
+            // }
+          );
+          u.error(o), this.event.emit("error", o), this.frozen = !1;
+          return;
+        }
+        const {
+          type: e,
+          data: r,
+          requestId: c
+        } = s, h = this.messages.findIndex((o) => o.messageId === c && o.role === I.User);
+        if (h < 0)
+          return;
+        const g = h + 1;
+        if (g >= this.messages.length) {
+          const o = {
+            messageId: "",
+            content: "",
+            role: I.Robot,
+            sessionId: this.id,
+            reference: [],
+            searchReference: [],
+            isLoading: !0,
+            messageTime: "",
+            actions: [],
+            comment: {
+              isAgree: 0
+            }
+          };
+          this.pushMessage(o);
+        }
+        e === "KG" ? (this.messages[g] = y(l({}, this.messages[g]), {
+          reference: r.knowledgeReference
+        }), yield this.fillMessageReference(this.messages[g]), this.event.emit("message:reference", r.knowledgeReference), this.event.emit("message:updated", this.messages[g])) : e === "WEB_SEARCH" && (this.messages[g] = y(l({}, this.messages[g]), {
+          searchReference: r.knowledgeReference
+        }), this.event.emit("message:searchreference", r.searchReference), this.event.emit("message:updated", this.messages[g]));
+      })), this.socket = n, n.open();
+    }));
+  }
+  sendQuestion(t, i) {
+    return p(this, null, function* () {
+      var m, n;
+      if (!this.active) {
+        if (yield this.activate(), !this.socket) {
+          const s = new Error(
+            k.SocketConnectionFailed
+            // {
+            //   cause: { code: -1, message: 'socket is null' },
+            // }
+          );
+          return u.warn("error", s), this.event.emit("error", s), !1;
+        }
+        (m = this.socket) == null || m.emit("question", t);
+        return;
+      }
+      (n = this.socket) == null || n.emit("question", t);
+    });
+  }
+  stopAnswer() {
+    var t;
+    (t = this.socket) == null || t.emit("stop_answer");
+  }
+}
+export {
+  J as default
+};

package/dist/es/BotClient.mjs

@@ -0,0 +1,236 @@
+var v = Object.defineProperty;
+var y = (p, i, t) => i in p ? v(p, i, { enumerable: !0, configurable: !0, writable: !0, value: t }) : p[i] = t;
+var c = (p, i, t) => (y(p, typeof i != "symbol" ? i + "" : i, t), t);
+var a = (p, i, t) => new Promise((e, s) => {
+  var l = (n) => {
+    try {
+      o(t.next(n));
+    } catch (r) {
+      s(r);
+    }
+  }, h = (n) => {
+    try {
+      o(t.throw(n));
+    } catch (r) {
+      s(r);
+    }
+  }, o = (n) => n.done ? e(n.value) : Promise.resolve(n.value).then(l, h);
+  o((t = t.apply(p, i)).next());
+});
+import { parseJWT as f } from "./utils.mjs";
+import g from "./Logger.mjs";
+import { getCVForceService as S } from "./api/index.mjs";
+import { configure as w, configuration as _ } from "./Config.mjs";
+import { ApplicationTypes as u } from "@glodon-aiot/apis";
+import m from "./AgentSession.mjs";
+import k from "./DialogSession.mjs";
+import A from "./EventBus.mjs";
+const x = new g("[BotClient] ");
+class q {
+  constructor(i, t, e = !0) {
+    c(this, "event", new A());
+    c(this, "cvforceApi");
+    //从token解析出来的用户信息
+    c(this, "_applicationId");
+    c(this, "_application", null);
+    //session列表，被加载过的session将会被缓存到这个列表中
+    c(this, "_sessions", []);
+    //活跃的session，只支持一个活跃的session
+    c(this, "_activeSession");
+    c(this, "_ready", !1);
+    //问题长度
+    c(this, "_questionContextMaxLength", 4e3);
+    c(this, "_promptVariables", []);
+    if (this._token = i, this.lazyload = e, w(t), this._applicationId = f(i).rid || _.applicationId, !this._applicationId)
+      throw "token is not vaild";
+    this.cvforceApi = this.getService(), this.loadApplication().then(() => {
+      this.event.emit("ready");
+    });
+  }
+  checkReady() {
+    if (!this.ready)
+      throw Error("client is not ready.");
+  }
+  get ready() {
+    return this._ready;
+  }
+  set ready(i) {
+    this._ready = i;
+  }
+  get questionContextMaxLength() {
+    return this.checkReady(), this._questionContextMaxLength;
+  }
+  get promptVariables() {
+    return this.checkReady(), this._promptVariables;
+  }
+  get knowledges() {
+    var i;
+    return this.checkReady(), (i = this._application) == null ? void 0 : i.arrange[0].knowledges;
+  }
+  get sessions() {
+    return this.checkReady(), this._sessions;
+  }
+  set sessions(i) {
+    this.checkReady(), this._sessions = i, this.event.emit("sessions:updated", i);
+  }
+  loadApplication() {
+    return a(this, null, function* () {
+      var t, e;
+      this._application = yield this.cvforceApi.getApplicationId(this._applicationId);
+      const i = (e = (t = this._application.arrange) == null ? void 0 : t[0]) == null ? void 0 : e.llmModel;
+      return i && (this._questionContextMaxLength = yield this.cvforceApi.getBaseModel(i).then((s) => s.maxModelLength)), this._promptVariables = yield this.cvforceApi.getPrompts(this._applicationId), this._ready = !0, this.event.emit("application:loaded"), this._application;
+    });
+  }
+  getService() {
+    return this.cvforceApi ? this.cvforceApi : S(_.apiRoot, this.token, (i, t) => x.error(t));
+  }
+  get token() {
+    return this._token;
+  }
+  set token(i) {
+    this.checkReady(), this._token = i, this.ready = !1;
+    const t = f(i).rid;
+    this._applicationId !== t && (this._applicationId = t, this._application = null, this.loadApplication().then(() => {
+      this.event.emit("ready");
+    }), this.clean());
+  }
+  get applicationId() {
+    return this._applicationId;
+  }
+  get application() {
+    return this.checkReady(), this._application;
+  }
+  get activeSession() {
+    var i;
+    return this.checkReady(), ((i = this._activeSession) == null ? void 0 : i.id) || "";
+  }
+  set activeSession(i) {
+    this.checkReady(), this.loadSession(i).then((t) => {
+      this._activeSession && this._activeSession.id !== i && this._activeSession.inactivate(), this._activeSession = t;
+    });
+  }
+  createSession(i, t, e, s, l) {
+    return a(this, null, function* () {
+      const h = this;
+      return new Promise((o, n) => {
+        const r = (d) => a(this, null, function* () {
+          if (d === u.Agent)
+            o(new m(i, t, e, s, l, h));
+          else if (d === u.Dialog)
+            o(new k(i, t, e, s, l, h));
+          else
+            throw Error(`不支持的应用类型:${d}`);
+        });
+        if (this.ready) {
+          if (!this._application)
+            throw Error("client is not ready");
+          r(this._application.type);
+        } else
+          this.event.on("ready", () => {
+            if (!this._application)
+              throw Error("_application is null");
+            r(this._application.type);
+          }, !0);
+      });
+    });
+  }
+  getSessions(i) {
+    return a(this, null, function* () {
+      return new Promise((t, e) => {
+        const s = () => this.cvforceApi.listSessions(this._applicationId, i).then((l) => {
+          const h = [];
+          return l.forEach((o) => a(this, null, function* () {
+            const n = this._sessions.find((r) => o.id === r.id);
+            if (n)
+              n.data = o, h.push(n);
+            else {
+              const r = yield this.createSession(this.token, o.id, o, this.lazyload, this._application || void 0);
+              h.push(r);
+            }
+          })), this.sessions = h, h;
+        });
+        if (this.ready) {
+          if (!this._application)
+            throw Error("_application is null");
+          t(s());
+        } else
+          this.event.on("ready", () => {
+            if (!this._application)
+              throw Error("_application is null");
+            t(s());
+          }, !0);
+      });
+    });
+  }
+  loadSession(i = "", t) {
+    return a(this, null, function* () {
+      this.checkReady();
+      let e = this.sessions.find((s) => s.id === i);
+      if (e || (e = yield this.createSession(this.token, i, t), this._sessions.push(e)), this._activeSession) {
+        if (this._activeSession === e)
+          return e;
+        this._activeSession.inactivate();
+      }
+      return e.id ? (yield e.activate(), this._activeSession = e) : e.addEventListener("data:loaded", () => {
+        this._activeSession !== e && (e == null || e.activate(), this._activeSession = e);
+      }, !0), e;
+    });
+  }
+  reload() {
+    return this.loadApplication().then(() => a(this, null, function* () {
+      return yield this.getSessions(), this.event.emit("ready"), this;
+    }));
+  }
+  reloadSession(i) {
+    return a(this, null, function* () {
+      yield this.reload();
+      let t = this._sessions.find((e) => e.id === i) || this._activeSession;
+      if (!t) {
+        console.warn("reloadSession session failed.");
+        return;
+      }
+      return yield t.reload(), t;
+    });
+  }
+  setTop(i, t, e = "top_time desc, sort_status desc, updated_at desc") {
+    return a(this, null, function* () {
+      let s = i ? this._sessions.find((l) => l.id === i) : this._activeSession;
+      if (!s) {
+        console.warn("setTop session failed.");
+        return;
+      }
+      return yield s.setTop(t), this.getSessions({
+        order: e
+      });
+    });
+  }
+  addEventListener(i, t, e = !1) {
+    if (typeof t != "function")
+      throw Error("event listener is not a function");
+    if (e) {
+      this.event.once(i, t);
+      return;
+    }
+    this.event.on(i, t);
+  }
+  removeEventListener(i, t) {
+    this.event.off(i, t);
+  }
+  removeSession(i, t) {
+    return a(this, null, function* () {
+      this.checkReady();
+      let e = -1;
+      if (typeof i == "string" ? e = this._sessions.findIndex((s) => s.id === i) : e = this._sessions.findIndex((s) => s.id === i.id), e > -1) {
+        const s = this._sessions[e];
+        t && (yield s.delete()), this._sessions.splice(e, 1);
+      } else
+        console.warn("session delete failed, session is not exist.");
+    });
+  }
+  clean() {
+    this._sessions.forEach((i) => i.destroy()), this.sessions = [];
+  }
+}
+export {
+  q as default
+};

package/dist/es/Config.mjs

@@ -0,0 +1,34 @@
+var v = Object.defineProperty, s = Object.defineProperties;
+var u = 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) => {
+  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));
+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 || {});
+let c = {
+  debug: !1,
+  env: "prod",
+  //AIoT 环境, 与apiRoot配对使用
+  apiRoot: "https://copilot.glodon.com/api/cvforce",
+  //https://${hostname}/${PRODUCD_CODE}
+  applicationId: "",
+  plugins: {}
+};
+function x(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);
+}
+export {
+  f as ApiUrl,
+  c as configuration,
+  x as configure
+};