From 0ba9342f5d9c6d99edc9aa8df12c458a76eb7caf Mon Sep 17 00:00:00 2001
From: Vasilije <8619304+Vasilije1990@users.noreply.github.com>
Date: Tue, 20 Feb 2024 17:41:40 +0100
Subject: [PATCH] Fixes to create db script

---
 .../database/graphdb/networkx/networkx_graph.py               | 4 ++--
 cognitive_architecture/database/relationaldb/database.py      | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/cognitive_architecture/database/graphdb/networkx/networkx_graph.py b/cognitive_architecture/database/graphdb/networkx/networkx_graph.py
index 137ae07ca..8fd1b4b84 100644
--- a/cognitive_architecture/database/graphdb/networkx/networkx_graph.py
+++ b/cognitive_architecture/database/graphdb/networkx/networkx_graph.py
@@ -9,8 +9,8 @@ config  = config.load()
 
 class NetworkXGraphDB:
     """A class to manage a graph database using NetworkX"""
-    graph_path = (Path(config.base_path) / config.graph_name).absolute()
-    def __init__(self, filename=graph_path):
+    # graph_path = (Path(config.db_path) / config.graph_name).absolute()
+    def __init__(self, filename="cognee_graph.pkl"):
         self.filename = filename
         try:
             self.graph = self.load_graph()  # Attempt to load an existing graph
diff --git a/cognitive_architecture/database/relationaldb/database.py b/cognitive_architecture/database/relationaldb/database.py
index b0410a557..47b50f841 100644
--- a/cognitive_architecture/database/relationaldb/database.py
+++ b/cognitive_architecture/database/relationaldb/database.py
@@ -14,14 +14,14 @@ RETRY_DELAY = 5
 def get_sqlalchemy_database_url(
     db_type = globalConfig.db_type,
     db_name = globalConfig.db_name,
-    base_path = globalConfig.db_path,
+    db_path = globalConfig.db_path,
     user = globalConfig.db_user,
     password = globalConfig.db_password,
     host = globalConfig.db_host,
     port = globalConfig.db_port,
 ):
     """Get the SQLAlchemy database URL based on parameters."""
-    db_path = (Path(base_path) / db_name).absolute()
+    db_path = (Path(db_path) / db_name).absolute()
     if db_type == "sqlite":
         return f"sqlite+aiosqlite:///{db_path}"  # SQLite uses file path
     elif db_type == "duckdb":