@lobehub/chat 1.142.0 → 1.142.2

package/src/features/Conversation/Messages/Assistant/MessageContent.tsx

@@ -1,3 +1,4 @@
+import { UIChatMessage } from '@lobechat/types';
 import { ReactNode, memo } from 'react';
 import { Flexbox } from 'react-layout-kit';
 
@@ -7,7 +8,6 @@ import ImageFileListViewer from '@/features/Conversation/Messages/User/ImageFile
 import VideoFileListViewer from '@/features/Conversation/Messages/User/VideoFileListViewer';
 import { useChatStore } from '@/store/chat';
 import { aiChatSelectors, chatSelectors } from '@/store/chat/selectors';
-import { ChatMessage } from '@/types/message';
 
 import { DefaultMessage } from '../Default';
 import FileChunks from './FileChunks';
@@ -17,7 +17,7 @@ import SearchGrounding from './SearchGrounding';
 import Tool from './Tool';
 
 export const AssistantMessageContent = memo<
-  ChatMessage & {
+  UIChatMessage & {
     editableContent: ReactNode;
   }
 >(({ id, tools, content, chunksList, search, imageList, videoList, children, ...props }) => {

package/src/features/Conversation/Messages/Assistant/Tool/Render/CustomRender.tsx

@@ -1,3 +1,4 @@
+import { UIChatMessage } from '@lobechat/types';
 import { ActionIcon } from '@lobehub/ui';
 import { App } from 'antd';
 import { Edit3Icon, PlayCircleIcon } from 'lucide-react';
@@ -10,7 +11,6 @@ import PluginResult from '@/features/Conversation/Messages/Assistant/Tool/Inspec
 import PluginRender from '@/features/PluginsUI/Render';
 import { useChatStore } from '@/store/chat';
 import { chatSelectors } from '@/store/chat/selectors';
-import { ChatMessage } from '@/types/message';
 
 import Arguments from './Arguments';
 import KeyValueEditor from './KeyValueEditor';
@@ -24,7 +24,7 @@ const safeParseJson = (str: string): Record<string, any> => {
   }
 };
 
-interface CustomRenderProps extends ChatMessage {
+interface CustomRenderProps extends UIChatMessage {
   requestArgs?: string;
   setShowPluginRender: (value: boolean) => void;
   showPluginRender: boolean;

package/src/features/Conversation/Messages/Assistant/Tool/Render/ErrorResponse.tsx

@@ -1,10 +1,9 @@
+import { ChatMessageError, ChatPluginPayload } from '@lobechat/types';
 import { Alert, Highlighter } from '@lobehub/ui';
 import { memo } from 'react';
 import { useTranslation } from 'react-i18next';
 import { Flexbox } from 'react-layout-kit';
 
-import { ChatMessageError, ChatPluginPayload } from '@/types/message';
-
 import PluginSettings from './PluginSettings';
 
 interface ErrorResponseProps extends ChatMessageError {

package/src/features/Conversation/Messages/Assistant/Tool/Render/PluginSettings.tsx

@@ -1,3 +1,4 @@
+import { ChatPluginPayload } from '@lobechat/types';
 import { Avatar, Button } from '@lobehub/ui';
 import { Divider } from 'antd';
 import { useTheme } from 'antd-style';
@@ -10,7 +11,6 @@ import PluginSettingsConfig from '@/features/PluginSettings';
 import { useChatStore } from '@/store/chat';
 import { pluginHelpers, useToolStore } from '@/store/tool';
 import { pluginSelectors } from '@/store/tool/selectors';
-import { ChatPluginPayload } from '@/types/message';
 
 import { ErrorActionContainer, useStyles } from '../../../../Error/style';
 

package/src/features/Conversation/Messages/Assistant/index.tsx

@@ -1,6 +1,6 @@
 'use client';
 
-import { ChatMessage } from '@lobechat/types';
+import { UIChatMessage } from '@lobechat/types';
 import { Tag } from '@lobehub/ui';
 import { useResponsive } from 'antd-style';
 import { ReactNode, memo, useCallback, useMemo } from 'react';
@@ -47,7 +47,7 @@ const isHtmlCode = (content: string, language: string) => {
 };
 const MOBILE_AVATAR_SIZE = 32;
 
-interface AssistantMessageProps extends ChatMessage {
+interface AssistantMessageProps extends UIChatMessage {
   disableEditing?: boolean;
   index: number;
   showTitle?: boolean;

package/src/features/Conversation/Messages/Default.tsx

@@ -1,13 +1,13 @@
+import { UIChatMessage } from '@lobechat/types';
 import { ReactNode, memo } from 'react';
 
 import BubblesLoading from '@/components/BubblesLoading';
 import { LOADING_FLAT } from '@/const/message';
 import { useChatStore } from '@/store/chat';
 import { chatSelectors } from '@/store/chat/selectors';
-import { ChatMessage } from '@/types/message';
 
 export const DefaultMessage = memo<
-  ChatMessage & {
+  UIChatMessage & {
     addIdOnDOM?: boolean;
     editableContent: ReactNode;
     isToolCallGenerating?: boolean;
@@ -22,6 +22,6 @@ export const DefaultMessage = memo<
   return <div id={addIdOnDOM ? id : undefined}>{editableContent}</div>;
 });
 
-export const DefaultBelowMessage = memo<ChatMessage>(() => {
+export const DefaultBelowMessage = memo<UIChatMessage>(() => {
   return null;
 });

package/src/features/Conversation/Messages/Supervisor/index.tsx

@@ -1,5 +1,6 @@
 'use client';
 
+import { UIChatMessage } from '@lobechat/types';
 import { ModelIcon } from '@lobehub/icons';
 import { Button, Text } from '@lobehub/ui';
 import { createStyles, useTheme } from 'antd-style';
@@ -12,7 +13,6 @@ import { DEFAULT_SUPERVISOR_AVATAR } from '@/const/meta';
 import { ChatItem } from '@/features/ChatItem';
 import { useChatStore } from '@/store/chat';
 import { ChatErrorType } from '@/types/fetch';
-import { ChatMessage } from '@/types/message';
 
 import TodoList, { TodoData } from './TodoList';
 
@@ -54,7 +54,7 @@ const parseMarkdownTodos = (content: string): TodoData => {
   };
 };
 
-interface SupervisorMessageProps extends ChatMessage {
+interface SupervisorMessageProps extends UIChatMessage {
   disableEditing?: boolean;
   index: number;
 }

package/src/features/Conversation/Messages/User/Actions.tsx

@@ -1,4 +1,4 @@
-import { ChatMessage } from '@lobechat/types';
+import { UIChatMessage } from '@lobechat/types';
 import { ActionIconGroup } from '@lobehub/ui';
 import { ActionIconGroupItemType } from '@lobehub/ui/es/ActionIconGroup';
 import { ActionIconGroupEvent } from '@lobehub/ui/es/ActionIconGroup/type';
@@ -17,7 +17,7 @@ import { InPortalThreadContext } from '../../context/InPortalThreadContext';
 import { useChatListActionsBar } from '../../hooks/useChatListActionsBar';
 
 interface UserActionsProps {
-  data: ChatMessage;
+  data: UIChatMessage;
   id: string;
   index: number;
 }

package/src/features/Conversation/Messages/User/FileListViewer/Item.tsx

@@ -1,3 +1,4 @@
+import { ChatFileItem } from '@lobechat/types';
 import { Text } from '@lobehub/ui';
 import { createStyles } from 'antd-style';
 import { memo } from 'react';
@@ -5,7 +6,6 @@ import { Flexbox } from 'react-layout-kit';
 
 import FileIcon from '@/components/FileIcon';
 import { useChatStore } from '@/store/chat';
-import { ChatFileItem } from '@/types/message';
 import { formatSize } from '@/utils/format';
 
 const useStyles = createStyles(({ css, token, isDarkMode }) => ({

package/src/features/Conversation/Messages/User/FileListViewer/index.tsx

@@ -1,8 +1,7 @@
+import { ChatFileItem } from '@lobechat/types';
 import { memo } from 'react';
 import { Flexbox } from 'react-layout-kit';
 
-import { ChatFileItem } from '@/types/message';
-
 import FileItem from './Item';
 
 interface FileListViewerProps {

package/src/features/Conversation/Messages/User/MessageContent.tsx

@@ -1,14 +1,13 @@
+import { UIChatMessage } from '@lobechat/types';
 import { ReactNode, memo } from 'react';
 import { Flexbox } from 'react-layout-kit';
 
-import { ChatMessage } from '@/types/message';
-
 import FileListViewer from './FileListViewer';
 import ImageFileListViewer from './ImageFileListViewer';
 import VideoFileListViewer from './VideoFileListViewer';
 
 export const UserMessageContent = memo<
-  ChatMessage & {
+  UIChatMessage & {
     editableContent: ReactNode;
   }
 >(({ id, editableContent, imageList, videoList, fileList }) => {

package/src/features/Conversation/Messages/User/VideoFileListViewer.tsx

@@ -1,8 +1,7 @@
+import { ChatVideoItem } from '@lobechat/types';
 import { memo } from 'react';
 import { Flexbox } from 'react-layout-kit';
 
-import { ChatVideoItem } from '@/types/message';
-
 interface VideoFileListViewerProps {
   items: ChatVideoItem[];
 }

package/src/features/Conversation/Messages/User/index.tsx

@@ -1,4 +1,4 @@
-import { ChatMessage } from '@lobechat/types';
+import { UIChatMessage } from '@lobechat/types';
 import { Tag } from '@lobehub/ui';
 import { useResponsive } from 'antd-style';
 import { ReactNode, memo, useCallback, useMemo } from 'react';
@@ -28,7 +28,7 @@ import { UserMessageExtra } from './Extra';
 import { MarkdownRender as UserMarkdownRender } from './MarkdownRender';
 import { UserMessageContent } from './MessageContent';
 
-interface UserMessageProps extends ChatMessage {
+interface UserMessageProps extends UIChatMessage {
   disableEditing?: boolean;
   index: number;
 }

package/src/features/Conversation/components/ChatItem/ShareMessageModal/SharePdf/index.tsx

@@ -1,3 +1,4 @@
+import { UIChatMessage } from '@lobechat/types';
 import { Button } from '@lobehub/ui';
 import { App } from 'antd';
 import { DownloadIcon, FileText } from 'lucide-react';
@@ -7,7 +8,6 @@ import { Flexbox } from 'react-layout-kit';
 
 import { useIsMobile } from '@/hooks/useIsMobile';
 import { useChatStore } from '@/store/chat';
-import { ChatMessage } from '@/types/message';
 
 import PdfPreview from './PdfPreview';
 import { useContainerStyles, useStyles } from './style';
@@ -15,7 +15,7 @@ import { generateMarkdown } from './template';
 import { usePdfGeneration } from './usePdfGeneration';
 
 interface SharePdfProps {
-  message: ChatMessage;
+  message: UIChatMessage;
 }
 
 const SharePdf = memo<SharePdfProps>(({ message }) => {

package/src/features/Conversation/components/ChatItem/ShareMessageModal/SharePdf/template.ts

@@ -1,7 +1,7 @@
+import { UIChatMessage } from '@lobechat/types';
 import { template } from 'lodash-es';
 
 import { LOADING_FLAT } from '@/const/message';
-import { ChatMessage } from '@/types/message';
 
 const markdownTemplate = template(`{{message.content}}`, {
   evaluate: /<%([\S\s]+?)%>/g,
@@ -9,7 +9,7 @@ const markdownTemplate = template(`{{message.content}}`, {
 });
 
 interface MarkdownParams {
-  message: ChatMessage;
+  message: UIChatMessage;
 }
 
 export const generateMarkdown = ({ message }: MarkdownParams) => {

package/src/features/Conversation/components/Extras/TTS/InitPlayer.tsx

@@ -1,10 +1,10 @@
+import { ChatMessageError, ChatTTS } from '@lobechat/types';
 import { memo, useCallback, useEffect, useState } from 'react';
 import { useTranslation } from 'react-i18next';
 
 import { useTTS } from '@/hooks/useTTS';
 import { useChatStore } from '@/store/chat';
 import { useFileStore } from '@/store/file';
-import { ChatMessageError, ChatTTS } from '@/types/message';
 import { getMessageError } from '@/utils/fetch';
 
 import Player from './Player';

package/src/features/Conversation/components/Extras/TTS/Player.tsx

@@ -1,3 +1,4 @@
+import { ChatMessageError } from '@lobechat/types';
 import { AudioPlayer, AudioPlayerProps } from '@lobehub/tts/react';
 import { ActionIcon, Alert, Button, Highlighter } from '@lobehub/ui';
 import { TrashIcon } from 'lucide-react';
@@ -5,8 +6,6 @@ import { memo } from 'react';
 import { useTranslation } from 'react-i18next';
 import { Flexbox } from 'react-layout-kit';
 
-import { ChatMessageError } from '@/types/message';
-
 interface PlayerProps extends AudioPlayerProps {
   error?: ChatMessageError;
   onDelete: () => void;

package/src/features/Conversation/components/Extras/Translate.tsx

@@ -1,3 +1,4 @@
+import { ChatTranslate } from '@lobechat/types';
 import { ActionIcon, Icon, Markdown, Tag, copyToClipboard } from '@lobehub/ui';
 import { App } from 'antd';
 import { useTheme } from 'antd-style';
@@ -8,7 +9,6 @@ import { Flexbox } from 'react-layout-kit';
 
 import BubblesLoading from '@/components/BubblesLoading';
 import { useChatStore } from '@/store/chat';
-import { ChatTranslate } from '@/types/message';
 
 interface TranslateProps extends ChatTranslate {
   id: string;

package/src/features/Conversation/components/Extras/Usage/UsageDetail/index.tsx

@@ -1,3 +1,4 @@
+import { MessageMetadata } from '@lobechat/types';
 import { Icon } from '@lobehub/ui';
 import { Divider, Popover } from 'antd';
 import { useTheme } from 'antd-style';
@@ -10,7 +11,6 @@ import InfoTooltip from '@/components/InfoTooltip';
 import { aiModelSelectors, useAiInfraStore } from '@/store/aiInfra';
 import { useGlobalStore } from '@/store/global';
 import { systemStatusSelectors } from '@/store/global/selectors';
-import { MessageMetadata } from '@/types/message';
 import { formatNumber } from '@/utils/format';
 
 import ModelCard from './ModelCard';

package/src/features/Conversation/components/Extras/Usage/UsageDetail/tokens.ts

@@ -1,6 +1,6 @@
+import type { ModelTokensUsage } from '@lobechat/types';
 import { LobeDefaultAiModelListItem } from 'model-bank';
 
-import type { ModelTokensUsage } from '@/types/message';
 import { getAudioInputUnitRate, getAudioOutputUnitRate } from '@/utils/pricing';
 
 import { getPrice } from './pricing';

package/src/features/Conversation/components/Extras/Usage/index.tsx

@@ -1,10 +1,9 @@
+import { MessageMetadata } from '@lobechat/types';
 import { ModelIcon } from '@lobehub/icons';
 import { createStyles } from 'antd-style';
 import { memo } from 'react';
 import { Center, Flexbox } from 'react-layout-kit';
 
-import { MessageMetadata } from '@/types/message';
-
 import TokenDetail from './UsageDetail';
 
 export const useStyles = createStyles(({ token, css, cx }) => ({

package/src/features/Conversation/components/ShareMessageModal/ShareImage/Preview.tsx

@@ -1,3 +1,4 @@
+import { UIChatMessage } from '@lobechat/types';
 import { ModelTag } from '@lobehub/icons';
 import { Avatar } from '@lobehub/ui';
 import { ChatHeaderTitle } from '@lobehub/ui/chat';
@@ -12,7 +13,6 @@ import { useAgentStore } from '@/store/agent';
 import { agentSelectors } from '@/store/agent/selectors';
 import { useSessionStore } from '@/store/session';
 import { sessionMetaSelectors, sessionSelectors } from '@/store/session/selectors';
-import { ChatMessage } from '@/types/message';
 
 import pkg from '../../../../../../package.json';
 import { useContainerStyles } from '../style';
@@ -20,7 +20,7 @@ import { useStyles } from './style';
 import { FieldType } from './type';
 
 interface PreviewProps extends FieldType {
-  message: ChatMessage;
+  message: UIChatMessage;
   previewId?: string;
   title?: string;
 }

package/src/features/Conversation/components/ShareMessageModal/ShareImage/index.tsx

@@ -1,3 +1,4 @@
+import { UIChatMessage } from '@lobechat/types';
 import { Button, Form, type FormItemProps, Segmented } from '@lobehub/ui';
 import { Switch } from 'antd';
 import { CopyIcon } from 'lucide-react';
@@ -11,7 +12,6 @@ import { useIsMobile } from '@/hooks/useIsMobile';
 import { ImageType, imageTypeOptions, useScreenshot } from '@/hooks/useScreenshot';
 import { useSessionStore } from '@/store/session';
 import { sessionMetaSelectors } from '@/store/session/selectors';
-import { ChatMessage } from '@/types/message';
 
 import { useStyles } from '../style';
 import Preview from './Preview';
@@ -23,7 +23,7 @@ const DEFAULT_FIELD_VALUE: FieldType = {
   withFooter: true,
 };
 
-const ShareImage = memo<{ message: ChatMessage; mobile?: boolean; uniqueId?: string }>(
+const ShareImage = memo<{ message: UIChatMessage; mobile?: boolean; uniqueId?: string }>(
   ({ message, uniqueId }) => {
     const currentAgentTitle = useSessionStore(sessionMetaSelectors.currentAgentTitle);
     const [fieldValue, setFieldValue] = useState<FieldType>(DEFAULT_FIELD_VALUE);

package/src/features/Conversation/components/ShareMessageModal/ShareText/index.tsx

@@ -1,3 +1,4 @@
+import { UIChatMessage } from '@lobechat/types';
 import { Button, copyToClipboard } from '@lobehub/ui';
 import { App } from 'antd';
 import isEqual from 'fast-deep-equal';
@@ -9,7 +10,6 @@ import { Flexbox } from 'react-layout-kit';
 import { useIsMobile } from '@/hooks/useIsMobile';
 import { useChatStore } from '@/store/chat';
 import { topicSelectors } from '@/store/chat/selectors';
-import { ChatMessage } from '@/types/message';
 import { exportFile } from '@/utils/client';
 
 import { useStyles } from '../style';
@@ -17,7 +17,7 @@ import Preview from './Preview';
 import { generateMarkdown } from './template';
 
 interface ShareTextProps {
-  item: ChatMessage;
+  item: UIChatMessage;
 }
 
 const ShareText = memo<ShareTextProps>(({ item }) => {

package/src/features/Conversation/components/ShareMessageModal/ShareText/template.test.ts

@@ -1,7 +1,7 @@
+import { UIChatMessage } from '@lobechat/types';
 import { describe, expect, it } from 'vitest';
 
 import { LOADING_FLAT } from '@/const/message';
-import { ChatMessage } from '@/types/message';
 
 import { generateMarkdown } from './template';
 
@@ -40,7 +40,7 @@ describe('generateMarkdown', () => {
       createdAt: Date.now(),
       tools: [{ name: 'calculator', result: '42' }],
     },
-  ] as ChatMessage[];
+  ] as UIChatMessage[];
 
   const defaultParams = {
     messages: mockMessages,
@@ -66,7 +66,7 @@ describe('generateMarkdown', () => {
         role: 'user',
         createdAt: Date.now(),
       },
-    ] as ChatMessage[];
+    ] as UIChatMessage[];
 
     const result = generateMarkdown({
       ...defaultParams,

package/src/features/Conversation/components/ShareMessageModal/ShareText/template.ts

@@ -1,7 +1,7 @@
+import { UIChatMessage } from '@lobechat/types';
 import { template } from 'lodash-es';
 
 import { LOADING_FLAT } from '@/const/message';
-import { ChatMessage } from '@/types/message';
 
 const markdownTemplate = template(
   `<% messages.forEach(function(chat) { %>
@@ -17,7 +17,7 @@ const markdownTemplate = template(
 );
 
 interface MarkdownParams {
-  messages: ChatMessage[];
+  messages: UIChatMessage[];
 }
 
 export const generateMarkdown = ({ messages }: MarkdownParams) =>

package/src/features/Conversation/components/ShareMessageModal/index.tsx

@@ -1,15 +1,15 @@
+import { UIChatMessage } from '@lobechat/types';
 import { Modal, Segmented, Tabs } from '@lobehub/ui';
 import { memo, useId, useMemo, useState } from 'react';
 import { useTranslation } from 'react-i18next';
 import { Flexbox } from 'react-layout-kit';
 
 import { isServerMode } from '@/const/version';
+import SharePdf from '@/features/ShareModal/SharePdf';
 import { useIsMobile } from '@/hooks/useIsMobile';
-import { ChatMessage } from '@/types/message';
 
 import ShareImage from './ShareImage';
 import ShareText from './ShareText';
-import SharePdf from '@/features/ShareModal/SharePdf';
 
 enum Tab {
   PDF = 'pdf',
@@ -18,7 +18,7 @@ enum Tab {
 }
 
 interface ShareModalProps {
-  message: ChatMessage;
+  message: UIChatMessage;
   onCancel: () => void;
   open: boolean;
 }

package/src/features/Conversation/types/index.ts

@@ -1,13 +1,13 @@
+import { UIChatMessage } from '@lobechat/types';
 import { type ChatItemProps } from '@lobehub/ui/chat';
 import { FC, ReactNode } from 'react';
 
 import { LLMRoleType } from '@/types/llm';
-import { ChatMessage } from '@/types/message';
 
 export type RenderRole = LLMRoleType | 'default' | 'history' | string;
-export type RenderMessage = FC<ChatMessage & { editableContent: ReactNode }>;
-export type RenderBelowMessage = FC<ChatMessage>;
-export type RenderMessageExtra = FC<ChatMessage>;
+export type RenderMessage = FC<UIChatMessage & { editableContent: ReactNode }>;
+export type RenderBelowMessage = FC<UIChatMessage>;
+export type RenderMessageExtra = FC<UIChatMessage>;
 export type MarkdownCustomRender = (props: {
   displayMode: 'chat' | 'docs';
   dom: ReactNode;
@@ -15,7 +15,7 @@ export type MarkdownCustomRender = (props: {
   text: string;
 }) => ReactNode;
 
-export type RenderItem = FC<{ key: string } & ChatMessage & ListItemProps>;
+export type RenderItem = FC<{ key: string } & UIChatMessage & ListItemProps>;
 
 export interface ListItemProps {
   groupNav?: ChatItemProps['avatarAddon'];

package/src/features/Portal/Home/Body/Files/FileList/Item.tsx

@@ -1,3 +1,4 @@
+import { ChatFileItem } from '@lobechat/types';
 import { Text } from '@lobehub/ui';
 import { createStyles } from 'antd-style';
 import { memo } from 'react';
@@ -5,7 +6,6 @@ import { Flexbox } from 'react-layout-kit';
 
 import FileIcon from '@/components/FileIcon';
 import { useChatStore } from '@/store/chat';
-import { ChatFileItem } from '@/types/message';
 import { formatSize } from '@/utils/format';
 
 const useStyles = createStyles(({ css, token }) => ({

package/src/features/Portal/Home/Body/Plugins/ArtifactList/Item/index.tsx

@@ -1,3 +1,4 @@
+import { ChatPluginPayload } from '@lobechat/types';
 import { Icon, Tag, Text } from '@lobehub/ui';
 import isEqual from 'fast-deep-equal';
 import { CircuitBoard } from 'lucide-react';
@@ -10,7 +11,6 @@ import { useYamlArguments } from '@/hooks/useYamlArguments';
 import { useChatStore } from '@/store/chat';
 import { pluginHelpers, useToolStore } from '@/store/tool';
 import { toolSelectors } from '@/store/tool/selectors';
-import { ChatPluginPayload } from '@/types/message';
 
 import { useStyles } from './style';
 

package/src/features/Portal/Thread/Chat/ChatInput/useSend.ts

@@ -1,3 +1,4 @@
+import { SendMessageParams } from '@lobechat/types';
 import { useMemo, useState } from 'react';
 
 import { useGeminiChineseWarning } from '@/hooks/useGeminiChineseWarning';
@@ -5,7 +6,6 @@ import { getAgentStoreState } from '@/store/agent';
 import { agentSelectors } from '@/store/agent/slices/chat';
 import { useChatStore } from '@/store/chat';
 import { threadSelectors } from '@/store/chat/selectors';
-import { SendMessageParams } from '@/types/message';
 
 export type UseSendMessageParams = Pick<
   SendMessageParams,

package/src/features/ShareModal/ShareJSON/generateMessages.test.ts

@@ -1,7 +1,7 @@
+import { UIChatMessage } from '@lobechat/types';
 import { describe, expect, it } from 'vitest';
 
 import { LOADING_FLAT } from '@/const/message';
-import { ChatMessage } from '@/types/message';
 
 import { generateMessages } from './generateMessages';
 
@@ -33,7 +33,7 @@ describe('generateMessages', () => {
       createdAt: Date.now(),
       tool_call_id: 'tool1',
     },
-  ] as ChatMessage[];
+  ] as UIChatMessage[];
 
   it('should filter out loading messages', () => {
     const result = generateMessages({
@@ -109,7 +109,7 @@ describe('generateMessages', () => {
         role: 'user',
         createdAt: Date.now(),
       },
-    ] as ChatMessage[];
+    ] as UIChatMessage[];
 
     const result = generateMessages({
       messages: messagesWithSpaces,

package/src/features/ShareModal/ShareJSON/generateMessages.ts

@@ -1,10 +1,11 @@
+import { UIChatMessage } from '@lobechat/types';
+
 import { LOADING_FLAT } from '@/const/message';
-import { ChatMessage } from '@/types/message';
 
 import { FieldType } from './type';
 
 interface JSONParams extends FieldType {
-  messages: ChatMessage[];
+  messages: UIChatMessage[];
   systemRole: string;
 }
 export const generateMessages = ({

package/src/features/ShareModal/SharePdf/index.tsx

@@ -1,3 +1,4 @@
+import { UIChatMessage } from '@lobechat/types';
 import { Button, Form, type FormItemProps } from '@lobehub/ui';
 import { App, Switch } from 'antd';
 import isEqual from 'fast-deep-equal';
@@ -18,7 +19,6 @@ import { FieldType } from '../ShareText/type';
 import { useContainerStyles, useStyles } from '../style';
 import PdfPreview from './PdfPreview';
 import { usePdfGeneration } from './usePdfGeneration';
-import { ChatMessage } from '@/types/message';
 
 const DEFAULT_FIELD_VALUE: FieldType = {
   includeTool: true,
@@ -27,7 +27,7 @@ const DEFAULT_FIELD_VALUE: FieldType = {
   withSystemRole: false,
 };
 
-const SharePdf = memo((props: {message?: ChatMessage}) => {
+const SharePdf = memo((props: { message?: UIChatMessage }) => {
   const [fieldValue, setFieldValue] = useState(DEFAULT_FIELD_VALUE);
   const { t } = useTranslation(['chat', 'common']);
   const { styles } = useStyles();

package/src/features/ShareModal/ShareText/template.test.ts

@@ -1,7 +1,7 @@
+import { UIChatMessage } from '@lobechat/types';
 import { describe, expect, it } from 'vitest';
 
 import { LOADING_FLAT } from '@/const/message';
-import { ChatMessage } from '@/types/message';
 
 import { generateMarkdown } from './template';
 
@@ -40,7 +40,7 @@ describe('generateMarkdown', () => {
       createdAt: Date.now(),
       tools: [{ name: 'calculator', result: '42' }],
     },
-  ] as ChatMessage[];
+  ] as UIChatMessage[];
 
   const defaultParams = {
     messages: mockMessages,
@@ -166,7 +166,7 @@ describe('generateMarkdown', () => {
         role: 'user',
         createdAt: Date.now(),
       },
-    ] as ChatMessage[];
+    ] as UIChatMessage[];
 
     const result = generateMarkdown({
       ...defaultParams,