Add exception on qdrant connection

Author: sudip-khanal

chatbotcore/database.py

@@ -31,7 +31,10 @@ class QdrantDatabase:
 
     def __post_init__(self):
         """Initialize database client"""
-        self.db_client = QdrantClient(host=self.host, port=self.port)
+        try:
+            self.db_client = QdrantClient(host=self.host, port=self.port)
+        except Exception as e:
+            logger.error(f"Failed to connect at {self.host}:{self.port}. Error: {e}")
 
     def _collection_exists(self, collection_name: str) -> bool:
         """Check if the collection in db already exists"""

content/tasks.py

@@ -3,8 +3,6 @@
 import requests
 from celery import shared_task
 from django.conf import settings
-from qdrant_client.http.exceptions import UnexpectedResponse
-from requests.exceptions import ConnectionError as RequestsConnectionError
 
 from chatbotcore.database import QdrantDatabase
 from chatbotcore.doc_loaders import LoaderFromText
@@ -41,18 +39,9 @@ def create_embedding_for_content_task(self, content_id):
         db.store_data(zip(response.json(), metadata))
         content.document_status = Content.DocumentStatus.ADDED_TO_VECTOR
 
-    except RequestsConnectionError as e:
-        # Qdrant connection failure
-        logger.error(f"Qdrant connection error: {str(e)}")
-        content.document_status = Content.DocumentStatus.FAILURE
-
-    except UnexpectedResponse as e:
-        # Qdrant server error
-        logger.error(f"Qdrant storage error [{e.status_code}]: {e.content}")
-        content.document_status = Content.DocumentStatus.FAILURE
-
     # NOTE: All exceptions have been handled with except
-    except Exception:
+    except Exception as e:
+        logger.error(f"Error:{str(e)}")
         content.document_status = Content.DocumentStatus.FAILURE
     content.save()