Update find_similar_news.py

find_similar_news.py

@@ -43,10 +43,10 @@ def get_milvus_collection():
     logger.warning('Exiting get_milvus_collection()')
     return collection
 
-def find_similar_news(text: str, collection, vectorizer, sent_model, ce_model, top_n: int=5):
+def find_similar_news(search_vec, collection, vectorizer, sent_model, ce_model, top_n: int=5):
     logger.warning('Entering find_similar_news')
     search_params = {"metric_type": "IP"}
-    search_vec = vectorizer.vectorize_(text)
+    # search_vec = vectorizer.vectorize_(text)
     logger.warning('Querying Milvus for most similar results')
     results = collection.search([search_vec],
                                 anns_field='article_embed', # annotations field specified in the schema definition