awesome-bookmarks/goose
3
0
Fork 0
mirror of https://github.com/block/goose.git synced 2026-07-17 12:56:20 +02:00
Code Issues Packages Projects Releases Wiki Activity

speed up loading extensions by loading in parallel (#4054)

Browse Source 
thx verified locally should be gtg 👍
This commit is contained in:
Zane
2025-08-13 10:30:12 -07:00
committed by GitHub
parent 6583480dc8
commit a7f158ee11
2 changed files with 14 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ui/desktop/src/extensions.tsx
-55
View File
@@ -213,61 +213,6 @@ export async function removeExtension(name: string, silent: boolean = false): Pr
}
}
// Store extension config in user_settings
function storeExtensionConfig(config: FullExtensionConfig) {
try {
const userSettingsStr = localStorage.getItem('user_settings');
const userSettings = userSettingsStr
? JSON.parse(userSettingsStr)
: { models: [], extensions: [] };
// Check if config already exists (based on cmd for stdio, uri for sse, name for builtin)
const extensionExists = userSettings.extensions.some(
(extension: { id: string }) => extension.id === config.id
);
if (!extensionExists) {
userSettings.extensions.push(config);
localStorage.setItem('user_settings', JSON.stringify(userSettings));
console.log('Extension config stored successfully in user_settings');
// Notify settings update through electron IPC
window.electron.emit('settings-updated');
} else {
console.log('Extension config already exists in user_settings');
}
} catch (error) {
console.error('Error storing extension config:', error);
}
}
export async function loadAndAddStoredExtensions() {
try {
const userSettingsStr = localStorage.getItem('user_settings');
if (userSettingsStr) {
const userSettings = JSON.parse(userSettingsStr);
const enabledExtensions = userSettings.extensions.filter(
(ext: FullExtensionConfig) => ext.enabled
);
console.log('Adding extensions from localStorage: ', enabledExtensions);
for (const ext of enabledExtensions) {
await addExtension(ext, true);
}
} else {
console.log('Saving default builtin extensions to localStorage');
// TODO - Revisit
BUILT_IN_EXTENSIONS.forEach(async (extension: FullExtensionConfig) => {
storeExtensionConfig(extension);
if (extension.enabled) {
await addExtension(extension, true);
}
});
}
} catch (error) {
console.error('Error loading and activating extensions from localStorage: ', error);
}
}
// Update the path to the binary based on the command
export async function replaceWithShims(cmd: string) {
const binaryPathMap: Record<string, string> = {
ui/desktop/src/utils/providerUtils.ts
+14 -6
View File
@@ -291,13 +291,21 @@ export const initializeSystem = async (
await syncBundledExtensions(refreshedExtensions, options.addExtension);
}
// Add enabled extensions to agent
for (const extensionEntry of refreshedExtensions) {
if (extensionEntry.enabled) {
const extensionConfig = extractExtensionConfig(extensionEntry);
await addToAgentOnStartup({ addToConfig: options.addExtension, extensionConfig });
// Add enabled extensions to agent in parallel
const enabledExtensions = refreshedExtensions.filter((ext) => ext.enabled);
const extensionLoadingPromises = enabledExtensions.map(async (extensionEntry) => {
const extensionConfig = extractExtensionConfig(extensionEntry);
const extensionName = extensionConfig.name;
try {
await addToAgentOnStartup({ addToConfig: options.addExtension!, extensionConfig });
} catch (error) {
console.error(`Failed to load extension ${extensionName}:`, error);
}
}
});
await Promise.allSettled(extensionLoadingPromises);
} catch (error) {
console.error('Failed to initialize agent:', error);
throw error;