@blueking/ai-blueking 0.3.29 → 0.4.1

package/README.md

@@ -154,7 +154,7 @@ AI 小鲸支持渲染可点击的交互式 HTML 元素。以下是如何实现
 
 <script lang="ts" setup>
   import { ref } from 'vue';
-  import AIBlueking, { ChatHelper, RoleType, MessageStatus } from '@blueking/ai-blueking';
+  import AIBlueking, { ChatHelper, RoleType, MessageStatus, isThinking } from '@blueking/ai-blueking';
   import '@blueking/ai-blueking/dist/vue3/style.css';
 
   interface ISendData {
@@ -178,32 +178,28 @@ AI 小鲸支持渲染可点击的交互式 HTML 元素。以下是如何实现
   };
 
   // 接收消息
-  const handleReceiveMessage = (message: string, id: number | string) => {
+  const handleReceiveMessage = (message: string, id: number | string, cover?: boolean) => {
     const currentMessage = messages.value.at(-1);
-    if (currentMessage.status === MessageStatus.Loading) {
+    if (currentMessage?.status === MessageStatus.Loading) {
       // 如果是loading状态，直接覆盖
       currentMessage.content = message;
       currentMessage.status = MessageStatus.Success;
-    } else if (currentMessage.status === MessageStatus.Success) {
-      // 如果是后续消息，就追加消息
-      currentMessage.content += message;
+    } else if (currentMessage?.status === MessageStatus.Success) {
+      // 根据 cover 参数决定是追加还是覆盖
+      currentMessage.content = cover ? message : currentMessage.content + message;
     }
   };
 
   // 聊天结束
-  const handleEnd = (id: number | string, message?: string) => {
+  const handleEnd = (id: number | string) => {
     loading.value = false;
     const currentMessage = messages.value.at(-1);
-    if (message) {
-      // done 的情况下，返回 message，直接覆盖
-      currentMessage.content = message;
-      currentMessage.status = MessageStatus.Success;
-    } else if (currentMessage.status === MessageStatus.Loading) {
-      // loading 情况下终止
+    // 在 loading 状态或思考状态下终止时，标记为错误
+    if (currentMessage?.status === MessageStatus.Loading || isThinking(currentMessage?.content || '')) {
       currentMessage.content = '聊天内容已中断';
       currentMessage.status = MessageStatus.Error;
     }
-  }
+  };
 
   // 错误处理
   const handleError = (message: string, code: string | number, id: number | string) => {
@@ -228,11 +224,12 @@ AI 小鲸支持渲染可点击的交互式 HTML 元素。以下是如何实现
     handleReceiveMessage,
     handleEnd,
     handleError,
+    messages.value,
   );
 
   // 清空消息
   const handleClear = () => {
-    messages.value = [];
+    messages.value.splice(0); // 必须使用这种方式清空消息, 不能使用 messages.value = []，否则 ChatHelper 无法感知消息数组的变化
   };
 
   // 发送消息

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

@@ -26,7 +26,7 @@ declare const _default: import("vue").DefineComponent<__VLS_WithDefaults<__VLS_T
     enablePopup: boolean;
 }>, {
     setInputMessage: (val: string) => void;
-    quickActions: (type: "translate" | "explanation", content: string) => void;
+    quickActions: (type: "explanation" | "translate", content: string) => void;
 }, unknown, {}, {}, import("vue").ComponentOptionsMixin, import("vue").ComponentOptionsMixin, {
     send: (data: ISendData) => void;
     close: () => void;
@@ -38,6 +38,12 @@ declare const _default: import("vue").DefineComponent<__VLS_WithDefaults<__VLS_T
     "update:model": (val: string) => void;
     "choose-prompt": (prompt: IPrompt) => void;
     "update:isShow": (val: boolean) => void;
+    "shortcut-click": (data: {
+        type: string;
+        label: string;
+        cite: string;
+        prompt: string;
+    }) => void;
 }, string, import("vue").PublicProps, Readonly<import("vue").ExtractPropTypes<__VLS_WithDefaults<__VLS_TypePropsToOption<IProps>, {
     enablePopup: boolean;
 }>>> & {
@@ -49,6 +55,12 @@ declare const _default: import("vue").DefineComponent<__VLS_WithDefaults<__VLS_T
     onSend?: ((data: ISendData) => any) | undefined;
     "onAi-click"?: ((val: string) => any) | undefined;
     "onScroll-load"?: (() => any) | undefined;
+    "onShortcut-click"?: ((data: {
+        type: string;
+        label: string;
+        cite: string;
+        prompt: string;
+    }) => any) | undefined;
     "onUpdate:model"?: ((val: string) => any) | undefined;
     "onUpdate:isShow"?: ((val: boolean) => any) | undefined;
 }, {

package/dist/chat-helper.d.ts

@@ -1,5 +1,8 @@
+import { ReferenceDocStyleManager } from './composables/use-reference-docs';
+import { ThinkStyleManager } from './composables/use-think';
+import { IMessage } from './types';
 type HandleStart = (id: number | string) => unknown;
-type HandleReceiveMessage = (message: string, id: number | string) => void;
+type HandleReceiveMessage = (message: string, id: number | string, cover?: boolean) => void;
 type HandleEnd = (id: number | string, message?: string) => void;
 type HandleError = (message: string, code: number | string, id: number | string) => unknown;
 export declare class ChatHelper {
@@ -8,8 +11,12 @@ export declare class ChatHelper {
     handleError: HandleError;
     handleReceiveMessage: HandleReceiveMessage;
     handleStart: HandleStart;
+    messages: IMessage[];
+    styleManager: InstanceType<typeof ReferenceDocStyleManager>;
+    thinkStyleManager: InstanceType<typeof ThinkStyleManager>;
     url: string;
-    constructor(url: string, handleStart: HandleStart, handleReceiveMessage: HandleReceiveMessage, handleEnd: HandleEnd, handleError: HandleError);
+    constructor(url: string, handleStart: HandleStart, handleReceiveMessage: HandleReceiveMessage, handleEnd: HandleEnd, handleError: HandleError, messages: IMessage[]);
+    destroy(): void;
     stop(id: number | string): void;
     stream(param: unknown, id: number | string, headers?: Record<string, string>): Promise<void>;
 }

package/dist/components/render-content.vue.d.ts

@@ -25,6 +25,12 @@ declare const _default: import("vue").DefineComponent<__VLS_WithDefaults<__VLS_T
     "choose-prompt": (prompt: IPrompt) => void;
     stop: () => void;
     "scroll-load": () => void;
+    "shortcut-click": (data: {
+        type: string;
+        label: string;
+        cite: string;
+        prompt: string;
+    }) => void;
 }, string, import("vue").PublicProps, Readonly<import("vue").ExtractPropTypes<__VLS_WithDefaults<__VLS_TypePropsToOption<IProps>, {
     background: string;
     scrollLoadingEnd: boolean;
@@ -36,6 +42,12 @@ declare const _default: import("vue").DefineComponent<__VLS_WithDefaults<__VLS_T
     onSend?: ((data: ISendData) => any) | undefined;
     "onAi-click"?: ((userInput: string) => any) | undefined;
     "onScroll-load"?: (() => any) | undefined;
+    "onShortcut-click"?: ((data: {
+        type: string;
+        label: string;
+        cite: string;
+        prompt: string;
+    }) => any) | undefined;
 }, {
     shortcuts: IShortCut[];
     background: string;

package/dist/components/render-popup.vue.d.ts

@@ -1,8 +1,21 @@
-declare const _default: import("vue").DefineComponent<{}, {}, {}, {}, {}, import("vue").ComponentOptionsMixin, import("vue").ComponentOptionsMixin, {
+import { IShortCut } from '../types';
+interface IProps {
+    shortcuts: IShortCut[];
+}
+declare const _default: import("vue").DefineComponent<__VLS_TypePropsToOption<IProps>, {}, unknown, {}, {}, import("vue").ComponentOptionsMixin, import("vue").ComponentOptionsMixin, {
     click: (...args: any[]) => void;
     shortcutClick: (...args: any[]) => void;
-}, string, import("vue").PublicProps, Readonly<import("vue").ExtractPropTypes<{}>> & {
+}, string, import("vue").PublicProps, Readonly<import("vue").ExtractPropTypes<__VLS_TypePropsToOption<IProps>>> & {
     onClick?: ((...args: any[]) => any) | undefined;
     onShortcutClick?: ((...args: any[]) => any) | undefined;
 }, {}, {}>;
 export default _default;
+type __VLS_NonUndefinedable<T> = T extends undefined ? never : T;
+type __VLS_TypePropsToOption<T> = {
+    [K in keyof T]-?: {} extends Pick<T, K> ? {
+        type: import('vue').PropType<__VLS_NonUndefinedable<T[K]>>;
+    } : {
+        type: import('vue').PropType<T[K]>;
+        required: true;
+    };
+};

package/dist/composables/use-reference-docs.d.ts

@@ -0,0 +1,19 @@
+import type { IDocument } from '../types/index';
+/**
+ * 获取引用资料的 HTML 内容
+ * @param documents
+ * @returns
+ */
+export declare const getHtmlContentFromDocuments: (documents: IDocument[]) => string;
+/**
+ * 移除引用资料的 HTML 内容
+ * @param content
+ * @returns
+ */
+export declare const removeReferenceDoc: (content: string) => string;
+export declare class ReferenceDocStyleManager {
+    private styleEle;
+    addStyle(): void;
+    removeStyle(): void;
+}
+export declare const useReferenceDoc: () => void;

package/dist/composables/use-think.d.ts

@@ -0,0 +1,26 @@
+/**
+ * 获取思考的 HTML 内容
+ * @param chatContent 聊天内容
+ * @param content 思考内容
+ * @param elapsedTime 思考时间
+ * @returns 完整的 chatContent
+ */
+export declare const getHtmlContentFromThink: (chatContent: string, content: string, elapsedTime?: number) => string;
+/**
+ * 判断是否是思考中
+ * @param content
+ * @returns
+ */
+export declare const isThinking: (content: string) => boolean | null;
+/**
+ * 移除思考的 HTML 内容
+ * @param content
+ * @returns
+ */
+export declare const removeThink: (content: string) => string;
+export declare class ThinkStyleManager {
+    private styleEle;
+    addStyle(): void;
+    removeStyle(): void;
+}
+export declare const useThink: () => void;

package/dist/lang/index.d.ts

@@ -29,7 +29,7 @@ export declare const langData: {
     readonly translateShortcut: "You are a highly skilled AI trained in language translation. I would like you to translate the text delimited by triple quotes (Translation of Chinese into English and other languages into Chinese), ensuring that the translation is colloquial and authentic.\n    Only give me the output and nothing else. Do not wrap responses in quotes\n    '''\n      {{ SELECTED_TEXT }}\n    '''\n    ";
     readonly explanationShortcut: "You are a professional explainer. Please provide a detailed explanation of \"{{ SELECTED_TEXT }}\". Your explanation should include: 1) basic meaning and conceptual explanation; 2) practical applications or use cases; 3) if it's a technical term, please provide relevant technical background; 4) where appropriate, provide specific examples to aid understanding. Use clear and accessible language to ensure non-experts can understand. If the word/phrase has multiple meanings, please list the main definitions. Keep your response concise and clear while ensuring completeness and accuracy of information.";
     readonly 翻译: "translate";
-    readonly 解释: "explan";
+    readonly 解释: "explanation";
     readonly 'AI\u5C0F\u9CB8\u6B63\u5728\u56DE\u590D\uFF0C\u8BF7\u7ED3\u675F\u540E\u518D\u6E05\u7A7A': "AI BK GPT is executing, please clear after completion";
 };
 export declare const zhLangData: {

package/dist/types/index.d.ts

@@ -46,3 +46,9 @@ export interface AIBluekingExpose {
     setInputMessage: (val: string) => void;
     quickActions: (type: 'explanation' | 'translate', content: string) => void;
 }
+export type IDocument = {
+    metadata: {
+        file_path: string;
+        path: string;
+    };
+};

package/dist/util/index.d.ts

@@ -30,3 +30,9 @@ export declare const debounce: <T, R>(fn: (P?: T) => R, delay?: number) => (para
  * @returns 处理后的提示词
  */
 export declare const processPromptTemplate: (prompt: string, selectedText: string) => string;
+/**
+ * 格式化时间
+ * @param val 时间
+ * @returns 格式化后的时间
+ */
+export declare function durationFormatter(val: number): string;