diff options
Diffstat (limited to 'pkgs/development/python-modules/chromadb/default.nix')
-rw-r--r-- | pkgs/development/python-modules/chromadb/default.nix | 18 |
1 files changed, 12 insertions, 6 deletions
diff --git a/pkgs/development/python-modules/chromadb/default.nix b/pkgs/development/python-modules/chromadb/default.nix index 867a380d7897c..0570f5c4df9b9 100644 --- a/pkgs/development/python-modules/chromadb/default.nix +++ b/pkgs/development/python-modules/chromadb/default.nix @@ -33,7 +33,6 @@ pytest-asyncio, pytestCheckHook, pythonOlder, - pythonRelaxDepsHook, pyyaml, requests, rustc, @@ -47,11 +46,12 @@ typing-extensions, uvicorn, zstd, + nixosTests, }: buildPythonPackage rec { pname = "chromadb"; - version = "0.5.3"; + version = "0.5.4"; pyproject = true; disabled = pythonOlder "3.9"; @@ -60,22 +60,24 @@ buildPythonPackage rec { owner = "chroma-core"; repo = "chroma"; rev = "refs/tags/${version}"; - hash = "sha256-czDL2b+Jj7mrYZCTfnaZArkOHBaWyTV0BTE2wvykHps="; + hash = "sha256-wzfzuWuNqLAjfAZC38p1iTtJHez/pJ9Ncgeo23o1dMo="; }; cargoDeps = rustPlatform.fetchCargoTarball { inherit src; name = "${pname}-${version}"; - hash = "sha256-eTVT1yowuDsajjceWojdUdX466FKneUt1i5QipBFdp4="; + hash = "sha256-0OE2i29oE6RpJRswQWI8+5dbA6lOWd3nhqe1RGlnjhk="; }; - pythonRelaxDeps = [ "orjson" ]; + pythonRelaxDeps = [ + "chroma-hnswlib" + "orjson" + ]; nativeBuildInputs = [ cargo pkg-config protobuf - pythonRelaxDepsHook rustc rustPlatform.cargoSetupHook setuptools @@ -156,6 +158,10 @@ buildPythonPackage rec { __darwinAllowLocalNetworking = true; + passthru.tests = { + inherit (nixosTests) chromadb; + }; + meta = with lib; { description = "AI-native open-source embedding database"; homepage = "https://github.com/chroma-core/chroma"; |