pyrox.dev
1{
2 lib,
3 buildPythonPackage,
4 fetchPypi,
5
6 # build-system
7 setuptools,
8 versioneer,
9
10 # dependencies
11 httpx,
12 httpx-ws,
13 httpx-sse,
14 pydantic,
15 pillow,
16
17 # optional dependencies
18 fastapi,
19 gitignore-parser,
20 openapi-spec-validator,
21 prance,
22 safetensors,
23 tabulate,
24 torch,
25 tqdm,
26}:
27
28buildPythonPackage rec {
29 pname = "fireworks-ai";
30 version = "0.15.13";
31 pyproject = true;
32
33 # no source available
34 src = fetchPypi {
35 pname = "fireworks_ai";
36 inherit version;
37 hash = "sha256-ZZSF4R1HOYpNmKnL2OPWoUwdSJJ2j2e3+hzW0QH55io=";
38 };
39
40 build-system = [
41 setuptools
42 versioneer
43 ];
44
45 dependencies = [
46 httpx
47 httpx-ws
48 httpx-sse
49 pydantic
50 pillow
51 ];
52
53 optional-dependencies = {
54 flumina = [
55 fastapi
56 gitignore-parser
57 openapi-spec-validator
58 prance
59 safetensors
60 tabulate
61 torch
62 tqdm
63 ];
64 };
65
66 # no tests available
67 doCheck = false;
68
69 pythonImportsCheck = [
70 "fireworks"
71 ];
72
73 meta = {
74 description = "Client library for the Fireworks.ai platform";
75 homepage = "https://pypi.org/project/fireworks-ai/";
76 license = lib.licenses.mit;
77 maintainers = with lib.maintainers; [ sarahec ];
78 };
79}