pkgs/development/python-modules/llama-index-cli/default.nix

   1 {
   2   lib,
   3   buildPythonPackage,
   4   fetchPypi,
   5   llama-index-core,
   6   llama-index-embeddings-openai,
   7   llama-index-llms-openai,
   8   llama-index-vector-stores-chroma,
   9   poetry-core,
  10   pythonOlder,
  11 }:
  12
  13 buildPythonPackage rec {
  14   pname = "llama-index-cli";
  15   version = "0.1.12";
  16   pyproject = true;
  17
  18   disabled = pythonOlder "3.8";
  19
  20   src = fetchPypi {
  21     pname = "llama_index_cli";
  22     inherit version;
  23     hash = "sha256-PPH3BsPGnGsaqwf8p/qtOVnbFwmAjv1QSRtmnTiwtYA=";
  24   };
  25
  26   build-system = [ poetry-core ];
  27
  28   dependencies = [
  29     llama-index-core
  30     llama-index-embeddings-openai
  31     llama-index-llms-openai
  32     llama-index-vector-stores-chroma
  33   ];
  34
  35   # Tests are only available in the mono repo
  36   doCheck = false;
  37
  38   pythonImportsCheck = [ "llama_index.cli" ];
  39
  40   meta = with lib; {
  41     description = "LlamaIndex CLI";
  42     homepage = "https://github.com/run-llama/llama_index/tree/main/llama-index-cli";
  43     license = licenses.mit;
  44     maintainers = with maintainers; [ fab ];
  45   };
  46 }