Unverified Commit c1633969 authored by Fabian Affolter's avatar Fabian Affolter Committed by GitHub
Browse files

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

parents 58675fa6 6101b565
Loading
Loading
Loading
Loading
+13 −0
Original line number Diff line number Diff line
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"]
+11 −2
Original line number Diff line number Diff line
@@ -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
    ];
  };
}