@lobehub/chat 1.68.11 → 1.69.1

package/src/libs/agent-runtime/utils/anthropicHelpers.ts

@@ -130,6 +130,7 @@ export const buildAnthropicMessage = async (
 
 export const buildAnthropicMessages = async (
   oaiMessages: OpenAIChatMessage[],
+  options: { enabledContextCaching?: boolean } = {},
 ): Promise<Anthropic.Messages.MessageParam[]> => {
   const messages: Anthropic.Messages.MessageParam[] = [];
   let pendingToolResults: Anthropic.ToolResultBlockParam[] = [];
@@ -180,13 +181,46 @@ export const buildAnthropicMessages = async (
     }
   }
 
+  const lastMessage = messages.at(-1);
+  if (options.enabledContextCaching && !!lastMessage) {
+    if (typeof lastMessage.content === 'string') {
+      lastMessage.content = [
+        {
+          cache_control: { type: 'ephemeral' },
+          text: lastMessage.content as string,
+          type: 'text',
+        },
+      ];
+    } else {
+      const lastContent = lastMessage.content.at(-1);
+
+      if (
+        lastContent &&
+        lastContent.type !== 'thinking' &&
+        lastContent.type !== 'redacted_thinking'
+      ) {
+        lastContent.cache_control = { type: 'ephemeral' };
+      }
+    }
+  }
   return messages;
 };
-export const buildAnthropicTools = (tools?: OpenAI.ChatCompletionTool[]) =>
-  tools?.map(
-    (tool): Anthropic.Tool => ({
+
+export const buildAnthropicTools = (
+  tools?: OpenAI.ChatCompletionTool[],
+  options: { enabledContextCaching?: boolean } = {},
+) => {
+  if (!tools) return;
+
+  return tools.map(
+    (tool, index): Anthropic.Tool => ({
+      cache_control:
+        options.enabledContextCaching && index === tools.length - 1
+          ? { type: 'ephemeral' }
+          : undefined,
       description: tool.function.description,
       input_schema: tool.function.parameters as Anthropic.Tool.InputSchema,
       name: tool.function.name,
     }),
   );
+};

package/src/libs/langchain/loaders/code/__tests__/long.json

@@ -164,11 +164,11 @@
     "metadata": { "loc": { "lines": { "from": 547, "to": 570 } } }
   },
   {
-    "pageContent": "const abortController = internal_toggleChatLoading(\n      true,\n      assistantId,\n      n('generateMessage(start)', { assistantId, messages }) as string,\n    );\n\n    const agentConfig = getAgentConfig();\n    const chatConfig = agentConfig.chatConfig;\n\n    const compiler = template(chatConfig.inputTemplate, { interpolate: /{{([\\S\\s]+?)}}/g });\n\n    // ================================== //\n    //   messages uniformly preprocess    //\n    // ================================== //\n\n    // 1. slice messages with config\n    let preprocessMsgs = chatHelpers.getSlicedMessagesWithConfig(messages, chatConfig);",
+    "pageContent": "const abortController = internal_toggleChatLoading(\n      true,\n      assistantId,\n      n('generateMessage(start)', { assistantId, messages }) as string,\n    );\n\n    const agentConfig = getAgentConfig();\n    const chatConfig = agentConfig.chatConfig;\n\n    const compiler = template(chatConfig.inputTemplate, { interpolate: /{{([\\S\\s]+?)}}/g });\n\n    // ================================== //\n    //   messages uniformly preprocess    //\n    // ================================== //\n\n    // 1. slice messages with config\n    let preprocessMsgs = chatHelpers.getSlicedMessages(messages, chatConfig);",
     "metadata": { "loc": { "lines": { "from": 566, "to": 582 } } }
   },
   {
-    "pageContent": "const compiler = template(chatConfig.inputTemplate, { interpolate: /{{([\\S\\s]+?)}}/g });\n\n    // ================================== //\n    //   messages uniformly preprocess    //\n    // ================================== //\n\n    // 1. slice messages with config\n    let preprocessMsgs = chatHelpers.getSlicedMessagesWithConfig(messages, chatConfig);\n\n    // 2. replace inputMessage template\n    preprocessMsgs = !chatConfig.inputTemplate\n      ? preprocessMsgs\n      : preprocessMsgs.map((m) => {\n          if (m.role === 'user') {\n            try {\n              return { ...m, content: compiler({ text: m.content }) };\n            } catch (error) {\n              console.error(error);\n\n              return m;\n            }\n          }\n\n          return m;\n        });",
+    "pageContent": "const compiler = template(chatConfig.inputTemplate, { interpolate: /{{([\\S\\s]+?)}}/g });\n\n    // ================================== //\n    //   messages uniformly preprocess    //\n    // ================================== //\n\n    // 1. slice messages with config\n    let preprocessMsgs = chatHelpers.getSlicedMessages(messages, chatConfig);\n\n    // 2. replace inputMessage template\n    preprocessMsgs = !chatConfig.inputTemplate\n      ? preprocessMsgs\n      : preprocessMsgs.map((m) => {\n          if (m.role === 'user') {\n            try {\n              return { ...m, content: compiler({ text: m.content }) };\n            } catch (error) {\n              console.error(error);\n\n              return m;\n            }\n          }\n\n          return m;\n        });",
     "metadata": { "loc": { "lines": { "from": 575, "to": 599 } } }
   },
   {

package/src/libs/langchain/loaders/code/__tests__/long.txt

@@ -579,7 +579,7 @@ export const chatMessage: StateCreator<
     // ================================== //
 
     // 1. slice messages with config
-    let preprocessMsgs = chatHelpers.getSlicedMessagesWithConfig(messages, chatConfig);
+    let preprocessMsgs = chatHelpers.getSlicedMessages(messages, chatConfig);
 
     // 2. replace inputMessage template
     preprocessMsgs = !chatConfig.inputTemplate

package/src/locales/default/chat.ts

@@ -33,7 +33,12 @@ export default {
   duplicateTitle: '{{title}} 副本',
   emptyAgent: '暂无助手',
   extendParams: {
+    disableContextCaching: {
+      desc: '单条对话生成成本最高可降低 90%，响应速度提升 4 倍（<1>了解更多</1>）。开启后将自动禁用历史消息数限制',
+      title: '开启上下文缓存',
+    },
     enableReasoning: {
+      desc: '基于 Claude Thinking 机制限制（<1>了解更多</1>），开启后将自动禁用历史消息数限制',
       title: '开启深度思考',
     },
     reasoningBudgetToken: {
@@ -41,6 +46,9 @@ export default {
     },
     title: '模型扩展功能',
   },
+  history: {
+    title: '助手将只记住最后{{count}}条消息',
+  },
   historyRange: '历史范围',
   historySummary: '历史消息总结',
   inbox: {

package/src/store/agent/initialState.ts

@@ -1,7 +1,7 @@
 import { AgentState, initialAgentChatState } from './slices/chat/initialState';
 
-export type SessionStoreState = AgentState;
+export type AgentStoreState = AgentState;
 
-export const initialState: SessionStoreState = {
+export const initialState: AgentStoreState = {
   ...initialAgentChatState,
 };

package/src/store/agent/selectors.ts

@@ -1 +1 @@
-export { agentSelectors } from './slices/chat/selectors';
+export { agentChatConfigSelectors,agentSelectors } from './slices/chat/selectors';

package/src/store/agent/slices/chat/{selectors.test.ts → selectors/agent.test.ts}

@@ -6,8 +6,8 @@ import { AgentStore } from '@/store/agent';
 import { AgentState } from '@/store/agent/slices/chat/initialState';
 import { merge } from '@/utils/merge';
 
-import { initialState } from '../../initialState';
-import { agentSelectors } from './selectors';
+import { initialState } from '../../../initialState';
+import { agentSelectors } from './agent';
 
 vi.mock('i18next', () => ({
   t: vi.fn((key) => key), // Simplified mock return value

package/src/store/agent/slices/chat/{selectors.ts → selectors/agent.ts}

@@ -7,64 +7,62 @@ import {
   DEFAULT_PROVIDER,
   DEFAUTT_AGENT_TTS_CONFIG,
 } from '@/const/settings';
-import { AgentStore } from '@/store/agent';
-import { LobeAgentChatConfig, LobeAgentConfig, LobeAgentTTSConfig } from '@/types/agent';
+import { AgentStoreState } from '@/store/agent/initialState';
+import { LobeAgentConfig, LobeAgentTTSConfig } from '@/types/agent';
 import { KnowledgeItem, KnowledgeType } from '@/types/knowledgeBase';
 import { merge } from '@/utils/merge';
 
-const isInboxSession = (s: AgentStore) => s.activeId === INBOX_SESSION_ID;
+const isInboxSession = (s: AgentStoreState) => s.activeId === INBOX_SESSION_ID;
 
 // ==========   Config   ============== //
 
-const inboxAgentConfig = (s: AgentStore) =>
+const inboxAgentConfig = (s: AgentStoreState) =>
   merge(DEFAULT_AGENT_CONFIG, s.agentMap[INBOX_SESSION_ID]);
-const inboxAgentModel = (s: AgentStore) => inboxAgentConfig(s).model;
+const inboxAgentModel = (s: AgentStoreState) => inboxAgentConfig(s).model;
 
 const getAgentConfigById =
   (id: string) =>
-  (s: AgentStore): LobeAgentConfig =>
+  (s: AgentStoreState): LobeAgentConfig =>
     merge(s.defaultAgentConfig, s.agentMap[id]);
 
-const currentAgentConfig = (s: AgentStore): LobeAgentConfig => getAgentConfigById(s.activeId)(s);
+export const currentAgentConfig = (s: AgentStoreState): LobeAgentConfig =>
+  getAgentConfigById(s.activeId)(s);
 
-const currentAgentChatConfig = (s: AgentStore): LobeAgentChatConfig =>
-  currentAgentConfig(s).chatConfig || {};
-
-const currentAgentSystemRole = (s: AgentStore) => {
+const currentAgentSystemRole = (s: AgentStoreState) => {
   return currentAgentConfig(s).systemRole;
 };
 
-const currentAgentModel = (s: AgentStore): string => {
+const currentAgentModel = (s: AgentStoreState): string => {
   const config = currentAgentConfig(s);
 
   return config?.model || DEFAULT_MODEL;
 };
 
-const currentAgentModelProvider = (s: AgentStore) => {
+const currentAgentModelProvider = (s: AgentStoreState) => {
   const config = currentAgentConfig(s);
 
   return config?.provider || DEFAULT_PROVIDER;
 };
 
-const currentAgentPlugins = (s: AgentStore) => {
+const currentAgentPlugins = (s: AgentStoreState) => {
   const config = currentAgentConfig(s);
 
   return config?.plugins || [];
 };
 
-const currentAgentKnowledgeBases = (s: AgentStore) => {
+const currentAgentKnowledgeBases = (s: AgentStoreState) => {
   const config = currentAgentConfig(s);
 
   return config?.knowledgeBases || [];
 };
 
-const currentAgentFiles = (s: AgentStore) => {
+const currentAgentFiles = (s: AgentStoreState) => {
   const config = currentAgentConfig(s);
 
   return config?.files || [];
 };
 
-const currentAgentTTS = (s: AgentStore): LobeAgentTTSConfig => {
+const currentAgentTTS = (s: AgentStoreState): LobeAgentTTSConfig => {
   const config = currentAgentConfig(s);
 
   return config?.tts || DEFAUTT_AGENT_TTS_CONFIG;
@@ -72,7 +70,7 @@ const currentAgentTTS = (s: AgentStore): LobeAgentTTSConfig => {
 
 const currentAgentTTSVoice =
   (lang: string) =>
-  (s: AgentStore): string => {
+  (s: AgentStoreState): string => {
     const { voice, ttsService } = currentAgentTTS(s);
     const voiceList = new VoiceList(lang);
     let currentVoice;
@@ -93,7 +91,7 @@ const currentAgentTTSVoice =
     return currentVoice || 'alloy';
   };
 
-const currentEnabledKnowledge = (s: AgentStore) => {
+const currentEnabledKnowledge = (s: AgentStoreState) => {
   const knowledgeBases = currentAgentKnowledgeBases(s);
   const files = currentAgentFiles(s);
 
@@ -107,29 +105,27 @@ const currentEnabledKnowledge = (s: AgentStore) => {
   ] as KnowledgeItem[];
 };
 
-const agentSearchMode = (s: AgentStore) => currentAgentChatConfig(s).searchMode || 'off';
-
-const hasSystemRole = (s: AgentStore) => {
+const hasSystemRole = (s: AgentStoreState) => {
   const config = currentAgentConfig(s);
 
   return !!config.systemRole;
 };
 
-const hasKnowledgeBases = (s: AgentStore) => {
+const hasKnowledgeBases = (s: AgentStoreState) => {
   const knowledgeBases = currentAgentKnowledgeBases(s);
 
   return knowledgeBases.length > 0;
 };
 
-const hasFiles = (s: AgentStore) => {
+const hasFiles = (s: AgentStoreState) => {
   const files = currentAgentFiles(s);
 
   return files.length > 0;
 };
 
-const hasKnowledge = (s: AgentStore) => hasKnowledgeBases(s) || hasFiles(s);
-const hasEnabledKnowledge = (s: AgentStore) => currentEnabledKnowledge(s).length > 0;
-const currentKnowledgeIds = (s: AgentStore) => {
+const hasKnowledge = (s: AgentStoreState) => hasKnowledgeBases(s) || hasFiles(s);
+const hasEnabledKnowledge = (s: AgentStoreState) => currentEnabledKnowledge(s).length > 0;
+const currentKnowledgeIds = (s: AgentStoreState) => {
   return {
     fileIds: currentAgentFiles(s)
       .filter((item) => item.enabled)
@@ -140,13 +136,9 @@ const currentKnowledgeIds = (s: AgentStore) => {
   };
 };
 
-const isAgentConfigLoading = (s: AgentStore) => !s.agentConfigInitMap[s.activeId];
-
-const isAgentEnableSearch = (s: AgentStore) => agentSearchMode(s) !== 'off';
+const isAgentConfigLoading = (s: AgentStoreState) => !s.agentConfigInitMap[s.activeId];
 
 export const agentSelectors = {
-  agentSearchMode,
-  currentAgentChatConfig,
   currentAgentConfig,
   currentAgentFiles,
   currentAgentKnowledgeBases,
@@ -165,6 +157,5 @@ export const agentSelectors = {
   inboxAgentConfig,
   inboxAgentModel,
   isAgentConfigLoading,
-  isAgentEnableSearch,
   isInboxSession,
 };

package/src/store/agent/slices/chat/selectors/chatConfig.test.ts

@@ -0,0 +1,184 @@
+import { describe, expect, it } from 'vitest';
+
+import { DEFAULT_AGENT_CHAT_CONFIG, DEFAULT_AGENT_CONFIG } from '@/const/settings';
+import { AgentStoreState, initialState } from '@/store/agent/initialState';
+import { LobeAgentChatConfig, LobeAgentConfig } from '@/types/agent';
+import { merge } from '@/utils/merge';
+
+import { agentChatConfigSelectors, currentAgentChatConfig } from './chatConfig';
+
+// Helper function to create mock state
+const createMockState = (
+  chatConfig?: Partial<LobeAgentChatConfig>,
+  config?: Partial<LobeAgentConfig>,
+): AgentStoreState =>
+  merge(initialState, {
+    agentMap: {
+      agent1: merge(DEFAULT_AGENT_CONFIG, { ...config, chatConfig }),
+    },
+    activeId: 'agent1',
+  }) as AgentStoreState;
+
+describe('agentChatConfigSelectors', () => {
+  describe('currentAgentChatConfig', () => {
+    it('should return default object when chatConfig is not defined', () => {
+      const state = createMockState({});
+      expect(currentAgentChatConfig(state)).toEqual(DEFAULT_AGENT_CHAT_CONFIG);
+    });
+
+    it('should return the chatConfig when defined', () => {
+      const chatConfig = { historyCount: 5 };
+      const state = createMockState(chatConfig);
+      expect(currentAgentChatConfig(state).historyCount).toEqual(5);
+    });
+  });
+
+  describe('enableHistoryCount', () => {
+    it('should return the enableHistoryCount value when defined', () => {
+      const state = createMockState({ enableHistoryCount: false });
+      expect(agentChatConfigSelectors.enableHistoryCount(state)).toBe(false);
+    });
+
+    it('should return false value when enable context caching with claude models', () => {
+      const state = createMockState(
+        { enableHistoryCount: true, disableContextCaching: false },
+        { model: 'claude-3-7-sonnet-20250219' },
+      );
+
+      expect(agentChatConfigSelectors.enableHistoryCount(merge(state))).toBe(false);
+    });
+
+    it('should return true value when enable context caching with other models', () => {
+      const state = createMockState(
+        { enableHistoryCount: true, disableContextCaching: false },
+        { model: 'gpt-4o-min' },
+      );
+
+      expect(agentChatConfigSelectors.enableHistoryCount(merge(state))).toBe(true);
+    });
+
+    it('should return false when enable search with claude 3.7 models', () => {
+      const state = createMockState(
+        { enableHistoryCount: true, disableContextCaching: true, searchMode: 'auto' },
+        { model: 'claude-3-7-sonnet-20250219' },
+      );
+
+      expect(agentChatConfigSelectors.enableHistoryCount(merge(state))).toBe(false);
+    });
+
+    it('should return true when disable search with claude 3.7 models', () => {
+      const state = createMockState(
+        { enableHistoryCount: true, disableContextCaching: true, searchMode: 'off' },
+        { model: 'claude-3-7-sonnet-20250219' },
+      );
+
+      expect(agentChatConfigSelectors.enableHistoryCount(merge(state))).toBe(true);
+    });
+
+    it('should return true when enable search with claude 3.5 models', () => {
+      const state = createMockState(
+        { enableHistoryCount: true, disableContextCaching: true, searchMode: 'auto' },
+        { model: 'claude-3-5-sonnet-latest' },
+      );
+
+      expect(agentChatConfigSelectors.enableHistoryCount(merge(state))).toBe(true);
+    });
+  });
+
+  describe('historyCount', () => {
+    it('should return undefined when historyCount is not defined', () => {
+      const state = createMockState();
+      expect(agentChatConfigSelectors.historyCount(state)).toBe(8);
+    });
+
+    it('should return the historyCount value when defined', () => {
+      const state = createMockState({ historyCount: 20 });
+      expect(agentChatConfigSelectors.historyCount(state)).toBe(20);
+    });
+  });
+
+  describe('agentSearchMode', () => {
+    it('should return "off" when searchMode is not defined', () => {
+      const state = createMockState();
+      expect(agentChatConfigSelectors.agentSearchMode(state)).toBe('off');
+    });
+
+    it('should return the searchMode value when defined', () => {
+      const state = createMockState({ searchMode: 'auto' });
+      expect(agentChatConfigSelectors.agentSearchMode(state)).toBe('auto');
+    });
+  });
+
+  describe('useModelBuiltinSearch', () => {
+    it('should return undefined when useModelBuiltinSearch is not defined', () => {
+      const state = createMockState();
+      expect(agentChatConfigSelectors.useModelBuiltinSearch(state)).toBeUndefined();
+    });
+
+    it('should return the useModelBuiltinSearch value when defined', () => {
+      const state = createMockState({ useModelBuiltinSearch: true });
+      expect(agentChatConfigSelectors.useModelBuiltinSearch(state)).toBe(true);
+    });
+  });
+
+  describe('isAgentEnableSearch', () => {
+    it('should return false when searchMode is "off"', () => {
+      const state = createMockState({ searchMode: 'off' });
+      expect(agentChatConfigSelectors.isAgentEnableSearch(state)).toBe(false);
+    });
+
+    it('should return true when searchMode is not "off"', () => {
+      const state = createMockState({ searchMode: 'auto' });
+      expect(agentChatConfigSelectors.isAgentEnableSearch(state)).toBe(true);
+    });
+
+    it('should return false when searchMode is undefined (defaults to "off")', () => {
+      const state = createMockState();
+      expect(agentChatConfigSelectors.isAgentEnableSearch(state)).toBe(false);
+    });
+  });
+
+  describe('displayMode', () => {
+    it('should return "chat" when displayMode is not defined', () => {
+      const state = createMockState();
+      expect(agentChatConfigSelectors.displayMode(state)).toBe('chat');
+    });
+
+    it('should return the displayMode value when defined', () => {
+      const state = createMockState({ displayMode: 'docs' });
+      expect(agentChatConfigSelectors.displayMode(state)).toBe('docs');
+    });
+  });
+
+  describe('enableHistoryDivider', () => {
+    it('should return false when enableHistoryCount is false', () => {
+      const state = createMockState({ enableHistoryCount: false, historyCount: 5 });
+      const selector = agentChatConfigSelectors.enableHistoryDivider(10, 5);
+      expect(selector(state)).toBe(false);
+    });
+
+    it('should return false when historyLength is less than or equal to historyCount', () => {
+      const state = createMockState({ enableHistoryCount: true, historyCount: 10 });
+      const selector = agentChatConfigSelectors.enableHistoryDivider(10, 5);
+      expect(selector(state)).toBe(false);
+    });
+
+    it('should return false when currentIndex is not at the historyCount position', () => {
+      const state = createMockState({ enableHistoryCount: true, historyCount: 5 });
+      const selector = agentChatConfigSelectors.enableHistoryDivider(10, 3);
+      expect(selector(state)).toBe(false);
+    });
+
+    it('should return true when all conditions are met', () => {
+      const state = createMockState({ enableHistoryCount: true, historyCount: 5 });
+      const selector = agentChatConfigSelectors.enableHistoryDivider(10, 5);
+      expect(selector(state)).toBe(true);
+    });
+
+    it('should handle undefined or null historyCount', () => {
+      const state = createMockState({ enableHistoryCount: true, historyCount: null as any });
+      const selector = agentChatConfigSelectors.enableHistoryDivider(10, 5);
+      expect(selector(state)).toBe(false);
+    });
+  });
+});

package/src/store/agent/slices/chat/selectors/chatConfig.ts

@@ -0,0 +1,65 @@
+import { contextCachingModels, thinkingWithToolClaudeModels } from '@/const/models';
+import { AgentStoreState } from '@/store/agent/initialState';
+import { LobeAgentChatConfig } from '@/types/agent';
+
+import { currentAgentConfig } from './agent';
+
+export const currentAgentChatConfig = (s: AgentStoreState): LobeAgentChatConfig =>
+  currentAgentConfig(s).chatConfig || {};
+
+const agentSearchMode = (s: AgentStoreState) => currentAgentChatConfig(s).searchMode || 'off';
+const isAgentEnableSearch = (s: AgentStoreState) => agentSearchMode(s) !== 'off';
+
+const useModelBuiltinSearch = (s: AgentStoreState) =>
+  currentAgentChatConfig(s).useModelBuiltinSearch;
+
+const enableHistoryCount = (s: AgentStoreState) => {
+  const config = currentAgentConfig(s);
+  const chatConfig = currentAgentChatConfig(s);
+
+  // 如果开启了上下文缓存，且当前模型类型匹配，则不开启历史记录
+  const enableContextCaching = !chatConfig.disableContextCaching;
+
+  if (enableContextCaching && contextCachingModels.has(config.model)) return false;
+
+  // 当开启搜索时，针对 claude 3.7 sonnet 模型不开启历史记录
+  const enableSearch = isAgentEnableSearch(s);
+
+  if (enableSearch && thinkingWithToolClaudeModels.has(config.model)) return false;
+
+  return chatConfig.enableHistoryCount;
+};
+
+const historyCount = (s: AgentStoreState) => {
+  const chatConfig = currentAgentChatConfig(s);
+
+  return chatConfig.historyCount;
+};
+
+const displayMode = (s: AgentStoreState) => {
+  const chatConfig = currentAgentChatConfig(s);
+
+  return chatConfig.displayMode || 'chat';
+};
+
+const enableHistoryDivider =
+  (historyLength: number, currentIndex: number) => (s: AgentStoreState) => {
+    const config = currentAgentChatConfig(s);
+
+    return (
+      enableHistoryCount(s) &&
+      historyLength > (config.historyCount ?? 0) &&
+      config.historyCount === historyLength - currentIndex
+    );
+  };
+
+export const agentChatConfigSelectors = {
+  agentSearchMode,
+  currentChatConfig: currentAgentChatConfig,
+  displayMode,
+  enableHistoryCount,
+  enableHistoryDivider,
+  historyCount,
+  isAgentEnableSearch,
+  useModelBuiltinSearch,
+};

package/src/store/agent/slices/chat/selectors/index.ts

@@ -0,0 +1,2 @@
+export { agentSelectors } from './agent';
+export { agentChatConfigSelectors } from './chatConfig';

package/src/store/agent/store.ts

@@ -3,12 +3,12 @@ import { createWithEqualityFn } from 'zustand/traditional';
 import { StateCreator } from 'zustand/vanilla';
 
 import { createDevtools } from '../middleware/createDevtools';
-import { SessionStoreState, initialState } from './initialState';
+import { AgentStoreState, initialState } from './initialState';
 import { AgentChatAction, createChatSlice } from './slices/chat/action';
 
 //  ===============  aggregate createStoreFn ============ //
 
-export interface AgentStore extends AgentChatAction, SessionStoreState {}
+export interface AgentStore extends AgentChatAction, AgentStoreState {}
 
 const createStore: StateCreator<AgentStore, [['zustand/devtools', never]]> = (...parameters) => ({
   ...initialState,

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

@@ -66,7 +66,7 @@ describe('chatHelpers', () => {
     });
   });
 
-  describe('getSlicedMessagesWithConfig', () => {
+  describe('getSlicedMessages', () => {
     const messages = [
       { id: '1', content: 'First' },
       { id: '2', content: 'Second' },
@@ -75,13 +75,13 @@ describe('chatHelpers', () => {
 
     it('returns all messages if history is disabled', () => {
       const config = { enableHistoryCount: false, historyCount: undefined } as LobeAgentChatConfig;
-      const slicedMessages = chatHelpers.getSlicedMessagesWithConfig(messages, config);
+      const slicedMessages = chatHelpers.getSlicedMessages(messages, config);
       expect(slicedMessages).toEqual(messages);
     });
 
     it('returns last N messages based on historyCount', () => {
       const config = { enableHistoryCount: true, historyCount: 2 } as LobeAgentChatConfig;
-      const slicedMessages = chatHelpers.getSlicedMessagesWithConfig(messages, config);
+      const slicedMessages = chatHelpers.getSlicedMessages(messages, config);
       expect(slicedMessages).toEqual([
         { id: '2', content: 'Second' },
         { id: '3', content: 'Third' },
@@ -90,25 +90,25 @@ describe('chatHelpers', () => {
 
     it('returns empty array when historyCount is negative', () => {
       const config = { enableHistoryCount: true, historyCount: -1 } as LobeAgentChatConfig;
-      const slicedMessages = chatHelpers.getSlicedMessagesWithConfig(messages, config);
+      const slicedMessages = chatHelpers.getSlicedMessages(messages, config);
       expect(slicedMessages).toEqual([]);
     });
 
     it('returns all messages if historyCount exceeds the array length', () => {
       const config = { enableHistoryCount: true, historyCount: 5 } as LobeAgentChatConfig;
-      const slicedMessages = chatHelpers.getSlicedMessagesWithConfig(messages, config);
+      const slicedMessages = chatHelpers.getSlicedMessages(messages, config);
       expect(slicedMessages).toEqual(messages);
     });
 
     it('returns an empty array for an empty message array', () => {
       const config = { enableHistoryCount: true, historyCount: 2 } as LobeAgentChatConfig;
-      const slicedMessages = chatHelpers.getSlicedMessagesWithConfig([], config);
+      const slicedMessages = chatHelpers.getSlicedMessages([], config);
       expect(slicedMessages).toEqual([]);
     });
 
     it('returns an empty array when historyCount is zero', () => {
       const config = { enableHistoryCount: true, historyCount: 0 } as LobeAgentChatConfig;
-      const slicedMessages = chatHelpers.getSlicedMessagesWithConfig(messages, config);
+      const slicedMessages = chatHelpers.getSlicedMessages(messages, config);
       expect(slicedMessages).toEqual([]);
     });
   });

package/src/store/chat/helpers.ts

@@ -1,4 +1,3 @@
-import { LobeAgentChatConfig } from '@/types/agent';
 import { ChatMessage } from '@/types/message';
 import { OpenAIChatMessage } from '@/types/openai/chat';
 import { encodeAsync } from '@/utils/tokenizer';
@@ -9,16 +8,21 @@ export const getMessagesTokenCount = async (messages: OpenAIChatMessage[]) =>
 export const getMessageById = (messages: ChatMessage[], id: string) =>
   messages.find((m) => m.id === id);
 
-const getSlicedMessagesWithConfig = (
+const getSlicedMessages = (
   messages: ChatMessage[],
-  config: LobeAgentChatConfig,
-  includeNewUserMessage?: boolean,
+  options: {
+    enableHistoryCount?: boolean;
+    historyCount?: number;
+    includeNewUserMessage?: boolean;
+  },
 ): ChatMessage[] => {
   // if historyCount is not enabled, return all messages
-  if (!config.enableHistoryCount || config.historyCount === undefined) return messages;
+  if (!options.enableHistoryCount || options.historyCount === undefined) return messages;
 
   // if user send message, history will include this message so the total length should +1
-  const messagesCount = !!includeNewUserMessage ? config.historyCount + 1 : config.historyCount;
+  const messagesCount = !!options.includeNewUserMessage
+    ? options.historyCount + 1
+    : options.historyCount;
 
   // if historyCount is negative or set to 0, return empty array
   if (messagesCount <= 0) return [];
@@ -30,5 +34,5 @@ const getSlicedMessagesWithConfig = (
 export const chatHelpers = {
   getMessageById,
   getMessagesTokenCount,
-  getSlicedMessagesWithConfig,
+  getSlicedMessages,
 };