log upon successful connection to db

Author: dminnear-rh

vector_db/elastic_provider.py

@@ -1,10 +1,13 @@
+import logging
 from typing import List
 
 from langchain_core.documents import Document
 from langchain_elasticsearch.vectorstores import ElasticsearchStore
 
 from vector_db.db_provider import DBProvider
 
+logger = logging.getLogger(__name__)
+
 
 class ElasticProvider(DBProvider):
     """
@@ -28,6 +31,7 @@ class ElasticProvider(DBProvider):
 
     def __init__(self, url: str, password: str, index: str, user: str):
         super().__init__()
+
         self.db = ElasticsearchStore(
             embedding=self.embeddings,
             es_url=url,
@@ -36,6 +40,8 @@ def __init__(self, url: str, password: str, index: str, user: str):
             index_name=index,
         )
 
+        logger.info("Connected to Elasticsearch at %s (index: %s)", url, index)
+
     def add_documents(self, docs: List[Document]) -> None:
         """
         Add documents to the Elasticsearch index.

vector_db/pgvector_provider.py

@@ -1,10 +1,14 @@
+import logging
 from typing import List
+from urllib.parse import urlparse
 
 from langchain_core.documents import Document
 from langchain_postgres import PGVector
 
 from vector_db.db_provider import DBProvider
 
+logger = logging.getLogger(__name__)
+
 
 class PGVectorProvider(DBProvider):
     """
@@ -27,12 +31,23 @@ class PGVectorProvider(DBProvider):
 
     def __init__(self, url: str, collection_name: str):
         super().__init__()
+
         self.db = PGVector(
             connection=url,
             collection_name=collection_name,
             embeddings=self.embeddings,
         )
 
+        parsed = urlparse(url)
+        postgres_location = (
+            f"{parsed.hostname}:{parsed.port or 5432}/{parsed.path.lstrip('/')}"
+        )
+        logger.info(
+            "Connected to PGVector at %s (collection: %s)",
+            postgres_location,
+            collection_name,
+        )
+
     def add_documents(self, docs: List[Document]) -> None:
         """
         Add a list of documents to the pgvector-backed vector store.

vector_db/qdrant_provider.py

@@ -1,3 +1,4 @@
+import logging
 from typing import List, Optional
 
 from langchain_core.documents import Document
@@ -7,6 +8,8 @@
 
 from vector_db.db_provider import DBProvider
 
+logger = logging.getLogger(__name__)
+
 
 class QdrantProvider(DBProvider):
     """
@@ -31,6 +34,7 @@ class QdrantProvider(DBProvider):
     def __init__(self, url: str, collection: str, api_key: Optional[str] = None):
         super().__init__()
         self.collection = collection
+        self.url = url
 
         self.client = QdrantClient(
             url=url,
@@ -46,11 +50,17 @@ def __init__(self, url: str, collection: str, api_key: Optional[str] = None):
             embedding=self.embeddings,
         )
 
+        logger.info(
+            "Connected to Qdrant instance at %s (collection: %s)",
+            self.url,
+            self.collection,
+        )
+
     def _collection_exists(self) -> bool:
         return self.client.collection_exists(self.collection)
 
     def _create_collection(self) -> None:
-        vector_size = self.embeddings.embed_query("test").__len__()
+        vector_size = len(self.embeddings.embed_query("test"))
         self.client.recreate_collection(
             collection_name=self.collection,
             vectors_config=VectorParams(size=vector_size, distance=Distance.COSINE),

vector_db/redis_provider.py

@@ -41,6 +41,7 @@ def __init__(self, url: str, index: str, schema: str):
         try:
             self.redis_client = redis.from_url(self.url)
             self.redis_client.ping()
+            logger.info("Connected to Redis instance at %s", self.url)
         except Exception:
             logger.exception("Failed to connect to Redis at %s", self.url)
             raise

vector_db/sqlserver_provider.py

@@ -59,6 +59,14 @@ def __init__(
         self.connection_string = self._build_connection_string(self.database)
 
         self._ensure_database_exists()
+
+        logger.info(
+            "Connected to SQL Server at %s:%s, database: %s",
+            self.host,
+            self.port,
+            self.database,
+        )
+
         self.db = SQLServer_VectorStore(
             connection_string=self.connection_string,
             embedding_function=self.embeddings,