depot/third_party/nixpkgs/pkgs/development/python-modules/openllm-client/default.nix
Default email 2c76a4cb41 Project import generated by Copybara.
GitOrigin-RevId: c757e9bd77b16ca2e03c89bf8bc9ecb28e0c06ad
2023-11-16 04:20:00 +00:00

63 lines
1.2 KiB
Nix

{ lib
, buildPythonPackage
, pythonOlder
, bentoml
, hatch-fancy-pypi-readme
, hatch-vcs
, hatchling
, attrs
, cattrs
, httpx
, openllm-core
, orjson
, soundfile
, transformers
}:
buildPythonPackage rec {
inherit (openllm-core) src version;
pname = "openllm-client";
pyproject = true;
disabled = pythonOlder "3.8";
sourceRoot = "source/openllm-client";
nativeBuildInputs = [
hatch-fancy-pypi-readme
hatch-vcs
hatchling
];
propagatedBuildInputs = [
attrs
cattrs
httpx
orjson
];
passthru.optional-dependencies = {
grpc = [
bentoml
] ++ bentoml.optional-dependencies.grpc;
agents = [
transformers
# diffusers
soundfile
] ++ transformers.agents;
full = passthru.optional-dependencies.grpc ++ passthru.optional-dependencies.agents;
};
# there is no tests
doCheck = false;
pythonImportsCheck = [ "openllm_client" ];
meta = with lib; {
description = "Interacting with OpenLLM HTTP/gRPC server, or any BentoML server";
homepage = "https://github.com/bentoml/OpenLLM/tree/main/openllm-client";
changelog = "https://github.com/bentoml/OpenLLM/blob/${src.rev}/CHANGELOG.md";
license = licenses.asl20;
maintainers = with maintainers; [ natsukium ];
};
}