Adding Loading time for semantic_search.py

backend/semantic_search.py

@@ -1,10 +1,31 @@
+import time
+import logging
 from qdrant_haystack import QdrantDocumentStore
 from haystack.nodes import EmbeddingRetriever
 from pathlib import Path
 
+# Setting up the logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# Start the timer for loading the QdrantDocumentStore
+start_time = time.perf_counter()
+
 proj_dir = Path(__file__).parents[1]
 qd_document_store = QdrantDocumentStore(path=str(proj_dir/'Qdrant'), index='RAGDemo')
+
+# Log the time taken to load the QdrantDocumentStore
+document_store_loading_time = time.perf_counter() - start_time
+logger.info(f"Time taken to load QdrantDocumentStore: {document_store_loading_time:.6f} seconds")
+
+# Start the timer for loading the EmbeddingRetriever
+start_time = time.perf_counter()
+
 qd_retriever = EmbeddingRetriever(document_store=qd_document_store,
                                   embedding_model="BAAI/bge-base-en-v1.5",
                                   model_format="sentence_transformers",
                                   use_gpu=True)
+
+# Log the time taken to load the EmbeddingRetriever
+retriever_loading_time = time.perf_counter() - start_time
+logger.info(f"Time taken to load EmbeddingRetriever: {retriever_loading_time:.6f} seconds")