pkraman06 commited on
Commit
14b6034
·
verified ·
1 Parent(s): 6c39801

Create memory/vector_store.py

Browse files
Files changed (1) hide show
  1. memory/vector_store.py +10 -0
memory/vector_store.py ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ from langchain.vectorstores import FAISS
2
+ from langchain.embeddings import HuggingFaceEmbeddings
3
+
4
+
5
+ def create_vector_store(documents):
6
+ embeddings = HuggingFaceEmbeddings(
7
+ model_name="sentence-transformers/all-MiniLM-L6-v2"
8
+ )
9
+
10
+ return FAISS.from_documents(documents, embeddings)