dd0f03a56c
They are not doing anything right now. This is in preparation for their complete removal from the tree. Note: several changes that affect the derivation inputs (e.g. removal of references to stub paths in build instructions) were left out. They will be cleaned up the next iteration and will require special care. Note: this PR is a result of a mix of ugly regex (not AST) based automation and some manual labor. For reference, the regex automation part was hacked in: https://github.com/booxter/nix-clean-apple_sdk Signed-off-by: Ihar Hrachyshka <ihar.hrachyshka@gmail.com>
184 lines
3.3 KiB
Nix
184 lines
3.3 KiB
Nix
{
|
|
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 src;
|
|
name = "${pname}-${version}";
|
|
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;
|
|
};
|
|
}
|