diff --git a/web-plugin/index.js b/web-plugin/index.js index 1b745db..8aafa2c 100644 --- a/web-plugin/index.js +++ b/web-plugin/index.js @@ -1999,6 +1999,25 @@ api.fetchApi = async (route, options) => { } } + if (route.startsWith("/object_info") && ext.native_mode) { + const info = await getSelectedWorkflowInfo(); + console.log("info, ", info); + if (info) { + const data = { + refresh_private: true, + refresh_public: true + } + await fetch("/comfyui-deploy/refresh-models", { + method: "POST", + headers: { + Authorization: `Bearer ${info.cd_token}`, + "Content-Type": "application/json", + }, + body: JSON.stringify(data), + }); + } + } + return await orginal_fetch_api.call(api, route, options); }; @@ -2157,4 +2176,4 @@ function createMenuRightButton(config) { } button.setAttribute('data-pd-tooltip', config.tooltip) return button -} \ No newline at end of file +}