Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-06 03:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add optional arguments to FalkorDBGraph constructor (#13459)

Browse Source 
**Description:** Add optional arguments to FalkorDBGraph constructor
**Tag maintainer:** baskaryan 
**Twitter handle:** @g_korland
This commit is contained in:
Guy Korland 2023-11-17 04:15:40 +02:00 committed by GitHub
parent e3a5cd7969
commit 7f8fd70ac4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
libs/langchain/langchain/graphs/falkordb_graph.py
View File

@ -1,4 +1,4 @@
from typing import Any, Dict, List
from typing import Any, Dict, List, Optional
from langchain.graphs.graph_document import GraphDocument
from langchain.graphs.graph_store import GraphStore
@ -48,7 +48,13 @@ class FalkorDBGraph(GraphStore):
"""
def __init__(
self, database: str, host: str = "localhost", port: int = 6379
self,
database: str,
host: str = "localhost",
port: int = 6379,
username: Optional[str] = None,
password: Optional[str] = None,
ssl: bool = False,
) -> None:
"""Create a new FalkorDB graph wrapper instance."""
try:
@ -60,7 +66,9 @@ class FalkorDBGraph(GraphStore):
"Please install it with `pip install redis`."
)
self._driver = redis.Redis(host=host, port=port)
self._driver = redis.Redis(
host=host, port=port, username=username, password=password, ssl=ssl
)
self._graph = Graph(self._driver, database)
self.schema: str = ""
self.structured_schema: Dict[str, Any] = {}