@blueking/ai-blueking 0.1.4 → 0.1.5-beta.1

package/README.md

@@ -79,6 +79,8 @@ npm i @blueking/ai-blueking
 
   // 发送消息
   const handleSend = (message: string) => {
+    // 记录当前消息记录
+    const chatHistory = [...messages.value];
     // 添加一条消息
     messages.value.push({
       role: 'user',
@@ -88,8 +90,7 @@ npm i @blueking/ai-blueking
     chatHelper.stream(
       {
         inputs: {
-          input: message,
-          chat_history: messages.value,
+          chat_history: chatHistory,
         },
       },
       1,

package/dist/chat-helper.js

@@ -1,32 +1,32 @@
-var u = Object.defineProperty;
-var g = (a, t, e) => t in a ? u(a, t, { enumerable: !0, configurable: !0, writable: !0, value: e }) : a[t] = e;
-var s = (a, t, e) => (g(a, typeof t != "symbol" ? t + "" : t, e), e);
-const E = (a) => {
+var b = Object.defineProperty;
+var f = (r, t, e) => t in r ? b(r, t, { enumerable: !0, configurable: !0, writable: !0, value: e }) : r[t] = e;
+var n = (r, t, e) => (f(r, typeof t != "symbol" ? t + "" : t, e), e);
+const m = (r) => {
   try {
-    return JSON.parse(a), !0;
+    return JSON.parse(r), !0;
   } catch {
     return !1;
   }
 };
-class m {
-  constructor(t, e, r, o, h) {
-    s(this, "controllerMap");
-    s(this, "handleEnd");
-    s(this, "handleError");
-    s(this, "handleReceiveMessage");
-    s(this, "handleStart");
-    s(this, "url");
-    this.url = t, this.handleStart = e, this.handleReceiveMessage = r, this.handleEnd = o, this.handleError = h, this.controllerMap = {};
+class y {
+  constructor(t, e, a, o, h) {
+    n(this, "controllerMap");
+    n(this, "handleEnd");
+    n(this, "handleError");
+    n(this, "handleReceiveMessage");
+    n(this, "handleStart");
+    n(this, "url");
+    this.url = t, this.handleStart = e, this.handleReceiveMessage = a, this.handleEnd = o, this.handleError = h, this.controllerMap = {};
   }
   stop(t) {
-    var e, r;
-    return (r = (e = this.controllerMap[t]) == null ? void 0 : e.abort) == null || r.call(e), this.handleEnd(t);
+    var e, a;
+    return (a = (e = this.controllerMap[t]) == null ? void 0 : e.abort) == null || a.call(e), this.handleEnd(t);
   }
   async stream(t, e) {
-    const r = new AbortController();
-    this.controllerMap[e] = r, fetch(this.url, {
+    const a = new AbortController();
+    this.controllerMap[e] = a, fetch(this.url, {
       method: "post",
-      signal: r.signal,
+      signal: a.signal,
       headers: {
         "Content-Type": "application/json"
       },
@@ -37,26 +37,33 @@ class m {
       const h = o.body.pipeThrough(new window.TextDecoderStream()).getReader();
       for (; ; )
         try {
-          const { value: n, done: c } = await h.read();
-          if (c)
+          const { value: s, done: d } = await h.read();
+          if (d)
             break;
-          n.toString().split(`
-`).forEach((i) => {
-            const l = i.trim();
-            if (E(l)) {
-              const { event: d, content: p } = JSON.parse(l);
-              switch (d) {
+          s.toString().split(`
+`).forEach((p) => {
+            const l = p.trim();
+            if (m(l)) {
+              const { event: u, content: g, result: E, code: c, message: i } = JSON.parse(l);
+              if (E === !1) {
+                this.handleError(i, c, e);
+                return;
+              }
+              switch (u) {
                 case "text":
-                  this.handleReceiveMessage(p, e);
+                  this.handleReceiveMessage(g, e);
                   break;
                 case "done":
                   this.handleEnd(e);
                   break;
+                case "error":
+                  this.handleError(i, c, e);
+                  break;
               }
             }
           });
-        } catch (n) {
-          (n == null ? void 0 : n.code) !== 20 && this.handleError(n.message, n.code, e);
+        } catch (s) {
+          (s == null ? void 0 : s.code) !== 20 && this.handleError(s.message, s.code, e);
           break;
         }
     }).catch((o) => {
@@ -65,5 +72,5 @@ class m {
   }
 }
 export {
-  m as ChatHelper
+  y as ChatHelper
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blueking/ai-blueking",
-  "version": "0.1.4",
+  "version": "0.1.5-beta.1",
   "description": "AI 小鲸",
   "license": "MIT",
   "author": "Tencent BlueKing",