quoc-khanh commited on
Commit
32e14d1
·
verified ·
1 Parent(s): 77263a0

Update file_loader.py

Browse files
Files changed (1) hide show
  1. 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 get_splits_and_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,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 all_splits, vectorstore
 
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