python312Packages.langchain-chroma: fix broken tests (#345803)

This commit is contained in:
Fabian Affolter 2024-10-02 08:36:00 +02:00 committed by GitHub
commit c163396967
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 24 additions and 2 deletions

View File

@ -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"]

View File

@ -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
];
};
}