community[patch]: support numpy2 (#28184)

Follows on from #27991, updates the langchain-community package to support numpy 2 versions --------- Co-authored-by: Chester Curme <[email protected]>
langchain-ai · Nov 27, 2024 · 5141f25 · 5141f25
1 parent 0901f11
commit 5141f25
Show file tree

Hide file tree

Showing 6 changed files with 69 additions and 8 deletions.
diff --git a/.github/scripts/check_diff.py b/.github/scripts/check_diff.py
@@ -137,9 +137,7 @@ def _get_configs_for_single_dir(job: str, dir_: str) -> List[Dict[str, str]]:
         py_versions = ["3.9", "3.13"]
 
     elif dir_ == "libs/community" and job == "extended-tests":
-        # community extended test resolution in 3.12 is slow
-        # even in uv
-        py_versions = ["3.9", "3.11"]
+        py_versions = ["3.9", "3.12"]
 
     elif dir_ == "libs/community" and job == "compile-integration-tests":
         # community integration deps are slow in 3.12

diff --git a/libs/community/extended_testing_deps.txt b/libs/community/extended_testing_deps.txt
@@ -55,7 +55,6 @@ openai<2
 openapi-pydantic>=0.3.2,<0.4
 oracle-ads>=2.9.1,<3
 oracledb>=2.2.0,<3
-outlines[test]>=0.1.0,<0.2
 pandas>=2.0.1,<3
 pdfminer-six>=20221105,<20240706
 pgvector>=0.1.6,<0.2

diff --git a/libs/community/langchain_community/vectorstores/oraclevs.py b/libs/community/langchain_community/vectorstores/oraclevs.py
@@ -762,7 +762,7 @@ def similarity_search_by_vector_returning_embeddings(
         k: int,
         filter: Optional[Dict[str, Any]] = None,
         **kwargs: Any,
-    ) -> List[Tuple[Document, float, np.ndarray[np.float32, Any]]]:
+    ) -> List[Tuple[Document, float, np.ndarray]]:
         embedding_arr: Any
         if self.insert_mode == "clob":
             embedding_arr = json.dumps(embedding)

diff --git a/libs/community/langchain_community/vectorstores/supabase.py b/libs/community/langchain_community/vectorstores/supabase.py
@@ -270,7 +270,7 @@ def similarity_search_by_vector_returning_embeddings(
         k: int,
         filter: Optional[Dict[str, Any]] = None,
         postgrest_filter: Optional[str] = None,
-    ) -> List[Tuple[Document, float, np.ndarray[np.float32, Any]]]:
+    ) -> List[Tuple[Document, float, np.ndarray]]:
         match_documents_params = self.match_args(query, filter)
         query_builder = self._client.rpc(self.query_name, match_documents_params)
 

diff --git a/libs/community/poetry.lock b/libs/community/poetry.lock
diff --git a/libs/community/pyproject.toml b/libs/community/pyproject.toml
@@ -49,7 +49,7 @@ version = ">=1.22.4,<2"
 python = "<3.12"
 
 [[tool.poetry.dependencies.numpy]]
-version = ">=1.26.2,<2"
+version = ">=1.26.2,<3"
 python = ">=3.12"
 
 [tool.ruff.lint]