Update scripts/chunk_and_embed.py
Browse files- scripts/chunk_and_embed.py +18 -19
scripts/chunk_and_embed.py
CHANGED
@@ -1,20 +1,19 @@
|
|
1 |
-
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
2 |
-
import pickle
|
3 |
-
import os
|
4 |
-
|
5 |
-
|
6 |
-
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
-
pickle.dump(chunks, f)
|
20 |
print(f"📦 Chunks saved to {CHUNKS_PATH}")
|
|
|
1 |
+
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
2 |
+
import pickle
|
3 |
+
import os
|
4 |
+
|
5 |
+
|
6 |
+
BASE_DIR = Path(__file__).resolve().parent.parent
|
7 |
+
DOCS_PATH = BASE_DIR / "output" / "all_docs.pkl"
|
8 |
+
CHUNKS_PATH = BASE_DIR / "output" / "chunks.pkl"
|
9 |
+
|
10 |
+
with open(DOCS_PATH, "rb") as f:
|
11 |
+
all_docs = pickle.load(f)
|
12 |
+
|
13 |
+
splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=100)
|
14 |
+
chunks = splitter.split_documents(all_docs)
|
15 |
+
print(f"✅ Split into {len(chunks)} chunks")
|
16 |
+
|
17 |
+
with open(CHUNKS_PATH, "wb") as f:
|
18 |
+
pickle.dump(chunks, f)
|
|
|
19 |
print(f"📦 Chunks saved to {CHUNKS_PATH}")
|