diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py
index 67ef3aab..8ac41721 100644
--- a/lightrag/lightrag.py
+++ b/lightrag/lightrag.py
@@ -48,6 +48,10 @@ from .utils import (
     set_logger,
 )
 from .types import KnowledgeGraph
+from dotenv import load_dotenv
+
+# Load environment variables
+load_dotenv(override=True)
 
 # TODO: TO REMOVE @Yannick
 config = configparser.ConfigParser()
diff --git a/lightrag/operate.py b/lightrag/operate.py
index 772815ce..a0111719 100644
--- a/lightrag/operate.py
+++ b/lightrag/operate.py
@@ -5,6 +5,7 @@ import json
 import re
 from typing import Any, AsyncIterator
 from collections import Counter, defaultdict
+
 from .utils import (
     logger,
     clean_str,
@@ -34,6 +35,10 @@ from .base import (
 )
 from .prompt import GRAPH_FIELD_SEP, PROMPTS
 import time
+from dotenv import load_dotenv
+
+# Load environment variables
+load_dotenv(override=True)
 
 
 def chunking_by_token_size(
diff --git a/lightrag/utils.py b/lightrag/utils.py
index d17ce87d..e7217def 100644
--- a/lightrag/utils.py
+++ b/lightrag/utils.py
@@ -15,8 +15,11 @@ from typing import Any, Callable
 import xml.etree.ElementTree as ET
 import numpy as np
 import tiktoken
-
 from lightrag.prompt import PROMPTS
+from dotenv import load_dotenv
+
+# Load environment variables
+load_dotenv(override=True)
 
 
 VERBOSE_DEBUG = os.getenv("VERBOSE", "false").lower() == "true"