Merge pull request #332625 from greg-hellings/graphrag-0.2.1

python312Packages.json-repair: init at 0.27.0, python312Packages.graphrag: 0.2.0 -> 0.2.2

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

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 ];
+  };
+}

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 { };