WebashalarForML commited on
Commit
1ca4c9d
·
verified ·
1 Parent(s): e0efaf2

Update retrival.py

Browse files
Files changed (1) hide show
  1. retrival.py +1 -1
retrival.py CHANGED
@@ -260,7 +260,7 @@ def split_text(documents: list[Document]):
260
 
261
  #def save_to_chroma(chunks: list[Document], name: str, tables: list[Document]):
262
  def save_to_chroma(chunks: list[Document], name: str):
263
- CHROMA_PATH = f"./VectorDB/chroma_{name}"
264
  #TABLE_PATH = f"./TableDB/chroma_{name}"
265
  if os.path.exists(CHROMA_PATH):
266
  shutil.rmtree(CHROMA_PATH)
 
260
 
261
  #def save_to_chroma(chunks: list[Document], name: str, tables: list[Document]):
262
  def save_to_chroma(chunks: list[Document], name: str):
263
+ CHROMA_PATH = f"./VectorDB/chroma_{name}"
264
  #TABLE_PATH = f"./TableDB/chroma_{name}"
265
  if os.path.exists(CHROMA_PATH):
266
  shutil.rmtree(CHROMA_PATH)