Fix import error for ChatMessage in token_counting (#11021)

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