Merge pull request #368 from AdityaKalraShorthillsAI/main
Added embedding_func argument in Neo4JStorage class's __init__ method in neo4j implementation
This commit is contained in:
@@ -27,8 +27,12 @@ class Neo4JStorage(BaseGraphStorage):
|
||||
def load_nx_graph(file_name):
|
||||
print("no preloading of graph with neo4j in production")
|
||||
|
||||
def __init__(self, namespace, global_config):
|
||||
super().__init__(namespace=namespace, global_config=global_config)
|
||||
def __init__(self, namespace, global_config, embedding_func):
|
||||
super().__init__(
|
||||
namespace=namespace,
|
||||
global_config=global_config,
|
||||
embedding_func=embedding_func,
|
||||
)
|
||||
self._driver = None
|
||||
self._driver_lock = asyncio.Lock()
|
||||
URI = os.environ["NEO4J_URI"]
|
||||
|
Reference in New Issue
Block a user