estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

feat: support show model display name (#887)

Browse Source
This commit is contained in:
Joel
2023-08-17 15:13:35 +08:00
committed by GitHub
parent f42e7d1a61
commit cd78adb0ab
5 changed files with 20 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
web/app/components/header/account-setting/model-page/model-selector/index.tsx
View File

@@ -47,6 +47,7 @@ const ModelSelector: FC<Props> = ({
[ModelType.embeddings]: embeddingsModelList,
[ModelType.speech2text]: speech2textModelList,
})[modelType]
const currModel = modelList.find(item => item.model_name === value?.modelName && item.model_provider.provider_name === value.providerName)
const allModelNames = (() => {
if (!search)
return {}
@@ -77,11 +78,12 @@ const ModelSelector: FC<Props> = ({
value: providerName,
})
const models = filteredModelList.filter(m => m.model_provider.provider_name === providerName)
models.forEach(({ model_name }) => {
models.forEach(({ model_name, model_display_name }) => {
res.push({
type: 'model',
providerName,
value: model_name,
modelDisplayName: model_display_name,
})
})
})
@@ -104,7 +106,7 @@ const ModelSelector: FC<Props> = ({
modelId={value.modelName}
providerName={value.providerName}
/>
<div className='mr-1.5 grow text-left text-sm text-gray-900 truncate'><ModelName modelId={value.modelName} /></div>
<div className='mr-1.5 grow text-left text-sm text-gray-900 truncate'><ModelName modelId={value.modelName} modelDisplayName={currModel?.model_display_name} /></div>
</>
)
: (
@@ -193,7 +195,7 @@ const ModelSelector: FC<Props> = ({
modelId={model.value}
providerName={model.providerName}
/>
<div className='grow text-left text-sm text-gray-900 truncate'><ModelName modelId={model.value} /></div>
<div className='grow text-left text-sm text-gray-900 truncate'><ModelName modelId={model.value} modelDisplayName={model.modelDisplayName} /></div>
{ (value?.providerName === model.providerName && value?.modelName === model.value) && <Check className='shrink-0 w-4 h-4 text-primary-600' /> }
</Popover.Button>
)