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

fix falkordb linting issues (#650)

Browse source 
Refactor FalkorDB driver: change port type to int and clean up whitespace
This commit is contained in:
Daniel Chalef 2025-06-30 12:55:05 -07:00 • committed by GitHub
parent 8213d10d44
commit daec70db65
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
graphiti_core/driver/falkordb_driver.py
View file

@ -66,7 +66,7 @@ class FalkorDriver(GraphDriver):
def __init__(
self,
host: str = 'localhost',
port: str = '6379',
port: int = 6379,
username: str | None = None,
password: str | None = None,
falkor_db: FalkorDB | None = None,
@ -74,8 +74,8 @@ class FalkorDriver(GraphDriver):
"""
Initialize the FalkorDB driver.
FalkorDB is a multi-tenant graph database.
To connect, provide the host and port.
FalkorDB is a multi-tenant graph database.
To connect, provide the host and port.
The default parameters assume a local (on-premises) FalkorDB instance.
"""
super().__init__()
@ -110,7 +110,7 @@ class FalkorDriver(GraphDriver):
# Convert the result header to a list of strings
header = [h[1] for h in result.header]
# Convert FalkorDB's result format (list of lists) to the format expected by Graphiti (list of dicts)
records = []
for row in result.result_set:
@ -122,7 +122,7 @@ class FalkorDriver(GraphDriver):
# If there are more fields in header than values in row, set to None
record[field_name] = None
records.append(record)
return records, header, None
def session(self, database: str | None) -> GraphDriverSession:
@ -131,7 +131,7 @@ class FalkorDriver(GraphDriver):
async def close(self) -> None:
"""Close the driver connection."""
if hasattr(self.client, 'aclose'):
await self.client.aclose()
await self.client.aclose() # type: ignore[reportUnknownMemberType]
elif hasattr(self.client.connection, 'aclose'):
await self.client.connection.aclose()
elif hasattr(self.client.connection, 'close'):