stuebinm
ff1a94e523
The nixpkgs-unstable channel's programs.sqlite was used to identify packages producing exactly one binary, and these automatically added to their package definitions wherever possible.
164 lines
2.8 KiB
Nix
164 lines
2.8 KiB
Nix
{ lib
|
|
, stdenv
|
|
, bcrypt
|
|
, build
|
|
, buildPythonPackage
|
|
, cargo
|
|
, chroma-hnswlib
|
|
, darwin
|
|
, fastapi
|
|
, fetchFromGitHub
|
|
, grpcio
|
|
, hypothesis
|
|
, importlib-resources
|
|
, kubernetes
|
|
, mmh3
|
|
, numpy
|
|
, onnxruntime
|
|
, openssl
|
|
, opentelemetry-api
|
|
, opentelemetry-exporter-otlp-proto-grpc
|
|
, opentelemetry-instrumentation-fastapi
|
|
, opentelemetry-sdk
|
|
, orjson
|
|
, overrides
|
|
, pkg-config
|
|
, posthog
|
|
, protobuf
|
|
, pulsar-client
|
|
, pydantic
|
|
, pypika
|
|
, pytest-asyncio
|
|
, pytestCheckHook
|
|
, pythonOlder
|
|
, pythonRelaxDepsHook
|
|
, pyyaml
|
|
, requests
|
|
, rustc
|
|
, rustPlatform
|
|
, setuptools
|
|
, setuptools-scm
|
|
, tenacity
|
|
, tokenizers
|
|
, tqdm
|
|
, typer
|
|
, typing-extensions
|
|
, uvicorn
|
|
, zstd
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "chromadb";
|
|
version = "0.4.23";
|
|
pyproject = true;
|
|
|
|
disabled = pythonOlder "3.9";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "chroma-core";
|
|
repo = "chroma";
|
|
rev = "refs/tags/${version}";
|
|
hash = "sha256-5gI+FE2jx4G/qahATLcYsONfPZZkk1RFFYK5nrpE0Ug=";
|
|
};
|
|
|
|
cargoDeps = rustPlatform.fetchCargoTarball {
|
|
inherit src;
|
|
name = "${pname}-${version}";
|
|
hash = "sha256-glItbT8gg5SAySnfx3A9TaPyFmd1R46JpAB1JnjBE5M=";
|
|
};
|
|
|
|
pythonRelaxDeps = [
|
|
"orjson"
|
|
];
|
|
|
|
nativeBuildInputs = [
|
|
cargo
|
|
pkg-config
|
|
protobuf
|
|
pythonRelaxDepsHook
|
|
rustc
|
|
rustPlatform.cargoSetupHook
|
|
setuptools
|
|
setuptools-scm
|
|
];
|
|
|
|
buildInputs = [
|
|
openssl
|
|
zstd
|
|
] ++ lib.optionals stdenv.isDarwin [
|
|
darwin.apple_sdk.frameworks.Security
|
|
];
|
|
|
|
propagatedBuildInputs = [
|
|
bcrypt
|
|
build
|
|
chroma-hnswlib
|
|
fastapi
|
|
grpcio
|
|
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
|
|
pytest-asyncio
|
|
pytestCheckHook
|
|
];
|
|
|
|
pythonImportsCheck = [
|
|
"chromadb"
|
|
];
|
|
|
|
env = {
|
|
ZSTD_SYS_USE_PKG_CONFIG = true;
|
|
};
|
|
|
|
pytestFlagsArray = [ "-x" ];
|
|
|
|
preCheck = ''
|
|
export HOME=$(mktemp -d)
|
|
'';
|
|
|
|
disabledTestPaths = [
|
|
# Tests require network access
|
|
"chromadb/test/property/test_cross_version_persist.py"
|
|
"chromadb/test/auth/test_simple_rbac_authz.py"
|
|
"chromadb/test/ef/test_default_ef.py"
|
|
"chromadb/test/test_api.py"
|
|
"chromadb/test/property/"
|
|
"chromadb/test/stress/"
|
|
];
|
|
|
|
__darwinAllowLocalNetworking = true;
|
|
|
|
meta = with lib; {
|
|
description = "The AI-native open-source embedding database";
|
|
mainProgram = "chroma";
|
|
homepage = "https://github.com/chroma-core/chroma";
|
|
changelog = "https://github.com/chroma-core/chroma/releases/tag/${version}";
|
|
license = licenses.asl20;
|
|
maintainers = with maintainers; [ fab ];
|
|
};
|
|
}
|