Schema refactoring (#261)

* Added format_schema function

* Added get_enhanced_schema function

* Adds docstrings

* More docstrings

* Updated docs

* Updated query_database

* Updated query_database

* Added enhanced schema unit tests

* Various test fixes

* Updated CHANGELOG

* Refactored format_schema

* Updated api.rst

* Updated api.rst

* Refactored get_enhanced_schema_cypher

* Refactored enhance_schema

* Bug fix

* Fixed flaky tests

* Makes helper functions private

* Update src/neo4j_graphrag/schema.py

Co-authored-by: Estelle Scifo <stellasia@users.noreply.github.com>

* Formatting

* Removed implicit query fallback execution from query_database

* Updated variables in NODE_PROPERTIES_QUERY and REL_PROPERTIES_QUERY

---------

Co-authored-by: Estelle Scifo <stellasia@users.noreply.github.com>

Author: alexthomas93

CHANGELOG.md

@@ -8,9 +8,14 @@
 
 - Ability to add event listener to get notifications about Pipeline progress.
 - Added py.typed so that mypy knows to use type annotations from the neo4j-graphrag package.
+- Support for creating enhanced schemas with detailed property statistics.
+- New utility functions for schema formatting and value sanitization.
+- Updated unit and integration tests to cover enhanced schema functionality.
 
 ### Changed
+
 - Changed the default behaviour of `FixedSizeSplitter` to avoid words cut-off in the chunks whenever it is possible.
+- Refactored schema creation code to reduce duplication and improve maintainability.
 
 ### Fixed
 - Removed the `uuid` package from dependencies (not needed with Python 3).

docs/source/api.rst

@@ -394,6 +394,8 @@ Database Interaction
 
 .. autofunction:: neo4j_graphrag.schema.get_schema
 
+.. autofunction:: neo4j_graphrag.schema.format_schema
+
 
 ******
 Errors