Pulastya0 commited on
Commit
a71b4b0
·
verified ·
1 Parent(s): 430d263

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -34,10 +34,12 @@ async def setup_endpoint(req: SetupRequest):
34
  kb_data = json.load(f)
35
 
36
  # Create ChromaDB collection if not exists
37
- chroma_client = chromadb.Client(Settings(
38
- chroma_db_impl="duckdb+parquet",
39
- persist_directory="/data/chroma_db"
40
- ))
 
 
41
 
42
  try:
43
  kb_collection = chroma_client.get_collection("kb_collection")
 
34
  kb_data = json.load(f)
35
 
36
  # Create ChromaDB collection if not exists
37
+ chroma_client = chromadb.Client(
38
+ Settings(
39
+ persist_directory="/data/chroma_db", # store DB here
40
+ chroma_db_impl="duckdb+parquet"
41
+ )
42
+ )
43
 
44
  try:
45
  kb_collection = chroma_client.get_collection("kb_collection")