Update agent.py

agent.py

@@ -59,7 +59,7 @@ from youtube_transcript_api._errors import TranscriptsDisabled, VideoUnavailable
 from langchain.schema import Document
 
 from langchain_community.vectorstores import FAISS
-from langchain_community.embeddings import HuggingFaceEmbeddings
+from langchain_huggingface import HuggingFaceEmbeddings
 from langchain.tools.retriever import create_retriever_tool
 #from langchain_community.tools import create_retriever_tool
 from typing import TypedDict, Annotated, List
@@ -429,18 +429,24 @@ embedding_model = HuggingFaceEmbeddings(
 # -----------------------------
 # Create FAISS index and save it
 # -----------------------------
-vector_store = FAISS.from_documents(
-    documents=docs,
-    embedding=embedding_model
-)
+try:
+    vector_store = FAISS.load_local(
+        "/home/wendy/Downloads/faiss_index/faiss_index",
+        embedding_model,
+        allow_dangerous_deserialization=True
+    )
+except Exception as e:
+    print(f"Error loading index: {str(e)}")
+    # Fallback to rebuilding index
+    vector_store = FAISS.from_documents(docs, embedding_model)
 
-vector_store.save_local("/home/wendy/Downloads/faiss_index")
 
 # -----------------------------
 # Load FAISS index properly
 # -----------------------------
 loaded_store = FAISS.load_local("/home/wendy/Downloads/faiss_index", embedding_model)
 
+
 # -----------------------------
 # Create LangChain Retriever Tool
 # -----------------------------