@lobehub/chat 0.160.5 → 0.160.7

package/src/store/chat/slices/enchance/action.ts

@@ -105,12 +105,13 @@ export const chatEnhance: StateCreator<
   },
 
   updateMessageTTS: async (id, data) => {
-    await messageService.updateMessage(id, { tts: data as ChatTTS });
+    await messageService.updateMessageTTS(id, data);
     await get().refreshMessages();
   },
 
   updateMessageTranslate: async (id, data) => {
-    await messageService.updateMessage(id, { translate: data as ChatTranslate });
+    await messageService.updateMessageTranslate(id, data);
+
     await get().refreshMessages();
   },
 });

package/src/store/chat/slices/plugin/action.test.ts

@@ -372,11 +372,9 @@ describe('ChatPluginAction', () => {
         await result.current.updatePluginState(messageId, pluginStateKey, pluginStateValue);
       });
 
-      expect(messageService.updateMessagePluginState).toHaveBeenCalledWith(
-        messageId,
-        pluginStateKey,
-        pluginStateValue,
-      );
+      expect(messageService.updateMessagePluginState).toHaveBeenCalledWith(messageId, {
+        [pluginStateKey]: pluginStateValue,
+      });
       expect(initialState.refreshMessages).toHaveBeenCalled();
     });
   });

package/src/store/chat/slices/plugin/action.ts

@@ -298,7 +298,7 @@ export const chatPlugin: StateCreator<
   updatePluginState: async (id, key, value) => {
     const { refreshMessages } = get();
 
-    await messageService.updateMessagePluginState(id, key, value);
+    await messageService.updateMessagePluginState(id, { [key]: value });
     await refreshMessages();
   },
 });

package/src/store/market/action.ts

@@ -12,6 +12,7 @@ import type { Store } from './store';
 export interface StoreAction {
   activateAgent: (identifier: string) => void;
   deactivateAgent: () => void;
+  getAgentById: (identifiers: string) => Promise<AgentsMarketItem>;
   setSearchKeywords: (keywords: string) => void;
   updateAgentMap: (key: string, value: AgentsMarketItem) => void;
   useFetchAgent: (identifier: string) => SWRResponse<AgentsMarketItem>;
@@ -30,6 +31,12 @@ export const createMarketAction: StateCreator<
   deactivateAgent: () => {
     set({ currentIdentifier: '' }, false, 'deactivateAgent');
   },
+
+  getAgentById: async (identifier) => {
+    const locale = globalHelpers.getCurrentLanguage();
+    return marketService.getAgentManifest(identifier, locale as string);
+  },
+
   setSearchKeywords: (keywords) => {
     set({ searchKeywords: keywords });
   },

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>;