AIMessage import

agent.py

@@ -1,20 +1,20 @@
 """LangGraph Agent"""
 import os
-from dotenv import load_dotenv
-from langgraph.graph import START, StateGraph, MessagesState
-from langgraph.prebuilt import tools_condition
-from langgraph.prebuilt import ToolNode
-from langchain_google_genai import ChatGoogleGenerativeAI
-from langchain_groq import ChatGroq
-from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint, HuggingFaceEmbeddings
+from langchain_core.messages import HumanMessage, SystemMessage, AIMessage
+from langchain_core.tools import tool
+from langchain_core.runnables.graph import StateGraph, START
+from langchain_core.runnables.history import MessagesState
+from langchain_core.runnables.utils import ToolNode, tools_condition
+from langchain_community.utilities import WikipediaLoader, ArxivLoader
 from langchain_community.tools.tavily_search import TavilySearchResults
-from langchain_community.document_loaders import WikipediaLoader
-from langchain_community.document_loaders import ArxivLoader
+from langchain_community.embeddings import HuggingFaceEmbeddings
 from langchain_community.vectorstores import SupabaseVectorStore
-from langchain_core.messages import SystemMessage, HumanMessage
-from langchain_core.tools import tool
-from langchain.tools.retriever import create_retriever_tool
-from supabase.client import Client, create_client
+from langchain_groq import ChatGroq
+from langchain_google_genai import ChatGoogleGenerativeAI
+from langchain_community.chat_models import ChatHuggingFace
+from langchain_community.llms import HuggingFaceEndpoint
+from supabase import create_client, Client
+from dotenv import load_dotenv
 
 load_dotenv()