@blueking/ai-blueking 0.0.5 → 0.1.0-beta.1

package/README.md

@@ -26,6 +26,7 @@ npm i @blueking/ai-blueking
       @clear="handleClear"
       @close="handleClose"
       @send="handleSend"
+      @scroll="handleScroll"
     />
   </div>
 </template>
@@ -114,6 +115,10 @@ npm i @blueking/ai-blueking
     console.log('trigger send', val);
   };
 
+  const handleScroll = (event: Event) => {
+    console.log('trigger scroll', event);
+  };
+
   const handleClose = () => {
     console.log('trigger close');
   };
@@ -148,6 +153,7 @@ npm i @blueking/bkui-library
       @clear="handleClear"
       @close="handleClose"
       @send="handleSend"
+      @scroll="handleScroll"
     />
   </div>
 </template>
@@ -236,6 +242,9 @@ npm i @blueking/bkui-library
       handleChoosePrompt(prompt) {
         console.log('choose prompt', prompt);
       },
+      handleScroll(event: Event) {
+        console.log('trigger scroll', event);
+      },
     },
   };
 </script>

package/dist/ai-blueking.vue.d.ts

@@ -15,6 +15,7 @@ declare const _default: import("vue").DefineComponent<__VLS_TypePropsToOption<{
     clear: () => void;
     send: (val: string) => void;
     "choose-prompt": (prompt: IPrompt) => void;
+    scroll: (data: Event) => void;
 }, string, import("vue").PublicProps, Readonly<import("vue").ExtractPropTypes<__VLS_TypePropsToOption<{
     messages: IMessage[];
     prompts?: IPrompt[] | undefined;
@@ -27,6 +28,7 @@ declare const _default: import("vue").DefineComponent<__VLS_TypePropsToOption<{
     userPhoto?: string | undefined;
     logo?: string | undefined;
 }>>> & {
+    onScroll?: ((data: Event) => any) | undefined;
     "onChoose-prompt"?: ((prompt: IPrompt) => any) | undefined;
     onClear?: (() => any) | undefined;
     onSend?: ((val: string) => any) | undefined;

package/dist/ai.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/components/render-content.vue.d.ts

@@ -10,6 +10,7 @@ declare const _default: import("vue").DefineComponent<__VLS_WithDefaults<__VLS_T
 }>, {}, unknown, {}, {}, import("vue").ComponentOptionsMixin, import("vue").ComponentOptionsMixin, {
     send: (userInput: string) => void;
     "choose-prompt": (prompt: IPrompt) => void;
+    scroll: (data: Event) => void;
 }, string, import("vue").PublicProps, Readonly<import("vue").ExtractPropTypes<__VLS_WithDefaults<__VLS_TypePropsToOption<{
     messages: IMessage[];
     prompts?: IPrompt[] | undefined;
@@ -19,6 +20,7 @@ declare const _default: import("vue").DefineComponent<__VLS_WithDefaults<__VLS_T
 }>, {
     background: string;
 }>>> & {
+    onScroll?: ((data: Event) => any) | undefined;
     "onChoose-prompt"?: ((prompt: IPrompt) => any) | undefined;
     onSend?: ((userInput: string) => any) | undefined;
 }, {

package/dist/index.es.min.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;