Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
6.1 kB · 277 lines
Nix
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278{ lib, stdenv, buildPythonPackage, fetchFromGitHub, fetchurl,
# build inputs cargo, pkg-config, protobuf, rustc, rustPlatform, pkgs, # zstd hidden by python3Packages.zstd openssl,
# dependencies bcrypt, build, fastapi, grpcio, httpx, importlib-resources, jsonschema, kubernetes, mmh3, numpy, onnxruntime, opentelemetry-api, opentelemetry-exporter-otlp-proto-grpc, opentelemetry-instrumentation-fastapi, opentelemetry-sdk, orjson, overrides, posthog, pydantic, pypika, pyyaml, requests, tenacity, tokenizers, tqdm, typer, typing-extensions, uvicorn,
# optional dependencies chroma-hnswlib,
# tests hnswlib, hypothesis, pandas, psutil, pytest-asyncio, pytest-xdist, pytestCheckHook, sqlite, starlette, writableTmpDirAsHomeHook,
# passthru nixosTests, nix-update-script,}:
buildPythonPackage rec { pname = "chromadb"; version = "1.0.12"; pyproject = true;
src = fetchFromGitHub { owner = "chroma-core"; repo = "chroma"; tag = version; hash = "sha256-Q4PhJTRNzJeVx6DIPWirnI9KksNb8vfOtqb/q9tSK3c="; };
cargoDeps = rustPlatform.fetchCargoVendor { inherit src; name = "${pname}-${version}-vendor"; hash = "sha256-+Ea2aRrsBGfVCLdOF41jeMehJhMurc8d0UKrpR6ndag="; };
# Can't use fetchFromGitHub as the build expects a zipfile swagger-ui = fetchurl { url = "https://github.com/swagger-api/swagger-ui/archive/refs/tags/v5.22.0.zip"; hash = "sha256-H+kXxA/6rKzYA19v7Zlx2HbIg/DGicD5FDIs0noVGSk="; };
patches = [ # The fastapi servers can't set up their networking in the test environment, so disable for testing ./disable-fastapi-fixtures.patch ];
postPatch = '' # Nixpkgs is taking the version from `chromadb_rust_bindings` which is versioned independently substituteInPlace pyproject.toml \ --replace-fail "dynamic = [\"version\"]" "version = \"${version}\"" '';
pythonRelaxDeps = [ "fastapi" ];
build-system = [ rustPlatform.maturinBuildHook ];
nativeBuildInputs = [ cargo pkg-config protobuf rustc rustPlatform.cargoSetupHook ];
buildInputs = [ openssl pkgs.zstd ];
dependencies = [ bcrypt build fastapi grpcio httpx importlib-resources jsonschema kubernetes mmh3 numpy onnxruntime opentelemetry-api opentelemetry-exporter-otlp-proto-grpc opentelemetry-instrumentation-fastapi opentelemetry-sdk orjson overrides posthog pydantic pypika pyyaml requests tenacity tokenizers tqdm typer typing-extensions uvicorn ];
optional-dependencies = { dev = [ chroma-hnswlib ]; };
nativeCheckInputs = [ chroma-hnswlib hnswlib hypothesis pandas psutil pytest-asyncio pytest-xdist pytestCheckHook sqlite starlette writableTmpDirAsHomeHook ];
# Disable on aarch64-linux due to broken onnxruntime # https://github.com/microsoft/onnxruntime/issues/10038 pythonImportsCheck = lib.optionals doCheck [ "chromadb" ];
# Test collection breaks on aarch64-linux doCheck = with stdenv.buildPlatform; !(isAarch && isLinux);
env = { ZSTD_SYS_USE_PKG_CONFIG = true; SWAGGER_UI_DOWNLOAD_URL = "file://${swagger-ui}"; };
pytestFlags = [ "-x" # these are slow tests, so stop on the first failure "-v" "-Wignore:DeprecationWarning" "-Wignore:PytestCollectionWarning" ];
preCheck = '' (($(ulimit -n) < 1024)) && ulimit -n 1024 export HOME=$(mktemp -d) '';
disabledTests = [ # Tests are flaky / timing sensitive "test_fastapi_server_token_authn_allows_when_it_should_allow" "test_fastapi_server_token_authn_rejects_when_it_should_reject"
# Issue with event loop "test_http_client_bw_compatibility"
# httpx ReadError "test_not_existing_collection_delete"
# Tests launch a server and try to connect to it # These either have https connection errors or name resolution errors "test_collection_query_with_invalid_collection_throws" "test_collection_update_with_invalid_collection_throws" "test_default_embedding" "test_invalid_index_params" "test_peek" "test_persist_index_loading" "test_query_id_filtering_e2e" "test_query_id_filtering_medium_dataset" "test_query_id_filtering_small_dataset" "test_ssl_self_signed_without_ssl_verify" "test_ssl_self_signed"
# Apparent race condition with sqlite # See https://github.com/chroma-core/chroma/issues/4661 "test_multithreaded_get_or_create" ];
disabledTestPaths = [ # Tests require network access "bin/rust_python_compat_test.py" "chromadb/test/configurations/test_collection_configuration.py" "chromadb/test/ef/test_default_ef.py" "chromadb/test/ef/test_onnx_mini_lm_l6_v2.py" "chromadb/test/ef/test_voyageai_ef.py" "chromadb/test/property/" "chromadb/test/property/test_cross_version_persist.py" "chromadb/test/stress/" "chromadb/test/test_api.py"
# Tests time out (waiting for server) "chromadb/test/test_cli.py"
# Cannot find protobuf file while loading test "chromadb/test/distributed/test_log_failover.py"
# ValueError: An instance of Chroma already exists for ephemeral with different settings "chromadb/test/test_chroma.py" "chromadb/test/ef/test_multimodal_ef.py" ];
__darwinAllowLocalNetworking = true;
passthru = { tests = { inherit (nixosTests) chromadb; };
updateScript = nix-update-script { # we have to update both the python hash and the cargo one, # so use nix-update-script extraArgs = [ "--version-regex" "([0-9].+)" ]; }; };
meta = { description = "AI-native open-source embedding database"; homepage = "https://github.com/chroma-core/chroma"; changelog = "https://github.com/chroma-core/chroma/releases/tag/${version}"; license = lib.licenses.asl20; maintainers = with lib.maintainers; [ fab sarahec ]; mainProgram = "chroma"; };}