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

Fix for using non default neo4j database (#329)

Browse source 
Pass database_ correctly to driver.session to fix using non default database
This commit is contained in:
neonconsultingllc 2025-04-18 12:06:31 -05:00 • committed by GitHub
parent f578ee2177
commit d0b1b2e5db
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
graphiti_core/utils/bulk_utils.py
View file

@ -26,7 +26,7 @@ from pydantic import BaseModel
from typing_extensions import Any from typing_extensions import Any
from graphiti_core.edges import Edge, EntityEdge, EpisodicEdge from graphiti_core.edges import Edge, EntityEdge, EpisodicEdge
from graphiti_core.helpers import semaphore_gather from graphiti_core.helpers import DEFAULT_DATABASE, semaphore_gather
from graphiti_core.llm_client import LLMClient from graphiti_core.llm_client import LLMClient
from graphiti_core.models.edges.edge_db_queries import ( from graphiti_core.models.edges.edge_db_queries import (
ENTITY_EDGE_SAVE_BULK, ENTITY_EDGE_SAVE_BULK,
@ -95,7 +95,7 @@ async def add_nodes_and_edges_bulk(
entity_nodes: list[EntityNode], entity_nodes: list[EntityNode],
entity_edges: list[EntityEdge], entity_edges: list[EntityEdge],
): ):
async with driver.session() as session: async with driver.session(database=DEFAULT_DATABASE) as session:
await session.execute_write( await session.execute_write(
add_nodes_and_edges_bulk_tx, episodic_nodes, episodic_edges, entity_nodes, entity_edges add_nodes_and_edges_bulk_tx, episodic_nodes, episodic_edges, entity_nodes, entity_edges
) )

8
graphiti_core/utils/maintenance/graph_data_operations.py
View file

@ -42,7 +42,7 @@ async def build_indices_and_constraints(driver: AsyncDriver, delete_existing: bo
driver.execute_query( driver.execute_query(
"""DROP INDEX $name""", """DROP INDEX $name""",
name=name, name=name,
_database=DEFAULT_DATABASE, database_=DEFAULT_DATABASE,
) )
for name in index_names for name in index_names
] ]
@ -87,7 +87,7 @@ async def build_indices_and_constraints(driver: AsyncDriver, delete_existing: bo
*[ *[
driver.execute_query( driver.execute_query(
query, query,
_database=DEFAULT_DATABASE, database_=DEFAULT_DATABASE,
) )
for query in index_queries for query in index_queries
] ]
@ -95,7 +95,7 @@ async def build_indices_and_constraints(driver: AsyncDriver, delete_existing: bo
async def clear_data(driver: AsyncDriver, group_ids: list[str] | None = None): async def clear_data(driver: AsyncDriver, group_ids: list[str] | None = None):
async with driver.session() as session: async with driver.session(database=DEFAULT_DATABASE) as session:
async def delete_all(tx): async def delete_all(tx):
await tx.run('MATCH (n) DETACH DELETE n') await tx.run('MATCH (n) DETACH DELETE n')
@ -150,7 +150,7 @@ async def retrieve_episodes(
reference_time=reference_time, reference_time=reference_time,
num_episodes=last_n, num_episodes=last_n,
group_ids=group_ids, group_ids=group_ids,
_database=DEFAULT_DATABASE, database_=DEFAULT_DATABASE,
) )
episodes = [ episodes = [
EpisodicNode( EpisodicNode(