Update agent.py

agent.py

@@ -14,7 +14,7 @@ from langchain_community.document_loaders import ArxivLoader
 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 langchain_community.tools import create_retriever_tool
 from supabase.client import Client, create_client
 from sentence_transformers import SentenceTransformer
 from sentence_transformers import SentenceTransformer
@@ -127,12 +127,12 @@ with open("system_prompt.txt", "r", encoding="utf-8") as f:
 sys_msg = SystemMessage(content=system_prompt)
 
 # Custom embedding class
-from sentence_transformers import SentenceTransformer
-from langchain_huggingface import HuggingFaceEmbeddings
-from supabase import create_client, Client
-from langchain_community.vectorstores import SupabaseVectorStore
-from langchain.tools import create_retriever_tool
-import os
+#from sentence_transformers import SentenceTransformer
+#from langchain_huggingface import HuggingFaceEmbeddings
+#from supabase import create_client, Client
+#from langchain_community.vectorstores import SupabaseVectorStore
+#from langchain.tools import create_retriever_tool
+#import os
 
 # Initialize SentenceTransformer and set max_seq_length
 sentence_transformer = SentenceTransformer("sentence-transformers/all-mpnet-base-v2")