aws[patch]: Remove numpy dependency (previously patch to support nump…

…y2) (#282) Increasing version of numpy to include 2.x.x for the langchain-aws package. Fixes #147 --------- Co-authored-by: Piyush Jain <[email protected]>
langchain-ai · Nov 22, 2024 · 1bf6ba5 · 1bf6ba5
1 parent cf4d4be
commit 1bf6ba5
Show file tree

Hide file tree

Showing 4 changed files with 332 additions and 280 deletions.
diff --git a/libs/aws/langchain_aws/utilities/math.py b/libs/aws/langchain_aws/utilities/math.py
@@ -67,7 +67,7 @@ def cosine_similarity_top_k(
     score_threshold = score_threshold or -1.0
     score_array[score_array < score_threshold] = 0
     top_k = min(top_k or len(score_array), np.count_nonzero(score_array))
-    top_k_idxs = np.argpartition(score_array, -top_k, axis=None)[-top_k:]
+    top_k_idxs = np.argpartition(score_array, -top_k, axis=None)[-top_k:]  # type: ignore
     top_k_idxs = top_k_idxs[np.argsort(score_array.ravel()[top_k_idxs])][::-1]
     ret_idxs = np.unravel_index(top_k_idxs, score_array.shape)
     scores = score_array.ravel()[top_k_idxs].tolist()

diff --git a/libs/aws/langchain_aws/vectorstores/inmemorydb/schema.py b/libs/aws/langchain_aws/vectorstores/inmemorydb/schema.py
@@ -96,7 +96,7 @@ class InMemoryDBVectorField(InMemoryDBField):
     dims: int = Field(...)
     algorithm: object = Field(...)
     datatype: str = Field(default="FLOAT32")
-    distance_metric: InMemoryDBDistanceMetric = Field(default="COSINE")
+    distance_metric: InMemoryDBDistanceMetric = Field(default="COSINE")  # type: ignore
     initial_cap: Optional[int] = None
 
     @field_validator("algorithm", "datatype", "distance_metric", mode="before")