From 4b3b7e137669599d4c923f26a550b4ec42973dca Mon Sep 17 00:00:00 2001
From: Lalit Moharana <105036971+lalitbyteiq@users.noreply.github.com>
Date: Tue, 20 Feb 2024 20:32:30 +0530
Subject: [PATCH] Fix import error for ChatMessage in token_counting (#11021)

---
 llama-index-core/llama_index/core/utilities/token_counting.py | 2 +-
 1 file changed, 1 insertion(+), 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
index a58b7f424d6d0..a06d0c14c3879 100644
--- 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