aman1762 commited on
Commit
b727ef9
·
verified ·
1 Parent(s): 10bcf8a

Update vectorstore.py

Browse files
Files changed (1) hide show
  1. vectorstore.py +6 -0
vectorstore.py CHANGED
@@ -2,7 +2,13 @@ from langchain_community.vectorstores import FAISS
2
  from langchain_community.embeddings import HuggingFaceEmbeddings
3
 
4
  def create_vectorstore(documents):
 
 
 
 
 
5
  embeddings = HuggingFaceEmbeddings(
6
  model_name="sentence-transformers/all-MiniLM-L6-v2"
7
  )
 
8
  return FAISS.from_documents(documents, embeddings)
 
2
  from langchain_community.embeddings import HuggingFaceEmbeddings
3
 
4
  def create_vectorstore(documents):
5
+ if not documents:
6
+ raise ValueError(
7
+ "No valid source code found to build vector index."
8
+ )
9
+
10
  embeddings = HuggingFaceEmbeddings(
11
  model_name="sentence-transformers/all-MiniLM-L6-v2"
12
  )
13
+
14
  return FAISS.from_documents(documents, embeddings)