@lobehub/chat 1.69.2 → 1.69.4

package/CHANGELOG.md

@@ -2,6 +2,56 @@
 
 # Changelog
 
+### [Version 1.69.4](https://github.com/lobehub/lobe-chat/compare/v1.69.3...v1.69.4)
+
+<sup>Released on **2025-03-09**</sup>
+
+#### 🐛 Bug Fixes
+
+- **misc**: Fix mistral can not chat.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### What's fixed
+
+- **misc**: Fix mistral can not chat, closes [#6828](https://github.com/lobehub/lobe-chat/issues/6828) ([00cba71](https://github.com/lobehub/lobe-chat/commit/00cba71))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
+### [Version 1.69.3](https://github.com/lobehub/lobe-chat/compare/v1.69.2...v1.69.3)
+
+<sup>Released on **2025-03-08**</sup>
+
+#### 💄 Styles
+
+- **misc**: Add login ui for next-auth.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### Styles
+
+- **misc**: Add login ui for next-auth, closes [#6434](https://github.com/lobehub/lobe-chat/issues/6434) ([541f275](https://github.com/lobehub/lobe-chat/commit/541f275))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
 ### [Version 1.69.2](https://github.com/lobehub/lobe-chat/compare/v1.69.1...v1.69.2)
 
 <sup>Released on **2025-03-07**</sup>

package/changelog/v1.json

@@ -1,4 +1,22 @@
 [
+  {
+    "children": {
+      "fixes": [
+        "Fix mistral can not chat."
+      ]
+    },
+    "date": "2025-03-09",
+    "version": "1.69.4"
+  },
+  {
+    "children": {
+      "improvements": [
+        "Add login ui for next-auth."
+      ]
+    },
+    "date": "2025-03-08",
+    "version": "1.69.3"
+  },
   {
     "children": {
       "improvements": [

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "1.69.2",
+  "version": "1.69.4",
   "description": "Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.",
   "keywords": [
     "framework",
@@ -122,7 +122,7 @@
     "@cyntler/react-doc-viewer": "^1.17.0",
     "@electric-sql/pglite": "0.2.13",
     "@google-cloud/vertexai": "^1.9.2",
-    "@google/generative-ai": "^0.22.0",
+    "@google/generative-ai": "^0.24.0",
     "@huggingface/inference": "^2.8.1",
     "@icons-pack/react-simple-icons": "9.6.0",
     "@khmyznikov/pwa-install": "0.3.9",
@@ -165,7 +165,7 @@
     "epub2": "^3.0.2",
     "fast-deep-equal": "^3.1.3",
     "file-type": "^20.0.0",
-    "framer-motion": "^11.16.0",
+    "framer-motion": "^12.0.0",
     "gpt-tokenizer": "^2.8.1",
     "html-to-text": "^9.0.5",
     "i18next": "^24.2.1",

package/packages/web-crawler/package.json

@@ -8,6 +8,7 @@
     "@mozilla/readability": "^0.5.0",
     "happy-dom": "^17.0.0",
     "node-html-markdown": "^1.3.0",
-    "query-string": "^9.1.1"
+    "query-string": "^9.1.1",
+    "url-join": "^5"
   }
 }

package/packages/web-crawler/src/crawImpl/browserless.ts

@@ -1,4 +1,5 @@
 import qs from 'query-string';
+import urlJoin from 'url-join';
 
 import { CrawlImpl, CrawlSuccessResult } from '../type';
 import { htmlToMarkdown } from '../utils/htmlToMarkdown';
@@ -25,7 +26,7 @@ export const browserless: CrawlImpl = async (url, { filterOptions }) => {
 
   try {
     const res = await fetch(
-      qs.stringifyUrl({ query: { token: BROWSERLESS_TOKEN }, url: `${BASE_URL}/content` }),
+      qs.stringifyUrl({ query: { token: BROWSERLESS_TOKEN }, url: urlJoin(BASE_URL, '/content') }),
       {
         body: JSON.stringify(input),
         headers: {

package/packages/web-crawler/src/urlRules.ts

@@ -61,4 +61,10 @@ export const crawUrlRules: CrawlUrlRule[] = [
     },
     urlPattern: 'https://www.qiumiwu.com/standings/(.*)',
   },
+
+  // mozilla use jina
+  {
+    impls: ['jina'],
+    urlPattern: 'https://developer.mozilla.org(.*)',
+  },
 ];

package/src/app/[variants]/(auth)/next-auth/signin/AuthSignInBox.tsx

@@ -0,0 +1,161 @@
+'use client';
+
+import { LobeChat } from '@lobehub/ui/brand';
+import { Button, Col, Flex, Row, Skeleton, Typography } from 'antd';
+import { createStyles } from 'antd-style';
+import { AuthError } from 'next-auth';
+import { signIn } from 'next-auth/react';
+import { useRouter, useSearchParams } from 'next/navigation';
+import { memo } from 'react';
+import { useTranslation } from 'react-i18next';
+
+import BrandWatermark from '@/components/BrandWatermark';
+import AuthIcons from '@/components/NextAuth/AuthIcons';
+import { DOCUMENTS_REFER_URL, PRIVACY_URL, TERMS_URL } from '@/const/url';
+import { useUserStore } from '@/store/user';
+
+const { Title, Paragraph } = Typography;
+
+const useStyles = createStyles(({ css, token }) => ({
+  button: css`
+    text-transform: capitalize;
+  `,
+  container: css`
+    min-width: 360px;
+    border: 1px solid ${token.colorBorder};
+    border-radius: ${token.borderRadiusLG}px;
+    background: ${token.colorBgContainer};
+  `,
+  contentCard: css`
+    padding-block: 2.5rem;
+    padding-inline: 2rem;
+  `,
+  description: css`
+    margin: 0;
+    color: ${token.colorTextSecondary};
+  `,
+  footer: css`
+    padding: 1rem;
+    border-block-start: 1px solid ${token.colorBorder};
+    border-radius: 0 0 8px 8px;
+
+    color: ${token.colorTextDescription};
+
+    background: ${token.colorBgElevated};
+  `,
+  text: css`
+    text-align: center;
+  `,
+  title: css`
+    margin: 0;
+    color: ${token.colorTextHeading};
+  `,
+}));
+
+const BtnListLoading = memo(() => {
+  return (
+    <Flex gap={'small'} vertical>
+      <Skeleton.Button active style={{ minWidth: 300 }} />
+      <Skeleton.Button active style={{ minWidth: 300 }} />
+      <Skeleton.Button active style={{ minWidth: 300 }} />
+    </Flex>
+  );
+});
+
+/**
+ * Follow the implementation from AuthJS official documentation,
+ * but using client components.
+ * ref: https://authjs.dev/guides/pages/signin
+ */
+export default memo(() => {
+  const { styles } = useStyles();
+  const { t } = useTranslation('clerk');
+  const router = useRouter();
+
+  const oAuthSSOProviders = useUserStore((s) => s.oAuthSSOProviders);
+
+  const searchParams = useSearchParams();
+
+  // Redirect back to the page url
+  const callbackUrl = searchParams.get('callbackUrl') ?? '';
+
+  const handleSignIn = async (provider: string) => {
+    try {
+      await signIn(provider, { redirectTo: callbackUrl });
+    } catch (error) {
+      // Signin can fail for a number of reasons, such as the user
+      // not existing, or the user not having the correct role.
+      // In some cases, you may want to redirect to a custom error
+      if (error instanceof AuthError) {
+        return router.push(`/next-auth/?error=${error.type}`);
+      }
+
+      // Otherwise if a redirects happens Next.js can handle it
+      // so you can just re-thrown the error and let Next.js handle it.
+      // Docs: https://nextjs.org/docs/app/api-reference/functions/redirect#server-component
+      throw error;
+    }
+  };
+
+  const footerBtns = [
+    { href: DOCUMENTS_REFER_URL, id: 0, label: t('footerPageLink__help') },
+    { href: PRIVACY_URL, id: 1, label: t('footerPageLink__privacy') },
+    { href: TERMS_URL, id: 2, label: t('footerPageLink__terms') },
+  ];
+
+  return (
+    <div className={styles.container}>
+      <div className={styles.contentCard}>
+        {/* Card Body */}
+        <Flex gap="large" vertical>
+          {/* Header */}
+          <div className={styles.text}>
+            <Title className={styles.title} level={4}>
+              <div>
+                <LobeChat size={48} />
+              </div>
+              {t('signIn.start.title', { applicationName: 'LobeChat' })}
+            </Title>
+            <Paragraph className={styles.description}>{t('signIn.start.subtitle')}</Paragraph>
+          </div>
+          {/* Content */}
+          <Flex gap="small" vertical>
+            {oAuthSSOProviders ? (
+              oAuthSSOProviders.map((provider) => (
+                <Button
+                  className={styles.button}
+                  icon={AuthIcons(provider, 16)}
+                  key={provider}
+                  onClick={() => handleSignIn(provider)}
+                >
+                  {provider}
+                </Button>
+              ))
+            ) : (
+              <BtnListLoading />
+            )}
+          </Flex>
+        </Flex>
+      </div>
+      <div className={styles.footer}>
+        {/* Footer */}
+        <Row>
+          <Col span={12}>
+            <Flex justify="left" style={{ height: '100%' }}>
+              <BrandWatermark />
+            </Flex>
+          </Col>
+          <Col offset={4} span={8}>
+            <Flex justify="right">
+              {footerBtns.map((btn) => (
+                <Button key={btn.id} onClick={() => router.push(btn.href)} size="small" type="text">
+                  {btn.label}
+                </Button>
+              ))}
+            </Flex>
+          </Col>
+        </Row>
+      </div>
+    </div>
+  );
+});

package/src/app/[variants]/(auth)/next-auth/signin/page.tsx

@@ -0,0 +1,11 @@
+import { Suspense } from 'react';
+
+import Loading from '@/components/Loading/BrandTextLoading';
+
+import AuthSignInBox from './AuthSignInBox';
+
+export default () => (
+  <Suspense fallback={<Loading />}>
+    <AuthSignInBox />
+  </Suspense>
+);

package/src/app/[variants]/(main)/profile/(home)/features/SSOProvidersList/index.tsx

@@ -10,7 +10,7 @@ import { userService } from '@/services/user';
 import { useUserStore } from '@/store/user';
 import { userProfileSelectors } from '@/store/user/selectors';
 
-import AuthIcons from './AuthIcons';
+import AuthIcons from '@/components/NextAuth/AuthIcons';
 
 const { Item } = List;
 

package/src/{app/[variants]/(main)/profile/(home)/features/SSOProvidersList → components/NextAuth}/AuthIcons.tsx

@@ -12,10 +12,6 @@ import {
 } from '@lobehub/ui/icons';
 import React from 'react';
 
-const iconProps = {
-  size: 32,
-};
-
 const iconComponents: { [key: string]: React.ElementType } = {
   'auth0': Auth0,
   'authelia': Authelia.Color,
@@ -29,9 +25,15 @@ const iconComponents: { [key: string]: React.ElementType } = {
   'zitadel': Zitadel.Color,
 };
 
-const AuthIcons = (id: string) => {
+/**
+ * Get the auth icons component for the given id
+ * @param id 
+ * @param size default is 36
+ * @returns 
+ */
+const AuthIcons = (id: string, size = 36) => {
   const IconComponent = iconComponents[id] || iconComponents.default;
-  return <IconComponent {...iconProps} />;
+  return <IconComponent size={size}/>;
 };
 
 export default AuthIcons;

package/src/features/Conversation/Extras/Usage/UsageDetail/ModelCard.tsx

@@ -112,12 +112,6 @@ const ModelCard = memo<ModelCardProps>(({ pricing, id, provider, displayName })
                 </Flexbox>
               </Tooltip>
             )}
-            <Tooltip title={t('messages.modelCard.pricing.inputTokens', { amount: inputPrice })}>
-              <Flexbox gap={2} horizontal>
-                <Icon icon={ArrowUpFromDot} />
-                {inputPrice}
-              </Flexbox>
-            </Tooltip>
             {pricing?.writeCacheInput && (
               <Tooltip
                 title={t('messages.modelCard.pricing.writeCacheInputTokens', {
@@ -130,6 +124,12 @@ const ModelCard = memo<ModelCardProps>(({ pricing, id, provider, displayName })
                 </Flexbox>
               </Tooltip>
             )}
+            <Tooltip title={t('messages.modelCard.pricing.inputTokens', { amount: inputPrice })}>
+              <Flexbox gap={2} horizontal>
+                <Icon icon={ArrowUpFromDot} />
+                {inputPrice}
+              </Flexbox>
+            </Tooltip>
             <Tooltip title={t('messages.modelCard.pricing.outputTokens', { amount: outputPrice })}>
               <Flexbox gap={2} horizontal>
                 <Icon icon={ArrowDownToDot} />

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

@@ -138,7 +138,7 @@ const TokenDetail = memo<TokenDetailProps>(({ usage, model, provider }) => {
               </Flexbox>
             )}
             {outputDetails.length > 1 && (
-              <>
+              <Flexbox gap={4}>
                 <Flexbox
                   align={'center'}
                   gap={4}
@@ -146,12 +146,12 @@ const TokenDetail = memo<TokenDetailProps>(({ usage, model, provider }) => {
                   justify={'space-between'}
                   width={'100%'}
                 >
-                  <div style={{ color: theme.colorTextDescription }}>
+                  <div style={{ color: theme.colorTextDescription, fontSize: 12 }}>
                     {t('messages.tokenDetails.outputTitle')}
                   </div>
                 </Flexbox>
                 <TokenProgress data={outputDetails} showIcon />
-              </>
+              </Flexbox>
             )}
             <Flexbox>
               <TokenProgress data={totalDetail} showIcon />

package/src/libs/agent-runtime/mistral/index.test.ts

@@ -57,9 +57,6 @@ describe('specific LobeMistralAI tests', () => {
         messages: [{ content: 'Hello', role: 'user' }],
         model: 'open-mistral-7b',
         stream: true,
-        stream_options: {
-          include_usage: true,
-        },
         temperature: 0.35,
         top_p: 1,
       },

package/src/libs/agent-runtime/mistral/index.ts

@@ -1,8 +1,8 @@
+import type { ChatModelCard } from '@/types/llm';
+
 import { ModelProvider } from '../types';
 import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
 
-import type { ChatModelCard } from '@/types/llm';
-
 export interface MistralModelCard {
   capabilities: {
     function_calling: boolean;
@@ -16,6 +16,9 @@ export interface MistralModelCard {
 export const LobeMistralAI = LobeOpenAICompatibleFactory({
   baseURL: 'https://api.mistral.ai/v1',
   chatCompletion: {
+    // Mistral API does not support stream_options: { include_usage: true }
+    // refs: https://github.com/lobehub/lobe-chat/issues/6825
+    excludeUsage: true,
     handlePayload: (payload) => ({
       ...(payload.max_tokens !== undefined && { max_tokens: payload.max_tokens }),
       messages: payload.messages as any,
@@ -33,12 +36,14 @@ export const LobeMistralAI = LobeOpenAICompatibleFactory({
   models: async ({ client }) => {
     const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
 
-    const modelsPage = await client.models.list() as any;
+    const modelsPage = (await client.models.list()) as any;
     const modelList: MistralModelCard[] = modelsPage.data;
 
     return modelList
       .map((model) => {
-        const knownModel = LOBE_DEFAULT_MODEL_LIST.find((m) => model.id.toLowerCase() === m.id.toLowerCase());
+        const knownModel = LOBE_DEFAULT_MODEL_LIST.find(
+          (m) => model.id.toLowerCase() === m.id.toLowerCase(),
+        );
 
         return {
           contextWindowTokens: model.max_context_length,
@@ -47,9 +52,7 @@ export const LobeMistralAI = LobeOpenAICompatibleFactory({
           enabled: knownModel?.enabled || false,
           functionCall: model.capabilities.function_calling,
           id: model.id,
-          reasoning:
-            knownModel?.abilities?.reasoning
-            || false,
+          reasoning: knownModel?.abilities?.reasoning || false,
           vision: model.capabilities.vision,
         };
       })

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

@@ -35,7 +35,7 @@ export const debugStream = async (stream: ReadableStream) => {
 
       console.log(`[chunk ${chunk}] ${getTime()}`);
       console.log(chunkValue);
-      console.log(`\n`);
+      console.log('');
 
       finished = done;
       chunk++;

package/src/libs/agent-runtime/utils/openaiCompatibleFactory/index.ts

@@ -57,6 +57,7 @@ interface OpenAICompatibleFactoryOptions<T extends Record<string, any> = any> {
   apiKey?: string;
   baseURL?: string;
   chatCompletion?: {
+    excludeUsage?: boolean;
     handleError?: (
       error: any,
       options: ConstructorOptions<T>,
@@ -224,12 +225,17 @@ export const LobeOpenAICompatibleFactory = <T extends Record<string, any> = any>
             ...postPayload,
             messages,
             ...(chatCompletion?.noUserId ? {} : { user: options?.user }),
-            stream_options: postPayload.stream ? { include_usage: true } : undefined,
+            stream_options:
+              postPayload.stream && !chatCompletion?.excludeUsage
+                ? { include_usage: true }
+                : undefined,
           };
 
           if (debug?.chatCompletion?.()) {
-            console.log('[requestPayload]:', JSON.stringify(finalPayload, null, 2));
+            console.log('[requestPayload]');
+            console.log(JSON.stringify(finalPayload), '\n');
           }
+
           response = await this.client.chat.completions.create(finalPayload, {
             // https://github.com/lobehub/lobe-chat/pull/318
             headers: { Accept: '*/*', ...options?.requestHeaders },

package/src/libs/next-auth/auth.config.ts

@@ -42,6 +42,7 @@ export default {
   debug: authEnv.NEXT_AUTH_DEBUG,
   pages: {
     error: '/next-auth/error',
+    signIn: '/next-auth/signin',
   },
   providers: initSSOProviders(),
   secret: authEnv.NEXT_AUTH_SECRET,

package/src/middleware.ts

@@ -36,7 +36,7 @@ export const config = {
 
     '/login(.*)',
     '/signup(.*)',
-    '/next-auth/error',
+    '/next-auth/(.*)',
     // ↓ cloud ↓
   ],
 };

package/src/services/chat.ts

@@ -17,13 +17,19 @@ import {
 } from '@/libs/agent-runtime';
 import { filesPrompts } from '@/prompts/files';
 import { BuiltinSystemRolePrompts } from '@/prompts/systemRole';
-import { aiModelSelectors, aiProviderSelectors, useAiInfraStore } from '@/store/aiInfra';
-import { getAgentChatConfig } from '@/store/chat/slices/aiChat/actions/helpers';
-import { useSessionStore } from '@/store/session';
+import { getAgentStoreState } from '@/store/agent';
+import { agentChatConfigSelectors } from '@/store/agent/selectors';
+import {
+  aiModelSelectors,
+  aiProviderSelectors,
+  getAiInfraStoreState,
+  useAiInfraStore,
+} from '@/store/aiInfra';
+import { getSessionStoreState } from '@/store/session';
 import { sessionMetaSelectors } from '@/store/session/selectors';
-import { useToolStore } from '@/store/tool';
+import { getToolStoreState } from '@/store/tool';
 import { pluginSelectors, toolSelectors } from '@/store/tool/selectors';
-import { useUserStore } from '@/store/user';
+import { getUserStoreState, useUserStore } from '@/store/user';
 import {
   modelConfigSelectors,
   modelProviderSelectors,
@@ -46,10 +52,10 @@ import { API_ENDPOINTS } from './_url';
 const isCanUseFC = (model: string, provider: string) => {
   // TODO: remove isDeprecatedEdition condition in V2.0
   if (isDeprecatedEdition) {
-    return modelProviderSelectors.isModelEnabledFunctionCall(model)(useUserStore.getState());
+    return modelProviderSelectors.isModelEnabledFunctionCall(model)(getUserStoreState());
   }
 
-  return aiModelSelectors.isModelSupportToolUse(model, provider)(useAiInfraStore.getState());
+  return aiModelSelectors.isModelSupportToolUse(model, provider)(getAiInfraStoreState());
 };
 
 /**
@@ -169,7 +175,7 @@ class ChatService {
     );
 
     // =================== 0. process search =================== //
-    const chatConfig = getAgentChatConfig();
+    const chatConfig = agentChatConfigSelectors.currentChatConfig(getAgentStoreState());
 
     const enabledSearch = chatConfig.searchMode !== 'off';
     const isModelHasBuiltinSearch = aiModelSelectors.isModelHasBuiltinSearch(
@@ -200,7 +206,7 @@ class ChatService {
 
     // ============  2. preprocess tools   ============ //
 
-    let filterTools = toolSelectors.enabledSchema(pluginIds)(useToolStore.getState());
+    let filterTools = toolSelectors.enabledSchema(pluginIds)(getToolStoreState());
 
     // check this model can use function call
     const canUseFC = isCanUseFC(payload.model, payload.provider!);
@@ -378,7 +384,7 @@ class ChatService {
    * @param options
    */
   runPluginApi = async (params: PluginRequestPayload, options?: FetchOptions) => {
-    const s = useToolStore.getState();
+    const s = getToolStoreState();
 
     const settings = pluginSelectors.getPluginSettingsById(params.identifier)(s);
     const manifest = pluginSelectors.getToolManifestById(params.identifier)(s);
@@ -537,7 +543,7 @@ class ChatService {
       const hasTools = tools && tools?.length > 0;
       const hasFC = hasTools && isCanUseFC(model, provider);
       const toolsSystemRoles =
-        hasFC && toolSelectors.enabledSystemRoles(tools)(useToolStore.getState());
+        hasFC && toolSelectors.enabledSystemRoles(tools)(getToolStoreState());
 
       const injectSystemRoles = BuiltinSystemRolePrompts({
         historySummary: options?.historySummary,
@@ -565,9 +571,9 @@ class ChatService {
   };
 
   private mapTrace = (trace?: TracePayload, tag?: TraceTagMap): TracePayload => {
-    const tags = sessionMetaSelectors.currentAgentMeta(useSessionStore.getState()).tags || [];
+    const tags = sessionMetaSelectors.currentAgentMeta(getSessionStoreState()).tags || [];
 
-    const enabled = preferenceSelectors.userAllowTrace(useUserStore.getState());
+    const enabled = preferenceSelectors.userAllowTrace(getUserStoreState());
 
     if (!enabled) return { ...trace, enabled: false };
 

package/src/store/agent/index.ts

@@ -1,2 +1,2 @@
 export type { AgentStore } from './store';
-export { useAgentStore } from './store';
+export { getAgentStoreState, useAgentStore } from './store';

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

@@ -1,4 +1,5 @@
 import { contextCachingModels, thinkingWithToolClaudeModels } from '@/const/models';
+import { DEFAULT_AGENT_CHAT_CONFIG } from '@/const/settings';
 import { AgentStoreState } from '@/store/agent/initialState';
 import { LobeAgentChatConfig } from '@/types/agent';
 
@@ -30,10 +31,10 @@ const enableHistoryCount = (s: AgentStoreState) => {
   return chatConfig.enableHistoryCount;
 };
 
-const historyCount = (s: AgentStoreState) => {
+const historyCount = (s: AgentStoreState): number => {
   const chatConfig = currentAgentChatConfig(s);
 
-  return chatConfig.historyCount;
+  return chatConfig.historyCount || (DEFAULT_AGENT_CHAT_CONFIG.historyCount as number);
 };
 
 const displayMode = (s: AgentStoreState) => {

package/src/store/agent/store.ts

@@ -20,3 +20,5 @@ const createStore: StateCreator<AgentStore, [['zustand/devtools', never]]> = (..
 const devtools = createDevtools('agent');
 
 export const useAgentStore = createWithEqualityFn<AgentStore>()(devtools(createStore), shallow);
+
+export const getAgentStoreState = () => useAgentStore.getState();

package/src/store/aiInfra/index.ts

@@ -1,2 +1,2 @@
 export * from './selectors';
-export { useAiInfraStore } from './store';
+export { getAiInfraStoreState,useAiInfraStore } from './store';

package/src/store/aiInfra/store.ts

@@ -23,3 +23,5 @@ const createStore: StateCreator<AiInfraStore, [['zustand/devtools', never]]> = (
 const devtools = createDevtools('aiInfra');
 
 export const useAiInfraStore = createWithEqualityFn<AiInfraStore>()(devtools(createStore), shallow);
+
+export const getAiInfraStoreState = () => useAiInfraStore.getState();

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

@@ -5,13 +5,14 @@ import { template } from 'lodash-es';
 import { StateCreator } from 'zustand/vanilla';
 
 import { LOADING_FLAT, MESSAGE_CANCEL_FLAT } from '@/const/message';
-import { DEFAULT_AGENT_CHAT_CONFIG } from '@/const/settings';
 import { TraceEventType, TraceNameMap } from '@/const/trace';
 import { isServerMode } from '@/const/version';
 import { knowledgeBaseQAPrompts } from '@/prompts/knowledgeBaseQA';
 import { chatService } from '@/services/chat';
 import { messageService } from '@/services/message';
 import { useAgentStore } from '@/store/agent';
+import { agentChatConfigSelectors, agentSelectors } from '@/store/agent/selectors';
+import { getAgentStoreState } from '@/store/agent/store';
 import { chatHelpers } from '@/store/chat/helpers';
 import { ChatStore } from '@/store/chat/store';
 import { messageMapKey } from '@/store/chat/utils/messageMapKey';
@@ -21,12 +22,6 @@ import { MessageSemanticSearchChunk } from '@/types/rag';
 import { setNamespace } from '@/utils/storeDebug';
 
 import { chatSelectors, topicSelectors } from '../../../selectors';
-import {
-  getAgentChatConfig,
-  getAgentConfig,
-  getAgentEnableHistoryCount,
-  getAgentKnowledge,
-} from './helpers';
 
 const n = setNamespace('ai');
 
@@ -163,7 +158,7 @@ export const generateAIChat: StateCreator<
       threadId: activeThreadId,
     };
 
-    const agentConfig = getAgentChatConfig();
+    const agentConfig = agentChatConfigSelectors.currentChatConfig(getAgentStoreState());
 
     let tempMessageId: string | undefined = undefined;
     let newTopicId: string | undefined = undefined;
@@ -288,7 +283,7 @@ export const generateAIChat: StateCreator<
     // create a new array to avoid the original messages array change
     const messages = [...originalMessages];
 
-    const { model, provider, chatConfig } = getAgentConfig();
+    const { model, provider, chatConfig } = agentSelectors.currentAgentConfig(getAgentStoreState());
 
     let fileChunks: MessageSemanticSearchChunk[] | undefined;
     let ragQueryId;
@@ -312,7 +307,7 @@ export const generateAIChat: StateCreator<
         chunks,
         userQuery: lastMsg.content,
         rewriteQuery,
-        knowledge: getAgentKnowledge(),
+        knowledge: agentSelectors.currentEnabledKnowledge(getAgentStoreState()),
       });
 
       // 3. add the retrieve context messages to the messages history
@@ -354,11 +349,10 @@ export const generateAIChat: StateCreator<
     }
 
     // 5. summary history if context messages is larger than historyCount
-    const historyCount =
-      chatConfig.historyCount || (DEFAULT_AGENT_CHAT_CONFIG.historyCount as number);
+    const historyCount = agentChatConfigSelectors.historyCount(getAgentStoreState());
 
     if (
-      chatConfig.enableHistoryCount &&
+      agentChatConfigSelectors.enableHistoryCount(getAgentStoreState()) &&
       chatConfig.enableCompressHistory &&
       originalMessages.length > historyCount
     ) {
@@ -387,7 +381,7 @@ export const generateAIChat: StateCreator<
       n('generateMessage(start)', { messageId, messages }) as string,
     );
 
-    const agentConfig = getAgentConfig();
+    const agentConfig = agentSelectors.currentAgentConfig(getAgentStoreState());
     const chatConfig = agentConfig.chatConfig;
 
     const compiler = template(chatConfig.inputTemplate, { interpolate: /{{([\S\s]+?)}}/g });
@@ -397,10 +391,13 @@ export const generateAIChat: StateCreator<
     // ================================== //
 
     // 1. slice messages with config
+    const historyCount = agentChatConfigSelectors.historyCount(getAgentStoreState());
+    const enableHistoryCount = agentChatConfigSelectors.enableHistoryCount(getAgentStoreState());
+
     let preprocessMsgs = chatHelpers.getSlicedMessages(messages, {
       includeNewUserMessage: true,
-      enableHistoryCount: getAgentEnableHistoryCount(),
-      historyCount: chatConfig.historyCount,
+      enableHistoryCount,
+      historyCount,
     });
 
     // 2. replace inputMessage template

package/src/store/session/index.ts

@@ -1,2 +1,2 @@
 export type { SessionStore } from './store';
-export { useSessionStore } from './store';
+export { getSessionStoreState,useSessionStore } from './store';

package/src/store/session/store.ts

@@ -31,3 +31,5 @@ export const useSessionStore = createWithEqualityFn<SessionStore>()(
   ),
   shallow,
 );
+
+export const getSessionStoreState = () => useSessionStore.getState();

package/src/store/tool/index.ts

@@ -1,2 +1,2 @@
 export * from './helpers';
-export { useToolStore } from './store';
+export { getToolStoreState, useToolStore } from './store';

package/src/store/tool/store.ts

@@ -30,3 +30,5 @@ const createStore: StateCreator<ToolStore, [['zustand/devtools', never]]> = (...
 const devtools = createDevtools('tools');
 
 export const useToolStore = createWithEqualityFn<ToolStore>()(devtools(createStore), shallow);
+
+export const getToolStoreState = () => useToolStore.getState();

package/src/store/user/store.ts

@@ -40,3 +40,5 @@ export const useUserStore = createWithEqualityFn<UserStore>()(
   subscribeWithSelector(devtools(createStore)),
   shallow,
 );
+
+export const getUserStoreState = () => useUserStore.getState();

package/src/types/agent/chatConfig.ts

@@ -0,0 +1,66 @@
+/* eslint-disable sort-keys-fix/sort-keys-fix, typescript-sort-keys/interface */
+import { z } from 'zod';
+
+import { SearchMode } from '@/types/search';
+
+export interface WorkingModel {
+  model: string;
+  provider: string;
+}
+
+export interface LobeAgentChatConfig {
+  displayMode?: 'chat' | 'docs';
+
+  enableAutoCreateTopic?: boolean;
+  autoCreateTopicThreshold: number;
+
+  enableMaxTokens?: boolean;
+
+  /**
+   * 是否开启推理
+   */
+  enableReasoning?: boolean;
+  /**
+   * 自定义推理强度
+   */
+  enableReasoningEffort?: boolean;
+  reasoningBudgetToken?: number;
+
+  /**
+   * 禁用上下文缓存
+   */
+  disableContextCaching?: boolean;
+  /**
+   * 历史消息条数
+   */
+  historyCount?: number;
+  /**
+   * 开启历史记录条数
+   */
+  enableHistoryCount?: boolean;
+  /**
+   * 历史消息长度压缩阈值
+   */
+  enableCompressHistory?: boolean;
+
+  inputTemplate?: string;
+
+  searchMode?: SearchMode;
+  searchFCModel?: WorkingModel;
+  useModelBuiltinSearch?: boolean;
+}
+/* eslint-enable */
+
+export const AgentChatConfigSchema = z.object({
+  autoCreateTopicThreshold: z.number().default(2),
+  displayMode: z.enum(['chat', 'docs']).optional(),
+  enableAutoCreateTopic: z.boolean().optional(),
+  enableCompressHistory: z.boolean().optional(),
+  enableHistoryCount: z.boolean().optional(),
+  enableMaxTokens: z.boolean().optional(),
+  enableReasoning: z.boolean().optional(),
+  enableReasoningEffort: z.boolean().optional(),
+  historyCount: z.number().optional(),
+  reasoningBudgetToken: z.number().optional(),
+  searchMode: z.enum(['off', 'on', 'auto']).optional(),
+});

package/src/types/agent/index.ts

@@ -1,9 +1,8 @@
-import { z } from 'zod';
-
 import { FileItem } from '@/types/files';
 import { KnowledgeBaseItem } from '@/types/knowledgeBase';
 import { FewShots, LLMParams } from '@/types/llm';
-import { SearchMode } from '@/types/search';
+
+import { LobeAgentChatConfig } from './chatConfig';
 
 export type TTSServer = 'openai' | 'edge' | 'microsoft';
 
@@ -55,64 +54,8 @@ export interface LobeAgentConfig {
   tts: LobeAgentTTSConfig;
 }
 
-/* eslint-disable sort-keys-fix/sort-keys-fix, typescript-sort-keys/interface */
-
-export interface LobeAgentChatConfig {
-  displayMode?: 'chat' | 'docs';
-
-  enableAutoCreateTopic?: boolean;
-  autoCreateTopicThreshold: number;
-
-  enableMaxTokens?: boolean;
-
-  /**
-   * 是否开启推理
-   */
-  enableReasoning?: boolean;
-  /**
-   * 自定义推理强度
-   */
-  enableReasoningEffort?: boolean;
-  reasoningBudgetToken?: number;
-
-  /**
-   * 禁用上下文缓存
-   */
-  disableContextCaching?: boolean;
-  /**
-   * 历史消息条数
-   */
-  historyCount?: number;
-  /**
-   * 开启历史记录条数
-   */
-  enableHistoryCount?: boolean;
-  /**
-   * 历史消息长度压缩阈值
-   */
-  enableCompressHistory?: boolean;
-
-  inputTemplate?: string;
-
-  searchMode?: SearchMode;
-  useModelBuiltinSearch?: boolean;
-}
-/* eslint-enable */
-
-export const AgentChatConfigSchema = z.object({
-  autoCreateTopicThreshold: z.number().default(2),
-  displayMode: z.enum(['chat', 'docs']).optional(),
-  enableAutoCreateTopic: z.boolean().optional(),
-  enableCompressHistory: z.boolean().optional(),
-  enableHistoryCount: z.boolean().optional(),
-  enableMaxTokens: z.boolean().optional(),
-  enableReasoning: z.boolean().optional(),
-  enableReasoningEffort: z.boolean().optional(),
-  historyCount: z.number().optional(),
-  reasoningBudgetToken: z.number().optional(),
-  searchMode: z.enum(['off', 'on', 'auto']).optional(),
-});
-
 export type LobeAgentConfigKeys =
   | keyof LobeAgentConfig
   | ['params', keyof LobeAgentConfig['params']];
+
+export * from './chatConfig';

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

@@ -1,13 +0,0 @@
-import { useAgentStore } from '@/store/agent';
-import { agentChatConfigSelectors, agentSelectors } from '@/store/agent/selectors';
-
-export const getAgentConfig = () => agentSelectors.currentAgentConfig(useAgentStore.getState());
-
-export const getAgentChatConfig = () =>
-  agentChatConfigSelectors.currentChatConfig(useAgentStore.getState());
-
-export const getAgentEnableHistoryCount = () =>
-  agentChatConfigSelectors.enableHistoryCount(useAgentStore.getState());
-
-export const getAgentKnowledge = () =>
-  agentSelectors.currentEnabledKnowledge(useAgentStore.getState());