This commit is contained in:
Timothy Jaeryang Baek 2025-10-02 13:26:41 -05:00
parent d9fdbb627f
commit abe70d1793

View file

@ -56,23 +56,16 @@
let version;
onMount(async () => {
if ($user === undefined || $user === null) {
await goto('/auth');
return;
}
if (!['user', 'admin'].includes($user?.role)) {
return;
}
const clearChatInputStorage = () => {
const chatInputKeys = Object.keys(localStorage).filter((key) => key.startsWith('chat-input'));
if (chatInputKeys.length > 0) {
chatInputKeys.forEach((key) => {
localStorage.removeItem(key);
});
}
};
const loadChatsFromDB = async () => {
const checkLocalDBChats = async () => {
try {
// Check if IndexedDB exists
DB = await openDB('Chats', 1);
@ -92,14 +85,18 @@
}
};
const loadUserSettings = async (): Promise<Parameters<(typeof settings)['set']>[0]> => {
const setUserSettings = async (cb) => {
const userSettings = await getUserSettings(localStorage.token).catch((error) => {
console.error(error);
return null;
});
if (userSettings) {
return userSettings.ui;
await settings.set(userSettings.ui);
if (cb) {
await cb();
}
}
try {
@ -110,18 +107,16 @@
}
};
const loadModels = async () => {
const setModels = async () => {
models.set(
await getModels(
localStorage.token,
$config?.features?.enable_direct_connections
? ($settings?.directConnections ?? null)
: null
$config?.features?.enable_direct_connections ? ($settings?.directConnections ?? null) : null
)
);
};
const loadToolServers = async () => {
const setToolServers = async () => {
let toolServersData = await getToolServersData($settings?.toolServers ?? []);
toolServersData = toolServersData.filter((data) => {
if (!data || data.error) {
@ -137,25 +132,32 @@
toolServers.set(toolServersData);
};
const loadBanners = async () => {
const setBanners = async () => {
const bannersData = await getBanners(localStorage.token);
banners.set(bannersData);
};
const loadTools = async () => {
const setTools = async () => {
const toolsData = await getTools(localStorage.token);
tools.set(toolsData);
};
// Parallel loading
onMount(async () => {
if ($user === undefined || $user === null) {
await goto('/auth');
return;
}
if (!['user', 'admin'].includes($user?.role)) {
return;
}
clearChatInputStorage();
await Promise.all([
loadChatsFromDB(),
loadBanners(),
loadTools(),
loadUserSettings().then((loadedSettings) => {
settings.set(loadedSettings);
// The following functions are dependent on the settings
return Promise.all([loadModels(), loadToolServers()]);
checkLocalDBChats(),
setBanners(),
setTools(),
setUserSettings(async () => {
await Promise.all([setModels(), setToolServers()]);
})
]);