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