Update rag_engine.py

rag_engine.py

@@ -1,10 +1,10 @@
 import os
 import glob
 from langchain_community.document_loaders import DirectoryLoader, TextLoader, PyPDFLoader
-from langchain_community.vectorstores import Qdrant
+from langchain_qdrant import Qdrant
 from langchain_huggingface import HuggingFaceEmbeddings
 from langchain_text_splitters import RecursiveCharacterTextSplitter
-from qdrant_client import QdrantClient
+from qdrant_client import QdrantClient, models
 
 class RAGEngine:
     def __init__(self, knowledge_base_dir="./knowledge_base"):
@@ -41,13 +41,23 @@ class RAGEngine:
         
         # Check if collection exists/is empty and build if needed
         try:
-            count = self.client.count(collection_name=self.collection_name).count
-            if count == 0:
+            if not self.client.collection_exists(self.collection_name):
+                print(f"⚠️ Collection '{self.collection_name}' not found. Creating...")
+                self.client.create_collection(
+                    collection_name=self.collection_name,
+                    vectors_config=models.VectorParams(size=384, distance=models.Distance.COSINE)
+                )
+                print(f"✅ Collection '{self.collection_name}' created!")
                 self._build_index()
             else:
-                print(f"✅ Qdrant Collection '{self.collection_name}' ready with {count} vectors.")
+                count = self.client.count(collection_name=self.collection_name).count
+                if count == 0:
+                    self._build_index()
+                else:
+                    print(f"✅ Qdrant Collection '{self.collection_name}' ready with {count} vectors.")
         except Exception as e:
-            print(f"⚠️ Collection check failed (might not exist): {e}")
+            print(f"⚠️ Collection check/creation failed: {e}")
+            # Try to build anyway, maybe wrapper handles it
             self._build_index()
 
     def _build_index(self):