minor fix

code/modules/vectorstore/store_manager.py

@@ -139,13 +139,15 @@ class VectorStoreManager:
         try:
             self.loaded_vector_db = self.vector_db._load_database(self.embedding_model)
         except Exception as e:
-            raise ValueError(f"Error loading database, check if it exists. if not run python -m modules.vectorstore.store_manager / Resteart the HF Space: {e}")
+            raise ValueError(
+                f"Error loading database, check if it exists. if not run python -m modules.vectorstore.store_manager / Resteart the HF Space: {e}"
+            )
             # print(f"Creating database")
             # self.create_database()
             # self.loaded_vector_db = self.vector_db._load_database(self.embedding_model)
         end_time = time.time()  # End time for loading database
         self.logger.info(
-            f"Time taken to load database {self.config['vectorstore']['db_option']} from Hugging Face: {end_time - start_time} seconds"
+            f"Time taken to load database {self.config['vectorstore']['db_option']}: {end_time - start_time} seconds"
         )
         self.logger.info("Loaded database")
         return self.loaded_vector_db
@@ -173,12 +175,16 @@ if __name__ == "__main__":
     vector_db = VectorStoreManager(config)
     if config["vectorstore"]["load_from_HF"]:
         if config["vectorstore"]["db_option"] in RETRIEVER_HF_PATHS:
-            vector_db.load_from_HF(HF_PATH = RETRIEVER_HF_PATHS[config["vectorstore"]["db_option"]])
+            vector_db.load_from_HF(
+                HF_PATH=RETRIEVER_HF_PATHS[config["vectorstore"]["db_option"]]
+            )
         else:
             # print(f"HF_PATH not available for {config['vectorstore']['db_option']}")
             # print("Creating database")
             # vector_db.create_database()
-            raise ValueError(f"HF_PATH not available for {config['vectorstore']['db_option']}")
+            raise ValueError(
+                f"HF_PATH not available for {config['vectorstore']['db_option']}"
+            )
     else:
         vector_db.create_database()
     print("Created database")