Fix import error for ChatMessage in token_counting (#11021)

run-llama · Feb 20, 2024 · 4b3b7e1 · 4b3b7e1
1 parent 8c807d8
commit 4b3b7e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llama-index-core/llama_index/core/utilities/token_counting.py b/llama-index-core/llama_index/core/utilities/token_counting.py
@@ -3,7 +3,7 @@
 
 from typing import Any, Callable, Dict, List, Optional
 
-from llama_index.core.llms import ChatMessage, MessageRole
+from llama_index.core.base.llms.types import ChatMessage, MessageRole
 from llama_index.core.utils import get_tokenizer