Update knowledge_base.py

knowledge_base.py

@@ -49,17 +49,16 @@ def create_vectorstore(chunks):
     return db
 
 def load_vectorstore():
+    print("📂 Loading from:", CHROMA_DIR)
     embeddings = HuggingFaceEmbeddings(model_name=MODEL_NAME)
     db = Chroma(persist_directory=CHROMA_DIR, embedding_function=embeddings)
 
-    # === DEBUG BLOCK ===
+    # Debug block
     try:
-        data = db.get()
-        doc_count = len(data["documents"])
-        print(f"📦 Vectorstore Loaded: {doc_count} documents")
-        if doc_count > 0:
-            print(f"🧾 First doc snippet: {data['documents'][0][:100]}...")
+        docs = db.get()
+        print(f"✅ Loaded vectorstore with {len(docs['documents'])} docs")
+        print(f"🧾 First doc snippet: {docs['documents'][0][:100]}...")
     except Exception as e:
-        print(f"❌ Failed to load vectorstore: {e}")
+        print(f"❌ Vectorstore load error: {e}")
 
-    return db
+    return db