@@ -12,6 +12,7 @@ export const ollamaService = new OllamaService()
1212export const registerLLMSessionHandlers = ( store : Store < StoreSchema > ) => {
1313 ipcMain . handle ( 'set-default-llm' , ( event , modelName : string ) => {
1414 store . set ( StoreKeys . DefaultLLM , modelName )
15+ event . sender . send ( 'llm-configs-changed' )
1516 } )
1617
1718 ipcMain . handle ( 'get-default-llm-name' , ( ) => store . get ( StoreKeys . DefaultLLM ) )
@@ -22,24 +23,29 @@ export const registerLLMSessionHandlers = (store: Store<StoreSchema>) => {
2223
2324 ipcMain . handle ( 'add-or-update-llm-config' , async ( event , llmConfig : LLMConfig ) => {
2425 await addOrUpdateLLMInStore ( store , llmConfig )
26+ event . sender . send ( 'llm-configs-changed' )
2527 } )
2628
2729 ipcMain . handle ( 'add-or-update-llm-api-config' , async ( event , llmAPIConfig : LLMAPIConfig ) => {
2830 await addOrUpdateLLMAPIInStore ( store , llmAPIConfig )
31+ event . sender . send ( 'llm-configs-changed' )
2932 } )
3033
3134 ipcMain . handle ( 'remove-llm' , async ( event , modelNameToDelete : string ) => {
3235 await removeLLM ( store , ollamaService , modelNameToDelete )
36+ event . sender . send ( 'llm-configs-changed' )
3337 } )
3438
3539 ipcMain . handle ( 'pull-ollama-model' , async ( event , modelName : string ) => {
3640 const handleProgress = ( progress : ProgressResponse ) => {
3741 event . sender . send ( 'ollamaDownloadProgress' , modelName , progress )
3842 }
3943 await ollamaService . pullModel ( modelName , handleProgress )
44+ event . sender . send ( 'llm-configs-changed' )
4045 } )
4146
4247 ipcMain . handle ( 'delete-llm' , async ( event , modelName : string ) => {
4348 await ollamaService . deleteModel ( modelName )
49+ event . sender . send ( 'llm-configs-changed' )
4450 } )
4551}
0 commit comments