pkgs/development/python-modules/fastembed/default.nix

   1 { lib
   2 , buildPythonPackage
   3 , fetchFromGitHub
   4 , huggingface-hub
   5 , pythonOlder
   6 , pythonRelaxDepsHook
   7 , poetry-core
   8 , onnx
   9 , onnxruntime
  10 , requests
  11 , tokenizers
  12 , tqdm
  13 , pytestCheckHook
  14 }:
  15
  16 buildPythonPackage rec {
  17   pname = "fastembed";
  18   version = "0.2.2";
  19   pyproject = true;
  20
  21   disabled = pythonOlder "3.8";
  22
  23   src = fetchFromGitHub {
  24     owner = "qdrant";
  25     repo = "fastembed";
  26     rev = "refs/tags/v${version}";
  27     hash = "sha256-ufgco5wPBG19GM99rZV7LKQqEzzCv24I8026SMz0CH4=";
  28   };
  29
  30   nativeBuildInputs = [
  31     poetry-core
  32     pythonRelaxDepsHook
  33   ];
  34
  35   propagatedBuildInputs = [
  36     huggingface-hub
  37     onnx
  38     onnxruntime
  39     requests
  40     tokenizers
  41     tqdm
  42   ];
  43
  44   pythonImportsCheck = [
  45     "fastembed"
  46   ];
  47
  48   pythonRelaxDeps = [
  49     "huggingface-hub"
  50   ];
  51
  52   nativeCheckInputs = [
  53     pytestCheckHook
  54   ];
  55
  56   # there is one test and it requires network
  57   doCheck = false;
  58
  59   meta = with lib; {
  60     description = "Fast, Accurate, Lightweight Python library to make State of the Art Embedding";
  61     homepage = "https://github.com/qdrant/fastembed";
  62     license = licenses.asl20;
  63     maintainers = with maintainers; [ happysalada ];
  64   };
  65 }