@chat-lab/ui 0.1.0-beta.53 → 0.1.0-beta.55

package/dist/utils/convertToAssistantMessage.cjs

@@ -5,6 +5,7 @@ var without = require('../node_modules/.pnpm/lodash-es@4.17.21/node_modules/loda
 var uniq = require('../node_modules/.pnpm/lodash-es@4.17.21/node_modules/lodash-es/uniq.cjs');
 var compact = require('../node_modules/.pnpm/lodash-es@4.17.21/node_modules/lodash-es/compact.cjs');
 var last = require('../node_modules/.pnpm/lodash-es@4.17.21/node_modules/lodash-es/last.cjs');
+var uniqueId = require('../node_modules/.pnpm/lodash-es@4.17.21/node_modules/lodash-es/uniqueId.cjs');
 var keys = require('../node_modules/.pnpm/lodash-es@4.17.21/node_modules/lodash-es/keys.cjs');
 
 const convertToAssistantMessage = (messages, options = {
@@ -29,6 +30,7 @@ const convertToAssistantMessage = (messages, options = {
         };
         const reasoningParts = [];
         const authorFirstMessage = new Set();
+        const attaches = [];
         const content = [...cur.content]
             .sort((a, b) => {
             const aAuthor = a.author;
@@ -48,6 +50,28 @@ const convertToAssistantMessage = (messages, options = {
                 authorFirstMessage.add(author);
             }
             if (part.type === 'text') {
+                try {
+                    const text = part.text;
+                    if (text?.[0] === '[' &&
+                        text?.[text.length - 1] === ']' &&
+                        cur.role === 'user') {
+                        const files = text.slice(1, -1).split(',');
+                        files.forEach(file => {
+                            attaches.push({
+                                id: uniqueId(),
+                                type: 'file',
+                                status: {
+                                    type: 'complete',
+                                },
+                                contentType: 'plain/text',
+                                name: file,
+                                content: [],
+                            });
+                        });
+                        return acc;
+                    }
+                }
+                catch (e) { }
                 if (lastPart &&
                     lastPart.type === 'text' &&
                     lastPart.author === author) {
@@ -80,7 +104,7 @@ const convertToAssistantMessage = (messages, options = {
                 }
             }
             else if (part.type === 'image') {
-                acc.push({
+                attaches.push({
                     type: part.type,
                     image: part.imageUrl.url,
                     ...(authors.length > 2 && { showAvatar, marginLeft }),
@@ -90,12 +114,12 @@ const convertToAssistantMessage = (messages, options = {
                 const data = part.fileUrl ||
                     `data:${part.mimeType};base64,${part.fileData}` ||
                     '';
-                acc.push({
+                attaches.push({
                     type: part.type,
                     data,
                     filename: part.name,
                     mimeType: part.mimeType,
-                    ...(authors.length > 2 && { showAvatar, marginLeft }),
+                    // ...(authors.length > 2 && { showAvatar, marginLeft }),
                 });
             }
             else if (part.type === 'tool_call') {
@@ -153,19 +177,22 @@ const convertToAssistantMessage = (messages, options = {
             id: cur.id,
             role: cur.role,
             status,
-            attachments: cur.attaches?.map(item => ({
-                status: { type: 'complete' },
-                id: item.id,
-                name: item.id,
-                type: 'image',
-                contentType: item.mimeType,
-                content: [
-                    {
-                        type: 'image',
-                        image: item.url,
-                    },
-                ],
-            })),
+            attachments: [
+                ...(cur.attaches?.map(item => ({
+                    status: { type: 'complete' },
+                    id: item.id,
+                    name: item.id,
+                    type: 'file',
+                    contentType: item.mimeType,
+                    content: [
+                        {
+                            type: item.type,
+                            image: item.url,
+                        },
+                    ],
+                })) || []),
+                ...attaches,
+            ],
             content: compact([
                 ...reasoningParts,
                 cur.role === 'assistant' &&

package/dist/utils/convertToAssistantMessage.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"convertToAssistantMessage.cjs","sources":["../../src/utils/convertToAssistantMessage.ts"],"sourcesContent":["import {\n  MessageStatus,\n  ReasoningMessagePart,\n  ThreadAssistantMessagePart,\n  ThreadMessageLike,\n  ToolCallMessagePart,\n} from '@assistant-ui/react';\nimport {\n  ChatMessage,\n  MessageStatus as MessageStatusEnum,\n} from '@chat-lab/core';\nimport { compact, keys, last, uniq, without } from 'lodash-es';\ntype Mutable<T> = {\n  -readonly [K in keyof T]: T[K]; // -readonly 表示移除 readonly 修饰符\n};\n\nconst convertToAssistantMessage = (\n  messages: Readonly<ChatMessage>[],\n  options: { toolDisplay?: 'group' | 'none' | 'step' } = {\n    toolDisplay: 'group',\n  },\n) => {\n  const authors = without(\n    uniq(\n      messages\n        .map(message =>\n          compact(\n            message.content.map(\n              part => part.type === 'text' && (part as any)?.author,\n            ),\n          ),\n        )\n        .flat(),\n    ),\n    'user',\n  );\n\n  const uiMessages = messages.map<ThreadMessageLike>(cur => {\n    let status: MessageStatus | undefined = undefined;\n    if (cur.role === 'assistant') {\n      status = ![MessageStatusEnum.PENDING, MessageStatusEnum.SENDING].includes(\n        cur.status,\n      )\n        ? { type: 'complete', reason: 'stop' }\n        : { type: 'running' };\n    }\n\n    const toolGroup: ToolCallMessagePart<Record<string, any>, null> = {\n      type: 'tool-call',\n      toolCallId: `tool-group-${cur.id}`,\n      toolName: 'tool-group',\n      args: {},\n      argsText: JSON.stringify({}),\n    };\n\n    const reasoningParts: (Mutable<ReasoningMessagePart> & {\n      author: string;\n      authorIdx: number;\n    })[] = [];\n\n    const authorFirstMessage = new Set();\n    const content = [...cur.content]\n      .sort((a, b) => {\n        const aAuthor = (a as any).author;\n        const bAuthor = (b as any).author;\n        if (aAuthor && bAuthor) {\n          return authors.indexOf(aAuthor) - authors.indexOf(bAuthor);\n        }\n        return 0;\n      })\n      .reduce((acc, part, index) => {\n        const lastPart = last(acc);\n        const author = (part as any).author;\n        const authorIdx = authors.indexOf(author);\n        const showAvatar =\n          authors.length >= 2 && !authorFirstMessage.has(author);\n        const marginLeft =\n          authors.length >= 2 && authorFirstMessage.has(author);\n\n        if (author && showAvatar) {\n          authorFirstMessage.add(author);\n        }\n\n        if (part.type === 'text') {\n          if (\n            lastPart &&\n            lastPart.type === 'text' &&\n            (lastPart as any).author === author\n          ) {\n            lastPart.text += '\\n' + part.text;\n          } else {\n            acc.push({\n              type: part.type,\n              text: part.text,\n              author,\n              authorIdx,\n              ...(authors.length >= 2 && { showAvatar, marginLeft }),\n            });\n          }\n        } else if (part.type === 'reasoning') {\n          const lastReasoningPart = last(reasoningParts);\n          if (\n            lastReasoningPart &&\n            (lastReasoningPart as any).author === author\n          ) {\n            lastReasoningPart.text += part.text;\n          } else {\n            reasoningParts.push({\n              type: 'reasoning',\n              text: part.text,\n              author,\n              authorIdx,\n              ...(authors.length > 2 && { showAvatar, marginLeft }),\n            });\n          }\n        } else if (part.type === 'image') {\n          acc.push({\n            type: part.type,\n            image: part.imageUrl.url,\n            ...(authors.length > 2 && { showAvatar, marginLeft }),\n          });\n        } else if (part.type === 'file') {\n          const data =\n            part.fileUrl ||\n            `data:${part.mimeType};base64,${part.fileData}` ||\n            '';\n          acc.push({\n            type: part.type,\n            data,\n            filename: part.name,\n            mimeType: part.mimeType,\n            ...(authors.length > 2 && { showAvatar, marginLeft }),\n          });\n        } else if (part.type === 'tool_call') {\n          if (options.toolDisplay === 'group') {\n            toolGroup.args[part.toolCallId] = {\n              type: 'tool-call',\n              index,\n              toolCallId: part.toolCallId,\n              toolName: part.toolCallName,\n              args: part.toolCallArgs as Record<string, string>,\n              argsText: JSON.stringify(part.toolCallArgs),\n              result: part.content,\n              author,\n              authorIdx,\n              ...(authors.length >= 2 && { showAvatar, marginLeft }),\n            };\n          } else if (options.toolDisplay === 'step') {\n            acc.push({\n              type: 'tool-call',\n              toolCallId: part.toolCallId,\n              toolName: part.toolCallName,\n              args: part.toolCallArgs as Record<string, string>,\n              argsText: JSON.stringify(part.toolCallArgs),\n              result: part.content,\n              author,\n              authorIdx,\n              ...(authors.length >= 2 && { showAvatar, marginLeft }),\n            });\n          }\n        }\n        return acc;\n      }, [] as any[]);\n\n    if (cur.status === MessageStatusEnum.ERROR || cur.error) {\n      return {\n        id: cur.id,\n        role: cur.role,\n        status: {\n          type: 'incomplete',\n          reason: 'error',\n          error:\n            (cur.error as { message: string }).message || (cur.error as any),\n        },\n        content: compact([\n          cur.role === 'assistant' &&\n          options.toolDisplay === 'group' &&\n          keys(toolGroup.args).length > 0\n            ? toolGroup\n            : undefined,\n          ...(content as ThreadAssistantMessagePart[]),\n        ]),\n        metadata: cur.metadata,\n      };\n    }\n\n    const uiMessage: Mutable<\n      Omit<ThreadMessageLike, 'content'> & {\n        content: Exclude<ThreadMessageLike['content'], string>;\n      }\n    > = {\n      id: cur.id,\n      role: cur.role,\n      status,\n      attachments: cur.attaches?.map(item => ({\n        status: { type: 'complete' },\n        id: item.id,\n        name: item.id,\n        type: 'image',\n        contentType: item.mimeType,\n        content: [\n          {\n            type: 'image',\n            image: item.url,\n          },\n        ],\n      })),\n      content: compact([\n        ...reasoningParts,\n        cur.role === 'assistant' &&\n        options.toolDisplay === 'group' &&\n        keys(toolGroup.args).length > 0\n          ? toolGroup\n          : undefined,\n        ...(content as ThreadAssistantMessagePart[]),\n      ]),\n      metadata: {\n        ...cur.metadata,\n        invocationIds:\n          keys(toolGroup.args).length > 0 ? keys(toolGroup.args) : undefined,\n      } as any,\n    };\n    return uiMessage;\n  });\n\n  return { uiMessages, authors };\n};\n\nexport default convertToAssistantMessage;\n"],"names":["MessageStatusEnum"],"mappings":";;;;;;;;;AAgBA,MAAM,yBAAyB,GAAG,CAChC,QAAiC,EACjC,OAAuD,GAAA;AACrD,IAAA,WAAW,EAAE,OAAO;AACrB,CAAA,KACC;AACF,IAAA,MAAM,OAAO,GAAG,OAAO,CACrB,IAAI,CACF,QAAQ;AACL,SAAA,GAAG,CAAC,OAAO,IACV,OAAO,CACL,OAAO,CAAC,OAAO,CAAC,GAAG,CACjB,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAK,IAAY,EAAE,MAAM,CACtD,CACF,CACF;AACA,SAAA,IAAI,EAAE,CACV,EACD,MAAM,CACP,CAAC;IAEF,MAAM,UAAU,GAAG,QAAQ,CAAC,GAAG,CAAoB,GAAG,IAAG;QACvD,IAAI,MAAM,GAA8B,SAAS,CAAC;AAClD,QAAA,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE;AAC5B,YAAA,MAAM,GAAG,CAAC,CAACA,mBAAiB,CAAC,OAAO,EAAEA,mBAAiB,CAAC,OAAO,CAAC,CAAC,QAAQ,CACvE,GAAG,CAAC,MAAM,CACX;kBACG,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE;AACtC,kBAAE,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;SACzB;AAED,QAAA,MAAM,SAAS,GAAmD;AAChE,YAAA,IAAI,EAAE,WAAW;AACjB,YAAA,UAAU,EAAE,CAAA,WAAA,EAAc,GAAG,CAAC,EAAE,CAAE,CAAA;AAClC,YAAA,QAAQ,EAAE,YAAY;AACtB,YAAA,IAAI,EAAE,EAAE;AACR,YAAA,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;SAC7B,CAAC;QAEF,MAAM,cAAc,GAGb,EAAE,CAAC;AAEV,QAAA,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAAE,CAAC;AACrC,QAAA,MAAM,OAAO,GAAG,CAAC,GAAG,GAAG,CAAC,OAAO,CAAC;AAC7B,aAAA,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,KAAI;AACb,YAAA,MAAM,OAAO,GAAI,CAAS,CAAC,MAAM,CAAC;AAClC,YAAA,MAAM,OAAO,GAAI,CAAS,CAAC,MAAM,CAAC;AAClC,YAAA,IAAI,OAAO,IAAI,OAAO,EAAE;AACtB,gBAAA,OAAO,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;aAC5D;AACD,YAAA,OAAO,CAAC,CAAC;AACX,SAAC,CAAC;aACD,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,KAAK,KAAI;AAC3B,YAAA,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;AAC3B,YAAA,MAAM,MAAM,GAAI,IAAY,CAAC,MAAM,CAAC;YACpC,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;AAC1C,YAAA,MAAM,UAAU,GACd,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;AACzD,YAAA,MAAM,UAAU,GACd,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;AAExD,YAAA,IAAI,MAAM,IAAI,UAAU,EAAE;AACxB,gBAAA,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;aAChC;AAED,YAAA,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE;AACxB,gBAAA,IACE,QAAQ;oBACR,QAAQ,CAAC,IAAI,KAAK,MAAM;AACvB,oBAAA,QAAgB,CAAC,MAAM,KAAK,MAAM,EACnC;oBACA,QAAQ,CAAC,IAAI,IAAI,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;iBACnC;qBAAM;oBACL,GAAG,CAAC,IAAI,CAAC;wBACP,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,MAAM;wBACN,SAAS;AACT,wBAAA,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;AACvD,qBAAA,CAAC,CAAC;iBACJ;aACF;AAAM,iBAAA,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE;AACpC,gBAAA,MAAM,iBAAiB,GAAG,IAAI,CAAC,cAAc,CAAC,CAAC;AAC/C,gBAAA,IACE,iBAAiB;AAChB,oBAAA,iBAAyB,CAAC,MAAM,KAAK,MAAM,EAC5C;AACA,oBAAA,iBAAiB,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;iBACrC;qBAAM;oBACL,cAAc,CAAC,IAAI,CAAC;AAClB,wBAAA,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,MAAM;wBACN,SAAS;AACT,wBAAA,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;AACtD,qBAAA,CAAC,CAAC;iBACJ;aACF;AAAM,iBAAA,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE;gBAChC,GAAG,CAAC,IAAI,CAAC;oBACP,IAAI,EAAE,IAAI,CAAC,IAAI;AACf,oBAAA,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG;AACxB,oBAAA,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;AACtD,iBAAA,CAAC,CAAC;aACJ;AAAM,iBAAA,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE;AAC/B,gBAAA,MAAM,IAAI,GACR,IAAI,CAAC,OAAO;AACZ,oBAAA,CAAA,KAAA,EAAQ,IAAI,CAAC,QAAQ,WAAW,IAAI,CAAC,QAAQ,CAAE,CAAA;AAC/C,oBAAA,EAAE,CAAC;gBACL,GAAG,CAAC,IAAI,CAAC;oBACP,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,IAAI;oBACJ,QAAQ,EAAE,IAAI,CAAC,IAAI;oBACnB,QAAQ,EAAE,IAAI,CAAC,QAAQ;AACvB,oBAAA,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;AACtD,iBAAA,CAAC,CAAC;aACJ;AAAM,iBAAA,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE;AACpC,gBAAA,IAAI,OAAO,CAAC,WAAW,KAAK,OAAO,EAAE;AACnC,oBAAA,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG;AAChC,wBAAA,IAAI,EAAE,WAAW;wBACjB,KAAK;wBACL,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,QAAQ,EAAE,IAAI,CAAC,YAAY;wBAC3B,IAAI,EAAE,IAAI,CAAC,YAAsC;wBACjD,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,YAAY,CAAC;wBAC3C,MAAM,EAAE,IAAI,CAAC,OAAO;wBACpB,MAAM;wBACN,SAAS;AACT,wBAAA,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;qBACvD,CAAC;iBACH;AAAM,qBAAA,IAAI,OAAO,CAAC,WAAW,KAAK,MAAM,EAAE;oBACzC,GAAG,CAAC,IAAI,CAAC;AACP,wBAAA,IAAI,EAAE,WAAW;wBACjB,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,QAAQ,EAAE,IAAI,CAAC,YAAY;wBAC3B,IAAI,EAAE,IAAI,CAAC,YAAsC;wBACjD,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,YAAY,CAAC;wBAC3C,MAAM,EAAE,IAAI,CAAC,OAAO;wBACpB,MAAM;wBACN,SAAS;AACT,wBAAA,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;AACvD,qBAAA,CAAC,CAAC;iBACJ;aACF;AACD,YAAA,OAAO,GAAG,CAAC;SACZ,EAAE,EAAW,CAAC,CAAC;AAElB,QAAA,IAAI,GAAG,CAAC,MAAM,KAAKA,mBAAiB,CAAC,KAAK,IAAI,GAAG,CAAC,KAAK,EAAE;YACvD,OAAO;gBACL,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,IAAI,EAAE,GAAG,CAAC,IAAI;AACd,gBAAA,MAAM,EAAE;AACN,oBAAA,IAAI,EAAE,YAAY;AAClB,oBAAA,MAAM,EAAE,OAAO;oBACf,KAAK,EACF,GAAG,CAAC,KAA6B,CAAC,OAAO,IAAK,GAAG,CAAC,KAAa;AACnE,iBAAA;gBACD,OAAO,EAAE,OAAO,CAAC;oBACf,GAAG,CAAC,IAAI,KAAK,WAAW;wBACxB,OAAO,CAAC,WAAW,KAAK,OAAO;wBAC/B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC;AAC7B,0BAAE,SAAS;AACX,0BAAE,SAAS;AACb,oBAAA,GAAI,OAAwC;iBAC7C,CAAC;gBACF,QAAQ,EAAE,GAAG,CAAC,QAAQ;aACvB,CAAC;SACH;AAED,QAAA,MAAM,SAAS,GAIX;YACF,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,MAAM;YACN,WAAW,EAAE,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,IAAI,KAAK;AACtC,gBAAA,MAAM,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE;gBAC5B,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,IAAI,EAAE,IAAI,CAAC,EAAE;AACb,gBAAA,IAAI,EAAE,OAAO;gBACb,WAAW,EAAE,IAAI,CAAC,QAAQ;AAC1B,gBAAA,OAAO,EAAE;AACP,oBAAA;AACE,wBAAA,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,IAAI,CAAC,GAAG;AAChB,qBAAA;AACF,iBAAA;AACF,aAAA,CAAC,CAAC;YACH,OAAO,EAAE,OAAO,CAAC;AACf,gBAAA,GAAG,cAAc;gBACjB,GAAG,CAAC,IAAI,KAAK,WAAW;oBACxB,OAAO,CAAC,WAAW,KAAK,OAAO;oBAC/B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC;AAC7B,sBAAE,SAAS;AACX,sBAAE,SAAS;AACb,gBAAA,GAAI,OAAwC;aAC7C,CAAC;AACF,YAAA,QAAQ,EAAE;gBACR,GAAG,GAAG,CAAC,QAAQ;gBACf,aAAa,EACX,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,SAAS;AAC9D,aAAA;SACT,CAAC;AACF,QAAA,OAAO,SAAS,CAAC;AACnB,KAAC,CAAC,CAAC;AAEH,IAAA,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC;AACjC;;;;"}
+{"version":3,"file":"convertToAssistantMessage.cjs","sources":["../../src/utils/convertToAssistantMessage.ts"],"sourcesContent":["import {\n  CompleteAttachment,\n  MessageStatus,\n  ReasoningMessagePart,\n  ThreadAssistantMessagePart,\n  ThreadMessageLike,\n  ToolCallMessagePart,\n} from '@assistant-ui/react';\nimport {\n  ChatMessage,\n  MessageStatus as MessageStatusEnum,\n} from '@chat-lab/core';\nimport { compact, keys, last, uniq, uniqueId, without } from 'lodash-es';\ntype Mutable<T> = {\n  -readonly [K in keyof T]: T[K]; // -readonly 表示移除 readonly 修饰符\n};\n\nconst convertToAssistantMessage = (\n  messages: Readonly<ChatMessage>[],\n  options: { toolDisplay?: 'group' | 'none' | 'step' } = {\n    toolDisplay: 'group',\n  },\n) => {\n  const authors = without(\n    uniq(\n      messages\n        .map(message =>\n          compact(\n            message.content.map(\n              part => part.type === 'text' && (part as any)?.author,\n            ),\n          ),\n        )\n        .flat(),\n    ),\n    'user',\n  );\n\n  const uiMessages = messages.map<ThreadMessageLike>(cur => {\n    let status: MessageStatus | undefined = undefined;\n    if (cur.role === 'assistant') {\n      status = ![MessageStatusEnum.PENDING, MessageStatusEnum.SENDING].includes(\n        cur.status,\n      )\n        ? { type: 'complete', reason: 'stop' }\n        : { type: 'running' };\n    }\n\n    const toolGroup: ToolCallMessagePart<Record<string, any>, null> = {\n      type: 'tool-call',\n      toolCallId: `tool-group-${cur.id}`,\n      toolName: 'tool-group',\n      args: {},\n      argsText: JSON.stringify({}),\n    };\n\n    const reasoningParts: (Mutable<ReasoningMessagePart> & {\n      author: string;\n      authorIdx: number;\n    })[] = [];\n\n    const authorFirstMessage = new Set();\n\n    const attaches: CompleteAttachment[] = [];\n    const content = [...cur.content]\n      .sort((a, b) => {\n        const aAuthor = (a as any).author;\n        const bAuthor = (b as any).author;\n        if (aAuthor && bAuthor) {\n          return authors.indexOf(aAuthor) - authors.indexOf(bAuthor);\n        }\n        return 0;\n      })\n      .reduce((acc, part, index) => {\n        const lastPart = last(acc);\n        const author = (part as any).author;\n        const authorIdx = authors.indexOf(author);\n        const showAvatar =\n          authors.length >= 2 && !authorFirstMessage.has(author);\n        const marginLeft =\n          authors.length >= 2 && authorFirstMessage.has(author);\n\n        if (author && showAvatar) {\n          authorFirstMessage.add(author);\n        }\n\n        if (part.type === 'text') {\n          try {\n            const text = part.text;\n            if (\n              text?.[0] === '[' &&\n              text?.[text.length - 1] === ']' &&\n              cur.role === 'user'\n            ) {\n              const files = text.slice(1, -1).split(',');\n              files.forEach(file => {\n                attaches.push({\n                  id: uniqueId(),\n                  type: 'file',\n                  status: {\n                    type: 'complete',\n                  },\n                  contentType: 'plain/text',\n                  name: file,\n                  content: [],\n                });\n              });\n\n              return acc;\n            }\n          } catch (e) {}\n          if (\n            lastPart &&\n            lastPart.type === 'text' &&\n            (lastPart as any).author === author\n          ) {\n            lastPart.text += '\\n' + part.text;\n          } else {\n            acc.push({\n              type: part.type,\n              text: part.text,\n              author,\n              authorIdx,\n              ...(authors.length >= 2 && { showAvatar, marginLeft }),\n            });\n          }\n        } else if (part.type === 'reasoning') {\n          const lastReasoningPart = last(reasoningParts);\n          if (\n            lastReasoningPart &&\n            (lastReasoningPart as any).author === author\n          ) {\n            lastReasoningPart.text += part.text;\n          } else {\n            reasoningParts.push({\n              type: 'reasoning',\n              text: part.text,\n              author,\n              authorIdx,\n              ...(authors.length > 2 && { showAvatar, marginLeft }),\n            });\n          }\n        } else if (part.type === 'image') {\n          attaches.push({\n            type: part.type,\n            image: part.imageUrl.url,\n            ...(authors.length > 2 && { showAvatar, marginLeft }),\n          });\n        } else if (part.type === 'file') {\n          const data =\n            part.fileUrl ||\n            `data:${part.mimeType};base64,${part.fileData}` ||\n            '';\n          attaches.push({\n            type: part.type,\n            data,\n            filename: part.name,\n            mimeType: part.mimeType,\n            // ...(authors.length > 2 && { showAvatar, marginLeft }),\n          });\n        } else if (part.type === 'tool_call') {\n          if (options.toolDisplay === 'group') {\n            toolGroup.args[part.toolCallId] = {\n              type: 'tool-call',\n              index,\n              toolCallId: part.toolCallId,\n              toolName: part.toolCallName,\n              args: part.toolCallArgs as Record<string, string>,\n              argsText: JSON.stringify(part.toolCallArgs),\n              result: part.content,\n              author,\n              authorIdx,\n              ...(authors.length >= 2 && { showAvatar, marginLeft }),\n            };\n          } else if (options.toolDisplay === 'step') {\n            acc.push({\n              type: 'tool-call',\n              toolCallId: part.toolCallId,\n              toolName: part.toolCallName,\n              args: part.toolCallArgs as Record<string, string>,\n              argsText: JSON.stringify(part.toolCallArgs),\n              result: part.content,\n              author,\n              authorIdx,\n              ...(authors.length >= 2 && { showAvatar, marginLeft }),\n            });\n          }\n        }\n        return acc;\n      }, [] as any[]);\n\n    if (cur.status === MessageStatusEnum.ERROR || cur.error) {\n      return {\n        id: cur.id,\n        role: cur.role,\n        status: {\n          type: 'incomplete',\n          reason: 'error',\n          error:\n            (cur.error as { message: string }).message || (cur.error as any),\n        },\n        content: compact([\n          cur.role === 'assistant' &&\n          options.toolDisplay === 'group' &&\n          keys(toolGroup.args).length > 0\n            ? toolGroup\n            : undefined,\n          ...(content as ThreadAssistantMessagePart[]),\n        ]),\n        metadata: cur.metadata,\n      };\n    }\n\n    const uiMessage: Mutable<\n      Omit<ThreadMessageLike, 'content'> & {\n        content: Exclude<ThreadMessageLike['content'], string>;\n      }\n    > = {\n      id: cur.id,\n      role: cur.role,\n      status,\n      attachments: [\n        ...((cur.attaches?.map(item => ({\n          status: { type: 'complete' },\n          id: item.id,\n          name: item.id,\n          type: 'file',\n          contentType: item.mimeType,\n          content: [\n            {\n              type: item.type,\n              image: item.url,\n            },\n          ],\n        })) || []) as CompleteAttachment[]),\n        ...attaches,\n      ],\n      content: compact([\n        ...reasoningParts,\n        cur.role === 'assistant' &&\n        options.toolDisplay === 'group' &&\n        keys(toolGroup.args).length > 0\n          ? toolGroup\n          : undefined,\n        ...(content as ThreadAssistantMessagePart[]),\n      ]),\n      metadata: {\n        ...cur.metadata,\n        invocationIds:\n          keys(toolGroup.args).length > 0 ? keys(toolGroup.args) : undefined,\n      } as any,\n    };\n    return uiMessage;\n  });\n\n  return { uiMessages, authors };\n};\n\nexport default convertToAssistantMessage;\n"],"names":["MessageStatusEnum"],"mappings":";;;;;;;;;;AAiBA,MAAM,yBAAyB,GAAG,CAChC,QAAiC,EACjC,OAAuD,GAAA;AACrD,IAAA,WAAW,EAAE,OAAO;AACrB,CAAA,KACC;AACF,IAAA,MAAM,OAAO,GAAG,OAAO,CACrB,IAAI,CACF,QAAQ;AACL,SAAA,GAAG,CAAC,OAAO,IACV,OAAO,CACL,OAAO,CAAC,OAAO,CAAC,GAAG,CACjB,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAK,IAAY,EAAE,MAAM,CACtD,CACF,CACF;AACA,SAAA,IAAI,EAAE,CACV,EACD,MAAM,CACP,CAAC;IAEF,MAAM,UAAU,GAAG,QAAQ,CAAC,GAAG,CAAoB,GAAG,IAAG;QACvD,IAAI,MAAM,GAA8B,SAAS,CAAC;AAClD,QAAA,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE;AAC5B,YAAA,MAAM,GAAG,CAAC,CAACA,mBAAiB,CAAC,OAAO,EAAEA,mBAAiB,CAAC,OAAO,CAAC,CAAC,QAAQ,CACvE,GAAG,CAAC,MAAM,CACX;kBACG,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE;AACtC,kBAAE,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;SACzB;AAED,QAAA,MAAM,SAAS,GAAmD;AAChE,YAAA,IAAI,EAAE,WAAW;AACjB,YAAA,UAAU,EAAE,CAAA,WAAA,EAAc,GAAG,CAAC,EAAE,CAAE,CAAA;AAClC,YAAA,QAAQ,EAAE,YAAY;AACtB,YAAA,IAAI,EAAE,EAAE;AACR,YAAA,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;SAC7B,CAAC;QAEF,MAAM,cAAc,GAGb,EAAE,CAAC;AAEV,QAAA,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAAE,CAAC;QAErC,MAAM,QAAQ,GAAyB,EAAE,CAAC;AAC1C,QAAA,MAAM,OAAO,GAAG,CAAC,GAAG,GAAG,CAAC,OAAO,CAAC;AAC7B,aAAA,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,KAAI;AACb,YAAA,MAAM,OAAO,GAAI,CAAS,CAAC,MAAM,CAAC;AAClC,YAAA,MAAM,OAAO,GAAI,CAAS,CAAC,MAAM,CAAC;AAClC,YAAA,IAAI,OAAO,IAAI,OAAO,EAAE;AACtB,gBAAA,OAAO,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;aAC5D;AACD,YAAA,OAAO,CAAC,CAAC;AACX,SAAC,CAAC;aACD,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,KAAK,KAAI;AAC3B,YAAA,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;AAC3B,YAAA,MAAM,MAAM,GAAI,IAAY,CAAC,MAAM,CAAC;YACpC,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;AAC1C,YAAA,MAAM,UAAU,GACd,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;AACzD,YAAA,MAAM,UAAU,GACd,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;AAExD,YAAA,IAAI,MAAM,IAAI,UAAU,EAAE;AACxB,gBAAA,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;aAChC;AAED,YAAA,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE;AACxB,gBAAA,IAAI;AACF,oBAAA,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;AACvB,oBAAA,IACE,IAAI,GAAG,CAAC,CAAC,KAAK,GAAG;wBACjB,IAAI,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,KAAK,GAAG;AAC/B,wBAAA,GAAG,CAAC,IAAI,KAAK,MAAM,EACnB;AACA,wBAAA,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;AAC3C,wBAAA,KAAK,CAAC,OAAO,CAAC,IAAI,IAAG;4BACnB,QAAQ,CAAC,IAAI,CAAC;gCACZ,EAAE,EAAE,QAAQ,EAAE;AACd,gCAAA,IAAI,EAAE,MAAM;AACZ,gCAAA,MAAM,EAAE;AACN,oCAAA,IAAI,EAAE,UAAU;AACjB,iCAAA;AACD,gCAAA,WAAW,EAAE,YAAY;AACzB,gCAAA,IAAI,EAAE,IAAI;AACV,gCAAA,OAAO,EAAE,EAAE;AACZ,6BAAA,CAAC,CAAC;AACL,yBAAC,CAAC,CAAC;AAEH,wBAAA,OAAO,GAAG,CAAC;qBACZ;iBACF;AAAC,gBAAA,OAAO,CAAC,EAAE,GAAE;AACd,gBAAA,IACE,QAAQ;oBACR,QAAQ,CAAC,IAAI,KAAK,MAAM;AACvB,oBAAA,QAAgB,CAAC,MAAM,KAAK,MAAM,EACnC;oBACA,QAAQ,CAAC,IAAI,IAAI,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;iBACnC;qBAAM;oBACL,GAAG,CAAC,IAAI,CAAC;wBACP,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,MAAM;wBACN,SAAS;AACT,wBAAA,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;AACvD,qBAAA,CAAC,CAAC;iBACJ;aACF;AAAM,iBAAA,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE;AACpC,gBAAA,MAAM,iBAAiB,GAAG,IAAI,CAAC,cAAc,CAAC,CAAC;AAC/C,gBAAA,IACE,iBAAiB;AAChB,oBAAA,iBAAyB,CAAC,MAAM,KAAK,MAAM,EAC5C;AACA,oBAAA,iBAAiB,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;iBACrC;qBAAM;oBACL,cAAc,CAAC,IAAI,CAAC;AAClB,wBAAA,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,MAAM;wBACN,SAAS;AACT,wBAAA,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;AACtD,qBAAA,CAAC,CAAC;iBACJ;aACF;AAAM,iBAAA,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE;gBAChC,QAAQ,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,IAAI,CAAC,IAAI;AACf,oBAAA,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG;AACxB,oBAAA,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;AACtD,iBAAA,CAAC,CAAC;aACJ;AAAM,iBAAA,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE;AAC/B,gBAAA,MAAM,IAAI,GACR,IAAI,CAAC,OAAO;AACZ,oBAAA,CAAA,KAAA,EAAQ,IAAI,CAAC,QAAQ,WAAW,IAAI,CAAC,QAAQ,CAAE,CAAA;AAC/C,oBAAA,EAAE,CAAC;gBACL,QAAQ,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,IAAI;oBACJ,QAAQ,EAAE,IAAI,CAAC,IAAI;oBACnB,QAAQ,EAAE,IAAI,CAAC,QAAQ;;AAExB,iBAAA,CAAC,CAAC;aACJ;AAAM,iBAAA,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE;AACpC,gBAAA,IAAI,OAAO,CAAC,WAAW,KAAK,OAAO,EAAE;AACnC,oBAAA,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG;AAChC,wBAAA,IAAI,EAAE,WAAW;wBACjB,KAAK;wBACL,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,QAAQ,EAAE,IAAI,CAAC,YAAY;wBAC3B,IAAI,EAAE,IAAI,CAAC,YAAsC;wBACjD,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,YAAY,CAAC;wBAC3C,MAAM,EAAE,IAAI,CAAC,OAAO;wBACpB,MAAM;wBACN,SAAS;AACT,wBAAA,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;qBACvD,CAAC;iBACH;AAAM,qBAAA,IAAI,OAAO,CAAC,WAAW,KAAK,MAAM,EAAE;oBACzC,GAAG,CAAC,IAAI,CAAC;AACP,wBAAA,IAAI,EAAE,WAAW;wBACjB,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,QAAQ,EAAE,IAAI,CAAC,YAAY;wBAC3B,IAAI,EAAE,IAAI,CAAC,YAAsC;wBACjD,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,YAAY,CAAC;wBAC3C,MAAM,EAAE,IAAI,CAAC,OAAO;wBACpB,MAAM;wBACN,SAAS;AACT,wBAAA,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;AACvD,qBAAA,CAAC,CAAC;iBACJ;aACF;AACD,YAAA,OAAO,GAAG,CAAC;SACZ,EAAE,EAAW,CAAC,CAAC;AAElB,QAAA,IAAI,GAAG,CAAC,MAAM,KAAKA,mBAAiB,CAAC,KAAK,IAAI,GAAG,CAAC,KAAK,EAAE;YACvD,OAAO;gBACL,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,IAAI,EAAE,GAAG,CAAC,IAAI;AACd,gBAAA,MAAM,EAAE;AACN,oBAAA,IAAI,EAAE,YAAY;AAClB,oBAAA,MAAM,EAAE,OAAO;oBACf,KAAK,EACF,GAAG,CAAC,KAA6B,CAAC,OAAO,IAAK,GAAG,CAAC,KAAa;AACnE,iBAAA;gBACD,OAAO,EAAE,OAAO,CAAC;oBACf,GAAG,CAAC,IAAI,KAAK,WAAW;wBACxB,OAAO,CAAC,WAAW,KAAK,OAAO;wBAC/B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC;AAC7B,0BAAE,SAAS;AACX,0BAAE,SAAS;AACb,oBAAA,GAAI,OAAwC;iBAC7C,CAAC;gBACF,QAAQ,EAAE,GAAG,CAAC,QAAQ;aACvB,CAAC;SACH;AAED,QAAA,MAAM,SAAS,GAIX;YACF,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,MAAM;AACN,YAAA,WAAW,EAAE;gBACX,IAAK,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,IAAI,KAAK;AAC9B,oBAAA,MAAM,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE;oBAC5B,EAAE,EAAE,IAAI,CAAC,EAAE;oBACX,IAAI,EAAE,IAAI,CAAC,EAAE;AACb,oBAAA,IAAI,EAAE,MAAM;oBACZ,WAAW,EAAE,IAAI,CAAC,QAAQ;AAC1B,oBAAA,OAAO,EAAE;AACP,wBAAA;4BACE,IAAI,EAAE,IAAI,CAAC,IAAI;4BACf,KAAK,EAAE,IAAI,CAAC,GAAG;AAChB,yBAAA;AACF,qBAAA;iBACF,CAAC,CAAC,IAAI,EAAE,CAA0B;AACnC,gBAAA,GAAG,QAAQ;AACZ,aAAA;YACD,OAAO,EAAE,OAAO,CAAC;AACf,gBAAA,GAAG,cAAc;gBACjB,GAAG,CAAC,IAAI,KAAK,WAAW;oBACxB,OAAO,CAAC,WAAW,KAAK,OAAO;oBAC/B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC;AAC7B,sBAAE,SAAS;AACX,sBAAE,SAAS;AACb,gBAAA,GAAI,OAAwC;aAC7C,CAAC;AACF,YAAA,QAAQ,EAAE;gBACR,GAAG,GAAG,CAAC,QAAQ;gBACf,aAAa,EACX,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,SAAS;AAC9D,aAAA;SACT,CAAC;AACF,QAAA,OAAO,SAAS,CAAC;AACnB,KAAC,CAAC,CAAC;AAEH,IAAA,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC;AACjC;;;;"}

package/dist/utils/convertToAssistantMessage.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"convertToAssistantMessage.d.ts","sourceRoot":"","sources":["../../src/utils/convertToAssistantMessage.ts"],"names":[],"mappings":"AAAA,OAAO,EAIL,iBAAiB,EAElB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,WAAW,EAEZ,MAAM,gBAAgB,CAAC;AAMxB,QAAA,MAAM,yBAAyB,GAC7B,UAAU,QAAQ,CAAC,WAAW,CAAC,EAAE,EACjC,UAAS;IAAE,WAAW,CAAC,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,CAAA;CAEjD;;;CA8MF,CAAC;AAEF,eAAe,yBAAyB,CAAC"}
+{"version":3,"file":"convertToAssistantMessage.d.ts","sourceRoot":"","sources":["../../src/utils/convertToAssistantMessage.ts"],"names":[],"mappings":"AAAA,OAAO,EAKL,iBAAiB,EAElB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,WAAW,EAEZ,MAAM,gBAAgB,CAAC;AAMxB,QAAA,MAAM,yBAAyB,GAC7B,UAAU,QAAQ,CAAC,WAAW,CAAC,EAAE,EACjC,UAAS;IAAE,WAAW,CAAC,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,CAAA;CAEjD;;;CA2OF,CAAC;AAEF,eAAe,yBAAyB,CAAC"}

package/dist/utils/convertToAssistantMessage.js

@@ -3,6 +3,7 @@ import without from '../node_modules/.pnpm/lodash-es@4.17.21/node_modules/lodash
 import uniq from '../node_modules/.pnpm/lodash-es@4.17.21/node_modules/lodash-es/uniq.js';
 import compact from '../node_modules/.pnpm/lodash-es@4.17.21/node_modules/lodash-es/compact.js';
 import last from '../node_modules/.pnpm/lodash-es@4.17.21/node_modules/lodash-es/last.js';
+import uniqueId from '../node_modules/.pnpm/lodash-es@4.17.21/node_modules/lodash-es/uniqueId.js';
 import keys from '../node_modules/.pnpm/lodash-es@4.17.21/node_modules/lodash-es/keys.js';
 
 const convertToAssistantMessage = (messages, options = {
@@ -27,6 +28,7 @@ const convertToAssistantMessage = (messages, options = {
         };
         const reasoningParts = [];
         const authorFirstMessage = new Set();
+        const attaches = [];
         const content = [...cur.content]
             .sort((a, b) => {
             const aAuthor = a.author;
@@ -46,6 +48,28 @@ const convertToAssistantMessage = (messages, options = {
                 authorFirstMessage.add(author);
             }
             if (part.type === 'text') {
+                try {
+                    const text = part.text;
+                    if (text?.[0] === '[' &&
+                        text?.[text.length - 1] === ']' &&
+                        cur.role === 'user') {
+                        const files = text.slice(1, -1).split(',');
+                        files.forEach(file => {
+                            attaches.push({
+                                id: uniqueId(),
+                                type: 'file',
+                                status: {
+                                    type: 'complete',
+                                },
+                                contentType: 'plain/text',
+                                name: file,
+                                content: [],
+                            });
+                        });
+                        return acc;
+                    }
+                }
+                catch (e) { }
                 if (lastPart &&
                     lastPart.type === 'text' &&
                     lastPart.author === author) {
@@ -78,7 +102,7 @@ const convertToAssistantMessage = (messages, options = {
                 }
             }
             else if (part.type === 'image') {
-                acc.push({
+                attaches.push({
                     type: part.type,
                     image: part.imageUrl.url,
                     ...(authors.length > 2 && { showAvatar, marginLeft }),
@@ -88,12 +112,12 @@ const convertToAssistantMessage = (messages, options = {
                 const data = part.fileUrl ||
                     `data:${part.mimeType};base64,${part.fileData}` ||
                     '';
-                acc.push({
+                attaches.push({
                     type: part.type,
                     data,
                     filename: part.name,
                     mimeType: part.mimeType,
-                    ...(authors.length > 2 && { showAvatar, marginLeft }),
+                    // ...(authors.length > 2 && { showAvatar, marginLeft }),
                 });
             }
             else if (part.type === 'tool_call') {
@@ -151,19 +175,22 @@ const convertToAssistantMessage = (messages, options = {
             id: cur.id,
             role: cur.role,
             status,
-            attachments: cur.attaches?.map(item => ({
-                status: { type: 'complete' },
-                id: item.id,
-                name: item.id,
-                type: 'image',
-                contentType: item.mimeType,
-                content: [
-                    {
-                        type: 'image',
-                        image: item.url,
-                    },
-                ],
-            })),
+            attachments: [
+                ...(cur.attaches?.map(item => ({
+                    status: { type: 'complete' },
+                    id: item.id,
+                    name: item.id,
+                    type: 'file',
+                    contentType: item.mimeType,
+                    content: [
+                        {
+                            type: item.type,
+                            image: item.url,
+                        },
+                    ],
+                })) || []),
+                ...attaches,
+            ],
             content: compact([
                 ...reasoningParts,
                 cur.role === 'assistant' &&

package/dist/utils/convertToAssistantMessage.js.map

@@ -1 +1 @@
-{"version":3,"file":"convertToAssistantMessage.js","sources":["../../src/utils/convertToAssistantMessage.ts"],"sourcesContent":["import {\n  MessageStatus,\n  ReasoningMessagePart,\n  ThreadAssistantMessagePart,\n  ThreadMessageLike,\n  ToolCallMessagePart,\n} from '@assistant-ui/react';\nimport {\n  ChatMessage,\n  MessageStatus as MessageStatusEnum,\n} from '@chat-lab/core';\nimport { compact, keys, last, uniq, without } from 'lodash-es';\ntype Mutable<T> = {\n  -readonly [K in keyof T]: T[K]; // -readonly 表示移除 readonly 修饰符\n};\n\nconst convertToAssistantMessage = (\n  messages: Readonly<ChatMessage>[],\n  options: { toolDisplay?: 'group' | 'none' | 'step' } = {\n    toolDisplay: 'group',\n  },\n) => {\n  const authors = without(\n    uniq(\n      messages\n        .map(message =>\n          compact(\n            message.content.map(\n              part => part.type === 'text' && (part as any)?.author,\n            ),\n          ),\n        )\n        .flat(),\n    ),\n    'user',\n  );\n\n  const uiMessages = messages.map<ThreadMessageLike>(cur => {\n    let status: MessageStatus | undefined = undefined;\n    if (cur.role === 'assistant') {\n      status = ![MessageStatusEnum.PENDING, MessageStatusEnum.SENDING].includes(\n        cur.status,\n      )\n        ? { type: 'complete', reason: 'stop' }\n        : { type: 'running' };\n    }\n\n    const toolGroup: ToolCallMessagePart<Record<string, any>, null> = {\n      type: 'tool-call',\n      toolCallId: `tool-group-${cur.id}`,\n      toolName: 'tool-group',\n      args: {},\n      argsText: JSON.stringify({}),\n    };\n\n    const reasoningParts: (Mutable<ReasoningMessagePart> & {\n      author: string;\n      authorIdx: number;\n    })[] = [];\n\n    const authorFirstMessage = new Set();\n    const content = [...cur.content]\n      .sort((a, b) => {\n        const aAuthor = (a as any).author;\n        const bAuthor = (b as any).author;\n        if (aAuthor && bAuthor) {\n          return authors.indexOf(aAuthor) - authors.indexOf(bAuthor);\n        }\n        return 0;\n      })\n      .reduce((acc, part, index) => {\n        const lastPart = last(acc);\n        const author = (part as any).author;\n        const authorIdx = authors.indexOf(author);\n        const showAvatar =\n          authors.length >= 2 && !authorFirstMessage.has(author);\n        const marginLeft =\n          authors.length >= 2 && authorFirstMessage.has(author);\n\n        if (author && showAvatar) {\n          authorFirstMessage.add(author);\n        }\n\n        if (part.type === 'text') {\n          if (\n            lastPart &&\n            lastPart.type === 'text' &&\n            (lastPart as any).author === author\n          ) {\n            lastPart.text += '\\n' + part.text;\n          } else {\n            acc.push({\n              type: part.type,\n              text: part.text,\n              author,\n              authorIdx,\n              ...(authors.length >= 2 && { showAvatar, marginLeft }),\n            });\n          }\n        } else if (part.type === 'reasoning') {\n          const lastReasoningPart = last(reasoningParts);\n          if (\n            lastReasoningPart &&\n            (lastReasoningPart as any).author === author\n          ) {\n            lastReasoningPart.text += part.text;\n          } else {\n            reasoningParts.push({\n              type: 'reasoning',\n              text: part.text,\n              author,\n              authorIdx,\n              ...(authors.length > 2 && { showAvatar, marginLeft }),\n            });\n          }\n        } else if (part.type === 'image') {\n          acc.push({\n            type: part.type,\n            image: part.imageUrl.url,\n            ...(authors.length > 2 && { showAvatar, marginLeft }),\n          });\n        } else if (part.type === 'file') {\n          const data =\n            part.fileUrl ||\n            `data:${part.mimeType};base64,${part.fileData}` ||\n            '';\n          acc.push({\n            type: part.type,\n            data,\n            filename: part.name,\n            mimeType: part.mimeType,\n            ...(authors.length > 2 && { showAvatar, marginLeft }),\n          });\n        } else if (part.type === 'tool_call') {\n          if (options.toolDisplay === 'group') {\n            toolGroup.args[part.toolCallId] = {\n              type: 'tool-call',\n              index,\n              toolCallId: part.toolCallId,\n              toolName: part.toolCallName,\n              args: part.toolCallArgs as Record<string, string>,\n              argsText: JSON.stringify(part.toolCallArgs),\n              result: part.content,\n              author,\n              authorIdx,\n              ...(authors.length >= 2 && { showAvatar, marginLeft }),\n            };\n          } else if (options.toolDisplay === 'step') {\n            acc.push({\n              type: 'tool-call',\n              toolCallId: part.toolCallId,\n              toolName: part.toolCallName,\n              args: part.toolCallArgs as Record<string, string>,\n              argsText: JSON.stringify(part.toolCallArgs),\n              result: part.content,\n              author,\n              authorIdx,\n              ...(authors.length >= 2 && { showAvatar, marginLeft }),\n            });\n          }\n        }\n        return acc;\n      }, [] as any[]);\n\n    if (cur.status === MessageStatusEnum.ERROR || cur.error) {\n      return {\n        id: cur.id,\n        role: cur.role,\n        status: {\n          type: 'incomplete',\n          reason: 'error',\n          error:\n            (cur.error as { message: string }).message || (cur.error as any),\n        },\n        content: compact([\n          cur.role === 'assistant' &&\n          options.toolDisplay === 'group' &&\n          keys(toolGroup.args).length > 0\n            ? toolGroup\n            : undefined,\n          ...(content as ThreadAssistantMessagePart[]),\n        ]),\n        metadata: cur.metadata,\n      };\n    }\n\n    const uiMessage: Mutable<\n      Omit<ThreadMessageLike, 'content'> & {\n        content: Exclude<ThreadMessageLike['content'], string>;\n      }\n    > = {\n      id: cur.id,\n      role: cur.role,\n      status,\n      attachments: cur.attaches?.map(item => ({\n        status: { type: 'complete' },\n        id: item.id,\n        name: item.id,\n        type: 'image',\n        contentType: item.mimeType,\n        content: [\n          {\n            type: 'image',\n            image: item.url,\n          },\n        ],\n      })),\n      content: compact([\n        ...reasoningParts,\n        cur.role === 'assistant' &&\n        options.toolDisplay === 'group' &&\n        keys(toolGroup.args).length > 0\n          ? toolGroup\n          : undefined,\n        ...(content as ThreadAssistantMessagePart[]),\n      ]),\n      metadata: {\n        ...cur.metadata,\n        invocationIds:\n          keys(toolGroup.args).length > 0 ? keys(toolGroup.args) : undefined,\n      } as any,\n    };\n    return uiMessage;\n  });\n\n  return { uiMessages, authors };\n};\n\nexport default convertToAssistantMessage;\n"],"names":["MessageStatusEnum"],"mappings":";;;;;;;AAgBA,MAAM,yBAAyB,GAAG,CAChC,QAAiC,EACjC,OAAuD,GAAA;AACrD,IAAA,WAAW,EAAE,OAAO;AACrB,CAAA,KACC;AACF,IAAA,MAAM,OAAO,GAAG,OAAO,CACrB,IAAI,CACF,QAAQ;AACL,SAAA,GAAG,CAAC,OAAO,IACV,OAAO,CACL,OAAO,CAAC,OAAO,CAAC,GAAG,CACjB,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAK,IAAY,EAAE,MAAM,CACtD,CACF,CACF;AACA,SAAA,IAAI,EAAE,CACV,EACD,MAAM,CACP,CAAC;IAEF,MAAM,UAAU,GAAG,QAAQ,CAAC,GAAG,CAAoB,GAAG,IAAG;QACvD,IAAI,MAAM,GAA8B,SAAS,CAAC;AAClD,QAAA,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE;AAC5B,YAAA,MAAM,GAAG,CAAC,CAACA,aAAiB,CAAC,OAAO,EAAEA,aAAiB,CAAC,OAAO,CAAC,CAAC,QAAQ,CACvE,GAAG,CAAC,MAAM,CACX;kBACG,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE;AACtC,kBAAE,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;SACzB;AAED,QAAA,MAAM,SAAS,GAAmD;AAChE,YAAA,IAAI,EAAE,WAAW;AACjB,YAAA,UAAU,EAAE,CAAA,WAAA,EAAc,GAAG,CAAC,EAAE,CAAE,CAAA;AAClC,YAAA,QAAQ,EAAE,YAAY;AACtB,YAAA,IAAI,EAAE,EAAE;AACR,YAAA,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;SAC7B,CAAC;QAEF,MAAM,cAAc,GAGb,EAAE,CAAC;AAEV,QAAA,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAAE,CAAC;AACrC,QAAA,MAAM,OAAO,GAAG,CAAC,GAAG,GAAG,CAAC,OAAO,CAAC;AAC7B,aAAA,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,KAAI;AACb,YAAA,MAAM,OAAO,GAAI,CAAS,CAAC,MAAM,CAAC;AAClC,YAAA,MAAM,OAAO,GAAI,CAAS,CAAC,MAAM,CAAC;AAClC,YAAA,IAAI,OAAO,IAAI,OAAO,EAAE;AACtB,gBAAA,OAAO,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;aAC5D;AACD,YAAA,OAAO,CAAC,CAAC;AACX,SAAC,CAAC;aACD,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,KAAK,KAAI;AAC3B,YAAA,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;AAC3B,YAAA,MAAM,MAAM,GAAI,IAAY,CAAC,MAAM,CAAC;YACpC,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;AAC1C,YAAA,MAAM,UAAU,GACd,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;AACzD,YAAA,MAAM,UAAU,GACd,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;AAExD,YAAA,IAAI,MAAM,IAAI,UAAU,EAAE;AACxB,gBAAA,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;aAChC;AAED,YAAA,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE;AACxB,gBAAA,IACE,QAAQ;oBACR,QAAQ,CAAC,IAAI,KAAK,MAAM;AACvB,oBAAA,QAAgB,CAAC,MAAM,KAAK,MAAM,EACnC;oBACA,QAAQ,CAAC,IAAI,IAAI,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;iBACnC;qBAAM;oBACL,GAAG,CAAC,IAAI,CAAC;wBACP,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,MAAM;wBACN,SAAS;AACT,wBAAA,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;AACvD,qBAAA,CAAC,CAAC;iBACJ;aACF;AAAM,iBAAA,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE;AACpC,gBAAA,MAAM,iBAAiB,GAAG,IAAI,CAAC,cAAc,CAAC,CAAC;AAC/C,gBAAA,IACE,iBAAiB;AAChB,oBAAA,iBAAyB,CAAC,MAAM,KAAK,MAAM,EAC5C;AACA,oBAAA,iBAAiB,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;iBACrC;qBAAM;oBACL,cAAc,CAAC,IAAI,CAAC;AAClB,wBAAA,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,MAAM;wBACN,SAAS;AACT,wBAAA,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;AACtD,qBAAA,CAAC,CAAC;iBACJ;aACF;AAAM,iBAAA,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE;gBAChC,GAAG,CAAC,IAAI,CAAC;oBACP,IAAI,EAAE,IAAI,CAAC,IAAI;AACf,oBAAA,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG;AACxB,oBAAA,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;AACtD,iBAAA,CAAC,CAAC;aACJ;AAAM,iBAAA,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE;AAC/B,gBAAA,MAAM,IAAI,GACR,IAAI,CAAC,OAAO;AACZ,oBAAA,CAAA,KAAA,EAAQ,IAAI,CAAC,QAAQ,WAAW,IAAI,CAAC,QAAQ,CAAE,CAAA;AAC/C,oBAAA,EAAE,CAAC;gBACL,GAAG,CAAC,IAAI,CAAC;oBACP,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,IAAI;oBACJ,QAAQ,EAAE,IAAI,CAAC,IAAI;oBACnB,QAAQ,EAAE,IAAI,CAAC,QAAQ;AACvB,oBAAA,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;AACtD,iBAAA,CAAC,CAAC;aACJ;AAAM,iBAAA,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE;AACpC,gBAAA,IAAI,OAAO,CAAC,WAAW,KAAK,OAAO,EAAE;AACnC,oBAAA,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG;AAChC,wBAAA,IAAI,EAAE,WAAW;wBACjB,KAAK;wBACL,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,QAAQ,EAAE,IAAI,CAAC,YAAY;wBAC3B,IAAI,EAAE,IAAI,CAAC,YAAsC;wBACjD,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,YAAY,CAAC;wBAC3C,MAAM,EAAE,IAAI,CAAC,OAAO;wBACpB,MAAM;wBACN,SAAS;AACT,wBAAA,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;qBACvD,CAAC;iBACH;AAAM,qBAAA,IAAI,OAAO,CAAC,WAAW,KAAK,MAAM,EAAE;oBACzC,GAAG,CAAC,IAAI,CAAC;AACP,wBAAA,IAAI,EAAE,WAAW;wBACjB,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,QAAQ,EAAE,IAAI,CAAC,YAAY;wBAC3B,IAAI,EAAE,IAAI,CAAC,YAAsC;wBACjD,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,YAAY,CAAC;wBAC3C,MAAM,EAAE,IAAI,CAAC,OAAO;wBACpB,MAAM;wBACN,SAAS;AACT,wBAAA,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;AACvD,qBAAA,CAAC,CAAC;iBACJ;aACF;AACD,YAAA,OAAO,GAAG,CAAC;SACZ,EAAE,EAAW,CAAC,CAAC;AAElB,QAAA,IAAI,GAAG,CAAC,MAAM,KAAKA,aAAiB,CAAC,KAAK,IAAI,GAAG,CAAC,KAAK,EAAE;YACvD,OAAO;gBACL,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,IAAI,EAAE,GAAG,CAAC,IAAI;AACd,gBAAA,MAAM,EAAE;AACN,oBAAA,IAAI,EAAE,YAAY;AAClB,oBAAA,MAAM,EAAE,OAAO;oBACf,KAAK,EACF,GAAG,CAAC,KAA6B,CAAC,OAAO,IAAK,GAAG,CAAC,KAAa;AACnE,iBAAA;gBACD,OAAO,EAAE,OAAO,CAAC;oBACf,GAAG,CAAC,IAAI,KAAK,WAAW;wBACxB,OAAO,CAAC,WAAW,KAAK,OAAO;wBAC/B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC;AAC7B,0BAAE,SAAS;AACX,0BAAE,SAAS;AACb,oBAAA,GAAI,OAAwC;iBAC7C,CAAC;gBACF,QAAQ,EAAE,GAAG,CAAC,QAAQ;aACvB,CAAC;SACH;AAED,QAAA,MAAM,SAAS,GAIX;YACF,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,MAAM;YACN,WAAW,EAAE,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,IAAI,KAAK;AACtC,gBAAA,MAAM,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE;gBAC5B,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,IAAI,EAAE,IAAI,CAAC,EAAE;AACb,gBAAA,IAAI,EAAE,OAAO;gBACb,WAAW,EAAE,IAAI,CAAC,QAAQ;AAC1B,gBAAA,OAAO,EAAE;AACP,oBAAA;AACE,wBAAA,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,IAAI,CAAC,GAAG;AAChB,qBAAA;AACF,iBAAA;AACF,aAAA,CAAC,CAAC;YACH,OAAO,EAAE,OAAO,CAAC;AACf,gBAAA,GAAG,cAAc;gBACjB,GAAG,CAAC,IAAI,KAAK,WAAW;oBACxB,OAAO,CAAC,WAAW,KAAK,OAAO;oBAC/B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC;AAC7B,sBAAE,SAAS;AACX,sBAAE,SAAS;AACb,gBAAA,GAAI,OAAwC;aAC7C,CAAC;AACF,YAAA,QAAQ,EAAE;gBACR,GAAG,GAAG,CAAC,QAAQ;gBACf,aAAa,EACX,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,SAAS;AAC9D,aAAA;SACT,CAAC;AACF,QAAA,OAAO,SAAS,CAAC;AACnB,KAAC,CAAC,CAAC;AAEH,IAAA,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC;AACjC;;;;"}
+{"version":3,"file":"convertToAssistantMessage.js","sources":["../../src/utils/convertToAssistantMessage.ts"],"sourcesContent":["import {\n  CompleteAttachment,\n  MessageStatus,\n  ReasoningMessagePart,\n  ThreadAssistantMessagePart,\n  ThreadMessageLike,\n  ToolCallMessagePart,\n} from '@assistant-ui/react';\nimport {\n  ChatMessage,\n  MessageStatus as MessageStatusEnum,\n} from '@chat-lab/core';\nimport { compact, keys, last, uniq, uniqueId, without } from 'lodash-es';\ntype Mutable<T> = {\n  -readonly [K in keyof T]: T[K]; // -readonly 表示移除 readonly 修饰符\n};\n\nconst convertToAssistantMessage = (\n  messages: Readonly<ChatMessage>[],\n  options: { toolDisplay?: 'group' | 'none' | 'step' } = {\n    toolDisplay: 'group',\n  },\n) => {\n  const authors = without(\n    uniq(\n      messages\n        .map(message =>\n          compact(\n            message.content.map(\n              part => part.type === 'text' && (part as any)?.author,\n            ),\n          ),\n        )\n        .flat(),\n    ),\n    'user',\n  );\n\n  const uiMessages = messages.map<ThreadMessageLike>(cur => {\n    let status: MessageStatus | undefined = undefined;\n    if (cur.role === 'assistant') {\n      status = ![MessageStatusEnum.PENDING, MessageStatusEnum.SENDING].includes(\n        cur.status,\n      )\n        ? { type: 'complete', reason: 'stop' }\n        : { type: 'running' };\n    }\n\n    const toolGroup: ToolCallMessagePart<Record<string, any>, null> = {\n      type: 'tool-call',\n      toolCallId: `tool-group-${cur.id}`,\n      toolName: 'tool-group',\n      args: {},\n      argsText: JSON.stringify({}),\n    };\n\n    const reasoningParts: (Mutable<ReasoningMessagePart> & {\n      author: string;\n      authorIdx: number;\n    })[] = [];\n\n    const authorFirstMessage = new Set();\n\n    const attaches: CompleteAttachment[] = [];\n    const content = [...cur.content]\n      .sort((a, b) => {\n        const aAuthor = (a as any).author;\n        const bAuthor = (b as any).author;\n        if (aAuthor && bAuthor) {\n          return authors.indexOf(aAuthor) - authors.indexOf(bAuthor);\n        }\n        return 0;\n      })\n      .reduce((acc, part, index) => {\n        const lastPart = last(acc);\n        const author = (part as any).author;\n        const authorIdx = authors.indexOf(author);\n        const showAvatar =\n          authors.length >= 2 && !authorFirstMessage.has(author);\n        const marginLeft =\n          authors.length >= 2 && authorFirstMessage.has(author);\n\n        if (author && showAvatar) {\n          authorFirstMessage.add(author);\n        }\n\n        if (part.type === 'text') {\n          try {\n            const text = part.text;\n            if (\n              text?.[0] === '[' &&\n              text?.[text.length - 1] === ']' &&\n              cur.role === 'user'\n            ) {\n              const files = text.slice(1, -1).split(',');\n              files.forEach(file => {\n                attaches.push({\n                  id: uniqueId(),\n                  type: 'file',\n                  status: {\n                    type: 'complete',\n                  },\n                  contentType: 'plain/text',\n                  name: file,\n                  content: [],\n                });\n              });\n\n              return acc;\n            }\n          } catch (e) {}\n          if (\n            lastPart &&\n            lastPart.type === 'text' &&\n            (lastPart as any).author === author\n          ) {\n            lastPart.text += '\\n' + part.text;\n          } else {\n            acc.push({\n              type: part.type,\n              text: part.text,\n              author,\n              authorIdx,\n              ...(authors.length >= 2 && { showAvatar, marginLeft }),\n            });\n          }\n        } else if (part.type === 'reasoning') {\n          const lastReasoningPart = last(reasoningParts);\n          if (\n            lastReasoningPart &&\n            (lastReasoningPart as any).author === author\n          ) {\n            lastReasoningPart.text += part.text;\n          } else {\n            reasoningParts.push({\n              type: 'reasoning',\n              text: part.text,\n              author,\n              authorIdx,\n              ...(authors.length > 2 && { showAvatar, marginLeft }),\n            });\n          }\n        } else if (part.type === 'image') {\n          attaches.push({\n            type: part.type,\n            image: part.imageUrl.url,\n            ...(authors.length > 2 && { showAvatar, marginLeft }),\n          });\n        } else if (part.type === 'file') {\n          const data =\n            part.fileUrl ||\n            `data:${part.mimeType};base64,${part.fileData}` ||\n            '';\n          attaches.push({\n            type: part.type,\n            data,\n            filename: part.name,\n            mimeType: part.mimeType,\n            // ...(authors.length > 2 && { showAvatar, marginLeft }),\n          });\n        } else if (part.type === 'tool_call') {\n          if (options.toolDisplay === 'group') {\n            toolGroup.args[part.toolCallId] = {\n              type: 'tool-call',\n              index,\n              toolCallId: part.toolCallId,\n              toolName: part.toolCallName,\n              args: part.toolCallArgs as Record<string, string>,\n              argsText: JSON.stringify(part.toolCallArgs),\n              result: part.content,\n              author,\n              authorIdx,\n              ...(authors.length >= 2 && { showAvatar, marginLeft }),\n            };\n          } else if (options.toolDisplay === 'step') {\n            acc.push({\n              type: 'tool-call',\n              toolCallId: part.toolCallId,\n              toolName: part.toolCallName,\n              args: part.toolCallArgs as Record<string, string>,\n              argsText: JSON.stringify(part.toolCallArgs),\n              result: part.content,\n              author,\n              authorIdx,\n              ...(authors.length >= 2 && { showAvatar, marginLeft }),\n            });\n          }\n        }\n        return acc;\n      }, [] as any[]);\n\n    if (cur.status === MessageStatusEnum.ERROR || cur.error) {\n      return {\n        id: cur.id,\n        role: cur.role,\n        status: {\n          type: 'incomplete',\n          reason: 'error',\n          error:\n            (cur.error as { message: string }).message || (cur.error as any),\n        },\n        content: compact([\n          cur.role === 'assistant' &&\n          options.toolDisplay === 'group' &&\n          keys(toolGroup.args).length > 0\n            ? toolGroup\n            : undefined,\n          ...(content as ThreadAssistantMessagePart[]),\n        ]),\n        metadata: cur.metadata,\n      };\n    }\n\n    const uiMessage: Mutable<\n      Omit<ThreadMessageLike, 'content'> & {\n        content: Exclude<ThreadMessageLike['content'], string>;\n      }\n    > = {\n      id: cur.id,\n      role: cur.role,\n      status,\n      attachments: [\n        ...((cur.attaches?.map(item => ({\n          status: { type: 'complete' },\n          id: item.id,\n          name: item.id,\n          type: 'file',\n          contentType: item.mimeType,\n          content: [\n            {\n              type: item.type,\n              image: item.url,\n            },\n          ],\n        })) || []) as CompleteAttachment[]),\n        ...attaches,\n      ],\n      content: compact([\n        ...reasoningParts,\n        cur.role === 'assistant' &&\n        options.toolDisplay === 'group' &&\n        keys(toolGroup.args).length > 0\n          ? toolGroup\n          : undefined,\n        ...(content as ThreadAssistantMessagePart[]),\n      ]),\n      metadata: {\n        ...cur.metadata,\n        invocationIds:\n          keys(toolGroup.args).length > 0 ? keys(toolGroup.args) : undefined,\n      } as any,\n    };\n    return uiMessage;\n  });\n\n  return { uiMessages, authors };\n};\n\nexport default convertToAssistantMessage;\n"],"names":["MessageStatusEnum"],"mappings":";;;;;;;;AAiBA,MAAM,yBAAyB,GAAG,CAChC,QAAiC,EACjC,OAAuD,GAAA;AACrD,IAAA,WAAW,EAAE,OAAO;AACrB,CAAA,KACC;AACF,IAAA,MAAM,OAAO,GAAG,OAAO,CACrB,IAAI,CACF,QAAQ;AACL,SAAA,GAAG,CAAC,OAAO,IACV,OAAO,CACL,OAAO,CAAC,OAAO,CAAC,GAAG,CACjB,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAK,IAAY,EAAE,MAAM,CACtD,CACF,CACF;AACA,SAAA,IAAI,EAAE,CACV,EACD,MAAM,CACP,CAAC;IAEF,MAAM,UAAU,GAAG,QAAQ,CAAC,GAAG,CAAoB,GAAG,IAAG;QACvD,IAAI,MAAM,GAA8B,SAAS,CAAC;AAClD,QAAA,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE;AAC5B,YAAA,MAAM,GAAG,CAAC,CAACA,aAAiB,CAAC,OAAO,EAAEA,aAAiB,CAAC,OAAO,CAAC,CAAC,QAAQ,CACvE,GAAG,CAAC,MAAM,CACX;kBACG,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE;AACtC,kBAAE,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;SACzB;AAED,QAAA,MAAM,SAAS,GAAmD;AAChE,YAAA,IAAI,EAAE,WAAW;AACjB,YAAA,UAAU,EAAE,CAAA,WAAA,EAAc,GAAG,CAAC,EAAE,CAAE,CAAA;AAClC,YAAA,QAAQ,EAAE,YAAY;AACtB,YAAA,IAAI,EAAE,EAAE;AACR,YAAA,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;SAC7B,CAAC;QAEF,MAAM,cAAc,GAGb,EAAE,CAAC;AAEV,QAAA,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAAE,CAAC;QAErC,MAAM,QAAQ,GAAyB,EAAE,CAAC;AAC1C,QAAA,MAAM,OAAO,GAAG,CAAC,GAAG,GAAG,CAAC,OAAO,CAAC;AAC7B,aAAA,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,KAAI;AACb,YAAA,MAAM,OAAO,GAAI,CAAS,CAAC,MAAM,CAAC;AAClC,YAAA,MAAM,OAAO,GAAI,CAAS,CAAC,MAAM,CAAC;AAClC,YAAA,IAAI,OAAO,IAAI,OAAO,EAAE;AACtB,gBAAA,OAAO,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;aAC5D;AACD,YAAA,OAAO,CAAC,CAAC;AACX,SAAC,CAAC;aACD,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,KAAK,KAAI;AAC3B,YAAA,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC;AAC3B,YAAA,MAAM,MAAM,GAAI,IAAY,CAAC,MAAM,CAAC;YACpC,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;AAC1C,YAAA,MAAM,UAAU,GACd,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;AACzD,YAAA,MAAM,UAAU,GACd,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;AAExD,YAAA,IAAI,MAAM,IAAI,UAAU,EAAE;AACxB,gBAAA,kBAAkB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;aAChC;AAED,YAAA,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE;AACxB,gBAAA,IAAI;AACF,oBAAA,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;AACvB,oBAAA,IACE,IAAI,GAAG,CAAC,CAAC,KAAK,GAAG;wBACjB,IAAI,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,KAAK,GAAG;AAC/B,wBAAA,GAAG,CAAC,IAAI,KAAK,MAAM,EACnB;AACA,wBAAA,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;AAC3C,wBAAA,KAAK,CAAC,OAAO,CAAC,IAAI,IAAG;4BACnB,QAAQ,CAAC,IAAI,CAAC;gCACZ,EAAE,EAAE,QAAQ,EAAE;AACd,gCAAA,IAAI,EAAE,MAAM;AACZ,gCAAA,MAAM,EAAE;AACN,oCAAA,IAAI,EAAE,UAAU;AACjB,iCAAA;AACD,gCAAA,WAAW,EAAE,YAAY;AACzB,gCAAA,IAAI,EAAE,IAAI;AACV,gCAAA,OAAO,EAAE,EAAE;AACZ,6BAAA,CAAC,CAAC;AACL,yBAAC,CAAC,CAAC;AAEH,wBAAA,OAAO,GAAG,CAAC;qBACZ;iBACF;AAAC,gBAAA,OAAO,CAAC,EAAE,GAAE;AACd,gBAAA,IACE,QAAQ;oBACR,QAAQ,CAAC,IAAI,KAAK,MAAM;AACvB,oBAAA,QAAgB,CAAC,MAAM,KAAK,MAAM,EACnC;oBACA,QAAQ,CAAC,IAAI,IAAI,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;iBACnC;qBAAM;oBACL,GAAG,CAAC,IAAI,CAAC;wBACP,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,MAAM;wBACN,SAAS;AACT,wBAAA,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;AACvD,qBAAA,CAAC,CAAC;iBACJ;aACF;AAAM,iBAAA,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE;AACpC,gBAAA,MAAM,iBAAiB,GAAG,IAAI,CAAC,cAAc,CAAC,CAAC;AAC/C,gBAAA,IACE,iBAAiB;AAChB,oBAAA,iBAAyB,CAAC,MAAM,KAAK,MAAM,EAC5C;AACA,oBAAA,iBAAiB,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;iBACrC;qBAAM;oBACL,cAAc,CAAC,IAAI,CAAC;AAClB,wBAAA,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,MAAM;wBACN,SAAS;AACT,wBAAA,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;AACtD,qBAAA,CAAC,CAAC;iBACJ;aACF;AAAM,iBAAA,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE;gBAChC,QAAQ,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,IAAI,CAAC,IAAI;AACf,oBAAA,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG;AACxB,oBAAA,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;AACtD,iBAAA,CAAC,CAAC;aACJ;AAAM,iBAAA,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE;AAC/B,gBAAA,MAAM,IAAI,GACR,IAAI,CAAC,OAAO;AACZ,oBAAA,CAAA,KAAA,EAAQ,IAAI,CAAC,QAAQ,WAAW,IAAI,CAAC,QAAQ,CAAE,CAAA;AAC/C,oBAAA,EAAE,CAAC;gBACL,QAAQ,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,IAAI;oBACJ,QAAQ,EAAE,IAAI,CAAC,IAAI;oBACnB,QAAQ,EAAE,IAAI,CAAC,QAAQ;;AAExB,iBAAA,CAAC,CAAC;aACJ;AAAM,iBAAA,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE;AACpC,gBAAA,IAAI,OAAO,CAAC,WAAW,KAAK,OAAO,EAAE;AACnC,oBAAA,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG;AAChC,wBAAA,IAAI,EAAE,WAAW;wBACjB,KAAK;wBACL,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,QAAQ,EAAE,IAAI,CAAC,YAAY;wBAC3B,IAAI,EAAE,IAAI,CAAC,YAAsC;wBACjD,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,YAAY,CAAC;wBAC3C,MAAM,EAAE,IAAI,CAAC,OAAO;wBACpB,MAAM;wBACN,SAAS;AACT,wBAAA,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;qBACvD,CAAC;iBACH;AAAM,qBAAA,IAAI,OAAO,CAAC,WAAW,KAAK,MAAM,EAAE;oBACzC,GAAG,CAAC,IAAI,CAAC;AACP,wBAAA,IAAI,EAAE,WAAW;wBACjB,UAAU,EAAE,IAAI,CAAC,UAAU;wBAC3B,QAAQ,EAAE,IAAI,CAAC,YAAY;wBAC3B,IAAI,EAAE,IAAI,CAAC,YAAsC;wBACjD,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,YAAY,CAAC;wBAC3C,MAAM,EAAE,IAAI,CAAC,OAAO;wBACpB,MAAM;wBACN,SAAS;AACT,wBAAA,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC;AACvD,qBAAA,CAAC,CAAC;iBACJ;aACF;AACD,YAAA,OAAO,GAAG,CAAC;SACZ,EAAE,EAAW,CAAC,CAAC;AAElB,QAAA,IAAI,GAAG,CAAC,MAAM,KAAKA,aAAiB,CAAC,KAAK,IAAI,GAAG,CAAC,KAAK,EAAE;YACvD,OAAO;gBACL,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,IAAI,EAAE,GAAG,CAAC,IAAI;AACd,gBAAA,MAAM,EAAE;AACN,oBAAA,IAAI,EAAE,YAAY;AAClB,oBAAA,MAAM,EAAE,OAAO;oBACf,KAAK,EACF,GAAG,CAAC,KAA6B,CAAC,OAAO,IAAK,GAAG,CAAC,KAAa;AACnE,iBAAA;gBACD,OAAO,EAAE,OAAO,CAAC;oBACf,GAAG,CAAC,IAAI,KAAK,WAAW;wBACxB,OAAO,CAAC,WAAW,KAAK,OAAO;wBAC/B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC;AAC7B,0BAAE,SAAS;AACX,0BAAE,SAAS;AACb,oBAAA,GAAI,OAAwC;iBAC7C,CAAC;gBACF,QAAQ,EAAE,GAAG,CAAC,QAAQ;aACvB,CAAC;SACH;AAED,QAAA,MAAM,SAAS,GAIX;YACF,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,MAAM;AACN,YAAA,WAAW,EAAE;gBACX,IAAK,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,IAAI,KAAK;AAC9B,oBAAA,MAAM,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE;oBAC5B,EAAE,EAAE,IAAI,CAAC,EAAE;oBACX,IAAI,EAAE,IAAI,CAAC,EAAE;AACb,oBAAA,IAAI,EAAE,MAAM;oBACZ,WAAW,EAAE,IAAI,CAAC,QAAQ;AAC1B,oBAAA,OAAO,EAAE;AACP,wBAAA;4BACE,IAAI,EAAE,IAAI,CAAC,IAAI;4BACf,KAAK,EAAE,IAAI,CAAC,GAAG;AAChB,yBAAA;AACF,qBAAA;iBACF,CAAC,CAAC,IAAI,EAAE,CAA0B;AACnC,gBAAA,GAAG,QAAQ;AACZ,aAAA;YACD,OAAO,EAAE,OAAO,CAAC;AACf,gBAAA,GAAG,cAAc;gBACjB,GAAG,CAAC,IAAI,KAAK,WAAW;oBACxB,OAAO,CAAC,WAAW,KAAK,OAAO;oBAC/B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC;AAC7B,sBAAE,SAAS;AACX,sBAAE,SAAS;AACb,gBAAA,GAAI,OAAwC;aAC7C,CAAC;AACF,YAAA,QAAQ,EAAE;gBACR,GAAG,GAAG,CAAC,QAAQ;gBACf,aAAa,EACX,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,SAAS;AAC9D,aAAA;SACT,CAAC;AACF,QAAA,OAAO,SAAS,CAAC;AACnB,KAAC,CAAC,CAAC;AAEH,IAAA,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC;AACjC;;;;"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@chat-lab/ui",
-  "version": "0.1.0-beta.53",
+  "version": "0.1.0-beta.55",
   "type": "module",
   "main": "dist/index.cjs",
   "module": "dist/index.js",