tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
1{
2 lib,
3 buildPythonPackage,
4 fetchPypi,
5
6 # build-system
7 pdm-backend,
8
9 # dependencies
10 asyncstdlib-fw,
11 betterproto-fw,
12 googleapis-common-protos,
13 grpcio,
14 grpclib,
15 httpx-sse,
16 httpx-ws,
17 httpx,
18 mmh3,
19 openai,
20 pillow,
21 protobuf,
22 pydantic,
23 python-dateutil,
24 rich,
25 toml,
26 typing-extensions,
27
28 # optional dependencies
29 fastapi,
30 gitignore-parser,
31 openapi-spec-validator,
32 prance,
33 safetensors,
34 tabulate,
35 torch,
36 tqdm,
37}:
38
39buildPythonPackage rec {
40 pname = "fireworks-ai";
41 version = "0.19.20";
42 pyproject = true;
43
44 # no source available
45 src = fetchPypi {
46 pname = "fireworks_ai";
47 inherit version;
48 hash = "sha256-zK8lO+vFnMEPPl79QGfqPdemZT7kQdCqAPiCrcXdqYQ=";
49 };
50
51 build-system = [
52 pdm-backend
53 ];
54
55 pythonRelaxDeps = [
56 "attrs"
57 "protobuf"
58 ];
59
60 dependencies = [
61 asyncstdlib-fw
62 betterproto-fw
63 googleapis-common-protos
64 grpcio
65 grpclib
66 httpx
67 httpx
68 httpx-sse
69 httpx-ws
70 mmh3
71 openai
72 pillow
73 protobuf
74 pydantic
75 python-dateutil
76 rich
77 toml
78 typing-extensions
79 ];
80
81 optional-dependencies = {
82 flumina = [
83 fastapi
84 gitignore-parser
85 openapi-spec-validator
86 prance
87 safetensors
88 tabulate
89 torch
90 tqdm
91 ];
92 };
93
94 # no tests available
95 doCheck = false;
96
97 pythonImportsCheck = [
98 "fireworks"
99 ];
100
101 meta = {
102 description = "Client library for the Fireworks.ai platform";
103 homepage = "https://pypi.org/project/fireworks-ai/";
104 license = lib.licenses.mit;
105 maintainers = with lib.maintainers; [ sarahec ];
106 };
107}