Fabian Affolter 2024-11-28 09:34:11 +01:00
parent a4dd838b66
commit 9230f0ec9d

View File

@ -29,6 +29,7 @@
rich,
tenacity,
tiktoken,
typer,
typing-extensions,
umap-learn,
nbformat,
@ -38,14 +39,14 @@
buildPythonPackage rec {
pname = "graphrag";
version = "0.3.6";
version = "0.5.0";
pyproject = true;
src = fetchFromGitHub {
owner = "microsoft";
repo = "graphrag";
rev = "refs/tags/v${version}";
hash = "sha256-H5ITK4m3l+rlIEYXoMIpsE9faCu6rRZuB5zaZQeArOU=";
hash = "sha256-QK6ZdBDSSKi/WUsDQeEY5JfxgsmW/vK7yDfjMseAO/k=";
};
build-system = [
@ -81,6 +82,7 @@ buildPythonPackage rec {
rich
tenacity
tiktoken
typer
typing-extensions
umap-learn
];
@ -104,6 +106,7 @@ buildPythonPackage rec {
"test_find"
"test_run_extract_entities_multiple_documents"
"test_run_extract_entities_single_document"
"test_sort_context"
"test_sort_context_max_tokens"
];