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

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

@@ -81,75 +81,224 @@ function _object_spread(target) {
 }
 import { ref } from 'vue';
 import { AGUIProtocol } from '../event/index.ts.js';
-import { useFetch } from '../http/fetch/index.ts.js';
-import { MessageContentStatus, MessageContentType, MessageRole } from '../message/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 = function() {
-        var _ref = _async_to_generator(function*(userInput, sessionCode, url, config) {
-            chatController = new AbortController();
-            const protocol = options.protocol || new AGUIProtocol();
-            // ag-ui 协议需要注入消息模块、sessionCode
-            if (protocol instanceof AGUIProtocol) {
-                protocol.injectMessageModule(message);
-                protocol.injectSessionCode(sessionCode);
-            }
-            // 先新增一个 message
-            yield message.plusMessage({
+    // 处理角色消息
+    const handleRole = (data, sessionCode)=>{
+        var _data_promptSetting_content, _data_promptSetting;
+        const lastRoleMessage = (_data_promptSetting = data.promptSetting) === null || _data_promptSetting === void 0 ? void 0 : (_data_promptSetting_content = _data_promptSetting.content) === null || _data_promptSetting_content === void 0 ? void 0 : _data_promptSetting_content.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,
                 sessionCode,
-                role: MessageRole.User,
-                content: [
-                    {
-                        type: MessageContentType.Text,
-                        data: {
-                            text: userInput
-                        },
-                        status: MessageContentStatus.Complete
+                property: {
+                    extra: {
+                        pause: true
                     }
-                ]
+                }
             });
-            // 开始聊天
-            fetchClient.streamRequest(_object_spread({
+        }
+    };
+    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
+                    session_code: sessionCode,
+                    execute_kwargs: {
+                        stream: true
+                    }
                 },
-                controller: chatController,
-                onDone: protocol.onDone.bind(protocol),
-                onError: protocol.onError.bind(protocol),
-                onMessage: protocol.onMessage.bind(protocol),
-                onStart: protocol.onStart.bind(protocol)
+                controller: abortController,
+                onDone,
+                onError,
+                onMessage,
+                onStart
             }, config));
         });
-        return function chat(userInput, sessionCode, url, config) {
+        return function streamRequest(sessionCode, url, config) {
             return _ref.apply(this, arguments);
         };
     }();
-    const stopChat = function() {
-        var _ref = _async_to_generator(function*() {
-            chatController === null || chatController === void 0 ? void 0 : chatController.abort();
-            chatController = null;
+    /**
+   * 发送聊天消息
+   * @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 stopChat() {
+        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;
+    };
+    /**
+   * 停止会话，后端中止
+   * @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(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

@@ -11,7 +11,6 @@ export declare class AGUIProtocol implements ISSEProtocol {
     private onErrorCallback?;
     private onMessageCallback?;
     private onStartCallback?;
-    private sessionCode;
     constructor(params?: {
         onDone?: () => void;
         onError?: (error: unknown) => void;
@@ -32,6 +31,34 @@ export declare class AGUIProtocol implements ISSEProtocol {
      * 处理自定义事件
      */
     handleCustomEvent(event: ICustomEvent): void;
+    /**
+     * 自定义事件 处理流程编排任务结束
+     */
+    handleFlowAgentEndCustomEvent(event: ICustomEvent): void;
+    /**
+     * 自定义事件 处理流程编排任务结果（增量更新节点状态）
+     */
+    handleFlowAgentResultCustomEvent(event: ICustomEvent): void;
+    /**
+     * 自定义事件 处理流程编排任务开始
+     */
+    handleFlowAgentStartCustomEvent(event: ICustomEvent): void;
+    /**
+     * 自定义事件 处理意图识别结束
+     */
+    handleKnowledgeRagEndCustomEvent(_event: ICustomEvent): void;
+    /**
+     * 自定义事件 处理意图识别结果
+     */
+    handleKnowledgeRagResultCustomEvent(event: ICustomEvent): void;
+    /**
+     * 自定义事件 处理意图识别开始
+     */
+    handleKnowledgeRagStartCustomEvent(_event: ICustomEvent): void;
+    /**
+     *  自定义事件 处理意图识别文本
+     */
+    handleKnowledgeRagTextContentCustomEvent(event: ICustomEvent): void;
     /**
      * 处理消息快照事件
      * 用于同步多端消息状态
@@ -53,7 +80,7 @@ export declare class AGUIProtocol implements ISSEProtocol {
     /**
      * 处理运行完成事件
      */
-    handleRunFinishedEvent(event: IRunFinishedEvent): void;
+    handleRunFinishedEvent(_event: IRunFinishedEvent): void;
     /**
      * 处理运行开始事件
      */
@@ -75,11 +102,15 @@ export declare class AGUIProtocol implements ISSEProtocol {
      */
     handleStepStartedEvent(_event: IStepStartedEvent): void;
     /**
-     * 处理文本消息块事件（流式输出）
+     * 自定义事件 处理临时消息，转换为 AI 文本消息
+     */
+    handleTempMessageCustomEvent(event: ICustomEvent): void;
+    /**
+     * 处理文本消息块事件
      */
     handleTextMessageChunkEvent(event: ITextMessageChunkEvent): void;
     /**
-     * 处理文本消息内容事件
+     * 处理文本消息内容事件（流式输出）
      */
     handleTextMessageContentEvent(event: ITextMessageContentEvent): void;
     /**
@@ -97,7 +128,7 @@ export declare class AGUIProtocol implements ISSEProtocol {
     /**
      * 处理思考开始事件
      */
-    handleThinkingStartEvent(event: IThinkingStartEvent): void;
+    handleThinkingStartEvent(_event: IThinkingStartEvent): void;
     /**
      * 处理思考文本消息内容事件
      */
@@ -117,7 +148,7 @@ export declare class AGUIProtocol implements ISSEProtocol {
     /**
      * 处理工具调用块事件
      */
-    handleToolCallChunkEvent(event: IToolCallChunkEvent): void;
+    handleToolCallChunkEvent(_event: IToolCallChunkEvent): void;
     /**
      * 处理工具调用结束事件
      */
@@ -131,7 +162,6 @@ export declare class AGUIProtocol implements ISSEProtocol {
      */
     handleToolCallStartEvent(event: IToolCallStartEvent): void;
     injectMessageModule(message: IMessageModule): void;
-    injectSessionCode(sessionCode: string): void;
     onDone(): void;
     onError(error: Error): void;
     onMessage(message: unknown): void;