@blueking/chat-helper 0.0.1-beta.9 → 0.0.3

package/dist/session/use-session.ts.js

@@ -51,13 +51,40 @@ function _async_to_generator(fn) {
         });
     };
 }
+function _define_property(obj, key, value) {
+    if (key in obj) {
+        Object.defineProperty(obj, key, {
+            value: value,
+            enumerable: true,
+            configurable: true,
+            writable: true
+        });
+    } else {
+        obj[key] = value;
+    }
+    return obj;
+}
+function _object_spread(target) {
+    for(var i = 1; i < arguments.length; i++){
+        var source = arguments[i] != null ? arguments[i] : {};
+        var ownKeys = Object.keys(source);
+        if (typeof Object.getOwnPropertySymbols === "function") {
+            ownKeys = ownKeys.concat(Object.getOwnPropertySymbols(source).filter(function(sym) {
+                return Object.getOwnPropertyDescriptor(source, sym).enumerable;
+            }));
+        }
+        ownKeys.forEach(function(key) {
+            _define_property(target, key, source[key]);
+        });
+    }
+    return target;
+}
 import { ref } from 'vue';
 /**
  * 使用会话模块，主要做业务的组合
- * @param http - HTTP 模块
- * @param message - 消息模块
+ * @param mediator - 中介者模块，用于获取其他模块的引用
  * @returns 会话模块
- */ export const useSession = (http, message, agent)=>{
+ */ export const useSession = (mediator)=>{
     const list = ref([]);
     const current = ref(null);
     const isCurrentLoading = ref(false);
@@ -65,67 +92,198 @@ import { ref } from 'vue';
     const isCreateLoading = ref(false);
     const isUpdateLoading = ref(false);
     const isDeleteLoading = ref(false);
+    const isBatchDeleteLoading = ref(false);
+    const isRenameLoading = ref(false);
+    /**
+   * 更新 list 中的 session，并同步更新 current
+   * @param session - Partial session
+   */ const updateSessionInList = (session)=>{
+        var _current_value;
+        // 先更新 list 中对应的 session
+        list.value = list.value.map((item)=>item.sessionCode === session.sessionCode ? _object_spread({}, item, session) : item);
+        // 如果 current 是被更新的 session，则让 current 重新指向 list 中更新后的对象
+        if (((_current_value = current.value) === null || _current_value === void 0 ? void 0 : _current_value.sessionCode) === session.sessionCode) {
+            var _list_value_find;
+            current.value = (_list_value_find = list.value.find((item)=>item.sessionCode === session.sessionCode)) !== null && _list_value_find !== void 0 ? _list_value_find : current.value;
+        }
+    };
     const getSessions = ()=>{
+        var _mediator_http;
         isListLoading.value = true;
-        return http.session.getSessions().then((res)=>{
+        return (_mediator_http = mediator.http) === null || _mediator_http === void 0 ? void 0 : _mediator_http.session.getSessions().then((res)=>{
             list.value = res;
         }).finally(()=>{
             isListLoading.value = false;
         });
     };
-    const chooseSession = function() {
-        var _ref = _async_to_generator(function*(sessionCode) {
-            yield agent.stopChat();
+    /**
+   * 切换会话
+   * @param sessionCode - 会话编码
+   * @param options - 选项
+   * @param options.loadMessages - 是否加载消息列表，默认 true。新创建的会话可设为 false 跳过加载
+   */ const chooseSession = function() {
+        var _ref = _async_to_generator(function*(sessionCode, options) {
+            var // 中止当前聊天
+            _mediator_agent;
+            (_mediator_agent = mediator.agent) === null || _mediator_agent === void 0 ? void 0 : _mediator_agent.abortChat();
             var _list_value_find;
             // 选择会话
             current.value = (_list_value_find = list.value.find((item)=>item.sessionCode === sessionCode)) !== null && _list_value_find !== void 0 ? _list_value_find : null;
-            // 获取会话内容
-            message.getMessages(sessionCode);
+            // 默认加载消息，但允许跳过（新会话消息必定为空，无需加载）
+            if ((options === null || options === void 0 ? void 0 : options.loadMessages) !== false) {
+                var _mediator_message, // 继续聊天
+                _mediator_agent1;
+                // 获取会话内容
+                yield (_mediator_message = mediator.message) === null || _mediator_message === void 0 ? void 0 : _mediator_message.getMessages(sessionCode);
+                (_mediator_agent1 = mediator.agent) === null || _mediator_agent1 === void 0 ? void 0 : _mediator_agent1.resumeStreamingChat(sessionCode);
+            } else {
+                // 新会话，清空消息列表
+                if (mediator.message) {
+                    mediator.message.list.value = [];
+                }
+            }
         });
-        return function chooseSession(sessionCode) {
+        return function chooseSession(sessionCode, options) {
             return _ref.apply(this, arguments);
         };
     }();
     const getSession = (sessionCode)=>{
+        var _mediator_http;
         isCurrentLoading.value = true;
-        return http.session.getSession(sessionCode).then((res)=>{
+        return (_mediator_http = mediator.http) === null || _mediator_http === void 0 ? void 0 : _mediator_http.session.getSession(sessionCode).then((res)=>{
             current.value = res;
         }).finally(()=>{
             isCurrentLoading.value = false;
         });
     };
-    const createSession = (session)=>{
-        isCreateLoading.value = true;
-        return http.session.plusSession(session).then((res)=>{
-            list.value.push(current.value);
-            chooseSession(res.sessionCode);
-        }).finally(()=>{
-            isCreateLoading.value = false;
+    /**
+   * 创建新会话
+   * @param session - 会话数据
+   * @param options - 选项
+   * @param options.loadMessages - 创建后是否加载消息列表，默认 false（新会话消息必定为空）
+   */ const createSession = function() {
+        var _ref = _async_to_generator(function*(session, options) {
+            isCreateLoading.value = true;
+            try {
+                var _mediator_http;
+                const res = yield (_mediator_http = mediator.http) === null || _mediator_http === void 0 ? void 0 : _mediator_http.session.plusSession(session);
+                if (res) {
+                    list.value.unshift(res); // 新会话插入到列表头部
+                    var _options_loadMessages;
+                    // 新会话默认不加载消息（消息必定为空），优化 UX
+                    yield chooseSession(res.sessionCode, {
+                        loadMessages: (_options_loadMessages = options === null || options === void 0 ? void 0 : options.loadMessages) !== null && _options_loadMessages !== void 0 ? _options_loadMessages : false
+                    });
+                }
+            } finally{
+                isCreateLoading.value = false;
+            }
         });
-    };
+        return function createSession(session, options) {
+            return _ref.apply(this, arguments);
+        };
+    }();
     const updateSession = (session)=>{
+        var _mediator_http;
         isUpdateLoading.value = true;
-        return http.session.modifySession(session).then((res)=>{
-            current.value = res;
-            list.value = list.value.map((item)=>item.sessionCode === session.sessionCode ? current.value : item);
+        return (_mediator_http = mediator.http) === null || _mediator_http === void 0 ? void 0 : _mediator_http.session.modifySession(session).then((res)=>{
+            updateSessionInList(res);
         }).finally(()=>{
             isUpdateLoading.value = false;
         });
     };
-    const deleteSession = (sessionCode)=>{
-        isDeleteLoading.value = true;
-        return http.session.deleteSession(sessionCode).then(()=>{
-            var _current_value;
-            list.value = list.value.filter((item)=>item.sessionCode !== sessionCode);
-            // 如果当前会话被删除，选择第一个会话
-            if (((_current_value = current.value) === null || _current_value === void 0 ? void 0 : _current_value.sessionCode) === sessionCode) {
-                var _list_value_;
-                chooseSession((_list_value_ = list.value[0]) === null || _list_value_ === void 0 ? void 0 : _list_value_.sessionCode);
+    const deleteSession = function() {
+        var _ref = _async_to_generator(function*(sessionCode) {
+            isDeleteLoading.value = true;
+            try {
+                var _mediator_http, _current_value, _list_value_;
+                yield (_mediator_http = mediator.http) === null || _mediator_http === void 0 ? void 0 : _mediator_http.session.deleteSession(sessionCode);
+                list.value = list.value.filter((item)=>item.sessionCode !== sessionCode);
+                // 如果当前会话被删除，选择第一个会话
+                if (((_current_value = current.value) === null || _current_value === void 0 ? void 0 : _current_value.sessionCode) === sessionCode && ((_list_value_ = list.value[0]) === null || _list_value_ === void 0 ? void 0 : _list_value_.sessionCode)) {
+                    yield chooseSession(list.value[0].sessionCode);
+                }
+            } finally{
+                isDeleteLoading.value = false;
+            }
+        });
+        return function deleteSession(sessionCode) {
+            return _ref.apply(this, arguments);
+        };
+    }();
+    /**
+   * 批量删除会话
+   * @param sessionCodes - 要删除的会话编码列表
+   */ const batchDeleteSessions = function() {
+        var _ref = _async_to_generator(function*(sessionCodes) {
+            if (sessionCodes.length === 0) return;
+            isBatchDeleteLoading.value = true;
+            try {
+                var _mediator_http;
+                yield (_mediator_http = mediator.http) === null || _mediator_http === void 0 ? void 0 : _mediator_http.session.batchDeleteSessions(sessionCodes);
+                const deletedSet = new Set(sessionCodes);
+                list.value = list.value.filter((item)=>!deletedSet.has(item.sessionCode));
+                if (current.value && deletedSet.has(current.value.sessionCode)) {
+                    if (list.value.length > 0) {
+                        yield chooseSession(list.value[0].sessionCode);
+                    } else {
+                        current.value = null;
+                        if (mediator.message) {
+                            mediator.message.list.value = [];
+                        }
+                    }
+                }
+            } finally{
+                isBatchDeleteLoading.value = false;
             }
+        });
+        return function batchDeleteSessions(sessionCodes) {
+            return _ref.apply(this, arguments);
+        };
+    }();
+    /**
+   * 提交会话反馈，sessionContentIds 只需要传 user 对应的 sessionContentId
+   * @param data - ISessionFeedback
+   */ const postSessionFeedback = (data)=>{
+        var _mediator_http;
+        return (_mediator_http = mediator.http) === null || _mediator_http === void 0 ? void 0 : _mediator_http.session.postSessionFeedback(data);
+    };
+    // 获取会话反馈原因
+    const getSessionFeedbackReasons = (rate)=>{
+        var _mediator_http;
+        return (_mediator_http = mediator.http) === null || _mediator_http === void 0 ? void 0 : _mediator_http.session.getSessionFeedbackReasons(rate);
+    };
+    // 会话重命名
+    const renameSession = (sessionCode)=>{
+        var _mediator_http;
+        isRenameLoading.value = true;
+        return (_mediator_http = mediator.http) === null || _mediator_http === void 0 ? void 0 : _mediator_http.session.renameSession(sessionCode).then((res)=>{
+            updateSessionInList({
+                sessionCode: res.sessionCode,
+                sessionName: res.sessionName
+            });
         }).finally(()=>{
-            isDeleteLoading.value = false;
+            isRenameLoading.value = false;
         });
     };
+    // 上传文件到会话
+    const uploadFile = (sessionCode, file)=>{
+        var _mediator_http;
+        return (_mediator_http = mediator.http) === null || _mediator_http === void 0 ? void 0 : _mediator_http.session.uploadFile(sessionCode, file);
+    };
+    /**
+   * 重置 session 模块状态
+   */ const reset = ()=>{
+        list.value = [];
+        current.value = null;
+        isCurrentLoading.value = false;
+        isListLoading.value = false;
+        isCreateLoading.value = false;
+        isUpdateLoading.value = false;
+        isDeleteLoading.value = false;
+        isBatchDeleteLoading.value = false;
+        isRenameLoading.value = false;
+    };
     return {
         list,
         current,
@@ -134,11 +292,18 @@ import { ref } from 'vue';
         isCreateLoading,
         isUpdateLoading,
         isDeleteLoading,
+        isBatchDeleteLoading,
         getSessions,
         chooseSession,
         getSession,
         createSession,
         updateSession,
-        deleteSession
+        deleteSession,
+        batchDeleteSessions,
+        postSessionFeedback,
+        getSessionFeedbackReasons,
+        renameSession,
+        uploadFile,
+        reset
     };
 };

package/dist/type.d.ts

@@ -2,10 +2,12 @@ import type { IRequestConfig, ISSEProtocol } from './http/fetch';
 import type { IResponse } from './http/fetch';
 export interface IUseChatHelperOptions {
     protocol?: ISSEProtocol;
+    /** 自定义拦截器，优先级高于内置 requestData 拦截器 */
     interceptors?: {
         request?: (config: IRequestConfig) => IRequestConfig;
         response?: (response: IResponse) => IResponse;
     };
+    /** 全局默认请求配置，通过内置拦截器自动合并到每次请求（函数形式可延迟求值） */
     requestData: {
         data?: (() => Record<string, unknown>) | Record<string, unknown>;
         headers?: (() => Record<string, string>) | Record<string, string>;

package/package.json

@@ -1,20 +1,22 @@
 {
   "name": "@blueking/chat-helper",
-  "version": "0.0.1-beta.9",
+  "version": "0.0.3",
   "description": "",
   "main": "./dist/index.ts.js",
-  "scripts": {
-    "build": "bk-cli-service build && vue-tsc --emitDeclarationOnly"
-  },
+  "types": "./dist/index.d.ts",
   "keywords": [
     "chat-helper"
   ],
   "files": [
     "dist"
   ],
-  "author": "",
+  "author": "Tencent BlueKing",
   "license": "ISC",
-  "packageManager": "pnpm@10.17.1",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/TencentBlueKing/bk-aidev-agent.git",
+    "directory": "src/frontend/ai-blueking/packages/chat-helper"
+  },
   "peerDependencies": {
     "vue": "^3.5.24"
   },
@@ -23,5 +25,8 @@
     "eslint-plugin-simple-import-sort": "^10.0.0",
     "typescript": "^5.5.4",
     "vue-tsc": "^3.1.4"
+  },
+  "scripts": {
+    "build": "bk-cli-service build && vue-tsc --emitDeclarationOnly"
   }
 }