@blueking/chat-helper 0.0.1-beta.2 → 0.0.1-beta.20

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

@@ -51,52 +51,245 @@ 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';
 import { AGUIProtocol } from '../event/index.ts.js';
-import { useFetch } from '../http/fetch/index.ts.js';
-export const useAgent = (options, http, message)=>{
-    const { fetchClient } = useFetch(options);
+import { MessageRole, MessageStatus } from '../message/index.ts.js';
+/**
+ * Agent 模块
+ * @param options - 配置选项
+ * @param mediator - 中介者模块，用于获取其他模块的引用
+ */ export const useAgent = (mediator, protocol)=>{
     const info = ref(null);
     const isInfoLoading = ref(false);
-    let chatController = null;
+    const isChatting = ref(false);
+    let usedProtocol = protocol || new AGUIProtocol();
+    let abortController = null;
     const getAgentInfo = ()=>{
+        var _mediator_http;
         isInfoLoading.value = true;
-        return http.agent.getAgentInfo().then((res)=>{
+        return (_mediator_http = mediator.http) === null || _mediator_http === void 0 ? void 0 : _mediator_http.agent.getAgentInfo().then((res)=>{
             info.value = res;
-        }).finally(()=>{
+        })['finally'](()=>{
             isInfoLoading.value = false;
         });
     };
-    const chat = ()=>{
-        chatController = new AbortController();
-        const protocol = options.protocol || new AGUIProtocol();
-        // ag-ui 协议需要注入消息模块
-        if (protocol instanceof AGUIProtocol) {
-            protocol.injectMessageModule(message);
+    // 处理角色消息
+    const handleRole = (data)=>{
+        var _data_promptSetting_collectionContent, _data_promptSetting;
+        const lastRoleMessage = (_data_promptSetting = data.promptSetting) === null || _data_promptSetting === void 0 ? void 0 : (_data_promptSetting_collectionContent = _data_promptSetting.collectionContent) === null || _data_promptSetting_collectionContent === void 0 ? void 0 : _data_promptSetting_collectionContent.at(-1);
+        if ((lastRoleMessage === null || lastRoleMessage === void 0 ? void 0 : lastRoleMessage.role) === MessageRole.Pause) {
+            var _mediator_message;
+            (_mediator_message = mediator.message) === null || _mediator_message === void 0 ? void 0 : _mediator_message.createAndPlusMessage({
+                role: MessageRole.Assistant,
+                content: lastRoleMessage.content,
+                status: MessageStatus.Complete
+            });
         }
-        fetchClient.streamRequest({
-            url: 'agent/chat/',
-            controller: chatController,
-            onDone: protocol.onDone,
-            onError: protocol.onError,
-            onMessage: protocol.onMessage,
-            onStart: protocol.onStart
+    };
+    const streamRequest = function() {
+        var _ref = _async_to_generator(function*(sessionCode, url, config) {
+            var // 发起聊天
+            _mediator_http;
+            // ag-ui 协议需要注入消息模块
+            if (usedProtocol instanceof AGUIProtocol) {
+                usedProtocol.injectMessageModule(mediator.message);
+            }
+            // 事件代理
+            const onDone = ()=>{
+                var _usedProtocol_onDone;
+                isChatting.value = false;
+                (_usedProtocol_onDone = usedProtocol.onDone) === null || _usedProtocol_onDone === void 0 ? void 0 : _usedProtocol_onDone.call(usedProtocol);
+            };
+            const onError = (error)=>{
+                var _usedProtocol_onError;
+                isChatting.value = false;
+                (_usedProtocol_onError = usedProtocol.onError) === null || _usedProtocol_onError === void 0 ? void 0 : _usedProtocol_onError.call(usedProtocol, error);
+            };
+            const onMessage = (event)=>{
+                var _usedProtocol_onMessage;
+                (_usedProtocol_onMessage = usedProtocol.onMessage) === null || _usedProtocol_onMessage === void 0 ? void 0 : _usedProtocol_onMessage.call(usedProtocol, event);
+            };
+            const onStart = ()=>{
+                var _usedProtocol_onStart;
+                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: abortController,
+                onDone,
+                onError,
+                onMessage,
+                onStart
+            }, config));
+        });
+        return function streamRequest(sessionCode, url, config) {
+            return _ref.apply(this, arguments);
+        };
+    }();
+    /**
+   * 发送聊天消息
+   * @param userInput - 用户输入的消息内容
+   * @param sessionCode - 会话代码
+   * @param url - 请求 URL（可选）
+   * @param config - 请求配置（可选）
+   * @param property - 消息属性，用于传递引用内容或快捷键相关信息（可选）
+   */ const chat = function() {
+        var _ref = _async_to_generator(function*(userInput, sessionCode, url, config, property) {
+            var _mediator_message;
+            // 先新增一个 message
+            yield (_mediator_message = mediator.message) === null || _mediator_message === void 0 ? void 0 : _mediator_message.createAndPlusMessage(_object_spread({
+                role: MessageRole.User,
+                content: userInput,
+                status: MessageStatus.Complete,
+                sessionCode
+            }, property && {
+                property
+            }));
+            // 发起聊天
+            streamRequest(sessionCode, url, config);
         });
+        return function chat(userInput, sessionCode, url, config, property) {
+            return _ref.apply(this, arguments);
+        };
+    }();
+    /**
+   * 恢复流式聊天
+   * 如果最后一条消息处于流式传输中或是用户消息，重新建立连接
+   * @param sessionCode - 会话代码
+   * @param url - 请求 URL（可选）
+   * @param config - 请求配置（可选）
+   */ const resumeStreamingChat = (sessionCode, url, config)=>{
+        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 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;
     };
-    const stopChat = function() {
-        var _ref = _async_to_generator(function*() {
-            chatController === null || chatController === void 0 ? void 0 : chatController.abort();
-            chatController = 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);
         };
     }();
+    /**
+   * 重新发送消息（乐观更新）
+   * 删除指定用户消息及其后续所有消息，同时创建新消息并重新发送
+   *
+   * @param messageId - 用户消息 ID（id 字段）
+   * @param sessionCode - 会话编码
+   * @param newContent - 新内容（可选，不传则使用原消息内容；支持多模态）
+   * @param url - 请求 URL（可选）
+   * @param config - 请求配置（可选）
+   */ const resendMessage = function() {
+        var _ref = _async_to_generator(function*(messageId, sessionCode, newContent, url, config) {
+            var _mediator_message, _mediator_message1, _mediator_message2;
+            const messages = ((_mediator_message = mediator.message) === null || _mediator_message === void 0 ? void 0 : _mediator_message.list.value) || [];
+            // 1. 找到目标用户消息
+            const messageIndex = messages.findIndex((m)=>String(m.id) === messageId);
+            if (messageIndex === -1) {
+                throw new Error(`Message not found: ${messageId}`);
+            }
+            const targetMessage = messages[messageIndex];
+            if (targetMessage.role !== MessageRole.User) {
+                throw new Error('Can only resend user messages');
+            }
+            // 2. 获取原消息内容和 property（在删除前保存）
+            const originalContent = targetMessage.content;
+            const originalProperty = targetMessage.property;
+            // 3. 确定最终发送的内容
+            const finalContent = newContent !== null && newContent !== void 0 ? newContent : originalContent;
+            // 4. 收集需要删除的消息（目标用户消息 + 后续所有消息）
+            const messagesToDelete = messages.slice(messageIndex);
+            // 5. 并行执行删除和创建（乐观更新：立即更新 UI，API 调用在后台进行）
+            // 不等待 API 返回，让 UI 立即响应
+            const deletePromise = (_mediator_message1 = mediator.message) === null || _mediator_message1 === void 0 ? void 0 : _mediator_message1.deleteMessages(messagesToDelete);
+            const createPromise = (_mediator_message2 = mediator.message) === null || _mediator_message2 === void 0 ? void 0 : _mediator_message2.createAndPlusMessage(_object_spread({
+                role: MessageRole.User,
+                content: finalContent,
+                status: MessageStatus.Complete,
+                sessionCode
+            }, originalProperty && {
+                property: originalProperty
+            }));
+            // 6. 立即发起流式请求（不等待删除和创建的 API 完成）
+            streamRequest(sessionCode, url, config);
+            // 7. 在后台等待 API 完成，处理可能的错误
+            Promise.all([
+                deletePromise,
+                createPromise
+            ])['catch']((error)=>{
+                console.error('[resendMessage] API error:', error);
+            });
+        });
+        return function resendMessage(messageId, sessionCode, newContent, url, config) {
+            return _ref.apply(this, arguments);
+        };
+    }();
+    const reset = (protocol)=>{
+        // 重置状态
+        usedProtocol = protocol || new AGUIProtocol();
+        info.value = null;
+        isInfoLoading.value = false;
+    };
     return {
         info,
         isInfoLoading,
+        isChatting,
         chat,
+        handleRole,
+        resendMessage,
+        resumeStreamingChat,
+        abortChat,
         stopChat,
-        getAgentInfo
+        getAgentInfo,
+        reset
     };
 };

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

@@ -31,6 +31,22 @@ export declare class AGUIProtocol implements ISSEProtocol {
      * 处理自定义事件
      */
     handleCustomEvent(event: ICustomEvent): void;
+    /**
+     * 自定义事件 处理意图识别结束
+     */
+    handleKnowledgeRagEndCustomEvent(_event: ICustomEvent): void;
+    /**
+     * 自定义事件 处理意图识别结果
+     */
+    handleKnowledgeRagResultCustomEvent(event: ICustomEvent): void;
+    /**
+     * 自定义事件 处理意图识别开始
+     */
+    handleKnowledgeRagStartCustomEvent(_event: ICustomEvent): void;
+    /**
+     *  自定义事件 处理意图识别文本
+     */
+    handleKnowledgeRagTextContentCustomEvent(event: ICustomEvent): void;
     /**
      * 处理消息快照事件
      * 用于同步多端消息状态
@@ -52,11 +68,11 @@ export declare class AGUIProtocol implements ISSEProtocol {
     /**
      * 处理运行完成事件
      */
-    handleRunFinishedEvent(event: IRunFinishedEvent): void;
+    handleRunFinishedEvent(_event: IRunFinishedEvent): void;
     /**
      * 处理运行开始事件
      */
-    handleRunStartedEvent(event: IRunStartedEvent): void;
+    handleRunStartedEvent(_event: IRunStartedEvent): void;
     /**
      * 处理状态增量更新事件
      */
@@ -74,11 +90,11 @@ export declare class AGUIProtocol implements ISSEProtocol {
      */
     handleStepStartedEvent(_event: IStepStartedEvent): void;
     /**
-     * 处理文本消息块事件（流式输出）
+     * 处理文本消息块事件
      */
     handleTextMessageChunkEvent(event: ITextMessageChunkEvent): void;
     /**
-     * 处理文本消息内容事件
+     * 处理文本消息内容事件（流式输出）
      */
     handleTextMessageContentEvent(event: ITextMessageContentEvent): void;
     /**
@@ -96,7 +112,7 @@ export declare class AGUIProtocol implements ISSEProtocol {
     /**
      * 处理思考开始事件
      */
-    handleThinkingStartEvent(event: IThinkingStartEvent): void;
+    handleThinkingStartEvent(_event: IThinkingStartEvent): void;
     /**
      * 处理思考文本消息内容事件
      */
@@ -116,7 +132,7 @@ export declare class AGUIProtocol implements ISSEProtocol {
     /**
      * 处理工具调用块事件
      */
-    handleToolCallChunkEvent(event: IToolCallChunkEvent): void;
+    handleToolCallChunkEvent(_event: IToolCallChunkEvent): void;
     /**
      * 处理工具调用结束事件
      */
@@ -131,7 +147,7 @@ export declare class AGUIProtocol implements ISSEProtocol {
     handleToolCallStartEvent(event: IToolCallStartEvent): void;
     injectMessageModule(message: IMessageModule): void;
     onDone(): void;
-    onError(error: unknown): void;
+    onError(error: Error): void;
     onMessage(message: unknown): void;
     onStart(): void;
 }