@blueking/ai-ui-sdk 0.0.7-beta.7 → 0.0.7-beta.8

package/dist/common/util.d.ts

@@ -17,3 +17,10 @@ export declare function durationFormatter(val: number): string;
  */
 export declare function handleDownLoad(source: string, filename?: string): void;
 export declare const handleCopy: (text: string) => void;
+/**
+ * 处理提示词模板,替换模板中的变量
+ * @param prompt 提示词模板
+ * @param selectedText 选中的文本
+ * @returns 处理后的提示词
+ */
+export declare const processPromptTemplate: (prompt: string, selectedText: string) => string;

package/dist/hooks/use-chat.d.ts

@@ -1,4 +1,4 @@
-import type { ISessionContent, ISession, ISessionPrompt, ChatCallbacks } from '../types/type.ts';
+import type { ISessionContent, ISession, ISessionPrompt, ChatCallbacks, BasicChatContent, ShortcutChatContent } from '../types/type.ts';
 import { SessionContentRole, SessionContentStatus } from '../types/enum';
 type SessionContentsMap = {
     [key: string]: ISessionContent[];
@@ -6,7 +6,7 @@ type SessionContentsMap = {
 type SessionLoadingMap = {
     [key: string]: boolean;
 };
-export declare const useChat: <T extends ISession = ISession>({ handleStart, handleText, handleReferenceDoc, handleThink, handleEnd, handleError, }?: ChatCallbacks) => {
+export declare const useChat: <T extends ISession = ISession>({ handleStart, handleText, handleReferenceDoc, handleThink, handleEnd, handleError, requestOptions, }?: ChatCallbacks) => {
     currentSession: import("vue").Ref<T | undefined, T | undefined>;
     sessionContents: import("vue").Ref<{
         id?: number | undefined;
@@ -21,6 +21,8 @@ export declare const useChat: <T extends ISession = ISession>({ handleStart, han
         rate?: number | undefined;
         status?: SessionContentStatus | undefined;
         sessionCode: string;
+        cite?: string | undefined;
+        time?: string | undefined;
     }[], ISessionContent[] | {
         id?: number | undefined;
         createdAt?: string | undefined;
@@ -34,16 +36,20 @@ export declare const useChat: <T extends ISession = ISession>({ handleStart, han
         rate?: number | undefined;
         status?: SessionContentStatus | undefined;
         sessionCode: string;
+        cite?: string | undefined;
+        time?: string | undefined;
     }[]>;
     sessionContentsMap: SessionContentsMap;
     sessionLoadingMap: import("vue").Ref<SessionLoadingMap, SessionLoadingMap>;
     prompts: import("vue").ComputedRef<ISessionPrompt[]>;
+    currentSessionLoading: import("vue").ComputedRef<boolean>;
     chat: ({ sessionCode, data, url, headers, }: {
         sessionCode: string;
         data?: Record<string, any>;
         url: string;
         headers?: Record<string, string>;
     }) => void;
+    sendChat: (content: BasicChatContent | ShortcutChatContent, callback?: () => void) => void;
     stopChat: (sessionCode: string) => void;
     plusSessionContent: (sessionCode: string, sessionContent: ISessionContent) => void;
     updateSessionContent: (sessionContent: ISessionContent) => void;
@@ -56,5 +62,11 @@ export declare const useChat: <T extends ISession = ISession>({ handleStart, han
     deleteSessionContents: (sessionCode: string, contentIds: number[]) => number[];
     handleStartChat: (sessionCode: string) => void | undefined;
     handleErrorChat: (sessionCode: string, message: string, code?: string) => void | undefined;
+    reGenerateChat: (chatIndex: number) => void;
+    reSendChat: (index: number, { message, cite }: {
+        message: string;
+        cite?: string;
+    }, callback?: () => void) => void;
+    deleteChat: (index: number) => void;
 };
 export {};

package/dist/main.js

@@ -844,6 +844,14 @@ const handleCopy = (text)=>{
     });
     document.body.removeChild(textarea);
 };
+/**
+ * 处理提示词模板,替换模板中的变量
+ * @param prompt 提示词模板
+ * @param selectedText 选中的文本
+ * @returns 处理后的提示词
+ */ const processPromptTemplate = (prompt, selectedText)=>{
+    return prompt.replace(/\{\{\s*SELECTED_TEXT\s*\}\}/g, selectedText || '');
+};
 
 ;// ./src/hooks/use-click-proxy.ts
 
@@ -1434,8 +1442,9 @@ class ChatHelper {
 
 
 
+
 // ai 聊天
-const useChat = ({ handleStart, handleText, handleReferenceDoc, handleThink, handleEnd, handleError } = {})=>{
+const useChat = ({ handleStart, handleText, handleReferenceDoc, handleThink, handleEnd, handleError, requestOptions } = {})=>{
     const startMessage = '内容正在生成中...';
     // 聊天上下文
     const currentSession = (0,external_vue_namespaceObject.ref)();
@@ -1477,6 +1486,12 @@ const useChat = ({ handleStart, handleText, handleReferenceDoc, handleThink, han
         }
         return calculatedSessionContents;
     });
+    // 当前会话是否正在加载
+    const currentSessionLoading = (0,external_vue_namespaceObject.computed)(()=>{
+        var _currentSession_value;
+        const sessionCode = (_currentSession_value = currentSession.value) === null || _currentSession_value === void 0 ? void 0 : _currentSession_value.sessionCode;
+        return sessionCode ? sessionLoadingMap.value[sessionCode] : false;
+    });
     // 计算当前的 prompt
     const prompts = (0,external_vue_namespaceObject.computed)(()=>{
         var _currentSession_value_roleInfo_content, _currentSession_value_roleInfo, _currentSession_value;
@@ -1738,16 +1753,90 @@ const useChat = ({ handleStart, handleText, handleReferenceDoc, handleThink, han
         // 调用cb
         return handleError === null || handleError === void 0 ? void 0 : handleError(sessionCode, sessionContent, code);
     }
+    // 重新生成对话 仅支持重新生成 ai 的回复
+    function reGenerateChat(chatIndex) {
+        const chat = sessionContents.value[chatIndex];
+        if (chat.role !== SessionContentRole.Ai) {
+            return;
+        }
+        const message = sessionContents.value[chatIndex - 1].content;
+        const cite = sessionContents.value[chatIndex - 1].cite;
+        sessionContents.value.splice(chatIndex - 1, 2);
+        sendChat({
+            message,
+            cite
+        });
+    }
     // 聊天
     function chat({ sessionCode, data, url, headers }) {
         // 发送请求
         chatHelper.stream({
             sessionCode,
-            url,
+            url: url,
             data,
-            headers
+            headers: headers || (requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.headers)
         });
     }
+    // 重新发送聊天, 需要将当前聊天内容以及下方的所有内容删除
+    function reSendChat(index, { message, cite }, callback) {
+        const chat = sessionContents.value[index];
+        if (chat.role !== SessionContentRole.User) {
+            return;
+        }
+        sessionContents.value.splice(index, sessionContents.value.length - index);
+        sendChat({
+            message,
+            cite
+        }, callback);
+    }
+    // 发送聊天
+    function sendChat(content, callback) {
+        var _currentSession_value, _currentSession_value1, _currentSession_value2;
+        if (!((_currentSession_value = currentSession.value) === null || _currentSession_value === void 0 ? void 0 : _currentSession_value.sessionCode) || !(requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.url) || currentSessionLoading.value) {
+            return;
+        }
+        const { message, cite, shortcut } = content;
+        let input = '';
+        if (shortcut) {
+            input = processPromptTemplate(shortcut.prompt, cite);
+        } else {
+            input = cite ? `${message}: "${cite}"` : message;
+        }
+        sessionContents.value.push({
+            sessionCode: (_currentSession_value1 = currentSession.value) === null || _currentSession_value1 === void 0 ? void 0 : _currentSession_value1.sessionCode,
+            content: message,
+            role: SessionContentRole.User,
+            status: SessionContentStatus.Success,
+            cite
+        });
+        // 发送请求
+        chatHelper.stream({
+            sessionCode: (_currentSession_value2 = currentSession.value) === null || _currentSession_value2 === void 0 ? void 0 : _currentSession_value2.sessionCode,
+            url: requestOptions.url,
+            headers: requestOptions === null || requestOptions === void 0 ? void 0 : requestOptions.headers,
+            data: {
+                inputs: {
+                    chat_history: prompts.value.slice(0, prompts.value.length - 1),
+                    input
+                }
+            }
+        });
+        callback === null || callback === void 0 ? void 0 : callback();
+    }
+    /**
+   * 删除聊天内容
+   * @param index 要删除的聊天内容索引
+   * @description 
+   * - 如果删除的是用户消息，则同时删除该消息及其对应的AI回复（共2条）
+   * - 如果删除的是其他类型消息（如AI回复），则只删除该条消息
+   */ function deleteChat(index) {
+        const chat = sessionContents.value[index];
+        if (chat.role === SessionContentRole.User) {
+            sessionContents.value.splice(index, 2);
+            return;
+        }
+        sessionContents.value.splice(index - 1, 2);
+    }
     // 停止聊天
     function stopChat(sessionCode) {
         chatHelper.stop(sessionCode);
@@ -1758,7 +1847,9 @@ const useChat = ({ handleStart, handleText, handleReferenceDoc, handleThink, han
         sessionContentsMap,
         sessionLoadingMap,
         prompts,
+        currentSessionLoading,
         chat,
+        sendChat,
         stopChat,
         plusSessionContent,
         updateSessionContent,
@@ -1770,7 +1861,10 @@ const useChat = ({ handleStart, handleText, handleReferenceDoc, handleThink, han
         deleteSessionContent,
         deleteSessionContents,
         handleStartChat,
-        handleErrorChat
+        handleErrorChat,
+        reGenerateChat,
+        reSendChat,
+        deleteChat
     };
 };
 

package/dist/types/type.d.ts

@@ -34,6 +34,8 @@ export interface ISessionContent {
     rate?: number;
     status?: SessionContentStatus;
     sessionCode: string;
+    cite?: string;
+    time?: string;
 }
 export interface ISessionPrompt {
     content: string;
@@ -46,9 +48,33 @@ export interface ChatCallbacks {
     handleThink?: (sessionCode: string, sessionContent: ISessionContent) => void;
     handleEnd?: (sessionCode: string, sessionContent: ISessionContent) => void;
     handleError?: (sessionCode: string, sessionContent: ISessionContent, code: string | undefined) => void;
+    requestOptions?: {
+        url: string;
+        headers?: Record<string, string>;
+    };
 }
 export interface SummaryCallbacks {
     handleStart?: () => void;
     handleEnd?: (text: string) => void;
     handleError?: (message: string, code?: string) => void;
 }
+export type BasicChatContent = {
+    message: string;
+    cite?: string;
+    shortcut?: never;
+};
+export type ShortcutChatContent = {
+    message: string;
+    cite: string;
+    shortcut: ShortCut;
+};
+export interface ShortCut {
+    label: string;
+    key: string;
+    prompt: string;
+    icon?: string;
+}
+export type ISendChat = {
+    content: BasicChatContent | ShortcutChatContent;
+    callback?: () => void;
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blueking/ai-ui-sdk",
-  "version": "0.0.7-beta.7",
+  "version": "0.0.7-beta.8",
   "description": "蓝鲸AI UI SDK",
   "module": "dist/main.js",
   "types": "dist/main.d.ts",