From 7f8fd70ac4c71d10c59e00ec69861efe3404113c Mon Sep 17 00:00:00 2001 From: Guy Korland Date: Fri, 17 Nov 2023 04:15:40 +0200 Subject: [PATCH] Add optional arguments to FalkorDBGraph constructor (#13459) **Description:** Add optional arguments to FalkorDBGraph constructor **Tag maintainer:** baskaryan **Twitter handle:** @g_korland --- libs/langchain/langchain/graphs/falkordb_graph.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/libs/langchain/langchain/graphs/falkordb_graph.py b/libs/langchain/langchain/graphs/falkordb_graph.py index 9829dac345..754455ce98 100644 --- a/libs/langchain/langchain/graphs/falkordb_graph.py +++ b/libs/langchain/langchain/graphs/falkordb_graph.py @@ -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] = {}