Update find_similar_news.py

find_similar_news.py

@@ -20,6 +20,7 @@ def load_sentence_transformer():
         logger.warning('Exiting load_sentence_transformer')
     except Exception as e:
         logger.warning(f"load_sentence_transformer error: {e}")
+        raise
         
     return sent_model, ce_model
 
@@ -51,7 +52,8 @@ def get_milvus_collection():
         print(f"Loaded collection")
         logger.warning('Exiting get_milvus_collection()')
     except Exception as e:
-            logger.warning(f"get_milvus_collection() error: {e}")
+        logger.warning(f"get_milvus_collection() error: {e}")
+        raise
         
     return collection
 
@@ -92,5 +94,7 @@ def find_similar_news(text, search_vec, collection, vectorizer, sent_model, ce_m
         logger.warning('Successfully generated HTML output')
         logger.warning('Exiting find_similar_news')
     except Exception as e:
-            logger.warning(f"find_similar_news() error: {e}")
+        logger.warning(f"find_similar_news() error: {e}")
+        raise
+        
     return html_output