Update ingest.py

ingest.py

@@ -5,10 +5,9 @@ import requests
 import os
 import openai
 from bs4 import BeautifulSoup
-from langchain.embeddings.openai import OpenAIEmbeddings
-from langchain.embeddings import CohereEmbeddings, HuggingFaceHubEmbeddings
-from langchain.vectorstores import FAISS, Annoy, DocArrayInMemorySearch
-from langchain.document_loaders import CSVLoader
+from langchain_openai import OpenAIEmbeddings
+from langchain_community.vectorstores import FAISS
+from langchain_community.document_loaders import CSVLoader
 import csv
 import pickle
 
@@ -65,12 +64,10 @@ def ingest_data():
     embeddings = OpenAIEmbeddings()
     db = FAISS.from_documents(data, embeddings)
 
-    return db
+    # Save vectorstore
 
-    # # Save vectorstore
-
-    # with open("db.pkl", "wb") as f:
-    #     pickle.dump(db, f)
+    with open("db.pkl", "wb") as f:
+        pickle.dump(db, f)
 
 if __name__ == "__main__":
     ingest_data()