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

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

@@ -22,136 +22,584 @@
  * 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 { MessageContentType } from '../../message/type.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 { ActivityType, MessageRole, MessageType } from '../../message/type.ts.js';
 /**
- * 将 API 返回的 message 数据转换为前端使用的 message 数据
- * @param data API 返回的 message 数据
- * @returns 前端使用的 message 数据
+ * 将 API 返回的消息数据转换为前端使用的消息数据
+ * @param data API 返回的消息数据
+ * @returns 前端使用的消息数据
  */ export const transferMessageApi2Message = (data)=>{
-    const message = {
+    const baseMessage = {
         id: data.id,
-        status: data.status,
+        messageId: data.message_id,
+        name: data.name,
         role: data.role,
         sessionCode: data.session_code,
-        content: data.content.map(transferMessageContentApi2MessageContent),
-        property: data.property
+        status: data.status
     };
-    return message;
+    // 处理不同类型的消息
+    switch(data.role){
+        case MessageRole.Activity:
+            {
+                const activityData = data;
+                const transferReferenceDocumentApi2ReferenceDocument = (doc)=>doc.map((item)=>({
+                            name: item.name,
+                            originFileUrl: item.origin_file_url,
+                            url: item.url
+                        }));
+                let content;
+                switch(activityData.activity_type){
+                    case ActivityType.ReferenceDocument:
+                        content = transferReferenceDocumentApi2ReferenceDocument(activityData.content);
+                        break;
+                    case ActivityType.KnowledgeRag:
+                        {
+                            const ragContent = activityData.content;
+                            content = {
+                                content: ragContent.content,
+                                referenceDocument: transferReferenceDocumentApi2ReferenceDocument(ragContent.reference_document)
+                            };
+                            break;
+                        }
+                    case ActivityType.FlowAgent:
+                    default:
+                        content = activityData.content;
+                        break;
+                }
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    activityType: activityData.activity_type,
+                    content,
+                    role: activityData.role
+                });
+                return result;
+            }
+        case MessageRole.Assistant:
+            {
+                var _assistantData_tool_calls;
+                const assistantData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: assistantData.content,
+                    property: assistantData.property,
+                    role: assistantData.role,
+                    toolCalls: (_assistantData_tool_calls = assistantData.tool_calls) === null || _assistantData_tool_calls === void 0 ? void 0 : _assistantData_tool_calls.map((toolCall)=>{
+                        return {
+                            function: {
+                                arguments: toolCall['function'].arguments,
+                                name: toolCall['function'].name,
+                                description: toolCall['function'].description,
+                                mcpName: toolCall['function'].mcp_name
+                            },
+                            id: toolCall.id,
+                            type: toolCall.type
+                        };
+                    })
+                });
+                return result;
+            }
+        case MessageRole.Developer:
+            {
+                const developerData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: developerData.content,
+                    role: developerData.role
+                });
+                return result;
+            }
+        case MessageRole.Guide:
+            {
+                const guideData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: guideData.content,
+                    role: guideData.role
+                });
+                return result;
+            }
+        case MessageRole.Hidden:
+            {
+                const hiddenData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: hiddenData.content,
+                    role: hiddenData.role
+                });
+                return result;
+            }
+        case MessageRole.HiddenAssistant:
+            {
+                const hiddenAssistantData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: hiddenAssistantData.content,
+                    role: hiddenAssistantData.role
+                });
+                return result;
+            }
+        case MessageRole.HiddenGuide:
+            {
+                const hiddenGuideData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: hiddenGuideData.content,
+                    role: hiddenGuideData.role
+                });
+                return result;
+            }
+        case MessageRole.HiddenSystem:
+            {
+                const hiddenSystemData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: hiddenSystemData.content,
+                    role: hiddenSystemData.role
+                });
+                return result;
+            }
+        case MessageRole.HiddenUser:
+            {
+                const hiddenUserData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: hiddenUserData.content,
+                    role: hiddenUserData.role
+                });
+                return result;
+            }
+        case MessageRole.Info:
+            {
+                const infoData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: infoData.content,
+                    role: infoData.role
+                });
+                return result;
+            }
+        case MessageRole.Pause:
+            {
+                const pauseData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: pauseData.content,
+                    role: pauseData.role
+                });
+                return result;
+            }
+        case MessageRole.Placeholder:
+            {
+                const placeholderData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: placeholderData.content,
+                    role: placeholderData.role
+                });
+                return result;
+            }
+        case MessageRole.Reasoning:
+            {
+                const reasoningData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: reasoningData.content,
+                    duration: reasoningData.duration,
+                    role: reasoningData.role
+                });
+                return result;
+            }
+        case MessageRole.System:
+            {
+                const systemData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: systemData.content,
+                    role: systemData.role
+                });
+                return result;
+            }
+        case MessageRole.TemplateAssistant:
+            {
+                const templateAssistantData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: templateAssistantData.content,
+                    role: templateAssistantData.role
+                });
+                return result;
+            }
+        case MessageRole.TemplateGuide:
+            {
+                const templateGuideData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: templateGuideData.content,
+                    role: templateGuideData.role
+                });
+                return result;
+            }
+        case MessageRole.TemplateHidden:
+            {
+                const templateHiddenData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: templateHiddenData.content,
+                    role: templateHiddenData.role
+                });
+                return result;
+            }
+        case MessageRole.TemplateSystem:
+            {
+                const templateSystemData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: templateSystemData.content,
+                    role: templateSystemData.role
+                });
+                return result;
+            }
+        case MessageRole.TemplateUser:
+            {
+                const templateUserData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: templateUserData.content,
+                    role: templateUserData.role
+                });
+                return result;
+            }
+        case MessageRole.Tool:
+            {
+                const toolData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: toolData.content,
+                    duration: toolData.duration,
+                    error: toolData.error,
+                    role: toolData.role,
+                    toolCallId: toolData.tool_call_id
+                });
+                return result;
+            }
+        case MessageRole.User:
+            {
+                const userData = data;
+                const userContent = userData.content;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: Array.isArray(userContent) ? userContent.map((item)=>{
+                        if (item.type === MessageType.Binary) {
+                            const binaryItem = item;
+                            return {
+                                data: binaryItem.data,
+                                filename: binaryItem.filename,
+                                id: binaryItem.id,
+                                mimeType: binaryItem.mime_type,
+                                type: binaryItem.type,
+                                url: binaryItem.url
+                            };
+                        }
+                        return item;
+                    }) : userContent,
+                    property: userData.property,
+                    role: userData.role
+                });
+                return result;
+            }
+    }
 };
 /**
- * 将前端使用的 message 数据转换为 API 返回的 message 数据
- * @param data 前端使用的 message 数据
- * @returns API 返回的 message 数据
+ * 将前端使用的消息数据转换为 API 使用的消息数据
+ * @param data 前端使用的消息数据
+ * @returns API 使用的消息数据
  */ export const transferMessage2MessageApi = (data)=>{
-    const messageApi = {
+    const baseMessage = {
         id: data.id,
-        status: data.status,
+        message_id: data.messageId,
+        name: data.name,
         role: data.role,
         session_code: data.sessionCode,
-        content: data.content.map(transferMessageContent2MessageContentApi),
-        property: data.property
+        status: data.status
     };
-    return messageApi;
-};
-/**
- * 将 API 返回的 message 数据中的内容数据转换为前端使用的 message 内容数据
- * @param content API 返回的 message 数据中的内容数据
- * @returns 前端使用的 message 内容数据
- */ const transferMessageContentApi2MessageContent = (content)=>{
-    switch(content.type){
-        case MessageContentType.Text:
-            return {
-                type: MessageContentType.Text,
-                id: content.id,
-                data: content.data,
-                status: content.status
-            };
-        case MessageContentType.Thinking:
-            return {
-                type: MessageContentType.Thinking,
-                data: content.data,
-                status: content.status
-            };
-        case MessageContentType.ToolCall:
-            return {
-                type: MessageContentType.ToolCall,
-                data: {
-                    toolCallId: content.data.tool_call_id,
-                    toolCallName: content.data.tool_call_name,
-                    parentMessageId: content.data.parent_message_id,
-                    args: content.data.args,
-                    chunk: content.data.chunk,
-                    result: content.data.result
-                },
-                status: content.status
-            };
-        case MessageContentType.Image:
-            return {
-                type: MessageContentType.Image,
-                data: content.data,
-                status: content.status
-            };
-        case MessageContentType.ReferenceDocument:
-            return {
-                type: MessageContentType.ReferenceDocument,
-                data: content.data.map((item)=>({
-                        title: item.title,
-                        icon: item.icon,
-                        url: item.url,
-                        originFileUrl: item.origin_file_url
-                    })),
-                status: content.status
-            };
-    }
-};
-/**
- * 将前端使用的 message 内容数据转换为 API 返回的 message 内容数据
- * @param data 前端使用的 message 内容数据
- * @returns API 返回的 message 内容数据
- */ const transferMessageContent2MessageContentApi = (content)=>{
-    switch(content.type){
-        case MessageContentType.Text:
-            return {
-                type: MessageContentType.Text,
-                id: content.id,
-                data: content.data,
-                status: content.status
-            };
-        case MessageContentType.Thinking:
-            return {
-                type: MessageContentType.Thinking,
-                data: content.data,
-                status: content.status
-            };
-        case MessageContentType.ToolCall:
-            return {
-                type: MessageContentType.ToolCall,
-                data: {
-                    tool_call_id: content.data.toolCallId,
-                    tool_call_name: content.data.toolCallName,
-                    parent_message_id: content.data.parentMessageId,
-                    args: content.data.args,
-                    chunk: content.data.chunk,
-                    result: content.data.result
-                },
-                status: content.status
-            };
-        case MessageContentType.Image:
-            return {
-                type: MessageContentType.Image,
-                data: content.data,
-                status: content.status
-            };
-        case MessageContentType.ReferenceDocument:
-            return {
-                type: MessageContentType.ReferenceDocument,
-                data: content.data.map((item)=>({
-                        title: item.title,
-                        icon: item.icon,
-                        url: item.url,
-                        origin_file_url: item.originFileUrl
-                    })),
-                status: content.status
-            };
+    // 处理不同类型的消息
+    switch(data.role){
+        case MessageRole.Activity:
+            {
+                const activityData = data;
+                const transferReferenceDocument2ReferenceDocumentApi = (doc)=>doc.map((item)=>({
+                            name: item.name,
+                            origin_file_url: item.originFileUrl,
+                            url: item.url
+                        }));
+                let content;
+                switch(activityData.activityType){
+                    case ActivityType.ReferenceDocument:
+                        content = transferReferenceDocument2ReferenceDocumentApi(activityData.content);
+                        break;
+                    case ActivityType.KnowledgeRag:
+                        {
+                            const ragContent = activityData.content;
+                            content = {
+                                content: ragContent.content,
+                                reference_document: transferReferenceDocument2ReferenceDocumentApi(ragContent.referenceDocument)
+                            };
+                            break;
+                        }
+                    case ActivityType.FlowAgent:
+                    default:
+                        content = activityData.content;
+                        break;
+                }
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    activity_type: activityData.activityType,
+                    content,
+                    role: activityData.role
+                });
+                return result;
+            }
+        case MessageRole.Assistant:
+            {
+                var _assistantData_toolCalls;
+                const assistantData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    property: assistantData.property,
+                    content: assistantData.content,
+                    role: assistantData.role,
+                    tool_calls: (_assistantData_toolCalls = assistantData.toolCalls) === null || _assistantData_toolCalls === void 0 ? void 0 : _assistantData_toolCalls.map((toolCall)=>{
+                        return {
+                            function: {
+                                arguments: toolCall['function'].arguments,
+                                name: toolCall['function'].name,
+                                description: toolCall['function'].description,
+                                mcp_name: toolCall['function'].mcpName
+                            },
+                            id: toolCall.id,
+                            type: toolCall.type
+                        };
+                    })
+                });
+                return result;
+            }
+        case MessageRole.Developer:
+            {
+                const developerData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: developerData.content,
+                    role: developerData.role
+                });
+                return result;
+            }
+        case MessageRole.Guide:
+            {
+                const guideData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: guideData.content,
+                    role: guideData.role
+                });
+                return result;
+            }
+        case MessageRole.Hidden:
+            {
+                const hiddenData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: hiddenData.content,
+                    role: hiddenData.role
+                });
+                return result;
+            }
+        case MessageRole.HiddenAssistant:
+            {
+                const hiddenAssistantData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: hiddenAssistantData.content,
+                    role: hiddenAssistantData.role
+                });
+                return result;
+            }
+        case MessageRole.HiddenGuide:
+            {
+                const hiddenGuideData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: hiddenGuideData.content,
+                    role: hiddenGuideData.role
+                });
+                return result;
+            }
+        case MessageRole.HiddenSystem:
+            {
+                const hiddenSystemData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: hiddenSystemData.content,
+                    role: hiddenSystemData.role
+                });
+                return result;
+            }
+        case MessageRole.HiddenUser:
+            {
+                const hiddenUserData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: hiddenUserData.content,
+                    role: hiddenUserData.role
+                });
+                return result;
+            }
+        case MessageRole.Info:
+            {
+                const infoData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: infoData.content,
+                    role: infoData.role
+                });
+                return result;
+            }
+        case MessageRole.Pause:
+            {
+                const pauseData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: pauseData.content,
+                    role: pauseData.role
+                });
+                return result;
+            }
+        case MessageRole.Placeholder:
+            {
+                const placeholderData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: placeholderData.content,
+                    role: placeholderData.role
+                });
+                return result;
+            }
+        case MessageRole.Reasoning:
+            {
+                const reasoningData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: reasoningData.content,
+                    duration: reasoningData.duration,
+                    role: reasoningData.role
+                });
+                return result;
+            }
+        case MessageRole.System:
+            {
+                const systemData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: systemData.content,
+                    role: systemData.role
+                });
+                return result;
+            }
+        case MessageRole.TemplateAssistant:
+            {
+                const templateAssistantData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: templateAssistantData.content,
+                    role: templateAssistantData.role
+                });
+                return result;
+            }
+        case MessageRole.TemplateGuide:
+            {
+                const templateGuideData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: templateGuideData.content,
+                    role: templateGuideData.role
+                });
+                return result;
+            }
+        case MessageRole.TemplateHidden:
+            {
+                const templateHiddenData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: templateHiddenData.content,
+                    role: templateHiddenData.role
+                });
+                return result;
+            }
+        case MessageRole.TemplateSystem:
+            {
+                const templateSystemData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: templateSystemData.content,
+                    role: templateSystemData.role
+                });
+                return result;
+            }
+        case MessageRole.TemplateUser:
+            {
+                const templateUserData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: templateUserData.content,
+                    role: templateUserData.role
+                });
+                return result;
+            }
+        case MessageRole.Tool:
+            {
+                const toolData = data;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: toolData.content,
+                    duration: toolData.duration,
+                    error: toolData.error,
+                    role: toolData.role,
+                    tool_call_id: toolData.toolCallId
+                });
+                return result;
+            }
+        case MessageRole.User:
+            {
+                const userData = data;
+                const userContent = userData.content;
+                const result = _object_spread_props(_object_spread({}, baseMessage), {
+                    content: Array.isArray(userContent) ? userContent.map((item)=>{
+                        if (item.type === MessageType.Binary) {
+                            const binaryItem = item;
+                            return {
+                                data: binaryItem.data,
+                                filename: binaryItem.filename,
+                                id: binaryItem.id,
+                                mime_type: binaryItem.mimeType,
+                                type: binaryItem.type,
+                                url: binaryItem.url
+                            };
+                        }
+                        return item;
+                    }) : userContent,
+                    property: userData.property,
+                    role: userData.role
+                });
+                return result;
+            }
     }
 };