Vishaltiwari2019 commited on
Commit
a2991c7
·
verified ·
1 Parent(s): c0546cc

Update tools.py

Browse files
Files changed (1) hide show
  1. tools.py +2 -2
tools.py CHANGED
@@ -4,7 +4,7 @@ from langchain.chat_models import ChatOpenAI
4
  from langchain.document_loaders import PyPDFLoader
5
  from langchain.text_splitter import CharacterTextSplitter
6
  from langchain.vectorstores import FAISS
7
- from langchain.embeddings import HuggingFaceEmbeddings
8
  import serpapi
9
 
10
  load_dotenv()
@@ -17,7 +17,7 @@ llm = ChatOpenAI(
17
  )
18
 
19
  # Embeddings (HuggingFace)
20
- embeddings = HuggingFaceEmbeddings(model_name="mistral3")
21
 
22
  # Load PDFs and create FAISS vectorstore
23
  def load_vectorstore(pdf_dir="pdfs/"):
 
4
  from langchain.document_loaders import PyPDFLoader
5
  from langchain.text_splitter import CharacterTextSplitter
6
  from langchain.vectorstores import FAISS
7
+ from langchain.embeddings import SentenceTransformerEmbeddings
8
  import serpapi
9
 
10
  load_dotenv()
 
17
  )
18
 
19
  # Embeddings (HuggingFace)
20
+ embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
21
 
22
  # Load PDFs and create FAISS vectorstore
23
  def load_vectorstore(pdf_dir="pdfs/"):