@lobehub/chat 1.118.4 → 1.118.5

package/packages/model-runtime/src/utils/modelParse.ts

@@ -25,16 +25,16 @@ export const MODEL_LIST_CONFIGS = {
     reasoningKeywords: ['thinking', '-2.5-'],
     visionKeywords: ['gemini', 'learnlm'],
   },
-  llama: {
-    functionCallKeywords: ['llama-3.2', 'llama-3.3', 'llama-4'],
-    reasoningKeywords: [],
-    visionKeywords: ['llava'],
-  },
   moonshot: {
     functionCallKeywords: ['moonshot', 'kimi'],
     reasoningKeywords: ['thinking'],
     visionKeywords: ['vision', 'kimi-latest', 'kimi-thinking-preview'],
   },
+  ollama: {
+    functionCallKeywords: ['llama-3.2', 'llama-3.3', 'llama-4'],
+    reasoningKeywords: [],
+    visionKeywords: ['llava'],
+  },
   openai: {
     excludeKeywords: ['audio'],
     functionCallKeywords: ['4o', '4.1', 'o3', 'o4', 'oss'],
@@ -86,8 +86,8 @@ export const PROVIDER_DETECTION_CONFIG = {
   anthropic: ['claude'],
   deepseek: ['deepseek'],
   google: ['gemini', 'imagen'],
-  llama: ['llama', 'llava'],
   moonshot: ['moonshot', 'kimi'],
+  ollama: ['llama', 'llava'],
   openai: ['o1', 'o3', 'o4', 'gpt-'],
   qwen: ['qwen', 'qwq', 'qvq'],
   v0: ['v0'],
@@ -119,12 +119,7 @@ export const IMAGE_MODEL_KEYWORDS = [
 ] as const;
 
 // 嵌入模型关键词配置
-export const EMBEDDING_MODEL_KEYWORDS = [
-  'embedding',
-  'embed',
-  'bge',
-  'm3e',
-] as const;
+export const EMBEDDING_MODEL_KEYWORDS = ['embedding', 'embed', 'bge', 'm3e'] as const;
 
 /**
  * 检测关键词列表是否匹配模型ID（支持多种匹配模式）
@@ -178,8 +173,9 @@ const findKnownModelByProvider = async (
 
   try {
     // 尝试动态导入对应的配置文件
-    const moduleImport = await import(`@/config/aiModels/${provider}.ts`);
-    const providerModels = moduleImport.default;
+    const modules = await import('model-bank');
+
+    const providerModels = modules[provider];
 
     // 如果导入成功且有数据，进行查找
     if (Array.isArray(providerModels)) {
@@ -287,9 +283,9 @@ const processModelCard = (
     )
       ? 'image'
       : isKeywordListMatch(
-        model.id.toLowerCase(),
-        EMBEDDING_MODEL_KEYWORDS.map((k) => k.toLowerCase()),
-      )
+            model.id.toLowerCase(),
+            EMBEDDING_MODEL_KEYWORDS.map((k) => k.toLowerCase()),
+          )
         ? 'embedding'
         : 'chat');
 
@@ -298,6 +294,34 @@ const processModelCard = (
     return undefined;
   }
 
+  const formatPricing = (pricing?: { input?: number; output?: number; units?: any[] }) => {
+    if (!pricing || typeof pricing !== 'object') return undefined;
+    if (Array.isArray(pricing.units)) {
+      return { units: pricing.units };
+    }
+    const { input, output } = pricing;
+    if (typeof input !== 'number' && typeof output !== 'number') return undefined;
+
+    const units = [];
+    if (typeof input === 'number') {
+      units.push({
+        name: 'textInput' as const,
+        rate: input,
+        strategy: 'fixed' as const,
+        unit: 'millionTokens' as const,
+      });
+    }
+    if (typeof output === 'number') {
+      units.push({
+        name: 'textOutput' as const,
+        rate: output,
+        strategy: 'fixed' as const,
+        unit: 'millionTokens' as const,
+      });
+    }
+    return { units };
+  };
+
   return {
     contextWindowTokens: model.contextWindowTokens ?? knownModel?.contextWindowTokens ?? undefined,
     description: model.description ?? knownModel?.description ?? '',
@@ -312,7 +336,7 @@ const processModelCard = (
         false),
     id: model.id,
     maxOutput: model.maxOutput ?? knownModel?.maxOutput ?? undefined,
-    // pricing: knownModel?.pricing ?? undefined,
+    pricing: formatPricing(model?.pricing) ?? undefined,
     reasoning:
       model.reasoning ??
       knownModel?.abilities?.reasoning ??
@@ -342,7 +366,7 @@ export const processModelList = async (
   config: ModelProcessorConfig,
   provider?: keyof typeof MODEL_LIST_CONFIGS,
 ): Promise<ChatModelCard[]> => {
-  const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
+  const { LOBE_DEFAULT_MODEL_LIST } = await import('model-bank');
 
   return Promise.all(
     modelList.map(async (model) => {
@@ -375,14 +399,15 @@ export const processMultiProviderModelList = async (
   modelList: Array<{ id: string }>,
   providerid?: ModelProviderKey,
 ): Promise<ChatModelCard[]> => {
-  const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
+  const { LOBE_DEFAULT_MODEL_LIST } = await import('model-bank');
 
   // 如果提供了 providerid，尝试获取该提供商的本地配置
   let providerLocalConfig: any[] | null = null;
   if (providerid) {
     try {
-      const moduleImport = await import(`@/config/aiModels/${providerid}.ts`);
-      providerLocalConfig = moduleImport.default;
+      const modules = await import('model-bank');
+
+      providerLocalConfig = modules[providerid];
     } catch {
       // 如果配置文件不存在或导入失败，保持为 null
       providerLocalConfig = null;

package/packages/model-runtime/src/utils/openaiCompatibleFactory/createImage.ts

@@ -1,9 +1,8 @@
 import { imageUrlToBase64 } from '@lobechat/utils';
 import createDebug from 'debug';
+import { RuntimeImageGenParamsValue } from 'model-bank';
 import OpenAI from 'openai';
 
-import { RuntimeImageGenParamsValue } from '@/libs/standard-parameters/index';
-
 import { CreateImagePayload, CreateImageResponse } from '../../types/image';
 import { convertImageUrlToFile } from '../openaiHelpers';
 import { parseDataUri } from '../uriParser';

package/packages/model-runtime/src/utils/openaiCompatibleFactory/index.ts

@@ -1,10 +1,10 @@
 import { getModelPropertyWithFallback } from '@lobechat/utils';
 import dayjs from 'dayjs';
 import utc from 'dayjs/plugin/utc';
+import { LOBE_DEFAULT_MODEL_LIST } from 'model-bank';
 import OpenAI, { ClientOptions } from 'openai';
 import { Stream } from 'openai/streaming';
 
-import { LOBE_DEFAULT_MODEL_LIST } from '@/config/aiModels';
 import type { AiModelType } from '@/types/aiModel';
 import type { ChatModelCard } from '@/types/llm';
 

package/packages/model-runtime/src/vllm/index.ts

@@ -13,7 +13,7 @@ export const LobeVLLMAI = createOpenAICompatibleRuntime({
     chatCompletion: () => process.env.DEBUG_VLLM_CHAT_COMPLETION === '1',
   },
   models: async ({ client }) => {
-    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
+    const { LOBE_DEFAULT_MODEL_LIST } = await import('model-bank');
 
     const modelsPage = (await client.models.list()) as any;
     const modelList: VLLMModelCard[] = modelsPage.data;

package/packages/model-runtime/src/xinference/index.ts

@@ -18,7 +18,7 @@ export const LobeXinferenceAI = createOpenAICompatibleRuntime({
     chatCompletion: () => process.env.DEBUG_XINFERENCE_CHAT_COMPLETION === '1',
   },
   models: async ({ client }) => {
-    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
+    const { LOBE_DEFAULT_MODEL_LIST } = await import('model-bank');
 
     const modelsPage = (await client.models.list()) as any;
     const modelList: XinferenceModelCard[] = modelsPage.data;

package/packages/types/src/aiModel.ts

@@ -1,7 +1,6 @@
+import { ModelParamsSchema } from 'model-bank';
 import { z } from 'zod';
 
-import { ModelParamsSchema } from '@/libs/standard-parameters';
-
 export type ModelPriceCurrency = 'CNY' | 'USD';
 
 export const AiModelSourceEnum = {

package/packages/types/src/llm.ts

@@ -1,6 +1,6 @@
+import { ModelParamsSchema } from 'model-bank';
 import { ReactNode } from 'react';
 
-import { ModelParamsSchema } from '@/libs/standard-parameters';
 import { AiModelType, Pricing } from '@/types/aiModel';
 import { AiProviderSettings } from '@/types/aiProvider';
 

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

@@ -3,7 +3,7 @@ import { describe, expect, it, vi } from 'vitest';
 import { getModelPropertyWithFallback } from './getFallbackModelProperty';
 
 // Mock LOBE_DEFAULT_MODEL_LIST for testing
-vi.mock('@/config/aiModels', () => ({
+vi.mock('model-bank', () => ({
   LOBE_DEFAULT_MODEL_LIST: [
     {
       id: 'gpt-4',

package/packages/utils/src/getFallbackModelProperty.ts

@@ -12,7 +12,7 @@ export const getModelPropertyWithFallback = async <T>(
   propertyName: keyof AiFullModelCard,
   providerId?: string,
 ): Promise<T> => {
-  const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
+  const { LOBE_DEFAULT_MODEL_LIST } = await import('model-bank');
 
   // Step 1: If providerId is provided, prioritize an exact match (same provider + same id)
   if (providerId) {

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

@@ -1,7 +1,6 @@
+import { LOBE_DEFAULT_MODEL_LIST, openaiChatModels } from 'model-bank';
 import { describe, expect, it } from 'vitest';
 
-import { LOBE_DEFAULT_MODEL_LIST } from '@/config/aiModels';
-import { openaiChatModels } from '@/config/aiModels/openai';
 import { AiFullModelCard } from '@/types/aiModel';
 
 import { extractEnabledModels, parseModelString, transformToAiModelList } from './parseModels';

package/packages/utils/src/parseModels.ts

@@ -141,7 +141,7 @@ export const transformToAiModelList = async ({
   }
 
   // 异步获取配置
-  const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
+  const { LOBE_DEFAULT_MODEL_LIST } = await import('model-bank');
 
   return produce(chatModels, (draft) => {
     // 处理添加或替换逻辑

package/src/app/[variants]/(main)/image/features/GenerationFeed/BatchItem.tsx

@@ -9,12 +9,12 @@ import dayjs from 'dayjs';
 import relativeTime from 'dayjs/plugin/relativeTime';
 import { omit } from 'lodash-es';
 import { CopyIcon, RotateCcwSquareIcon, Trash2 } from 'lucide-react';
+import { RuntimeImageGenParams } from 'model-bank';
 import { memo, useMemo } from 'react';
 import { useTranslation } from 'react-i18next';
 import { Flexbox } from 'react-layout-kit';
 
 import InvalidAPIKey from '@/components/InvalidAPIKey';
-import { RuntimeImageGenParams } from '@/libs/standard-parameters/index';
 import { useImageStore } from '@/store/image';
 import { AsyncTaskErrorType } from '@/types/asyncTask';
 import { GenerationBatch } from '@/types/generation';

package/src/app/[variants]/(main)/profile/features/ClerkProfile.tsx

@@ -48,7 +48,7 @@ export const useStyles = createStyles(
         height: 100%;
       `,
       scrollBox: css`
-        background: transparent;
+        background: transparent !important;
       `,
     }) as Partial<Record<keyof ElementsConfig, any>>,
 );

package/src/app/[variants]/(main)/settings/_layout/Desktop/index.tsx

@@ -2,7 +2,7 @@
 
 import { useResponsive, useTheme } from 'antd-style';
 import { usePathname } from 'next/navigation';
-import { PropsWithChildren, memo, useEffect, useRef } from 'react';
+import { PropsWithChildren, memo, useRef } from 'react';
 import { Flexbox } from 'react-layout-kit';
 
 import InitClientDB from '@/features/InitClientDB';
@@ -31,10 +31,6 @@ const Layout = memo<LayoutProps>(({ children, category }) => {
   const { md = true } = useResponsive();
   const theme = useTheme();
 
-  useEffect(() => {
-    console.log('settings render');
-  });
-
   return (
     <Flexbox
       height={'100%'}

package/src/locales/default/models.ts

@@ -1,4 +1,4 @@
-import { LOBE_DEFAULT_MODEL_LIST } from '@/config/aiModels';
+import { LOBE_DEFAULT_MODEL_LIST } from 'model-bank';
 
 const locales: {
   [key: string]: {

package/src/server/globalConfig/genServerAiProviderConfig.test.ts

@@ -3,8 +3,8 @@ import { beforeEach, describe, expect, it, vi } from 'vitest';
 import { genServerAiProvidersConfig } from './genServerAiProviderConfig';
 
 // Mock dependencies using importOriginal to preserve real provider data
-vi.mock('@/config/aiModels', async (importOriginal) => {
-  const actual = await importOriginal<typeof import('@/config/aiModels')>();
+vi.mock('model-bank', async (importOriginal) => {
+  const actual = await importOriginal<typeof import('model-bank')>();
   return {
     ...actual,
     // Keep the original exports but we can override specific ones if needed
@@ -189,7 +189,7 @@ describe('genServerAiProvidersConfig Error Handling', () => {
     vi.resetModules();
 
     // Mock dependencies with a missing provider scenario
-    vi.doMock('@/config/aiModels', () => ({
+    vi.doMock('model-bank', () => ({
       // Explicitly set openai to undefined to simulate missing provider
       openai: undefined,
       anthropic: [

package/src/server/globalConfig/genServerAiProviderConfig.ts

@@ -1,6 +1,6 @@
 import { ModelProvider } from '@lobechat/model-runtime';
+import * as AiModels from 'model-bank';
 
-import * as AiModels from '@/config/aiModels';
 import { getLLMConfig } from '@/config/llm';
 import { AiFullModelCard } from '@/types/aiModel';
 import { ProviderConfig } from '@/types/user/settings';

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

@@ -1,11 +1,11 @@
 import debug from 'debug';
+import { RuntimeImageGenParams } from 'model-bank';
 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 { RuntimeImageGenParams } from '@/libs/standard-parameters/index';
 import { asyncAuthedProcedure, asyncRouter as router } from '@/libs/trpc/async';
 import { initModelRuntimeWithUserPayload } from '@/server/modules/ModelRuntime';
 import { GenerationService } from '@/server/services/generation';

package/src/server/services/discover/index.test.ts

@@ -23,7 +23,7 @@ vi.mock('@/locales/resources', () => ({
 process.env.MARKET_BASE_URL = 'http://localhost:8787/api';
 
 // Mock constants with inline data
-vi.mock('@/config/aiModels', () => ({
+vi.mock('model-bank', () => ({
   LOBE_DEFAULT_MODEL_LIST: [
     {
       id: 'gpt-4',

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

@@ -43,7 +43,11 @@ import {
   ProviderQueryParams,
   ProviderSorts,
 } from '@/types/discover';
-import { getAudioInputUnitRate, getTextInputUnitRate, getTextOutputUnitRate } from '@/utils/pricing';
+import {
+  getAudioInputUnitRate,
+  getTextInputUnitRate,
+  getTextOutputUnitRate,
+} from '@/utils/pricing';
 
 const log = debug('lobe-server:discover');
 
@@ -728,7 +732,7 @@ export class DiscoverService {
   private _getProviderList = async (): Promise<DiscoverProviderItem[]> => {
     log('_getProviderList: fetching provider list');
     const [{ LOBE_DEFAULT_MODEL_LIST }, { DEFAULT_MODEL_PROVIDER_LIST }] = await Promise.all([
-      import('@/config/aiModels'),
+      import('model-bank'),
       import('@/config/modelProviders'),
     ]);
     const result = DEFAULT_MODEL_PROVIDER_LIST.map((item) => {
@@ -754,7 +758,7 @@ export class DiscoverService {
   }): Promise<DiscoverProviderDetail | undefined> => {
     log('getProviderDetail: params=%O', params);
     const { identifier, locale, withReadme } = params;
-    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
+    const { LOBE_DEFAULT_MODEL_LIST } = await import('model-bank');
     const all = await this._getProviderList();
     let provider = all.find((item) => item.identifier === identifier);
     if (!provider) {
@@ -890,7 +894,7 @@ export class DiscoverService {
 
   private _getRawModelList = async (): Promise<DiscoverModelItem[]> => {
     log('_getRawModelList: fetching raw model list');
-    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
+    const { LOBE_DEFAULT_MODEL_LIST } = await import('model-bank');
     const result = LOBE_DEFAULT_MODEL_LIST.map((item) => {
       const identifier = (item.id.split('/').at(-1) || item.id).toLowerCase();
       const providers = uniq(
@@ -983,7 +987,7 @@ export class DiscoverService {
   getModelCategories = async (params: CategoryListQuery = {}): Promise<CategoryItem[]> => {
     log('getModelCategories: params=%O', params);
     const { q } = params;
-    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
+    const { LOBE_DEFAULT_MODEL_LIST } = await import('model-bank');
     let list = LOBE_DEFAULT_MODEL_LIST;
     if (q) {
       const originalCount = list.length;
@@ -1018,7 +1022,7 @@ export class DiscoverService {
   }): Promise<DiscoverModelDetail | undefined> => {
     log('getModelDetail: params=%O', params);
     const [{ LOBE_DEFAULT_MODEL_LIST }, { DEFAULT_MODEL_PROVIDER_LIST }] = await Promise.all([
-      import('@/config/aiModels'),
+      import('model-bank'),
       import('@/config/modelProviders'),
     ]);
     const { identifier } = params;
@@ -1158,9 +1162,13 @@ export class DiscoverService {
         case ModelSorts.OutputPrice: {
           list = list.sort((a, b) => {
             if (order === 'asc') {
-              return (getTextOutputUnitRate(a.pricing) || 0) - (getTextOutputUnitRate(b.pricing) || 0);
+              return (
+                (getTextOutputUnitRate(a.pricing) || 0) - (getTextOutputUnitRate(b.pricing) || 0)
+              );
             } else {
-              return (getTextOutputUnitRate(b.pricing) || 0) - (getTextOutputUnitRate(a.pricing) || 0);
+              return (
+                (getTextOutputUnitRate(b.pricing) || 0) - (getTextOutputUnitRate(a.pricing) || 0)
+              );
             }
           });
           break;

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

@@ -232,7 +232,7 @@ export const createAiProviderSlice: StateCreator<
       !isDeprecatedEdition ? [AiProviderSwrKey.fetchAiProviderRuntimeState, isLogin] : null,
       async ([, isLogin]) => {
         const [{ LOBE_DEFAULT_MODEL_LIST: builtinAiModelList }, { DEFAULT_MODEL_PROVIDER_LIST }] =
-          await Promise.all([import('@/config/aiModels'), import('@/config/modelProviders')]);
+          await Promise.all([import('model-bank'), import('@/config/modelProviders')]);
 
         if (isLogin) {
           const data = await aiProviderService.getAiProviderRuntimeState();

package/src/store/image/slices/generationConfig/action.test.ts

@@ -1,12 +1,8 @@
 import { act, renderHook } from '@testing-library/react';
+import { ModelParamsSchema, RuntimeImageGenParams, extractDefaultValues } from 'model-bank';
+import { fluxSchnellParamsSchema } from 'model-bank';
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { fluxSchnellParamsSchema } from '@/config/aiModels/fal';
-import {
-  ModelParamsSchema,
-  RuntimeImageGenParams,
-  extractDefaultValues,
-} from '@/libs/standard-parameters/index';
 import { useImageStore } from '@/store/image';
 import { AIImageModelCard } from '@/types/aiModel';
 

package/src/store/image/slices/generationConfig/action.ts

@@ -1,12 +1,12 @@
-import { StateCreator } from 'zustand/vanilla';
-
 import {
   ModelParamsSchema,
   RuntimeImageGenParams,
   RuntimeImageGenParamsKeys,
   RuntimeImageGenParamsValue,
   extractDefaultValues,
-} from '@/libs/standard-parameters/index';
+} from 'model-bank';
+import { StateCreator } from 'zustand/vanilla';
+
 import { aiProviderSelectors, getAiInfraStoreState } from '@/store/aiInfra';
 import { AIImageModelCard } from '@/types/aiModel';
 

package/src/store/image/slices/generationConfig/hooks.test.ts

@@ -1,8 +1,8 @@
 import { act, renderHook } from '@testing-library/react';
+import { ModelParamsSchema, RuntimeImageGenParams } from 'model-bank';
+import { fluxSchnellParamsSchema } from 'model-bank';
 import { describe, expect, it, vi } from 'vitest';
 
-import { fluxSchnellParamsSchema } from '@/config/aiModels/fal';
-import { ModelParamsSchema, RuntimeImageGenParams } from '@/libs/standard-parameters/index';
 import { useImageStore } from '@/store/image';
 import { AIImageModelCard } from '@/types/aiModel';
 

package/src/store/image/slices/generationConfig/hooks.ts

@@ -1,7 +1,7 @@
+import { RuntimeImageGenParams, RuntimeImageGenParamsKeys } from 'model-bank';
 import { useCallback, useMemo } from 'react';
 
 import { DEFAULT_ASPECT_RATIO, PRESET_ASPECT_RATIOS } from '@/const/image';
-import { RuntimeImageGenParams, RuntimeImageGenParamsKeys } from '@/libs/standard-parameters/index';
 
 import { useImageStore } from '../../store';
 import { imageGenerationConfigSelectors } from './selectors';

package/src/store/image/slices/generationConfig/initialState.ts

@@ -1,12 +1,11 @@
 /* eslint-disable sort-keys-fix/sort-keys-fix, typescript-sort-keys/interface */
 import { ModelProvider } from '@lobechat/model-runtime';
-
-import { gptImage1ParamsSchema } from '@/config/aiModels/openai';
 import {
   ModelParamsSchema,
   RuntimeImageGenParams,
   extractDefaultValues,
-} from '@/libs/standard-parameters/index';
+  gptImage1ParamsSchema,
+} from 'model-bank';
 
 export const DEFAULT_AI_IMAGE_PROVIDER = ModelProvider.OpenAI;
 export const DEFAULT_AI_IMAGE_MODEL = 'gpt-image-1';

package/src/store/image/slices/generationConfig/selectors.test.ts

@@ -1,7 +1,6 @@
+import { ModelParamsSchema, RuntimeImageGenParams, gptImage1ParamsSchema } from 'model-bank';
 import { describe, expect, it, vi } from 'vitest';
 
-import { gptImage1ParamsSchema } from '@/config/aiModels/openai';
-import { ModelParamsSchema, RuntimeImageGenParams } from '@/libs/standard-parameters/index';
 import { ImageStore } from '@/store/image';
 import { initialState } from '@/store/image/initialState';
 import { AIImageModelCard } from '@/types/aiModel';

package/src/store/image/slices/generationConfig/selectors.ts

@@ -1,4 +1,4 @@
-import { RuntimeImageGenParamsKeys } from '@/libs/standard-parameters/index';
+import { RuntimeImageGenParamsKeys } from 'model-bank';
 
 import { GenerationConfigState } from './initialState';