Spaces:
Sleeping
Sleeping
Update file_loader.py
Browse files- file_loader.py +2 -2
file_loader.py
CHANGED
@@ -10,7 +10,7 @@ from helpers import (
|
|
10 |
get_json_splits_only, # Xử lý file JSON (FAQ)
|
11 |
get_urls_splits, # Xử lý dữ liệu từ web
|
12 |
)
|
13 |
-
def
|
14 |
### Xử lý tất cả các tài liệu và nhét vào database
|
15 |
folder_path = "syllabus_nct_word_format/"
|
16 |
docx_files = list_docx_files(folder_path)
|
@@ -30,4 +30,4 @@ def get_splits_and_vectorstore()
|
|
30 |
embedding = GoogleGenerativeAIEmbeddings(model="models/text-embedding-004")
|
31 |
vectorstore = FAISS.from_documents(documents=all_splits, embedding=embedding)
|
32 |
|
33 |
-
return
|
|
|
10 |
get_json_splits_only, # Xử lý file JSON (FAQ)
|
11 |
get_urls_splits, # Xử lý dữ liệu từ web
|
12 |
)
|
13 |
+
def get_vectorstore():
|
14 |
### Xử lý tất cả các tài liệu và nhét vào database
|
15 |
folder_path = "syllabus_nct_word_format/"
|
16 |
docx_files = list_docx_files(folder_path)
|
|
|
30 |
embedding = GoogleGenerativeAIEmbeddings(model="models/text-embedding-004")
|
31 |
vectorstore = FAISS.from_documents(documents=all_splits, embedding=embedding)
|
32 |
|
33 |
+
return vectorstore
|