Update rag_engine.py

rag_engine.py

@@ -17,16 +17,23 @@ class RAGEngine:
         
         # Qdrant Cloud Configuration
         # Prioritize Env Vars, fallback to Hardcoded (User provided)
-        self.qdrant_url = os.environ.get("QDRANT_URL") or "https://abd29675-7fb9-4d95-8941-e6130b09bf7f.us-east4-0.gcp.cloud.qdrant.io"
-        self.qdrant_api_key = os.environ.get("QDRANT_API_KEY") or "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhY2Nlc3MiOiJtIn0.L0aAAAbxRypLfBeGCtFr2xX06iveGb76NrA3BPJQiNM"
+        env_qdrant_url = os.environ.get("QDRANT_URL")
+        print(f"DEBUG: QDRANT_URL from env: '{env_qdrant_url}'")
+        
+        self.qdrant_url = env_qdrant_url or "https://abd29675-7fb9-4d95-8941-e6130b09bf7f.us-east4-0.gcp.cloud.qdrant.io"
+        self.qdrant_api_key = os.environ.get("QDRANT_API_KEY") # Don't default key if using local URL
+        
+        if not self.qdrant_api_key and "qdrant.io" in self.qdrant_url:
+             self.qdrant_api_key = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhY2Nlc3MiOiJtIn0.L0aAAAbxRypLfBeGCtFr2xX06iveGb76NrA3BPJQiNM"
+
         self.collection_name = "phishing_knowledge"
         
-        if not self.qdrant_url or not self.qdrant_api_key:
-            print("⚠️ QDRANT_URL or QDRANT_API_KEY not set. RAG will not function correctly.")
+        if not self.qdrant_url:
+            print("⚠️ QDRANT_URL not set. RAG will not function correctly.")
             self.vector_store = None
             return
 
-        print(f"☁️ Connecting to Qdrant Cloud: {self.qdrant_url}...")
+        print(f"☁️ Connecting to Qdrant: {self.qdrant_url}...")
         
         # Initialize Qdrant Client
         self.client = QdrantClient(