From 0032c4949b950d5ce2b42f09221c2b2ae3ff72a3 Mon Sep 17 00:00:00 2001
From: Tienson Qin <tiensonqin@gmail.com>
Date: Tue, 12 Aug 2025 12:52:16 +0800
Subject: [PATCH] fix: slow switch to large graph with local embedding enabled

---
 src/main/frontend/handler.cljs                   | 16 ++++++++--------
 .../db_based/vector_search_background_tasks.cljs |  2 ++
 2 files changed, 10 insertions(+), 8 deletions(-)

diff --git a/src/main/frontend/handler.cljs b/src/main/frontend/handler.cljs
index bb2103ed05..e9e21c1721 100644
--- a/src/main/frontend/handler.cljs
+++ b/src/main/frontend/handler.cljs
@@ -168,13 +168,7 @@
                repo (or (state/get-current-repo) (:url (first repos)))
                _ (if (empty? repos)
                    (repo-handler/new-db! config/demo-repo)
-                   (restore-and-setup! repo))
-               webgpu-available? (db-browser/<check-webgpu-available?)]
-         (log/info :webgpu-available? webgpu-available?)
-         (when webgpu-available?
-           (p/do! (db-browser/start-inference-worker!)
-                  (db-browser/<connect-db-worker-and-infer-worker!)
-                  (reset! vector-search-flows/*infer-worker-ready true)))
+                   (restore-and-setup! repo))]
          (set-network-watcher!)
 
          (when (util/electron?)
@@ -184,7 +178,13 @@
        (p/catch (fn [e]
                   (js/console.error "Error while restoring repos: " e)))
        (p/finally (fn []
-                    (state/set-db-restoring! false))))
+                    (state/set-db-restoring! false)
+                    (p/let [webgpu-available? (db-browser/<check-webgpu-available?)]
+                      (log/info :webgpu-available? webgpu-available?)
+                      (when webgpu-available?
+                        (p/do! (db-browser/start-inference-worker!)
+                               (db-browser/<connect-db-worker-and-infer-worker!)
+                               (reset! vector-search-flows/*infer-worker-ready true)))))))
 
    (util/<app-wake-up-from-sleep-loop (atom false))
 
diff --git a/src/main/frontend/handler/db_based/vector_search_background_tasks.cljs b/src/main/frontend/handler/db_based/vector_search_background_tasks.cljs
index 684d625ba1..a2622974c5 100644
--- a/src/main/frontend/handler/db_based/vector_search_background_tasks.cljs
+++ b/src/main/frontend/handler/db_based/vector_search_background_tasks.cljs
@@ -21,6 +21,8 @@
     (m/?< vector-search-flows/infer-worker-ready-flow)
     (when-let [repo (m/?< flows/current-repo-flow)]
       (try
+        ;; Don't block ui render (ui needs data from db-worker)
+        (m/? (m/sleep 1000))
         (c.m/<? (state/<invoke-db-worker :thread-api/vec-search-init-embedding-model repo))
         (m/?< (c.m/clock (* 30 1000)))
         (c.m/<? (state/<invoke-db-worker :thread-api/vec-search-embedding-graph repo))