Fix M.env AX_GRAPH_NODES not working problem

lightrag/kg/neo4j_impl.py

@@ -29,12 +29,20 @@ from neo4j import (  # type: ignore
     AsyncManagedTransaction,
 )
 
-config = configparser.ConfigParser()
+from dotenv import load_dotenv
-config.read("config.ini", "utf-8")
+
+# use the .env that is inside the current folder
+# allows to use different .env file for each lightrag instance
+# the OS environment variables take precedence over the .env file
+load_dotenv(dotenv_path=".env", override=False)
 
 # Get maximum number of graph nodes from environment variable, default is 1000
 MAX_GRAPH_NODES = int(os.getenv("MAX_GRAPH_NODES", 1000))
 
+config = configparser.ConfigParser()
+config.read("config.ini", "utf-8")
+
+
 # Set neo4j logger level to ERROR to suppress warning logs
 logging.getLogger("neo4j").setLevel(logging.ERROR)
 

lightrag/kg/networkx_impl.py

@@ -21,6 +21,13 @@ from .shared_storage import (
     set_all_update_flags,
 )
 
+from dotenv import load_dotenv
+
+# use the .env that is inside the current folder
+# allows to use different .env file for each lightrag instance
+# the OS environment variables take precedence over the .env file
+load_dotenv(dotenv_path=".env", override=False)
+
 MAX_GRAPH_NODES = int(os.getenv("MAX_GRAPH_NODES", 1000))
 
 

lightrag/kg/postgres_impl.py

@@ -35,6 +35,13 @@ if not pm.is_installed("asyncpg"):
 import asyncpg  # type: ignore
 from asyncpg import Pool  # type: ignore
 
+from dotenv import load_dotenv
+
+# use the .env that is inside the current folder
+# allows to use different .env file for each lightrag instance
+# the OS environment variables take precedence over the .env file
+load_dotenv(dotenv_path=".env", override=False)
+
 # Get maximum number of graph nodes from environment variable, default is 1000
 MAX_GRAPH_NODES = int(os.getenv("MAX_GRAPH_NODES", 1000))