103 lines
2.8 KiB
TypeScript
103 lines
2.8 KiB
TypeScript
import { useState, useEffect } from 'react'
|
|
|
|
import { InferenceEngine } from '@janhq/core'
|
|
|
|
import { Button } from '@janhq/joi'
|
|
import { useSetAtom } from 'jotai'
|
|
import { SettingsIcon, PlusIcon } from 'lucide-react'
|
|
|
|
import { MainViewState } from '@/constants/screens'
|
|
|
|
import { localEngines } from '@/utils/modelEngine'
|
|
|
|
import { extensionManager } from '@/extension'
|
|
import { mainViewStateAtom } from '@/helpers/atoms/App.atom'
|
|
import { selectedSettingAtom } from '@/helpers/atoms/Setting.atom'
|
|
|
|
type Props = {
|
|
engine: InferenceEngine
|
|
}
|
|
|
|
const SetupRemoteModel = ({ engine }: Props) => {
|
|
const setSelectedSetting = useSetAtom(selectedSettingAtom)
|
|
const setMainViewState = useSetAtom(mainViewStateAtom)
|
|
|
|
const [extensionHasSettings, setExtensionHasSettings] = useState<
|
|
{ name?: string; setting: string; apiKey: string; provider: string }[]
|
|
>([])
|
|
|
|
useEffect(() => {
|
|
const getAllSettings = async () => {
|
|
const extensionsMenu: {
|
|
name?: string
|
|
setting: string
|
|
apiKey: string
|
|
provider: string
|
|
}[] = []
|
|
const extensions = extensionManager.getAll()
|
|
|
|
for (const extension of extensions) {
|
|
if (typeof extension.getSettings === 'function') {
|
|
const settings = await extension.getSettings()
|
|
|
|
if (
|
|
(settings && settings.length > 0) ||
|
|
(await extension.installationState()) !== 'NotRequired'
|
|
) {
|
|
extensionsMenu.push({
|
|
name: extension.productName,
|
|
setting: extension.name,
|
|
apiKey:
|
|
'apiKey' in extension && typeof extension.apiKey === 'string'
|
|
? extension.apiKey
|
|
: '',
|
|
provider:
|
|
'provider' in extension &&
|
|
typeof extension.provider === 'string'
|
|
? extension.provider
|
|
: '',
|
|
})
|
|
}
|
|
}
|
|
}
|
|
setExtensionHasSettings(extensionsMenu)
|
|
}
|
|
getAllSettings()
|
|
}, [])
|
|
|
|
const onSetupItemClick = (setting: InferenceEngine) => {
|
|
setMainViewState(MainViewState.Settings)
|
|
setSelectedSetting(
|
|
extensionHasSettings.filter((x) =>
|
|
x.provider.toLowerCase().includes(setting)
|
|
)[0]?.setting
|
|
)
|
|
}
|
|
|
|
const apiKey = !localEngines.includes(engine)
|
|
? extensionHasSettings.filter((x) => x.provider === engine)[0]?.apiKey
|
|
.length > 1
|
|
: true
|
|
|
|
return (
|
|
<Button
|
|
theme="icon"
|
|
variant="outline"
|
|
onClick={() => {
|
|
onSetupItemClick(engine)
|
|
}}
|
|
>
|
|
{apiKey ? (
|
|
<SettingsIcon
|
|
size={14}
|
|
className="text-[hsla(var(--text-secondary))]"
|
|
/>
|
|
) : (
|
|
<PlusIcon size={14} className="text-[hsla(var(--text-secondary))]" />
|
|
)}
|
|
</Button>
|
|
)
|
|
}
|
|
|
|
export default SetupRemoteModel
|