Update app.py

app.py

@@ -19,32 +19,40 @@ class TicketRequest(BaseModel):
 class SetupRequest(BaseModel):
     kb_file: str  # path to KB.json
 
+# -------------------------------
+# Global variable for collection
+# -------------------------------
+kb_collection = None
+
 # -------------------------------
 # KB Setup Endpoint
 # -------------------------------
 @app.post("/setup")
 async def setup_endpoint(req: SetupRequest):
-    """Embed KB.json and store in ChromaDB"""
     global kb_collection
+
     if not os.path.exists(req.kb_file):
         raise HTTPException(status_code=404, detail="KB.json file not found")
 
-    # Load KB
+    # Load KB.json
     with open(req.kb_file, "r") as f:
         kb_data = json.load(f)
 
-    # Create ChromaDB collection if not exists
+    # Create Chroma client (new API)
     chroma_client = chromadb.Client(
-    persist_directory="/data/chroma_db",  # where DB will be stored
-    chroma_db_impl="duckdb+parquet"
-)
+        Settings(
+            persist_directory="/data/chroma_db",  # make sure this dir exists and writable
+            anonymized_telemetry=False
+        )
+    )
 
+    # Create or get collection
     try:
         kb_collection = chroma_client.get_collection("kb_collection")
     except:
         kb_collection = chroma_client.create_collection("kb_collection")
 
-    # Setup embedding function
+    # Embedding function
     embedding_func = embedding_functions.SentenceTransformerEmbeddingFunction(
         model_name="sentence-transformers/multi-qa-mpnet-base-dot-v1"
     )
@@ -63,7 +71,9 @@ async def setup_endpoint(req: SetupRequest):
             embedding_function=embedding_func
         )
 
+    # Persist to disk
     kb_collection.persist()
+
     return {"status": "KB embedded and stored successfully"}
 
 # -------------------------------