diff --git a/lightrag/base.py b/lightrag/base.py
index aa8e6d9e..210b0ca6 100644
--- a/lightrag/base.py
+++ b/lightrag/base.py
@@ -1,4 +1,5 @@
 import os
+from dotenv import load_dotenv
 from dataclasses import dataclass, field
 from enum import Enum
 from typing import (
@@ -9,12 +10,11 @@ from typing import (
     TypeVar,
     Union,
 )
-
 import numpy as np
-
 from .utils import EmbeddingFunc
 from .types import KnowledgeGraph
 
+load_dotenv()
 
 class TextChunkSchema(TypedDict):
     tokens: int