@blueking/chat-helper 0.0.1-beta.1

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

@@ -0,0 +1,102 @@
+/*
+ * Tencent is pleased to support the open source community by making
+ * 蓝鲸智云PaaS平台 (BlueKing PaaS) available.
+ *
+ * Copyright (C) 2021 THL A29 Limited, a Tencent company.  All rights reserved.
+ *
+ * 蓝鲸智云PaaS平台 (BlueKing PaaS) is licensed under the MIT License.
+ *
+ * License for 蓝鲸智云PaaS平台 (BlueKing PaaS):
+ *
+ * ---------------------------------------------------
+ * Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
+ * documentation files (the "Software"), to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and
+ * to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all copies or substantial portions of
+ * the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO
+ * THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * 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.
+ */ function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) {
+    try {
+        var info = gen[key](arg);
+        var value = info.value;
+    } catch (error) {
+        reject(error);
+        return;
+    }
+    if (info.done) {
+        resolve(value);
+    } else {
+        Promise.resolve(value).then(_next, _throw);
+    }
+}
+function _async_to_generator(fn) {
+    return function() {
+        var self = this, args = arguments;
+        return new Promise(function(resolve, reject) {
+            var gen = fn.apply(self, args);
+            function _next(value) {
+                asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value);
+            }
+            function _throw(err) {
+                asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err);
+            }
+            _next(undefined);
+        });
+    };
+}
+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);
+    const info = ref(null);
+    const isInfoLoading = ref(false);
+    let chatController = null;
+    const getAgentInfo = ()=>{
+        isInfoLoading.value = true;
+        return http.agent.getAgentInfo().then((res)=>{
+            info.value = res;
+        }).finally(()=>{
+            isInfoLoading.value = false;
+        });
+    };
+    const chat = ()=>{
+        chatController = new AbortController();
+        const protocol = options.protocol || new AGUIProtocol();
+        // ag-ui 协议需要注入消息模块
+        if (protocol instanceof AGUIProtocol) {
+            protocol.injectMessageModule(message);
+        }
+        fetchClient.streamRequest({
+            url: 'agent/chat/',
+            controller: chatController,
+            onDone: protocol.onDone,
+            onError: protocol.onError,
+            onMessage: protocol.onMessage,
+            onStart: protocol.onStart
+        });
+    };
+    const stopChat = function() {
+        var _ref = _async_to_generator(function*() {
+            chatController === null || chatController === void 0 ? void 0 : chatController.abort();
+            chatController = null;
+        });
+        return function stopChat() {
+            return _ref.apply(this, arguments);
+        };
+    }();
+    return {
+        info,
+        isInfoLoading,
+        chat,
+        stopChat,
+        getAgentInfo
+    };
+};

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

@@ -0,0 +1,137 @@
+import { type IMessageModule } from '../message';
+import { type IActivityDeltaEvent, type IActivitySnapshotEvent, type ICustomEvent, type IEvent, type IMessagesSnapshotEvent, type IRawEvent, type IRunErrorEvent, type IRunFinishedEvent, type IRunStartedEvent, type IStateDeltaEvent, type IStateSnapshotEvent, type IStepFinishedEvent, type IStepStartedEvent, type ITextMessageChunkEvent, type ITextMessageContentEvent, type ITextMessageEndEvent, type ITextMessageStartEvent, type IThinkingEndEvent, type IThinkingStartEvent, type IThinkingTextMessageContentEvent, type IThinkingTextMessageEndEvent, type IThinkingTextMessageStartEvent, type IToolCallArgsEvent, type IToolCallChunkEvent, type IToolCallEndEvent, type IToolCallResultEvent, type IToolCallStartEvent } from './type';
+import type { ISSEProtocol } from '../http/fetch';
+/**
+ * AGUI 协议
+ * @param message - 消息模块
+ */
+export declare class AGUIProtocol implements ISSEProtocol {
+    private messageModule;
+    private onDoneCallback?;
+    private onErrorCallback?;
+    private onMessageCallback?;
+    private onStartCallback?;
+    constructor(params?: {
+        onDone?: () => void;
+        onError?: (error: unknown) => void;
+        onMessage?: (message: IEvent) => void;
+        onStart?: () => void;
+    });
+    /**
+     * 处理活动增量更新事件
+     * 使用 JSON Patch 格式增量更新活动数据
+     */
+    handleActivityDeltaEvent(_event: IActivityDeltaEvent): void;
+    /**
+     * 处理活动快照事件
+     * 记录活动的完整状态
+     */
+    handleActivitySnapshotEvent(_event: IActivitySnapshotEvent): void;
+    /**
+     * 处理自定义事件
+     */
+    handleCustomEvent(event: ICustomEvent): void;
+    /**
+     * 处理消息快照事件
+     * 用于同步多端消息状态
+     */
+    handleMessagesSnapshotEvent(event: IMessagesSnapshotEvent): void;
+    /**
+     * 处理原始事件
+     * 透传底层 AI 服务的原始事件
+     */
+    handleRawEvent(_event: IRawEvent): void;
+    /**
+     * 自定义事件 处理参考文档事件
+     */
+    handleReferenceDocumentCustomEvent(event: ICustomEvent): void;
+    /**
+     * 处理运行错误事件
+     */
+    handleRunErrorEvent(event: IRunErrorEvent): void;
+    /**
+     * 处理运行完成事件
+     */
+    handleRunFinishedEvent(event: IRunFinishedEvent): void;
+    /**
+     * 处理运行开始事件
+     */
+    handleRunStartedEvent(event: IRunStartedEvent): void;
+    /**
+     * 处理状态增量更新事件
+     */
+    handleStateDeltaEvent(_event: IStateDeltaEvent): void;
+    /**
+     * 处理状态快照事件
+     */
+    handleStateSnapshotEvent(_event: IStateSnapshotEvent): void;
+    /**
+     * 处理步骤完成事件
+     */
+    handleStepFinishedEvent(_event: IStepFinishedEvent): void;
+    /**
+     * 处理步骤开始事件
+     */
+    handleStepStartedEvent(_event: IStepStartedEvent): void;
+    /**
+     * 处理文本消息块事件（流式输出）
+     */
+    handleTextMessageChunkEvent(event: ITextMessageChunkEvent): void;
+    /**
+     * 处理文本消息内容事件
+     */
+    handleTextMessageContentEvent(event: ITextMessageContentEvent): void;
+    /**
+     * 处理文本消息结束事件
+     */
+    handleTextMessageEndEvent(event: ITextMessageEndEvent): void;
+    /**
+     * 处理文本消息开始事件
+     */
+    handleTextMessageStartEvent(event: ITextMessageStartEvent): void;
+    /**
+     * 处理思考结束事件
+     */
+    handleThinkingEndEvent(event: IThinkingEndEvent): void;
+    /**
+     * 处理思考开始事件
+     */
+    handleThinkingStartEvent(event: IThinkingStartEvent): void;
+    /**
+     * 处理思考文本消息内容事件
+     */
+    handleThinkingTextMessageContentEvent(event: IThinkingTextMessageContentEvent): void;
+    /**
+     * 处理思考文本消息结束事件
+     */
+    handleThinkingTextMessageEndEvent(_event: IThinkingTextMessageEndEvent): void;
+    /**
+     * 处理思考文本消息开始事件
+     */
+    handleThinkingTextMessageStartEvent(_event: IThinkingTextMessageStartEvent): void;
+    /**
+     * 处理工具调用参数事件
+     */
+    handleToolCallArgsEvent(event: IToolCallArgsEvent): void;
+    /**
+     * 处理工具调用块事件
+     */
+    handleToolCallChunkEvent(event: IToolCallChunkEvent): void;
+    /**
+     * 处理工具调用结束事件
+     */
+    handleToolCallEndEvent(event: IToolCallEndEvent): void;
+    /**
+     * 处理工具调用结果事件
+     */
+    handleToolCallResultEvent(event: IToolCallResultEvent): void;
+    /**
+     * 处理工具调用开始事件
+     */
+    handleToolCallStartEvent(event: IToolCallStartEvent): void;
+    injectMessageModule(message: IMessageModule): void;
+    onDone(): void;
+    onError(error: unknown): void;
+    onMessage(message: unknown): void;
+    onStart(): void;
+}

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

@@ -0,0 +1,434 @@
+/*
+ * Tencent is pleased to support the open source community by making
+ * 蓝鲸智云PaaS平台 (BlueKing PaaS) available.
+ *
+ * Copyright (C) 2021 THL A29 Limited, a Tencent company.  All rights reserved.
+ *
+ * 蓝鲸智云PaaS平台 (BlueKing PaaS) is licensed under the MIT License.
+ *
+ * License for 蓝鲸智云PaaS平台 (BlueKing PaaS):
+ *
+ * ---------------------------------------------------
+ * Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
+ * documentation files (the "Software"), to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and
+ * to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all copies or substantial portions of
+ * the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO
+ * THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * 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.
+ */ 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;
+}
+import { transferMessageApi2Message } from '../http/transform/message.ts.js';
+import { MessageContentStatus, MessageContentType, MessageRole, MessageStatus } from '../message/index.ts.js';
+import { CustomEventType, EventType } from './type.ts.js';
+// 临时存储时间戳
+let tempTimestamp = 0;
+/**
+ * AGUI 协议
+ * @param message - 消息模块
+ */ export class AGUIProtocol {
+    /**
+   * 处理活动增量更新事件
+   * 使用 JSON Patch 格式增量更新活动数据
+   */ handleActivityDeltaEvent(_event) {
+    // 可以在这里实现 JSON Patch 逻辑
+    // console.log('ActivityDelta:', event);
+    }
+    /**
+   * 处理活动快照事件
+   * 记录活动的完整状态
+   */ handleActivitySnapshotEvent(_event) {
+    // 可以在这里存储活动快照
+    // console.log('ActivitySnapshot:', event);
+    }
+    /**
+   * 处理自定义事件
+   */ handleCustomEvent(event) {
+        switch(event.value.type){
+            case CustomEventType.ReferenceDocument:
+                this.handleReferenceDocumentCustomEvent(event);
+                break;
+            default:
+                break;
+        }
+    }
+    /**
+   * 处理消息快照事件
+   * 用于同步多端消息状态
+   */ handleMessagesSnapshotEvent(event) {
+        this.messageModule.list.value = event.messages.map(transferMessageApi2Message);
+    }
+    /**
+   * 处理原始事件
+   * 透传底层 AI 服务的原始事件
+   */ handleRawEvent(_event) {
+    // console.log('Raw:', event);
+    }
+    /**
+   * 自定义事件 处理参考文档事件
+   */ handleReferenceDocumentCustomEvent(event) {
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message && message.role === MessageRole.Assistant) {
+            const referenceDocumentMessageContent = {
+                type: MessageContentType.ReferenceDocument,
+                data: event.value.data,
+                status: MessageContentStatus.Streaming
+            };
+            message.content.push(referenceDocumentMessageContent);
+        }
+    }
+    /**
+   * 处理运行错误事件
+   */ handleRunErrorEvent(event) {
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message) {
+            message.status = MessageStatus.Error;
+            message.content = [
+                {
+                    type: MessageContentType.Text,
+                    data: event.message,
+                    status: MessageContentStatus.Error
+                }
+            ];
+        }
+    }
+    /**
+   * 处理运行完成事件
+   */ handleRunFinishedEvent(event) {
+        const message = this.messageModule.getMessageById(event.runId);
+        if (message) {
+            message.status = MessageStatus.Complete;
+        }
+    }
+    /**
+   * 处理运行开始事件
+   */ handleRunStartedEvent(event) {
+        this.messageModule.plusMessage({
+            id: event.runId,
+            role: MessageRole.Assistant,
+            content: [],
+            status: MessageStatus.Streaming
+        });
+    }
+    /**
+   * 处理状态增量更新事件
+   */ handleStateDeltaEvent(_event) {
+    // 可以在这里实现 JSON Patch 逻辑
+    // console.log('StateDelta:', event);
+    }
+    /**
+   * 处理状态快照事件
+   */ handleStateSnapshotEvent(_event) {
+    // console.log('StateSnapshot:', event);
+    }
+    /**
+   * 处理步骤完成事件
+   */ handleStepFinishedEvent(_event) {
+    // console.log('StepFinished:', event);
+    }
+    /**
+   * 处理步骤开始事件
+   */ handleStepStartedEvent(_event) {
+    // console.log('StepStarted:', event);
+    }
+    /**
+   * 处理文本消息块事件（流式输出）
+   */ handleTextMessageChunkEvent(event) {
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message) {
+            const messageContent = message.content.find((item)=>item.type === MessageContentType.Text && item.id === event.messageId);
+            if (messageContent) {
+                // 增量
+                messageContent.data += event.delta;
+            } else {
+                // 新增
+                message.content.push({
+                    id: event.messageId,
+                    status: MessageContentStatus.Complete,
+                    type: MessageContentType.Text,
+                    data: event.delta
+                });
+            }
+        }
+    }
+    /**
+   * 处理文本消息内容事件
+   */ handleTextMessageContentEvent(event) {
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message) {
+            const messageContent = message.content.find((item)=>item.type === MessageContentType.Text && item.id === event.messageId);
+            if (messageContent) {
+                messageContent.data += event.delta;
+            }
+        }
+    }
+    /**
+   * 处理文本消息结束事件
+   */ handleTextMessageEndEvent(event) {
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message) {
+            const messageContent = message.content.find((item)=>item.type === MessageContentType.Text && item.id === event.messageId);
+            if (messageContent) {
+                messageContent.status = MessageContentStatus.Complete;
+            }
+        }
+    }
+    /**
+   * 处理文本消息开始事件
+   */ handleTextMessageStartEvent(event) {
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message) {
+            message.content.push({
+                id: event.messageId,
+                status: MessageContentStatus.Streaming,
+                type: MessageContentType.Text,
+                data: ''
+            });
+        }
+    }
+    /**
+   * 处理思考结束事件
+   */ handleThinkingEndEvent(event) {
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message && message.role === MessageRole.Assistant) {
+            const lastThinkingMessageContent = message.content.findLast((item)=>item.type === MessageContentType.Thinking);
+            if (lastThinkingMessageContent) {
+                lastThinkingMessageContent.status = MessageContentStatus.Complete;
+                lastThinkingMessageContent.data.duration = event.timestamp - tempTimestamp;
+            }
+        }
+    }
+    /**
+   * 处理思考开始事件
+   */ handleThinkingStartEvent(event) {
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message && message.role === MessageRole.Assistant) {
+            // 临时存储时间戳
+            tempTimestamp = event.timestamp;
+            // 创建思考消息内容
+            const thinkingMessageContent = {
+                status: MessageContentStatus.Streaming,
+                type: MessageContentType.Thinking,
+                data: {
+                    text: '',
+                    title: event.title
+                }
+            };
+            message.content.push(thinkingMessageContent);
+        }
+    }
+    /**
+   * 处理思考文本消息内容事件
+   */ handleThinkingTextMessageContentEvent(event) {
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message && message.role === MessageRole.Assistant) {
+            const lastThinkingMessageContent = message.content.findLast((item)=>item.type === MessageContentType.Thinking);
+            if (lastThinkingMessageContent) {
+                lastThinkingMessageContent.data.text += event.delta;
+            }
+        }
+    }
+    /**
+   * 处理思考文本消息结束事件
+   */ handleThinkingTextMessageEndEvent(_event) {
+    // console.log(event);
+    }
+    /**
+   * 处理思考文本消息开始事件
+   */ handleThinkingTextMessageStartEvent(_event) {
+    // console.log(event);
+    }
+    /**
+   * 处理工具调用参数事件
+   */ handleToolCallArgsEvent(event) {
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message && message.role === MessageRole.Assistant) {
+            const toolCallMessageContent = message.content.find((item)=>item.type === MessageContentType.ToolCall && item.data.toolCallId === event.toolCallId);
+            if ((toolCallMessageContent === null || toolCallMessageContent === void 0 ? void 0 : toolCallMessageContent.type) === MessageContentType.ToolCall) {
+                toolCallMessageContent.data.args += event.delta;
+            }
+        }
+    }
+    /**
+   * 处理工具调用块事件
+   */ handleToolCallChunkEvent(event) {
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message && message.role === MessageRole.Assistant) {
+            const toolCallMessageContent = message.content.find((item)=>item.type === MessageContentType.ToolCall && item.data.toolCallId === event.toolCallId);
+            if ((toolCallMessageContent === null || toolCallMessageContent === void 0 ? void 0 : toolCallMessageContent.type) === MessageContentType.ToolCall) {
+                toolCallMessageContent.data.chunk += event.delta;
+            }
+        }
+    }
+    /**
+   * 处理工具调用结束事件
+   */ handleToolCallEndEvent(event) {
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message && message.role === MessageRole.Assistant) {
+            const toolCallMessageContent = message.content.find((item)=>item.type === MessageContentType.ToolCall && item.data.toolCallId === event.toolCallId);
+            if ((toolCallMessageContent === null || toolCallMessageContent === void 0 ? void 0 : toolCallMessageContent.type) === MessageContentType.ToolCall) {
+                toolCallMessageContent.status = MessageContentStatus.Complete;
+            }
+        }
+    }
+    /**
+   * 处理工具调用结果事件
+   */ handleToolCallResultEvent(event) {
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message && message.role === MessageRole.Assistant) {
+            const toolCallMessageContent = message.content.find((item)=>item.type === MessageContentType.ToolCall && item.data.toolCallId === event.toolCallId);
+            if ((toolCallMessageContent === null || toolCallMessageContent === void 0 ? void 0 : toolCallMessageContent.type) === MessageContentType.ToolCall) {
+                toolCallMessageContent.data.result += event.content;
+            }
+        }
+    }
+    /**
+   * 处理工具调用开始事件
+   */ handleToolCallStartEvent(event) {
+        const message = this.messageModule.getCurrentLoadingMessage();
+        if (message && message.role === MessageRole.Assistant) {
+            const toolCallMessageContent = {
+                status: MessageContentStatus.Streaming,
+                type: MessageContentType.ToolCall,
+                data: {
+                    toolCallId: event.toolCallId,
+                    toolCallName: event.toolCallName,
+                    parentMessageId: event.parentMessageId
+                }
+            };
+            message.content.push(toolCallMessageContent);
+        }
+    }
+    injectMessageModule(message) {
+        this.messageModule = message;
+    }
+    onDone() {
+        var _this_onDoneCallback, _this;
+        (_this_onDoneCallback = (_this = this).onDoneCallback) === null || _this_onDoneCallback === void 0 ? void 0 : _this_onDoneCallback.call(_this);
+    }
+    onError(error) {
+        var _this_onErrorCallback, _this;
+        (_this_onErrorCallback = (_this = this).onErrorCallback) === null || _this_onErrorCallback === void 0 ? void 0 : _this_onErrorCallback.call(_this, error);
+    }
+    onMessage(message) {
+        var // 回调给上层
+        _this_onMessageCallback, _this;
+        const event = message;
+        (_this_onMessageCallback = (_this = this).onMessageCallback) === null || _this_onMessageCallback === void 0 ? void 0 : _this_onMessageCallback.call(_this, event);
+        // 处理事件
+        switch(event.type){
+            case EventType.ActivityDelta:
+                this.handleActivityDeltaEvent(event);
+                break;
+            case EventType.ActivitySnapshot:
+                this.handleActivitySnapshotEvent(event);
+                break;
+            case EventType.Custom:
+                this.handleCustomEvent(event);
+                break;
+            case EventType.MessagesSnapshot:
+                this.handleMessagesSnapshotEvent(event);
+                break;
+            case EventType.Raw:
+                this.handleRawEvent(event);
+                break;
+            case EventType.RunError:
+                this.handleRunErrorEvent(event);
+                break;
+            case EventType.RunFinished:
+                this.handleRunFinishedEvent(event);
+                break;
+            case EventType.RunStarted:
+                this.handleRunStartedEvent(event);
+                break;
+            case EventType.StateDelta:
+                this.handleStateDeltaEvent(event);
+                break;
+            case EventType.StateSnapshot:
+                this.handleStateSnapshotEvent(event);
+                break;
+            case EventType.StepFinished:
+                this.handleStepFinishedEvent(event);
+                break;
+            case EventType.StepStarted:
+                this.handleStepStartedEvent(event);
+                break;
+            case EventType.TextMessageChunk:
+                this.handleTextMessageChunkEvent(event);
+                break;
+            case EventType.TextMessageContent:
+                this.handleTextMessageContentEvent(event);
+                break;
+            case EventType.TextMessageEnd:
+                this.handleTextMessageEndEvent(event);
+                break;
+            case EventType.TextMessageStart:
+                this.handleTextMessageStartEvent(event);
+                break;
+            case EventType.ThinkingEnd:
+                this.handleThinkingEndEvent(event);
+                break;
+            case EventType.ThinkingStart:
+                this.handleThinkingStartEvent(event);
+                break;
+            case EventType.ThinkingTextMessageContent:
+                this.handleThinkingTextMessageContentEvent(event);
+                break;
+            case EventType.ThinkingTextMessageEnd:
+                this.handleThinkingTextMessageEndEvent(event);
+                break;
+            case EventType.ThinkingTextMessageStart:
+                this.handleThinkingTextMessageStartEvent(event);
+                break;
+            case EventType.ToolCallArgs:
+                this.handleToolCallArgsEvent(event);
+                break;
+            case EventType.ToolCallChunk:
+                this.handleToolCallChunkEvent(event);
+                break;
+            case EventType.ToolCallEnd:
+                this.handleToolCallEndEvent(event);
+                break;
+            case EventType.ToolCallResult:
+                this.handleToolCallResultEvent(event);
+                break;
+            case EventType.ToolCallStart:
+                this.handleToolCallStartEvent(event);
+                break;
+            default:
+                break;
+        }
+    }
+    onStart() {
+        var _this_onStartCallback, _this;
+        (_this_onStartCallback = (_this = this).onStartCallback) === null || _this_onStartCallback === void 0 ? void 0 : _this_onStartCallback.call(_this);
+    }
+    constructor(params){
+        _define_property(this, "messageModule", void 0);
+        _define_property(this, "onDoneCallback", void 0);
+        _define_property(this, "onErrorCallback", void 0);
+        _define_property(this, "onMessageCallback", void 0);
+        _define_property(this, "onStartCallback", void 0);
+        this.onDoneCallback = params === null || params === void 0 ? void 0 : params.onDone;
+        this.onErrorCallback = params === null || params === void 0 ? void 0 : params.onError;
+        this.onMessageCallback = params === null || params === void 0 ? void 0 : params.onMessage;
+        this.onStartCallback = params === null || params === void 0 ? void 0 : params.onStart;
+    }
+}

package/dist/event/index.d.ts

@@ -0,0 +1,2 @@
+export * from './ag-ui';
+export * from './type';