pkgs/development/python-modules/llama-index-llms-ollama/default.nix

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