@blueking/ai-blueking 0.1.4 → 0.1.5

package/README.md

@@ -64,13 +64,29 @@ npm i @blueking/ai-blueking
 
   // 错误处理
   const handleError = (message: string, code: string, id: number | string) => {
-    const currentMessage = messages.value.at(-1);
-    currentMessage.status = 'error';
-    currentMessage.content = message;
-    loading.value = false;
+    if (message.includes('user authentication failed')) {
+      // 未登录，跳转登录
+      const loginUrl = new URL(process.env.BK_LOGIN_URL);
+      loginUrl.searchParams.append('c_url', location.origin);
+      window.location.href = loginUrl.href;
+    } else {
+      // 处理错误消息
+      const currentMessage = messages.value.at(-1);
+      currentMessage.status = 'fail';
+      currentMessage.content = message;
+      loading.value = false;
+    }
   };
 
-  const chatHelper = new ChatHelper('<接口地址>', handleStart, handleReceiveMessage, handleEnd, handleError);
+  // 需要将 <网关名> 替换成插件部署后生成的网关名
+  const prefix = process.env.BK_API_URL_TMPL.replace('{api_name}', '<网关名>').replace('http', 'https');
+  const chatHelper = new ChatHelper(
+    `${prefix}/prod/bk_plugin/plugin_api/assistant/`,
+    handleStart,
+    handleReceiveMessage,
+    handleEnd,
+    handleError,
+  );
 
   // 清空消息
   const handleClear = () => {
@@ -79,6 +95,8 @@ npm i @blueking/ai-blueking
 
   // 发送消息
   const handleSend = (message: string) => {
+    // 记录当前消息记录
+    const chatHistory = [...messages.value];
     // 添加一条消息
     messages.value.push({
       role: 'user',
@@ -89,7 +107,7 @@ npm i @blueking/ai-blueking
       {
         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",
   "description": "AI 小鲸",
   "license": "MIT",
   "author": "Tencent BlueKing",