Merge branch 'main' into fix/get-embedding-for-near-vector

langchain-ai · Mar 25, 2024 · c8bbf02 · c8bbf02
2 parents d0552ac + e2984f4
commit c8bbf02
Show file tree

Hide file tree

Showing 4 changed files with 144 additions and 66 deletions.
diff --git a/langchain_weaviate/_math.py b/langchain_weaviate/_math.py
@@ -4,7 +4,7 @@
 from typing import List, Optional, Tuple, Union
 
 import numpy as np
-import simsimd as simd
+import simsimd
 
 logger = logging.getLogger(__name__)
 
@@ -26,7 +26,7 @@ def cosine_similarity(X: Matrix, Y: Matrix) -> np.ndarray:
 
     X = np.array(X, dtype=np.float32)
     Y = np.array(Y, dtype=np.float32)
-    Z = 1 - simd.cdist(X, Y, metric="cosine")
+    Z = 1 - np.array(simsimd.cdist(X, Y, metric="cosine"))
     if isinstance(Z, float):
         return np.array([Z])
     return Z