diff --git a/lightrag/wrapper/llama_index_impl.py b/lightrag/wrapper/llama_index_impl.py
index f79dade5..4e1618b1 100644
--- a/lightrag/wrapper/llama_index_impl.py
+++ b/lightrag/wrapper/llama_index_impl.py
@@ -5,6 +5,7 @@ from llama_index.core.llms import (
     ChatResponse,
 )
 from typing import List, Optional
+from lightrag.utils import logger
 
 # Install required dependencies
 if not pm.is_installed("llama-index"):