SCR_Course_ChatBot / scripts /chunk_and_embed.py
MaryamKarimi080's picture
Upload 3 files
5f096cc verified
raw
history blame
691 Bytes
from langchain.text_splitter import RecursiveCharacterTextSplitter
import pickle
import os
DOCS_PATH = "E:/courses/LangChain Project/main root/output/all_docs.pkl"
CHUNKS_PATH = "E:/courses/LangChain Project/main root/output/chunks.pkl"
if not os.path.exists(DOCS_PATH):
raise FileNotFoundError("Run load_documents.py first")
with open(DOCS_PATH, "rb") as f:
all_docs = pickle.load(f)
splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=100)
chunks = splitter.split_documents(all_docs)
print(f"βœ… Split into {len(chunks)} chunks")
with open(CHUNKS_PATH, "wb") as f:
pickle.dump(chunks, f)
print(f"πŸ“¦ Chunks saved to {CHUNKS_PATH}")