Update knowledge_base.py

knowledge_base.py

@@ -51,14 +51,14 @@ def create_vectorstore(chunks):
 def load_vectorstore():
     embeddings = HuggingFaceEmbeddings(model_name=MODEL_NAME)
     db = Chroma(persist_directory=CHROMA_DIR, embedding_function=embeddings)
-    
+
+    # Debug block
     try:
-        sample = db.get()["documents"][:5]
-        print("✅ Sample documents from vectorstore:")
-        for i, s in enumerate(sample):
-            print(f"[{i+1}] {s[:100]}...")  # print first 100 chars
+        docs = db.get()
+        print("📦 Loaded vectorstore:")
+        print(f"- Total docs: {len(docs['documents'])}")
+        print(f"- First doc: {docs['documents'][0][:100]}...")
     except Exception as e:
-        print(f"❌ Error loading documents: {e}")
-    
-    return db
+        print(f"❌ Failed to fetch documents: {e}")
 
+    return db