fix: clickable remote model (#2970)

This commit is contained in:
Faisal Amir 2024-05-31 10:14:27 +07:00 committed by GitHub
parent 8dbb8d6536
commit f0f273bf07
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -324,6 +324,7 @@ const ModelDropdown = ({
<h6 className="mb-1 mt-3 px-3 font-medium text-[hsla(var(--text-secondary))]"> <h6 className="mb-1 mt-3 px-3 font-medium text-[hsla(var(--text-secondary))]">
Cortex Cortex
</h6> </h6>
{searchText.length === 0 ? (
<ul className="pb-2"> <ul className="pb-2">
{featuredModel.map((model) => { {featuredModel.map((model) => {
const isDownloading = downloadingModels.some( const isDownloading = downloadingModels.some(
@ -341,7 +342,10 @@ const ModelDropdown = ({
> >
{model.name} {model.name}
</p> </p>
<ModelLabel metadata={model.metadata} compact /> <ModelLabel
metadata={model.metadata}
compact
/>
</div> </div>
<div className="flex items-center gap-2 text-[hsla(var(--text-tertiary))]"> <div className="flex items-center gap-2 text-[hsla(var(--text-tertiary))]">
<span className="font-medium"> <span className="font-medium">
@ -359,6 +363,53 @@ const ModelDropdown = ({
) )
})} })}
</ul> </ul>
) : (
<ul className="pb-2">
{configuredModels
.filter((x) => x.engine === InferenceEngine.nitro)
.filter((e) =>
e.name
.toLowerCase()
.includes(searchText.toLowerCase().trim())
)
.map((model) => {
const isDownloading = downloadingModels.some(
(md) => md.id === model.id
)
return (
<li
key={model.id}
className="flex items-center justify-between gap-4 px-3 py-2 hover:bg-[hsla(var(--dropdown-menu-hover-bg))]"
>
<div className="flex items-center gap-2">
<p
className="line-clamp-1 text-[hsla(var(--text-secondary))]"
title={model.name}
>
{model.name}
</p>
<ModelLabel
metadata={model.metadata}
compact
/>
</div>
<div className="flex items-center gap-2 text-[hsla(var(--text-tertiary))]">
<span className="font-medium">
{toGibibytes(model.metadata.size)}
</span>
{!isDownloading && (
<DownloadCloudIcon
size={18}
className="cursor-pointer text-[hsla(var(--app-link))]"
onClick={() => downloadModel(model)}
/>
)}
</div>
</li>
)
})}
</ul>
)}
</div> </div>
</div> </div>
)} )}
@ -397,12 +448,15 @@ const ModelDropdown = ({
InferenceEngine.nitro_tensorrt_llm && InferenceEngine.nitro_tensorrt_llm &&
'cursor-default text-[hsla(var(--text-tertiary))]' 'cursor-default text-[hsla(var(--text-tertiary))]'
)} )}
onClick={() => onClick={() => {
if (
apiKey || apiKey ||
(model.engine === model.engine ===
InferenceEngine.nitro_tensorrt_llm && InferenceEngine.nitro_tensorrt_llm
onClickModelItem(model.id)) ) {
onClickModelItem(model.id)
} }
}}
> >
<div className="flex flex-shrink-0 gap-x-2"> <div className="flex flex-shrink-0 gap-x-2">
{engineHasLogo.map((x) => { {engineHasLogo.map((x) => {