diff --git a/BCEmbedding/tools/llama_index/bce_rerank.py b/BCEmbedding/tools/llama_index/bce_rerank.py
index 6f53cb8..84e193e 100644
--- a/BCEmbedding/tools/llama_index/bce_rerank.py
+++ b/BCEmbedding/tools/llama_index/bce_rerank.py
@@ -8,10 +8,10 @@
 from typing import Any, List, Optional
 
 from pydantic.v1 import Field, PrivateAttr
-from llama_index.callbacks import CBEventType, EventPayload
-from llama_index.postprocessor.types import BaseNodePostprocessor
-from llama_index.schema import MetadataMode, NodeWithScore, QueryBundle
-from llama_index.utils import infer_torch_device
+from llama_index.core.callbacks import CBEventType, EventPayload
+from llama_index.core.postprocessor.types import BaseNodePostprocessor
+from llama_index.core.schema import MetadataMode, NodeWithScore, QueryBundle
+from llama_index.core.utils import infer_torch_device
 
 
 class BCERerank(BaseNodePostprocessor):