@lobehub/chat 1.142.1 → 1.142.2

package/src/services/chat/chat.test.ts

@@ -1,5 +1,7 @@
 import { LobeTool } from '@lobechat/types';
-import { ChatImageItem, ChatMessage } from '@lobechat/types';
+import { UIChatMessage } from '@lobechat/types';
+import { ChatErrorType } from '@lobechat/types';
+import { ChatStreamPayload } from '@lobechat/types';
 import { LobeChatPluginManifest } from '@lobehub/chat-plugin-sdk';
 import { act } from '@testing-library/react';
 import { type Mock, afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
@@ -15,8 +17,6 @@ import { toolSelectors } from '@/store/tool/selectors';
 import { modelProviderSelectors } from '@/store/user/selectors';
 import { DalleManifest } from '@/tools/dalle';
 import { WebBrowsingManifest } from '@/tools/web-browsing';
-import { ChatErrorType } from '@/types/index';
-import { ChatStreamPayload, type OpenAIChatMessage } from '@/types/openai/chat';
 
 import { API_ENDPOINTS } from '../_url';
 import * as helpers from './helper';
@@ -80,7 +80,7 @@ describe('ChatService', () => {
   describe('createAssistantMessage', () => {
     it('should process messages and call getChatCompletion with the right parameters', async () => {
       const getChatCompletionSpy = vi.spyOn(chatService, 'getChatCompletion');
-      const messages = [{ content: 'Hello', role: 'user' }] as ChatMessage[];
+      const messages = [{ content: 'Hello', role: 'user' }] as UIChatMessage[];
       const enabledPlugins = ['plugin1'];
       await act(async () => {
         useToolStore.setState({
@@ -127,7 +127,7 @@ describe('ChatService', () => {
     describe('extendParams functionality', () => {
       it('should add reasoning parameters when model supports enableReasoning and user enables it', async () => {
         const getChatCompletionSpy = vi.spyOn(chatService, 'getChatCompletion');
-        const messages = [{ content: 'Test reasoning', role: 'user' }] as ChatMessage[];
+        const messages = [{ content: 'Test reasoning', role: 'user' }] as UIChatMessage[];
 
         // Mock aiModelSelectors for extend params support
         vi.spyOn(aiModelSelectors, 'isModelHasExtendParams').mockReturnValue(() => true);
@@ -160,7 +160,7 @@ describe('ChatService', () => {
 
       it('should disable reasoning when model supports enableReasoning but user disables it', async () => {
         const getChatCompletionSpy = vi.spyOn(chatService, 'getChatCompletion');
-        const messages = [{ content: 'Test no reasoning', role: 'user' }] as ChatMessage[];
+        const messages = [{ content: 'Test no reasoning', role: 'user' }] as UIChatMessage[];
 
         // Mock aiModelSelectors for extend params support
         vi.spyOn(aiModelSelectors, 'isModelHasExtendParams').mockReturnValue(() => true);
@@ -192,7 +192,7 @@ describe('ChatService', () => {
 
       it('should use default budget when reasoningBudgetToken is not set', async () => {
         const getChatCompletionSpy = vi.spyOn(chatService, 'getChatCompletion');
-        const messages = [{ content: 'Test default budget', role: 'user' }] as ChatMessage[];
+        const messages = [{ content: 'Test default budget', role: 'user' }] as UIChatMessage[];
 
         // Mock aiModelSelectors for extend params support
         vi.spyOn(aiModelSelectors, 'isModelHasExtendParams').mockReturnValue(() => true);
@@ -225,7 +225,7 @@ describe('ChatService', () => {
 
       it('should set reasoning_effort when model supports reasoningEffort and user configures it', async () => {
         const getChatCompletionSpy = vi.spyOn(chatService, 'getChatCompletion');
-        const messages = [{ content: 'Test reasoning effort', role: 'user' }] as ChatMessage[];
+        const messages = [{ content: 'Test reasoning effort', role: 'user' }] as UIChatMessage[];
 
         // Mock aiModelSelectors for extend params support
         vi.spyOn(aiModelSelectors, 'isModelHasExtendParams').mockReturnValue(() => true);
@@ -254,7 +254,7 @@ describe('ChatService', () => {
 
       it('should set thinkingBudget when model supports thinkingBudget and user configures it', async () => {
         const getChatCompletionSpy = vi.spyOn(chatService, 'getChatCompletion');
-        const messages = [{ content: 'Test thinking budget', role: 'user' }] as ChatMessage[];
+        const messages = [{ content: 'Test thinking budget', role: 'user' }] as UIChatMessage[];
 
         // Mock aiModelSelectors for extend params support
         vi.spyOn(aiModelSelectors, 'isModelHasExtendParams').mockReturnValue(() => true);
@@ -302,7 +302,7 @@ describe('ChatService', () => {
               },
             ],
           }, // Message with files
-        ] as ChatMessage[];
+        ] as UIChatMessage[];
 
         const getChatCompletionSpy = vi.spyOn(chatService, 'getChatCompletion');
         await chatService.createAssistantMessage({
@@ -342,7 +342,7 @@ describe('ChatService', () => {
         const messages = [
           { content: 'Hello', role: 'user', files: ['file2'] }, // Message with files
           { content: 'Hey', role: 'assistant' }, // Regular user message
-        ] as ChatMessage[];
+        ] as UIChatMessage[];
 
         const getChatCompletionSpy = vi.spyOn(chatService, 'getChatCompletion');
         await chatService.createAssistantMessage({ messages, plugins: [] });
@@ -391,7 +391,7 @@ describe('ChatService', () => {
             meta: {},
             updatedAt: Date.now(),
           },
-        ] as ChatMessage[];
+        ] as UIChatMessage[];
 
         // Spy on processImageList method
         // const processImageListSpy = vi.spyOn(chatService as any, 'processImageList');
@@ -463,7 +463,7 @@ describe('ChatService', () => {
             meta: {},
             updatedAt: Date.now(),
           },
-        ] as ChatMessage[];
+        ] as UIChatMessage[];
 
         // Spy on processImageList method
         const getChatCompletionSpy = vi.spyOn(chatService, 'getChatCompletion');
@@ -550,7 +550,7 @@ describe('ChatService', () => {
             meta: {},
             updatedAt: Date.now(),
           },
-        ] as ChatMessage[];
+        ] as UIChatMessage[];
 
         const getChatCompletionSpy = vi.spyOn(chatService, 'getChatCompletion');
 
@@ -605,7 +605,7 @@ describe('ChatService', () => {
               avatar: DEFAULT_USER_AVATAR,
             },
           },
-        ] as ChatMessage[];
+        ] as UIChatMessage[];
 
         act(() => {
           useToolStore.setState({
@@ -705,7 +705,7 @@ describe('ChatService', () => {
             role: 'user',
             content: 'https://vercel.com/ 请分析 chatGPT 关键词\n\n',
           },
-        ] as ChatMessage[];
+        ] as UIChatMessage[];
 
         act(() => {
           useToolStore.setState({
@@ -807,7 +807,7 @@ describe('ChatService', () => {
             role: 'user',
             content: 'https://vercel.com/ 请分析 chatGPT 关键词\n\n',
           },
-        ] as ChatMessage[];
+        ] as UIChatMessage[];
 
         await chatService.createAssistantMessage({
           messages,
@@ -847,7 +847,7 @@ describe('ChatService', () => {
               avatar: DEFAULT_USER_AVATAR,
             },
           },
-        ] as ChatMessage[];
+        ] as UIChatMessage[];
 
         await chatService.createAssistantMessage({
           messages,
@@ -870,7 +870,7 @@ describe('ChatService', () => {
       it('should add WebBrowsingManifest when search is enabled and not using model built-in search', async () => {
         const getChatCompletionSpy = vi.spyOn(chatService, 'getChatCompletion');
 
-        const messages = [{ content: 'Search for something', role: 'user' }] as ChatMessage[];
+        const messages = [{ content: 'Search for something', role: 'user' }] as UIChatMessage[];
 
         // Mock agent store state with search enabled
         vi.spyOn(agentChatConfigSelectors, 'currentChatConfig').mockReturnValueOnce({
@@ -921,7 +921,7 @@ describe('ChatService', () => {
       it('should enable built-in search when model supports it and useModelBuiltinSearch is true', async () => {
         const getChatCompletionSpy = vi.spyOn(chatService, 'getChatCompletion');
 
-        const messages = [{ content: 'Search for something', role: 'user' }] as ChatMessage[];
+        const messages = [{ content: 'Search for something', role: 'user' }] as UIChatMessage[];
 
         // Mock agent store state with search enabled and useModelBuiltinSearch enabled
         vi.spyOn(agentChatConfigSelectors, 'currentChatConfig').mockReturnValueOnce({
@@ -966,7 +966,7 @@ describe('ChatService', () => {
       it('should not enable search when searchMode is off', async () => {
         const getChatCompletionSpy = vi.spyOn(chatService, 'getChatCompletion');
 
-        const messages = [{ content: 'Search for something', role: 'user' }] as ChatMessage[];
+        const messages = [{ content: 'Search for something', role: 'user' }] as UIChatMessage[];
 
         // Mock agent store state with search disabled
         vi.spyOn(agentChatConfigSelectors, 'currentChatConfig').mockReturnValueOnce({
@@ -1278,7 +1278,7 @@ describe('ChatService private methods', () => {
   describe('extendParams', () => {
     it('should set enabledContextCaching to false when model supports disableContextCaching and user enables it', async () => {
       const getChatCompletionSpy = vi.spyOn(chatService, 'getChatCompletion');
-      const messages = [{ content: 'Test context caching', role: 'user' }] as ChatMessage[];
+      const messages = [{ content: 'Test context caching', role: 'user' }] as UIChatMessage[];
 
       // Mock aiModelSelectors for extend params support
       vi.spyOn(aiModelSelectors, 'isModelHasExtendParams').mockReturnValue(() => true);
@@ -1309,7 +1309,9 @@ describe('ChatService private methods', () => {
 
     it('should not set enabledContextCaching when disableContextCaching is false', async () => {
       const getChatCompletionSpy = vi.spyOn(chatService, 'getChatCompletion');
-      const messages = [{ content: 'Test context caching enabled', role: 'user' }] as ChatMessage[];
+      const messages = [
+        { content: 'Test context caching enabled', role: 'user' },
+      ] as UIChatMessage[];
 
       // Mock aiModelSelectors for extend params support
       vi.spyOn(aiModelSelectors, 'isModelHasExtendParams').mockReturnValue(() => true);
@@ -1337,7 +1339,7 @@ describe('ChatService private methods', () => {
 
     it('should set reasoning_effort when model supports reasoningEffort and user configures it', async () => {
       const getChatCompletionSpy = vi.spyOn(chatService, 'getChatCompletion');
-      const messages = [{ content: 'Test reasoning effort', role: 'user' }] as ChatMessage[];
+      const messages = [{ content: 'Test reasoning effort', role: 'user' }] as UIChatMessage[];
 
       // Mock aiModelSelectors for extend params support
       vi.spyOn(aiModelSelectors, 'isModelHasExtendParams').mockReturnValue(() => true);
@@ -1366,7 +1368,7 @@ describe('ChatService private methods', () => {
 
     it('should set thinkingBudget when model supports thinkingBudget and user configures it', async () => {
       const getChatCompletionSpy = vi.spyOn(chatService, 'getChatCompletion');
-      const messages = [{ content: 'Test thinking budget', role: 'user' }] as ChatMessage[];
+      const messages = [{ content: 'Test thinking budget', role: 'user' }] as UIChatMessage[];
 
       // Mock aiModelSelectors for extend params support
       vi.spyOn(aiModelSelectors, 'isModelHasExtendParams').mockReturnValue(() => true);

package/src/services/chat/contextEngineering.test.ts

@@ -1,4 +1,4 @@
-import { ChatMessage } from '@lobechat/types';
+import { UIChatMessage } from '@lobechat/types';
 import { afterEach, describe, expect, it, vi } from 'vitest';
 
 import * as isCanUseFCModule from '@/helpers/isCanUseFC';
@@ -72,7 +72,7 @@ describe('contextEngineering', () => {
           ],
         }, // Message with files
         { content: 'Hey', role: 'assistant' }, // Regular user message
-      ] as ChatMessage[];
+      ] as UIChatMessage[];
 
       const output = await contextEngineering({
         messages,
@@ -140,7 +140,7 @@ describe('contextEngineering', () => {
           ],
         }, // Message with files
         { content: 'Hey', role: 'assistant' }, // Regular user message
-      ] as ChatMessage[];
+      ] as UIChatMessage[];
       const output = await contextEngineering({
         messages,
         provider: 'openai',
@@ -192,7 +192,7 @@ describe('contextEngineering', () => {
         role: 'assistant',
         tool_calls: [],
       },
-    ] as ChatMessage[];
+    ] as UIChatMessage[];
 
     const result = await contextEngineering({
       messages,
@@ -222,7 +222,7 @@ describe('contextEngineering', () => {
           signature: 'thinking_process',
         },
       },
-    ] as ChatMessage[];
+    ] as UIChatMessage[];
 
     const result = await contextEngineering({
       messages,
@@ -250,7 +250,7 @@ describe('contextEngineering', () => {
 
   it('should inject INBOX_GUIDE_SYSTEM_ROLE for welcome questions in inbox session', async () => {
     // Don't mock INBOX_GUIDE_SYSTEMROLE, use the real one
-    const messages: ChatMessage[] = [
+    const messages: UIChatMessage[] = [
       {
         role: 'user',
         content: 'Hello, this is my first question',
@@ -281,7 +281,7 @@ describe('contextEngineering', () => {
   it('should inject historySummary into system message when provided', async () => {
     const historySummary = 'Previous conversation summary: User discussed AI topics.';
 
-    const messages: ChatMessage[] = [
+    const messages: UIChatMessage[] = [
       {
         role: 'user',
         content: 'Continue our discussion',
@@ -310,7 +310,7 @@ describe('contextEngineering', () => {
       // Mock isCanUseVision to return true for vision models
       vi.spyOn(helpers, 'isCanUseVision').mockReturnValue(true);
 
-      const messages: ChatMessage[] = [
+      const messages: UIChatMessage[] = [
         {
           role: 'assistant',
           content: 'Here is an image.',
@@ -337,7 +337,7 @@ describe('contextEngineering', () => {
       // Mock isCanUseVision to return true for vision models
       vi.spyOn(helpers, 'isCanUseVision').mockReturnValue(true);
 
-      const messages: ChatMessage[] = [
+      const messages: UIChatMessage[] = [
         {
           role: 'assistant',
           content: '',
@@ -364,7 +364,7 @@ describe('contextEngineering', () => {
     // Mock isCanUseFC to return false
     vi.spyOn(isCanUseFCModule, 'isCanUseFC').mockReturnValue(false);
 
-    const messages: ChatMessage[] = [
+    const messages: UIChatMessage[] = [
       {
         role: 'assistant',
         content: 'I have a tool call.',
@@ -396,7 +396,7 @@ describe('contextEngineering', () => {
 
   describe('Process placeholder variables', () => {
     it('should process placeholder variables in string content', async () => {
-      const messages: ChatMessage[] = [
+      const messages: UIChatMessage[] = [
         {
           role: 'user',
           content: 'Hello {{username}}, today is {{date}} and the time is {{time}}',
@@ -461,7 +461,7 @@ describe('contextEngineering', () => {
     });
 
     it('should handle missing placeholder variables gracefully', async () => {
-      const messages: ChatMessage[] = [
+      const messages: UIChatMessage[] = [
         {
           role: 'user',
           content: 'Hello {{username}}, missing: {{missing_var}}',
@@ -482,7 +482,7 @@ describe('contextEngineering', () => {
     });
 
     it('should not modify messages without placeholder variables', async () => {
-      const messages: ChatMessage[] = [
+      const messages: UIChatMessage[] = [
         {
           role: 'user',
           content: 'Hello there, no variables here',
@@ -506,7 +506,7 @@ describe('contextEngineering', () => {
       isServerMode = true;
       vi.spyOn(helpers, 'isCanUseVision').mockReturnValue(true);
 
-      const messages: ChatMessage[] = [
+      const messages: UIChatMessage[] = [
         {
           role: 'user',
           content: 'Hello {{username}}, check this image from {{date}}',
@@ -549,7 +549,7 @@ describe('contextEngineering', () => {
 
   describe('Message preprocessing processors', () => {
     it('should truncate message history when enabled', async () => {
-      const messages: ChatMessage[] = [
+      const messages: UIChatMessage[] = [
         {
           role: 'user',
           content: 'Message 1',
@@ -611,7 +611,7 @@ describe('contextEngineering', () => {
     });
 
     it('should apply input template to user messages', async () => {
-      const messages: ChatMessage[] = [
+      const messages: UIChatMessage[] = [
         {
           role: 'user',
           content: 'Original user input',
@@ -652,7 +652,7 @@ describe('contextEngineering', () => {
     });
 
     it('should inject system role at the beginning', async () => {
-      const messages: ChatMessage[] = [
+      const messages: UIChatMessage[] = [
         {
           role: 'user',
           content: 'User message',
@@ -678,7 +678,7 @@ describe('contextEngineering', () => {
     });
 
     it('should combine all preprocessing steps correctly', async () => {
-      const messages: ChatMessage[] = [
+      const messages: UIChatMessage[] = [
         {
           role: 'user',
           content: 'Old message 1',
@@ -733,7 +733,7 @@ describe('contextEngineering', () => {
     });
 
     it('should skip preprocessing when no configuration is provided', async () => {
-      const messages: ChatMessage[] = [
+      const messages: UIChatMessage[] = [
         {
           role: 'user',
           content: 'Simple message',
@@ -760,7 +760,7 @@ describe('contextEngineering', () => {
     });
 
     it('should handle history truncation with system role injection correctly', async () => {
-      const messages: ChatMessage[] = [
+      const messages: UIChatMessage[] = [
         {
           role: 'user',
           content: 'Message 1',
@@ -810,7 +810,7 @@ describe('contextEngineering', () => {
     });
 
     it('should handle input template compilation errors gracefully', async () => {
-      const messages: ChatMessage[] = [
+      const messages: UIChatMessage[] = [
         {
           role: 'user',
           content: 'User message',

package/src/services/chat/contextEngineering.ts

@@ -15,7 +15,7 @@ import {
   ToolSystemRoleProvider,
 } from '@lobechat/context-engine';
 import { historySummaryPrompt } from '@lobechat/prompts';
-import { ChatMessage, OpenAIChatMessage } from '@lobechat/types';
+import { OpenAIChatMessage, UIChatMessage } from '@lobechat/types';
 import { VARIABLE_GENERATORS } from '@lobechat/utils/client';
 
 import { isCanUseFC } from '@/helpers/isCanUseFC';
@@ -30,7 +30,7 @@ interface ContextEngineeringContext {
   historySummary?: string;
   inputTemplate?: string;
   isWelcomeQuestion?: boolean;
-  messages: ChatMessage[];
+  messages: UIChatMessage[];
   model: string;
   provider: string;
   sessionId?: string;

package/src/services/chat/index.ts

@@ -1,5 +1,5 @@
 import { AgentRuntimeError, ChatCompletionErrorPayload } from '@lobechat/model-runtime';
-import { ChatErrorType, TracePayload, TraceTagMap , ChatMessage } from '@lobechat/types';
+import { ChatErrorType, TracePayload, TraceTagMap, UIChatMessage } from '@lobechat/types';
 import { PluginRequestPayload, createHeadersWithPluginSettings } from '@lobehub/chat-plugin-sdk';
 import { merge } from 'lodash-es';
 import { ModelProvider } from 'model-bank';
@@ -41,11 +41,11 @@ import { findDeploymentName, isEnableFetchOnClient, resolveRuntimeProvider } fro
 import { FetchOptions } from './types';
 
 interface GetChatCompletionPayload extends Partial<Omit<ChatStreamPayload, 'messages'>> {
-  messages: ChatMessage[];
+  messages: UIChatMessage[];
 }
 
 type ChatStreamInputParams = Partial<Omit<ChatStreamPayload, 'messages'>> & {
-  messages?: (ChatMessage | OpenAIChatMessage)[];
+  messages?: (UIChatMessage | OpenAIChatMessage)[];
 };
 
 interface FetchAITaskResultParams extends FetchSSEOptions {

package/src/services/message/_deprecated.test.ts

@@ -1,9 +1,9 @@
 import {
-  ChatMessage,
   ChatMessageError,
   ChatPluginPayload,
   ChatTTS,
   ChatTranslate,
+  UIChatMessage,
 } from '@lobechat/types';
 import dayjs from 'dayjs';
 import { Mock, describe, expect, it, vi } from 'vitest';
@@ -47,7 +47,7 @@ describe('MessageClientService', () => {
     updatedAt: 100,
     role: 'user',
     // ... other properties
-  } as ChatMessage;
+  } as UIChatMessage;
   const mockMessages = [mockMessage];
 
   beforeEach(() => {

package/src/services/message/_deprecated.ts

@@ -1,11 +1,11 @@
 import {
   ChatFileItem,
-  ChatMessage,
   ChatMessageError,
   ChatTTS,
   ChatTranslate,
   CreateMessageParams,
   ModelRankItem,
+  UIChatMessage,
 } from '@lobechat/types';
 import dayjs from 'dayjs';
 
@@ -23,11 +23,11 @@ export class ClientService implements IMessageService {
   }
 
   // @ts-ignore
-  async batchCreateMessages(messages: ChatMessage[]) {
+  async batchCreateMessages(messages: UIChatMessage[]) {
     return MessageModel.batchCreate(messages);
   }
 
-  async getMessages(sessionId: string, topicId?: string): Promise<ChatMessage[]> {
+  async getMessages(sessionId: string, topicId?: string): Promise<UIChatMessage[]> {
     const messages = await MessageModel.query({ sessionId, topicId });
 
     const fileList = (await Promise.all(
@@ -49,7 +49,7 @@ export class ClientService implements IMessageService {
     }));
   }
 
-  async getGroupMessages(groupId: string, topicId?: string): Promise<ChatMessage[]> {
+  async getGroupMessages(groupId: string, topicId?: string): Promise<UIChatMessage[]> {
     // For the deprecated service, group messages are the same as regular messages
     // since the old schema doesn't differentiate between session and group messages
     return this.getMessages(groupId, topicId);

package/src/services/message/client.test.ts

@@ -1,10 +1,10 @@
 import {
-  ChatMessage,
   ChatMessageError,
   ChatTTS,
   ChatTranslate,
   CreateMessageParams,
-  MessageItem,
+  DBMessageItem,
+  UIChatMessage,
 } from '@lobechat/types';
 import { and, eq } from 'drizzle-orm';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
@@ -34,7 +34,7 @@ const mockMessage = {
   content: 'Mock message content',
   sessionId,
   role: 'user',
-} as ChatMessage;
+} as UIChatMessage;
 
 const mockMessages = [mockMessage];
 
@@ -98,7 +98,7 @@ describe('MessageClientService', () => {
           sessionId,
           role: 'user',
         },
-      ] as MessageItem[]);
+      ] as DBMessageItem[]);
       const count = await clientDB.$count(messages);
 
       // Assert

package/src/services/message/client.ts

@@ -1,4 +1,4 @@
-import { ChatMessage } from '@lobechat/types';
+import { UIChatMessage } from '@lobechat/types';
 
 import { INBOX_SESSION_ID } from '@/const/session';
 import { clientDB } from '@/database/client/db';
@@ -42,7 +42,7 @@ export class ClientService extends BaseClientService implements IMessageService
       },
     );
 
-    return data as unknown as ChatMessage[];
+    return data as unknown as UIChatMessage[];
   };
 
   getGroupMessages: IMessageService['getGroupMessages'] = async (groupId, topicId) => {
@@ -62,13 +62,13 @@ export class ClientService extends BaseClientService implements IMessageService
       },
     );
 
-    return data as unknown as ChatMessage[];
+    return data as unknown as UIChatMessage[];
   };
 
   getAllMessages: IMessageService['getAllMessages'] = async () => {
     const data = await this.messageModel.queryAll();
 
-    return data as unknown as ChatMessage[];
+    return data as unknown as UIChatMessage[];
   };
 
   countMessages: IMessageService['countMessages'] = async (params) => {
@@ -90,7 +90,7 @@ export class ClientService extends BaseClientService implements IMessageService
   getAllMessagesInSession: IMessageService['getAllMessagesInSession'] = async (sessionId) => {
     const data = this.messageModel.queryBySessionId(this.toDbSessionId(sessionId));
 
-    return data as unknown as ChatMessage[];
+    return data as unknown as UIChatMessage[];
   };
 
   updateMessageError: IMessageService['updateMessageError'] = async (id, error) => {

package/src/services/message/server.ts

@@ -1,5 +1,5 @@
 /* eslint-disable @typescript-eslint/no-unused-vars */
-import { ChatMessage, ChatTranslate } from '@lobechat/types';
+import { ChatTranslate, UIChatMessage } from '@lobechat/types';
 
 import { INBOX_SESSION_ID } from '@/const/session';
 import { lambdaClient } from '@/libs/trpc/client';
@@ -25,7 +25,7 @@ export class ServerService implements IMessageService {
       topicId,
     });
 
-    return data as unknown as ChatMessage[];
+    return data as unknown as UIChatMessage[];
   };
 
   getGroupMessages: IMessageService['getGroupMessages'] = async (groupId, topicId) => {
@@ -33,7 +33,7 @@ export class ServerService implements IMessageService {
       groupId,
       topicId,
     });
-    return data as unknown as ChatMessage[];
+    return data as unknown as UIChatMessage[];
   };
 
   getAllMessages: IMessageService['getAllMessages'] = async () => {

package/src/services/message/type.ts

@@ -1,12 +1,12 @@
 import {
-  ChatMessage,
   ChatMessageError,
   ChatMessagePluginError,
   ChatTTS,
   ChatTranslate,
   CreateMessageParams,
-  MessageItem,
+  DBMessageItem,
   ModelRankItem,
+  UIChatMessage,
   UpdateMessageParams,
   UpdateMessageRAGParams,
 } from '@lobechat/types';
@@ -16,12 +16,12 @@ import type { HeatmapsProps } from '@lobehub/charts';
 
 export interface IMessageService {
   createMessage(data: CreateMessageParams): Promise<string>;
-  batchCreateMessages(messages: MessageItem[]): Promise<any>;
+  batchCreateMessages(messages: DBMessageItem[]): Promise<any>;
 
-  getMessages(sessionId: string, topicId?: string, groupId?: string): Promise<ChatMessage[]>;
-  getGroupMessages(groupId: string, topicId?: string): Promise<ChatMessage[]>;
-  getAllMessages(): Promise<ChatMessage[]>;
-  getAllMessagesInSession(sessionId: string): Promise<ChatMessage[]>;
+  getMessages(sessionId: string, topicId?: string, groupId?: string): Promise<UIChatMessage[]>;
+  getGroupMessages(groupId: string, topicId?: string): Promise<UIChatMessage[]>;
+  getAllMessages(): Promise<UIChatMessage[]>;
+  getAllMessagesInSession(sessionId: string): Promise<UIChatMessage[]>;
   countMessages(params?: {
     endDate?: string;
     range?: [string, string];

package/src/store/chat/helpers.test.ts

@@ -1,8 +1,8 @@
-import { ChatMessage } from '@lobechat/types';
+import { UIChatMessage } from '@lobechat/types';
+import { LobeAgentChatConfig } from '@lobechat/types';
+import { OpenAIChatMessage } from '@lobechat/types';
 import { describe, expect, it, vi } from 'vitest';
 
-import { LobeAgentChatConfig, LobeAgentConfig } from '@/types/agent';
-import { OpenAIChatMessage } from '@/types/openai/chat';
 import { encodeAsync } from '@/utils/tokenizer';
 import * as tokenizerObj from '@/utils/tokenizer';
 
@@ -48,7 +48,7 @@ describe('chatHelpers', () => {
     const messages = [
       { id: '1', content: 'Hello' },
       { id: '2', content: 'World' },
-    ] as ChatMessage[];
+    ] as UIChatMessage[];
 
     it('finds a message by id', () => {
       const message = chatHelpers.getMessageById(messages, '1');
@@ -71,7 +71,7 @@ describe('chatHelpers', () => {
       { id: '1', content: 'First' },
       { id: '2', content: 'Second' },
       { id: '3', content: 'Third' },
-    ] as ChatMessage[];
+    ] as UIChatMessage[];
 
     it('returns all messages if history is disabled', () => {
       const config = { enableHistoryCount: false, historyCount: undefined } as LobeAgentChatConfig;