@lobehub/lobehub 2.0.0-next.14 → 2.0.0-next.16

package/src/store/user/slices/modelList/reducers/customModelCard.test.ts

@@ -1,204 +0,0 @@
-import { describe, expect, it } from 'vitest';
-
-import { ChatModelCard } from '@/types/llm';
-
-import {
-  AddCustomModelCard,
-  DeleteCustomModelCard,
-  UpdateCustomModelCard,
-  customModelCardsReducer,
-} from './customModelCard';
-
-describe('customModelCardsReducer', () => {
-  const initialState: ChatModelCard[] = [
-    {
-      id: 'model1',
-      displayName: 'Model 1',
-      description: 'A helpful assistant',
-      files: true,
-      functionCall: false,
-      enabled: true,
-      isCustom: true,
-      legacy: false,
-      maxOutput: 1000,
-      contextWindowTokens: 2048,
-      vision: false,
-    },
-    {
-      id: 'model2',
-      displayName: 'Model 2',
-      description: 'A friendly chatbot',
-      files: false,
-      functionCall: true,
-      isCustom: true,
-      legacy: true,
-      maxOutput: 500,
-      contextWindowTokens: 1024,
-      vision: true,
-    },
-  ];
-
-  it('should add a new custom model card', () => {
-    const newModelCard: ChatModelCard = {
-      id: 'model3',
-      displayName: 'Model 3',
-      description: 'A versatile assistant',
-      files: true,
-      functionCall: true,
-      enabled: true,
-      isCustom: true,
-      legacy: false,
-      maxOutput: 2000,
-      contextWindowTokens: 4096,
-      vision: false,
-    };
-
-    const action: AddCustomModelCard = {
-      type: 'add',
-      modelCard: newModelCard,
-    };
-
-    const newState = customModelCardsReducer(initialState, action);
-
-    expect(newState).toContainEqual(newModelCard);
-    expect(newState.length).toBe(initialState.length + 1);
-  });
-
-  it('should not add a duplicate custom model card', () => {
-    const duplicateModelCard: ChatModelCard = {
-      id: 'model1',
-      displayName: 'Duplicate Model 1',
-      description: 'A duplicate model',
-      files: true,
-      functionCall: false,
-      enabled: true,
-      isCustom: true,
-      legacy: false,
-      maxOutput: 1000,
-      contextWindowTokens: 2048,
-      vision: false,
-    };
-
-    const action: AddCustomModelCard = {
-      type: 'add',
-      modelCard: duplicateModelCard,
-    };
-
-    const newState = customModelCardsReducer(initialState, action);
-
-    expect(newState).toEqual(initialState);
-  });
-
-  it('should delete a custom model card', () => {
-    const action: DeleteCustomModelCard = {
-      type: 'delete',
-      id: 'model1',
-    };
-
-    const newState = customModelCardsReducer(initialState, action);
-
-    expect(newState).not.toContainEqual(initialState[0]);
-    expect(newState.length).toBe(initialState.length - 1);
-  });
-
-  it('should update a custom model card', () => {
-    const action: UpdateCustomModelCard = {
-      type: 'update',
-      id: 'model1',
-      value: { displayName: 'Updated Model 1' },
-    };
-
-    const newState = customModelCardsReducer(initialState, action);
-
-    expect(newState.find((card) => card.id === 'model1')?.displayName).toBe('Updated Model 1');
-    expect(newState.length).toBe(initialState.length);
-  });
-
-  it('should throw an error for unhandled action type', () => {
-    const invalidAction = {
-      type: 'invalid',
-    };
-
-    expect(() => customModelCardsReducer(initialState, invalidAction as any)).toThrowError(
-      'Unhandled action type in customModelCardsReducer',
-    );
-  });
-
-  it('should return the original state if the model card is not found during update', () => {
-    const action: UpdateCustomModelCard = {
-      type: 'update',
-      id: 'nonexistent',
-      value: { displayName: 'Updated Nonexistent Model' },
-    };
-
-    const newState = customModelCardsReducer(initialState, action);
-
-    expect(newState).toEqual(initialState);
-  });
-
-  it('should return the original state if the model card ID is missing during add', () => {
-    const newModelCard: ChatModelCard = {
-      id: '',
-      displayName: 'Model 4',
-      description: 'A new model',
-      files: false,
-      functionCall: false,
-      enabled: true,
-      isCustom: true,
-      legacy: false,
-      maxOutput: 1500,
-      contextWindowTokens: 2048,
-      vision: false,
-    };
-
-    const action: AddCustomModelCard = {
-      type: 'add',
-      modelCard: newModelCard,
-    };
-
-    const newState = customModelCardsReducer(initialState, action);
-
-    expect(newState).toEqual(initialState);
-  });
-
-  it('should handle optional properties correctly', () => {
-    const newModelCard: ChatModelCard = {
-      id: 'model4',
-    };
-
-    const action: AddCustomModelCard = {
-      type: 'add',
-      modelCard: newModelCard,
-    };
-
-    const newState = customModelCardsReducer(initialState, action);
-
-    expect(newState).toContainEqual(newModelCard);
-  });
-
-  it('should handle an undefined initial state', () => {
-    const newModelCard: ChatModelCard = {
-      id: 'model4',
-      displayName: 'Model 4',
-      description: 'A new model',
-      files: false,
-      functionCall: false,
-      enabled: true,
-      isCustom: true,
-      legacy: false,
-      maxOutput: 1500,
-      contextWindowTokens: 2048,
-      vision: false,
-    };
-
-    const action: AddCustomModelCard = {
-      type: 'add',
-      modelCard: newModelCard,
-    };
-
-    const newState = customModelCardsReducer(undefined, action);
-
-    expect(newState).toContainEqual(newModelCard);
-    expect(newState.length).toBe(1);
-  });
-});

package/src/store/user/slices/modelList/reducers/customModelCard.ts

@@ -1,64 +0,0 @@
-import { produce } from 'immer';
-
-import { ChatModelCard } from '@/types/llm';
-
-export interface AddCustomModelCard {
-  modelCard: ChatModelCard;
-  type: 'add';
-}
-
-export interface DeleteCustomModelCard {
-  id: string;
-  type: 'delete';
-}
-
-export interface UpdateCustomModelCard {
-  id: string;
-  type: 'update';
-  value: Partial<ChatModelCard>;
-}
-
-export type CustomModelCardDispatch =
-  | AddCustomModelCard
-  | DeleteCustomModelCard
-  | UpdateCustomModelCard;
-
-export const customModelCardsReducer = (
-  state: ChatModelCard[] | undefined,
-  payload: CustomModelCardDispatch,
-): ChatModelCard[] => {
-  switch (payload.type) {
-    case 'add': {
-      return produce(state || [], (draftState) => {
-        const { id } = payload.modelCard;
-        if (!id) return;
-        if (draftState.some((card) => card.id === id)) return;
-
-        draftState.push(payload.modelCard);
-      });
-    }
-
-    case 'delete': {
-      return produce(state || [], (draftState) => {
-        const index = draftState.findIndex((card) => card.id === payload.id);
-        if (index !== -1) {
-          draftState.splice(index, 1);
-        }
-      });
-    }
-
-    case 'update': {
-      return produce(state || [], (draftState) => {
-        const index = draftState.findIndex((card) => card.id === payload.id);
-        if (index !== -1) {
-          const card = draftState[index];
-          Object.assign(card, payload.value);
-        }
-      });
-    }
-
-    default: {
-      throw new Error('Unhandled action type in customModelCardsReducer');
-    }
-  }
-};

package/src/store/user/slices/modelList/selectors/index.ts

@@ -1,3 +0,0 @@
-export { keyVaultsConfigSelectors } from './keyVaults';
-export { modelConfigSelectors } from './modelConfig';
-export { modelProviderSelectors } from './modelProvider';

package/src/store/user/slices/modelList/selectors/keyVaults.test.ts

@@ -1,201 +0,0 @@
-import { describe, expect, it } from 'vitest';
-
-import { UserStore } from '@/store/user';
-import {
-  AWSBedrockKeyVault,
-  AzureOpenAIKeyVault,
-  OpenAICompatibleKeyVault,
-} from '@/types/user/settings';
-import { merge } from '@/utils/merge';
-
-import { initialSettingsState } from '../../settings/initialState';
-import { keyVaultsConfigSelectors } from './keyVaults';
-
-describe('keyVaultsConfigSelectors', () => {
-  describe('isProviderEndpointNotEmpty', () => {
-    describe('OpenAICompatibleKeyVault', () => {
-      it('should return true if provider endpoint is not empty', () => {
-        const s = merge(initialSettingsState, {
-          settings: {
-            keyVaults: {
-              openai: {
-                endpoint: 'endpoint',
-              } as OpenAICompatibleKeyVault,
-            },
-          },
-        }) as unknown as UserStore;
-        expect(keyVaultsConfigSelectors.isProviderEndpointNotEmpty('openai')(s)).toBe(true);
-      });
-
-      it('should return false if provider endpoint is empty', () => {
-        const s = merge(initialSettingsState, {
-          settings: {
-            keyVaults: {
-              openai: {
-                endpoint: undefined,
-              } as OpenAICompatibleKeyVault,
-            },
-          },
-        }) as unknown as UserStore;
-        expect(keyVaultsConfigSelectors.isProviderEndpointNotEmpty('openai')(s)).toBe(false);
-      });
-    });
-
-    describe('AzureOpenAIKeyVault', () => {
-      it('should return true if provider endpoint is not empty', () => {
-        const s = merge(initialSettingsState, {
-          settings: {
-            keyVaults: {
-              azure: {
-                baseURL: 'baseURL',
-              } as AzureOpenAIKeyVault,
-            },
-          },
-        }) as unknown as UserStore;
-        expect(keyVaultsConfigSelectors.isProviderEndpointNotEmpty('azure')(s)).toBe(true);
-      });
-
-      it('should return false if provider endpoint is empty', () => {
-        const s = merge(initialSettingsState, {
-          settings: {
-            keyVaults: {
-              azure: {
-                baseURL: undefined,
-              } as AzureOpenAIKeyVault,
-            },
-          },
-        }) as unknown as UserStore;
-        expect(keyVaultsConfigSelectors.isProviderEndpointNotEmpty('azure')(s)).toBe(false);
-      });
-    });
-
-    // Always return false for AWSBedrockKeyVault
-    describe('AWSBedrockKeyVault', () => {
-      it('should return false if provider region is not empty for AWSBedrockKeyVault', () => {
-        const s = merge(initialSettingsState, {
-          settings: {
-            keyVaults: {
-              bedrock: {
-                region: 'region',
-              } as AWSBedrockKeyVault,
-            },
-          },
-        }) as unknown as UserStore;
-        expect(keyVaultsConfigSelectors.isProviderEndpointNotEmpty('bedrock')(s)).toBe(false);
-      });
-
-      it('should return false if provider region is empty for AWSBedrockKeyVault', () => {
-        const s = merge(initialSettingsState, {
-          settings: {
-            keyVaults: {
-              bedrock: {
-                region: undefined,
-              } as AWSBedrockKeyVault,
-            },
-          },
-        }) as unknown as UserStore;
-        expect(keyVaultsConfigSelectors.isProviderEndpointNotEmpty('bedrock')(s)).toBe(false);
-      });
-    });
-  });
-
-  describe('isProviderApiKeyNotEmpty', () => {
-    describe('OpenAICompatibleKeyVault', () => {
-      it('should return true if provider apikey is not empty', () => {
-        const s = merge(initialSettingsState, {
-          settings: {
-            keyVaults: {
-              openai: {
-                apiKey: 'apikey',
-              } as OpenAICompatibleKeyVault,
-            },
-          },
-        }) as unknown as UserStore;
-        expect(keyVaultsConfigSelectors.isProviderApiKeyNotEmpty('openai')(s)).toBe(true);
-      });
-
-      it('should return false if provider apikey is empty', () => {
-        const s = merge(initialSettingsState, {
-          settings: {
-            keyVaults: {
-              openai: {
-                apiKey: undefined,
-              } as OpenAICompatibleKeyVault,
-            },
-          },
-        }) as unknown as UserStore;
-        expect(keyVaultsConfigSelectors.isProviderApiKeyNotEmpty('openai')(s)).toBe(false);
-      });
-    });
-
-    describe('AzureOpenAIKeyVault', () => {
-      it('should return true if provider apikey is not empty', () => {
-        const s = merge(initialSettingsState, {
-          settings: {
-            keyVaults: {
-              azure: {
-                apiKey: 'apikey',
-              } as AzureOpenAIKeyVault,
-            },
-          },
-        }) as unknown as UserStore;
-        expect(keyVaultsConfigSelectors.isProviderApiKeyNotEmpty('azure')(s)).toBe(true);
-      });
-
-      it('should return false if provider apikey is empty', () => {
-        const s = merge(initialSettingsState, {
-          settings: {
-            keyVaults: {
-              azure: {
-                apiKey: undefined,
-              } as AzureOpenAIKeyVault,
-            },
-          },
-        }) as unknown as UserStore;
-        expect(keyVaultsConfigSelectors.isProviderApiKeyNotEmpty('azure')(s)).toBe(false);
-      });
-    });
-
-    describe('AWSBedrockKeyVault', () => {
-      it('should return true if provider accessKeyId is not empty for AWSBedrockKeyVault', () => {
-        const s = merge(initialSettingsState, {
-          settings: {
-            keyVaults: {
-              bedrock: {
-                accessKeyId: 'accessKeyId',
-              } as AWSBedrockKeyVault,
-            },
-          },
-        }) as unknown as UserStore;
-        expect(keyVaultsConfigSelectors.isProviderApiKeyNotEmpty('bedrock')(s)).toBe(true);
-      });
-
-      it('should return true if provider secretAccessKey is not empty for AWSBedrockKeyVault', () => {
-        const s = merge(initialSettingsState, {
-          settings: {
-            keyVaults: {
-              bedrock: {
-                secretAccessKey: 'secretAccessKey',
-              } as AWSBedrockKeyVault,
-            },
-          },
-        }) as unknown as UserStore;
-        expect(keyVaultsConfigSelectors.isProviderApiKeyNotEmpty('bedrock')(s)).toBe(true);
-      });
-
-      it('should return false if provider accessKeyId and secretAccessKey are both empty for AWSBedrockKeyVault', () => {
-        const s = merge(initialSettingsState, {
-          settings: {
-            keyVaults: {
-              bedrock: {
-                accessKeyId: undefined,
-                secretAccessKey: undefined,
-              } as AWSBedrockKeyVault,
-            },
-          },
-        }) as unknown as UserStore;
-        expect(keyVaultsConfigSelectors.isProviderApiKeyNotEmpty('bedrock')(s)).toBe(false);
-      });
-    });
-  });
-});

package/src/store/user/slices/modelList/selectors/keyVaults.ts

@@ -1,50 +0,0 @@
-import { UserStore } from '@/store/user';
-import {
-  AWSBedrockKeyVault,
-  AzureOpenAIKeyVault,
-  ComfyUIKeyVault,
-  GlobalLLMProviderKey,
-  OpenAICompatibleKeyVault,
-  UserKeyVaults,
-} from '@/types/user/settings';
-
-import { currentSettings } from '../../settings/selectors/settings';
-
-export const keyVaultsSettings = (s: UserStore): UserKeyVaults =>
-  currentSettings(s).keyVaults || {};
-
-const openAIConfig = (s: UserStore) => keyVaultsSettings(s).openai || {};
-const bedrockConfig = (s: UserStore) => keyVaultsSettings(s).bedrock || {};
-const ollamaConfig = (s: UserStore) => keyVaultsSettings(s).ollama || {};
-const azureConfig = (s: UserStore) => keyVaultsSettings(s).azure || {};
-const cloudflareConfig = (s: UserStore) => keyVaultsSettings(s).cloudflare || {};
-const getVaultByProvider = (provider: GlobalLLMProviderKey) => (s: UserStore) =>
-  (keyVaultsSettings(s)[provider] || {}) as OpenAICompatibleKeyVault &
-    AzureOpenAIKeyVault &
-    AWSBedrockKeyVault &
-    ComfyUIKeyVault;
-
-const isProviderEndpointNotEmpty = (provider: string) => (s: UserStore) => {
-  const vault = getVaultByProvider(provider as GlobalLLMProviderKey)(s);
-  return !!vault?.baseURL || !!vault?.endpoint;
-};
-
-const isProviderApiKeyNotEmpty = (provider: string) => (s: UserStore) => {
-  const vault = getVaultByProvider(provider as GlobalLLMProviderKey)(s);
-  return !!vault?.apiKey || !!vault?.accessKeyId || !!vault?.secretAccessKey;
-};
-
-const password = (s: UserStore) => keyVaultsSettings(s).password || '';
-
-export const keyVaultsConfigSelectors = {
-  azureConfig,
-  bedrockConfig,
-  cloudflareConfig,
-  getVaultByProvider,
-  isProviderApiKeyNotEmpty,
-  isProviderEndpointNotEmpty,
-  keyVaultsSettings,
-  ollamaConfig,
-  openAIConfig,
-  password,
-};