pkgs/by-name/sk/skypilot/package.nix

   1 {
   2   lib,
   3   fetchFromGitHub,
   4   python3Packages,
   5 }:
   6
   7 python3Packages.buildPythonApplication rec {
   8   pname = "skypilot";
   9   version = "0.6.1";
  10
  11   src = fetchFromGitHub {
  12     owner = "skypilot-org";
  13     repo = "skypilot";
  14     rev = "refs/tags/v${version}";
  15     hash = "sha256-ZrNI9s7U39SMHqIzOtyuth8Wrkn+T2KSsMfpqO1pxoI=";
  16   };
  17
  18   pyproject = true;
  19
  20   build-system = with python3Packages; [ setuptools ];
  21
  22   # when updating, please ensure package version constraints stipulaed
  23   # in setup.py are met
  24   propagatedBuildInputs = with python3Packages; [
  25     cachetools
  26     click
  27     colorama
  28     cryptography
  29     filelock
  30     jinja2
  31     jsonschema
  32     networkx
  33     packaging
  34     pandas
  35     pendulum
  36     prettytable
  37     psutil
  38     python-dotenv
  39     pyyaml
  40     pulp
  41     requests
  42     rich
  43     tabulate
  44     typing-extensions
  45     wheel
  46   ];
  47
  48   meta = {
  49     description = "Run LLMs and AI on any Cloud";
  50     longDescription = ''
  51       SkyPilot is a framework for running LLMs, AI, and batch jobs on any
  52       cloud, offering maximum cost savings, highest GPU availability, and
  53       managed execution.
  54     '';
  55     homepage = "https://github.com/skypilot-org/skypilot";
  56     license = lib.licenses.asl20;
  57     maintainers = with lib.maintainers; [ seanrmurphy ];
  58     mainProgram = "sky";
  59   };
  60 }