@lobehub/chat 0.115.2 → 0.115.4

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

@@ -0,0 +1,29 @@
+import { ChatMessage } from '@/types/message';
+
+export interface ChatMessageState {
+  abortController?: AbortController;
+  /**
+   * @title 当前活动的会话
+   * @description 当前正在编辑或查看的会话
+   */
+  activeId: string;
+  /**
+   * @deprecated
+   */
+  chatLoadingId?: string;
+  inputMessage: string;
+  messageLoadingIds: [];
+  messages: ChatMessage[];
+  /**
+   * whether messages have fetched
+   */
+  messagesInit: boolean;
+}
+
+export const initialMessageState: ChatMessageState = {
+  activeId: 'inbox',
+  inputMessage: '',
+  messageLoadingIds: [],
+  messages: [],
+  messagesInit: false,
+};

package/src/store/chat/{reducers/message.test.ts → slices/message/reducer.test.ts}

@@ -1,6 +1,6 @@
-import { ChatMessage, ChatMessageMap } from '@/types/chatMessage';
+import { ChatMessage } from '@/types/message';
 
-import { MessageDispatch, messagesReducer } from './message';
+import { MessageDispatch, messagesReducer } from './reducer';
 
 describe('messagesReducer', () => {
   let initialState: ChatMessage[];

package/src/store/chat/{reducers/message.ts → slices/message/reducer.ts}

@@ -1,7 +1,7 @@
 import isEqual from 'fast-deep-equal';
 import { produce } from 'immer';
 
-import { ChatMessage } from '@/types/chatMessage';
+import { ChatMessage } from '@/types/message';
 import { merge } from '@/utils/merge';
 
 interface UpdateMessage {

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

@@ -5,11 +5,11 @@ import { INBOX_SESSION_ID } from '@/const/session';
 import { ChatStore } from '@/store/chat';
 import { initialState } from '@/store/chat/initialState';
 import { useSessionStore } from '@/store/session';
-import { ChatMessage } from '@/types/chatMessage';
+import { ChatMessage } from '@/types/message';
 import { MetaData } from '@/types/meta';
 import { merge } from '@/utils/merge';
 
-import { chatSelectors } from '../chat';
+import { chatSelectors } from './selectors';
 
 vi.mock('i18next', () => ({
   t: vi.fn((key) => key), // Simplified mock return value

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

@@ -6,12 +6,12 @@ import { INBOX_SESSION_ID } from '@/const/session';
 import { useGlobalStore } from '@/store/global';
 import { useSessionStore } from '@/store/session';
 import { agentSelectors } from '@/store/session/selectors';
-import { ChatMessage } from '@/types/chatMessage';
+import { ChatMessage } from '@/types/message';
 import { MetaData } from '@/types/meta';
 import { merge } from '@/utils/merge';
 
-import { chatHelpers } from '../helpers';
-import type { ChatStore } from '../store';
+import { chatHelpers } from '../../helpers';
+import type { ChatStore } from '../../store';
 
 const getMeta = (message: ChatMessage) => {
   switch (message.role) {

package/src/store/chat/{actions/share.ts → slices/share/action.ts}

@@ -8,8 +8,8 @@ import { useSessionStore } from '@/store/session';
 import { agentSelectors } from '@/store/session/selectors';
 import { ShareGPTConversation } from '@/types/share';
 
-import { chatSelectors } from '../selectors';
-import { ChatStore } from '../store';
+import { chatSelectors } from '../../selectors';
+import { ChatStore } from '../../store';
 
 interface ShareMessage {
   from: 'human' | 'gpt';

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

@@ -0,0 +1,7 @@
+export interface ChatShareState {
+  shareLoading?: boolean;
+}
+
+export const initialShareState: ChatShareState = {
+  shareLoading: false,
+};

package/src/store/chat/{actions/plugin.ts → slices/tool/action.ts}

@@ -7,11 +7,11 @@ import { messageService } from '@/services/message';
 import { ChatStore } from '@/store/chat/store';
 import { useToolStore } from '@/store/tool';
 import { pluginSelectors } from '@/store/tool/selectors';
-import { ChatPluginPayload } from '@/types/chatMessage';
+import { ChatPluginPayload } from '@/types/message';
 import { OpenAIFunctionCall } from '@/types/openai/functionCall';
 import { setNamespace } from '@/utils/storeDebug';
 
-import { chatSelectors } from '../selectors';
+import { chatSelectors } from '../../selectors';
 
 const n = setNamespace('plugin');
 
@@ -85,7 +85,6 @@ export const chatPlugin: StateCreator<
     const chats = chatSelectors.currentChats(get());
     await coreProcessMessage(chats, id);
   },
-
   triggerFunctionCall: async (id) => {
     const { invokeDefaultTypePlugin, invokeBuiltinTool, refreshMessages } = get();
 
@@ -146,7 +145,6 @@ export const chatPlugin: StateCreator<
       }
     }
   },
-
   updatePluginState: async (id, key, value) => {
     const { refreshMessages } = get();
 

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

@@ -0,0 +1,7 @@
+export interface ChatToolState {
+  dalleImageLoading: Record<string, boolean>;
+}
+
+export const initialToolState: ChatToolState = {
+  dalleImageLoading: {},
+};

package/src/store/chat/{actions/topic.test.ts → slices/topic/action.test.ts}

@@ -6,10 +6,10 @@ import { LOADING_FLAT } from '@/const/message';
 import { chatService } from '@/services/chat';
 import { messageService } from '@/services/message';
 import { topicService } from '@/services/topic';
-import { ChatMessage } from '@/types/chatMessage';
+import { ChatMessage } from '@/types/message';
 import { ChatTopic } from '@/types/topic';
 
-import { useChatStore } from '../store';
+import { useChatStore } from '../../store';
 
 // Mock topicService 和 messageService
 vi.mock('@/services/topic', () => ({

package/src/store/chat/{actions/topic.ts → slices/topic/action.ts}

@@ -12,11 +12,12 @@ import { chatService } from '@/services/chat';
 import { messageService } from '@/services/message';
 import { topicService } from '@/services/topic';
 import { ChatStore } from '@/store/chat';
-import { ChatMessage } from '@/types/chatMessage';
+import { ChatMessage } from '@/types/message';
 import { ChatTopic } from '@/types/topic';
 import { setNamespace } from '@/utils/storeDebug';
 
-import { chatSelectors, topicSelectors } from '../selectors';
+import { chatSelectors } from '../message/selectors';
+import { topicSelectors } from './selectors';
 
 const n = setNamespace('topic');
 

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

@@ -0,0 +1,23 @@
+import { ChatTopic } from '@/types/topic';
+
+export interface ChatTopicState {
+  activeTopicId?: string;
+  isSearchingTopic: boolean;
+  searchTopics: ChatTopic[];
+  topicLoadingId?: string;
+  topicRenamingId?: string;
+  topicSearchKeywords: string;
+  topics: ChatTopic[];
+  /**
+   * whether topics have fetched
+   */
+  topicsInit: boolean;
+}
+
+export const initialTopicState: ChatTopicState = {
+  isSearchingTopic: false,
+  searchTopics: [],
+  topicSearchKeywords: '',
+  topics: [],
+  topicsInit: false,
+};

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

@@ -4,7 +4,7 @@ import { ChatStore } from '@/store/chat';
 import { initialState } from '@/store/chat/initialState';
 import { merge } from '@/utils/merge';
 
-import { topicSelectors } from '..';
+import { topicSelectors } from '../../selectors';
 
 const initialStore = initialState as ChatStore;
 

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

@@ -1,6 +1,6 @@
 import { ChatTopic } from '@/types/topic';
 
-import { ChatStore } from '../store';
+import { ChatStore } from '../../store';
 
 const currentTopics = (s: ChatStore): ChatTopic[] => s.topics;
 

package/src/store/chat/store.ts

@@ -6,12 +6,12 @@ import { StateCreator } from 'zustand/vanilla';
 import { createHyperStorage } from '@/store/middleware/createHyperStorage';
 import { isDev } from '@/utils/env';
 
-import { ChatEnhanceAction, chatEnhance } from './actions/enhance';
-import { ChatMessageAction, chatMessage } from './actions/message';
-import { ChatPluginAction, chatPlugin } from './actions/plugin';
-import { ShareAction, chatShare } from './actions/share';
-import { ChatTopicAction, chatTopic } from './actions/topic';
 import { ChatStoreState, initialState } from './initialState';
+import { ChatEnhanceAction, chatEnhance } from './slices/enchance/action';
+import { ChatMessageAction, chatMessage } from './slices/message/action';
+import { ShareAction, chatShare } from './slices/share/action';
+import { ChatPluginAction, chatPlugin } from './slices/tool/action';
+import { ChatTopicAction, chatTopic } from './slices/topic/action';
 
 export interface ChatStoreAction
   extends ChatMessageAction,

package/src/store/global/helpers.ts

@@ -1,4 +1,4 @@
-import { settingsSelectors } from './selectors';
+import { settingsSelectors } from './slices/settings/selectors';
 import { useGlobalStore } from './store';
 
 const getCurrentLanguage = () => settingsSelectors.currentLanguage(useGlobalStore.getState());

package/src/store/global/index.ts

@@ -1,3 +1,2 @@
 export * from './hooks';
-export * from './selectors';
 export * from './store';

package/src/store/global/initialState.ts

@@ -1,66 +1,11 @@
-import { DEFAULT_SETTINGS } from '@/const/settings';
-import type { GlobalServerConfig, GlobalSettings } from '@/types/settings';
+import { GlobalCommonState, initialCommonState } from './slices/common/initialState';
+import { GlobalSettingsState, initialSettingsState } from './slices/settings/initialState';
 
-export enum SidebarTabKey {
-  Chat = 'chat',
-  Market = 'market',
-  Setting = 'settings',
-}
+export { SettingsTabs, SidebarTabKey } from './slices/common/initialState';
 
-export enum SettingsTabs {
-  Agent = 'agent',
-  Common = 'common',
-  LLM = 'llm',
-  TTS = 'tts',
-}
-
-export interface Guide {
-  // Topic 引导
-  topic?: boolean;
-}
-
-export interface GlobalPreference {
-  guide?: Guide;
-  inputHeight: number;
-  mobileShowTopic?: boolean;
-  sessionGroupKeys: string[];
-  sessionsWidth: number;
-  showChatSideBar?: boolean;
-  showSessionPanel?: boolean;
-  showSystemRole?: boolean;
-}
-
-export interface GlobalState {
-  hasNewVersion?: boolean;
-  latestVersion?: string;
-  /**
-   *  用户偏好的 UI 状态
-   *  @localStorage
-   */
-  preference: GlobalPreference;
-  serverConfig: GlobalServerConfig;
-  /**
-   * @localStorage
-   * 用户设置
-   */
-  settings: GlobalSettings;
-  settingsTab: SettingsTabs;
-  sidebarKey: SidebarTabKey;
-}
+export type GlobalState = GlobalCommonState & GlobalSettingsState;
 
 export const initialState: GlobalState = {
-  preference: {
-    guide: {},
-    inputHeight: 200,
-    mobileShowTopic: false,
-    sessionGroupKeys: ['pinned', 'sessionList'],
-    sessionsWidth: 320,
-    showChatSideBar: true,
-    showSessionPanel: true,
-    showSystemRole: false,
-  },
-  serverConfig: {},
-  settings: DEFAULT_SETTINGS,
-  settingsTab: SettingsTabs.Common,
-  sidebarKey: SidebarTabKey.Chat,
+  ...initialCommonState,
+  ...initialSettingsState,
 };

package/src/store/global/selectors.ts

@@ -1,2 +1,2 @@
-export * from './selectors/preference';
-export * from './selectors/settings';
+export * from './slices/common/selectors';
+export * from './slices/settings/selectors';

package/src/store/global/slices/{common.ts → common/action.ts}

@@ -5,12 +5,12 @@ import type { StateCreator } from 'zustand/vanilla';
 
 import { CURRENT_VERSION } from '@/const/version';
 import { globalService } from '@/services/global';
+import type { GlobalStore } from '@/store/global';
 import type { GlobalServerConfig } from '@/types/settings';
 import { merge } from '@/utils/merge';
 import { setNamespace } from '@/utils/storeDebug';
 
-import type { GlobalPreference, GlobalState, Guide, SidebarTabKey } from '../initialState';
-import type { GlobalStore } from '../store';
+import type { GlobalCommonState, GlobalPreference, Guide, SidebarTabKey } from './initialState';
 
 const n = setNamespace('settings');
 
@@ -66,7 +66,7 @@ export const createCommonSlice: StateCreator<
   },
   updatePreference: (preference, action) => {
     set(
-      produce((draft: GlobalState) => {
+      produce((draft: GlobalCommonState) => {
         draft.preference = merge(draft.preference, preference);
       }),
       false,

package/src/store/global/slices/common/initialState.ts

@@ -0,0 +1,55 @@
+export enum SidebarTabKey {
+  Chat = 'chat',
+  Market = 'market',
+  Setting = 'settings',
+}
+
+export enum SettingsTabs {
+  Agent = 'agent',
+  Common = 'common',
+  LLM = 'llm',
+  TTS = 'tts',
+}
+
+export interface Guide {
+  // Topic 引导
+  topic?: boolean;
+}
+
+export interface GlobalPreference {
+  guide?: Guide;
+  inputHeight: number;
+  mobileShowTopic?: boolean;
+  sessionGroupKeys: string[];
+  sessionsWidth: number;
+  showChatSideBar?: boolean;
+  showSessionPanel?: boolean;
+  showSystemRole?: boolean;
+}
+
+export interface GlobalCommonState {
+  hasNewVersion?: boolean;
+  latestVersion?: string;
+  /**
+   *  用户偏好的 UI 状态
+   *  @localStorage
+   */
+  preference: GlobalPreference;
+  settingsTab: SettingsTabs;
+  sidebarKey: SidebarTabKey;
+}
+
+export const initialCommonState: GlobalCommonState = {
+  preference: {
+    guide: {},
+    inputHeight: 200,
+    mobileShowTopic: false,
+    sessionGroupKeys: ['pinned', 'sessionList'],
+    sessionsWidth: 320,
+    showChatSideBar: true,
+    showSessionPanel: true,
+    showSystemRole: false,
+  },
+  settingsTab: SettingsTabs.Common,
+  sidebarKey: SidebarTabKey.Chat,
+};

package/src/store/global/{selectors/preference.ts → slices/common/selectors.ts}

@@ -1,6 +1,6 @@
 import { GlobalStore } from '@/store/global';
 
-import { initialState } from '../initialState';
+import { initialState } from '../../initialState';
 
 const sessionGroupKeys = (s: GlobalStore): string[] =>
   s.preference.sessionGroupKeys || initialState.preference.sessionGroupKeys;

package/src/store/global/slices/{settings.ts → settings/action.ts}

@@ -5,13 +5,13 @@ import { DeepPartial } from 'utility-types';
 import type { StateCreator } from 'zustand/vanilla';
 
 import { DEFAULT_AGENT, DEFAULT_SETTINGS } from '@/const/settings';
+import type { GlobalStore } from '@/store/global';
 import { SettingsTabs } from '@/store/global/initialState';
+import { LobeAgentSettings } from '@/types/session';
 import type { GlobalSettings, OpenAIConfig } from '@/types/settings';
 import { merge } from '@/utils/merge';
 import { setNamespace } from '@/utils/storeDebug';
 
-import type { GlobalStore } from '../store';
-
 const n = setNamespace('settings');
 
 /**
@@ -24,6 +24,7 @@ export interface SettingsAction {
    */
   resetSettings: () => void;
   setOpenAIConfig: (config: Partial<OpenAIConfig>) => void;
+
   /**
    * 设置部分配置设置
    * @param settings - 部分配置设置
@@ -35,6 +36,7 @@ export interface SettingsAction {
    * @param themeMode - 主题模式
    */
   switchThemeMode: (themeMode: ThemeMode) => void;
+  updateDefaultAgent: (agent: DeepPartial<LobeAgentSettings>) => void;
 }
 
 export const createSettingsSlice: StateCreator<
@@ -46,7 +48,7 @@ export const createSettingsSlice: StateCreator<
   importAppSettings: (importAppSettings) => {
     const { setSettings } = get();
     // eslint-disable-next-line @typescript-eslint/no-unused-vars
-    const { OPENAI_API_KEY: _, password: __, ...settings } = importAppSettings;
+    const { password: _, ...settings } = importAppSettings;
 
     setSettings({
       ...settings,
@@ -78,8 +80,14 @@ export const createSettingsSlice: StateCreator<
   switchSettingTabs: (tab) => {
     set({ settingsTab: tab });
   },
-
   switchThemeMode: (themeMode) => {
     get().setSettings({ themeMode });
   },
+  updateDefaultAgent: (agent) => {
+    const settings = produce(get().settings, (draft: GlobalSettings) => {
+      draft.defaultAgent = merge(draft.defaultAgent, agent);
+    });
+
+    set({ settings }, false, n('updateDefaultAgent', agent));
+  },
 });

package/src/store/global/slices/settings/initialState.ts

@@ -0,0 +1,15 @@
+import { DEFAULT_SETTINGS } from '@/const/settings';
+import type { GlobalServerConfig, GlobalSettings } from '@/types/settings';
+
+export interface GlobalSettingsState {
+  serverConfig: GlobalServerConfig;
+  /**
+   * @localStorage
+   */
+  settings: GlobalSettings;
+}
+
+export const initialSettingsState: GlobalSettingsState = {
+  serverConfig: {},
+  settings: DEFAULT_SETTINGS,
+};

package/src/store/global/{selectors/settings.test.ts → slices/settings/selectors.test.ts}

@@ -1,7 +1,7 @@
 import { LanguageModel } from '@/types/llm';
 
-import { GlobalStore } from '../store';
-import { settingsSelectors } from './settings';
+import { GlobalStore } from '../../store';
+import { settingsSelectors } from './selectors';
 
 describe('settingsSelectors', () => {
   describe('currentSettings', () => {

package/src/store/global/{selectors/settings.ts → slices/settings/selectors.ts}

@@ -12,7 +12,7 @@ import { CustomModels, GlobalSettings } from '@/types/settings';
 import { isOnServerSide } from '@/utils/env';
 import { merge } from '@/utils/merge';
 
-import { GlobalStore } from '../store';
+import { GlobalStore } from '../../store';
 
 const currentSettings = (s: GlobalStore) => merge(DEFAULT_SETTINGS, s.settings);
 
@@ -24,8 +24,7 @@ const defaultAgentConfig = (s: GlobalStore) => merge(DEFAULT_AGENT_CONFIG, defau
 
 const defaultAgentMeta = (s: GlobalStore) => merge(DEFAULT_AGENT_META, defaultAgent(s).meta);
 
-const openAIAPIKeySelectors = (s: GlobalStore) =>
-  s.settings.languageModel.openAI.OPENAI_API_KEY || s.settings.OPENAI_API_KEY;
+const openAIAPIKeySelectors = (s: GlobalStore) => s.settings.languageModel.openAI.OPENAI_API_KEY;
 
 const openAIProxyUrlSelectors = (s: GlobalStore) => s.settings.languageModel.openAI.endpoint;
 
@@ -68,7 +67,7 @@ const modelListSelectors = (s: GlobalStore) => {
 
 export const exportSettings = (s: GlobalStore) => {
   // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  const { OPENAI_API_KEY: _, password: __, ...settings } = s.settings;
+  const { password: _, ...settings } = s.settings;
 
   return settings as GlobalSettings;
 };

package/src/store/global/store.ts

@@ -9,18 +9,16 @@ import { isDev } from '@/utils/env';
 
 import { createHyperStorage } from '../middleware/createHyperStorage';
 import { type GlobalState, initialState } from './initialState';
-import { type AgentAction, createAgentSlice } from './slices/agent';
-import { type CommonAction, createCommonSlice } from './slices/common';
-import { type SettingsAction, createSettingsSlice } from './slices/settings';
+import { type CommonAction, createCommonSlice } from './slices/common/action';
+import { type SettingsAction, createSettingsSlice } from './slices/settings/action';
 
 //  ===============  聚合 createStoreFn ============ //
 
-export type GlobalStore = CommonAction & GlobalState & AgentAction & SettingsAction;
+export type GlobalStore = CommonAction & GlobalState & SettingsAction;
 
 const createStore: StateCreator<GlobalStore, [['zustand/devtools', never]]> = (...parameters) => ({
   ...initialState,
   ...createCommonSlice(...parameters),
-  ...createAgentSlice(...parameters),
   ...createSettingsSlice(...parameters),
 });
 
@@ -38,24 +36,12 @@ const persistOptions: PersistOptions<GlobalStore, GlobalPersist> = {
         if (!draft.defaultAgent) {
           draft.defaultAgent = DEFAULT_AGENT;
         }
-        delete draft.enableMaxTokens;
-        delete draft.enableHistoryCount;
-        delete draft.historyCount;
-        delete draft.enableCompressThreshold;
-        delete draft.compressThreshold;
 
         // migration to new data model
         if (!draft.languageModel) {
           draft.languageModel = {
-            openAI: {
-              ...DEFAULT_LLM_CONFIG.openAI,
-              OPENAI_API_KEY: draft.OPENAI_API_KEY || DEFAULT_LLM_CONFIG.openAI.OPENAI_API_KEY,
-              endpoint: draft.endpoint || DEFAULT_LLM_CONFIG.openAI.OPENAI_API_KEY,
-            },
+            openAI: DEFAULT_LLM_CONFIG.openAI,
           };
-
-          delete draft.OPENAI_API_KEY;
-          delete draft.endpoint;
         }
       }),
     };

package/src/store/session/slices/agent/selectors.ts

@@ -4,7 +4,8 @@ import { t } from 'i18next';
 import { DEFAULT_OPENAI_MODEL_LIST, VISION_MODEL_WHITE_LIST } from '@/const/llm';
 import { DEFAULT_AVATAR, DEFAULT_BACKGROUND_COLOR, DEFAULT_INBOX_AVATAR } from '@/const/meta';
 import { DEFAULT_AGENT_CONFIG, DEFAUTT_AGENT_TTS_CONFIG } from '@/const/settings';
-import { settingsSelectors, useGlobalStore } from '@/store/global';
+import { useGlobalStore } from '@/store/global';
+import { settingsSelectors } from '@/store/global/selectors';
 import { SessionStore } from '@/store/session';
 import { LobeAgentTTSConfig } from '@/types/agent';
 import { LanguageModel } from '@/types/llm';

package/src/store/session/slices/session/action.ts

@@ -5,7 +5,8 @@ import { StateCreator } from 'zustand/vanilla';
 import { INBOX_SESSION_ID } from '@/const/session';
 import { SESSION_CHAT_URL } from '@/const/url';
 import { sessionService } from '@/services/session';
-import { settingsSelectors, useGlobalStore } from '@/store/global';
+import { useGlobalStore } from '@/store/global';
+import { settingsSelectors } from '@/store/global/selectors';
 import { SessionStore } from '@/store/session';
 import {
   LobeAgentSession,

package/src/tools/dalle/Render/ToolBar.tsx

@@ -5,7 +5,8 @@ import { Flexbox } from 'react-layout-kit';
 
 import { useChatStore } from '@/store/chat';
 import { chatEnhanceSelectors } from '@/store/chat/selectors';
-import { settingsSelectors, useGlobalStore } from '@/store/global';
+import { useGlobalStore } from '@/store/global';
+import { settingsSelectors } from '@/store/global/selectors';
 import { DallEImageItem } from '@/types/tool/dalle';
 
 interface ToolBarProps {

package/src/types/exportConfig.ts

@@ -1,4 +1,4 @@
-import { ChatMessage } from '@/types/chatMessage';
+import { ChatMessage } from '@/types/message';
 import { LobeSessions } from '@/types/session';
 import { GlobalSettings } from '@/types/settings';
 import { ChatTopic } from '@/types/topic';

package/src/types/{chatMessage.ts → message/index.ts}

@@ -1,8 +1,8 @@
 import { ErrorType } from '@/types/fetch';
 import { Translate } from '@/types/translate';
 
-import { LLMRoleType } from './llm';
-import { BaseDataModel } from './meta';
+import { LLMRoleType } from '../llm';
+import { BaseDataModel } from '../meta';
 
 /**
  * 聊天消息错误对象