@lobehub/chat 0.100.5 → 0.101.0

package/src/store/session/slices/session/reducers/session.test.ts

@@ -206,6 +206,103 @@ describe('sessionsReducer', () => {
     });
   });
 
+  describe('updateSessionFiles', () => {
+    it('should update session files when valid id, key, and value are provided', () => {
+      const state: LobeSessions = {
+        'session-id': {
+          id: 'session-id',
+          config: {
+            model: 'gpt-3.5-turbo',
+            params: {},
+            systemRole: 'system-role',
+          },
+          type: 'agent',
+          meta: {
+            avatar: 'avatar-url',
+            backgroundColor: 'background-color',
+            description: 'description',
+            tags: ['tag1', 'tag2'],
+            title: 'title',
+          },
+        } as LobeAgentSession,
+      };
+      const id = 'session-id';
+      const files = ['new-file'];
+      const payload: SessionDispatch = { type: 'updateSessionFiles', id, files };
+
+      const newState = sessionsReducer(state, payload);
+
+      expect(newState).toEqual({
+        'session-id': {
+          ...state['session-id'],
+          files: ['new-file'],
+        },
+      });
+    });
+
+    it('should not change state when invalid id, key, and value are provided', () => {
+      const state: LobeSessions = {
+        'session-id': {
+          id: 'session-id',
+          config: {
+            model: 'gpt-3.5-turbo',
+            params: {},
+            systemRole: 'system-role',
+          },
+          type: 'agent',
+          meta: {
+            avatar: 'avatar-url',
+            backgroundColor: 'background-color',
+            description: 'description',
+            tags: ['tag1', 'tag2'],
+            title: 'title',
+          },
+        } as LobeAgentSession,
+      };
+      const id = 'non-existent-id';
+      const files = ['new-file'];
+      const payload: SessionDispatch = { type: 'updateSessionFiles', id, files };
+
+      const newState = sessionsReducer(state, payload);
+
+      expect(newState).toEqual(state);
+    });
+
+    it('should not change state when valid id, invalid files  are provided', () => {
+      const state: LobeSessions = {
+        'session-id': {
+          id: 'session-id',
+          config: {
+            model: 'gpt-3.5-turbo',
+            params: {},
+            systemRole: 'system-role',
+          },
+          type: 'agent',
+          meta: {
+            avatar: 'avatar-url',
+            backgroundColor: 'background-color',
+            description: 'description',
+            tags: ['tag1', 'tag2'],
+            title: 'title',
+          },
+        } as LobeAgentSession,
+      };
+      const id = 'session-id';
+
+      const errFile1 = 123213412 as any;
+      const res1 = sessionsReducer(state, { type: 'updateSessionFiles', id, files: errFile1 });
+      expect(res1).toEqual(state);
+
+      const errFile2 = undefined as any;
+      const res2 = sessionsReducer(state, { type: 'updateSessionFiles', id, files: errFile2 });
+      expect(res2).toEqual(state);
+
+      const errFile3 = ['state', 123] as any;
+      const res3 = sessionsReducer(state, { type: 'updateSessionFiles', id, files: errFile3 });
+      expect(res3).toEqual(state);
+    });
+  });
+
   describe('updateSessionConfig', () => {
     it('should update session config when valid id and partial config are provided', () => {
       const state: LobeSessions = {
@@ -345,7 +442,7 @@ describe('sessionsReducer', () => {
     });
   });
 
-  test('should not change state when invalid operation type is provided', () => {
+  it('should not change state when invalid operation type is provided', () => {
     const state: LobeSessions = {
       session1: {
         id: 'session1',

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

@@ -1,4 +1,5 @@
 import { produce } from 'immer';
+import { z } from 'zod';
 
 import { ChatMessageMap } from '@/types/chatMessage';
 import { MetaData } from '@/types/meta';
@@ -39,7 +40,7 @@ interface UpdateSessionChat {
 }
 
 /**
- * @title 更新会话聊天上下文
+ * 更新会话聊天上下文
  */
 interface UpdateSessionTopic {
   /**
@@ -51,6 +52,15 @@ interface UpdateSessionTopic {
   type: 'updateSessionTopic';
 }
 
+/**
+ * 更新会话聊天上下文
+ */
+interface UpdateSessionFiles {
+  files: string[];
+  id: string;
+  type: 'updateSessionFiles';
+}
+
 interface UpdateSessionMeta {
   id: string;
   key: keyof MetaData;
@@ -76,6 +86,7 @@ export type SessionDispatch =
   | UpdateSessionMeta
   | UpdateSessionAgentConfig
   | UpdateSessionTopic
+  | UpdateSessionFiles
   | ToggleSessionPinned;
 
 export const sessionsReducer = (state: LobeSessions, payload: SessionDispatch): LobeSessions => {
@@ -136,6 +147,20 @@ export const sessionsReducer = (state: LobeSessions, payload: SessionDispatch):
       });
     }
 
+    case 'updateSessionFiles': {
+      return produce(state, (draft) => {
+        const session = draft[payload.id];
+        if (!session || !payload.files) return;
+
+        const schema = z.array(z.string());
+
+        const { success } = schema.safeParse(payload.files);
+        if (!success) return;
+
+        session.files = payload.files;
+      });
+    }
+
     case 'updateSessionConfig': {
       return produce(state, (draft) => {
         const { id, config } = payload;

package/src/types/chatMessage.ts

@@ -36,18 +36,19 @@ export interface ChatMessage extends BaseDataModel {
     translate?: ChatTranslate;
   } & Record<string, any>;
 
+  files?: string[];
   /**
    * replace with plugin
    * @deprecated
    */
   function_call?: OpenAIFunctionCall;
+
   name?: string;
 
   parentId?: string;
-
   plugin?: PluginRequestPayload;
-  pluginState?: any;
 
+  pluginState?: any;
   // 引用
   quotaId?: string;
   /**

package/src/types/database/db.ts

@@ -0,0 +1,4 @@
+export type DBModel<T> = T & {
+  createdAt: number;
+  id: string;
+};

package/src/types/database/files.ts

@@ -0,0 +1,38 @@
+import { z } from 'zod';
+
+export const LocalFileSchema = z.object({
+  /**
+   * create Time
+   */
+  createdAt: z.number().optional(),
+  /**
+   * file data array buffer
+   */
+  data: z.instanceof(ArrayBuffer),
+  /**
+   * file type
+   * @example 'image/png'
+   */
+  fileType: z.string(),
+  /**
+   * file name
+   * @example 'test.png'
+   */
+  name: z.string(),
+  /**
+   * the mode database save the file
+   * local mean save the raw file into data
+   * url mean upload the file to a cdn and then save the url
+   */
+  saveMode: z.enum(['local', 'url']),
+  /**
+   * file size
+   */
+  size: z.number(),
+  /**
+   * file url if saveMode is url
+   */
+  url: z.string().url().optional(),
+});
+
+export type LocalFile = z.infer<typeof LocalFileSchema>;

package/src/types/files.ts

@@ -0,0 +1,8 @@
+import { LocalFile } from './database/files';
+
+export interface FilePreview extends Pick<LocalFile, 'saveMode' | 'fileType'> {
+  base64Url?: string;
+  data?: ArrayBuffer;
+  name: string;
+  url: string;
+}

package/src/types/openai/chat.ts

@@ -1,12 +1,26 @@
 import { OpenAIFunctionCall } from '@/types/chatMessage';
 import { LLMRoleType } from '@/types/llm';
 
+interface UserMessageContentPartText {
+  text: string;
+  type: 'text';
+}
+interface UserMessageContentPartImage {
+  image_url: {
+    detail?: 'auto' | 'low' | 'high';
+    url: string;
+  };
+  type: 'image_url';
+}
+
+export type UserMessageContentPart = UserMessageContentPartText | UserMessageContentPartImage;
+
 export interface OpenAIChatMessage {
   /**
    * @title 内容
    * @description 消息内容
    */
-  content: string;
+  content: string | UserMessageContentPart[];
 
   function_call?: OpenAIFunctionCall;
   name?: string;

package/src/types/session.ts

@@ -19,6 +19,7 @@ interface LobeSessionBase extends BaseDataModel {
    * 聊天记录
    */
   chats: ChatMessageMap;
+  files?: string[];
   /**
    * 置顶
    */

package/tests/setup.ts

@@ -0,0 +1,8 @@
+import '@testing-library/jest-dom';
+// remove antd hash on test
+import { theme } from 'antd';
+// mock indexedDB to test with dexie
+// refs: https://github.com/dumbmatter/fakeIndexedDB#dexie-and-other-indexeddb-api-wrappers
+import 'fake-indexeddb/auto';
+
+theme.defaultConfig.hashed = false;

package/vitest.config.ts

@@ -7,10 +7,12 @@ export default defineConfig({
       '@': resolve(__dirname, './src'),
     },
     coverage: {
+      exclude: ['__mocks__/**'],
       provider: 'v8',
       reporter: ['text', 'json', 'lcov', 'text-summary'],
     },
     environment: 'jsdom',
     globals: true,
+    setupFiles: './tests/setup.ts',
   },
 });

package/src/app/chat/(desktop)/features/ChatInput/Footer.tsx

@@ -1,63 +0,0 @@
-import { Icon } from '@lobehub/ui';
-import { Button } from 'antd';
-import { createStyles } from 'antd-style';
-import { ArrowBigUp, CornerDownLeft, Loader2 } from 'lucide-react';
-import { memo } from 'react';
-import { useTranslation } from 'react-i18next';
-import { Flexbox } from 'react-layout-kit';
-
-import SaveTopic from '@/app/chat/features/ChatInput/Topic';
-import { useSessionStore } from '@/store/session';
-
-import { useSendMessage } from './useSend';
-
-const useStyles = createStyles(({ css }) => ({
-  footerBar: css`
-    display: flex;
-    flex: none;
-    gap: 8px;
-    align-items: center;
-    justify-content: flex-end;
-
-    padding: 0 24px;
-  `,
-}));
-
-const Footer = memo(() => {
-  const { t } = useTranslation('chat');
-  const { styles, theme } = useStyles();
-  const [loading, onStop] = useSessionStore((s) => [!!s.chatLoadingId, s.stopGenerateMessage]);
-
-  const onSend = useSendMessage();
-
-  return (
-    <div className={styles.footerBar}>
-      <Flexbox
-        gap={4}
-        horizontal
-        style={{ color: theme.colorTextDescription, fontSize: 12, marginRight: 12 }}
-      >
-        <Icon icon={CornerDownLeft} />
-        <span>{t('send')}</span>
-        <span>/</span>
-        <Flexbox horizontal>
-          <Icon icon={ArrowBigUp} />
-          <Icon icon={CornerDownLeft} />
-        </Flexbox>
-        <span>{t('warp')}</span>
-      </Flexbox>
-      <SaveTopic />
-      {loading ? (
-        <Button icon={loading && <Icon icon={Loader2} spin />} onClick={onStop}>
-          {t('stop')}
-        </Button>
-      ) : (
-        <Button onClick={() => onSend()} type={'primary'}>
-          {t('send')}
-        </Button>
-      )}
-    </div>
-  );
-});
-
-export default Footer;