From 048ae19dfee189a59459555824e6b0feabe6fc8a Mon Sep 17 00:00:00 2001
From: Sarah Clark <seclark@nextquestion.net>
Date: Mon, 28 Apr 2025 17:05:08 -0700
Subject: [PATCH] python3Packages.fireworks-ai: init at 0.15.12

---
 .../python-modules/fireworks-ai/default.nix   | 79 +++++++++++++++++++
 pkgs/top-level/python-packages.nix            |  2 +
 2 files changed, 81 insertions(+)
 create mode 100644 pkgs/development/python-modules/fireworks-ai/default.nix

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 1f055b73bc97..bc58713588ad 100644
--- a/pkgs/top-level/python-packages.nix
+++ b/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 { };