@@ -242,7 +242,7 @@ export async function handleMarketplaceMessages(
242242 try {
243243 await marketplaceManager
244244 . installMarketplaceItem ( message . mpItem , message . mpInstallOptions )
245- . then ( async ( r ) => r === "$COMMIT" && ( await provider . postStateToWebview ( ) ) )
245+ . then ( async ( r ) => r === "$COMMIT" && ( await _onCommit ( ) ) )
246246 } catch ( error ) {
247247 vscode . window . showErrorMessage (
248248 `Failed to install item "${ message . mpItem . name } ":\n${ error instanceof Error ? error . message : String ( error ) } ` ,
@@ -263,7 +263,7 @@ export async function handleMarketplaceMessages(
263263 try {
264264 await marketplaceManager
265265 . installMarketplaceItem ( item , { parameters } )
266- . then ( async ( r ) => r === "$COMMIT" && ( await provider . postStateToWebview ( ) ) )
266+ . then ( async ( r ) => r === "$COMMIT" && ( await _onCommit ( ) ) )
267267 } catch ( error ) {
268268 console . error ( `Error submitting marketplace parameters: ${ error } ` )
269269 vscode . window . showErrorMessage (
@@ -287,7 +287,7 @@ export async function handleMarketplaceMessages(
287287 try {
288288 await marketplaceManager
289289 . removeInstalledMarketplaceItem ( message . mpItem , message . mpInstallOptions )
290- . then ( async ( r ) => r === "$COMMIT" && ( await provider . postStateToWebview ( ) ) )
290+ . then ( async ( r ) => r === "$COMMIT" && ( await _onCommit ( ) ) )
291291 } catch ( error ) {
292292 vscode . window . showErrorMessage (
293293 `Failed to remove item "${ message . mpItem . name } ":\n${ error instanceof Error ? error . message : String ( error ) } ` ,
@@ -302,4 +302,12 @@ export async function handleMarketplaceMessages(
302302 default :
303303 return false
304304 }
305+
306+ async function _onCommit ( ) {
307+ await Promise . all ( [
308+ provider . getMcpHub ( ) ?. reloadMcpServers ?.( ) ,
309+ provider . customModesManager ?. refreshMergedState ?.( ) ,
310+ provider . postStateToWebview ( ) ,
311+ ] )
312+ }
305313}
0 commit comments