diff --git a/examples/quickstart/quickstart_falkordb.py b/examples/quickstart/quickstart_falkordb.py index 82f6a94..b8839e7 100644 --- a/examples/quickstart/quickstart_falkordb.py +++ b/examples/quickstart/quickstart_falkordb.py @@ -24,6 +24,7 @@ from logging import INFO from dotenv import load_dotenv from graphiti_core import Graphiti +from graphiti_core.driver.falkordb_driver import FalkorDriver from graphiti_core.nodes import EpisodeType from graphiti_core.search.search_config_recipes import NODE_HYBRID_SEARCH_RRF @@ -47,6 +48,8 @@ load_dotenv() # FalkorDB connection parameters # Make sure FalkorDB on premises is running, see https://docs.falkordb.com/ falkor_uri = os.environ.get('FALKORDB_URI', 'falkor://localhost:6379') +falkor_user = os.environ.get('FALKORDB_USER', 'falkor') +falkor_password = os.environ.get('FALKORDB_PASSWORD', '') if not falkor_uri: raise ValueError('FALKORDB_URI must be set') @@ -62,7 +65,8 @@ async def main(): ################################################# # Initialize Graphiti with FalkorDB connection - graphiti = Graphiti(falkor_uri) + falkor_driver = FalkorDriver(uri=falkor_uri, user=falkor_user, password=falkor_password) + graphiti = Graphiti(uri=falkor_uri, graph_driver=falkor_driver) try: # Initialize the graph database with graphiti's indices. This only needs to be done once. diff --git a/graphiti_core/driver/falkordb_driver.py b/graphiti_core/driver/falkordb_driver.py index 1736966..7747c41 100644 --- a/graphiti_core/driver/falkordb_driver.py +++ b/graphiti_core/driver/falkordb_driver.py @@ -71,12 +71,11 @@ class FalkorDriver(GraphDriver): password: str, ): super().__init__() - if user and password: - uri_parts = uri.split('://', 1) - uri = f'{uri_parts[0]}://{user}:{password}@{uri_parts[1]}' + uri_parts = uri.split('://', 1) + uri = f'{uri_parts[0]}://{user}:{password}@{uri_parts[1]}' - self.client = FalkorDB.from_url( - url=uri, + self.client = FalkorDB( + host='your-db.falkor.cloud', port=6380, password='your_password', ssl=True ) def _get_graph(self, graph_name: str | None) -> FalkorGraph: