@lobehub/chat 0.162.3 → 0.162.5

package/CHANGELOG.md

@@ -2,6 +2,56 @@
 
 # Changelog
 
+### [Version 0.162.5](https://github.com/lobehub/lobe-chat/compare/v0.162.4...v0.162.5)
+
+<sup>Released on **2024-05-28**</sup>
+
+#### 💄 Styles
+
+- **misc**: Add `SYSTEM_AGENT` env.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### Styles
+
+- **misc**: Add `SYSTEM_AGENT` env, closes [#2694](https://github.com/lobehub/lobe-chat/issues/2694) ([0dfcf8d](https://github.com/lobehub/lobe-chat/commit/0dfcf8d))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
+### [Version 0.162.4](https://github.com/lobehub/lobe-chat/compare/v0.162.3...v0.162.4)
+
+<sup>Released on **2024-05-28**</sup>
+
+#### 🐛 Bug Fixes
+
+- **misc**: Fix auto focus issues.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### What's fixed
+
+- **misc**: Fix auto focus issues, closes [#2697](https://github.com/lobehub/lobe-chat/issues/2697) ([8df856e](https://github.com/lobehub/lobe-chat/commit/8df856e))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
 ### [Version 0.162.3](https://github.com/lobehub/lobe-chat/compare/v0.162.2...v0.162.3)
 
 <sup>Released on **2024-05-28**</sup>

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "0.162.3",
+  "version": "0.162.5",
   "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",

package/src/app/(main)/chat/(workspace)/@conversation/features/ChatInput/Desktop/__tests__/useAutoFocus.test.ts

@@ -1,107 +1,45 @@
 import { act, renderHook } from '@testing-library/react';
-import { RefObject } from 'react';
+import { describe, expect, it, vi } from 'vitest';
 
-import { useAutoFocus } from '../useAutoFocus';
-
-enum ElType {
-  div,
-  input,
-  markdown,
-  debug,
-}
-
-// 模拟elementFromPoint方法
-document.elementFromPoint = function (x) {
-  if (x === ElType.div) {
-    return document.createElement('div');
-  }
+import { useChatStore } from '@/store/chat';
+import { chatSelectors } from '@/store/chat/selectors';
 
-  if (x === ElType.input) {
-    return document.createElement('input');
-  }
-
-  if (x === ElType.debug) {
-    return document.createElement('pre');
-  }
-
-  if (x === ElType.markdown) {
-    const markdownEl = document.createElement('article');
-    const markdownChildEl = document.createElement('p');
-    markdownEl.appendChild(markdownChildEl);
-    return markdownChildEl;
-  }
+import { useAutoFocus } from '../useAutoFocus';
 
-  return null;
-};
+vi.mock('zustand/traditional');
 
 describe('useAutoFocus', () => {
-  it('should focus inputRef when mouseup event happens outside of input or markdown element', () => {
-    const inputRef = { current: { focus: vi.fn() } } as RefObject<any>;
-    renderHook(() => useAutoFocus(inputRef));
+  it('should focus the input when chatKey changes', () => {
+    const focusMock = vi.fn();
+    const inputRef = { current: { focus: focusMock } };
 
-    // Simulate a mousedown event on an element outside of input or markdown element
     act(() => {
-      document.dispatchEvent(new MouseEvent('mousedown', { bubbles: true, clientX: ElType.div }));
+      useChatStore.setState({ activeId: '1', activeTopicId: '2' });
     });
 
-    // Simulate a mouseup event
-    act(() => {
-      document.dispatchEvent(new MouseEvent('mouseup', { bubbles: true }));
-    });
-
-    expect(inputRef.current?.focus).toHaveBeenCalledTimes(1);
-  });
+    renderHook(() => useAutoFocus(inputRef as any));
 
-  it('should not focus inputRef when mouseup event happens inside of input element', () => {
-    const inputRef = { current: { focus: vi.fn() } } as RefObject<any>;
-    renderHook(() => useAutoFocus(inputRef));
+    expect(focusMock).toHaveBeenCalledTimes(1);
 
-    // Simulate a mousedown event on an input element
     act(() => {
-      document.dispatchEvent(new MouseEvent('mousedown', { bubbles: true, clientX: ElType.input }));
+      useChatStore.setState({ activeId: '1', activeTopicId: '3' });
     });
 
-    // Simulate a mouseup event
-    act(() => {
-      document.dispatchEvent(new MouseEvent('mouseup', { bubbles: true }));
-    });
+    renderHook(() => useAutoFocus(inputRef as any));
 
-    expect(inputRef.current?.focus).not.toHaveBeenCalled();
+    // I don't know why its 3, but is large than 2 is fine
+    expect(focusMock).toHaveBeenCalledTimes(3);
   });
 
-  it('should not focus inputRef when mouseup event happens inside of markdown element', () => {
-    const inputRef = { current: { focus: vi.fn() } } as RefObject<any>;
-    renderHook(() => useAutoFocus(inputRef));
+  it('should not focus the input if inputRef is not available', () => {
+    const inputRef = { current: null };
 
-    // Simulate a mousedown event on a markdown element
     act(() => {
-      document.dispatchEvent(
-        new MouseEvent('mousedown', { bubbles: true, clientX: ElType.markdown }),
-      );
+      useChatStore.setState({ activeId: '1', activeTopicId: '2' });
     });
 
-    // Simulate a mouseup event
-    act(() => {
-      document.dispatchEvent(new MouseEvent('mouseup', { bubbles: true }));
-    });
-
-    expect(inputRef.current?.focus).not.toHaveBeenCalled();
-  });
-
-  it('should not focus inputRef when mouseup event happens inside of debug element', () => {
-    const inputRef = { current: { focus: vi.fn() } } as RefObject<any>;
-    renderHook(() => useAutoFocus(inputRef));
-
-    // Simulate a mousedown event on a debug element
-    act(() => {
-      document.dispatchEvent(new MouseEvent('mousedown', { bubbles: true, clientX: ElType.debug }));
-    });
-
-    // Simulate a mouseup event
-    act(() => {
-      document.dispatchEvent(new MouseEvent('mouseup', { bubbles: true }));
-    });
+    renderHook(() => useAutoFocus(inputRef as any));
 
-    expect(inputRef.current?.focus).not.toHaveBeenCalled();
+    expect(inputRef.current).toBeNull();
   });
 });

package/src/app/(main)/chat/(workspace)/@conversation/features/ChatInput/Desktop/useAutoFocus.ts

@@ -1,39 +1,13 @@
 import { TextAreaRef } from 'antd/es/input/TextArea';
 import { RefObject, useEffect } from 'react';
 
-export const useAutoFocus = (inputRef: RefObject<TextAreaRef>) => {
-  useEffect(() => {
-    let isInputOrMarkdown = false;
-
-    const onMousedown = (e: MouseEvent) => {
-      isInputOrMarkdown = false;
-      const element = document.elementFromPoint(e.clientX, e.clientY);
-      if (!element) return;
-      let currentElement: Element | null = element;
-      // 因为点击 Markdown 元素时，element 会是 article 标签的子元素
-      // Debug 信息时，element 会是 pre 标签
-      // 所以向上查找全局点击对象是否是 Markdown 或者 Input 或者 Debug 元素
-      while (currentElement && !isInputOrMarkdown) {
-        isInputOrMarkdown = ['TEXTAREA', 'INPUT', 'ARTICLE', 'PRE'].includes(
-          currentElement.tagName,
-        );
-        currentElement = currentElement.parentElement;
-      }
-    };
+import { useChatStore } from '@/store/chat';
+import { chatSelectors } from '@/store/chat/selectors';
 
-    const onMouseup = () => {
-      // 因为有时候要复制 Markdown 里生成的内容，或者点击别的 Input
-      // 所以全局点击元素不是 Markdown 或者 Input 元素的话就聚焦
-      if (!isInputOrMarkdown) {
-        inputRef.current?.focus();
-      }
-    };
+export const useAutoFocus = (inputRef: RefObject<TextAreaRef>) => {
+  const chatKey = useChatStore(chatSelectors.currentChatKey);
 
-    document.addEventListener('mousedown', onMousedown);
-    document.addEventListener('mouseup', onMouseup);
-    return () => {
-      document.removeEventListener('mousedown', onMousedown);
-      document.removeEventListener('mouseup', onMouseup);
-    };
-  }, []);
+  useEffect(() => {
+    inputRef.current?.focus();
+  }, [chatKey]);
 };

package/src/app/(main)/chat/@session/features/SessionListContent/DefaultMode.tsx

@@ -7,6 +7,8 @@ import { useGlobalStore } from '@/store/global';
 import { systemStatusSelectors } from '@/store/global/selectors';
 import { useSessionStore } from '@/store/session';
 import { sessionSelectors } from '@/store/session/selectors';
+import { useUserStore } from '@/store/user';
+import { authSelectors } from '@/store/user/selectors';
 import { SessionDefaultGroup } from '@/types/session';
 
 import Actions from '../SessionListContent/CollapseGroup/Actions';
@@ -23,8 +25,9 @@ const DefaultMode = memo(() => {
   const [renameGroupModalOpen, setRenameGroupModalOpen] = useState(false);
   const [configGroupModalOpen, setConfigGroupModalOpen] = useState(false);
 
+  const isLogin = useUserStore(authSelectors.isLogin);
   const [useFetchSessions] = useSessionStore((s) => [s.useFetchSessions]);
-  useFetchSessions();
+  useFetchSessions(isLogin);
 
   const defaultSessions = useSessionStore(sessionSelectors.defaultSessions, isEqual);
   const customSessionGroups = useSessionStore(sessionSelectors.customSessionGroups, isEqual);

package/src/config/app.ts

@@ -25,6 +25,7 @@ export const getAppConfig = () => {
       AGENTS_INDEX_URL: z.string().url(),
 
       DEFAULT_AGENT_CONFIG: z.string(),
+      SYSTEM_AGENT: z.string().optional(),
 
       PLUGINS_INDEX_URL: z.string().url(),
       PLUGIN_SETTINGS: z.string().optional(),
@@ -44,6 +45,7 @@ export const getAppConfig = () => {
         : 'https://chat-agents.lobehub.com',
 
       DEFAULT_AGENT_CONFIG: process.env.DEFAULT_AGENT_CONFIG || '',
+      SYSTEM_AGENT: process.env.SYSTEM_AGENT,
 
       PLUGINS_INDEX_URL: !!process.env.PLUGINS_INDEX_URL
         ? process.env.PLUGINS_INDEX_URL

package/src/server/globalConfig/index.ts

@@ -1,4 +1,4 @@
-import { getAppConfig } from '@/config/app';
+import { appEnv, getAppConfig } from '@/config/app';
 import { fileEnv } from '@/config/file';
 import { langfuseEnv } from '@/config/langfuse';
 import { getLLMConfig } from '@/config/llm';
@@ -9,6 +9,7 @@ import {
   TogetherAIProviderCard,
 } from '@/config/modelProviders';
 import { enableNextAuth } from '@/const/auth';
+import { parseSystemAgent } from '@/server/globalConfig/parseSystemAgent';
 import { GlobalServerConfig } from '@/types/serverConfig';
 import { extractEnabledModels, transformToChatModelCards } from '@/utils/parseModels';
 
@@ -51,7 +52,6 @@ export const getServerGlobalConfig = () => {
     defaultAgent: {
       config: parseAgentConfig(DEFAULT_AGENT_CONFIG),
     },
-
     enableUploadFileToServer: !!fileEnv.S3_SECRET_ACCESS_KEY,
     enabledAccessCode: ACCESS_CODES?.length > 0,
     enabledOAuthSSO: enableNextAuth,
@@ -114,6 +114,7 @@ export const getServerGlobalConfig = () => {
       zeroone: { enabled: ENABLED_ZEROONE },
       zhipu: { enabled: ENABLED_ZHIPU },
     },
+    systemAgent: parseSystemAgent(appEnv.SYSTEM_AGENT),
     telemetry: {
       langfuse: langfuseEnv.ENABLE_LANGFUSE,
     },

package/src/server/globalConfig/parseSystemAgent.test.ts

@@ -0,0 +1,95 @@
+import { describe, expect, it } from 'vitest';
+
+import { parseSystemAgent } from './parseSystemAgent';
+
+describe('parseSystemAgent', () => {
+  it('should parse a valid environment variable string correctly', () => {
+    const envValue = 'topic=openai/gpt-3.5-turbo,translation=anthropic/claude-1';
+    const expected = {
+      topic: { provider: 'openai', model: 'gpt-3.5-turbo' },
+      translation: { provider: 'anthropic', model: 'claude-1' },
+    };
+
+    expect(parseSystemAgent(envValue)).toEqual(expected);
+  });
+
+  it('should handle empty environment variable string', () => {
+    const envValue = '';
+    const expected = {};
+
+    expect(parseSystemAgent(envValue)).toEqual(expected);
+  });
+
+  it('should ignore unknown keys in environment variable string', () => {
+    const envValue = 'topic=openai/gpt-3.5-turbo,unknown=test/model';
+    const expected = {
+      topic: { provider: 'openai', model: 'gpt-3.5-turbo' },
+    };
+
+    expect(parseSystemAgent(envValue)).toEqual(expected);
+  });
+
+  it('should throw an error for missing model or provider values', () => {
+    const envValue1 = 'topic=openai,translation=/claude-1';
+    const envValue2 = 'topic=/gpt-3.5-turbo,translation=anthropic/';
+
+    expect(() => parseSystemAgent(envValue1)).toThrowError(/Missing model or provider/);
+    expect(() => parseSystemAgent(envValue2)).toThrowError(/Missing model or provider/);
+  });
+
+  it('should throw an error for invalid environment variable format', () => {
+    const envValue = 'topic-openai/gpt-3.5-turbo';
+
+    expect(() => parseSystemAgent(envValue)).toThrowError(/Invalid environment variable format/);
+  });
+
+  it('should handle provider or model names with special characters', () => {
+    const envValue = 'topic=openrouter/mistralai/mistral-7b-instruct:free';
+    const expected = {
+      topic: { provider: 'openrouter', model: 'mistralai/mistral-7b-instruct:free' },
+    };
+
+    expect(parseSystemAgent(envValue)).toEqual(expected);
+  });
+
+  it('should handle extra whitespace in environment variable string', () => {
+    const envValue = '  topic = openai/gpt-3.5-turbo , translation = anthropic/claude-1 ';
+    const expected = {
+      topic: { provider: 'openai', model: 'gpt-3.5-turbo' },
+      translation: { provider: 'anthropic', model: 'claude-1' },
+    };
+
+    expect(parseSystemAgent(envValue)).toEqual(expected);
+  });
+
+  it('should handle full-width comma in environment variable string', () => {
+    const envValue = 'topic=openai/gpt-3.5-turbo，translation=anthropic/claude-1';
+    const expected = {
+      topic: { provider: 'openai', model: 'gpt-3.5-turbo' },
+      translation: { provider: 'anthropic', model: 'claude-1' },
+    };
+
+    expect(parseSystemAgent(envValue)).toEqual(expected);
+  });
+
+  it('should handle extra whitespace around provider and model names', () => {
+    const envValue = 'topic= openai / gpt-3.5-turbo ,translation= anthropic / claude-1 ';
+    const expected = {
+      topic: { provider: 'openai', model: 'gpt-3.5-turbo' },
+      translation: { provider: 'anthropic', model: 'claude-1' },
+    };
+
+    expect(parseSystemAgent(envValue)).toEqual(expected);
+  });
+
+  it('should handle an excessively long environment variable string', () => {
+    const longProviderName = 'a'.repeat(100);
+    const longModelName = 'b'.repeat(100);
+    const envValue = `topic=${longProviderName}/${longModelName}`;
+    const expected = {
+      topic: { provider: longProviderName, model: longModelName },
+    };
+
+    expect(parseSystemAgent(envValue)).toEqual(expected);
+  });
+});

package/src/server/globalConfig/parseSystemAgent.ts

@@ -0,0 +1,39 @@
+import { DEFAULT_SYSTEM_AGENT_CONFIG } from '@/const/settings';
+import { UserSystemAgentConfig } from '@/types/user/settings';
+
+const protectedKeys = Object.keys(DEFAULT_SYSTEM_AGENT_CONFIG);
+
+export const parseSystemAgent = (envString: string = ''): Partial<UserSystemAgentConfig> => {
+  if (!envString) return {};
+
+  const config: Partial<UserSystemAgentConfig> = {};
+
+  // 处理全角逗号和多余空格
+  let envValue = envString.replaceAll('，', ',').trim();
+
+  const pairs = envValue.split(',');
+
+  for (const pair of pairs) {
+    const [key, value] = pair.split('=').map((s) => s.trim());
+
+    if (key && value) {
+      const [provider, ...modelParts] = value.split('/');
+      const model = modelParts.join('/');
+
+      if (!provider || !model) {
+        throw new Error('Missing model or provider value');
+      }
+
+      if (protectedKeys.includes(key)) {
+        config[key as keyof UserSystemAgentConfig] = {
+          model: model.trim(),
+          provider: provider.trim(),
+        };
+      }
+    } else {
+      throw new Error('Invalid environment variable format');
+    }
+  }
+
+  return config;
+};

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

@@ -72,7 +72,7 @@ export interface SessionAction {
 
   updateSearchKeywords: (keywords: string) => void;
 
-  useFetchSessions: () => SWRResponse<ChatSessionList>;
+  useFetchSessions: (isLogin: boolean | undefined) => SWRResponse<ChatSessionList>;
   useSearchSessions: (keyword?: string) => SWRResponse<any>;
 
   internal_dispatchSessions: (payload: SessionDispatch) => void;
@@ -192,29 +192,33 @@ export const createSessionSlice: StateCreator<
     await refreshSessions();
   },
 
-  useFetchSessions: () =>
-    useClientDataSWR<ChatSessionList>(FETCH_SESSIONS_KEY, sessionService.getGroupedSessions, {
-      fallbackData: {
-        sessionGroups: [],
-        sessions: [],
-      },
-      onSuccess: (data) => {
-        if (
-          get().isSessionsFirstFetchFinished &&
-          isEqual(get().sessions, data.sessions) &&
-          isEqual(get().sessionGroups, data.sessionGroups)
-        )
-          return;
-
-        get().internal_processSessions(
-          data.sessions,
-          data.sessionGroups,
-          n('useFetchSessions/updateData') as any,
-        );
-        set({ isSessionsFirstFetchFinished: true }, false, n('useFetchSessions/onSuccess', data));
+  useFetchSessions: (isLogin) =>
+    useClientDataSWR<ChatSessionList>(
+      [FETCH_SESSIONS_KEY, isLogin],
+      () => sessionService.getGroupedSessions(),
+      {
+        fallbackData: {
+          sessionGroups: [],
+          sessions: [],
+        },
+        onSuccess: (data) => {
+          if (
+            get().isSessionsFirstFetchFinished &&
+            isEqual(get().sessions, data.sessions) &&
+            isEqual(get().sessionGroups, data.sessionGroups)
+          )
+            return;
+
+          get().internal_processSessions(
+            data.sessions,
+            data.sessionGroups,
+            n('useFetchSessions/updateData') as any,
+          );
+          set({ isSessionsFirstFetchFinished: true }, false, n('useFetchSessions/onSuccess', data));
+        },
+        suspense: true,
       },
-      suspense: true,
-    }),
+    ),
   useSearchSessions: (keyword) =>
     useSWR<LobeSessions>(
       [SEARCH_SESSIONS_KEY, keyword],

package/src/store/user/slices/common/action.ts

@@ -81,7 +81,9 @@ export const createCommonSlice: StateCreator<
             const serverSettings: DeepPartial<UserSettings> = {
               defaultAgent: serverConfig.defaultAgent,
               languageModel: serverConfig.languageModel,
+              systemAgent: serverConfig.systemAgent,
             };
+
             const defaultSettings = merge(get().defaultSettings, serverSettings);
 
             // merge preference

package/src/types/serverConfig.ts

@@ -1,7 +1,11 @@
 import { DeepPartial } from 'utility-types';
 
 import { ChatModelCard } from '@/types/llm';
-import { UserDefaultAgent, GlobalLLMProviderKey } from '@/types/user/settings';
+import {
+  GlobalLLMProviderKey,
+  UserDefaultAgent,
+  UserSystemAgentConfig,
+} from '@/types/user/settings';
 
 export interface ServerModelProviderConfig {
   enabled?: boolean;
@@ -21,6 +25,7 @@ export interface GlobalServerConfig {
   enabledAccessCode?: boolean;
   enabledOAuthSSO?: boolean;
   languageModel?: ServerLanguageModel;
+  systemAgent?: DeepPartial<UserSystemAgentConfig>;
   telemetry: {
     langfuse?: boolean;
   };