python3Packages.fireworks-ai: init at 0.15.12

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 ];
+  };
+}

pkgs/top-level/python-packages.nix

@@ -4930,6 +4930,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 { };