diff --git a/pkgs/development/python-modules/graphrag/default.nix b/pkgs/development/python-modules/graphrag/default.nix index ef62b9eb8fc5..d00cc1b4b26e 100644 --- a/pkgs/development/python-modules/graphrag/default.nix +++ b/pkgs/development/python-modules/graphrag/default.nix @@ -14,6 +14,7 @@ environs, fastparquet, graspologic, + json-repair, lancedb, networkx, nltk, @@ -39,14 +40,14 @@ buildPythonPackage rec { pname = "graphrag"; - version = "0.2.0"; + version = "0.2.2"; pyproject = true; src = fetchFromGitHub { owner = "microsoft"; repo = "graphrag"; rev = "refs/tags/v${version}"; - hash = "sha256-VOcUjfRik4sdk6xbrAe5I7788d2/l4tuUByJajSjo4Q="; + hash = "sha256-X4mhnKSaQQo0i10EsdaDtMgKwfBxFvOOrkDrkUM2cQI="; }; build-system = [ @@ -57,9 +58,11 @@ buildPythonPackage rec { pythonRelaxDeps = [ "aiofiles" "azure-identity" + "json-repair" "lancedb" "scipy" "tenacity" + "textual" "tiktoken" ]; @@ -74,6 +77,7 @@ buildPythonPackage rec { environs fastparquet graspologic + json-repair lancedb networkx nltk diff --git a/pkgs/development/python-modules/json-repair/default.nix b/pkgs/development/python-modules/json-repair/default.nix new file mode 100644 index 000000000000..72037ce322b3 --- /dev/null +++ b/pkgs/development/python-modules/json-repair/default.nix @@ -0,0 +1,35 @@ +{ + lib, + buildPythonPackage, + fetchFromGitHub, + setuptools, + pytestCheckHook, +}: + +buildPythonPackage rec { + pname = "json-repair"; + version = "0.27.0"; + pyproject = true; + + src = fetchFromGitHub { + owner = "mangiucugna"; + repo = "json_repair"; + rev = "refs/tags/${version}"; + hash = "sha256-kaTwdS2zRt94pJko9AKEvszLR4z62u3ZrKlfXkJr5TQ="; + }; + + build-system = [ setuptools ]; + + nativeBuildInputs = [ pytestCheckHook ]; + + disabledTestPaths = [ "tests/test_performance.py" ]; + + pythonImportsCheck = [ "json_repair" ]; + + meta = with lib; { + homepage = "https://github.com/mangiucugna/json_repair/"; + description = "repair invalid JSON, commonly used to parse the output of LLMs"; + license = licenses.mit; + maintainers = with maintainers; [ greg ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 890224f6109e..8723e43860ce 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -6361,6 +6361,8 @@ self: super: with self; { json-merge-patch = callPackage ../development/python-modules/json-merge-patch { }; + json-repair = callPackage ../development/python-modules/json-repair { }; + json-schema-for-humans = callPackage ../development/python-modules/json-schema-for-humans { }; json-stream = callPackage ../development/python-modules/json-stream { };