diff --git a/ra_aid/llm.py b/ra_aid/llm.py
index a505638..32d842d 100644
--- a/ra_aid/llm.py
+++ b/ra_aid/llm.py
@@ -3,7 +3,6 @@ from typing import Any, Dict, List, Optional
 
 from langchain_anthropic import ChatAnthropic
 from langchain_core.language_models import BaseChatModel
-from langchain_core.messages import BaseMessage
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_openai import ChatOpenAI
 from openai import OpenAI