python312Packages.graphrag: init at 0.1.1

{

  lib,

  buildPythonPackage,

  fetchFromGitHub,

  poetry-core,

  poetry-dynamic-versioning,

  aiofiles,

  aiolimiter,

  azure-identity,

  azure-search-documents,

  azure-storage-blob,

  datashaper,

  devtools,

  environs,

  fastparquet,

  graspologic,

  lancedb,

  networkx,

  nltk,

  numba,

  numpy,

  openai,

  pyaml-env,

  pydantic,

  python-dotenv,

  pyyaml,

  rich,

  scipy,

  swifter,

  tenacity,

  textual,

  tiktoken,

  typing-extensions,

  uvloop,

  nbformat,

  pytest-asyncio,

  pytestCheckHook,

}:

buildPythonPackage rec {

  pname = "graphrag";

  version = "0.1.1";

  pyproject = true;

  src = fetchFromGitHub {

    owner = "microsoft";

    repo = "graphrag";

    rev = "refs/tags/v${version}";

    hash = "sha256-hIAQOIqm9S9AtssE6UxcXfaIbSt3+506ueMrlathNaQ=";

  };

  build-system = [

    poetry-core

    poetry-dynamic-versioning

  ];

  pythonRelaxDeps = [

    "aiofiles"

    "azure-identity"

    "scipy"

    "tiktoken"

  ];

  dependencies = [

    aiofiles

    aiolimiter

    azure-identity

    azure-search-documents

    azure-storage-blob

    datashaper

    devtools

    environs

    fastparquet

    graspologic

    lancedb

    networkx

    nltk

    numba

    numpy

    openai

    pyaml-env

    pydantic

    python-dotenv

    pyyaml

    rich

    scipy

    swifter

    tenacity

    textual

    tiktoken

    typing-extensions

    uvloop

  ];

  env.NUMBA_CACHE_DIR = "$TMPDIR";

  pythonImportsCheck = [ "graphrag" ];

  nativeCheckInputs = [

    nbformat

    pytest-asyncio

    pytestCheckHook

  ];

  pytestFlagsArray = [ "tests/unit" ];

  disabledTests = [

    # touch the network

    "test_child"

    "test_dotprefix"

    "test_find"

    "test_run_extract_entities_multiple_documents"

    "test_run_extract_entities_single_document"

  ];

  meta = {

    description = "Modular graph-based Retrieval-Augmented Generation (RAG) system";

    homepage = "https://github.com/microsoft/graphrag";

    license = lib.licenses.mit;

    maintainers = with lib.maintainers; [ natsukium ];

  };

}

  graphql-subscription-manager = callPackage ../development/python-modules/graphql-subscription-manager { };

  graphrag = callPackage ../development/python-modules/graphrag { };

  graph-tool = callPackage ../development/python-modules/graph-tool { };

  graphtage = callPackage ../development/python-modules/graphtage { };