@blueking/ai-blueking 0.0.6 → 0.1.0-beta.2

package/dist/chat-helper.d.ts

@@ -0,0 +1,21 @@
+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;
+type HandleError = (message: string, code: string, id: number | string) => unknown;
+export declare class ChatHelper {
+    controllerMap: Record<number | string, AbortController>;
+    handleEnd: HandleEnd;
+    handleError: HandleError;
+    handleReceiveMessage: HandleReceiveMessage;
+    handleStart: HandleStart;
+    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>;
+}
+export {};

package/dist/chat-helper.js

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

package/dist/types/enum.d.ts

@@ -0,0 +1,9 @@
+export declare enum AgentMessageType {
+    AgentAction = "AgentAction",
+    AgentActionInfo = "AgentActionInfo"
+}
+export declare enum RoleType {
+    Assistant = "assistant",
+    System = "system",
+    User = "user"
+}

package/dist/types/index.d.ts

@@ -1,6 +1,5 @@
-export interface IMessage {
-    content: string;
-    type: 'ai' | 'user';
+import { RoleType } from './enum';
+export interface IMessage extends IChatHistory {
     status?: 'error' | 'loading';
 }
 export interface IPrompt {
@@ -23,3 +22,7 @@ export interface IStartPosition {
     left?: number;
     right?: number;
 }
+export interface IChatHistory {
+    role: RoleType;
+    content: string;
+}

package/dist/util/index.d.ts

@@ -3,3 +3,9 @@
  * @param {*} name cookie 的名称
  */
 export declare const getCookieByName: (name: string) => string;
+/**
+ * 判断是否是json字符串
+ * @param str
+ * @returns
+ */
+export declare const isJSON: (str: string) => boolean;