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

fix constraints (#25)

Browse source
This commit is contained in:
Preston Rasmussen 2024-08-22 19:10:18 -04:00 • committed by GitHub
parent 8a55f48f5e
commit 7152a211ae
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 2 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
core/utils/maintenance/graph_data_operations.py
View file

@ -13,17 +13,6 @@ logger = logging.getLogger(__name__)
async def build_indices_and_constraints(driver: AsyncDriver):
constraints: list[LiteralString] = [
"""
CREATE CONSTRAINT entity_name IF NOT EXISTS
FOR (n:Entity) REQUIRE n.name IS UNIQUE
""",
"""
CREATE CONSTRAINT edge_facts IF NOT EXISTS
FOR ()-[e:RELATES_TO]-() REQUIRE e.fact IS UNIQUE
""",
]
range_indices: list[LiteralString] = [
'CREATE INDEX entity_uuid IF NOT EXISTS FOR (n:Entity) ON (n.uuid)',
'CREATE INDEX episode_uuid IF NOT EXISTS FOR (n:Episodic) ON (n.uuid)',
@ -64,7 +53,7 @@ async def build_indices_and_constraints(driver: AsyncDriver):
""",
]
index_queries: list[LiteralString] = (
constraints + range_indices + fulltext_indices + vector_indices
range_indices + fulltext_indices + vector_indices
)
await asyncio.gather(*[driver.execute_query(query) for query in index_queries])

1
examples/podcast/podcast_runner.py
View file

@ -42,6 +42,7 @@ async def main(use_bulk: bool = True):
setup_logging()
client = Graphiti(neo4j_uri, neo4j_user, neo4j_password)
await clear_data(client.driver)
await client.build_indices_and_constraints()
messages = parse_podcast_messages()
if not use_bulk: