160 lines
2.4 KiB
Nix
160 lines
2.4 KiB
Nix
{
|
|
lib,
|
|
buildPythonPackage,
|
|
fetchPypi,
|
|
|
|
# build-system
|
|
pdm-backend,
|
|
|
|
# local dependencies
|
|
black,
|
|
mypy,
|
|
|
|
# dependencies
|
|
grpcio,
|
|
grpclib,
|
|
httpx-sse,
|
|
httpx-ws,
|
|
httpx,
|
|
mmh3,
|
|
openai,
|
|
pillow,
|
|
protobuf,
|
|
pydantic,
|
|
python-dateutil,
|
|
rich,
|
|
typing-extensions,
|
|
|
|
# optional dependencies
|
|
fastapi,
|
|
gitignore-parser,
|
|
openapi-spec-validator,
|
|
prance,
|
|
safetensors,
|
|
tabulate,
|
|
torch,
|
|
tqdm,
|
|
}:
|
|
|
|
let
|
|
asyncstdlib-fw = buildPythonPackage rec {
|
|
pname = "asyncstdlib_fw";
|
|
version = "3.13.2";
|
|
pyproject = true;
|
|
|
|
src = fetchPypi {
|
|
inherit pname version;
|
|
hash = "sha256-Ua0JTCBMWTbDBA84wy/W1UmzkcmA8h8foJW2X7aAah8=";
|
|
};
|
|
|
|
build-system = [
|
|
pdm-backend
|
|
];
|
|
|
|
dependencies = [
|
|
black
|
|
mypy
|
|
];
|
|
|
|
pythonImportsCheck = [
|
|
"asyncstdlib"
|
|
];
|
|
};
|
|
|
|
betterproto-fw = buildPythonPackage rec {
|
|
pname = "betterproto_fw";
|
|
version = "2.0.3";
|
|
pyproject = true;
|
|
|
|
src = fetchPypi {
|
|
inherit version pname;
|
|
hash = "sha256-ut5GchUiTygHhC2hj+gSWKCoVnZrrV8KIKFHTFzba5M=";
|
|
};
|
|
|
|
build-system = [
|
|
pdm-backend
|
|
];
|
|
|
|
dependencies = [
|
|
grpclib
|
|
python-dateutil
|
|
typing-extensions
|
|
];
|
|
|
|
pythonImportsCheck = [
|
|
"betterproto"
|
|
];
|
|
|
|
};
|
|
in
|
|
buildPythonPackage rec {
|
|
pname = "fireworks-ai";
|
|
version = "0.17.16";
|
|
pyproject = true;
|
|
|
|
# no source available
|
|
src = fetchPypi {
|
|
pname = "fireworks_ai";
|
|
inherit version;
|
|
hash = "sha256-WblcAaYjnzwPS4n5rixNHbHLNGTE3bTPXvQ9lYZ1f9A=";
|
|
};
|
|
|
|
build-system = [
|
|
pdm-backend
|
|
];
|
|
|
|
pythonRelaxDeps = [
|
|
"protobuf"
|
|
];
|
|
|
|
dependencies = [
|
|
asyncstdlib-fw
|
|
betterproto-fw
|
|
grpcio
|
|
grpclib
|
|
httpx
|
|
httpx
|
|
httpx-sse
|
|
httpx-sse
|
|
httpx-ws
|
|
httpx-ws
|
|
mmh3
|
|
openai
|
|
pillow
|
|
pillow
|
|
protobuf
|
|
pydantic
|
|
pydantic
|
|
python-dateutil
|
|
rich
|
|
typing-extensions
|
|
];
|
|
|
|
optional-dependencies = {
|
|
flumina = [
|
|
fastapi
|
|
gitignore-parser
|
|
openapi-spec-validator
|
|
prance
|
|
safetensors
|
|
tabulate
|
|
torch
|
|
tqdm
|
|
];
|
|
};
|
|
|
|
# no tests available
|
|
doCheck = false;
|
|
|
|
pythonImportsCheck = [
|
|
"fireworks"
|
|
];
|
|
|
|
meta = {
|
|
description = "Client library for the Fireworks.ai platform";
|
|
homepage = "https://pypi.org/project/fireworks-ai/";
|
|
license = lib.licenses.mit;
|
|
maintainers = with lib.maintainers; [ sarahec ];
|
|
};
|
|
}
|