mirror of
https://github.com/ItzCrazyKns/Perplexica.git
synced 2025-10-19 05:48:15 +00:00
feat(emptyChatInput): add model selector
This commit is contained in:
@@ -66,12 +66,12 @@ const EmptyChatMessageInput = () => {
|
|||||||
placeholder="Ask anything..."
|
placeholder="Ask anything..."
|
||||||
/>
|
/>
|
||||||
<div className="flex flex-row items-center justify-between mt-4">
|
<div className="flex flex-row items-center justify-between mt-4">
|
||||||
<div className="flex flex-row items-center space-x-2 lg:space-x-4">
|
|
||||||
<Focus />
|
|
||||||
<Attach showText />
|
|
||||||
</div>
|
|
||||||
<div className="flex flex-row items-center space-x-1 sm:space-x-4">
|
|
||||||
<Optimization />
|
<Optimization />
|
||||||
|
<div className="flex flex-row items-center space-x-2">
|
||||||
|
<div className="flex flex-row items-center space-x-1">
|
||||||
|
<ModelSelector />
|
||||||
|
<Attach />
|
||||||
|
</div>
|
||||||
<button
|
<button
|
||||||
disabled={message.trim().length === 0}
|
disabled={message.trim().length === 0}
|
||||||
className="bg-sky-500 text-white disabled:text-black/50 dark:disabled:text-white/50 disabled:bg-[#e0e0dc] dark:disabled:bg-[#ececec21] hover:bg-opacity-85 transition duration-100 rounded-full p-2"
|
className="bg-sky-500 text-white disabled:text-black/50 dark:disabled:text-white/50 disabled:bg-[#e0e0dc] dark:disabled:bg-[#ececec21] hover:bg-opacity-85 transition duration-100 rounded-full p-2"
|
||||||
|
198
src/components/MessageInputActions/ModelSelector.tsx
Normal file
198
src/components/MessageInputActions/ModelSelector.tsx
Normal file
@@ -0,0 +1,198 @@
|
|||||||
|
'use client';
|
||||||
|
|
||||||
|
import { Cpu, Loader2, Search } from 'lucide-react';
|
||||||
|
import { cn } from '@/lib/utils';
|
||||||
|
import {
|
||||||
|
Popover,
|
||||||
|
PopoverButton,
|
||||||
|
PopoverPanel,
|
||||||
|
Transition,
|
||||||
|
} from '@headlessui/react';
|
||||||
|
import { Fragment, useEffect, useState } from 'react';
|
||||||
|
import { MinimalProvider } from '@/lib/models/types';
|
||||||
|
|
||||||
|
const ModelSelector = () => {
|
||||||
|
const [providers, setProviders] = useState<MinimalProvider[]>([]);
|
||||||
|
const [isLoading, setIsLoading] = useState(true);
|
||||||
|
const [searchQuery, setSearchQuery] = useState('');
|
||||||
|
const [selectedModel, setSelectedModel] = useState<{
|
||||||
|
providerId: string;
|
||||||
|
modelKey: string;
|
||||||
|
} | null>(null);
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
const loadProviders = async () => {
|
||||||
|
try {
|
||||||
|
setIsLoading(true);
|
||||||
|
const res = await fetch('/api/providers');
|
||||||
|
|
||||||
|
if (!res.ok) {
|
||||||
|
throw new Error('Failed to fetch providers');
|
||||||
|
}
|
||||||
|
|
||||||
|
const data = await res.json();
|
||||||
|
setProviders(data.providers || []);
|
||||||
|
|
||||||
|
const savedProviderId = localStorage.getItem('chatModelProviderId');
|
||||||
|
const savedModelKey = localStorage.getItem('chatModelKey');
|
||||||
|
|
||||||
|
if (savedProviderId && savedModelKey) {
|
||||||
|
setSelectedModel({
|
||||||
|
providerId: savedProviderId,
|
||||||
|
modelKey: savedModelKey,
|
||||||
|
});
|
||||||
|
} else if (data.providers && data.providers.length > 0) {
|
||||||
|
const firstProvider = data.providers.find(
|
||||||
|
(p: MinimalProvider) => p.chatModels.length > 0,
|
||||||
|
);
|
||||||
|
if (firstProvider && firstProvider.chatModels[0]) {
|
||||||
|
setSelectedModel({
|
||||||
|
providerId: firstProvider.id,
|
||||||
|
modelKey: firstProvider.chatModels[0].key,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.error('Error loading providers:', error);
|
||||||
|
} finally {
|
||||||
|
setIsLoading(false);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
loadProviders();
|
||||||
|
}, []);
|
||||||
|
|
||||||
|
const handleModelSelect = (providerId: string, modelKey: string) => {
|
||||||
|
setSelectedModel({ providerId, modelKey });
|
||||||
|
localStorage.setItem('chatModelProviderId', providerId);
|
||||||
|
localStorage.setItem('chatModelKey', modelKey);
|
||||||
|
};
|
||||||
|
|
||||||
|
const filteredProviders = providers
|
||||||
|
.map((provider) => ({
|
||||||
|
...provider,
|
||||||
|
chatModels: provider.chatModels.filter(
|
||||||
|
(model) =>
|
||||||
|
model.name.toLowerCase().includes(searchQuery.toLowerCase()) ||
|
||||||
|
provider.name.toLowerCase().includes(searchQuery.toLowerCase()),
|
||||||
|
),
|
||||||
|
}))
|
||||||
|
.filter((provider) => provider.chatModels.length > 0);
|
||||||
|
|
||||||
|
return (
|
||||||
|
<Popover className="relative w-full max-w-[15rem] md:max-w-md lg:max-w-lg">
|
||||||
|
<PopoverButton
|
||||||
|
type="button"
|
||||||
|
className="active:border-none hover:bg-light-200 hover:dark:bg-dark-200 p-2 rounded-lg focus:outline-none headless-open:text-black dark:headless-open:text-white text-black/50 dark:text-white/50 active:scale-95 transition duration-200 hover:text-black dark:hover:text-white"
|
||||||
|
>
|
||||||
|
<Cpu size={16} className="text-sky-500" />
|
||||||
|
</PopoverButton>
|
||||||
|
<Transition
|
||||||
|
as={Fragment}
|
||||||
|
enter="transition ease-out duration-100"
|
||||||
|
enterFrom="opacity-0 translate-y-1"
|
||||||
|
enterTo="opacity-100 translate-y-0"
|
||||||
|
leave="transition ease-in duration-100"
|
||||||
|
leaveFrom="opacity-100 translate-y-0"
|
||||||
|
leaveTo="opacity-0 translate-y-1"
|
||||||
|
>
|
||||||
|
<PopoverPanel className="absolute z-10 w-[230px] sm:w-[270px] md:w-[300px] -right-4">
|
||||||
|
<div className="bg-light-primary dark:bg-dark-primary border rounded-lg border-light-200 dark:border-dark-200 w-full flex flex-col shadow-lg overflow-hidden">
|
||||||
|
<div className="p-4 border-b border-light-200 dark:border-dark-200">
|
||||||
|
<div className="relative">
|
||||||
|
<Search
|
||||||
|
size={16}
|
||||||
|
className="absolute left-3 top-1/2 -translate-y-1/2 text-black/40 dark:text-white/40"
|
||||||
|
/>
|
||||||
|
<input
|
||||||
|
type="text"
|
||||||
|
placeholder="Search models..."
|
||||||
|
value={searchQuery}
|
||||||
|
onChange={(e) => setSearchQuery(e.target.value)}
|
||||||
|
className="w-full pl-9 pr-3 py-2 bg-light-secondary dark:bg-dark-secondary rounded-lg text-sm text-black dark:text-white placeholder:text-black/40 dark:placeholder:text-white/40 focus:outline-none focus:ring-2 focus:ring-sky-500/20 border border-transparent focus:border-sky-500/30 transition duration-200"
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="max-h-[320px] overflow-y-auto">
|
||||||
|
{isLoading ? (
|
||||||
|
<div className="flex items-center justify-center py-16">
|
||||||
|
<Loader2
|
||||||
|
className="animate-spin text-black/40 dark:text-white/40"
|
||||||
|
size={24}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
) : filteredProviders.length === 0 ? (
|
||||||
|
<div className="text-center py-16 px-4 text-black/60 dark:text-white/60 text-sm">
|
||||||
|
{searchQuery
|
||||||
|
? 'No models found'
|
||||||
|
: 'No chat models configured'}
|
||||||
|
</div>
|
||||||
|
) : (
|
||||||
|
<div className="flex flex-col">
|
||||||
|
{filteredProviders.map((provider, providerIndex) => (
|
||||||
|
<div key={provider.id}>
|
||||||
|
<div className="px-4 py-2.5 sticky top-0 bg-light-primary dark:bg-dark-primary border-b border-light-200/50 dark:border-dark-200/50">
|
||||||
|
<p className="text-xs text-black/50 dark:text-white/50 uppercase tracking-wider">
|
||||||
|
{provider.name}
|
||||||
|
</p>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="flex flex-col px-2 py-2 space-y-0.5">
|
||||||
|
{provider.chatModels.map((model) => (
|
||||||
|
<PopoverButton
|
||||||
|
key={model.key}
|
||||||
|
onClick={() =>
|
||||||
|
handleModelSelect(provider.id, model.key)
|
||||||
|
}
|
||||||
|
className={cn(
|
||||||
|
'px-3 py-2 flex items-center justify-between text-start duration-200 cursor-pointer transition rounded-lg group',
|
||||||
|
selectedModel?.providerId === provider.id &&
|
||||||
|
selectedModel?.modelKey === model.key
|
||||||
|
? 'bg-light-secondary dark:bg-dark-secondary'
|
||||||
|
: 'hover:bg-light-secondary dark:hover:bg-dark-secondary',
|
||||||
|
)}
|
||||||
|
>
|
||||||
|
<div className="flex items-center space-x-2.5 min-w-0 flex-1">
|
||||||
|
<Cpu
|
||||||
|
size={15}
|
||||||
|
className={cn(
|
||||||
|
'shrink-0',
|
||||||
|
selectedModel?.providerId === provider.id &&
|
||||||
|
selectedModel?.modelKey === model.key
|
||||||
|
? 'text-sky-500'
|
||||||
|
: 'text-black/50 dark:text-white/50 group-hover:text-black/70 group-hover:dark:text-white/70',
|
||||||
|
)}
|
||||||
|
/>
|
||||||
|
<p
|
||||||
|
className={cn(
|
||||||
|
'text-sm truncate',
|
||||||
|
selectedModel?.providerId === provider.id &&
|
||||||
|
selectedModel?.modelKey === model.key
|
||||||
|
? 'text-sky-500 font-medium'
|
||||||
|
: 'text-black/70 dark:text-white/70 group-hover:text-black dark:group-hover:text-white',
|
||||||
|
)}
|
||||||
|
>
|
||||||
|
{model.name}
|
||||||
|
</p>
|
||||||
|
</div>
|
||||||
|
</PopoverButton>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{providerIndex < filteredProviders.length - 1 && (
|
||||||
|
<div className="h-px bg-light-200 dark:bg-dark-200" />
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</PopoverPanel>
|
||||||
|
</Transition>
|
||||||
|
</Popover>
|
||||||
|
);
|
||||||
|
};
|
||||||
|
|
||||||
|
export default ModelSelector;
|
@@ -14,13 +14,13 @@ const OptimizationModes = [
|
|||||||
key: 'speed',
|
key: 'speed',
|
||||||
title: 'Speed',
|
title: 'Speed',
|
||||||
description: 'Prioritize speed and get the quickest possible answer.',
|
description: 'Prioritize speed and get the quickest possible answer.',
|
||||||
icon: <Zap size={20} className="text-[#FF9800]" />,
|
icon: <Zap size={16} className="text-[#FF9800]" />,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
key: 'balanced',
|
key: 'balanced',
|
||||||
title: 'Balanced',
|
title: 'Balanced',
|
||||||
description: 'Find the right balance between speed and accuracy',
|
description: 'Find the right balance between speed and accuracy',
|
||||||
icon: <Sliders size={20} className="text-[#4CAF50]" />,
|
icon: <Sliders size={16} className="text-[#4CAF50]" />,
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
key: 'quality',
|
key: 'quality',
|
||||||
@@ -40,22 +40,24 @@ const Optimization = () => {
|
|||||||
|
|
||||||
return (
|
return (
|
||||||
<Popover className="relative w-full max-w-[15rem] md:max-w-md lg:max-w-lg">
|
<Popover className="relative w-full max-w-[15rem] md:max-w-md lg:max-w-lg">
|
||||||
|
{({ open }) => (
|
||||||
|
<>
|
||||||
<PopoverButton
|
<PopoverButton
|
||||||
type="button"
|
type="button"
|
||||||
className="p-2 text-black/50 dark:text-white/50 rounded-xl hover:bg-light-secondary dark:hover:bg-dark-secondary active:scale-95 transition duration-200 hover:text-black dark:hover:text-white"
|
className="p-2 text-black/50 dark:text-white/50 rounded-xl hover:bg-light-secondary dark:hover:bg-dark-secondary active:scale-95 transition duration-200 hover:text-black dark:hover:text-white focus:outline-none"
|
||||||
>
|
>
|
||||||
<div className="flex flex-row items-center space-x-1">
|
<div className="flex flex-row items-center space-x-1">
|
||||||
{
|
{
|
||||||
OptimizationModes.find((mode) => mode.key === optimizationMode)
|
OptimizationModes.find((mode) => mode.key === optimizationMode)
|
||||||
?.icon
|
?.icon
|
||||||
}
|
}
|
||||||
<p className="text-xs font-medium">
|
<ChevronDown
|
||||||
{
|
size={16}
|
||||||
OptimizationModes.find((mode) => mode.key === optimizationMode)
|
className={cn(
|
||||||
?.title
|
open ? 'rotate-180' : 'rotate-0',
|
||||||
}
|
'transition duration:200',
|
||||||
</p>
|
)}
|
||||||
<ChevronDown size={20} />
|
/>
|
||||||
</div>
|
</div>
|
||||||
</PopoverButton>
|
</PopoverButton>
|
||||||
<Transition
|
<Transition
|
||||||
@@ -67,7 +69,7 @@ const Optimization = () => {
|
|||||||
leaveFrom="opacity-100 translate-y-0"
|
leaveFrom="opacity-100 translate-y-0"
|
||||||
leaveTo="opacity-0 translate-y-1"
|
leaveTo="opacity-0 translate-y-1"
|
||||||
>
|
>
|
||||||
<PopoverPanel className="absolute z-10 w-64 md:w-[250px] right-0">
|
<PopoverPanel className="absolute z-10 w-64 md:w-[250px] left-0">
|
||||||
<div className="flex flex-col gap-2 bg-light-primary dark:bg-dark-primary border rounded-lg border-light-200 dark:border-dark-200 w-full p-4 max-h-[200px] md:max-h-none overflow-y-auto">
|
<div className="flex flex-col gap-2 bg-light-primary dark:bg-dark-primary border rounded-lg border-light-200 dark:border-dark-200 w-full p-4 max-h-[200px] md:max-h-none overflow-y-auto">
|
||||||
{OptimizationModes.map((mode, i) => (
|
{OptimizationModes.map((mode, i) => (
|
||||||
<PopoverButton
|
<PopoverButton
|
||||||
@@ -75,7 +77,7 @@ const Optimization = () => {
|
|||||||
key={i}
|
key={i}
|
||||||
disabled={mode.key === 'quality'}
|
disabled={mode.key === 'quality'}
|
||||||
className={cn(
|
className={cn(
|
||||||
'p-2 rounded-lg flex flex-col items-start justify-start text-start space-y-1 duration-200 cursor-pointer transition',
|
'p-2 rounded-lg flex flex-col items-start justify-start text-start space-y-1 duration-200 cursor-pointer transition focus:outline-none',
|
||||||
optimizationMode === mode.key
|
optimizationMode === mode.key
|
||||||
? 'bg-light-secondary dark:bg-dark-secondary'
|
? 'bg-light-secondary dark:bg-dark-secondary'
|
||||||
: 'hover:bg-light-secondary dark:hover:bg-dark-secondary',
|
: 'hover:bg-light-secondary dark:hover:bg-dark-secondary',
|
||||||
@@ -94,6 +96,8 @@ const Optimization = () => {
|
|||||||
</div>
|
</div>
|
||||||
</PopoverPanel>
|
</PopoverPanel>
|
||||||
</Transition>
|
</Transition>
|
||||||
|
</>
|
||||||
|
)}
|
||||||
</Popover>
|
</Popover>
|
||||||
);
|
);
|
||||||
};
|
};
|
||||||
|
Reference in New Issue
Block a user