@lobehub/chat 1.142.1 → 1.142.2

package/src/store/chat/helpers.ts

@@ -1,22 +1,21 @@
-import { ChatMessage } from '@lobechat/types';
+import { OpenAIChatMessage, UIChatMessage } from '@lobechat/types';
 
-import { OpenAIChatMessage } from '@/types/openai/chat';
 import { encodeAsync } from '@/utils/tokenizer';
 
 export const getMessagesTokenCount = async (messages: OpenAIChatMessage[]) =>
   encodeAsync(messages.map((m) => m.content).join(''));
 
-export const getMessageById = (messages: ChatMessage[], id: string) =>
+export const getMessageById = (messages: UIChatMessage[], id: string) =>
   messages.find((m) => m.id === id);
 
 const getSlicedMessages = (
-  messages: ChatMessage[],
+  messages: UIChatMessage[],
   options: {
     enableHistoryCount?: boolean;
     historyCount?: number;
     includeNewUserMessage?: boolean;
   },
-): ChatMessage[] => {
+): UIChatMessage[] => {
   // if historyCount is not enabled, return all messages
   if (!options.enableHistoryCount || options.historyCount === undefined) return messages;
 

package/src/store/chat/initialState.ts

@@ -2,7 +2,6 @@
 import { ChatToolState, initialToolState } from './slices/builtinTool/initialState';
 import { ChatPortalState, initialChatPortalState } from './slices/portal/initialState';
 import { ChatMessageState, initialMessageState } from './slices/message/initialState';
-import { ChatShareState, initialShareState } from './slices/share/initialState';
 import { ChatTopicState, initialTopicState } from './slices/topic/initialState';
 import { ChatAIChatState, initialAiChatState } from './slices/aiChat/initialState';
 import { ChatThreadState, initialThreadState } from './slices/thread/initialState';
@@ -11,7 +10,6 @@ export type ChatStoreState = ChatTopicState &
   ChatMessageState &
   ChatAIChatState &
   ChatToolState &
-  ChatShareState &
   ChatThreadState &
   ChatPortalState;
 
@@ -20,7 +18,6 @@ export const initialState: ChatStoreState = {
   ...initialAiChatState,
   ...initialTopicState,
   ...initialToolState,
-  ...initialShareState,
   ...initialThreadState,
   ...initialChatPortalState,
 

package/src/store/chat/slices/aiChat/actions/__tests__/fixtures.ts

@@ -1,4 +1,4 @@
-import { ChatMessage } from '@lobechat/types';
+import { UIChatMessage } from '@lobechat/types';
 
 import { DEFAULT_AGENT_CHAT_CONFIG, DEFAULT_AGENT_CONFIG } from '@/const/settings';
 
@@ -22,7 +22,7 @@ export const TEST_CONTENT = {
 } as const;
 
 // Mock Data Factories
-export const createMockMessage = (overrides: Partial<ChatMessage> = {}): ChatMessage => {
+export const createMockMessage = (overrides: Partial<UIChatMessage> = {}): UIChatMessage => {
   const base: any = {
     content: TEST_CONTENT.USER_MESSAGE,
     createdAt: Date.now(),
@@ -34,10 +34,10 @@ export const createMockMessage = (overrides: Partial<ChatMessage> = {}): ChatMes
   };
 
   // Merge overrides, preserving all provided properties
-  return { ...base, ...overrides } as ChatMessage;
+  return { ...base, ...overrides } as UIChatMessage;
 };
 
-export const createMockMessages = (count: number): ChatMessage[] =>
+export const createMockMessages = (count: number): UIChatMessage[] =>
   Array.from({ length: count }, (_, i) =>
     createMockMessage({
       content: `Message ${i}`,

package/src/store/chat/slices/aiChat/actions/__tests__/generateAIChatV2.test.ts

@@ -1,4 +1,4 @@
-import { ChatMessage } from '@lobechat/types';
+import { UIChatMessage } from '@lobechat/types';
 import { act, renderHook } from '@testing-library/react';
 import { TRPCClientError } from '@trpc/client';
 import { Mock, afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
@@ -376,7 +376,7 @@ describe('generateAIChatV2 actions', () => {
         content: TEST_CONTENT.USER_MESSAGE,
         sessionId: TEST_IDS.SESSION_ID,
         topicId: TEST_IDS.TOPIC_ID,
-      } as ChatMessage;
+      } as UIChatMessage;
       const messages = [userMessage];
 
       const streamSpy = vi.spyOn(chatService, 'createAssistantMessageStream');

package/src/store/chat/slices/aiChat/actions/__tests__/rag.test.ts

@@ -1,4 +1,4 @@
-import { ChatMessage } from '@lobechat/types';
+import { UIChatMessage } from '@lobechat/types';
 import { act, renderHook } from '@testing-library/react';
 import { Mock, beforeEach, describe, expect, it, vi } from 'vitest';
 
@@ -49,7 +49,7 @@ describe('chatRAG actions', () => {
       act(() => {
         useChatStore.setState({
           messagesMap: {
-            default: [{ id: messageId }] as ChatMessage[],
+            default: [{ id: messageId }] as UIChatMessage[],
           },
         });
       });
@@ -75,7 +75,7 @@ describe('chatRAG actions', () => {
           ({
             id: messageId,
             ragQuery: existingRagQuery,
-          }) as ChatMessage,
+          }) as UIChatMessage,
       );
 
       // Mock the semantic search response
@@ -117,7 +117,7 @@ describe('chatRAG actions', () => {
         () =>
           ({
             id: messageId,
-          }) as ChatMessage,
+          }) as UIChatMessage,
       );
 
       // Mock the rewrite query function
@@ -240,12 +240,12 @@ describe('chatRAG actions', () => {
           ({
             id: messageId,
             content,
-          }) as ChatMessage,
+          }) as UIChatMessage,
       );
 
       vi.spyOn(chatSelectors, 'mainAIChatsWithHistoryConfig').mockReturnValue([
         { content: 'history' },
-      ] as ChatMessage[]);
+      ] as UIChatMessage[]);
 
       const rewriteSpy = vi.spyOn(result.current, 'internal_rewriteQuery');
       const deleteSpy = vi.spyOn(result.current, 'deleteUserMessageRagQuery');

package/src/store/chat/slices/aiChat/actions/generateAIChat.ts

@@ -4,12 +4,12 @@ import { LOADING_FLAT, MESSAGE_CANCEL_FLAT, isDesktop, isServerMode } from '@lob
 import { knowledgeBaseQAPrompts } from '@lobechat/prompts';
 import {
   ChatImageItem,
-  ChatMessage,
   CreateMessageParams,
   MessageSemanticSearchChunk,
   SendMessageParams,
   TraceEventType,
   TraceNameMap,
+  UIChatMessage,
 } from '@lobechat/types';
 import { t } from 'i18next';
 import { produce } from 'immer';
@@ -76,7 +76,7 @@ export interface AIGenerateAction {
    * including preprocessing and postprocessing steps
    */
   internal_coreProcessMessage: (
-    messages: ChatMessage[],
+    messages: UIChatMessage[],
     parentId: string,
     params?: ProcessMessageParams,
   ) => Promise<void>;
@@ -84,7 +84,7 @@ export interface AIGenerateAction {
    * Retrieves an AI-generated chat message from the backend service
    */
   internal_fetchAIChatMessage: (input: {
-    messages: ChatMessage[];
+    messages: UIChatMessage[];
     messageId: string;
     params?: ProcessMessageParams;
     model: string;
@@ -101,7 +101,7 @@ export interface AIGenerateAction {
     id: string,
     params?: {
       traceId?: string;
-      messages?: ChatMessage[];
+      messages?: UIChatMessage[];
       threadId?: string;
       inPortalThread?: boolean;
     },
@@ -341,7 +341,7 @@ export const generateAIChat: StateCreator<
 
       ragQueryId = queryId;
 
-      const lastMsg = messages.pop() as ChatMessage;
+      const lastMsg = messages.pop() as UIChatMessage;
 
       // 2. build the retrieve context messages
       const knowledgeBaseQAContext = knowledgeBaseQAPrompts({
@@ -780,7 +780,7 @@ export const generateAIChat: StateCreator<
 
     const currentMessage = chats[currentIndex];
 
-    let contextMessages: ChatMessage[] = [];
+    let contextMessages: UIChatMessage[] = [];
 
     switch (currentMessage.role) {
       case 'tool':

package/src/store/chat/slices/aiChat/actions/generateAIChatV2.ts

@@ -4,13 +4,13 @@ import { DEFAULT_AGENT_CHAT_CONFIG, INBOX_SESSION_ID, isDesktop } from '@lobecha
 import { knowledgeBaseQAPrompts } from '@lobechat/prompts';
 import {
   ChatImageItem,
-  ChatMessage,
   ChatTopic,
   ChatVideoItem,
   MessageSemanticSearchChunk,
   SendMessageParams,
   SendMessageServerResponse,
   TraceNameMap,
+  UIChatMessage,
 } from '@lobechat/types';
 import { TRPCClientError } from '@trpc/client';
 import { t } from 'i18next';
@@ -47,7 +47,7 @@ export interface AIGenerateV2Action {
   clearSendMessageError: () => void;
   internal_refreshAiChat: (params: {
     topics?: ChatTopic[];
-    messages: ChatMessage[];
+    messages: UIChatMessage[];
     sessionId: string;
     topicId?: string;
   }) => void;
@@ -56,7 +56,7 @@ export interface AIGenerateV2Action {
    * including preprocessing and postprocessing steps
    */
   internal_execAgentRuntime: (params: {
-    messages: ChatMessage[];
+    messages: UIChatMessage[];
     userMessageId: string;
     assistantMessageId: string;
     isWelcomeQuestion?: boolean;
@@ -353,7 +353,7 @@ export const generateAIChatV2: StateCreator<
 
       ragQueryId = queryId;
 
-      const lastMsg = messages.pop() as ChatMessage;
+      const lastMsg = messages.pop() as UIChatMessage;
 
       // 2. build the retrieve context messages
       const knowledgeBaseQAContext = knowledgeBaseQAPrompts({

package/src/store/chat/slices/aiChat/actions/generateAIGroupChat.ts

@@ -5,7 +5,12 @@ import {
   buildGroupChatSystemPrompt,
   filterMessagesForAgent,
 } from '@lobechat/prompts';
-import { ChatMessage, CreateMessageParams, SendGroupMessageParams } from '@lobechat/types';
+import {
+  ChatErrorType,
+  CreateMessageParams,
+  SendGroupMessageParams,
+  UIChatMessage,
+} from '@lobechat/types';
 import { produce } from 'immer';
 import { StateCreator } from 'zustand/vanilla';
 
@@ -17,7 +22,6 @@ import { useSessionStore } from '@/store/session';
 import { sessionSelectors } from '@/store/session/selectors';
 import { userProfileSelectors } from '@/store/user/selectors';
 import { getUserStoreState } from '@/store/user/store';
-import { ChatErrorType } from '@/types/fetch';
 import { merge } from '@/utils/merge';
 import { setNamespace } from '@/utils/storeDebug';
 
@@ -99,7 +103,7 @@ const extractMentionsFromContent = (
 /**
  * Check if a message is a tool calling message that requires a follow-up
  */
-const isToolCallMessage = (message: ChatMessage): boolean => {
+const isToolCallMessage = (message: UIChatMessage): boolean => {
   return message.role === 'assistant' && !!message.tools && message.tools.length > 0;
 };
 
@@ -107,7 +111,7 @@ const isToolCallMessage = (message: ChatMessage): boolean => {
  * Count consecutive assistant messages from the end of the message list
  * This helps enforce maxResponseInRow limit
  */
-const countConsecutiveAssistantMessages = (messages: ChatMessage[]): number => {
+const countConsecutiveAssistantMessages = (messages: UIChatMessage[]): number => {
   let count = 0;
 
   // Count from the end of the array backwards
@@ -135,7 +139,7 @@ const countConsecutiveAssistantMessages = (messages: ChatMessage[]): number => {
  * Returns true if the conversation flow should continue without supervisor intervention
  */
 const shouldAvoidSupervisorDecision = (
-  messages: ChatMessage[],
+  messages: UIChatMessage[],
   maxResponseInRow?: number,
   isManualTrigger: boolean = false,
 ): boolean => {
@@ -666,7 +670,7 @@ export const chatAiGroupChat: StateCreator<
 
         const assistantId = await internal_createMessage(agentMessage);
 
-        const systemMessage: ChatMessage = {
+        const systemMessage: UIChatMessage = {
           id: 'group-system',
           role: 'system',
           content: groupChatSystemPrompt,

package/src/store/chat/slices/aiChat/actions/memory.ts

@@ -1,5 +1,5 @@
 import { chainSummaryHistory } from '@lobechat/prompts';
-import { TraceNameMap , ChatMessage } from '@lobechat/types';
+import { TraceNameMap, UIChatMessage } from '@lobechat/types';
 import { StateCreator } from 'zustand/vanilla';
 
 import { chatService } from '@/services/chat';
@@ -9,7 +9,7 @@ import { useUserStore } from '@/store/user';
 import { systemAgentSelectors } from '@/store/user/selectors';
 
 export interface ChatMemoryAction {
-  internal_summaryHistory: (messages: ChatMessage[]) => Promise<void>;
+  internal_summaryHistory: (messages: UIChatMessage[]) => Promise<void>;
 }
 
 export const chatMemory: StateCreator<

package/src/store/chat/slices/builtinTool/actions/__tests__/dalle.test.ts

@@ -1,4 +1,4 @@
-import { ChatMessage } from '@lobechat/types';
+import { UIChatMessage } from '@lobechat/types';
 import { act, renderHook } from '@testing-library/react';
 import { describe, expect, it, vi } from 'vitest';
 
@@ -24,7 +24,7 @@ describe('chatToolSlice - dalle', () => {
           ({
             id,
             content: initialMessageContent,
-          }) as ChatMessage,
+          }) as UIChatMessage,
       );
 
       const messageId = 'message-id';
@@ -84,7 +84,7 @@ describe('chatToolSlice - dalle', () => {
           ({
             id,
             content: initialMessageContent,
-          }) as ChatMessage,
+          }) as UIChatMessage,
       );
       vi.spyOn(messageService, 'updateMessage').mockResolvedValueOnce(undefined);
 

package/src/store/chat/slices/builtinTool/actions/__tests__/search.test.ts

@@ -1,6 +1,6 @@
 import { crawlResultsPrompt, searchResultsPrompt } from '@lobechat/prompts';
 import { SearchContent, SearchQuery, UniformSearchResponse } from '@lobechat/types';
-import { ChatMessage } from '@lobechat/types';
+import { UIChatMessage } from '@lobechat/types';
 import { act, renderHook } from '@testing-library/react';
 import { Mock, beforeEach, describe, expect, it, vi } from 'vitest';
 
@@ -247,7 +247,7 @@ describe('search actions', () => {
     it('should save search result as tool message', async () => {
       const messageId = 'test-message-id';
       const parentId = 'parent-message-id';
-      const mockMessage: Partial<ChatMessage> = {
+      const mockMessage: Partial<UIChatMessage> = {
         id: messageId,
         parentId,
         content: 'test content',
@@ -265,7 +265,7 @@ describe('search actions', () => {
       };
 
       vi.spyOn(chatSelectors, 'getMessageById').mockImplementation(
-        () => () => mockMessage as ChatMessage,
+        () => () => mockMessage as UIChatMessage,
       );
 
       const { result } = renderHook(() => useChatStore());

package/src/store/chat/slices/message/action.test.ts

@@ -1,5 +1,5 @@
 import { TraceEventType } from '@lobechat/types';
-import { ChatMessage } from '@lobechat/types';
+import { UIChatMessage } from '@lobechat/types';
 import * as lobeUIModules from '@lobehub/ui';
 import { act, renderHook, waitFor } from '@testing-library/react';
 import { mutate } from 'swr';
@@ -206,7 +206,7 @@ describe('chatMessage actions', () => {
           activeId: 'session-id',
           activeTopicId: undefined,
           messagesMap: {
-            [messageMapKey('session-id')]: [{ id: messageId } as ChatMessage],
+            [messageMapKey('session-id')]: [{ id: messageId } as UIChatMessage],
           },
         });
       });
@@ -229,9 +229,9 @@ describe('chatMessage actions', () => {
           activeTopicId: undefined,
           messagesMap: {
             [messageMapKey('session-id')]: [
-              { id: messageId, tools: [{ id: 'tool1' }, { id: 'tool2' }] } as ChatMessage,
-              { id: '2', tool_call_id: 'tool1', role: 'tool' } as ChatMessage,
-              { id: '3', tool_call_id: 'tool2', role: 'tool' } as ChatMessage,
+              { id: messageId, tools: [{ id: 'tool1' }, { id: 'tool2' }] } as UIChatMessage,
+              { id: '2', tool_call_id: 'tool1', role: 'tool' } as UIChatMessage,
+              { id: '3', tool_call_id: 'tool2', role: 'tool' } as UIChatMessage,
             ],
           },
         });
@@ -292,9 +292,14 @@ describe('chatMessage actions', () => {
                 id: messageId,
                 role: 'assistant',
                 tools: [{ id: 'tool1' }, { id: 'tool2' }],
-              } as ChatMessage,
-              { id: '2', parentId: messageId, tool_call_id: 'tool1', role: 'tool' } as ChatMessage,
-              { id: '3', tool_call_id: 'tool2', role: 'tool' } as ChatMessage,
+              } as UIChatMessage,
+              {
+                id: '2',
+                parentId: messageId,
+                tool_call_id: 'tool1',
+                role: 'tool',
+              } as UIChatMessage,
+              { id: '3', tool_call_id: 'tool2', role: 'tool' } as UIChatMessage,
             ],
           },
         });

package/src/store/chat/slices/message/action.ts

@@ -3,7 +3,6 @@
 import {
   ChatErrorType,
   ChatImageItem,
-  ChatMessage,
   ChatMessageError,
   ChatMessagePluginError,
   CreateMessageParams,
@@ -13,6 +12,7 @@ import {
   ModelReasoning,
   TraceEventPayloads,
   TraceEventType,
+  UIChatMessage,
   UpdateMessageRAGParams,
 } from '@lobechat/types';
 import { nanoid } from '@lobechat/utils';
@@ -62,10 +62,10 @@ export interface ChatMessageAction {
     messageContextId: string,
     activeTopicId?: string,
     type?: 'session' | 'group',
-  ) => SWRResponse<ChatMessage[]>;
+  ) => SWRResponse<UIChatMessage[]>;
   copyMessage: (id: string, content: string) => Promise<void>;
   refreshMessages: () => Promise<void>;
-  replaceMessages: (messages: ChatMessage[]) => void;
+  replaceMessages: (messages: UIChatMessage[]) => void;
   // =========  ↓ Internal Method ↓  ========== //
   // ========================================== //
   // ========================================== //
@@ -306,7 +306,7 @@ export const chatMessage: StateCreator<
    * @param messageContextId - Can be sessionId or groupId
    */
   useFetchMessages: (enable, messageContextId, activeTopicId, type = 'session') =>
-    useClientDataSWR<ChatMessage[]>(
+    useClientDataSWR<UIChatMessage[]>(
       enable ? [SWR_USE_FETCH_MESSAGES, messageContextId, activeTopicId, type] : null,
       async ([, sessionId, topicId, type]: [string, string, string | undefined, string]) =>
         type === 'session'

package/src/store/chat/slices/message/initialState.ts

@@ -1,4 +1,4 @@
-import { ChatMessage } from '@lobechat/types';
+import { UIChatMessage } from '@lobechat/types';
 
 import { ChatGroupAgentItem, ChatGroupItem } from '@/database/schemas/chatGroup';
 
@@ -40,7 +40,7 @@ export interface ChatMessageState {
    * whether messages have fetched
    */
   messagesInit: boolean;
-  messagesMap: Record<string, ChatMessage[]>;
+  messagesMap: Record<string, UIChatMessage[]>;
   /**
    * Supervisor decision debounce timers by group ID
    */

package/src/store/chat/slices/message/reducer.test.ts

@@ -1,9 +1,9 @@
-import { ChatMessage, ChatToolPayload } from '@lobechat/types';
+import { ChatToolPayload, UIChatMessage } from '@lobechat/types';
 
 import { MessageDispatch, messagesReducer } from './reducer';
 
 describe('messagesReducer', () => {
-  let initialState: ChatMessage[];
+  let initialState: UIChatMessage[];
 
   beforeEach(() => {
     initialState = [
@@ -115,7 +115,7 @@ describe('messagesReducer', () => {
             role: 'user',
             meta: {},
             extra: { abc: '1' },
-          } as ChatMessage,
+          } as UIChatMessage,
           ...initialState,
         ],
         payload,
@@ -189,7 +189,7 @@ describe('messagesReducer', () => {
 
   describe('updateMessagePlugin', () => {
     it('should update the plugin of a tool message', () => {
-      const toolMessage: ChatMessage = {
+      const toolMessage: UIChatMessage = {
         id: 'toolMessage',
         role: 'tool',
         content: 'Tool content',
@@ -302,7 +302,7 @@ describe('messagesReducer', () => {
       };
 
       const newState = messagesReducer(
-        [...initialState, { id: messageId, role: 'assistant', content: '' } as ChatMessage],
+        [...initialState, { id: messageId, role: 'assistant', content: '' } as UIChatMessage],
         payload,
       );
       const updatedMessage = newState.find((m) => m.id === messageId);
@@ -330,7 +330,7 @@ describe('messagesReducer', () => {
       };
 
       const newState = messagesReducer(
-        [...initialState, { id: messageId, role: 'assistant', content: '' } as ChatMessage],
+        [...initialState, { id: messageId, role: 'assistant', content: '' } as UIChatMessage],
         payload,
       );
       const updatedMessage = newState.find((m) => m.id === messageId);

package/src/store/chat/slices/message/reducer.ts

@@ -1,9 +1,9 @@
 import {
-  ChatMessage,
   ChatMessageExtra,
   ChatPluginPayload,
   ChatToolPayload,
   CreateMessageParams,
+  UIChatMessage,
 } from '@lobechat/types';
 import isEqual from 'fast-deep-equal';
 import { produce } from 'immer';
@@ -12,13 +12,13 @@ import { merge } from '@/utils/merge';
 
 interface UpdateMessages {
   type: 'updateMessages';
-  value: ChatMessage[];
+  value: UIChatMessage[];
 }
 
 interface UpdateMessage {
   id: string;
   type: 'updateMessage';
-  value: Partial<ChatMessage>;
+  value: Partial<UIChatMessage>;
 }
 
 interface CreateMessage {
@@ -88,7 +88,10 @@ export type MessageDispatch =
   | DeleteMessageTool
   | DeleteMessages;
 
-export const messagesReducer = (state: ChatMessage[], payload: MessageDispatch): ChatMessage[] => {
+export const messagesReducer = (
+  state: UIChatMessage[],
+  payload: MessageDispatch,
+): UIChatMessage[] => {
   switch (payload.type) {
     case 'updateMessage': {
       return produce(state, (draftState) => {

package/src/store/chat/slices/message/selectors.test.ts

@@ -1,4 +1,5 @@
-import { ChatMessage } from '@lobechat/types';
+import { UIChatMessage } from '@lobechat/types';
+import { LobeAgentConfig } from '@lobechat/types';
 import { act } from '@testing-library/react';
 import { describe, expect, it } from 'vitest';
 
@@ -9,7 +10,6 @@ import { ChatStore } from '@/store/chat';
 import { initialState } from '@/store/chat/initialState';
 import { messageMapKey } from '@/store/chat/utils/messageMapKey';
 import { createServerConfigStore } from '@/store/serverConfig/store';
-import { LobeAgentConfig } from '@/types/agent';
 import { merge } from '@/utils/merge';
 
 import { chatSelectors } from './selectors';
@@ -45,7 +45,7 @@ const mockMessages = [
       },
     ],
   },
-] as ChatMessage[];
+] as UIChatMessage[];
 
 const mockReasoningMessages = [
   {
@@ -66,7 +66,7 @@ const mockReasoningMessages = [
       content: 'Reasoning Content',
     },
   },
-] as ChatMessage[];
+] as UIChatMessage[];
 
 const mockedChats = [
   {
@@ -105,7 +105,7 @@ const mockedChats = [
       },
     ],
   },
-] as ChatMessage[];
+] as UIChatMessage[];
 
 const mockChatStore = {
   messagesMap: {
@@ -172,7 +172,7 @@ describe('chatSelectors', () => {
           apiName: 'ttt',
           type: 'default',
         },
-      } as ChatMessage;
+      } as UIChatMessage;
       const state = merge(initialStore, {
         messagesMap: {
           [messageMapKey('abc')]: [...mockMessages, toolMessage],
@@ -350,7 +350,7 @@ describe('chatSelectors', () => {
         { id: '3', role: 'tool', content: 'Tool message 1' },
         { id: '4', role: 'user', content: 'Query' },
         { id: '5', role: 'tool', tools: [] },
-      ] as ChatMessage[];
+      ] as UIChatMessage[];
       const state: Partial<ChatStore> = {
         activeId: 'test-id',
         messagesMap: {
@@ -366,7 +366,7 @@ describe('chatSelectors', () => {
       const messages = [
         { id: '1', role: 'user', content: 'Hello' },
         { id: '2', role: 'assistant', content: 'Hi' },
-      ] as ChatMessage[];
+      ] as UIChatMessage[];
       const state: Partial<ChatStore> = {
         activeId: 'test-id',
         messagesMap: {
@@ -455,7 +455,7 @@ describe('chatSelectors', () => {
           groupId: 'group-123',
           agentId: 'agent-456',
         },
-      ] as ChatMessage[];
+      ] as UIChatMessage[];
 
       const state = merge(initialStore, {
         messagesMap: {