Update scripts/setup_vectorstore.py
Browse files
scripts/setup_vectorstore.py
CHANGED
@@ -7,6 +7,8 @@ def main():
|
|
7 |
BASE_DIR = Path(__file__).resolve().parent.parent
|
8 |
CHUNKS_PATH = BASE_DIR / "output" / "chunks.pkl"
|
9 |
DB_DIR = BASE_DIR / "db"
|
|
|
|
|
10 |
|
11 |
with open(CHUNKS_PATH, "rb") as f:
|
12 |
chunks = pickle.load(f)
|
|
|
7 |
BASE_DIR = Path(__file__).resolve().parent.parent
|
8 |
CHUNKS_PATH = BASE_DIR / "output" / "chunks.pkl"
|
9 |
DB_DIR = BASE_DIR / "db"
|
10 |
+
|
11 |
+
DB_DIR.mkdir(parents=True, exist_ok=True) # <-- create the db folder if not exists
|
12 |
|
13 |
with open(CHUNKS_PATH, "rb") as f:
|
14 |
chunks = pickle.load(f)
|