yujunjun/graphiti
1
0
Fork 0
mirror of https://github.com/getzep/graphiti.git synced 2025-06-27 02:00:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

update falkordb (#589)

Browse Source 
* update falkordb

* updates

* updates
This commit is contained in:
Preston Rasmussen 2025-06-16 12:02:08 -04:00 committed by GitHub
parent 1c51a18430
commit fb6d674fc2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
examples/quickstart/quickstart_falkordb.py
View File

@ -24,6 +24,7 @@ from logging import INFO
from dotenv import load_dotenv from dotenv import load_dotenv
from graphiti_core import Graphiti from graphiti_core import Graphiti
from graphiti_core.driver.falkordb_driver import FalkorDriver
from graphiti_core.nodes import EpisodeType from graphiti_core.nodes import EpisodeType
from graphiti_core.search.search_config_recipes import NODE_HYBRID_SEARCH_RRF from graphiti_core.search.search_config_recipes import NODE_HYBRID_SEARCH_RRF
@ -47,6 +48,8 @@ load_dotenv()
# FalkorDB connection parameters # FalkorDB connection parameters
# Make sure FalkorDB on premises is running, see https://docs.falkordb.com/ # Make sure FalkorDB on premises is running, see https://docs.falkordb.com/
falkor_uri = os.environ.get('FALKORDB_URI', 'falkor://localhost:6379') 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: if not falkor_uri:
raise ValueError('FALKORDB_URI must be set') raise ValueError('FALKORDB_URI must be set')
@ -62,7 +65,8 @@ async def main():
################################################# #################################################
# Initialize Graphiti with FalkorDB connection # 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: try:
# Initialize the graph database with graphiti's indices. This only needs to be done once. # Initialize the graph database with graphiti's indices. This only needs to be done once.

9
graphiti_core/driver/falkordb_driver.py
View File

@ -71,12 +71,11 @@ class FalkorDriver(GraphDriver):
password: str, password: str,
): ):
super().__init__() super().__init__()
if user and password: uri_parts = uri.split('://', 1)
uri_parts = uri.split('://', 1) uri = f'{uri_parts[0]}://{user}:{password}@{uri_parts[1]}'
uri = f'{uri_parts[0]}://{user}:{password}@{uri_parts[1]}'
self.client = FalkorDB.from_url( self.client = FalkorDB(
url=uri, host='your-db.falkor.cloud', port=6380, password='your_password', ssl=True
) )
def _get_graph(self, graph_name: str | None) -> FalkorGraph: def _get_graph(self, graph_name: str | None) -> FalkorGraph: