@blueking/chat-helper 0.0.1-beta.10 → 0.0.1-beta.12

package/dist/agent/use-agent.d.ts

@@ -537,7 +537,8 @@ export declare const useAgent: (mediator: IMediatorModule, protocol: ISSEProtoco
     chat: (userInput: IUserMessage["content"], sessionCode: string, url?: string, config?: IRequestConfig, property?: IMessageProperty) => Promise<void>;
     resendMessage: (messageId: string, sessionCode: string, newContent?: string, url?: string, config?: IRequestConfig) => Promise<void>;
     resumeStreamingChat: (sessionCode: string, url?: string, config?: IRequestConfig) => void;
-    stopChat: () => Promise<void>;
+    abortChat: () => void;
+    stopChat: (sessionCode: string) => Promise<void>;
     getAgentInfo: () => Promise<void>;
     reset: (protocol: ISSEProtocol) => void;
 };

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

@@ -90,8 +90,8 @@ import { MessageRole, MessageStatus } from '../message/index.ts.js';
     const info = ref(null);
     const isInfoLoading = ref(false);
     const isChatting = ref(false);
-    let chatController = null;
     let usedProtocol = protocol || new AGUIProtocol();
+    let abortController = null;
     const getAgentInfo = ()=>{
         var _mediator_http;
         isInfoLoading.value = true;
@@ -105,8 +105,6 @@ import { MessageRole, MessageStatus } from '../message/index.ts.js';
         var _ref = _async_to_generator(function*(sessionCode, url, config) {
             var // 发起聊天
             _mediator_http;
-            // 创建一个 AbortController 实例
-            chatController = new AbortController();
             // ag-ui 协议需要注入消息模块
             if (usedProtocol instanceof AGUIProtocol) {
                 usedProtocol.injectMessageModule(mediator.message);
@@ -131,13 +129,15 @@ import { MessageRole, MessageStatus } from '../message/index.ts.js';
                 isChatting.value = true;
                 (_usedProtocol_onStart = usedProtocol.onStart) === null || _usedProtocol_onStart === void 0 ? void 0 : _usedProtocol_onStart.call(usedProtocol);
             };
+            // 创建 AbortController
+            abortController = new AbortController();
             (_mediator_http = mediator.http) === null || _mediator_http === void 0 ? void 0 : _mediator_http.fetchClient.streamRequest(_object_spread({
                 url: url || 'chat_completion/',
                 method: 'POST',
                 data: {
                     session_code: sessionCode
                 },
-                controller: chatController,
+                controller: abortController,
                 onDone,
                 onError,
                 onMessage,
@@ -176,22 +176,33 @@ import { MessageRole, MessageStatus } from '../message/index.ts.js';
     }();
     /**
    * 恢复流式聊天
-   * 如果最后一条消息处于流式传输中，重新建立连接
+   * 如果最后一条消息处于流式传输中或是用户消息，重新建立连接
    * @param sessionCode - 会话代码
    * @param url - 请求 URL（可选）
    * @param config - 请求配置（可选）
    */ const resumeStreamingChat = (sessionCode, url, config)=>{
-        var _mediator_message_list_value_at, _mediator_message;
-        if (((_mediator_message = mediator.message) === null || _mediator_message === void 0 ? void 0 : (_mediator_message_list_value_at = _mediator_message.list.value.at(-1)) === null || _mediator_message_list_value_at === void 0 ? void 0 : _mediator_message_list_value_at.status) === MessageStatus.Streaming) {
+        var _mediator_message;
+        const lastMessage = (_mediator_message = mediator.message) === null || _mediator_message === void 0 ? void 0 : _mediator_message.list.value.at(-1);
+        if ((lastMessage === null || lastMessage === void 0 ? void 0 : lastMessage.status) === MessageStatus.Streaming || (lastMessage === null || lastMessage === void 0 ? void 0 : lastMessage.role) === MessageRole.User) {
             streamRequest(sessionCode, url, config);
         }
     };
-    const stopChat = function() {
-        var _ref = _async_to_generator(function*() {
-            chatController === null || chatController === void 0 ? void 0 : chatController.abort();
-            chatController = null;
+    /**
+   * 中止聊天（纯前端中止，后端继续处理）
+   */ const abortChat = ()=>{
+        var _abortController_abort;
+        abortController === null || abortController === void 0 ? void 0 : (_abortController_abort = abortController.abort) === null || _abortController_abort === void 0 ? void 0 : _abortController_abort.call(abortController);
+        abortController = null;
+    };
+    /**
+   * 停止会话，后端中止
+   * @param sessionCode - 会话代码
+   */ const stopChat = function() {
+        var _ref = _async_to_generator(function*(sessionCode) {
+            var _mediator_http_message;
+            return (_mediator_http_message = mediator.http.message) === null || _mediator_http_message === void 0 ? void 0 : _mediator_http_message.stopChat(sessionCode);
         });
-        return function stopChat() {
+        return function stopChat(sessionCode) {
             return _ref.apply(this, arguments);
         };
     }();
@@ -250,8 +261,6 @@ import { MessageRole, MessageStatus } from '../message/index.ts.js';
         };
     }();
     const reset = (protocol)=>{
-        // 停止正在进行的聊天
-        stopChat();
         // 重置状态
         usedProtocol = protocol || new AGUIProtocol();
         info.value = null;
@@ -264,6 +273,7 @@ import { MessageRole, MessageStatus } from '../message/index.ts.js';
         chat,
         resendMessage,
         resumeStreamingChat,
+        abortChat,
         stopChat,
         getAgentInfo,
         reset

package/dist/event/ag-ui.ts.js

@@ -336,6 +336,12 @@ import { CustomEventName, EventType } from './type.ts.js';
     onError(error) {
         var // 回调给上层
         _this_onErrorCallback, _this;
+        // 创建一个错误消息
+        this.messageModule.plusMessage({
+            role: MessageRole.Assistant,
+            content: error.message,
+            status: MessageStatus.Error
+        });
         (_this_onErrorCallback = (_this = this).onErrorCallback) === null || _this_onErrorCallback === void 0 ? void 0 : _this_onErrorCallback.call(_this, error);
     }
     onMessage(message) {

package/dist/http/fetch/fetch.ts.js

@@ -415,6 +415,7 @@ export class FetchClient {
                             var _config_onMessage;
                             const json = JSON.parse(item);
                             (_config_onMessage = config.onMessage) === null || _config_onMessage === void 0 ? void 0 : _config_onMessage.call(config, json);
+                            temp = '';
                         } else if (item) {
                             temp = item;
                         }

package/dist/http/index.d.ts

@@ -28,6 +28,9 @@ export declare const useHttp: (options: IUseChatHelperOptions) => {
         }>;
         getSessionFeedbackReasons: (rate: number, config?: import("./fetch").IRequestConfig) => Promise<string[]>;
         renameSession: (sessionCode: string, config?: import("./fetch").IRequestConfig) => Promise<import("..").ISession<unknown, unknown>>;
+        uploadFile: (sessionCode: string, file: File, config?: import("./fetch").IRequestConfig) => Promise<{
+            download_url?: string;
+        }>;
     };
     message: {
         getMessages: (sessionCode: string, limit?: number, config?: import("./fetch").IRequestConfig) => Promise<import("..").IMessage[]>;
@@ -36,6 +39,7 @@ export declare const useHttp: (options: IUseChatHelperOptions) => {
         deleteMessage: (id: string, config?: import("./fetch").IRequestConfig) => Promise<import("..").IMessage>;
         batchDeleteMessages: (ids: string[], config?: import("./fetch").IRequestConfig) => Promise<number>;
         shareMessages: (sessionCode: string, contentIds: string[], expiredAt?: number, config?: import("./fetch").IRequestConfig) => Promise<import("./module/message").IShareMessagesResponse>;
+        stopChat: (sessionCode: string, config?: import("./fetch").IRequestConfig) => Promise<void>;
     };
     fetchClient: import("./fetch").FetchClient;
 };

package/dist/http/module/index.d.ts

@@ -25,6 +25,9 @@ export declare const useModule: (fetchClient: FetchClient) => {
         }>;
         getSessionFeedbackReasons: (rate: number, config?: import("../fetch").IRequestConfig) => Promise<string[]>;
         renameSession: (sessionCode: string, config?: import("../fetch").IRequestConfig) => Promise<import("../..").ISession<unknown, unknown>>;
+        uploadFile: (sessionCode: string, file: File, config?: import("../fetch").IRequestConfig) => Promise<{
+            download_url?: string;
+        }>;
     };
     message: {
         getMessages: (sessionCode: string, limit?: number, config?: import("../fetch").IRequestConfig) => Promise<import("../..").IMessage[]>;
@@ -33,6 +36,7 @@ export declare const useModule: (fetchClient: FetchClient) => {
         deleteMessage: (id: string, config?: import("../fetch").IRequestConfig) => Promise<import("../..").IMessage>;
         batchDeleteMessages: (ids: string[], config?: import("../fetch").IRequestConfig) => Promise<number>;
         shareMessages: (sessionCode: string, contentIds: string[], expiredAt?: number, config?: import("../fetch").IRequestConfig) => Promise<import("./message").IShareMessagesResponse>;
+        stopChat: (sessionCode: string, config?: import("../fetch").IRequestConfig) => Promise<void>;
     };
     fetchClient: FetchClient;
 };

package/dist/http/module/message.d.ts

@@ -25,4 +25,5 @@ export declare const useMessage: (fetchClient: FetchClient) => {
     deleteMessage: (id: string, config?: IRequestConfig) => Promise<IMessage>;
     batchDeleteMessages: (ids: string[], config?: IRequestConfig) => Promise<number>;
     shareMessages: (sessionCode: string, contentIds: string[], expiredAt?: number, config?: IRequestConfig) => Promise<IShareMessagesResponse>;
+    stopChat: (sessionCode: string, config?: IRequestConfig) => Promise<void>;
 };

package/dist/http/module/message.ts.js

@@ -66,7 +66,7 @@ import { transferMessage2MessageApi, transferMessageApi2Message } from '../trans
     // 修改会话内容
     const modifyMessage = (data, config)=>fetchClient.put(`session_content/${data.id}/`, transferMessage2MessageApi(data), config).then(transferMessageApi2Message);
     // 删除会话内容
-    const deleteMessage = (id, config)=>fetchClient.delete(`session_content/${id}/`, undefined, config);
+    const deleteMessage = (id, config)=>fetchClient['delete'](`session_content/${id}/`, undefined, config);
     // 批量删除聊天内容
     const batchDeleteMessages = (ids, config)=>fetchClient.post(`session_content/batch_delete/`, {
             ids
@@ -78,12 +78,17 @@ import { transferMessage2MessageApi, transferMessageApi2Message } from '../trans
         }, expiredAt ? {
             expired_at: expiredAt
         } : {}), config);
+    // 停止会话
+    const stopChat = (sessionCode, config)=>fetchClient.post(`session_content/stop/`, {
+            session_code: sessionCode
+        }, config);
     return {
         getMessages,
         plusMessage,
         modifyMessage,
         deleteMessage,
         batchDeleteMessages,
-        shareMessages
+        shareMessages,
+        stopChat
     };
 };

package/dist/http/module/session.d.ts

@@ -22,4 +22,7 @@ export declare const useSession: (fetchClient: FetchClient) => {
     }>;
     getSessionFeedbackReasons: (rate: number, config?: IRequestConfig) => Promise<string[]>;
     renameSession: (sessionCode: string, config?: IRequestConfig) => Promise<ISession<unknown, unknown>>;
+    uploadFile: (sessionCode: string, file: File, config?: IRequestConfig) => Promise<{
+        download_url?: string;
+    }>;
 };

package/dist/http/module/session.ts.js

@@ -22,7 +22,59 @@
  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF
  * CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  * IN THE SOFTWARE.
- */ import { transferSession2SessionApi, transferSessionApi2Session, transferSessionFeedback2SessionFeedbackApi, transferSessionFeedbackApi2SessionFeedback } from '../transform/session.ts.js';
+ */ 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;
+}
+function ownKeys(object, enumerableOnly) {
+    var keys = Object.keys(object);
+    if (Object.getOwnPropertySymbols) {
+        var symbols = Object.getOwnPropertySymbols(object);
+        if (enumerableOnly) {
+            symbols = symbols.filter(function(sym) {
+                return Object.getOwnPropertyDescriptor(object, sym).enumerable;
+            });
+        }
+        keys.push.apply(keys, symbols);
+    }
+    return keys;
+}
+function _object_spread_props(target, source) {
+    source = source != null ? source : {};
+    if (Object.getOwnPropertyDescriptors) {
+        Object.defineProperties(target, Object.getOwnPropertyDescriptors(source));
+    } else {
+        ownKeys(Object(source)).forEach(function(key) {
+            Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key));
+        });
+    }
+    return target;
+}
+import { transferSession2SessionApi, transferSessionApi2Session, transferSessionFeedback2SessionFeedbackApi, transferSessionFeedbackApi2SessionFeedback } from '../transform/session.ts.js';
 /**
  * session 相关 http 接口
  * @param fetchClient - 请求客户端
@@ -37,7 +89,7 @@
     // 修改会话
     const modifySession = (data, config)=>fetchClient.put(`session/${data.sessionCode}/`, transferSession2SessionApi(data), config).then((res)=>transferSessionApi2Session(res));
     // 删除会话
-    const deleteSession = (sessionCode, config)=>fetchClient['delete'](`session/${sessionCode}/`, undefined, config);
+    const deleteSession = (sessionCode, config)=>fetchClient.delete(`session/${sessionCode}/`, undefined, config);
     // 批量删除会话
     const batchDeleteSessions = (sessionCodes, config)=>fetchClient.post(`session/batch_delete/`, {
             session_codes: sessionCodes
@@ -52,6 +104,15 @@
         }, config);
     // 会话重命名
     const renameSession = (sessionCode, config)=>fetchClient.post(`session/${sessionCode}/ai_rename/`, undefined, config).then((res)=>transferSessionApi2Session(res));
+    // 上传文件到会话
+    const uploadFile = (sessionCode, file, config)=>{
+        const fileName = encodeURIComponent(file.name);
+        return file.arrayBuffer().then((content)=>fetchClient.post(`session/${sessionCode}/upload/${fileName}/`, content, _object_spread_props(_object_spread({}, config), {
+                headers: {
+                    'Content-Disposition': `attachment; filename="${file.name}"`
+                }
+            })));
+    };
     return {
         clearSession,
         getSessions,
@@ -62,6 +123,7 @@
         getSession,
         postSessionFeedback,
         getSessionFeedbackReasons,
-        renameSession
+        renameSession,
+        uploadFile
     };
 };

package/dist/index.d.ts

@@ -539,7 +539,8 @@ export declare const useChatHelper: (options: IUseChatHelperOptions) => {
         chat: (userInput: import("./message").IUserMessage["content"], sessionCode: string, url?: string, config?: import("./http").IRequestConfig, property?: import("./message").IMessageProperty) => Promise<void>;
         resendMessage: (messageId: string, sessionCode: string, newContent?: string, url?: string, config?: import("./http").IRequestConfig) => Promise<void>;
         resumeStreamingChat: (sessionCode: string, url?: string, config?: import("./http").IRequestConfig) => void;
-        stopChat: () => Promise<void>;
+        abortChat: () => void;
+        stopChat: (sessionCode: string) => Promise<void>;
         getAgentInfo: () => Promise<void>;
         reset: (protocol: import("./http").ISSEProtocol) => void;
     };
@@ -2434,6 +2435,9 @@ export declare const useChatHelper: (options: IUseChatHelperOptions) => {
         }>;
         getSessionFeedbackReasons: (rate: number) => Promise<string[]>;
         renameSession: (sessionCode: string) => Promise<void>;
+        uploadFile: (sessionCode: string, file: File) => Promise<{
+            download_url?: string;
+        }>;
         reset: () => void;
     };
     message: {
@@ -3352,6 +3356,9 @@ export declare const useChatHelper: (options: IUseChatHelperOptions) => {
             }>;
             getSessionFeedbackReasons: (rate: number, config?: import("./http").IRequestConfig) => Promise<string[]>;
             renameSession: (sessionCode: string, config?: import("./http").IRequestConfig) => Promise<import("./session").ISession<unknown, unknown>>;
+            uploadFile: (sessionCode: string, file: File, config?: import("./http").IRequestConfig) => Promise<{
+                download_url?: string;
+            }>;
         };
         message: {
             getMessages: (sessionCode: string, limit?: number, config?: import("./http").IRequestConfig) => Promise<import("./message").IMessage[]>;
@@ -3360,6 +3367,7 @@ export declare const useChatHelper: (options: IUseChatHelperOptions) => {
             deleteMessage: (id: string, config?: import("./http").IRequestConfig) => Promise<import("./message").IMessage>;
             batchDeleteMessages: (ids: string[], config?: import("./http").IRequestConfig) => Promise<number>;
             shareMessages: (sessionCode: string, contentIds: string[], expiredAt?: number, config?: import("./http").IRequestConfig) => Promise<import("./http/module/message").IShareMessagesResponse>;
+            stopChat: (sessionCode: string, config?: import("./http").IRequestConfig) => Promise<void>;
         };
         fetchClient: import("./http").FetchClient;
     };

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

@@ -1896,6 +1896,9 @@ export declare const useSession: (mediator: IMediatorModule) => {
     }>;
     getSessionFeedbackReasons: (rate: number) => Promise<string[]>;
     renameSession: (sessionCode: string) => Promise<void>;
+    uploadFile: (sessionCode: string, file: File) => Promise<{
+        download_url?: string;
+    }>;
     reset: () => void;
 };
 export type ISessionModule = ReturnType<typeof useSession>;

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

@@ -122,8 +122,9 @@ import { ref } from 'vue';
    * @param options.loadMessages - 是否加载消息列表，默认 true。新创建的会话可设为 false 跳过加载
    */ const chooseSession = function() {
         var _ref = _async_to_generator(function*(sessionCode, options) {
-            var _mediator_agent;
-            yield (_mediator_agent = mediator.agent) === null || _mediator_agent === void 0 ? void 0 : _mediator_agent.stopChat();
+            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;
@@ -234,6 +235,11 @@ import { ref } from 'vue';
             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 = ()=>{
@@ -263,6 +269,7 @@ import { ref } from 'vue';
         postSessionFeedback,
         getSessionFeedbackReasons,
         renameSession,
+        uploadFile,
         reset
     };
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blueking/chat-helper",
-  "version": "0.0.1-beta.10",
+  "version": "0.0.1-beta.12",
   "description": "",
   "main": "./dist/index.ts.js",
   "types": "./dist/index.d.ts",