Fix on embeddings list

aurelio-labs · Sep 20, 2024 · 2764adf · 2764adf
1 parent 79f50b4
commit 2764adf
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/semantic_router/encoders/huggingface.py b/semantic_router/encoders/huggingface.py
@@ -220,7 +220,10 @@ def __call__(self, docs: List[str]) -> List[List[float]]:
                 outputs = self.query({"inputs": batch, "parameters": {}})
                 if not outputs or len(outputs) == 0:
                     raise ValueError("No embeddings returned from the query.")
-                embeddings = embeddings + outputs
+                if isinstance(outputs[0], list):
+                    embeddings.extend(outputs)
+                else:
+                    embeddings.append(outputs)
             except Exception as e:
                 raise ValueError(f"No embeddings returned for batch. Error: {e}") from e