ai-codebase-intelligence / memory /vector_store.py
pkraman06's picture
Create memory/vector_store.py
14b6034 verified
raw
history blame contribute delete
295 Bytes
from langchain.vectorstores import FAISS
from langchain.embeddings import HuggingFaceEmbeddings
def create_vector_store(documents):
embeddings = HuggingFaceEmbeddings(
model_name="sentence-transformers/all-MiniLM-L6-v2"
)
return FAISS.from_documents(documents, embeddings)