Update agent.py

agent.py

@@ -25,7 +25,6 @@ import uuid
 import requests
 import json
 from langchain_core.documents import Document
-from langchain_community.vectorstores import FAISS
 from youtube_transcript_api import YouTubeTranscriptApi
 from youtube_transcript_api._errors import TranscriptsDisabled, VideoUnavailable
 import re
@@ -62,7 +61,8 @@ from langchain.schema import Document
 
 from langchain_community.vectorstores import FAISS
 from langchain_community.embeddings import HuggingFaceEmbeddings
-from langchain.tools.retriever import create_retriever_tool
+#from langchain.tools.retriever import create_retriever_tool
+from langchain_community.tools import create_retriever_tool
 
 load_dotenv()
 
@@ -343,12 +343,7 @@ for name in enabled_tool_names:
 # Prepare Documents
 # -----------------------------
 # Define the URL where the JSON file is hosted
-from typing import TypedDict, Annotated, List
-import gradio as gr
-from langchain.schema import Document
-import json
-from langchain.embeddings import HuggingFaceEmbeddings
-from langchain.vectorstores import FAISS
+
 import faiss
 
 # 1. Type-Checked State for Gradio
@@ -397,14 +392,8 @@ def create_documents(data_source: str, data: List[dict]) -> List[Document]:
     return docs
 
 # 4. Vector Store Integration
-import os
-import json
+
 import faiss
-from langchain.vectorstores import FAISS
-from langchain.embeddings import HuggingFaceEmbeddings
-from langchain.tools import create_retriever_tool
-from langchain.llms import HuggingFaceEndpoint
-from your_module import create_documents  # Replace with your actual import
 
 # Custom FAISS wrapper (optional, if you still want it)
 class MyVector_Store: