gmakstutis/graphiti
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Implement build_indices_and_constraints for Kuzu and Neptune drivers (#1048)

Browse source 
* Implement build_indices_and_constraints for Kuzu and Neptune drivers

Both Kuzu and Neptune drivers were missing the build_indices_and_constraints
method required by the abstract GraphDriver base class. Kuzu uses a no-op
implementation since indices are created during schema setup. Neptune delegates
to its existing OpenSearch (AOSS) index creation logic.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

* Skip fulltext search tests for Kuzu driver

Kuzu requires explicit schema definition and doesn't support dynamic fulltext
index creation like Neo4j or FalkorDB. These tests were already being skipped
for FalkorDB for similar reasons. Updated all four fulltext search tests to
skip for both FalkorDB and Kuzu providers.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

---------

Co-authored-by: Claude <noreply@anthropic.com>
This commit is contained in:
Daniel Chalef 2025-11-05 08:51:59 -08:00 • committed by GitHub
parent 50659cc805
commit f7816c29b3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 20 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
graphiti_core/driver/kuzu_driver.py
View file

@ -140,6 +140,12 @@ class KuzuDriver(GraphDriver):
def delete_all_indexes(self, database_: str):
pass
async def build_indices_and_constraints(self, delete_existing: bool = False):
# Kuzu doesn't support dynamic index creation like Neo4j or FalkorDB
# Schema and indices are created during setup_schema()
# This method is required by the abstract base class but is a no-op for Kuzu
pass
def setup_schema(self):
conn = kuzu.Connection(self.db)
conn.execute(SCHEMA_QUERIES)

6
graphiti_core/driver/neptune_driver.py
View file

@ -243,6 +243,12 @@ class NeptuneDriver(GraphDriver):
if client.indices.exists(index=index_name):
client.indices.delete(index=index_name)
async def build_indices_and_constraints(self, delete_existing: bool = False):
# Neptune uses OpenSearch (AOSS) for indexing
if delete_existing:
await self.delete_aoss_indices()
await self.create_aoss_indices()
def run_aoss_query(self, name: str, query_text: str, limit: int = 10) -> dict[str, Any]:
for index in aoss_indices:
if name.lower() == index['index_name']:

16
tests/test_graphiti_mock.py
View file

@ -918,8 +918,8 @@ async def test_get_communities_by_nodes(graph_driver, mock_embedder):
async def test_edge_fulltext_search(
graph_driver, mock_embedder, mock_llm_client, mock_cross_encoder_client
):
if graph_driver.provider == GraphProvider.FALKORDB:
pytest.skip('Skipping as tests fail on Falkordb')
if graph_driver.provider in [GraphProvider.FALKORDB, GraphProvider.KUZU]:
pytest.skip('Skipping as fulltext indexing not supported for FalkorDB and Kuzu')
graphiti = Graphiti(
graph_driver=graph_driver,
@ -1307,8 +1307,8 @@ async def test_edge_bfs_search(graph_driver, mock_embedder):
async def test_node_fulltext_search(
graph_driver, mock_embedder, mock_llm_client, mock_cross_encoder_client
):
if graph_driver.provider == GraphProvider.FALKORDB:
pytest.skip('Skipping as tests fail on Falkordb')
if graph_driver.provider in [GraphProvider.FALKORDB, GraphProvider.KUZU]:
pytest.skip('Skipping as fulltext indexing not supported for FalkorDB and Kuzu')
graphiti = Graphiti(
graph_driver=graph_driver,
@ -1516,8 +1516,8 @@ async def test_node_bfs_search(graph_driver, mock_embedder):
async def test_episode_fulltext_search(
graph_driver, mock_embedder, mock_llm_client, mock_cross_encoder_client
):
if graph_driver.provider == GraphProvider.FALKORDB:
pytest.skip('Skipping as tests fail on Falkordb')
if graph_driver.provider in [GraphProvider.FALKORDB, GraphProvider.KUZU]:
pytest.skip('Skipping as fulltext indexing not supported for FalkorDB and Kuzu')
graphiti = Graphiti(
graph_driver=graph_driver,
@ -1567,8 +1567,8 @@ async def test_episode_fulltext_search(
async def test_community_fulltext_search(
graph_driver, mock_embedder, mock_llm_client, mock_cross_encoder_client
):
if graph_driver.provider == GraphProvider.FALKORDB:
pytest.skip('Skipping as tests fail on Falkordb')
if graph_driver.provider in [GraphProvider.FALKORDB, GraphProvider.KUZU]:
pytest.skip('Skipping as fulltext indexing not supported for FalkorDB and Kuzu')
graphiti = Graphiti(
graph_driver=graph_driver,