python312Packages.langchain-chroma: fix broken tests (#345803)
This commit is contained in:
commit
c163396967
@ -0,0 +1,13 @@
|
||||
diff --git a/libs/partners/chroma/tests/integration_tests/test_vectorstores.py b/libs/partners/chroma/tests/integration_tests/test_vectorstores.py
|
||||
index 382b24cb5..f99a34112 100644
|
||||
--- a/tests/integration_tests/test_vectorstores.py
|
||||
+++ b/tests/integration_tests/test_vectorstores.py
|
||||
@@ -36,7 +36,7 @@ def test_chroma() -> None:
|
||||
|
||||
assert output == [Document(page_content="foo")]
|
||||
|
||||
-
|
||||
+@pytest.mark.asyncio
|
||||
async def test_chroma_async() -> None:
|
||||
"""Test end to end construction and search."""
|
||||
texts = ["foo", "bar", "baz"]
|
@ -7,6 +7,7 @@
|
||||
numpy,
|
||||
poetry-core,
|
||||
pytestCheckHook,
|
||||
pytest-asyncio,
|
||||
}:
|
||||
|
||||
buildPythonPackage rec {
|
||||
@ -23,6 +24,8 @@ buildPythonPackage rec {
|
||||
|
||||
sourceRoot = "${src.name}/libs/partners/chroma";
|
||||
|
||||
patches = [ ./001-async-test.patch ];
|
||||
|
||||
build-system = [ poetry-core ];
|
||||
|
||||
pythonRelaxDeps = [ "chromadb" ];
|
||||
@ -35,7 +38,10 @@ buildPythonPackage rec {
|
||||
|
||||
pythonImportsCheck = [ "langchain_chroma" ];
|
||||
|
||||
nativeCheckInputs = [ pytestCheckHook ];
|
||||
nativeCheckInputs = [
|
||||
pytest-asyncio
|
||||
pytestCheckHook
|
||||
];
|
||||
|
||||
passthru = {
|
||||
inherit (langchain-core) updateScript;
|
||||
@ -46,6 +52,9 @@ buildPythonPackage rec {
|
||||
description = "Integration package connecting Chroma and LangChain";
|
||||
homepage = "https://github.com/langchain-ai/langchain/tree/master/libs/partners/chroma";
|
||||
license = lib.licenses.mit;
|
||||
maintainers = with lib.maintainers; [ natsukium ];
|
||||
maintainers = with lib.maintainers; [
|
||||
natsukium
|
||||
sarahec
|
||||
];
|
||||
};
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user