Jaita commited on
Commit
ace2d6c
·
verified ·
1 Parent(s): d0b39ff

Update kb_embed.py

Browse files
Files changed (1) hide show
  1. kb_embed.py +3 -0
kb_embed.py CHANGED
@@ -11,6 +11,9 @@ CHROMA_DIR = BASE_DIR / "chroma_db"
11
  MODEL_DIR = BASE_DIR / "all-MiniLM-L6-v2"
12
  DOCS_DIR = BASE_DIR / "GenericSOPsForTesting"
13
 
 
 
 
14
  # --- ChromaDB persistent client ---
15
  client = chromadb.PersistentClient(
16
  path=str(CHROMA_DIR),
 
11
  MODEL_DIR = BASE_DIR / "all-MiniLM-L6-v2"
12
  DOCS_DIR = BASE_DIR / "GenericSOPsForTesting"
13
 
14
+ # Ensure persistence folder exists
15
+ CHROMA_DIR.mkdir(parents=True, exist_ok=True)
16
+
17
  # --- ChromaDB persistent client ---
18
  client = chromadb.PersistentClient(
19
  path=str(CHROMA_DIR),