fix bug

tangwang
1 parent 70dab99f
Showing 1 changed file with 13 additions and 13 deletions Show diff stats
embeddings/text_encoder.py
@@ -113,7 +113,19 @@ class BgeEncoder:
         uncached_indices: List[int] = []
         uncached_texts: List[str] = []
-        # Prepare request data for uncached texts
+        # Process response
+        # Each element can be np.ndarray or None (表示该文本没有可用的向量）
+        embeddings: List[Optional[np.ndarray]] = [None] * len(sentences)
+
+        for i, text in enumerate(sentences):
+            cached = self._get_cached_embedding(text, 'en')  # Use 'en' as default language for title embedding
+            if cached is not None:
+                embeddings[i] = cached
+            else:
+                uncached_indices.append(i)
+                uncached_texts.append(text)
+        
+        # Prepare request data for uncached texts (after cache check)
         request_data = []
         for i, text in enumerate(uncached_texts):
             request_item = {
@@ -127,18 +139,6 @@ class BgeEncoder:
             request_item["name_ru"] = None
             request_data.append(request_item)
-
-        # Process response
-        # Each element can be np.ndarray or None (表示该文本没有可用的向量）
-        embeddings: List[Optional[np.ndarray]] = [None] * len(sentences)
-
-        for i, text in enumerate(sentences):
-            cached = self._get_cached_embedding(text, 'en')  # Use 'en' as default language for title embedding
-            if cached is not None:
-                embeddings[i] = cached
-            else:
-                uncached_indices.append(i)
-                uncached_texts.append(text)
         # If there are uncached texts, call service
         if uncached_texts: