@lobehub/chat 0.148.8 → 0.148.10

package/CHANGELOG.md

@@ -2,6 +2,56 @@
 
 # Changelog
 
+### [Version 0.148.10](https://github.com/lobehub/lobe-chat/compare/v0.148.9...v0.148.10)
+
+<sup>Released on **2024-04-24**</sup>
+
+#### 💄 Styles
+
+- **misc**: Update Ollama model 240421.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### Styles
+
+- **misc**: Update Ollama model 240421, closes [#2130](https://github.com/lobehub/lobe-chat/issues/2130) ([e797af0](https://github.com/lobehub/lobe-chat/commit/e797af0))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
+### [Version 0.148.9](https://github.com/lobehub/lobe-chat/compare/v0.148.8...v0.148.9)
+
+<sup>Released on **2024-04-23**</sup>
+
+#### ♻ Code Refactoring
+
+- **misc**: Refactor for session server mode.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### Code refactoring
+
+- **misc**: Refactor for session server mode, closes [#2163](https://github.com/lobehub/lobe-chat/issues/2163) ([e012597](https://github.com/lobehub/lobe-chat/commit/e012597))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
 ### [Version 0.148.8](https://github.com/lobehub/lobe-chat/compare/v0.148.7...v0.148.8)
 
 <sup>Released on **2024-04-23**</sup>

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "0.148.8",
+  "version": "0.148.10",
   "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",
@@ -173,7 +173,7 @@
     "@next/eslint-plugin-next": "^14.2.2",
     "@peculiar/webcrypto": "^1.4.6",
     "@testing-library/jest-dom": "^6.4.2",
-    "@testing-library/react": "^15.0.2",
+    "@testing-library/react": "^15.0.4",
     "@types/chroma-js": "^2.4.4",
     "@types/debug": "^4.1.12",
     "@types/diff": "^5.2.0",

package/src/app/chat/features/SessionListContent/CollapseGroup/Actions.tsx

@@ -27,7 +27,7 @@ const Actions = memo<ActionsProps>(
   ({ id, openRenameModal, openConfigModal, onOpenChange, isCustomGroup, isPinned }) => {
     const { t } = useTranslation('chat');
     const { styles } = useStyles();
-    const { modal } = App.useApp();
+    const { modal, message } = App.useApp();
 
     const [createSession, removeSessionGroup] = useSessionStore((s) => [
       s.createSession,
@@ -48,9 +48,15 @@ const Actions = memo<ActionsProps>(
       icon: <Icon icon={Plus} />,
       key: 'newAgent',
       label: t('newAgent'),
-      onClick: ({ domEvent }) => {
+      onClick: async ({ domEvent }) => {
         domEvent.stopPropagation();
-        createSession({ group: id, pinned: isPinned });
+        const key = 'createNewAgentInGroup';
+        message.loading({ content: t('sessionGroup.creatingAgent'), duration: 0, key });
+
+        await createSession({ group: id, pinned: isPinned });
+
+        message.destroy(key);
+        message.success({ content: t('sessionGroup.createAgentSuccess') });
       },
     };
 
@@ -83,9 +89,9 @@ const Actions = memo<ActionsProps>(
             modal.confirm({
               centered: true,
               okButtonProps: { danger: true },
-              onOk: () => {
+              onOk: async () => {
                 if (!id) return;
-                removeSessionGroup(id);
+                await removeSessionGroup(id);
               },
               rootClassName: styles.modalRoot,
               title: t('sessionGroup.confirmRemoveGroupAlert'),

package/src/app/chat/features/SessionListContent/DefaultMode.tsx

@@ -1,10 +1,12 @@
 import { CollapseProps } from 'antd';
+import isEqual from 'fast-deep-equal';
 import { memo, useMemo, useState } from 'react';
 import { useTranslation } from 'react-i18next';
 
 import { useGlobalStore } from '@/store/global';
 import { preferenceSelectors } from '@/store/global/selectors';
 import { useSessionStore } from '@/store/session';
+import { sessionSelectors } from '@/store/session/selectors';
 import { SessionDefaultGroup } from '@/types/session';
 
 import Actions from '../SessionListContent/CollapseGroup/Actions';
@@ -22,11 +24,11 @@ const SessionListContent = memo(() => {
   const [configGroupModalOpen, setConfigGroupModalOpen] = useState(false);
 
   const [useFetchSessions] = useSessionStore((s) => [s.useFetchSessions]);
-  const { data } = useFetchSessions();
+  useFetchSessions();
 
-  const pinnedSessions = data?.pinned;
-  const defaultSessions = data?.default;
-  const customSessionGroups = data?.customGroup;
+  const defaultSessions = useSessionStore(sessionSelectors.defaultSessions, isEqual);
+  const customSessionGroups = useSessionStore(sessionSelectors.customSessionGroups, isEqual);
+  const pinnedSessions = useSessionStore(sessionSelectors.pinnedSessions, isEqual);
 
   const [sessionGroupKeys, updatePreference] = useGlobalStore((s) => [
     preferenceSelectors.sessionGroupKeys(s),

package/src/app/chat/features/SessionListContent/List/AddButton.tsx

@@ -12,18 +12,13 @@ const AddButton = memo<{ groupId?: string }>(({ groupId }) => {
   const { t } = useTranslation('chat');
   const createSession = useSessionStore((s) => s.createSession);
 
-  const { mutate, isValidating } = useActionSWR('session.createSession', (groupId) =>
-    createSession({ group: groupId }),
-  );
+  const { mutate, isValidating } = useActionSWR(['session.createSession', groupId], () => {
+    return createSession({ group: groupId });
+  });
 
   return (
     <Flexbox style={{ margin: '12px 16px' }}>
-      <Button
-        block
-        icon={<Icon icon={Plus} />}
-        loading={isValidating}
-        onClick={() => mutate(groupId)}
-      >
+      <Button block icon={<Icon icon={Plus} />} loading={isValidating} onClick={() => mutate()}>
         {t('newAgent')}
       </Button>
     </Flexbox>

package/src/app/chat/features/SessionListContent/Modals/ConfigGroupModal/GroupItem.tsx

@@ -1,5 +1,5 @@
 import { ActionIcon, EditableText, SortableList } from '@lobehub/ui';
-import { App, Popconfirm } from 'antd';
+import { App } from 'antd';
 import { createStyles } from 'antd-style';
 import { PencilLine, Trash } from 'lucide-react';
 import { memo, useState } from 'react';
@@ -25,7 +25,7 @@ const useStyles = createStyles(({ css }) => ({
 const GroupItem = memo<SessionGroupItem>(({ id, name }) => {
   const { t } = useTranslation('chat');
   const { styles } = useStyles();
-  const { message } = App.useApp();
+  const { message, modal } = App.useApp();
 
   const [editing, setEditing] = useState(false);
   const [updateSessionGroupName, removeSessionGroup] = useSessionStore((s) => [
@@ -40,29 +40,34 @@ const GroupItem = memo<SessionGroupItem>(({ id, name }) => {
         <>
           <span className={styles.title}>{name}</span>
           <ActionIcon icon={PencilLine} onClick={() => setEditing(true)} size={'small'} />
-          <Popconfirm
-            arrow={false}
-            okButtonProps={{
-              danger: true,
-              type: 'primary',
+          <ActionIcon
+            icon={Trash}
+            onClick={() => {
+              modal.confirm({
+                centered: true,
+                okButtonProps: {
+                  danger: true,
+                  type: 'primary',
+                },
+                onOk: async () => {
+                  await removeSessionGroup(id);
+                },
+                title: t('sessionGroup.confirmRemoveGroupAlert'),
+              });
             }}
-            onConfirm={() => {
-              removeSessionGroup(id);
-            }}
-            title={t('sessionGroup.confirmRemoveGroupAlert')}
-          >
-            <ActionIcon icon={Trash} size={'small'} />
-          </Popconfirm>
+            size={'small'}
+          />
         </>
       ) : (
         <EditableText
           editing={editing}
-          onChangeEnd={(input) => {
+          onChangeEnd={async (input) => {
             if (name !== input) {
               if (!input) return;
               if (input.length === 0 || input.length > 20)
                 return message.warning(t('sessionGroup.tooLong'));
-              updateSessionGroupName(id, input);
+
+              await updateSessionGroupName(id, input);
               message.success(t('sessionGroup.renameSuccess'));
             }
             setEditing(false);

package/src/app/chat/features/SessionListContent/Modals/ConfigGroupModal/index.tsx

@@ -3,7 +3,7 @@ import { Button } from 'antd';
 import { createStyles } from 'antd-style';
 import isEqual from 'fast-deep-equal';
 import { Plus } from 'lucide-react';
-import { memo } from 'react';
+import { memo, useState } from 'react';
 import { useTranslation } from 'react-i18next';
 import { Flexbox } from 'react-layout-kit';
 
@@ -35,6 +35,7 @@ const ConfigGroupModal = memo<ModalProps>(({ open, onCancel }) => {
     s.addSessionGroup,
     s.updateSessionGroupSort,
   ]);
+  const [loading, setLoading] = useState(false);
 
   return (
     <Modal
@@ -67,7 +68,12 @@ const ConfigGroupModal = memo<ModalProps>(({ open, onCancel }) => {
         <Button
           block
           icon={<Icon icon={Plus} />}
-          onClick={() => addSessionGroup(t('sessionGroup.newGroup'))}
+          loading={loading}
+          onClick={async () => {
+            setLoading(true);
+            await addSessionGroup(t('sessionGroup.newGroup'));
+            setLoading(false);
+          }}
         >
           {t('sessionGroup.createGroup')}
         </Button>

package/src/app/chat/features/SessionListContent/Modals/CreateGroupModal.tsx

@@ -22,21 +22,29 @@ const CreateGroupModal = memo<CreateGroupModalProps>(
       s.addSessionGroup,
     ]);
     const [input, setInput] = useState('');
+    const [loading, setLoading] = useState(false);
 
     return (
       <div onClick={(e) => e.stopPropagation()}>
         <Modal
           allowFullscreen
-          onCancel={onCancel}
+          destroyOnClose
+          okButtonProps={{ loading }}
+          onCancel={(e) => {
+            setInput('');
+            onCancel?.(e);
+          }}
           onOk={async (e: MouseEvent<HTMLButtonElement>) => {
             if (!input) return;
 
             if (input.length === 0 || input.length > 20)
               return message.warning(t('sessionGroup.tooLong'));
 
+            setLoading(true);
             const groupId = await addCustomGroup(input);
             await updateSessionGroup(id, groupId);
             toggleExpandSessionGroup(groupId, true);
+            setLoading(false);
 
             message.success(t('sessionGroup.createSuccess'));
             onCancel?.(e);

package/src/app/chat/features/SessionListContent/Modals/RenameGroupModal.tsx

@@ -18,18 +18,27 @@ const RenameGroupModal = memo<RenameGroupModalProps>(({ id, open, onCancel }) =>
   const group = useSessionStore((s) => sessionGroupSelectors.getGroupById(id)(s), isEqual);
 
   const [input, setInput] = useState<string>();
+  const [loading, setLoading] = useState(false);
 
   const { message } = App.useApp();
   return (
     <Modal
       allowFullscreen
-      onCancel={onCancel}
-      onOk={(e) => {
+      destroyOnClose
+      okButtonProps={{ loading }}
+      onCancel={(e) => {
+        setInput(group?.name);
+        onCancel?.(e);
+      }}
+      onOk={async (e) => {
         if (!input) return;
         if (input.length === 0 || input.length > 20)
           return message.warning(t('sessionGroup.tooLong'));
-        updateSessionGroupName(id, input);
+        setLoading(true);
+        await updateSessionGroupName(id, input);
         message.success(t('sessionGroup.renameSuccess'));
+        setLoading(false);
+
         onCancel?.(e);
       }}
       open={open}

package/src/app/home/Redirect.tsx

@@ -8,8 +8,8 @@ import { sessionService } from '@/services/session';
 
 const checkHasConversation = async () => {
   const hasMessages = await messageService.hasMessages();
-  const hasAgents = await sessionService.countSessions();
-  return hasMessages || hasAgents === 0;
+  const hasAgents = await sessionService.hasSessions();
+  return hasMessages || hasAgents;
 };
 
 const Redirect = memo(() => {

package/src/app/welcome/(desktop)/features/Footer.tsx

@@ -16,7 +16,7 @@ const Footer = memo(() => {
   return (
     <Flexbox align={'center'} horizontal justify={'space-between'} style={{ padding: 16 }}>
       <span style={{ color: theme.colorTextDescription }}>
-        ©{new Date().getFullYear()} LobeHub
+        © 2023 - {new Date().getFullYear()} LobeHub, LLC
       </span>
       <Flexbox horizontal>
         <ActionIcon

package/src/app/welcome/features/Banner/index.tsx

@@ -3,12 +3,12 @@
 import { Icon } from '@lobehub/ui';
 import { Button } from 'antd';
 import { SendHorizonal } from 'lucide-react';
+import Link from 'next/link';
 import { useRouter } from 'next/navigation';
 import { memo } from 'react';
 import { useTranslation } from 'react-i18next';
 import { Flexbox } from 'react-layout-kit';
 
-import DataImporter from '@/features/DataImporter';
 import { useGlobalStore } from '@/store/global';
 
 import Hero from './Hero';
@@ -32,15 +32,11 @@ const Banner = memo<{ mobile?: boolean }>(({ mobile }) => {
         justify={'center'}
         width={'100%'}
       >
-        <DataImporter
-          onFinishImport={() => {
-            router.push('/chat');
-          }}
-        >
-          <Button block={mobile} size={'large'}>
-            {t('button.import')}
+        <Link href={'/market'}>
+          <Button block={mobile} size={'large'} type={'default'}>
+            {t('button.market')}
           </Button>
-        </DataImporter>
+        </Link>
         <Button
           block={mobile}
           onClick={() => (isMobile ? router.push('/chat') : switchBackToChat())}

package/src/config/modelProviders/ollama.ts

@@ -2,148 +2,148 @@ import { ModelProviderCard } from '@/types/llm';
 
 const Ollama: ModelProviderCard = {
   chatModels: [
+    {
+      displayName: 'Command R 35B',
+      enabled: true,
+      id: 'command-r',
+      tokens: 128_000,
+    },
+    {
+      displayName: 'Command R+ 104B (Q2_K)',
+      id: 'command-r-plus:104b-q2_K',
+      tokens: 128_000,
+    },
     {
       displayName: 'Gemma 7B',
       enabled: true,
-      functionCall: false,
       id: 'gemma',
       tokens: 4000,
-      vision: false,
     },
     {
       displayName: 'Gemma 2B',
       enabled: true,
-      functionCall: false,
       id: 'gemma:2b',
       tokens: 4000,
-      vision: false,
     },
     {
       displayName: 'Llama2 Chat 13B',
-      functionCall: false,
       id: 'llama2:13b',
       tokens: 4000,
-      vision: false,
     },
     {
       displayName: 'Llama2 Chat 7B',
-      enabled: true,
-      functionCall: false,
       id: 'llama2',
       tokens: 4000,
-      vision: false,
     },
     {
       displayName: 'Llama2 Chat 70B',
-      functionCall: false,
       id: 'llama2:70b',
       tokens: 4000,
-      vision: false,
     },
     {
       displayName: 'Llama2 CN 13B',
-      functionCall: false,
       id: 'llama2-chinese:13b',
       tokens: 4000,
-      vision: false,
     },
     {
       displayName: 'Llama2 CN 7B',
-      functionCall: false,
       id: 'llama2-chinese',
       tokens: 4000,
-      vision: false,
+    },
+    {
+      displayName: 'Llama3 8B',
+      enabled: true,
+      id: 'llama3',
+      tokens: 8000,
+    },
+    {
+      displayName: 'Llama3 70B',
+      id: 'llama3:70b',
+      tokens: 8000,
+    },
+    {
+      displayName: 'WizardLM 2 7B',
+      enabled: true,
+      id: 'wizardlm2',
+      tokens: 65_536,
+    },
+    {
+      displayName: 'WizardLM 2 8x22B',
+      id: 'wizardlm2:8x22b',
+      tokens: 65_536,
     },
     {
       displayName: 'Code Llama 7B',
-      functionCall: false,
       id: 'codellama',
       tokens: 16_000,
-      vision: false,
     },
     {
       displayName: 'Code Llama 34B',
-      functionCall: false,
       id: 'codellama:34b',
       tokens: 16_000,
-      vision: false,
     },
     {
       displayName: 'Code Llama 70B',
-      functionCall: false,
       id: 'codellama:70b',
       tokens: 16_000,
-      vision: false,
     },
     {
       displayName: 'Code Llama 7B (Python)',
-      functionCall: false,
       id: 'codellama:python',
       tokens: 16_000,
-      vision: false,
     },
     {
       displayName: 'Mistral',
       enabled: true,
-      functionCall: false,
       id: 'mistral',
       tokens: 4800,
-      vision: false,
     },
     {
       displayName: 'Mixtral 8x7B',
       enabled: true,
-      functionCall: false,
       id: 'mixtral',
       tokens: 32_000,
-      vision: false,
+    },
+    {
+      displayName: 'Mixtral 8x22B',
+      id: 'mixtral:8x22b',
+      tokens: 64_000,
     },
     {
       displayName: 'Qwen Chat 4B',
-      functionCall: false,
       id: 'qwen',
       tokens: 32_768,
-      vision: false,
     },
     {
       displayName: 'Qwen Chat 7B',
       enabled: true,
-      functionCall: false,
       id: 'qwen:7b',
       tokens: 32_768,
-      vision: false,
     },
     {
       displayName: 'Qwen Chat 14B',
-      functionCall: false,
       id: 'qwen:14b',
       tokens: 32_768,
-      vision: false,
     },
     {
       displayName: 'Qwen Chat 72B',
-      functionCall: false,
       id: 'qwen:72b',
       tokens: 32_768,
-      vision: false,
     },
     {
       displayName: 'LLaVA 7B',
-      functionCall: false,
       id: 'llava',
       tokens: 4000,
       vision: true,
     },
     {
       displayName: 'LLaVA 13B',
-      functionCall: false,
       id: 'llava:13b',
       tokens: 4000,
       vision: true,
     },
     {
       displayName: 'LLaVA 34B',
-      functionCall: false,
       id: 'llava:34b',
       tokens: 4000,
       vision: true,

package/src/database/client/models/__tests__/session.test.ts

@@ -179,8 +179,8 @@ describe('SessionModel', () => {
       await SessionModel.create('agent', sessionData);
 
       const sessionsWithGroups = await SessionModel.queryWithGroups();
-      expect(sessionsWithGroups.all).toHaveLength(1);
-      expect(sessionsWithGroups.all[0]).toEqual(expect.objectContaining(sessionData));
+      expect(sessionsWithGroups.sessions).toHaveLength(1);
+      expect(sessionsWithGroups.sessions[0]).toEqual(expect.objectContaining(sessionData));
     });
   });
 

package/src/database/client/models/session.ts

@@ -43,21 +43,11 @@ class _SessionModel extends BaseModel {
   }
 
   async queryWithGroups(): Promise<ChatSessionList> {
-    const groups = await SessionGroupModel.query();
-    const customGroups = await this.queryByGroupIds(groups.map((item) => item.id));
-    const defaultItems = await this.querySessionsByGroupId(SessionDefaultGroup.Default);
-    const pinnedItems = await this.getPinnedSessions();
+    const sessionGroups = await SessionGroupModel.query();
 
-    const all = await this.query();
-    return {
-      all,
-      customGroup: groups.map((group) => ({
-        ...group,
-        children: customGroups[group.id],
-      })),
-      default: defaultItems,
-      pinned: pinnedItems,
-    };
+    const sessions = await this.query();
+
+    return { sessionGroups, sessions };
   }
 
   /**