@lobehub/chat 1.111.10 → 1.111.12

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

@@ -1,6 +1,5 @@
-import type { ChatModelCard } from '@/types/llm';
-
 import { ModelProvider } from '../types';
+import { processMultiProviderModelList } from '../utils/modelParse';
 import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
 export interface GiteeAIModelCard {
@@ -13,44 +12,10 @@ export const LobeGiteeAI = createOpenAICompatibleRuntime({
     chatCompletion: () => process.env.DEBUG_GITEE_AI_CHAT_COMPLETION === '1',
   },
   models: async ({ client }) => {
-    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
-
-    const functionCallKeywords = ['qwen2.5', 'glm-4'];
-
-    const visionKeywords = ['internvl', 'qwen2-vl'];
-
-    const reasoningKeywords = ['deepseek-r1', 'qwq'];
-
     const modelsPage = (await client.models.list()) as any;
     const modelList: GiteeAIModelCard[] = modelsPage.data;
 
-    return modelList
-      .map((model) => {
-        const knownModel = LOBE_DEFAULT_MODEL_LIST.find(
-          (m) => model.id.toLowerCase() === m.id.toLowerCase(),
-        );
-
-        return {
-          contextWindowTokens: knownModel?.contextWindowTokens ?? undefined,
-          displayName: knownModel?.displayName ?? undefined,
-          enabled: knownModel?.enabled || false,
-          functionCall:
-            (functionCallKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) &&
-              !model.id.toLowerCase().includes('qwen2.5-coder')) ||
-            knownModel?.abilities?.functionCall ||
-            false,
-          id: model.id,
-          reasoning:
-            reasoningKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
-            knownModel?.abilities?.reasoning ||
-            false,
-          vision:
-            visionKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
-            knownModel?.abilities?.vision ||
-            false,
-        };
-      })
-      .filter(Boolean) as ChatModelCard[];
+    return await processMultiProviderModelList(modelList, 'giteeai');
   },
   provider: ModelProvider.GiteeAI,
 });

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

@@ -1,17 +1,26 @@
-import type { ChatModelCard } from '@/types/llm';
-
 import { AgentRuntimeErrorType } from '../error';
 import { ModelProvider } from '../types';
+import { processMultiProviderModelList } from '../utils/modelParse';
 import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 import { pruneReasoningPayload } from '../utils/openaiHelpers';
 
 export interface GithubModelCard {
-  description: string;
-  friendly_name: string;
+  capabilities: string[];
+  html_url: string;
   id: string;
+  limits: {
+    max_input_tokens: number;
+    max_output_tokens: number;
+  };
   name: string;
+  publisher: string;
+  rate_limit_tier: string;
+  registry: string;
+  summary: string;
+  supported_input_modalities: string[];
+  supported_output_modalities: string[];
   tags: string[];
-  task: string;
+  version: string;
 }
 
 /* eslint-enable typescript-sort-keys/interface */
@@ -40,47 +49,27 @@ export const LobeGithubAI = createOpenAICompatibleRuntime({
     bizError: AgentRuntimeErrorType.ProviderBizError,
     invalidAPIKey: AgentRuntimeErrorType.InvalidGithubToken,
   },
-  models: async ({ client }) => {
-    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
-
-    const functionCallKeywords = ['function', 'tool'];
-
-    const visionKeywords = ['vision'];
-
-    const reasoningKeywords = ['deepseek-r1', 'o1', 'o3', 'grok-3-mini'];
-
-    const modelsPage = (await client.models.list()) as any;
-    const modelList: GithubModelCard[] = modelsPage.body;
+  models: async () => {
+    const response = await fetch('https://models.github.ai/catalog/models');
+    const modelList: GithubModelCard[] = await response.json();
 
-    return modelList
-      .map((model) => {
-        const knownModel = LOBE_DEFAULT_MODEL_LIST.find(
-          (m) => model.name.toLowerCase() === m.id.toLowerCase(),
-        );
+    const formattedModels = modelList.map((model) => ({
+      contextWindowTokens: model.limits?.max_input_tokens + model.limits?.max_output_tokens,
+      description: model.summary,
+      displayName: model.name,
+      functionCall: model.capabilities?.includes('tool-calling') ?? undefined,
+      id: model.id,
+      maxOutput: model.limits?.max_output_tokens ?? undefined,
+      reasoning: model.tags?.includes('reasoning') ?? undefined,
+      releasedAt:
+        model.version && /^\d{4}-\d{2}-\d{2}$/.test(model.version) ? model.version : undefined,
+      vision:
+        (model.tags?.includes('multimodal') ||
+          model.supported_input_modalities?.includes('image')) ??
+        undefined,
+    }));
 
-        return {
-          contextWindowTokens: knownModel?.contextWindowTokens ?? undefined,
-          description: model.description,
-          displayName: model.friendly_name,
-          enabled: knownModel?.enabled || false,
-          functionCall:
-            functionCallKeywords.some((keyword) =>
-              model.description.toLowerCase().includes(keyword),
-            ) ||
-            knownModel?.abilities?.functionCall ||
-            false,
-          id: model.name,
-          reasoning:
-            reasoningKeywords.some((keyword) => model.name.toLowerCase().includes(keyword)) ||
-            knownModel?.abilities?.reasoning ||
-            false,
-          vision:
-            visionKeywords.some((keyword) => model.description.toLowerCase().includes(keyword)) ||
-            knownModel?.abilities?.vision ||
-            false,
-        };
-      })
-      .filter(Boolean) as ChatModelCard[];
+    return await processMultiProviderModelList(formattedModels, 'github');
   },
   provider: ModelProvider.Github,
 });

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

@@ -1,6 +1,5 @@
-import type { ChatModelCard } from '@/types/llm';
-
 import { ModelProvider } from '../types';
+import { processMultiProviderModelList } from '../utils/modelParse';
 import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
 export interface ModelScopeModelCard {
@@ -16,46 +15,11 @@ export const LobeModelScopeAI = createOpenAICompatibleRuntime({
     chatCompletion: () => process.env.DEBUG_MODELSCOPE_CHAT_COMPLETION === '1',
   },
   models: async ({ client }) => {
-    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
-
-    const functionCallKeywords = ['qwen', 'deepseek', 'llama'];
-
-    const visionKeywords = ['qwen-vl', 'qwen2-vl', 'llava'];
-
-    const reasoningKeywords = ['qwq', 'deepseek-r1'];
-
     try {
       const modelsPage = (await client.models.list()) as any;
       const modelList: ModelScopeModelCard[] = modelsPage.data || [];
 
-      return modelList
-        .map((model) => {
-          const knownModel = LOBE_DEFAULT_MODEL_LIST.find(
-            (m) => model.id.toLowerCase() === m.id.toLowerCase(),
-          );
-
-          const modelId = model.id.toLowerCase();
-
-          return {
-            contextWindowTokens: knownModel?.contextWindowTokens ?? undefined,
-            displayName: knownModel?.displayName ?? model.id,
-            enabled: knownModel?.enabled || false,
-            functionCall:
-              functionCallKeywords.some((keyword) => modelId.includes(keyword)) ||
-              knownModel?.abilities?.functionCall ||
-              false,
-            id: model.id,
-            reasoning:
-              reasoningKeywords.some((keyword) => modelId.includes(keyword)) ||
-              knownModel?.abilities?.reasoning ||
-              false,
-            vision:
-              visionKeywords.some((keyword) => modelId.includes(keyword)) ||
-              knownModel?.abilities?.vision ||
-              false,
-          };
-        })
-        .filter(Boolean) as ChatModelCard[];
+      return await processMultiProviderModelList(modelList, 'modelscope');
     } catch (error) {
       console.warn(
         'Failed to fetch ModelScope models. Please ensure your ModelScope API key is valid and your Alibaba Cloud account is properly bound:',

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

@@ -1,6 +1,5 @@
-import type { ChatModelCard } from '@/types/llm';
-
 import { ChatStreamPayload, ModelProvider } from '../types';
+import { MODEL_LIST_CONFIGS, processModelList } from '../utils/modelParse';
 import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
 
 export interface MoonshotModelCard {
@@ -45,43 +44,10 @@ export const LobeMoonshotAI = createOpenAICompatibleRuntime({
     chatCompletion: () => process.env.DEBUG_MOONSHOT_CHAT_COMPLETION === '1',
   },
   models: async ({ client }) => {
-    const { LOBE_DEFAULT_MODEL_LIST } = await import('@/config/aiModels');
-
-    const functionCallKeywords = ['moonshot-v1', 'kimi-latest'];
-
-    const visionKeywords = ['kimi-latest', 'kimi-thinking', 'vision'];
-
-    const reasoningKeywords = ['thinking'];
-
     const modelsPage = (await client.models.list()) as any;
     const modelList: MoonshotModelCard[] = modelsPage.data;
 
-    return modelList
-      .map((model) => {
-        const knownModel = LOBE_DEFAULT_MODEL_LIST.find(
-          (m) => model.id.toLowerCase() === m.id.toLowerCase(),
-        );
-
-        return {
-          contextWindowTokens: knownModel?.contextWindowTokens ?? undefined,
-          displayName: knownModel?.displayName ?? undefined,
-          enabled: knownModel?.enabled || false,
-          functionCall:
-            functionCallKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
-            knownModel?.abilities?.functionCall ||
-            false,
-          id: model.id,
-          reasoning:
-            reasoningKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
-            knownModel?.abilities?.reasoning ||
-            false,
-          vision:
-            visionKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
-            knownModel?.abilities?.vision ||
-            false,
-        };
-      })
-      .filter(Boolean) as ChatModelCard[];
+    return processModelList(modelList, MODEL_LIST_CONFIGS.moonshot, 'moonshot');
   },
   provider: ModelProvider.Moonshot,
 });

package/packages/model-runtime/src/novita/__snapshots__/index.test.ts.snap

@@ -5,53 +5,57 @@ exports[`NovitaAI > models > should get models 1`] = `
   {
     "contextWindowTokens": 8192,
     "description": "Meta's latest class of model (Llama 3) launched with a variety of sizes & flavors. This 8B instruct-tuned version was optimized for high quality dialogue usecases. It has demonstrated strong performance compared to leading closed-source models in human evaluations.",
-    "displayName": "meta-llama/llama-3-8b-instruct",
+    "displayName": "Llama 3 8B Instruct",
     "enabled": false,
     "functionCall": false,
     "id": "meta-llama/llama-3-8b-instruct",
     "maxOutput": undefined,
     "reasoning": false,
+    "releasedAt": "2024-04-25",
     "type": "chat",
     "vision": false,
   },
   {
     "contextWindowTokens": 8192,
     "description": "Meta's latest class of model (Llama 3) launched with a variety of sizes & flavors. This 70B instruct-tuned version was optimized for high quality dialogue usecases. It has demonstrated strong performance compared to leading closed-source models in human evaluations.",
-    "displayName": "meta-llama/llama-3-70b-instruct",
+    "displayName": "Llama 3 70B Instruct",
     "enabled": false,
     "functionCall": false,
     "id": "meta-llama/llama-3-70b-instruct",
     "maxOutput": undefined,
     "reasoning": false,
+    "releasedAt": "2024-04-25",
     "type": "chat",
     "vision": false,
   },
   {
-    "contextWindowTokens": 8192,
+    "contextWindowTokens": 16384,
     "description": "Meta's latest class of models, Llama 3.1, launched with a variety of sizes and configurations. The 8B instruct-tuned version is particularly fast and efficient. It has demonstrated strong performance in human evaluations, outperforming several leading closed-source models.",
-    "displayName": "meta-llama/llama-3.1-8b-instruct",
+    "displayName": "Llama 3.1 8B Instruct",
     "enabled": false,
     "functionCall": false,
     "id": "meta-llama/llama-3.1-8b-instruct",
     "maxOutput": undefined,
     "reasoning": false,
+    "releasedAt": "2024-07-24",
     "type": "chat",
     "vision": false,
   },
   {
-    "contextWindowTokens": 8192,
+    "contextWindowTokens": 32768,
     "description": "Meta's latest class of models, Llama 3.1, has launched with a variety of sizes and configurations. The 70B instruct-tuned version is optimized for high-quality dialogue use cases. It has demonstrated strong performance in human evaluations compared to leading closed-source models.",
     "displayName": "meta-llama/llama-3.1-70b-instruct",
-    "enabled": true,
+    "enabled": false,
     "functionCall": false,
     "id": "meta-llama/llama-3.1-70b-instruct",
     "maxOutput": undefined,
     "reasoning": false,
+    "releasedAt": "2024-07-24",
     "type": "chat",
     "vision": false,
   },
   {
-    "contextWindowTokens": 32768,
+    "contextWindowTokens": undefined,
     "description": "Meta's latest class of models, Llama 3.1, launched with a variety of sizes and configurations. This 405B instruct-tuned version is optimized for high-quality dialogue use cases. It has demonstrated strong performance compared to leading closed-source models, including GPT-4o and Claude 3.5 Sonnet, in evaluations.",
     "displayName": "meta-llama/llama-3.1-405b-instruct",
     "enabled": false,
@@ -59,6 +63,7 @@ exports[`NovitaAI > models > should get models 1`] = `
     "id": "meta-llama/llama-3.1-405b-instruct",
     "maxOutput": undefined,
     "reasoning": false,
+    "releasedAt": "2024-07-24",
     "type": "chat",
     "vision": false,
   },
@@ -66,17 +71,18 @@ exports[`NovitaAI > models > should get models 1`] = `
     "contextWindowTokens": 8192,
     "description": "Gemma 2 9B by Google is an advanced, open-source language model that sets a new standard for efficiency and performance in its size class.
 Designed for a wide variety of tasks, it empowers developers and researchers to build innovative applications, while maintaining accessibility, safety, and cost-effectiveness.",
-    "displayName": "google/gemma-2-9b-it",
+    "displayName": "Gemma 2 9B Instruct",
     "enabled": false,
     "functionCall": false,
     "id": "google/gemma-2-9b-it",
     "maxOutput": undefined,
     "reasoning": false,
+    "releasedAt": "2024-07-18",
     "type": "chat",
     "vision": false,
   },
   {
-    "contextWindowTokens": 4096,
+    "contextWindowTokens": undefined,
     "description": "This is a fine-tuned Llama-2 model designed to support longer and more detailed writing prompts, as well as next-chapter generation. It also includes an experimental role-playing instruction set with multi-round dialogues, character interactions, and varying numbers of participants",
     "displayName": "jondurbin/airoboros-l2-70b",
     "enabled": false,
@@ -84,83 +90,90 @@ Designed for a wide variety of tasks, it empowers developers and researchers to
     "id": "jondurbin/airoboros-l2-70b",
     "maxOutput": undefined,
     "reasoning": false,
+    "releasedAt": "2024-07-17",
     "type": "chat",
     "vision": false,
   },
   {
     "contextWindowTokens": 8192,
     "description": "Hermes 2 Pro is an upgraded, retrained version of Nous Hermes 2, consisting of an updated and cleaned version of the OpenHermes 2.5 Dataset, as well as a newly introduced Function Calling and JSON Mode dataset developed in-house.",
-    "displayName": "nousresearch/hermes-2-pro-llama-3-8b",
+    "displayName": "Hermes 2 Pro Llama 3 8B",
     "enabled": false,
-    "functionCall": true,
+    "functionCall": false,
     "id": "nousresearch/hermes-2-pro-llama-3-8b",
     "maxOutput": undefined,
     "reasoning": false,
+    "releasedAt": "2024-06-27",
     "type": "chat",
     "vision": false,
   },
   {
     "contextWindowTokens": 32768,
     "description": "A high-performing, industry-standard 7.3B parameter model, with optimizations for speed and context length.",
-    "displayName": "mistralai/mistral-7b-instruct",
+    "displayName": "Mistral 7B Instruct",
     "enabled": false,
     "functionCall": false,
     "id": "mistralai/mistral-7b-instruct",
     "maxOutput": undefined,
     "reasoning": false,
+    "releasedAt": "2024-06-27",
     "type": "chat",
     "vision": false,
   },
   {
     "contextWindowTokens": 16000,
     "description": "Dolphin 2.9 is designed for instruction following, conversational, and coding. This model is a finetune of Mixtral 8x22B Instruct. It features a 64k context length and was fine-tuned with a 16k sequence length using ChatML templates.The model is uncensored and is stripped of alignment and bias. It requires an external alignment layer for ethical use.",
-    "displayName": "cognitivecomputations/dolphin-mixtral-8x22b",
+    "displayName": "Dolphin Mixtral 8x22B",
     "enabled": false,
     "functionCall": false,
     "id": "cognitivecomputations/dolphin-mixtral-8x22b",
     "maxOutput": undefined,
     "reasoning": false,
+    "releasedAt": "2024-06-27",
     "type": "chat",
     "vision": false,
   },
   {
-    "contextWindowTokens": 16000,
+    "contextWindowTokens": 8192,
     "description": "The uncensored llama3 model is a powerhouse of creativity, excelling in both roleplay and story writing. It offers a liberating experience during roleplays, free from any restrictions. This model stands out for its immense creativity, boasting a vast array of unique ideas and plots, truly a treasure trove for those seeking originality. Its unrestricted nature during roleplays allows for the full breadth of imagination to unfold, akin to an enhanced, big-brained version of Stheno. Perfect for creative minds seeking a boundless platform for their imaginative expressions, the uncensored llama3 model is an ideal choice",
-    "displayName": "sao10k/l3-70b-euryale-v2.1",
+    "displayName": "L3 70B Euryale v2.1",
     "enabled": false,
     "functionCall": false,
     "id": "sao10k/l3-70b-euryale-v2.1",
     "maxOutput": undefined,
     "reasoning": true,
+    "releasedAt": "2024-06-18",
     "type": "chat",
     "vision": false,
   },
   {
     "contextWindowTokens": 4096,
     "description": "A merge with a complex family tree, this model was crafted for roleplaying and storytelling. Midnight Rose is a successor to Rogue Rose and Aurora Nights and improves upon them both. It wants to produce lengthy output by default and is the best creative writing merge produced so far by sophosympatheia.",
-    "displayName": "sophosympatheia/midnight-rose-70b",
+    "displayName": "Midnight Rose 70B",
     "enabled": false,
     "functionCall": false,
     "id": "sophosympatheia/midnight-rose-70b",
     "maxOutput": undefined,
     "reasoning": false,
+    "releasedAt": "2024-06-17",
     "type": "chat",
     "vision": false,
   },
   {
     "contextWindowTokens": 4096,
     "description": "The idea behind this merge is that each layer is composed of several tensors, which are in turn responsible for specific functions. Using MythoLogic-L2's robust understanding as its input and Huginn's extensive writing capability as its output seems to have resulted in a model that exceeds at both, confirming my theory. (More details to be released at a later time).",
-    "displayName": "gryphe/mythomax-l2-13b",
+    "displayName": "MythoMax l2 13B",
     "enabled": false,
     "functionCall": false,
     "id": "gryphe/mythomax-l2-13b",
     "maxOutput": undefined,
     "reasoning": false,
+    "releasedAt": "2024-04-25",
     "type": "chat",
     "vision": false,
   },
   {
-    "contextWindowTokens": 4096,
+    "contextWindowTokens": undefined,
     "description": "Nous-Hermes-Llama2-13b is a state-of-the-art language model fine-tuned on over 300,000 instructions. This model was fine-tuned by Nous Research, with Teknium and Emozilla leading the fine tuning process and dataset curation, Redmond AI sponsoring the compute, and several other contributors.",
     "displayName": "nousresearch/nous-hermes-llama2-13b",
     "enabled": false,
@@ -168,11 +181,12 @@ Designed for a wide variety of tasks, it empowers developers and researchers to
     "id": "nousresearch/nous-hermes-llama2-13b",
     "maxOutput": undefined,
     "reasoning": false,
+    "releasedAt": "2024-04-25",
     "type": "chat",
     "vision": false,
   },
   {
-    "contextWindowTokens": 32768,
+    "contextWindowTokens": undefined,
     "description": "Nous Hermes 2 Mixtral 8x7B DPO is the new flagship Nous Research model trained over the Mixtral 8x7B MoE LLM. The model was trained on over 1,000,000 entries of primarily GPT-4 generated data, as well as other high quality data from open datasets across the AI landscape, achieving state of the art performance on a variety of tasks.",
     "displayName": "Nous-Hermes-2-Mixtral-8x7B-DPO",
     "enabled": false,
@@ -180,11 +194,12 @@ Designed for a wide variety of tasks, it empowers developers and researchers to
     "id": "Nous-Hermes-2-Mixtral-8x7B-DPO",
     "maxOutput": undefined,
     "reasoning": false,
+    "releasedAt": "2024-04-25",
     "type": "chat",
     "vision": false,
   },
   {
-    "contextWindowTokens": 4096,
+    "contextWindowTokens": undefined,
     "description": "A Mythomax/MLewd_13B-style merge of selected 70B models. A multi-model merge of several LLaMA2 70B finetunes for roleplaying and creative work. The goal was to create a model that combines creativity with intelligence for an enhanced experience.",
     "displayName": "lzlv_70b",
     "enabled": false,
@@ -192,11 +207,12 @@ Designed for a wide variety of tasks, it empowers developers and researchers to
     "id": "lzlv_70b",
     "maxOutput": undefined,
     "reasoning": false,
+    "releasedAt": "2024-04-25",
     "type": "chat",
     "vision": false,
   },
   {
-    "contextWindowTokens": 4096,
+    "contextWindowTokens": undefined,
     "description": "OpenHermes 2.5 Mistral 7B is a state of the art Mistral Fine-tune, a continuation of OpenHermes 2 model, which trained on additional code datasets.",
     "displayName": "teknium/openhermes-2.5-mistral-7b",
     "enabled": false,
@@ -204,18 +220,20 @@ Designed for a wide variety of tasks, it empowers developers and researchers to
     "id": "teknium/openhermes-2.5-mistral-7b",
     "maxOutput": undefined,
     "reasoning": false,
+    "releasedAt": "2024-04-24",
     "type": "chat",
     "vision": false,
   },
   {
     "contextWindowTokens": 65535,
     "description": "WizardLM-2 8x22B is Microsoft AI's most advanced Wizard model. It demonstrates highly competitive performance compared to leading proprietary models, and it consistently outperforms all existing state-of-the-art opensource models.",
-    "displayName": "microsoft/wizardlm-2-8x22b",
+    "displayName": "WizardLM-2 8x22B",
     "enabled": false,
     "functionCall": false,
     "id": "microsoft/wizardlm-2-8x22b",
     "maxOutput": undefined,
     "reasoning": false,
+    "releasedAt": "2024-04-24",
     "type": "chat",
     "vision": false,
   },

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

@@ -1,5 +1,3 @@
-import type { ChatModelCard } from '@/types/llm';
-
 import { ModelProvider } from '../types';
 import { processMultiProviderModelList } from '../utils/modelParse';
 import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
@@ -16,39 +14,10 @@ export const LobeNovitaAI = createOpenAICompatibleRuntime({
     chatCompletion: () => process.env.DEBUG_NOVITA_CHAT_COMPLETION === '1',
   },
   models: async ({ client }) => {
-    const reasoningKeywords = ['deepseek-r1'];
-
     const modelsPage = (await client.models.list()) as any;
     const modelList: NovitaModelCard[] = modelsPage.data;
 
-    // 解析模型能力
-    const baseModels = await processMultiProviderModelList(modelList);
-
-    // 合并 Novita 获取的模型信息
-    return baseModels
-      .map((baseModel) => {
-        const model = modelList.find((m) => m.id === baseModel.id);
-
-        if (!model) return baseModel;
-
-        return {
-          ...baseModel,
-          contextWindowTokens: model.context_size,
-          description: model.description,
-          displayName: model.title,
-          functionCall:
-            baseModel.functionCall ||
-            model.description.toLowerCase().includes('function calling') ||
-            false,
-          reasoning:
-            baseModel.reasoning ||
-            model.description.toLowerCase().includes('reasoning task') ||
-            reasoningKeywords.some((keyword) => model.id.toLowerCase().includes(keyword)) ||
-            false,
-          vision: baseModel.vision || model.description.toLowerCase().includes('vision') || false,
-        };
-      })
-      .filter(Boolean) as ChatModelCard[];
+    return await processMultiProviderModelList(modelList, 'novita');
   },
   provider: ModelProvider.Novita,
 });

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

@@ -15,7 +15,7 @@ export const LobeNvidiaAI = createOpenAICompatibleRuntime({
     const modelsPage = (await client.models.list()) as any;
     const modelList: NvidiaModelCard[] = modelsPage.data;
 
-    return processMultiProviderModelList(modelList);
+    return processMultiProviderModelList(modelList, 'nvidia');
   },
   provider: ModelProvider.Nvidia,
 });