@blueking/ai-blueking 0.3.28 → 0.4.0

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,6 +224,7 @@ AI 小鲸支持渲染可点击的交互式 HTML 元素。以下是如何实现
     handleReceiveMessage,
     handleEnd,
     handleError,
+    messages.value,
   );
 
   // 清空消息

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-modal.vue.d.ts

@@ -8,6 +8,7 @@ declare const _default: __VLS_WithTemplateSlots<import("vue").DefineComponent<__
     startPosition?: IStartPosition | undefined;
     models?: IModel[] | undefined;
     model?: string | undefined;
+    loading?: boolean | undefined;
 }>, {
     headBackground: string;
     positionLimit: () => {
@@ -40,6 +41,7 @@ declare const _default: __VLS_WithTemplateSlots<import("vue").DefineComponent<__
     startPosition?: IStartPosition | undefined;
     models?: IModel[] | undefined;
     model?: string | undefined;
+    loading?: boolean | undefined;
 }>, {
     headBackground: string;
     positionLimit: () => {

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

@@ -30,6 +30,7 @@ export declare const langData: {
     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 '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: {
     translateShortcut: string;

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;