@lobehub/chat 0.160.5 → 0.160.6

package/src/store/user/slices/auth/selectors.ts

@@ -43,5 +43,5 @@ const isLogin = (s: UserStore) => {
 export const authSelectors = {
   isLogin,
   isLoginWithAuth: (s: UserStore) => s.isSignedIn,
-  isLoginWithClerk: (s: UserStore) => s.isSignedIn && enableClerk,
+  isLoginWithClerk: (s: UserStore): boolean => (s.isSignedIn && enableClerk) || false,
 };

package/src/store/user/slices/preference/action.test.ts

@@ -3,6 +3,7 @@ import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 import { withSWR } from '~test-utils';
 
 import { DEFAULT_PREFERENCE } from '@/const/user';
+import { userService } from '@/services/user';
 import { useUserStore } from '@/store/user';
 import { UserGuide, UserPreference } from '@/types/user';
 
@@ -44,9 +45,7 @@ describe('createPreferenceSlice', () => {
     it('should return false when userId is empty', async () => {
       const { result } = renderHook(() => useUserStore());
 
-      vi.spyOn(result.current.preferenceStorage, 'getFromLocalStorage').mockResolvedValueOnce(
-        {} as any,
-      );
+      vi.spyOn(userService, 'getPreference').mockResolvedValueOnce({} as any);
 
       const { result: prefernce } = renderHook(() => result.current.useInitPreference(), {
         wrapper: withSWR,
@@ -60,9 +59,7 @@ describe('createPreferenceSlice', () => {
     it('should return default preference when local storage is empty', async () => {
       const { result } = renderHook(() => useUserStore());
 
-      vi.spyOn(result.current.preferenceStorage, 'getFromLocalStorage').mockResolvedValueOnce(
-        {} as any,
-      );
+      vi.spyOn(userService, 'getPreference').mockResolvedValueOnce({} as any);
 
       renderHook(() => result.current.useInitPreference(), {
         wrapper: withSWR,
@@ -82,9 +79,7 @@ describe('createPreferenceSlice', () => {
         guide: { topic: false, moveSettingsToAvatar: true },
       };
 
-      vi.spyOn(result.current.preferenceStorage, 'getFromLocalStorage').mockResolvedValueOnce(
-        savedPreference,
-      );
+      vi.spyOn(userService, 'getPreference').mockResolvedValueOnce(savedPreference);
 
       const { result: prefernce } = renderHook(() => result.current.useInitPreference(), {
         wrapper: withSWR,

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

@@ -3,6 +3,7 @@ import type { StateCreator } from 'zustand/vanilla';
 
 import { DEFAULT_PREFERENCE } from '@/const/user';
 import { useClientDataSWR } from '@/libs/swr';
+import { userService } from '@/services/user';
 import type { UserStore } from '@/store/user';
 import { UserGuide, UserPreference } from '@/types/user';
 import { merge } from '@/utils/merge';
@@ -11,8 +12,8 @@ import { setNamespace } from '@/utils/storeDebug';
 const n = setNamespace('preference');
 
 export interface PreferenceAction {
-  updateGuideState: (guide: Partial<UserGuide>) => void;
-  updatePreference: (preference: Partial<UserPreference>, action?: any) => void;
+  updateGuideState: (guide: Partial<UserGuide>) => Promise<void>;
+  updatePreference: (preference: Partial<UserPreference>, action?: any) => Promise<void>;
   useInitPreference: () => SWRResponse;
 }
 
@@ -22,33 +23,29 @@ export const createPreferenceSlice: StateCreator<
   [],
   PreferenceAction
 > = (set, get) => ({
-  updateGuideState: (guide) => {
+  updateGuideState: async (guide) => {
     const { updatePreference } = get();
     const nextGuide = merge(get().preference.guide, guide);
-    updatePreference({ guide: nextGuide });
+    await updatePreference({ guide: nextGuide });
   },
-  updatePreference: (preference, action) => {
+  updatePreference: async (preference, action) => {
     const nextPreference = merge(get().preference, preference);
 
     set({ preference: nextPreference }, false, action || n('updatePreference'));
 
-    get().preferenceStorage.saveToLocalStorage(nextPreference);
+    await userService.updatePreference(nextPreference);
   },
 
   useInitPreference: () =>
-    useClientDataSWR<UserPreference>(
-      'initUserPreference',
-      () => get().preferenceStorage.getFromLocalStorage(),
-      {
-        onSuccess: (preference) => {
-          const isEmpty = Object.keys(preference).length === 0;
-
-          set(
-            { isPreferenceInit: true, preference: isEmpty ? DEFAULT_PREFERENCE : preference },
-            false,
-            n('initPreference'),
-          );
-        },
+    useClientDataSWR<UserPreference>('initUserPreference', userService.getPreference, {
+      onSuccess: (preference) => {
+        const isEmpty = Object.keys(preference).length === 0;
+
+        set(
+          { isPreferenceInit: true, preference: isEmpty ? DEFAULT_PREFERENCE : preference },
+          false,
+          n('initPreference'),
+        );
       },
-    ),
+    }),
 });

package/src/store/user/slices/settings/selectors/modelConfig.test.ts

@@ -1,8 +1,8 @@
 import { describe, expect, it } from 'vitest';
 
+import { UserStore } from '@/store/user';
 import { merge } from '@/utils/merge';
 
-import { UserStore, useUserStore } from '../../../store';
 import { UserSettingsState, initialSettingsState } from '../initialState';
 import { modelConfigSelectors } from './modelConfig';
 
@@ -33,6 +33,34 @@ describe('modelConfigSelectors', () => {
     });
   });
 
+  describe('isProviderFetchOnClient', () => {
+    it('client fetch should disabled on default', () => {
+      const s = merge(initialSettingsState, {
+        settings: {
+          languageModel: {
+            azure: {
+              endpoint: 'endpoint',
+              apiKey: 'apikey',
+            },
+          },
+        },
+      } as UserSettingsState) as unknown as UserStore;
+
+      expect(modelConfigSelectors.isProviderFetchOnClient('azure')(s)).toBe(false);
+    });
+
+    it('client fetch should enabled if user set it enabled', () => {
+      const s = merge(initialSettingsState, {
+        settings: {
+          languageModel: {
+            azure: { fetchOnClient: true },
+          },
+        },
+      } as UserSettingsState) as unknown as UserStore;
+      expect(modelConfigSelectors.isProviderFetchOnClient('azure')(s)).toBe(true);
+    });
+  });
+
   describe('getCustomModelCardById', () => {
     it('should return the custom model card with the given id and provider', () => {
       const s = merge(initialSettingsState, {

package/src/store/user/slices/settings/selectors/modelConfig.ts

@@ -13,7 +13,7 @@ const isProviderFetchOnClient = (provider: GlobalLLMProviderKey | string) => (s:
   const config = getProviderConfigById(provider)(s);
   if (typeof config?.fetchOnClient !== 'undefined') return config?.fetchOnClient;
 
-  return isProviderEndpointNotEmpty(provider)(s);
+  return false;
 };
 
 const getCustomModelCard =

package/src/tools/dalle/Render/Item/Error.tsx

@@ -28,7 +28,7 @@ const Error = memo<ErrorProps>(({ messageId, index }) => {
         <Alert
           extra={
             <Highlighter copyButtonSize={'small'} language={'json'}>
-              {JSON.stringify(error.body, null, 2)}
+              {JSON.stringify(error?.body || error, null, 2)}
             </Highlighter>
           }
           extraDefaultExpand

package/src/types/files.ts

@@ -1,3 +1,5 @@
+import { z } from 'zod';
+
 export interface FilePreview {
   base64Url?: string;
   data?: ArrayBuffer;
@@ -6,3 +8,34 @@ export interface FilePreview {
   saveMode: 'local' | 'url';
   url: string;
 }
+
+export const UploadFileSchema = z.object({
+  data: z.instanceof(ArrayBuffer).optional(),
+  /**
+   * file type
+   * @example 'image/png'
+   */
+  fileType: z.string(),
+  metadata: z.any().optional(),
+  /**
+   * 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().optional(),
+});
+
+export type UploadFileParams = z.infer<typeof UploadFileSchema>;