diff --git a/pkgs/development/python-modules/fireworks-ai/default.nix b/pkgs/development/python-modules/fireworks-ai/default.nix index 7006fbfddf3f..9fcd5105dfe8 100644 --- a/pkgs/development/python-modules/fireworks-ai/default.nix +++ b/pkgs/development/python-modules/fireworks-ai/default.nix @@ -6,11 +6,10 @@ # build-system pdm-backend, - # local dependencies - black, - mypy, - # dependencies + asyncstdlib-fw, + betterproto-fw, + googleapis-common-protos, grpcio, grpclib, httpx-sse, @@ -23,6 +22,7 @@ pydantic, python-dateutil, rich, + toml, typing-extensions, # optional dependencies @@ -36,67 +36,16 @@ 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"; + version = "0.19.20"; pyproject = true; # no source available src = fetchPypi { pname = "fireworks_ai"; inherit version; - hash = "sha256-WblcAaYjnzwPS4n5rixNHbHLNGTE3bTPXvQ9lYZ1f9A="; + hash = "sha256-zK8lO+vFnMEPPl79QGfqPdemZT7kQdCqAPiCrcXdqYQ="; }; build-system = [ @@ -104,29 +53,28 @@ buildPythonPackage rec { ]; pythonRelaxDeps = [ + "attrs" "protobuf" ]; dependencies = [ asyncstdlib-fw betterproto-fw + googleapis-common-protos grpcio grpclib httpx httpx httpx-sse - httpx-sse - httpx-ws httpx-ws mmh3 openai pillow - pillow protobuf pydantic - pydantic python-dateutil rich + toml typing-extensions ];