diff --git a/pkgs/development/python-modules/fireworks-ai/default.nix b/pkgs/development/python-modules/fireworks-ai/default.nix new file mode 100644 index 000000000000..cea3bd3ef9b9 --- /dev/null +++ b/pkgs/development/python-modules/fireworks-ai/default.nix @@ -0,0 +1,79 @@ +{ + lib, + buildPythonPackage, + fetchPypi, + + # build-system + setuptools, + versioneer, + + # dependencies + httpx, + httpx-ws, + httpx-sse, + pydantic, + pillow, + + # optional dependencies + fastapi, + gitignore-parser, + openapi-spec-validator, + prance, + safetensors, + tabulate, + torch, + tqdm, +}: + +buildPythonPackage rec { + pname = "fireworks-ai"; + version = "0.15.12"; + pyproject = true; + + # no source available + src = fetchPypi { + pname = "fireworks_ai"; + inherit version; + hash = "sha256-I4ClPZIkTGCP05j42XuXOA2Jnz/3EAkfS1CRe3URnsI="; + }; + + build-system = [ + setuptools + versioneer + ]; + + dependencies = [ + httpx + httpx-ws + httpx-sse + pydantic + pillow + ]; + + 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 ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 1613dfd9e0b5..36c2db9e9a03 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -4932,6 +4932,8 @@ self: super: with self; { firetv = callPackage ../development/python-modules/firetv { }; + fireworks-ai = callPackage ../development/python-modules/fireworks-ai { }; + first = callPackage ../development/python-modules/first { }; fissix = callPackage ../development/python-modules/fissix { };