pkgs/development/python-modules/llama-index-agent-openai/default.nix

   1 {
   2   lib,
   3   buildPythonPackage,
   4   fetchPypi,
   5   llama-index-core,
   6   llama-index-llms-openai,
   7   poetry-core,
   8   pythonOlder,
   9   pythonRelaxDepsHook,
  10 }:
  11
  12 buildPythonPackage rec {
  13   pname = "llama-index-agent-openai";
  14   version = "0.2.3";
  15   pyproject = true;
  16
  17   disabled = pythonOlder "3.8";
  18
  19   src = fetchPypi {
  20     pname = "llama_index_agent_openai";
  21     inherit version;
  22     hash = "sha256-yJnZCzIDZlao74bQ8DeNQWjgDrLXWhCQHqtYulsmVqQ=";
  23   };
  24
  25   pythonRelaxDeps = [ "llama-index-llms-openai" ];
  26
  27   build-system = [
  28     poetry-core
  29     pythonRelaxDepsHook
  30   ];
  31
  32   dependencies = [
  33     llama-index-core
  34     llama-index-llms-openai
  35   ];
  36
  37   pythonImportsCheck = [ "llama_index.agent.openai" ];
  38
  39   meta = with lib; {
  40     description = "LlamaIndex Agent Integration for OpenAI";
  41     homepage = "https://github.com/run-llama/llama_index/tree/main/llama-index-integrations/agent/llama-index-agent-openai";
  42     license = licenses.mit;
  43     maintainers = with maintainers; [ fab ];
  44   };
  45 }