Merge pull request #645 from wonderwhy-er/pr-620
refactor: refactored Setting component, added useSettings Hook to get settings details from one single place
This commit is contained in:
@@ -17,7 +17,6 @@ import Cookies from 'js-cookie';
|
|||||||
import * as Tooltip from '@radix-ui/react-tooltip';
|
import * as Tooltip from '@radix-ui/react-tooltip';
|
||||||
|
|
||||||
import styles from './BaseChat.module.scss';
|
import styles from './BaseChat.module.scss';
|
||||||
import type { ProviderInfo } from '~/utils/types';
|
|
||||||
import { ExportChatButton } from '~/components/chat/chatExportAndImport/ExportChatButton';
|
import { ExportChatButton } from '~/components/chat/chatExportAndImport/ExportChatButton';
|
||||||
import { ImportButtons } from '~/components/chat/chatExportAndImport/ImportButtons';
|
import { ImportButtons } from '~/components/chat/chatExportAndImport/ImportButtons';
|
||||||
import { ExamplePrompts } from '~/components/chat/ExamplePrompts';
|
import { ExamplePrompts } from '~/components/chat/ExamplePrompts';
|
||||||
@@ -26,6 +25,7 @@ import GitCloneButton from './GitCloneButton';
|
|||||||
import FilePreview from './FilePreview';
|
import FilePreview from './FilePreview';
|
||||||
import { ModelSelector } from '~/components/chat/ModelSelector';
|
import { ModelSelector } from '~/components/chat/ModelSelector';
|
||||||
import { SpeechRecognitionButton } from '~/components/chat/SpeechRecognition';
|
import { SpeechRecognitionButton } from '~/components/chat/SpeechRecognition';
|
||||||
|
import type { IProviderSetting, ProviderInfo } from '~/types/model';
|
||||||
|
|
||||||
const TEXTAREA_MIN_HEIGHT = 76;
|
const TEXTAREA_MIN_HEIGHT = 76;
|
||||||
|
|
||||||
@@ -45,6 +45,7 @@ interface BaseChatProps {
|
|||||||
setModel?: (model: string) => void;
|
setModel?: (model: string) => void;
|
||||||
provider?: ProviderInfo;
|
provider?: ProviderInfo;
|
||||||
setProvider?: (provider: ProviderInfo) => void;
|
setProvider?: (provider: ProviderInfo) => void;
|
||||||
|
providerList?: ProviderInfo[];
|
||||||
handleStop?: () => void;
|
handleStop?: () => void;
|
||||||
sendMessage?: (event: React.UIEvent, messageInput?: string) => void;
|
sendMessage?: (event: React.UIEvent, messageInput?: string) => void;
|
||||||
handleInputChange?: (event: React.ChangeEvent<HTMLTextAreaElement>) => void;
|
handleInputChange?: (event: React.ChangeEvent<HTMLTextAreaElement>) => void;
|
||||||
@@ -70,6 +71,7 @@ export const BaseChat = React.forwardRef<HTMLDivElement, BaseChatProps>(
|
|||||||
setModel,
|
setModel,
|
||||||
provider,
|
provider,
|
||||||
setProvider,
|
setProvider,
|
||||||
|
providerList,
|
||||||
input = '',
|
input = '',
|
||||||
enhancingPrompt,
|
enhancingPrompt,
|
||||||
handleInputChange,
|
handleInputChange,
|
||||||
@@ -108,48 +110,10 @@ export const BaseChat = React.forwardRef<HTMLDivElement, BaseChatProps>(
|
|||||||
const [recognition, setRecognition] = useState<SpeechRecognition | null>(null);
|
const [recognition, setRecognition] = useState<SpeechRecognition | null>(null);
|
||||||
const [transcript, setTranscript] = useState('');
|
const [transcript, setTranscript] = useState('');
|
||||||
|
|
||||||
// Load enabled providers from cookies
|
|
||||||
const [enabledProviders, setEnabledProviders] = useState(() => {
|
|
||||||
const savedProviders = Cookies.get('providers');
|
|
||||||
|
|
||||||
if (savedProviders) {
|
|
||||||
try {
|
|
||||||
const parsedProviders = JSON.parse(savedProviders);
|
|
||||||
return PROVIDER_LIST.filter((p) => parsedProviders[p.name]);
|
|
||||||
} catch (error) {
|
|
||||||
console.error('Failed to parse providers from cookies:', error);
|
|
||||||
return PROVIDER_LIST;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return PROVIDER_LIST;
|
|
||||||
});
|
|
||||||
|
|
||||||
// Update enabled providers when cookies change
|
|
||||||
useEffect(() => {
|
|
||||||
const updateProvidersFromCookies = () => {
|
|
||||||
const savedProviders = Cookies.get('providers');
|
|
||||||
|
|
||||||
if (savedProviders) {
|
|
||||||
try {
|
|
||||||
const parsedProviders = JSON.parse(savedProviders);
|
|
||||||
setEnabledProviders(PROVIDER_LIST.filter((p) => parsedProviders[p.name]));
|
|
||||||
} catch (error) {
|
|
||||||
console.error('Failed to parse providers from cookies:', error);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
updateProvidersFromCookies();
|
|
||||||
|
|
||||||
const interval = setInterval(updateProvidersFromCookies, 1000);
|
|
||||||
|
|
||||||
return () => clearInterval(interval);
|
|
||||||
}, [PROVIDER_LIST]);
|
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
console.log(transcript);
|
console.log(transcript);
|
||||||
}, [transcript]);
|
}, [transcript]);
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
// Load API keys from cookies on component mount
|
// Load API keys from cookies on component mount
|
||||||
try {
|
try {
|
||||||
@@ -169,7 +133,26 @@ export const BaseChat = React.forwardRef<HTMLDivElement, BaseChatProps>(
|
|||||||
Cookies.remove('apiKeys');
|
Cookies.remove('apiKeys');
|
||||||
}
|
}
|
||||||
|
|
||||||
initializeModelList().then((modelList) => {
|
let providerSettings: Record<string, IProviderSetting> | undefined = undefined;
|
||||||
|
|
||||||
|
try {
|
||||||
|
const savedProviderSettings = Cookies.get('providers');
|
||||||
|
|
||||||
|
if (savedProviderSettings) {
|
||||||
|
const parsedProviderSettings = JSON.parse(savedProviderSettings);
|
||||||
|
|
||||||
|
if (typeof parsedProviderSettings === 'object' && parsedProviderSettings !== null) {
|
||||||
|
providerSettings = parsedProviderSettings;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.error('Error loading Provider Settings from cookies:', error);
|
||||||
|
|
||||||
|
// Clear invalid cookie data
|
||||||
|
Cookies.remove('providers');
|
||||||
|
}
|
||||||
|
|
||||||
|
initializeModelList(providerSettings).then((modelList) => {
|
||||||
setModelList(modelList);
|
setModelList(modelList);
|
||||||
});
|
});
|
||||||
|
|
||||||
@@ -369,10 +352,10 @@ export const BaseChat = React.forwardRef<HTMLDivElement, BaseChatProps>(
|
|||||||
modelList={modelList}
|
modelList={modelList}
|
||||||
provider={provider}
|
provider={provider}
|
||||||
setProvider={setProvider}
|
setProvider={setProvider}
|
||||||
providerList={PROVIDER_LIST}
|
providerList={providerList || PROVIDER_LIST}
|
||||||
apiKeys={apiKeys}
|
apiKeys={apiKeys}
|
||||||
/>
|
/>
|
||||||
{enabledProviders.length > 0 && provider && (
|
{(providerList || []).length > 0 && provider && (
|
||||||
<APIKeyManager
|
<APIKeyManager
|
||||||
provider={provider}
|
provider={provider}
|
||||||
apiKey={apiKeys[provider.name] || ''}
|
apiKey={apiKeys[provider.name] || ''}
|
||||||
@@ -468,7 +451,7 @@ export const BaseChat = React.forwardRef<HTMLDivElement, BaseChatProps>(
|
|||||||
<SendButton
|
<SendButton
|
||||||
show={input.length > 0 || isStreaming || uploadedFiles.length > 0}
|
show={input.length > 0 || isStreaming || uploadedFiles.length > 0}
|
||||||
isStreaming={isStreaming}
|
isStreaming={isStreaming}
|
||||||
disabled={enabledProviders.length === 0}
|
disabled={!providerList || providerList.length === 0}
|
||||||
onClick={(event) => {
|
onClick={(event) => {
|
||||||
if (isStreaming) {
|
if (isStreaming) {
|
||||||
handleStop?.();
|
handleStop?.();
|
||||||
@@ -528,7 +511,7 @@ export const BaseChat = React.forwardRef<HTMLDivElement, BaseChatProps>(
|
|||||||
!isModelSettingsCollapsed,
|
!isModelSettingsCollapsed,
|
||||||
})}
|
})}
|
||||||
onClick={() => setIsModelSettingsCollapsed(!isModelSettingsCollapsed)}
|
onClick={() => setIsModelSettingsCollapsed(!isModelSettingsCollapsed)}
|
||||||
disabled={enabledProviders.length === 0}
|
disabled={!providerList || providerList.length === 0}
|
||||||
>
|
>
|
||||||
<div className={`i-ph:caret-${isModelSettingsCollapsed ? 'right' : 'down'} text-lg`} />
|
<div className={`i-ph:caret-${isModelSettingsCollapsed ? 'right' : 'down'} text-lg`} />
|
||||||
{isModelSettingsCollapsed ? <span className="text-xs">{model}</span> : <span />}
|
{isModelSettingsCollapsed ? <span className="text-xs">{model}</span> : <span />}
|
||||||
|
|||||||
@@ -17,8 +17,9 @@ import { cubicEasingFn } from '~/utils/easings';
|
|||||||
import { createScopedLogger, renderLogger } from '~/utils/logger';
|
import { createScopedLogger, renderLogger } from '~/utils/logger';
|
||||||
import { BaseChat } from './BaseChat';
|
import { BaseChat } from './BaseChat';
|
||||||
import Cookies from 'js-cookie';
|
import Cookies from 'js-cookie';
|
||||||
import type { ProviderInfo } from '~/utils/types';
|
|
||||||
import { debounce } from '~/utils/debounce';
|
import { debounce } from '~/utils/debounce';
|
||||||
|
import { useSettings } from '~/lib/hooks/useSettings';
|
||||||
|
import type { ProviderInfo } from '~/types/model';
|
||||||
|
|
||||||
const toastAnimation = cssTransition({
|
const toastAnimation = cssTransition({
|
||||||
enter: 'animated fadeInRight',
|
enter: 'animated fadeInRight',
|
||||||
@@ -91,6 +92,7 @@ export const ChatImpl = memo(
|
|||||||
const [chatStarted, setChatStarted] = useState(initialMessages.length > 0);
|
const [chatStarted, setChatStarted] = useState(initialMessages.length > 0);
|
||||||
const [uploadedFiles, setUploadedFiles] = useState<File[]>([]); // Move here
|
const [uploadedFiles, setUploadedFiles] = useState<File[]>([]); // Move here
|
||||||
const [imageDataList, setImageDataList] = useState<string[]>([]); // Move here
|
const [imageDataList, setImageDataList] = useState<string[]>([]); // Move here
|
||||||
|
const { activeProviders } = useSettings();
|
||||||
|
|
||||||
const [model, setModel] = useState(() => {
|
const [model, setModel] = useState(() => {
|
||||||
const savedModel = Cookies.get('selectedModel');
|
const savedModel = Cookies.get('selectedModel');
|
||||||
@@ -316,6 +318,7 @@ export const ChatImpl = memo(
|
|||||||
setModel={handleModelChange}
|
setModel={handleModelChange}
|
||||||
provider={provider}
|
provider={provider}
|
||||||
setProvider={handleProviderChange}
|
setProvider={handleProviderChange}
|
||||||
|
providerList={activeProviders}
|
||||||
messageRef={messageRef}
|
messageRef={messageRef}
|
||||||
scrollRef={scrollRef}
|
scrollRef={scrollRef}
|
||||||
handleInputChange={(e) => {
|
handleInputChange={(e) => {
|
||||||
|
|||||||
@@ -46,7 +46,7 @@
|
|||||||
padding: 1rem;
|
padding: 1rem;
|
||||||
margin-bottom: 1rem;
|
margin-bottom: 1rem;
|
||||||
border-style: solid;
|
border-style: solid;
|
||||||
border-color: var(--bolt-elements-button-danger-backgroundHover) ;
|
border-color: var(--bolt-elements-button-danger-backgroundHover);
|
||||||
border-width: thin;
|
border-width: thin;
|
||||||
|
|
||||||
button {
|
button {
|
||||||
@@ -60,4 +60,4 @@
|
|||||||
background-color: var(--bolt-elements-button-danger-backgroundHover);
|
background-color: var(--bolt-elements-button-danger-backgroundHover);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,17 +1,16 @@
|
|||||||
import * as RadixDialog from '@radix-ui/react-dialog';
|
import * as RadixDialog from '@radix-ui/react-dialog';
|
||||||
import { motion } from 'framer-motion';
|
import { motion } from 'framer-motion';
|
||||||
import { useState } from 'react';
|
import { useState, type ReactElement } from 'react';
|
||||||
import { classNames } from '~/utils/classNames';
|
import { classNames } from '~/utils/classNames';
|
||||||
import { DialogTitle, dialogVariants, dialogBackdropVariants } from '~/components/ui/Dialog';
|
import { DialogTitle, dialogVariants, dialogBackdropVariants } from '~/components/ui/Dialog';
|
||||||
import { IconButton } from '~/components/ui/IconButton';
|
import { IconButton } from '~/components/ui/IconButton';
|
||||||
import { providersList } from '~/lib/stores/settings';
|
|
||||||
import { db, getAll, deleteById } from '~/lib/persistence';
|
|
||||||
import { toast } from 'react-toastify';
|
|
||||||
import { useNavigate } from '@remix-run/react';
|
|
||||||
import commit from '~/commit.json';
|
|
||||||
import Cookies from 'js-cookie';
|
|
||||||
import styles from './Settings.module.scss';
|
import styles from './Settings.module.scss';
|
||||||
import { Switch } from '~/components/ui/Switch';
|
import ChatHistoryTab from './chat-history/ChatHistoryTab';
|
||||||
|
import ProvidersTab from './providers/ProvidersTab';
|
||||||
|
import { useSettings } from '~/lib/hooks/useSettings';
|
||||||
|
import FeaturesTab from './features/FeaturesTab';
|
||||||
|
import DebugTab from './debug/DebugTab';
|
||||||
|
import ConnectionsTab from './connections/ConnectionsTab';
|
||||||
|
|
||||||
interface SettingsProps {
|
interface SettingsProps {
|
||||||
open: boolean;
|
open: boolean;
|
||||||
@@ -21,206 +20,27 @@ interface SettingsProps {
|
|||||||
type TabType = 'chat-history' | 'providers' | 'features' | 'debug' | 'connection';
|
type TabType = 'chat-history' | 'providers' | 'features' | 'debug' | 'connection';
|
||||||
|
|
||||||
// Providers that support base URL configuration
|
// Providers that support base URL configuration
|
||||||
const URL_CONFIGURABLE_PROVIDERS = ['Ollama', 'LMStudio', 'OpenAILike'];
|
|
||||||
|
|
||||||
export const SettingsWindow = ({ open, onClose }: SettingsProps) => {
|
export const SettingsWindow = ({ open, onClose }: SettingsProps) => {
|
||||||
const navigate = useNavigate();
|
const { debug } = useSettings();
|
||||||
const [activeTab, setActiveTab] = useState<TabType>('chat-history');
|
const [activeTab, setActiveTab] = useState<TabType>('chat-history');
|
||||||
const [isDebugEnabled, setIsDebugEnabled] = useState(() => {
|
|
||||||
const savedDebugState = Cookies.get('isDebugEnabled');
|
|
||||||
return savedDebugState === 'true';
|
|
||||||
});
|
|
||||||
const [searchTerm, setSearchTerm] = useState('');
|
|
||||||
const [isDeleting, setIsDeleting] = useState(false);
|
|
||||||
const [githubUsername, setGithubUsername] = useState(Cookies.get('githubUsername') || '');
|
|
||||||
const [githubToken, setGithubToken] = useState(Cookies.get('githubToken') || '');
|
|
||||||
const [isLocalModelsEnabled, setIsLocalModelsEnabled] = useState(() => {
|
|
||||||
const savedLocalModelsState = Cookies.get('isLocalModelsEnabled');
|
|
||||||
return savedLocalModelsState === 'true';
|
|
||||||
});
|
|
||||||
|
|
||||||
// Load base URLs from cookies
|
const tabs: { id: TabType; label: string; icon: string; component?: ReactElement }[] = [
|
||||||
const [baseUrls, setBaseUrls] = useState(() => {
|
{ id: 'chat-history', label: 'Chat History', icon: 'i-ph:book', component: <ChatHistoryTab /> },
|
||||||
const savedUrls = Cookies.get('providerBaseUrls');
|
{ id: 'providers', label: 'Providers', icon: 'i-ph:key', component: <ProvidersTab /> },
|
||||||
|
{ id: 'features', label: 'Features', icon: 'i-ph:star', component: <FeaturesTab /> },
|
||||||
if (savedUrls) {
|
{ id: 'connection', label: 'Connection', icon: 'i-ph:link', component: <ConnectionsTab /> },
|
||||||
try {
|
...(debug
|
||||||
return JSON.parse(savedUrls);
|
? [
|
||||||
} catch (error) {
|
{
|
||||||
console.error('Failed to parse base URLs from cookies:', error);
|
id: 'debug' as TabType,
|
||||||
return {
|
label: 'Debug Tab',
|
||||||
Ollama: 'http://localhost:11434',
|
icon: 'i-ph:bug',
|
||||||
LMStudio: 'http://localhost:1234',
|
component: <DebugTab />,
|
||||||
OpenAILike: '',
|
},
|
||||||
};
|
]
|
||||||
}
|
: []),
|
||||||
}
|
|
||||||
|
|
||||||
return {
|
|
||||||
Ollama: 'http://localhost:11434',
|
|
||||||
LMStudio: 'http://localhost:1234',
|
|
||||||
OpenAILike: '',
|
|
||||||
};
|
|
||||||
});
|
|
||||||
|
|
||||||
const handleBaseUrlChange = (provider: string, url: string) => {
|
|
||||||
setBaseUrls((prev: Record<string, string>) => {
|
|
||||||
const newUrls = { ...prev, [provider]: url };
|
|
||||||
Cookies.set('providerBaseUrls', JSON.stringify(newUrls));
|
|
||||||
|
|
||||||
return newUrls;
|
|
||||||
});
|
|
||||||
};
|
|
||||||
|
|
||||||
const tabs: { id: TabType; label: string; icon: string }[] = [
|
|
||||||
{ id: 'chat-history', label: 'Chat History', icon: 'i-ph:book' },
|
|
||||||
{ id: 'providers', label: 'Providers', icon: 'i-ph:key' },
|
|
||||||
{ id: 'features', label: 'Features', icon: 'i-ph:star' },
|
|
||||||
{ id: 'connection', label: 'Connection', icon: 'i-ph:link' },
|
|
||||||
...(isDebugEnabled ? [{ id: 'debug' as TabType, label: 'Debug Tab', icon: 'i-ph:bug' }] : []),
|
|
||||||
];
|
];
|
||||||
|
|
||||||
// Load providers from cookies on mount
|
|
||||||
const [providers, setProviders] = useState(() => {
|
|
||||||
const savedProviders = Cookies.get('providers');
|
|
||||||
|
|
||||||
if (savedProviders) {
|
|
||||||
try {
|
|
||||||
const parsedProviders = JSON.parse(savedProviders);
|
|
||||||
|
|
||||||
// Merge saved enabled states with the base provider list
|
|
||||||
return providersList.map((provider) => ({
|
|
||||||
...provider,
|
|
||||||
isEnabled: parsedProviders[provider.name] || false,
|
|
||||||
}));
|
|
||||||
} catch (error) {
|
|
||||||
console.error('Failed to parse providers from cookies:', error);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return providersList;
|
|
||||||
});
|
|
||||||
|
|
||||||
const handleToggleProvider = (providerName: string, enabled: boolean) => {
|
|
||||||
setProviders((prevProviders) => {
|
|
||||||
const newProviders = prevProviders.map((provider) =>
|
|
||||||
provider.name === providerName ? { ...provider, isEnabled: enabled } : provider,
|
|
||||||
);
|
|
||||||
|
|
||||||
// Save to cookies
|
|
||||||
const enabledStates = newProviders.reduce(
|
|
||||||
(acc, provider) => ({
|
|
||||||
...acc,
|
|
||||||
[provider.name]: provider.isEnabled,
|
|
||||||
}),
|
|
||||||
{},
|
|
||||||
);
|
|
||||||
Cookies.set('providers', JSON.stringify(enabledStates));
|
|
||||||
|
|
||||||
return newProviders;
|
|
||||||
});
|
|
||||||
};
|
|
||||||
|
|
||||||
const filteredProviders = providers
|
|
||||||
.filter((provider) => {
|
|
||||||
const isLocalModelProvider = ['OpenAILike', 'LMStudio', 'Ollama'].includes(provider.name);
|
|
||||||
return isLocalModelsEnabled || !isLocalModelProvider;
|
|
||||||
})
|
|
||||||
.filter((provider) => provider.name.toLowerCase().includes(searchTerm.toLowerCase()))
|
|
||||||
.sort((a, b) => a.name.localeCompare(b.name));
|
|
||||||
|
|
||||||
const handleCopyToClipboard = () => {
|
|
||||||
const debugInfo = {
|
|
||||||
OS: navigator.platform,
|
|
||||||
Browser: navigator.userAgent,
|
|
||||||
ActiveFeatures: providers.filter((provider) => provider.isEnabled).map((provider) => provider.name),
|
|
||||||
BaseURLs: {
|
|
||||||
Ollama: process.env.REACT_APP_OLLAMA_URL,
|
|
||||||
OpenAI: process.env.REACT_APP_OPENAI_URL,
|
|
||||||
LMStudio: process.env.REACT_APP_LM_STUDIO_URL,
|
|
||||||
},
|
|
||||||
Version: versionHash,
|
|
||||||
};
|
|
||||||
navigator.clipboard.writeText(JSON.stringify(debugInfo, null, 2)).then(() => {
|
|
||||||
alert('Debug information copied to clipboard!');
|
|
||||||
});
|
|
||||||
};
|
|
||||||
|
|
||||||
const downloadAsJson = (data: any, filename: string) => {
|
|
||||||
const blob = new Blob([JSON.stringify(data, null, 2)], { type: 'application/json' });
|
|
||||||
const url = URL.createObjectURL(blob);
|
|
||||||
const link = document.createElement('a');
|
|
||||||
link.href = url;
|
|
||||||
link.download = filename;
|
|
||||||
document.body.appendChild(link);
|
|
||||||
link.click();
|
|
||||||
document.body.removeChild(link);
|
|
||||||
URL.revokeObjectURL(url);
|
|
||||||
};
|
|
||||||
|
|
||||||
const handleDeleteAllChats = async () => {
|
|
||||||
if (!db) {
|
|
||||||
toast.error('Database is not available');
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
setIsDeleting(true);
|
|
||||||
|
|
||||||
const allChats = await getAll(db);
|
|
||||||
|
|
||||||
// Delete all chats one by one
|
|
||||||
await Promise.all(allChats.map((chat) => deleteById(db!, chat.id)));
|
|
||||||
|
|
||||||
toast.success('All chats deleted successfully');
|
|
||||||
navigate('/', { replace: true });
|
|
||||||
} catch (error) {
|
|
||||||
toast.error('Failed to delete chats');
|
|
||||||
console.error(error);
|
|
||||||
} finally {
|
|
||||||
setIsDeleting(false);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
const handleExportAllChats = async () => {
|
|
||||||
if (!db) {
|
|
||||||
toast.error('Database is not available');
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
const allChats = await getAll(db);
|
|
||||||
const exportData = {
|
|
||||||
chats: allChats,
|
|
||||||
exportDate: new Date().toISOString(),
|
|
||||||
};
|
|
||||||
|
|
||||||
downloadAsJson(exportData, `all-chats-${new Date().toISOString()}.json`);
|
|
||||||
toast.success('Chats exported successfully');
|
|
||||||
} catch (error) {
|
|
||||||
toast.error('Failed to export chats');
|
|
||||||
console.error(error);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
const versionHash = commit.commit; // Get the version hash from commit.json
|
|
||||||
|
|
||||||
const handleSaveConnection = () => {
|
|
||||||
Cookies.set('githubUsername', githubUsername);
|
|
||||||
Cookies.set('githubToken', githubToken);
|
|
||||||
toast.success('GitHub credentials saved successfully!');
|
|
||||||
};
|
|
||||||
|
|
||||||
const handleToggleDebug = (enabled: boolean) => {
|
|
||||||
setIsDebugEnabled(enabled);
|
|
||||||
Cookies.set('isDebugEnabled', String(enabled));
|
|
||||||
};
|
|
||||||
|
|
||||||
const handleToggleLocalModels = (enabled: boolean) => {
|
|
||||||
setIsLocalModelsEnabled(enabled);
|
|
||||||
Cookies.set('isLocalModelsEnabled', String(enabled));
|
|
||||||
};
|
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<RadixDialog.Root open={open}>
|
<RadixDialog.Root open={open}>
|
||||||
<RadixDialog.Portal>
|
<RadixDialog.Portal>
|
||||||
@@ -284,192 +104,7 @@ export const SettingsWindow = ({ open, onClose }: SettingsProps) => {
|
|||||||
</div>
|
</div>
|
||||||
|
|
||||||
<div className="flex-1 flex flex-col p-8 pt-10 bg-bolt-elements-background-depth-2">
|
<div className="flex-1 flex flex-col p-8 pt-10 bg-bolt-elements-background-depth-2">
|
||||||
<div className="flex-1 overflow-y-auto">
|
<div className="flex-1 overflow-y-auto">{tabs.find((tab) => tab.id === activeTab)?.component}</div>
|
||||||
{activeTab === 'chat-history' && (
|
|
||||||
<div className="p-4">
|
|
||||||
<h3 className="text-lg font-medium text-bolt-elements-textPrimary mb-4">Chat History</h3>
|
|
||||||
<button
|
|
||||||
onClick={handleExportAllChats}
|
|
||||||
className={classNames(
|
|
||||||
'bg-bolt-elements-button-primary-background',
|
|
||||||
'rounded-lg px-4 py-2 mb-4 transition-colors duration-200',
|
|
||||||
'hover:bg-bolt-elements-button-primary-backgroundHover',
|
|
||||||
'text-bolt-elements-button-primary-text',
|
|
||||||
)}
|
|
||||||
>
|
|
||||||
Export All Chats
|
|
||||||
</button>
|
|
||||||
|
|
||||||
<div
|
|
||||||
className={classNames(
|
|
||||||
'text-bolt-elements-textPrimary rounded-lg py-4 mb-4',
|
|
||||||
styles['settings-danger-area'],
|
|
||||||
)}
|
|
||||||
>
|
|
||||||
<h4 className="font-semibold">Danger Area</h4>
|
|
||||||
<p className="mb-2">This action cannot be undone!</p>
|
|
||||||
<button
|
|
||||||
onClick={handleDeleteAllChats}
|
|
||||||
disabled={isDeleting}
|
|
||||||
className={classNames(
|
|
||||||
'bg-bolt-elements-button-danger-background',
|
|
||||||
'rounded-lg px-4 py-2 transition-colors duration-200',
|
|
||||||
isDeleting
|
|
||||||
? 'opacity-50 cursor-not-allowed'
|
|
||||||
: 'hover:bg-bolt-elements-button-danger-backgroundHover',
|
|
||||||
'text-bolt-elements-button-danger-text',
|
|
||||||
)}
|
|
||||||
>
|
|
||||||
{isDeleting ? 'Deleting...' : 'Delete All Chats'}
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
{activeTab === 'providers' && (
|
|
||||||
<div className="p-4">
|
|
||||||
<div className="flex mb-4">
|
|
||||||
<input
|
|
||||||
type="text"
|
|
||||||
placeholder="Search providers..."
|
|
||||||
value={searchTerm}
|
|
||||||
onChange={(e) => setSearchTerm(e.target.value)}
|
|
||||||
className="w-full bg-white dark:bg-bolt-elements-background-depth-4 relative px-2 py-1.5 rounded-md focus:outline-none placeholder-bolt-elements-textTertiary text-bolt-elements-textPrimary dark:text-bolt-elements-textPrimary border border-bolt-elements-borderColor"
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
{filteredProviders.map((provider) => (
|
|
||||||
<div
|
|
||||||
key={provider.name}
|
|
||||||
className="flex flex-col mb-2 provider-item hover:bg-bolt-elements-bg-depth-3 p-4 rounded-lg border border-bolt-elements-borderColor "
|
|
||||||
>
|
|
||||||
<div className="flex items-center justify-between mb-2">
|
|
||||||
<span className="text-bolt-elements-textPrimary">{provider.name}</span>
|
|
||||||
<Switch
|
|
||||||
className="ml-auto"
|
|
||||||
checked={provider.isEnabled}
|
|
||||||
onCheckedChange={(enabled) => handleToggleProvider(provider.name, enabled)}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
{/* Base URL input for configurable providers */}
|
|
||||||
{URL_CONFIGURABLE_PROVIDERS.includes(provider.name) && provider.isEnabled && (
|
|
||||||
<div className="mt-2">
|
|
||||||
<label className="block text-sm text-bolt-elements-textSecondary mb-1">Base URL:</label>
|
|
||||||
<input
|
|
||||||
type="text"
|
|
||||||
value={baseUrls[provider.name]}
|
|
||||||
onChange={(e) => handleBaseUrlChange(provider.name, e.target.value)}
|
|
||||||
placeholder={`Enter ${provider.name} base URL`}
|
|
||||||
className="w-full bg-white dark:bg-bolt-elements-background-depth-4 relative px-2 py-1.5 rounded-md focus:outline-none placeholder-bolt-elements-textTertiary text-bolt-elements-textPrimary dark:text-bolt-elements-textPrimary border border-bolt-elements-borderColor"
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
))}
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
{activeTab === 'features' && (
|
|
||||||
<div className="p-4 bg-bolt-elements-bg-depth-2 border border-bolt-elements-borderColor rounded-lg mb-4">
|
|
||||||
<div className="mb-6">
|
|
||||||
<h3 className="text-lg font-medium text-bolt-elements-textPrimary mb-4">Optional Features</h3>
|
|
||||||
<div className="flex items-center justify-between mb-2">
|
|
||||||
<span className="text-bolt-elements-textPrimary">Debug Info</span>
|
|
||||||
<Switch className="ml-auto" checked={isDebugEnabled} onCheckedChange={handleToggleDebug} />
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="mb-6 border-t border-bolt-elements-borderColor pt-4">
|
|
||||||
<h3 className="text-lg font-medium text-bolt-elements-textPrimary mb-4">
|
|
||||||
Experimental Features
|
|
||||||
</h3>
|
|
||||||
<p className="text-sm text-bolt-elements-textSecondary mb-4">
|
|
||||||
Disclaimer: Experimental features may be unstable and are subject to change.
|
|
||||||
</p>
|
|
||||||
<div className="flex items-center justify-between mb-2">
|
|
||||||
<span className="text-bolt-elements-textPrimary">Enable Local Models</span>
|
|
||||||
<Switch
|
|
||||||
className="ml-auto"
|
|
||||||
checked={isLocalModelsEnabled}
|
|
||||||
onCheckedChange={handleToggleLocalModels}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
{activeTab === 'debug' && isDebugEnabled && (
|
|
||||||
<div className="p-4">
|
|
||||||
<h3 className="text-lg font-medium text-bolt-elements-textPrimary mb-4">Debug Tab</h3>
|
|
||||||
<button
|
|
||||||
onClick={handleCopyToClipboard}
|
|
||||||
className="bg-blue-500 text-white rounded-lg px-4 py-2 hover:bg-blue-600 mb-4 transition-colors duration-200"
|
|
||||||
>
|
|
||||||
Copy to Clipboard
|
|
||||||
</button>
|
|
||||||
|
|
||||||
<h4 className="text-md font-medium text-bolt-elements-textPrimary">System Information</h4>
|
|
||||||
<p className="text-bolt-elements-textSecondary">OS: {navigator.platform}</p>
|
|
||||||
<p className="text-bolt-elements-textSecondary">Browser: {navigator.userAgent}</p>
|
|
||||||
|
|
||||||
<h4 className="text-md font-medium text-bolt-elements-textPrimary mt-4">Active Features</h4>
|
|
||||||
<ul>
|
|
||||||
{providers
|
|
||||||
.filter((provider) => provider.isEnabled)
|
|
||||||
.map((provider) => (
|
|
||||||
<li key={provider.name} className="text-bolt-elements-textSecondary">
|
|
||||||
{provider.name}
|
|
||||||
</li>
|
|
||||||
))}
|
|
||||||
</ul>
|
|
||||||
|
|
||||||
<h4 className="text-md font-medium text-bolt-elements-textPrimary mt-4">Base URLs</h4>
|
|
||||||
<ul>
|
|
||||||
<li className="text-bolt-elements-textSecondary">Ollama: {process.env.REACT_APP_OLLAMA_URL}</li>
|
|
||||||
<li className="text-bolt-elements-textSecondary">OpenAI: {process.env.REACT_APP_OPENAI_URL}</li>
|
|
||||||
<li className="text-bolt-elements-textSecondary">
|
|
||||||
LM Studio: {process.env.REACT_APP_LM_STUDIO_URL}
|
|
||||||
</li>
|
|
||||||
</ul>
|
|
||||||
|
|
||||||
<h4 className="text-md font-medium text-bolt-elements-textPrimary mt-4">Version Information</h4>
|
|
||||||
<p className="text-bolt-elements-textSecondary">Version Hash: {versionHash}</p>
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
{activeTab === 'connection' && (
|
|
||||||
<div className="p-4 mb-4 border border-bolt-elements-borderColor rounded-lg bg-bolt-elements-background-depth-3">
|
|
||||||
<h3 className="text-lg font-medium text-bolt-elements-textPrimary mb-4">GitHub Connection</h3>
|
|
||||||
<div className="flex mb-4">
|
|
||||||
<div className="flex-1 mr-2">
|
|
||||||
<label className="block text-sm text-bolt-elements-textSecondary mb-1">
|
|
||||||
GitHub Username:
|
|
||||||
</label>
|
|
||||||
<input
|
|
||||||
type="text"
|
|
||||||
value={githubUsername}
|
|
||||||
onChange={(e) => setGithubUsername(e.target.value)}
|
|
||||||
className="w-full bg-white dark:bg-bolt-elements-background-depth-4 relative px-2 py-1.5 rounded-md focus:outline-none placeholder-bolt-elements-textTertiary text-bolt-elements-textPrimary dark:text-bolt-elements-textPrimary border border-bolt-elements-borderColor"
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
<div className="flex-1">
|
|
||||||
<label className="block text-sm text-bolt-elements-textSecondary mb-1">
|
|
||||||
Personal Access Token:
|
|
||||||
</label>
|
|
||||||
<input
|
|
||||||
type="password"
|
|
||||||
value={githubToken}
|
|
||||||
onChange={(e) => setGithubToken(e.target.value)}
|
|
||||||
className="w-full bg-white dark:bg-bolt-elements-background-depth-4 relative px-2 py-1.5 rounded-md focus:outline-none placeholder-bolt-elements-textTertiary text-bolt-elements-textPrimary dark:text-bolt-elements-textPrimary border border-bolt-elements-borderColor"
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
<div className="flex mb-4">
|
|
||||||
<button
|
|
||||||
onClick={handleSaveConnection}
|
|
||||||
className="bg-bolt-elements-button-primary-background rounded-lg px-4 py-2 mr-2 transition-colors duration-200 hover:bg-bolt-elements-button-primary-backgroundHover text-bolt-elements-button-primary-text"
|
|
||||||
>
|
|
||||||
Save Connection
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
<RadixDialog.Close asChild onClick={onClose}>
|
<RadixDialog.Close asChild onClick={onClose}>
|
||||||
|
|||||||
105
app/components/settings/chat-history/ChatHistoryTab.tsx
Normal file
105
app/components/settings/chat-history/ChatHistoryTab.tsx
Normal file
@@ -0,0 +1,105 @@
|
|||||||
|
import { useNavigate } from '@remix-run/react';
|
||||||
|
import React, { useState } from 'react';
|
||||||
|
import { toast } from 'react-toastify';
|
||||||
|
import { db, deleteById, getAll } from '~/lib/persistence';
|
||||||
|
import { classNames } from '~/utils/classNames';
|
||||||
|
import styles from '~/components/settings/Settings.module.scss';
|
||||||
|
|
||||||
|
export default function ChatHistoryTab() {
|
||||||
|
const navigate = useNavigate();
|
||||||
|
const [isDeleting, setIsDeleting] = useState(false);
|
||||||
|
const downloadAsJson = (data: any, filename: string) => {
|
||||||
|
const blob = new Blob([JSON.stringify(data, null, 2)], { type: 'application/json' });
|
||||||
|
const url = URL.createObjectURL(blob);
|
||||||
|
const link = document.createElement('a');
|
||||||
|
link.href = url;
|
||||||
|
link.download = filename;
|
||||||
|
document.body.appendChild(link);
|
||||||
|
link.click();
|
||||||
|
document.body.removeChild(link);
|
||||||
|
URL.revokeObjectURL(url);
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleDeleteAllChats = async () => {
|
||||||
|
if (!db) {
|
||||||
|
toast.error('Database is not available');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
setIsDeleting(true);
|
||||||
|
|
||||||
|
const allChats = await getAll(db);
|
||||||
|
|
||||||
|
// Delete all chats one by one
|
||||||
|
await Promise.all(allChats.map((chat) => deleteById(db!, chat.id)));
|
||||||
|
|
||||||
|
toast.success('All chats deleted successfully');
|
||||||
|
navigate('/', { replace: true });
|
||||||
|
} catch (error) {
|
||||||
|
toast.error('Failed to delete chats');
|
||||||
|
console.error(error);
|
||||||
|
} finally {
|
||||||
|
setIsDeleting(false);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleExportAllChats = async () => {
|
||||||
|
if (!db) {
|
||||||
|
toast.error('Database is not available');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
const allChats = await getAll(db);
|
||||||
|
const exportData = {
|
||||||
|
chats: allChats,
|
||||||
|
exportDate: new Date().toISOString(),
|
||||||
|
};
|
||||||
|
|
||||||
|
downloadAsJson(exportData, `all-chats-${new Date().toISOString()}.json`);
|
||||||
|
toast.success('Chats exported successfully');
|
||||||
|
} catch (error) {
|
||||||
|
toast.error('Failed to export chats');
|
||||||
|
console.error(error);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
return (
|
||||||
|
<>
|
||||||
|
<div className="p-4">
|
||||||
|
<h3 className="text-lg font-medium text-bolt-elements-textPrimary mb-4">Chat History</h3>
|
||||||
|
<button
|
||||||
|
onClick={handleExportAllChats}
|
||||||
|
className={classNames(
|
||||||
|
'bg-bolt-elements-button-primary-background',
|
||||||
|
'rounded-lg px-4 py-2 mb-4 transition-colors duration-200',
|
||||||
|
'hover:bg-bolt-elements-button-primary-backgroundHover',
|
||||||
|
'text-bolt-elements-button-primary-text',
|
||||||
|
)}
|
||||||
|
>
|
||||||
|
Export All Chats
|
||||||
|
</button>
|
||||||
|
|
||||||
|
<div
|
||||||
|
className={classNames('text-bolt-elements-textPrimary rounded-lg py-4 mb-4', styles['settings-danger-area'])}
|
||||||
|
>
|
||||||
|
<h4 className="font-semibold">Danger Area</h4>
|
||||||
|
<p className="mb-2">This action cannot be undone!</p>
|
||||||
|
<button
|
||||||
|
onClick={handleDeleteAllChats}
|
||||||
|
disabled={isDeleting}
|
||||||
|
className={classNames(
|
||||||
|
'bg-bolt-elements-button-danger-background',
|
||||||
|
'rounded-lg px-4 py-2 transition-colors duration-200',
|
||||||
|
isDeleting ? 'opacity-50 cursor-not-allowed' : 'hover:bg-bolt-elements-button-danger-backgroundHover',
|
||||||
|
'text-bolt-elements-button-danger-text',
|
||||||
|
)}
|
||||||
|
>
|
||||||
|
{isDeleting ? 'Deleting...' : 'Delete All Chats'}
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</>
|
||||||
|
);
|
||||||
|
}
|
||||||
48
app/components/settings/connections/ConnectionsTab.tsx
Normal file
48
app/components/settings/connections/ConnectionsTab.tsx
Normal file
@@ -0,0 +1,48 @@
|
|||||||
|
import React, { useState } from 'react';
|
||||||
|
import { toast } from 'react-toastify';
|
||||||
|
import Cookies from 'js-cookie';
|
||||||
|
|
||||||
|
export default function ConnectionsTab() {
|
||||||
|
const [githubUsername, setGithubUsername] = useState(Cookies.get('githubUsername') || '');
|
||||||
|
const [githubToken, setGithubToken] = useState(Cookies.get('githubToken') || '');
|
||||||
|
|
||||||
|
const handleSaveConnection = () => {
|
||||||
|
Cookies.set('githubUsername', githubUsername);
|
||||||
|
Cookies.set('githubToken', githubToken);
|
||||||
|
toast.success('GitHub credentials saved successfully!');
|
||||||
|
};
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="p-4 mb-4 border border-bolt-elements-borderColor rounded-lg bg-bolt-elements-background-depth-3">
|
||||||
|
<h3 className="text-lg font-medium text-bolt-elements-textPrimary mb-4">GitHub Connection</h3>
|
||||||
|
<div className="flex mb-4">
|
||||||
|
<div className="flex-1 mr-2">
|
||||||
|
<label className="block text-sm text-bolt-elements-textSecondary mb-1">GitHub Username:</label>
|
||||||
|
<input
|
||||||
|
type="text"
|
||||||
|
value={githubUsername}
|
||||||
|
onChange={(e) => setGithubUsername(e.target.value)}
|
||||||
|
className="w-full bg-white dark:bg-bolt-elements-background-depth-4 relative px-2 py-1.5 rounded-md focus:outline-none placeholder-bolt-elements-textTertiary text-bolt-elements-textPrimary dark:text-bolt-elements-textPrimary border border-bolt-elements-borderColor"
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
<div className="flex-1">
|
||||||
|
<label className="block text-sm text-bolt-elements-textSecondary mb-1">Personal Access Token:</label>
|
||||||
|
<input
|
||||||
|
type="password"
|
||||||
|
value={githubToken}
|
||||||
|
onChange={(e) => setGithubToken(e.target.value)}
|
||||||
|
className="w-full bg-white dark:bg-bolt-elements-background-depth-4 relative px-2 py-1.5 rounded-md focus:outline-none placeholder-bolt-elements-textTertiary text-bolt-elements-textPrimary dark:text-bolt-elements-textPrimary border border-bolt-elements-borderColor"
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div className="flex mb-4">
|
||||||
|
<button
|
||||||
|
onClick={handleSaveConnection}
|
||||||
|
className="bg-bolt-elements-button-primary-background rounded-lg px-4 py-2 mr-2 transition-colors duration-200 hover:bg-bolt-elements-button-primary-backgroundHover text-bolt-elements-button-primary-text"
|
||||||
|
>
|
||||||
|
Save Connection
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
69
app/components/settings/debug/DebugTab.tsx
Normal file
69
app/components/settings/debug/DebugTab.tsx
Normal file
@@ -0,0 +1,69 @@
|
|||||||
|
import React, { useCallback, useEffect, useState } from 'react';
|
||||||
|
import { useSettings } from '~/lib/hooks/useSettings';
|
||||||
|
import commit from '~/commit.json';
|
||||||
|
|
||||||
|
const versionHash = commit.commit; // Get the version hash from commit.json
|
||||||
|
|
||||||
|
export default function DebugTab() {
|
||||||
|
const { providers } = useSettings();
|
||||||
|
const [activeProviders, setActiveProviders] = useState<string[]>([]);
|
||||||
|
useEffect(() => {
|
||||||
|
setActiveProviders(
|
||||||
|
Object.entries(providers)
|
||||||
|
.filter(([_key, provider]) => provider.settings.enabled)
|
||||||
|
.map(([_key, provider]) => provider.name),
|
||||||
|
);
|
||||||
|
}, [providers]);
|
||||||
|
|
||||||
|
const handleCopyToClipboard = useCallback(() => {
|
||||||
|
const debugInfo = {
|
||||||
|
OS: navigator.platform,
|
||||||
|
Browser: navigator.userAgent,
|
||||||
|
ActiveFeatures: activeProviders,
|
||||||
|
BaseURLs: {
|
||||||
|
Ollama: process.env.REACT_APP_OLLAMA_URL,
|
||||||
|
OpenAI: process.env.REACT_APP_OPENAI_URL,
|
||||||
|
LMStudio: process.env.REACT_APP_LM_STUDIO_URL,
|
||||||
|
},
|
||||||
|
Version: versionHash,
|
||||||
|
};
|
||||||
|
navigator.clipboard.writeText(JSON.stringify(debugInfo, null, 2)).then(() => {
|
||||||
|
alert('Debug information copied to clipboard!');
|
||||||
|
});
|
||||||
|
}, [providers]);
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="p-4">
|
||||||
|
<h3 className="text-lg font-medium text-bolt-elements-textPrimary mb-4">Debug Tab</h3>
|
||||||
|
<button
|
||||||
|
onClick={handleCopyToClipboard}
|
||||||
|
className="bg-blue-500 text-white rounded-lg px-4 py-2 hover:bg-blue-600 mb-4 transition-colors duration-200"
|
||||||
|
>
|
||||||
|
Copy to Clipboard
|
||||||
|
</button>
|
||||||
|
|
||||||
|
<h4 className="text-md font-medium text-bolt-elements-textPrimary">System Information</h4>
|
||||||
|
<p className="text-bolt-elements-textSecondary">OS: {navigator.platform}</p>
|
||||||
|
<p className="text-bolt-elements-textSecondary">Browser: {navigator.userAgent}</p>
|
||||||
|
|
||||||
|
<h4 className="text-md font-medium text-bolt-elements-textPrimary mt-4">Active Features</h4>
|
||||||
|
<ul>
|
||||||
|
{activeProviders.map((name) => (
|
||||||
|
<li key={name} className="text-bolt-elements-textSecondary">
|
||||||
|
{name}
|
||||||
|
</li>
|
||||||
|
))}
|
||||||
|
</ul>
|
||||||
|
|
||||||
|
<h4 className="text-md font-medium text-bolt-elements-textPrimary mt-4">Base URLs</h4>
|
||||||
|
<ul>
|
||||||
|
<li className="text-bolt-elements-textSecondary">Ollama: {process.env.REACT_APP_OLLAMA_URL}</li>
|
||||||
|
<li className="text-bolt-elements-textSecondary">OpenAI: {process.env.REACT_APP_OPENAI_URL}</li>
|
||||||
|
<li className="text-bolt-elements-textSecondary">LM Studio: {process.env.REACT_APP_LM_STUDIO_URL}</li>
|
||||||
|
</ul>
|
||||||
|
|
||||||
|
<h4 className="text-md font-medium text-bolt-elements-textPrimary mt-4">Version Information</h4>
|
||||||
|
<p className="text-bolt-elements-textSecondary">Version Hash: {versionHash}</p>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
29
app/components/settings/features/FeaturesTab.tsx
Normal file
29
app/components/settings/features/FeaturesTab.tsx
Normal file
@@ -0,0 +1,29 @@
|
|||||||
|
import React from 'react';
|
||||||
|
import { Switch } from '~/components/ui/Switch';
|
||||||
|
import { useSettings } from '~/lib/hooks/useSettings';
|
||||||
|
|
||||||
|
export default function FeaturesTab() {
|
||||||
|
const { debug, enableDebugMode, isLocalModel, enableLocalModels } = useSettings();
|
||||||
|
return (
|
||||||
|
<div className="p-4 bg-bolt-elements-bg-depth-2 border border-bolt-elements-borderColor rounded-lg mb-4">
|
||||||
|
<div className="mb-6">
|
||||||
|
<h3 className="text-lg font-medium text-bolt-elements-textPrimary mb-4">Optional Features</h3>
|
||||||
|
<div className="flex items-center justify-between mb-2">
|
||||||
|
<span className="text-bolt-elements-textPrimary">Debug Info</span>
|
||||||
|
<Switch className="ml-auto" checked={debug} onCheckedChange={enableDebugMode} />
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="mb-6 border-t border-bolt-elements-borderColor pt-4">
|
||||||
|
<h3 className="text-lg font-medium text-bolt-elements-textPrimary mb-4">Experimental Features</h3>
|
||||||
|
<p className="text-sm text-bolt-elements-textSecondary mb-4">
|
||||||
|
Disclaimer: Experimental features may be unstable and are subject to change.
|
||||||
|
</p>
|
||||||
|
<div className="flex items-center justify-between mb-2">
|
||||||
|
<span className="text-bolt-elements-textPrimary">Enable Local Models</span>
|
||||||
|
<Switch className="ml-auto" checked={isLocalModel} onCheckedChange={enableLocalModels} />
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
78
app/components/settings/providers/ProvidersTab.tsx
Normal file
78
app/components/settings/providers/ProvidersTab.tsx
Normal file
@@ -0,0 +1,78 @@
|
|||||||
|
import React, { useEffect, useState } from 'react';
|
||||||
|
import { Switch } from '~/components/ui/Switch';
|
||||||
|
import { useSettings } from '~/lib/hooks/useSettings';
|
||||||
|
import { LOCAL_PROVIDERS, URL_CONFIGURABLE_PROVIDERS } from '~/lib/stores/settings';
|
||||||
|
import type { IProviderConfig } from '~/types/model';
|
||||||
|
|
||||||
|
export default function ProvidersTab() {
|
||||||
|
const { providers, updateProviderSettings, isLocalModel } = useSettings();
|
||||||
|
const [filteredProviders, setFilteredProviders] = useState<IProviderConfig[]>([]);
|
||||||
|
|
||||||
|
// Load base URLs from cookies
|
||||||
|
const [searchTerm, setSearchTerm] = useState('');
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
let newFilteredProviders: IProviderConfig[] = Object.entries(providers).map(([key, value]) => ({
|
||||||
|
...value,
|
||||||
|
name: key,
|
||||||
|
}));
|
||||||
|
|
||||||
|
if (searchTerm && searchTerm.length > 0) {
|
||||||
|
newFilteredProviders = newFilteredProviders.filter((provider) =>
|
||||||
|
provider.name.toLowerCase().includes(searchTerm.toLowerCase()),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!isLocalModel) {
|
||||||
|
newFilteredProviders = newFilteredProviders.filter((provider) => !LOCAL_PROVIDERS.includes(provider.name));
|
||||||
|
}
|
||||||
|
|
||||||
|
newFilteredProviders.sort((a, b) => a.name.localeCompare(b.name));
|
||||||
|
|
||||||
|
setFilteredProviders(newFilteredProviders);
|
||||||
|
}, [providers, searchTerm, isLocalModel]);
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="p-4">
|
||||||
|
<div className="flex mb-4">
|
||||||
|
<input
|
||||||
|
type="text"
|
||||||
|
placeholder="Search providers..."
|
||||||
|
value={searchTerm}
|
||||||
|
onChange={(e) => setSearchTerm(e.target.value)}
|
||||||
|
className="w-full bg-white dark:bg-bolt-elements-background-depth-4 relative px-2 py-1.5 rounded-md focus:outline-none placeholder-bolt-elements-textTertiary text-bolt-elements-textPrimary dark:text-bolt-elements-textPrimary border border-bolt-elements-borderColor"
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
{filteredProviders.map((provider) => (
|
||||||
|
<div
|
||||||
|
key={provider.name}
|
||||||
|
className="flex flex-col mb-2 provider-item hover:bg-bolt-elements-bg-depth-3 p-4 rounded-lg border border-bolt-elements-borderColor "
|
||||||
|
>
|
||||||
|
<div className="flex items-center justify-between mb-2">
|
||||||
|
<span className="text-bolt-elements-textPrimary">{provider.name}</span>
|
||||||
|
<Switch
|
||||||
|
className="ml-auto"
|
||||||
|
checked={provider.settings.enabled}
|
||||||
|
onCheckedChange={(enabled) => updateProviderSettings(provider.name, { ...provider.settings, enabled })}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
{/* Base URL input for configurable providers */}
|
||||||
|
{URL_CONFIGURABLE_PROVIDERS.includes(provider.name) && provider.settings.enabled && (
|
||||||
|
<div className="mt-2">
|
||||||
|
<label className="block text-sm text-bolt-elements-textSecondary mb-1">Base URL:</label>
|
||||||
|
<input
|
||||||
|
type="text"
|
||||||
|
value={provider.settings.baseUrl || ''}
|
||||||
|
onChange={(e) =>
|
||||||
|
updateProviderSettings(provider.name, { ...provider.settings, baseUrl: e.target.value })
|
||||||
|
}
|
||||||
|
placeholder={`Enter ${provider.name} base URL`}
|
||||||
|
className="w-full bg-white dark:bg-bolt-elements-background-depth-4 relative px-2 py-1.5 rounded-md focus:outline-none placeholder-bolt-elements-textTertiary text-bolt-elements-textPrimary dark:text-bolt-elements-textPrimary border border-bolt-elements-borderColor"
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
@@ -11,6 +11,7 @@ import { createOpenRouter } from '@openrouter/ai-sdk-provider';
|
|||||||
import { createMistral } from '@ai-sdk/mistral';
|
import { createMistral } from '@ai-sdk/mistral';
|
||||||
import { createCohere } from '@ai-sdk/cohere';
|
import { createCohere } from '@ai-sdk/cohere';
|
||||||
import type { LanguageModelV1 } from 'ai';
|
import type { LanguageModelV1 } from 'ai';
|
||||||
|
import type { IProviderSetting } from '~/types/model';
|
||||||
|
|
||||||
export const DEFAULT_NUM_CTX = process.env.DEFAULT_NUM_CTX ? parseInt(process.env.DEFAULT_NUM_CTX, 10) : 32768;
|
export const DEFAULT_NUM_CTX = process.env.DEFAULT_NUM_CTX ? parseInt(process.env.DEFAULT_NUM_CTX, 10) : 32768;
|
||||||
|
|
||||||
@@ -127,14 +128,20 @@ export function getXAIModel(apiKey: OptionalApiKey, model: string) {
|
|||||||
return openai(model);
|
return openai(model);
|
||||||
}
|
}
|
||||||
|
|
||||||
export function getModel(provider: string, model: string, env: Env, apiKeys?: Record<string, string>) {
|
export function getModel(
|
||||||
|
provider: string,
|
||||||
|
model: string,
|
||||||
|
env: Env,
|
||||||
|
apiKeys?: Record<string, string>,
|
||||||
|
providerSettings?: Record<string, IProviderSetting>,
|
||||||
|
) {
|
||||||
/*
|
/*
|
||||||
* let apiKey; // Declare first
|
* let apiKey; // Declare first
|
||||||
* let baseURL;
|
* let baseURL;
|
||||||
*/
|
*/
|
||||||
|
|
||||||
const apiKey = getAPIKey(env, provider, apiKeys); // Then assign
|
const apiKey = getAPIKey(env, provider, apiKeys); // Then assign
|
||||||
const baseURL = getBaseURL(env, provider);
|
const baseURL = providerSettings?.[provider].baseUrl || getBaseURL(env, provider);
|
||||||
|
|
||||||
switch (provider) {
|
switch (provider) {
|
||||||
case 'Anthropic':
|
case 'Anthropic':
|
||||||
|
|||||||
@@ -3,6 +3,7 @@ import { getModel } from '~/lib/.server/llm/model';
|
|||||||
import { MAX_TOKENS } from './constants';
|
import { MAX_TOKENS } from './constants';
|
||||||
import { getSystemPrompt } from './prompts';
|
import { getSystemPrompt } from './prompts';
|
||||||
import { DEFAULT_MODEL, DEFAULT_PROVIDER, getModelList, MODEL_REGEX, PROVIDER_REGEX } from '~/utils/constants';
|
import { DEFAULT_MODEL, DEFAULT_PROVIDER, getModelList, MODEL_REGEX, PROVIDER_REGEX } from '~/utils/constants';
|
||||||
|
import type { IProviderSetting } from '~/types/model';
|
||||||
|
|
||||||
interface ToolResult<Name extends string, Args, Result> {
|
interface ToolResult<Name extends string, Args, Result> {
|
||||||
toolCallId: string;
|
toolCallId: string;
|
||||||
@@ -58,15 +59,17 @@ function extractPropertiesFromMessage(message: Message): { model: string; provid
|
|||||||
return { model, provider, content: cleanedContent };
|
return { model, provider, content: cleanedContent };
|
||||||
}
|
}
|
||||||
|
|
||||||
export async function streamText(
|
export async function streamText(props: {
|
||||||
messages: Messages,
|
messages: Messages;
|
||||||
env: Env,
|
env: Env;
|
||||||
options?: StreamingOptions,
|
options?: StreamingOptions;
|
||||||
apiKeys?: Record<string, string>,
|
apiKeys?: Record<string, string>;
|
||||||
) {
|
providerSettings?: Record<string, IProviderSetting>;
|
||||||
|
}) {
|
||||||
|
const { messages, env, options, apiKeys, providerSettings } = props;
|
||||||
let currentModel = DEFAULT_MODEL;
|
let currentModel = DEFAULT_MODEL;
|
||||||
let currentProvider = DEFAULT_PROVIDER.name;
|
let currentProvider = DEFAULT_PROVIDER.name;
|
||||||
const MODEL_LIST = await getModelList(apiKeys || {});
|
const MODEL_LIST = await getModelList(apiKeys || {}, providerSettings);
|
||||||
const processedMessages = messages.map((message) => {
|
const processedMessages = messages.map((message) => {
|
||||||
if (message.role === 'user') {
|
if (message.role === 'user') {
|
||||||
const { model, provider, content } = extractPropertiesFromMessage(message);
|
const { model, provider, content } = extractPropertiesFromMessage(message);
|
||||||
@@ -88,7 +91,7 @@ export async function streamText(
|
|||||||
const dynamicMaxTokens = modelDetails && modelDetails.maxTokenAllowed ? modelDetails.maxTokenAllowed : MAX_TOKENS;
|
const dynamicMaxTokens = modelDetails && modelDetails.maxTokenAllowed ? modelDetails.maxTokenAllowed : MAX_TOKENS;
|
||||||
|
|
||||||
return _streamText({
|
return _streamText({
|
||||||
model: getModel(currentProvider, currentModel, env, apiKeys) as any,
|
model: getModel(currentProvider, currentModel, env, apiKeys, providerSettings) as any,
|
||||||
system: getSystemPrompt(),
|
system: getSystemPrompt(),
|
||||||
maxTokens: dynamicMaxTokens,
|
maxTokens: dynamicMaxTokens,
|
||||||
messages: convertToCoreMessages(processedMessages as any),
|
messages: convertToCoreMessages(processedMessages as any),
|
||||||
|
|||||||
97
app/lib/hooks/useSettings.tsx
Normal file
97
app/lib/hooks/useSettings.tsx
Normal file
@@ -0,0 +1,97 @@
|
|||||||
|
import { useStore } from '@nanostores/react';
|
||||||
|
import { isDebugMode, isLocalModelsEnabled, LOCAL_PROVIDERS, providersStore } from '~/lib/stores/settings';
|
||||||
|
import { useCallback, useEffect, useState } from 'react';
|
||||||
|
import Cookies from 'js-cookie';
|
||||||
|
import type { IProviderSetting, ProviderInfo } from '~/types/model';
|
||||||
|
|
||||||
|
export function useSettings() {
|
||||||
|
const providers = useStore(providersStore);
|
||||||
|
const debug = useStore(isDebugMode);
|
||||||
|
const isLocalModel = useStore(isLocalModelsEnabled);
|
||||||
|
const [activeProviders, setActiveProviders] = useState<ProviderInfo[]>([]);
|
||||||
|
|
||||||
|
// reading values from cookies on mount
|
||||||
|
useEffect(() => {
|
||||||
|
const savedProviders = Cookies.get('providers');
|
||||||
|
|
||||||
|
if (savedProviders) {
|
||||||
|
try {
|
||||||
|
const parsedProviders: Record<string, IProviderSetting> = JSON.parse(savedProviders);
|
||||||
|
Object.keys(parsedProviders).forEach((provider) => {
|
||||||
|
const currentProvider = providers[provider];
|
||||||
|
providersStore.setKey(provider, {
|
||||||
|
...currentProvider,
|
||||||
|
settings: {
|
||||||
|
...parsedProviders[provider],
|
||||||
|
enabled: parsedProviders[provider].enabled || true,
|
||||||
|
},
|
||||||
|
});
|
||||||
|
});
|
||||||
|
} catch (error) {
|
||||||
|
console.error('Failed to parse providers from cookies:', error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// load debug mode from cookies
|
||||||
|
const savedDebugMode = Cookies.get('isDebugEnabled');
|
||||||
|
|
||||||
|
if (savedDebugMode) {
|
||||||
|
isDebugMode.set(savedDebugMode === 'true');
|
||||||
|
}
|
||||||
|
|
||||||
|
// load local models from cookies
|
||||||
|
const savedLocalModels = Cookies.get('isLocalModelsEnabled');
|
||||||
|
|
||||||
|
if (savedLocalModels) {
|
||||||
|
isLocalModelsEnabled.set(savedLocalModels === 'true');
|
||||||
|
}
|
||||||
|
}, []);
|
||||||
|
|
||||||
|
// writing values to cookies on change
|
||||||
|
useEffect(() => {
|
||||||
|
const providers = providersStore.get();
|
||||||
|
const providerSetting: Record<string, IProviderSetting> = {};
|
||||||
|
Object.keys(providers).forEach((provider) => {
|
||||||
|
providerSetting[provider] = providers[provider].settings;
|
||||||
|
});
|
||||||
|
Cookies.set('providers', JSON.stringify(providerSetting));
|
||||||
|
}, [providers]);
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
let active = Object.entries(providers)
|
||||||
|
.filter(([_key, provider]) => provider.settings.enabled)
|
||||||
|
.map(([_k, p]) => p);
|
||||||
|
|
||||||
|
if (!isLocalModel) {
|
||||||
|
active = active.filter((p) => !LOCAL_PROVIDERS.includes(p.name));
|
||||||
|
}
|
||||||
|
|
||||||
|
setActiveProviders(active);
|
||||||
|
}, [providers, isLocalModel]);
|
||||||
|
|
||||||
|
// helper function to update settings
|
||||||
|
const updateProviderSettings = useCallback((provider: string, config: IProviderSetting) => {
|
||||||
|
const settings = providers[provider].settings;
|
||||||
|
providersStore.setKey(provider, { ...providers[provider], settings: { ...settings, ...config } });
|
||||||
|
}, []);
|
||||||
|
|
||||||
|
const enableDebugMode = useCallback((enabled: boolean) => {
|
||||||
|
isDebugMode.set(enabled);
|
||||||
|
Cookies.set('isDebugEnabled', String(enabled));
|
||||||
|
}, []);
|
||||||
|
|
||||||
|
const enableLocalModels = useCallback((enabled: boolean) => {
|
||||||
|
isLocalModelsEnabled.set(enabled);
|
||||||
|
Cookies.set('isLocalModelsEnabled', String(enabled));
|
||||||
|
}, []);
|
||||||
|
|
||||||
|
return {
|
||||||
|
providers,
|
||||||
|
activeProviders,
|
||||||
|
updateProviderSettings,
|
||||||
|
debug,
|
||||||
|
enableDebugMode,
|
||||||
|
isLocalModel,
|
||||||
|
enableLocalModels,
|
||||||
|
};
|
||||||
|
}
|
||||||
@@ -1,5 +1,7 @@
|
|||||||
import { map } from 'nanostores';
|
import { atom, map } from 'nanostores';
|
||||||
import { workbenchStore } from './workbench';
|
import { workbenchStore } from './workbench';
|
||||||
|
import { PROVIDER_LIST } from '~/utils/constants';
|
||||||
|
import type { IProviderConfig } from '~/types/model';
|
||||||
|
|
||||||
export interface Shortcut {
|
export interface Shortcut {
|
||||||
key: string;
|
key: string;
|
||||||
@@ -15,32 +17,10 @@ export interface Shortcuts {
|
|||||||
toggleTerminal: Shortcut;
|
toggleTerminal: Shortcut;
|
||||||
}
|
}
|
||||||
|
|
||||||
export interface Provider {
|
export const URL_CONFIGURABLE_PROVIDERS = ['Ollama', 'LMStudio', 'OpenAILike'];
|
||||||
name: string;
|
export const LOCAL_PROVIDERS = ['OpenAILike', 'LMStudio', 'Ollama'];
|
||||||
isEnabled: boolean;
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface Settings {
|
export type ProviderSetting = Record<string, IProviderConfig>;
|
||||||
shortcuts: Shortcuts;
|
|
||||||
providers: Provider[];
|
|
||||||
}
|
|
||||||
|
|
||||||
export const providersList: Provider[] = [
|
|
||||||
{ name: 'Groq', isEnabled: false },
|
|
||||||
{ name: 'HuggingFace', isEnabled: false },
|
|
||||||
{ name: 'OpenAI', isEnabled: false },
|
|
||||||
{ name: 'Anthropic', isEnabled: false },
|
|
||||||
{ name: 'OpenRouter', isEnabled: false },
|
|
||||||
{ name: 'Google', isEnabled: false },
|
|
||||||
{ name: 'Ollama', isEnabled: false },
|
|
||||||
{ name: 'OpenAILike', isEnabled: false },
|
|
||||||
{ name: 'Together', isEnabled: false },
|
|
||||||
{ name: 'Deepseek', isEnabled: false },
|
|
||||||
{ name: 'Mistral', isEnabled: false },
|
|
||||||
{ name: 'Cohere', isEnabled: false },
|
|
||||||
{ name: 'LMStudio', isEnabled: false },
|
|
||||||
{ name: 'xAI', isEnabled: false },
|
|
||||||
];
|
|
||||||
|
|
||||||
export const shortcutsStore = map<Shortcuts>({
|
export const shortcutsStore = map<Shortcuts>({
|
||||||
toggleTerminal: {
|
toggleTerminal: {
|
||||||
@@ -50,14 +30,17 @@ export const shortcutsStore = map<Shortcuts>({
|
|||||||
},
|
},
|
||||||
});
|
});
|
||||||
|
|
||||||
export const settingsStore = map<Settings>({
|
const initialProviderSettings: ProviderSetting = {};
|
||||||
shortcuts: shortcutsStore.get(),
|
PROVIDER_LIST.forEach((provider) => {
|
||||||
providers: providersList,
|
initialProviderSettings[provider.name] = {
|
||||||
|
...provider,
|
||||||
|
settings: {
|
||||||
|
enabled: false,
|
||||||
|
},
|
||||||
|
};
|
||||||
});
|
});
|
||||||
|
export const providersStore = map<ProviderSetting>(initialProviderSettings);
|
||||||
|
|
||||||
shortcutsStore.subscribe((shortcuts) => {
|
export const isDebugMode = atom(false);
|
||||||
settingsStore.set({
|
|
||||||
...settingsStore.get(),
|
export const isLocalModelsEnabled = atom(true);
|
||||||
shortcuts,
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|||||||
@@ -3,6 +3,7 @@ import { MAX_RESPONSE_SEGMENTS, MAX_TOKENS } from '~/lib/.server/llm/constants';
|
|||||||
import { CONTINUE_PROMPT } from '~/lib/.server/llm/prompts';
|
import { CONTINUE_PROMPT } from '~/lib/.server/llm/prompts';
|
||||||
import { streamText, type Messages, type StreamingOptions } from '~/lib/.server/llm/stream-text';
|
import { streamText, type Messages, type StreamingOptions } from '~/lib/.server/llm/stream-text';
|
||||||
import SwitchableStream from '~/lib/.server/llm/switchable-stream';
|
import SwitchableStream from '~/lib/.server/llm/switchable-stream';
|
||||||
|
import type { IProviderSetting } from '~/types/model';
|
||||||
|
|
||||||
export async function action(args: ActionFunctionArgs) {
|
export async function action(args: ActionFunctionArgs) {
|
||||||
return chatAction(args);
|
return chatAction(args);
|
||||||
@@ -38,6 +39,9 @@ async function chatAction({ context, request }: ActionFunctionArgs) {
|
|||||||
|
|
||||||
// Parse the cookie's value (returns an object or null if no cookie exists)
|
// Parse the cookie's value (returns an object or null if no cookie exists)
|
||||||
const apiKeys = JSON.parse(parseCookies(cookieHeader || '').apiKeys || '{}');
|
const apiKeys = JSON.parse(parseCookies(cookieHeader || '').apiKeys || '{}');
|
||||||
|
const providerSettings: Record<string, IProviderSetting> = JSON.parse(
|
||||||
|
parseCookies(cookieHeader || '').providers || '{}',
|
||||||
|
);
|
||||||
|
|
||||||
const stream = new SwitchableStream();
|
const stream = new SwitchableStream();
|
||||||
|
|
||||||
@@ -60,13 +64,13 @@ async function chatAction({ context, request }: ActionFunctionArgs) {
|
|||||||
messages.push({ role: 'assistant', content });
|
messages.push({ role: 'assistant', content });
|
||||||
messages.push({ role: 'user', content: CONTINUE_PROMPT });
|
messages.push({ role: 'user', content: CONTINUE_PROMPT });
|
||||||
|
|
||||||
const result = await streamText(messages, context.cloudflare.env, options, apiKeys);
|
const result = await streamText({ messages, env: context.cloudflare.env, options, apiKeys, providerSettings });
|
||||||
|
|
||||||
return stream.switchSource(result.toAIStream());
|
return stream.switchSource(result.toAIStream());
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
|
|
||||||
const result = await streamText(messages, context.cloudflare.env, options, apiKeys);
|
const result = await streamText({ messages, env: context.cloudflare.env, options, apiKeys, providerSettings });
|
||||||
|
|
||||||
stream.switchSource(result.toAIStream());
|
stream.switchSource(result.toAIStream());
|
||||||
|
|
||||||
|
|||||||
@@ -2,7 +2,7 @@ import { type ActionFunctionArgs } from '@remix-run/cloudflare';
|
|||||||
import { StreamingTextResponse, parseStreamPart } from 'ai';
|
import { StreamingTextResponse, parseStreamPart } from 'ai';
|
||||||
import { streamText } from '~/lib/.server/llm/stream-text';
|
import { streamText } from '~/lib/.server/llm/stream-text';
|
||||||
import { stripIndents } from '~/utils/stripIndent';
|
import { stripIndents } from '~/utils/stripIndent';
|
||||||
import type { ProviderInfo } from '~/types/model';
|
import type { IProviderSetting, ProviderInfo } from '~/types/model';
|
||||||
|
|
||||||
const encoder = new TextEncoder();
|
const encoder = new TextEncoder();
|
||||||
const decoder = new TextDecoder();
|
const decoder = new TextDecoder();
|
||||||
@@ -11,8 +11,28 @@ export async function action(args: ActionFunctionArgs) {
|
|||||||
return enhancerAction(args);
|
return enhancerAction(args);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function parseCookies(cookieHeader: string) {
|
||||||
|
const cookies: any = {};
|
||||||
|
|
||||||
|
// Split the cookie string by semicolons and spaces
|
||||||
|
const items = cookieHeader.split(';').map((cookie) => cookie.trim());
|
||||||
|
|
||||||
|
items.forEach((item) => {
|
||||||
|
const [name, ...rest] = item.split('=');
|
||||||
|
|
||||||
|
if (name && rest) {
|
||||||
|
// Decode the name and value, and join value parts in case it contains '='
|
||||||
|
const decodedName = decodeURIComponent(name.trim());
|
||||||
|
const decodedValue = decodeURIComponent(rest.join('=').trim());
|
||||||
|
cookies[decodedName] = decodedValue;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
return cookies;
|
||||||
|
}
|
||||||
|
|
||||||
async function enhancerAction({ context, request }: ActionFunctionArgs) {
|
async function enhancerAction({ context, request }: ActionFunctionArgs) {
|
||||||
const { message, model, provider, apiKeys } = await request.json<{
|
const { message, model, provider } = await request.json<{
|
||||||
message: string;
|
message: string;
|
||||||
model: string;
|
model: string;
|
||||||
provider: ProviderInfo;
|
provider: ProviderInfo;
|
||||||
@@ -36,9 +56,17 @@ async function enhancerAction({ context, request }: ActionFunctionArgs) {
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const cookieHeader = request.headers.get('Cookie');
|
||||||
|
|
||||||
|
// Parse the cookie's value (returns an object or null if no cookie exists)
|
||||||
|
const apiKeys = JSON.parse(parseCookies(cookieHeader || '').apiKeys || '{}');
|
||||||
|
const providerSettings: Record<string, IProviderSetting> = JSON.parse(
|
||||||
|
parseCookies(cookieHeader || '').providers || '{}',
|
||||||
|
);
|
||||||
|
|
||||||
try {
|
try {
|
||||||
const result = await streamText(
|
const result = await streamText({
|
||||||
[
|
messages: [
|
||||||
{
|
{
|
||||||
role: 'user',
|
role: 'user',
|
||||||
content:
|
content:
|
||||||
@@ -73,10 +101,10 @@ async function enhancerAction({ context, request }: ActionFunctionArgs) {
|
|||||||
`,
|
`,
|
||||||
},
|
},
|
||||||
],
|
],
|
||||||
context.cloudflare.env,
|
env: context.cloudflare.env,
|
||||||
undefined,
|
|
||||||
apiKeys,
|
apiKeys,
|
||||||
);
|
providerSettings,
|
||||||
|
});
|
||||||
|
|
||||||
const transformStream = new TransformStream({
|
const transformStream = new TransformStream({
|
||||||
transform(chunk, controller) {
|
transform(chunk, controller) {
|
||||||
|
|||||||
@@ -3,9 +3,17 @@ import type { ModelInfo } from '~/utils/types';
|
|||||||
export type ProviderInfo = {
|
export type ProviderInfo = {
|
||||||
staticModels: ModelInfo[];
|
staticModels: ModelInfo[];
|
||||||
name: string;
|
name: string;
|
||||||
getDynamicModels?: (apiKeys?: Record<string, string>) => Promise<ModelInfo[]>;
|
getDynamicModels?: (apiKeys?: Record<string, string>, providerSettings?: IProviderSetting) => Promise<ModelInfo[]>;
|
||||||
getApiKeyLink?: string;
|
getApiKeyLink?: string;
|
||||||
labelForGetApiKey?: string;
|
labelForGetApiKey?: string;
|
||||||
icon?: string;
|
icon?: string;
|
||||||
isEnabled?: boolean;
|
};
|
||||||
|
|
||||||
|
export interface IProviderSetting {
|
||||||
|
enabled?: boolean;
|
||||||
|
baseUrl?: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
export type IProviderConfig = ProviderInfo & {
|
||||||
|
settings: IProviderSetting;
|
||||||
};
|
};
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
import Cookies from 'js-cookie';
|
import Cookies from 'js-cookie';
|
||||||
import type { ModelInfo, OllamaApiResponse, OllamaModel } from './types';
|
import type { ModelInfo, OllamaApiResponse, OllamaModel } from './types';
|
||||||
import type { ProviderInfo } from '~/types/model';
|
import type { ProviderInfo, IProviderSetting } from '~/types/model';
|
||||||
import { createScopedLogger } from './logger';
|
import { createScopedLogger } from './logger';
|
||||||
|
|
||||||
export const WORK_DIR_NAME = 'project';
|
export const WORK_DIR_NAME = 'project';
|
||||||
@@ -298,13 +298,16 @@ const staticModels: ModelInfo[] = PROVIDER_LIST.map((p) => p.staticModels).flat(
|
|||||||
|
|
||||||
export let MODEL_LIST: ModelInfo[] = [...staticModels];
|
export let MODEL_LIST: ModelInfo[] = [...staticModels];
|
||||||
|
|
||||||
export async function getModelList(apiKeys: Record<string, string>) {
|
export async function getModelList(
|
||||||
|
apiKeys: Record<string, string>,
|
||||||
|
providerSettings?: Record<string, IProviderSetting>,
|
||||||
|
) {
|
||||||
MODEL_LIST = [
|
MODEL_LIST = [
|
||||||
...(
|
...(
|
||||||
await Promise.all(
|
await Promise.all(
|
||||||
PROVIDER_LIST.filter(
|
PROVIDER_LIST.filter(
|
||||||
(p): p is ProviderInfo & { getDynamicModels: () => Promise<ModelInfo[]> } => !!p.getDynamicModels,
|
(p): p is ProviderInfo & { getDynamicModels: () => Promise<ModelInfo[]> } => !!p.getDynamicModels,
|
||||||
).map((p) => p.getDynamicModels(apiKeys)),
|
).map((p) => p.getDynamicModels(apiKeys, providerSettings?.[p.name])),
|
||||||
)
|
)
|
||||||
).flat(),
|
).flat(),
|
||||||
...staticModels,
|
...staticModels,
|
||||||
@@ -312,9 +315,9 @@ export async function getModelList(apiKeys: Record<string, string>) {
|
|||||||
return MODEL_LIST;
|
return MODEL_LIST;
|
||||||
}
|
}
|
||||||
|
|
||||||
async function getTogetherModels(apiKeys?: Record<string, string>): Promise<ModelInfo[]> {
|
async function getTogetherModels(apiKeys?: Record<string, string>, settings?: IProviderSetting): Promise<ModelInfo[]> {
|
||||||
try {
|
try {
|
||||||
const baseUrl = import.meta.env.TOGETHER_API_BASE_URL || '';
|
const baseUrl = settings?.baseUrl || import.meta.env.TOGETHER_API_BASE_URL || '';
|
||||||
const provider = 'Together';
|
const provider = 'Together';
|
||||||
|
|
||||||
if (!baseUrl) {
|
if (!baseUrl) {
|
||||||
@@ -353,8 +356,8 @@ async function getTogetherModels(apiKeys?: Record<string, string>): Promise<Mode
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
const getOllamaBaseUrl = () => {
|
const getOllamaBaseUrl = (settings?: IProviderSetting) => {
|
||||||
const defaultBaseUrl = import.meta.env.OLLAMA_API_BASE_URL || 'http://localhost:11434';
|
const defaultBaseUrl = settings?.baseUrl || import.meta.env.OLLAMA_API_BASE_URL || 'http://localhost:11434';
|
||||||
|
|
||||||
// Check if we're in the browser
|
// Check if we're in the browser
|
||||||
if (typeof window !== 'undefined') {
|
if (typeof window !== 'undefined') {
|
||||||
@@ -368,7 +371,7 @@ const getOllamaBaseUrl = () => {
|
|||||||
return isDocker ? defaultBaseUrl.replace('localhost', 'host.docker.internal') : defaultBaseUrl;
|
return isDocker ? defaultBaseUrl.replace('localhost', 'host.docker.internal') : defaultBaseUrl;
|
||||||
};
|
};
|
||||||
|
|
||||||
async function getOllamaModels(): Promise<ModelInfo[]> {
|
async function getOllamaModels(apiKeys?: Record<string, string>, settings?: IProviderSetting): Promise<ModelInfo[]> {
|
||||||
/*
|
/*
|
||||||
* if (typeof window === 'undefined') {
|
* if (typeof window === 'undefined') {
|
||||||
* return [];
|
* return [];
|
||||||
@@ -376,7 +379,7 @@ async function getOllamaModels(): Promise<ModelInfo[]> {
|
|||||||
*/
|
*/
|
||||||
|
|
||||||
try {
|
try {
|
||||||
const baseUrl = getOllamaBaseUrl();
|
const baseUrl = getOllamaBaseUrl(settings);
|
||||||
const response = await fetch(`${baseUrl}/api/tags`);
|
const response = await fetch(`${baseUrl}/api/tags`);
|
||||||
const data = (await response.json()) as OllamaApiResponse;
|
const data = (await response.json()) as OllamaApiResponse;
|
||||||
|
|
||||||
@@ -392,20 +395,21 @@ async function getOllamaModels(): Promise<ModelInfo[]> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async function getOpenAILikeModels(): Promise<ModelInfo[]> {
|
async function getOpenAILikeModels(
|
||||||
|
apiKeys?: Record<string, string>,
|
||||||
|
settings?: IProviderSetting,
|
||||||
|
): Promise<ModelInfo[]> {
|
||||||
try {
|
try {
|
||||||
const baseUrl = import.meta.env.OPENAI_LIKE_API_BASE_URL || '';
|
const baseUrl = settings?.baseUrl || import.meta.env.OPENAI_LIKE_API_BASE_URL || '';
|
||||||
|
|
||||||
if (!baseUrl) {
|
if (!baseUrl) {
|
||||||
return [];
|
return [];
|
||||||
}
|
}
|
||||||
|
|
||||||
let apiKey = import.meta.env.OPENAI_LIKE_API_KEY ?? '';
|
let apiKey = '';
|
||||||
|
|
||||||
const apikeys = JSON.parse(Cookies.get('apiKeys') || '{}');
|
if (apiKeys && apiKeys.OpenAILike) {
|
||||||
|
apiKey = apiKeys.OpenAILike;
|
||||||
if (apikeys && apikeys.OpenAILike) {
|
|
||||||
apiKey = apikeys.OpenAILike;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
const response = await fetch(`${baseUrl}/models`, {
|
const response = await fetch(`${baseUrl}/models`, {
|
||||||
@@ -459,13 +463,13 @@ async function getOpenRouterModels(): Promise<ModelInfo[]> {
|
|||||||
}));
|
}));
|
||||||
}
|
}
|
||||||
|
|
||||||
async function getLMStudioModels(): Promise<ModelInfo[]> {
|
async function getLMStudioModels(_apiKeys?: Record<string, string>, settings?: IProviderSetting): Promise<ModelInfo[]> {
|
||||||
if (typeof window === 'undefined') {
|
if (typeof window === 'undefined') {
|
||||||
return [];
|
return [];
|
||||||
}
|
}
|
||||||
|
|
||||||
try {
|
try {
|
||||||
const baseUrl = import.meta.env.LMSTUDIO_API_BASE_URL || 'http://localhost:1234';
|
const baseUrl = settings?.baseUrl || import.meta.env.LMSTUDIO_API_BASE_URL || 'http://localhost:1234';
|
||||||
const response = await fetch(`${baseUrl}/v1/models`);
|
const response = await fetch(`${baseUrl}/v1/models`);
|
||||||
const data = (await response.json()) as any;
|
const data = (await response.json()) as any;
|
||||||
|
|
||||||
@@ -480,7 +484,7 @@ async function getLMStudioModels(): Promise<ModelInfo[]> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async function initializeModelList(): Promise<ModelInfo[]> {
|
async function initializeModelList(providerSettings?: Record<string, IProviderSetting>): Promise<ModelInfo[]> {
|
||||||
let apiKeys: Record<string, string> = {};
|
let apiKeys: Record<string, string> = {};
|
||||||
|
|
||||||
try {
|
try {
|
||||||
@@ -501,7 +505,7 @@ async function initializeModelList(): Promise<ModelInfo[]> {
|
|||||||
await Promise.all(
|
await Promise.all(
|
||||||
PROVIDER_LIST.filter(
|
PROVIDER_LIST.filter(
|
||||||
(p): p is ProviderInfo & { getDynamicModels: () => Promise<ModelInfo[]> } => !!p.getDynamicModels,
|
(p): p is ProviderInfo & { getDynamicModels: () => Promise<ModelInfo[]> } => !!p.getDynamicModels,
|
||||||
).map((p) => p.getDynamicModels(apiKeys)),
|
).map((p) => p.getDynamicModels(apiKeys, providerSettings?.[p.name])),
|
||||||
)
|
)
|
||||||
).flat(),
|
).flat(),
|
||||||
...staticModels,
|
...staticModels,
|
||||||
|
|||||||
@@ -26,12 +26,3 @@ export interface ModelInfo {
|
|||||||
provider: string;
|
provider: string;
|
||||||
maxTokenAllowed: number;
|
maxTokenAllowed: number;
|
||||||
}
|
}
|
||||||
|
|
||||||
export interface ProviderInfo {
|
|
||||||
staticModels: ModelInfo[];
|
|
||||||
name: string;
|
|
||||||
getDynamicModels?: () => Promise<ModelInfo[]>;
|
|
||||||
getApiKeyLink?: string;
|
|
||||||
labelForGetApiKey?: string;
|
|
||||||
icon?: string;
|
|
||||||
}
|
|
||||||
|
|||||||
Reference in New Issue
Block a user