From 09a347cae4b05ff879ce98cba6983a03b9ed49ec Mon Sep 17 00:00:00 2001 From: duanyongcheng Date: Sun, 16 Feb 2025 19:36:16 +0800 Subject: [PATCH] feat: show provider in mesage --- src/renderer/src/services/ModelService.ts | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/renderer/src/services/ModelService.ts b/src/renderer/src/services/ModelService.ts index 2c3481924..159905e3c 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 { t } from 'i18next' import { pick } from 'lodash' export const getModelUniqId = (m?: Model) => { @@ -17,5 +18,8 @@ export const hasModel = (m?: Model) => { } export function getModelName(model?: Model) { - return model?.name || model?.id || '' + const provider = store.getState().llm.providers.find((p) => p.id === model?.provider) + const modelName = model?.name || model?.id || '' + const providerName = provider?.isSystem ? t(`provider.${provider.id}`) : provider?.name + return `${modelName} | ${providerName}` }