Something went wrong. Try again.
lol
Something went wrong. Try again.
7.0 kB · 296 lines
Nix
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297{ lib, stdenv, buildPythonPackage, fetchFromGitHub, fetchurl,
# build inputs cargo, openssl, pkg-config, protobuf, rustc, rustPlatform, zstd-c,
# 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, pybase64, pydantic, pydantic-settings, 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 (finalAttrs: { pname = "chromadb"; version = "1.5.7"; pyproject = true;
src = fetchFromGitHub { owner = "chroma-core"; repo = "chroma"; tag = finalAttrs.version; hash = "sha256-JrkfLwEL7iTL9P/4UDM4hFQtRL1JYH47dgZ1d+Mphqw="; };
cargoDeps = rustPlatform.fetchCargoVendor { inherit (finalAttrs) pname version src; hash = "sha256-Szy2mSTriMwMViVTbI+0XaizcQBKh1Ncipf84moDREI="; };
# 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="; };
postPatch = # Nixpkgs is taking the version from `chromadb_rust_bindings` which is versioned independently '' substituteInPlace pyproject.toml \ --replace-fail "dynamic = [\"version\"]" "version = \"${finalAttrs.version}\"" '' # Flip anonymized telemetry to opt in versus current opt-in out for privacy + '' substituteInPlace chromadb/config.py \ --replace-fail "anonymized_telemetry: bool = True" \ "anonymized_telemetry: bool = False" '' # error: queries overflow the depth limit! # https://github.com/chroma-core/chroma/issues/6891 # https://github.com/chroma-core/chroma/issues/6892 # https://github.com/chroma-core/chroma/issues/6687 + '' sed -i '1i #![recursion_limit = "256"]' rust/blockstore/src/lib.rs sed -i '1i #![recursion_limit = "256"]' rust/index/src/lib.rs sed -i '1i #![recursion_limit = "256"]' rust/segment/src/lib.rs '';
pythonRelaxDeps = [ "fastapi" "posthog" ];
build-system = [ rustPlatform.maturinBuildHook ];
nativeBuildInputs = [ cargo pkg-config protobuf rustc rustPlatform.cargoSetupHook ];
buildInputs = [ openssl zstd-c ];
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 pybase64 pydantic pydantic-settings 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 finalAttrs.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://${finalAttrs.swagger-ui}"; };
pytestFlags = [ "-v" "-Wignore:DeprecationWarning" "-Wignore:PytestCollectionWarning" ];
# Skip the distributed and integration tests # See https://github.com/chroma-core/chroma/issues/5315 preCheck = '' (($(ulimit -n) < 1024)) && ulimit -n 1024 export CHROMA_RUST_BINDINGS_TEST_ONLY=1 '';
enabledTestPaths = [ "chromadb/test" ];
disabledTests = [ # Failure in name resolution "test_collection_query_with_invalid_collection_throws" "test_collection_update_with_invalid_collection_throws" "test_default_embedding" "test_persist_index_loading"
# Deadlocks intermittently "test_app"
# Depends on specific floating-point precision "test_base64_conversion_is_identity_f16"
# No such file or directory: 'openssl' "test_ssl_self_signed_without_ssl_verify" "test_ssl_self_signed"
# https://github.com/chroma-core/chroma/issues/6029 "test_embedding_function_config_roundtrip"
# Requires network access "test_persistent_client_close" "test_persistent_client_context_manager" "test_ephemeral_client_close" "test_ephemeral_client_context_manager" "test_client_close_idempotent" ] ++ lib.optionals stdenv.hostPlatform.isDarwin [ # Fails in nixpkgs-review on Darwin due to concurrent copies running and the lack of network namespaces. "test_add_then_delete_n_minus_1" ];
disabledTestPaths = [ # Tests require network access "chromadb/test/distributed" "chromadb/test/ef" "chromadb/test/property/test_cross_version_persist.py" "chromadb/test/stress" "chromadb/test/api/test_schema_e2e.py"
# Excessively slow "chromadb/test/property/test_add.py" "chromadb/test/property/test_persist.py"
# ValueError: An instance of Chroma already exists for ephemeral with different settings "chromadb/test/test_chroma.py"
# RuntimeError: There is no current event loop in thread 'MainThread'. # https://github.com/chroma-core/chroma/issues/6659 "chromadb/test/test_client.py::test_http_client_with_inconsistent_host_settings[async_client]" "chromadb/test/test_client.py::test_http_client_with_inconsistent_port_settings[async_client]" "chromadb/test/test_client.py::test_http_client[async_client]" ];
__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/${finalAttrs.src.tag}"; license = lib.licenses.asl20; maintainers = with lib.maintainers; [ fab sarahec ]; mainProgram = "chroma"; };})