Update app/api/routes.py

app/api/routes.py

@@ -303,7 +303,6 @@
 
 
 
-
 
 
 # app/api/routes.py
@@ -320,14 +319,17 @@ from qdrant_client.http.models import Filter, FieldCondition, MatchValue
 from app.core.llm_engine import ask_gemini
 from app.core.mongo import conversations
 from qdrant_client import QdrantClient
-from app.core.config import QDRANT_URL
+from app.core.config import QDRANT_URL, QDRANT_API_KEY
 
 router = APIRouter()
 UPLOAD_DIR = "uploads"
 os.makedirs(UPLOAD_DIR, exist_ok=True)
 
 # Qdrant Client
-qdrant_client = QdrantClient(url=QDRANT_URL)
+qdrant_client = QdrantClient(
+    url=QDRANT_URL,
+    api_key=QDRANT_API_KEY,
+    check_compatibility=False)
 
 # In-memory (for fallback)
 chat_histories = defaultdict(list)
@@ -427,6 +429,41 @@ async def get_all_chats():
 # ---------------------------------------------------
 # ✅ Delete Chat (Qdrant + Mongo + File)
 # ---------------------------------------------------
+# @router.delete("/chat/{chat_id}")
+# async def delete_chat(chat_id: str):
+#     try:
+#         chat = conversations.find_one({"_id": ObjectId(chat_id)})
+#         if not chat:
+#             raise HTTPException(status_code=404, detail="Chat not found")
+
+#         doc_id = chat.get("doc_id")
+#         qdrant_collection = chat.get("qdrant_collection")
+
+#         # ✅ Delete embeddings from Qdrant
+#         try:
+#             qdrant_client.delete(
+#                 collection_name=qdrant_collection,
+#                 points_selector=Filter(
+#                     must=[FieldCondition(key="doc_id", match=MatchValue(value=doc_id))]
+#                 ),
+#             )
+#         except Exception as e:
+#             print(f"⚠️ Qdrant delete failed: {e}")
+
+#         # ✅ Delete uploaded file
+#         file_path = chat.get("file_path")
+#         if file_path and os.path.exists(file_path):
+#             os.remove(file_path)
+
+#         # ✅ Delete from MongoDB
+#         conversations.delete_one({"_id": ObjectId(chat_id)})
+
+#         return {"status": "success", "message": "Chat deleted successfully"}
+
+#     except Exception as e:
+#         raise HTTPException(status_code=500, detail=str(e))
+
+
 
 @router.delete("/chat/{chat_id}")
 async def delete_chat(chat_id: str):
@@ -486,14 +523,26 @@ async def query_pdf(doc_id: str = Form(...), question: str = Form(...)):
 
     # ✅ Vector Search
     question_vector = embedder.encode([question])[0].tolist()
-    hits = qdrant.search(
+    # hits = qdrant.search(
+    #     collection_name=COLLECTION_NAME,
+    #     query_vector=question_vector,
+    #     query_filter=Filter(
+    #         must=[FieldCondition(key="doc_id", match=MatchValue(value=doc_id))]
+    #     ),
+    #     limit=5,
+    # )
+
+
+    hits = qdrant_client.query_points(
         collection_name=COLLECTION_NAME,
-        query_vector=question_vector,
+        query=question_vector,
         query_filter=Filter(
             must=[FieldCondition(key="doc_id", match=MatchValue(value=doc_id))]
         ),
         limit=5,
-    )
+    ).points
+
+
     context = "\n".join([hit.payload["text"] for hit in hits])
     full_context = f"{structured_history}\n\n{context}"
 
@@ -521,4 +570,3 @@ async def get_conversation(doc_id: str):
     if not doc:
         return {"history": []}
     return {"history": doc.get("history", [])}
-