Update vectorstore.py
Browse files- 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)
|