Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2026-01-01 01:30:51 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Support API models with provider_name field in getModelName

Browse Source 
- 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
This commit is contained in:
suyao 2025-09-20 16:05:18 +08:00
parent 34b05a138b
commit 117e390cf1
No known key found for this signature in database
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/renderer/src/services/ModelService.ts
View File

@ -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}`
}