From bfed2e3bc6cbaa85f1aef81907263ac4db2404a7 Mon Sep 17 00:00:00 2001
From: Tienson Qin <tiensonqin@gmail.com>
Date: Tue, 15 Jul 2025 17:49:01 +0800
Subject: [PATCH] Add qwen3 embedding

---
 src/main/frontend/inference_worker/text_embedding.cljs | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/main/frontend/inference_worker/text_embedding.cljs b/src/main/frontend/inference_worker/text_embedding.cljs
index 172efdf4cd..ab33804646 100644
--- a/src/main/frontend/inference_worker/text_embedding.cljs
+++ b/src/main/frontend/inference_worker/text_embedding.cljs
@@ -97,12 +97,12 @@
     (when (model-loaded?)
       (let [hnsw (or (get-hnsw-index repo) (new-hnsw-index! repo))
             {:keys [data _type dims _size]} (worker-util/profile :<text-embedding
-                                              (c.m/<? (<text-embedding text-array)))
+                                                                 (c.m/<? (<text-embedding text-array)))
             data-coll (split-into-chunks data (last dims))
             _ (assert (= (count text-array) (count data-coll)))]
         (when (seq delete-labels) (.markDeleteItems hnsw (into-array delete-labels)))
         (worker-util/profile (keyword "add-items" (str (alength data-coll)))
-          (add-items hnsw data-coll replace-deleted?))))))
+                             (add-items hnsw data-coll replace-deleted?))))))
 
 (def ^:private write-index-wait-delays-flow
   (m/ap
@@ -172,7 +172,9 @@
   {"Xenova/all-MiniLM-L6-v2" {:tf-config {:dtype "fp32"}
                               :hnsw-config {:dims 384}}
    "Xenova/jina-embeddings-v2-base-zh" {:tf-config {:dtype "fp32"}
-                                        :hnsw-config {:dims 768}}})
+                                        :hnsw-config {:dims 768}}
+   "onnx-community/Qwen3-Embedding-0.6B-ONNX" {:tf-config {:dtype "fp32"}
+                                               :hnsw-config {:dims 384}}})
 
 (def ^:private *load-model-progress (atom nil))