@blueking/ai-blueking 0.1.3 → 0.1.4

package/README.md

@@ -85,11 +85,15 @@ npm i @blueking/ai-blueking
       content: message,
     });
     // ai 消息，id是唯一标识当前流，调用 chatHelper.stop 的时候需要传入
-    chatHelper.stream({
-      id: 1,
-      input: message,
-      chatHistory: messages.value,
-    });
+    chatHelper.stream(
+      {
+        inputs: {
+          input: message,
+          chat_history: messages.value,
+        },
+      },
+      1,
+    );
   };
 </script>
 ```

package/dist/chat-helper.d.ts

@@ -1,4 +1,3 @@
-import type { IChatHistory } from './types/index';
 type HandleStart = (id: number | string) => unknown;
 type HandleReceiveMessage = (message: string, id: number | string) => void;
 type HandleEnd = (id: number | string) => void;
@@ -12,10 +11,6 @@ export declare class ChatHelper {
     url: string;
     constructor(url: string, handleStart: HandleStart, handleReceiveMessage: HandleReceiveMessage, handleEnd: HandleEnd, handleError: HandleError);
     stop(id: number | string): void;
-    stream({ id, input, chatHistory }: {
-        id: number | string;
-        input?: string;
-        chatHistory: IChatHistory[];
-    }): Promise<void>;
+    stream(param: unknown, id: number | string): Promise<void>;
 }
 export {};

package/dist/chat-helper.js

@@ -1,6 +1,6 @@
-var g = Object.defineProperty;
-var y = (a, e, t) => e in a ? g(a, e, { enumerable: !0, configurable: !0, writable: !0, value: t }) : a[e] = t;
-var s = (a, e, t) => (y(a, typeof e != "symbol" ? e + "" : e, t), t);
+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) => {
   try {
     return JSON.parse(a), !0;
@@ -9,46 +9,45 @@ const E = (a) => {
   }
 };
 class m {
-  constructor(e, t, n, h, o) {
+  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 = e, this.handleStart = t, this.handleReceiveMessage = n, this.handleEnd = h, this.handleError = o, this.controllerMap = {};
+    this.url = t, this.handleStart = e, this.handleReceiveMessage = r, this.handleEnd = o, this.handleError = h, this.controllerMap = {};
   }
-  stop(e) {
-    var t, n;
-    return (n = (t = this.controllerMap[e]) == null ? void 0 : t.abort) == null || n.call(t), this.handleEnd(e);
+  stop(t) {
+    var e, r;
+    return (r = (e = this.controllerMap[t]) == null ? void 0 : e.abort) == null || r.call(e), this.handleEnd(t);
   }
-  async stream({ id: e, input: t, chatHistory: n }) {
-    await this.handleStart(e);
-    const h = new AbortController();
-    this.controllerMap[e] = h, fetch(this.url, {
+  async stream(t, e) {
+    const r = new AbortController();
+    this.controllerMap[e] = r, fetch(this.url, {
       method: "post",
-      signal: h.signal,
+      signal: r.signal,
       headers: {
         "Content-Type": "application/json"
       },
       mode: "cors",
       credentials: "include",
-      body: JSON.stringify({ inputs: { input: t, chat_history: n } })
+      body: JSON.stringify(t)
     }).then(async (o) => {
-      const c = o.body.pipeThrough(new window.TextDecoderStream()).getReader();
+      const h = o.body.pipeThrough(new window.TextDecoderStream()).getReader();
       for (; ; )
         try {
-          const { value: r, done: i } = await c.read();
-          if (i)
+          const { value: n, done: c } = await h.read();
+          if (c)
             break;
-          r.toString().split(`
-`).forEach((d) => {
-            const l = d.trim();
+          n.toString().split(`
+`).forEach((i) => {
+            const l = i.trim();
             if (E(l)) {
-              const { event: p, content: u } = JSON.parse(l);
-              switch (p) {
+              const { event: d, content: p } = JSON.parse(l);
+              switch (d) {
                 case "text":
-                  this.handleReceiveMessage(u, e);
+                  this.handleReceiveMessage(p, e);
                   break;
                 case "done":
                   this.handleEnd(e);
@@ -56,13 +55,13 @@ class m {
               }
             }
           });
-        } catch (r) {
-          (r == null ? void 0 : r.code) !== 20 && this.handleError(r.message, r.code, e);
+        } catch (n) {
+          (n == null ? void 0 : n.code) !== 20 && this.handleError(n.message, n.code, e);
           break;
         }
     }).catch((o) => {
       this.handleError(o.message, o.code, e);
-    });
+    }), await this.handleStart(e);
   }
 }
 export {

package/package.json

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