Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
3.4 kB · 182 lines
Nix
at flake-libs
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183{ lib, stdenv, bcrypt, build, buildPythonPackage, cargo, chroma-hnswlib, fastapi, fetchFromGitHub, grpcio, httpx, hypothesis, importlib-resources, kubernetes, mmh3, nixosTests, numpy, onnxruntime, openssl, opentelemetry-api, opentelemetry-exporter-otlp-proto-grpc, opentelemetry-instrumentation-fastapi, opentelemetry-sdk, orjson, overrides, pkg-config, posthog, protobuf, psutil, pulsar-client, pydantic, pypika, pytest-asyncio, pytestCheckHook, pythonOlder, pyyaml, requests, rustc, rustPlatform, setuptools-scm, setuptools, tenacity, tokenizers, tqdm, typer, typing-extensions, uvicorn, zstd,}:
buildPythonPackage rec { pname = "chromadb"; version = "0.5.20"; pyproject = true;
disabled = pythonOlder "3.9";
src = fetchFromGitHub { owner = "chroma-core"; repo = "chroma"; tag = version; hash = "sha256-DQHkgCHtrn9xi7Kp7TZ5NP1EtFtTH5QOvne9PUvxsWc="; };
cargoDeps = rustPlatform.fetchCargoVendor { inherit pname version src; hash = "sha256-ZtCTg8qNCiqlH7RsZxaWUNAoazdgmXP2GtpjDpRdvbk="; };
pythonRelaxDeps = [ "chroma-hnswlib" "orjson" ];
build-system = [ setuptools setuptools-scm ];
nativeBuildInputs = [ cargo pkg-config protobuf rustc rustPlatform.cargoSetupHook ];
buildInputs = [ openssl zstd ];
dependencies = [ bcrypt build chroma-hnswlib fastapi grpcio httpx importlib-resources kubernetes mmh3 numpy onnxruntime opentelemetry-api opentelemetry-exporter-otlp-proto-grpc opentelemetry-instrumentation-fastapi opentelemetry-sdk orjson overrides posthog pulsar-client pydantic pypika pyyaml requests tenacity tokenizers tqdm typer typing-extensions uvicorn ];
nativeCheckInputs = [ hypothesis psutil pytest-asyncio pytestCheckHook ];
pythonImportsCheck = [ "chromadb" ];
env = { ZSTD_SYS_USE_PKG_CONFIG = true; };
pytestFlagsArray = [ "-x" ];
preCheck = '' (($(ulimit -n) < 1024)) && ulimit -n 1024 export HOME=$(mktemp -d) '';
disabledTests = [ # Tests are laky / 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" # Issue with httpx "test_not_existing_collection_delete" ];
disabledTestPaths = [ # Tests require network access "chromadb/test/auth/test_simple_rbac_authz.py" "chromadb/test/db/test_system.py" "chromadb/test/ef/test_default_ef.py" "chromadb/test/property/" "chromadb/test/property/test_cross_version_persist.py" "chromadb/test/stress/" "chromadb/test/test_api.py" ];
__darwinAllowLocalNetworking = true;
passthru.tests = { inherit (nixosTests) chromadb; };
meta = with lib; { 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 = licenses.asl20; maintainers = with maintainers; [ fab ]; mainProgram = "chroma"; broken = stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64; };}