bakaEC/OliveSensorAPI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix bug

Browse Source
This commit is contained in:
zealot52099 2024-03-18 10:46:09 +08:00
parent 74db6d9893
commit 98ecdda78d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
rag/src/data_processing.py
View File

@ -63,8 +63,8 @@ def split_document(data_path, chunk_size=1000, chunk_overlap=100):
# logger.info(f"splits type {type(splits[0])}")
# logger.info(f'splits size {len(splits)}')
split_docs += splits
elif file.endswith('.txt'):
file_path = os.path.join(root, file)
elif data_path.endswith('.txt'):
file_path = os.path.join(root, data_path)
# logger.info(f'splitting file {file_path}')
text_loader = TextLoader(file_path, encoding='utf-8')
text = text_loader.load()