@lobehub/chat 1.119.1 → 1.119.2

package/packages/utils/src/number.test.ts

@@ -1,7 +1,6 @@
+import { MAX_SEED } from 'model-bank';
 import { describe, expect, it, vi } from 'vitest';
 
-import { MAX_SEED } from '@/const/image';
-
 import { calculateThumbnailDimensions, generateUniqueSeeds } from './number';
 
 describe('number utilities', () => {

package/packages/utils/src/number.ts

@@ -1,7 +1,6 @@
+import { IMAGE_GENERATION_CONFIG, MAX_SEED } from 'model-bank';
 import prand from 'pure-rand';
 
-import { IMAGE_GENERATION_CONFIG, MAX_SEED } from '@/const/image';
-
 export function generateUniqueSeeds(seedCount: number): number[] {
   // Use current timestamp as the initial seed
   const initialSeed = Date.now();

package/packages/utils/src/parseModels.test.ts

@@ -1,8 +1,7 @@
 import { LOBE_DEFAULT_MODEL_LIST, openaiChatModels } from 'model-bank';
+import { AiFullModelCard } from 'model-bank';
 import { describe, expect, it } from 'vitest';
 
-import { AiFullModelCard } from '@/types/aiModel';
-
 import { extractEnabledModels, parseModelString, transformToAiModelList } from './parseModels';
 
 describe('parseModelString', () => {

package/packages/utils/src/parseModels.ts

@@ -1,8 +1,7 @@
+import { getModelPropertyWithFallback } from '@lobechat/model-runtime';
 import { produce } from 'immer';
+import { AiFullModelCard, AiModelType } from 'model-bank';
 
-import { AiFullModelCard, AiModelType } from '@/types/aiModel';
-
-import { getModelPropertyWithFallback } from './getFallbackModelProperty';
 import { merge } from './merge';
 
 /**

package/packages/utils/src/pricing.test.ts

@@ -1,7 +1,6 @@
+import { Pricing } from 'model-bank';
 import { describe, expect, it } from 'vitest';
 
-import { Pricing } from '@/types/aiModel';
-
 import {
   getAudioInputUnitRate,
   getAudioOutputUnitRate,

package/packages/utils/src/pricing.ts

@@ -1,4 +1,4 @@
-import { Pricing, PricingUnit, PricingUnitName } from '@/types/aiModel';
+import { Pricing, PricingUnit, PricingUnitName } from 'model-bank';
 
 /**
  * Internal helper to extract the displayed unit rate from a pricing unit by strategy

package/packages/utils/src/server/xor.ts

@@ -1,4 +1,6 @@
-import { ClientSecretPayload, SECRET_XOR_KEY } from '@/const/auth';
+import { ClientSecretPayload } from '@lobechat/types';
+
+import { SECRET_XOR_KEY } from '@/const/auth';
 
 /**
  * 将 Base64 字符串转换为 Uint8Array

package/src/app/(backend)/middleware/auth/index.ts

@@ -4,12 +4,11 @@ import {
   ChatCompletionErrorPayload,
   ModelRuntime,
 } from '@lobechat/model-runtime';
-import { ChatErrorType } from '@lobechat/types';
+import { ChatErrorType, ClientSecretPayload } from '@lobechat/types';
 import { getXorPayload } from '@lobechat/utils/server';
 import { NextRequest } from 'next/server';
 
 import {
-  ClientSecretPayload,
   LOBE_CHAT_AUTH_HEADER,
   LOBE_CHAT_OIDC_AUTH_HEADER,
   OAUTH_AUTHORIZED,

package/src/app/(backend)/webapi/chat/vertexai/route.ts

@@ -1,7 +1,7 @@
 import { ModelProvider, ModelRuntime } from '@lobechat/model-runtime';
+import { LobeVertexAI } from '@lobechat/model-runtime/vertexai';
 
 import { checkAuth } from '@/app/(backend)/middleware/auth';
-import { LobeVertexAI } from '@/libs/model-runtime/vertexai';
 import { safeParseJSON } from '@/utils/safeParseJSON';
 
 import { POST as UniverseRoute } from '../[provider]/route';

package/src/app/(backend)/webapi/text-to-image/[provider]/route.ts

@@ -1,9 +1,8 @@
-import { ChatCompletionErrorPayload } from '@lobechat/model-runtime';
+import { ChatCompletionErrorPayload, TextToImagePayload } from '@lobechat/model-runtime';
 import { ChatErrorType } from '@lobechat/types';
 import { NextResponse } from 'next/server';
 
 import { checkAuth } from '@/app/(backend)/middleware/auth';
-import { TextToImagePayload } from '@/libs/model-runtime/types';
 import { initModelRuntimeWithUserPayload } from '@/server/modules/ModelRuntime';
 import { createErrorResponse } from '@/utils/errorResponse';
 

package/src/app/[variants]/(main)/discover/(list)/model/features/List/ModelTypeIcon.tsx

@@ -12,10 +12,9 @@ import {
   PhoneIcon,
   VideoIcon,
 } from 'lucide-react';
+import { AiModelType } from 'model-bank';
 import { memo } from 'react';
 
-import { AiModelType } from '@/types/aiModel';
-
 const icons: Record<AiModelType, LucideIcon> = {
   chat: MessageSquareTextIcon,
   embedding: BoltIcon,

package/src/app/[variants]/(main)/image/@menu/components/SeedNumberInput/index.tsx

@@ -2,11 +2,11 @@
 
 import { Button, InputNumber, Tooltip } from '@lobehub/ui';
 import { Dices } from 'lucide-react';
+import { MAX_SEED } from 'model-bank';
 import { CSSProperties, memo, useCallback } from 'react';
 import { useTranslation } from 'react-i18next';
 import { Flexbox } from 'react-layout-kit';
 
-import { MAX_SEED } from '@/const/image';
 import { generateUniqueSeeds } from '@/utils/number';
 
 export interface SeedNumberInputProps {

package/src/app/[variants]/(main)/image/@menu/features/ConfigPanel/hooks/useAutoDimensions.ts

@@ -1,8 +1,9 @@
-import { DEFAULT_DIMENSION_CONSTRAINTS } from '@lobechat/const';
+import { DEFAULT_DIMENSION_CONSTRAINTS } from 'model-bank';
 
-import { constrainDimensions } from '@/app/[variants]/(main)/image/@menu/features/ConfigPanel/utils/dimensionConstraints';
 import { useImageStore } from '@/store/image';
-import { imageGenerationConfigSelectors } from '@/store/image/slices/generationConfig/selectors';
+import { imageGenerationConfigSelectors } from '@/store/image/selectors';
+
+import { constrainDimensions } from '../utils/dimensionConstraints';
 
 /**
  * Extract URL and dimensions from callback data (supports both old and new API)

package/src/app/[variants]/(main)/settings/provider/features/ModelList/CreateNewModelModal/Form.tsx

@@ -5,7 +5,7 @@ import { useTranslation } from 'react-i18next';
 
 import MaxTokenSlider from '@/components/MaxTokenSlider';
 import { useIsMobile } from '@/hooks/useIsMobile';
-import { AiModelType } from '@/types/aiModel';
+import { AiModelType } from '../../../../../../../../../packages/model-bank/src/types/aiModel';
 import { ChatModelCard } from '@/types/llm';
 
 interface ModelConfigFormProps {

package/src/app/[variants]/(main)/settings/provider/features/ModelList/ModelItem.tsx

@@ -10,7 +10,7 @@ import { Flexbox } from 'react-layout-kit';
 import { ModelInfoTags } from '@/components/ModelSelect';
 import { useIsMobile } from '@/hooks/useIsMobile';
 import { aiModelSelectors, useAiInfraStore } from '@/store/aiInfra';
-import { AiModelSourceEnum, AiProviderModelListItem } from '@/types/aiModel';
+import { AiModelSourceEnum, AiProviderModelListItem } from '../../../../../../../../packages/model-bank/src/types/aiModel';
 import { formatPriceByCurrency } from '@/utils/format';
 import {
   getAudioInputUnitRate,

package/src/app/[variants]/(main)/settings/provider/features/ModelList/SortModelModal/ListItem.tsx

@@ -3,7 +3,7 @@ import { SortableList } from '@lobehub/ui';
 import { memo } from 'react';
 import { Flexbox } from 'react-layout-kit';
 
-import { AiProviderModelListItem } from '@/types/aiModel';
+import { AiProviderModelListItem } from '../../../../../../../../../packages/model-bank/src/types/aiModel';
 
 const ListItem = memo<AiProviderModelListItem>(({ id, displayName }) => {
   return (

package/src/app/[variants]/(main)/settings/provider/features/ModelList/SortModelModal/index.tsx

@@ -6,7 +6,7 @@ import { useTranslation } from 'react-i18next';
 import { Flexbox } from 'react-layout-kit';
 
 import { useAiInfraStore } from '@/store/aiInfra';
-import { AiProviderModelListItem } from '@/types/aiModel';
+import { AiProviderModelListItem } from '../../../../../../../../../packages/model-bank/src/types/aiModel';
 
 import ListItem from './ListItem';
 

package/src/components/ModelSelect/index.tsx

@@ -15,7 +15,7 @@ import { FC, memo } from 'react';
 import { useTranslation } from 'react-i18next';
 import { Flexbox } from 'react-layout-kit';
 
-import { ModelAbilities } from '@/types/aiModel';
+import { ModelAbilities } from '../../../packages/model-bank/src/types/aiModel';
 import { AiProviderSourceType } from '@/types/aiProvider';
 import { ChatModelCard } from '@/types/llm';
 import { formatTokenNumber } from '@/utils/format';

package/src/database/_deprecated/core/migrations/migrateSettingsToUser/type.ts

@@ -1,9 +1,6 @@
+import { FewShots, LobeAgentTTSConfig, MetaData, STTServer } from '@lobechat/types';
 import type { ThemeMode } from 'antd-style';
-
-import { LobeAgentTTSConfig } from '@/types/agent';
-import { FewShots, LLMParams } from '@/types/llm';
-import { MetaData } from '@/types/meta';
-import { STTServer } from '@/types/user/settings';
+import { LLMParams } from 'model-bank';
 
 interface V4LobeAgentConfig {
   autoCreateTopicThreshold: number;

package/src/features/Conversation/Extras/Usage/UsageDetail/ModelCard.tsx

@@ -9,7 +9,7 @@ import { Flexbox } from 'react-layout-kit';
 import { getPrice } from '@/features/Conversation/Extras/Usage/UsageDetail/pricing';
 import { useGlobalStore } from '@/store/global';
 import { systemStatusSelectors } from '@/store/global/selectors';
-import { LobeDefaultAiModelListItem } from '@/types/aiModel';
+import { LobeDefaultAiModelListItem } from '../../../../../../packages/model-bank/src/types/aiModel';
 import { getCachedTextInputUnitRate, getWriteCacheInputUnitRate } from '@/utils/pricing';
 
 export const useStyles = createStyles(({ css, token }) => {

package/src/features/Conversation/Extras/Usage/UsageDetail/pricing.ts

@@ -1,12 +1,11 @@
-import { Pricing } from '@/types/aiModel';
-import { ModelPriceCurrency } from '@/types/llm';
-import { formatPriceByCurrency } from '@/utils/format';
 import {
+  formatPriceByCurrency,
   getCachedTextInputUnitRate,
   getTextInputUnitRate,
   getTextOutputUnitRate,
   getWriteCacheInputUnitRate,
-} from '@/utils/pricing';
+} from '@lobechat/utils';
+import { ModelPriceCurrency, Pricing } from 'model-bank';
 
 export const getPrice = (pricing: Pricing) => {
   const inputRate = getTextInputUnitRate(pricing);

package/src/features/Conversation/Extras/Usage/UsageDetail/tokens.test.ts

@@ -1,6 +1,6 @@
 import { describe, expect, it } from 'vitest';
 
-import { LobeDefaultAiModelListItem } from '@/types/aiModel';
+import { LobeDefaultAiModelListItem } from '../../../../../../packages/model-bank/src/types/aiModel';
 import { ModelTokensUsage } from '@/types/message';
 
 import { getDetailsToken } from './tokens';

package/src/features/Conversation/Extras/Usage/UsageDetail/tokens.ts

@@ -1,5 +1,6 @@
-import { LobeDefaultAiModelListItem } from '@/types/aiModel';
-import { ModelTokensUsage } from '@/types/message';
+import type { ModelTokensUsage } from '@lobechat/model-runtime';
+import { LobeDefaultAiModelListItem } from 'model-bank';
+
 import { getAudioInputUnitRate, getAudioOutputUnitRate } from '@/utils/pricing';
 
 import { getPrice } from './pricing';

package/src/libs/trpc/async/context.ts

@@ -1,7 +1,8 @@
+import { ClientSecretPayload } from '@lobechat/types';
 import debug from 'debug';
 import { NextRequest } from 'next/server';
 
-import { ClientSecretPayload, LOBE_CHAT_AUTH_HEADER } from '@/const/auth';
+import { LOBE_CHAT_AUTH_HEADER } from '@/const/auth';
 import { LobeChatDatabase } from '@/database/type';
 import { KeyVaultsGateKeeper } from '@/server/modules/KeyVaultsEncrypt';
 

package/src/libs/trpc/edge/context.ts

@@ -1,12 +1,8 @@
+import { ClientSecretPayload } from '@lobechat/types';
 import { User } from 'next-auth';
 import { NextRequest } from 'next/server';
 
-import {
-  ClientSecretPayload,
-  LOBE_CHAT_AUTH_HEADER,
-  enableClerk,
-  enableNextAuth,
-} from '@/const/auth';
+import { LOBE_CHAT_AUTH_HEADER, enableClerk, enableNextAuth } from '@/const/auth';
 import { ClerkAuth, IClerkAuth } from '@/libs/clerk-auth';
 
 export interface AuthContext {

package/src/libs/trpc/lambda/context.ts

@@ -1,10 +1,10 @@
+import { ClientSecretPayload } from '@lobechat/types';
 import { parse } from 'cookie';
 import debug from 'debug';
 import { User } from 'next-auth';
 import { NextRequest } from 'next/server';
 
 import {
-  ClientSecretPayload,
   LOBE_CHAT_AUTH_HEADER,
   LOBE_CHAT_OIDC_AUTH_HEADER,
   enableClerk,

package/src/migrations/FromV5ToV6/types/v5.ts

@@ -1,5 +1,5 @@
-import { LobeAgentTTSConfig } from '@/types/agent';
-import { FewShots, LLMParams } from '@/types/llm';
+import { FewShots, LobeAgentTTSConfig } from '@lobechat/types';
+import { LLMParams } from 'model-bank';
 
 /**
  * Lobe Agent

package/src/migrations/FromV5ToV6/types/v6.ts

@@ -1,4 +1,5 @@
-import { FewShots, LLMParams } from '@/types/llm';
+import { FewShots } from '@lobechat/types';
+import { LLMParams } from 'model-bank';
 
 export type TTSServer = 'openai' | 'edge' | 'microsoft';
 

package/src/server/globalConfig/genServerAiProviderConfig.ts

@@ -1,10 +1,10 @@
 import { ModelProvider } from '@lobechat/model-runtime';
+import { ProviderConfig } from '@lobechat/types';
+import { extractEnabledModels, transformToAiModelList } from '@lobechat/utils';
 import * as AiModels from 'model-bank';
+import { AiFullModelCard } from 'model-bank';
 
 import { getLLMConfig } from '@/config/llm';
-import { AiFullModelCard } from '@/types/aiModel';
-import { ProviderConfig } from '@/types/user/settings';
-import { extractEnabledModels, transformToAiModelList } from '@/utils/parseModels';
 
 interface ProviderSpecificConfig {
   enabled?: boolean;

package/src/server/modules/ModelRuntime/index.test.ts

@@ -21,9 +21,9 @@ import {
   ModelProvider,
 } from '@lobechat/model-runtime';
 import { ModelRuntime } from '@lobechat/model-runtime';
+import { ClientSecretPayload } from '@lobechat/types';
 import { describe, expect, it, vi } from 'vitest';
 
-import { ClientSecretPayload } from '@/const/auth';
 import { LobeStepfunAI } from '@/libs/model-runtime/stepfun';
 
 import { initModelRuntimeWithUserPayload } from './index';

package/src/server/modules/ModelRuntime/index.ts

@@ -1,7 +1,7 @@
 import { ModelProvider, ModelRuntime } from '@lobechat/model-runtime';
+import { ClientSecretPayload } from '@lobechat/types';
 
 import { getLLMConfig } from '@/config/llm';
-import { ClientSecretPayload } from '@/const/auth';
 
 import apiKeyManager from './apiKeyManager';
 

package/src/server/routers/async/caller.ts

@@ -1,9 +1,10 @@
+import { ClientSecretPayload } from '@lobechat/types';
 import { createTRPCClient, httpLink } from '@trpc/client';
 import superjson from 'superjson';
 import urlJoin from 'url-join';
 
 import { serverDBEnv } from '@/config/db';
-import { ClientSecretPayload, LOBE_CHAT_AUTH_HEADER } from '@/const/auth';
+import { LOBE_CHAT_AUTH_HEADER } from '@/const/auth';
 import { isDesktop } from '@/const/version';
 import { appEnv } from '@/envs/app';
 import { createAsyncCallerFactory } from '@/libs/trpc/async';

package/src/server/routers/async/image.ts

@@ -1,3 +1,5 @@
+import { AgentRuntimeErrorType } from '@lobechat/model-runtime';
+import { AsyncTaskError, AsyncTaskErrorType, AsyncTaskStatus } from '@lobechat/types';
 import debug from 'debug';
 import { RuntimeImageGenParams } from 'model-bank';
 import { z } from 'zod';
@@ -5,11 +7,9 @@ import { z } from 'zod';
 import { ASYNC_TASK_TIMEOUT, AsyncTaskModel } from '@/database/models/asyncTask';
 import { FileModel } from '@/database/models/file';
 import { GenerationModel } from '@/database/models/generation';
-import { AgentRuntimeErrorType } from '@/libs/model-runtime/error';
 import { asyncAuthedProcedure, asyncRouter as router } from '@/libs/trpc/async';
 import { initModelRuntimeWithUserPayload } from '@/server/modules/ModelRuntime';
 import { GenerationService } from '@/server/services/generation';
-import { AsyncTaskError, AsyncTaskErrorType, AsyncTaskStatus } from '@/types/asyncTask';
 
 const log = debug('lobe-image:async');
 

package/src/server/routers/lambda/aiModel.ts

@@ -12,7 +12,7 @@ import {
   CreateAiModelSchema,
   ToggleAiModelEnableSchema,
   UpdateAiModelSchema,
-} from '@/types/aiModel';
+} from '../../../../packages/model-bank/src/types/aiModel';
 import { ProviderConfig } from '@/types/user/settings';
 
 const aiModelProcedure = authedProcedure.use(serverDatabase).use(async (opts) => {

package/src/server/services/chunk/index.ts

@@ -1,4 +1,5 @@
-import { ClientSecretPayload } from '@/const/auth';
+import { ClientSecretPayload } from '@lobechat/types';
+
 import { AsyncTaskModel } from '@/database/models/asyncTask';
 import { FileModel } from '@/database/models/file';
 import { LobeChatDatabase } from '@/database/type';

package/src/server/services/generation/index.ts

@@ -1,12 +1,12 @@
+import { parseDataUri } from '@lobechat/model-runtime';
 import debug from 'debug';
 import { sha256 } from 'js-sha256';
 import mime from 'mime';
+import { IMAGE_GENERATION_CONFIG } from 'model-bank';
 import { nanoid } from 'nanoid';
 import sharp from 'sharp';
 
-import { IMAGE_GENERATION_CONFIG } from '@/const/image';
 import { LobeChatDatabase } from '@/database/type';
-import { parseDataUri } from '@/libs/model-runtime/utils/uriParser';
 import { FileService } from '@/server/services/file';
 import { calculateThumbnailDimensions } from '@/utils/number';
 import { getYYYYmmddHHMMss } from '@/utils/time';

package/src/services/_auth.ts

@@ -1,6 +1,7 @@
 import { ModelProvider } from '@lobechat/model-runtime';
+import { ClientSecretPayload } from '@lobechat/types';
 
-import { ClientSecretPayload, LOBE_CHAT_AUTH_HEADER } from '@/const/auth';
+import { LOBE_CHAT_AUTH_HEADER } from '@/const/auth';
 import { isDeprecatedEdition } from '@/const/version';
 import { aiProviderSelectors, useAiInfraStore } from '@/store/aiInfra';
 import { useUserStore } from '@/store/user';

package/src/services/aiModel/server.test.ts

@@ -1,7 +1,7 @@
+import { AiProviderModelListItem } from 'model-bank';
 import { describe, expect, it, vi } from 'vitest';
 
 import { lambdaClient } from '@/libs/trpc/client';
-import { AiProviderModelListItem } from '@/types/aiModel';
 
 import { ServerService } from './server';
 

package/src/services/aiModel/type.ts

@@ -5,7 +5,7 @@ import {
   CreateAiModelParams,
   ToggleAiModelEnableParams,
   UpdateAiModelParams,
-} from '@/types/aiModel';
+} from 'model-bank';
 
 export interface IAiModelService {
   createAiModel: (params: CreateAiModelParams) => Promise<any>;

package/src/services/chat.ts

@@ -3,6 +3,7 @@ import {
   ChatCompletionErrorPayload,
   ModelProvider,
   ModelRuntime,
+  parseDataUri,
 } from '@lobechat/model-runtime';
 import { BuiltinSystemRolePrompts, filesPrompts } from '@lobechat/prompts';
 import { ChatErrorType, TracePayload, TraceTagMap } from '@lobechat/types';
@@ -15,7 +16,6 @@ import { INBOX_GUIDE_SYSTEMROLE } from '@/const/guide';
 import { INBOX_SESSION_ID } from '@/const/session';
 import { DEFAULT_AGENT_CONFIG } from '@/const/settings';
 import { isDeprecatedEdition, isDesktop, isServerMode } from '@/const/version';
-import { parseDataUri } from '@/libs/model-runtime/utils/uriParser';
 import { getAgentStoreState } from '@/store/agent';
 import { agentChatConfigSelectors } from '@/store/agent/selectors';
 import { aiModelSelectors, aiProviderSelectors, getAiInfraStoreState } from '@/store/aiInfra';

package/src/services/upload.ts

@@ -1,15 +1,15 @@
+import { isDesktop, isServerMode } from '@lobechat/const';
+import { parseDataUri } from '@lobechat/model-runtime';
+import { uuid } from '@lobechat/utils';
 import dayjs from 'dayjs';
 import { sha256 } from 'js-sha256';
 
 import { fileEnv } from '@/config/file';
-import { isDesktop, isServerMode } from '@/const/version';
-import { parseDataUri } from '@/libs/model-runtime/utils/uriParser';
 import { edgeClient } from '@/libs/trpc/client';
 import { API_ENDPOINTS } from '@/services/_url';
 import { clientS3Storage } from '@/services/file/ClientS3';
 import { FileMetadata, UploadBase64ToS3Result } from '@/types/files';
 import { FileUploadState, FileUploadStatus } from '@/types/files/upload';
-import { uuid } from '@/utils/uuid';
 
 export const UPLOAD_NETWORK_ERROR = 'NetWorkError';
 

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

@@ -15,7 +15,7 @@ import { LobeAgentChatConfig, LobeAgentConfig } from '@/types/agent';
 import { KnowledgeItem } from '@/types/knowledgeBase';
 import { merge } from '@/utils/merge';
 
-import { AgentStore } from '../../store';
+import type { AgentStore } from '../../store';
 import { agentSelectors } from './selectors';
 
 /**

package/src/store/aiInfra/slices/aiModel/action.ts

@@ -1,16 +1,16 @@
 import isEqual from 'fast-deep-equal';
+import {
+  AiModelSortMap,
+  AiProviderModelListItem,
+  CreateAiModelParams,
+  ToggleAiModelEnableParams,
+} from 'model-bank';
 import { SWRResponse, mutate } from 'swr';
 import { StateCreator } from 'zustand/vanilla';
 
 import { useClientDataSWR } from '@/libs/swr';
 import { aiModelService } from '@/services/aiModel';
 import { AiInfraStore } from '@/store/aiInfra/store';
-import {
-  AiModelSortMap,
-  AiProviderModelListItem,
-  CreateAiModelParams,
-  ToggleAiModelEnableParams,
-} from '@/types/aiModel';
 
 const FETCH_AI_PROVIDER_MODEL_LIST_KEY = 'FETCH_AI_PROVIDER_MODELS';
 

package/src/store/aiInfra/slices/aiModel/initialState.ts

@@ -1,4 +1,4 @@
-import { AiProviderModelListItem, LobeDefaultAiModelListItem } from '@/types/aiModel';
+import { AiProviderModelListItem, LobeDefaultAiModelListItem } from 'model-bank';
 
 export interface AIModelsState {
   aiModelLoadingIds: string[];

package/src/store/aiInfra/slices/aiModel/selectors.test.ts

@@ -1,7 +1,7 @@
+import { AiModelSourceEnum } from 'model-bank';
 import { describe, expect, it } from 'vitest';
 
 import { AIProviderStoreState } from '@/store/aiInfra/initialState';
-import { AiModelSourceEnum } from '@/types/aiModel';
 
 import { aiModelSelectors } from './selectors';
 

package/src/store/aiInfra/slices/aiModel/selectors.ts

@@ -1,5 +1,6 @@
+import { AiModelSourceEnum } from 'model-bank';
+
 import { AIProviderStoreState } from '@/store/aiInfra/initialState';
-import { AiModelSourceEnum } from '@/types/aiModel';
 import { ModelSearchImplement } from '@/types/search';
 
 const aiProviderChatModelListIds = (s: AIProviderStoreState) =>

package/src/store/aiInfra/slices/aiProvider/__tests__/action.test.ts

@@ -1,14 +1,9 @@
+import * as runtimeModule from '@lobechat/model-runtime';
+import type { EnabledAiModel, ModelAbilities } from 'model-bank';
 import { describe, expect, it, vi } from 'vitest';
 
-import type { EnabledAiModel, ModelAbilities } from '@/types/aiModel';
-
 import { getModelListByType } from '../action';
 
-// Mock getModelPropertyWithFallback
-vi.mock('@/utils/getFallbackModelProperty', () => ({
-  getModelPropertyWithFallback: vi.fn().mockResolvedValue({ size: '1024x1024' }),
-}));
-
 describe('getModelListByType', () => {
   const mockChatModels: EnabledAiModel[] = [
     {
@@ -100,6 +95,11 @@ describe('getModelListByType', () => {
     });
 
     it('should use fallback parameters for image models without parameters', async () => {
+      // Mock getModelPropertyWithFallback
+      vi.spyOn(runtimeModule, 'getModelPropertyWithFallback').mockResolvedValueOnce({
+        size: '1024x1024',
+      });
+
       const result = await getModelListByType(allModels, 'midjourney', 'image');
 
       expect(result[0]).toEqual({

package/src/store/aiInfra/slices/aiProvider/action.ts

@@ -1,17 +1,18 @@
+import { isDeprecatedEdition, isDesktop, isUsePgliteDB } from '@lobechat/const';
+import { getModelPropertyWithFallback } from '@lobechat/model-runtime';
 import { uniqBy } from 'lodash-es';
+import {
+  AIImageModelCard,
+  EnabledAiModel,
+  LobeDefaultAiModelListItem,
+  ModelAbilities,
+} from 'model-bank';
 import { SWRResponse, mutate } from 'swr';
 import { StateCreator } from 'zustand/vanilla';
 
-import { isDeprecatedEdition, isDesktop, isUsePgliteDB } from '@/const/version';
 import { useClientDataSWR } from '@/libs/swr';
 import { aiProviderService } from '@/services/aiProvider';
 import { AiInfraStore } from '@/store/aiInfra/store';
-import {
-  AIImageModelCard,
-  EnabledAiModel,
-  LobeDefaultAiModelListItem,
-  ModelAbilities,
-} from '@/types/aiModel';
 import {
   AiProviderDetailItem,
   AiProviderListItem,
@@ -24,7 +25,6 @@ import {
   UpdateAiProviderConfigParams,
   UpdateAiProviderParams,
 } from '@/types/aiProvider';
-import { getModelPropertyWithFallback } from '@/utils/getFallbackModelProperty';
 
 /**
  * Get models by provider ID and type, with proper formatting and deduplication

package/src/store/aiInfra/slices/aiProvider/initialState.ts

@@ -1,4 +1,5 @@
-import { EnabledAiModel } from '@/types/aiModel';
+import { EnabledAiModel } from 'model-bank';
+
 import {
   AiProviderDetailItem,
   AiProviderListItem,

package/src/store/electron/actions/app.ts

@@ -8,7 +8,7 @@ import { electronSystemService } from '@/services/electron/system';
 import { globalAgentContextManager } from '@/utils/client/GlobalAgentContextManager';
 import { merge } from '@/utils/merge';
 
-import { ElectronStore } from '../store';
+import type { ElectronStore } from '../store';
 
 // ======== Action Interface ======== //