From 2bf8c7c4d356ede81603f470e8bbacc6c70d08a1 Mon Sep 17 00:00:00 2001
From: hailin <hailin@gdzx.xyz>
Date: Sun, 11 May 2025 13:59:49 +0800
Subject: [PATCH] .

---
 app/core/embedding.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/app/core/embedding.py b/app/core/embedding.py
index e3aa4ca..2a852b2 100644
--- a/app/core/embedding.py
+++ b/app/core/embedding.py
@@ -27,5 +27,9 @@ class BGEEmbedding:
         input_mask_expanded = attention_mask.unsqueeze(-1).expand(token_embeddings.size()).float()
         return torch.sum(token_embeddings * input_mask_expanded, 1) / torch.clamp(input_mask_expanded.sum(1), min=1e-9)
 
+    def get_agg_embedding_from_queries(self, queries):
+        embeddings = self.encode(queries)
+        return np.mean(embeddings, axis=0)  # 聚合多个嵌入
+    
 embedder = BGEEmbedding()