Spaces:
Runtime error
Runtime error
Fixed bug aligning the names of the vector stores on train versus inference
Browse files
src/training/train_rag.py
CHANGED
@@ -60,7 +60,7 @@ def connect_to_product_db(args) -> None:
|
|
60 |
|
61 |
|
62 |
def get_out_vector_store_client(args, allow_reset: bool = False) -> chromadb.Client:
|
63 |
-
out_dir = os.path.join(data_dir, 'vector_stores', args.out_vectors)
|
64 |
chroma_settings = Settings()
|
65 |
chroma_settings.allow_reset = allow_reset
|
66 |
return chromadb.PersistentClient(path=out_dir, settings=chroma_settings)
|
|
|
60 |
|
61 |
|
62 |
def get_out_vector_store_client(args, allow_reset: bool = False) -> chromadb.Client:
|
63 |
+
out_dir = os.path.join(data_dir, 'vector_stores', f"{args.out_vectors}_chroma")
|
64 |
chroma_settings = Settings()
|
65 |
chroma_settings.allow_reset = allow_reset
|
66 |
return chromadb.PersistentClient(path=out_dir, settings=chroma_settings)
|