fix constraints (#25)
This commit is contained in:
parent
8a55f48f5e
commit
7152a211ae
2 changed files with 2 additions and 12 deletions
|
|
@ -13,17 +13,6 @@ logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
async def build_indices_and_constraints(driver: AsyncDriver):
|
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] = [
|
range_indices: list[LiteralString] = [
|
||||||
'CREATE INDEX entity_uuid IF NOT EXISTS FOR (n:Entity) ON (n.uuid)',
|
'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)',
|
'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] = (
|
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])
|
await asyncio.gather(*[driver.execute_query(query) for query in index_queries])
|
||||||
|
|
|
||||||
|
|
@ -42,6 +42,7 @@ async def main(use_bulk: bool = True):
|
||||||
setup_logging()
|
setup_logging()
|
||||||
client = Graphiti(neo4j_uri, neo4j_user, neo4j_password)
|
client = Graphiti(neo4j_uri, neo4j_user, neo4j_password)
|
||||||
await clear_data(client.driver)
|
await clear_data(client.driver)
|
||||||
|
await client.build_indices_and_constraints()
|
||||||
messages = parse_podcast_messages()
|
messages = parse_podcast_messages()
|
||||||
|
|
||||||
if not use_bulk:
|
if not use_bulk:
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue