From 117e390cf1e090b285dc50123a50228502b5154d Mon Sep 17 00:00:00 2001
From: suyao <sy20010504@gmail.com>
Date: Sat, 20 Sep 2025 16:05:18 +0800
Subject: [PATCH] Support API models with provider_name field in getModelName

- Add ApiModel type import and update function signature to accept ApiModel
- Return formatted name using provider_name field for API models
- Maintain backward compatibility for legacy models by looking up provider in store
---
 src/renderer/src/services/ModelService.ts | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/src/renderer/src/services/ModelService.ts b/src/renderer/src/services/ModelService.ts
index 52049f229..c26ba75cf 100644
--- a/src/renderer/src/services/ModelService.ts
+++ b/src/renderer/src/services/ModelService.ts
@@ -1,5 +1,6 @@
 import store from '@renderer/store'
 import { Model } from '@renderer/types'
+import { ApiModel } from '@renderer/types/apiModels'
 import { pick } from 'lodash'
 
 import { getProviderName } from './ProviderService'
@@ -18,12 +19,19 @@ export const hasModel = (m?: Model) => {
   return allModels.find((model) => model.id === m?.id)
 }
 
-export function getModelName(model?: Model) {
-  const provider = store.getState().llm.providers.find((p) => p.id === model?.provider)
+export function getModelName(model?: Model | ApiModel) {
   const modelName = model?.name || model?.id || ''
 
+  // For API models that have provider_name field, use it directly
+  const apiModel = model as ApiModel
+  if (apiModel?.provider_name) {
+    return `${modelName} | ${apiModel.provider_name}`
+  }
+
+  // For legacy models, look up the provider in the store
+  const provider = store.getState().llm.providers.find((p) => p.id === model?.provider)
   if (provider) {
-    const providerName = getProviderName(model)
+    const providerName = getProviderName(model as Model)
     return `${modelName} | ${providerName}`
   }