Renamed JSONRepairError to InvalidJSONError

Author: willtai

docs/source/api.rst

@@ -511,8 +511,8 @@ PipelineStatusUpdateError
    :show-inheritance:
 
 
-JSONRepairError
-===============
+InvalidJSONError
+================
 
-.. autoclass:: neo4j_graphrag.experimental.pipeline.exceptions.JSONRepairError
+.. autoclass:: neo4j_graphrag.experimental.pipeline.exceptions.InvalidJSONError
    :show-inheritance:

src/neo4j_graphrag/experimental/components/entity_relation_extractor.py

@@ -37,7 +37,7 @@
     TextChunks,
 )
 from neo4j_graphrag.experimental.pipeline.component import Component
-from neo4j_graphrag.experimental.pipeline.exceptions import JSONRepairError
+from neo4j_graphrag.experimental.pipeline.exceptions import InvalidJSONError
 from neo4j_graphrag.generation.prompts import ERExtractionTemplate, PromptTemplate
 from neo4j_graphrag.llm import LLMInterface
 
@@ -111,9 +111,9 @@ def fix_invalid_json(raw_json: str) -> str:
         repaired_json = ""
 
     if repaired_json == '""':
-        raise JSONRepairError("JSON repair resulted in an empty or invalid JSON.")
+        raise InvalidJSONError("JSON repair resulted in an empty or invalid JSON.")
     if not repaired_json:
-        raise JSONRepairError("JSON repair resulted in an empty string.")
+        raise InvalidJSONError("JSON repair resulted in an empty string.")
     return repaired_json
 
 
@@ -218,7 +218,7 @@ async def extract_for_chunk(
         try:
             llm_generated_json = fix_invalid_json(llm_result.content)
             result = json.loads(llm_generated_json)
-        except (json.JSONDecodeError, JSONRepairError) as e:
+        except (json.JSONDecodeError, InvalidJSONError) as e:
             if self.on_error == OnError.RAISE:
                 raise LLMGenerationError(
                     f"LLM response is not valid JSON {llm_result.content}: {e}"

src/neo4j_graphrag/experimental/pipeline/exceptions.py

@@ -33,7 +33,7 @@ class PipelineStatusUpdateError(Neo4jGraphRagError):
     pass
 
 
-class JSONRepairError(Neo4jGraphRagError):
+class InvalidJSONError(Neo4jGraphRagError):
     """Raised when JSON repair fails to produce valid JSON."""
 
     pass

tests/unit/experimental/components/test_entity_relation_extractor.py

@@ -31,7 +31,7 @@
     TextChunk,
     TextChunks,
 )
-from neo4j_graphrag.experimental.pipeline.exceptions import JSONRepairError
+from neo4j_graphrag.experimental.pipeline.exceptions import InvalidJSONError
 from neo4j_graphrag.llm import LLMInterface, LLMResponse
 
 
@@ -215,7 +215,7 @@ def test_fix_invalid_json_empty_result() -> None:
     json_string = "invalid json"
 
     with patch("json_repair.repair_json", return_value=""):
-        with pytest.raises(JSONRepairError):
+        with pytest.raises(InvalidJSONError):
             fix_invalid_json(json_string)