commit

controllers/vectorizer.py

@@ -147,7 +147,9 @@ def vectorize(article):
     Returns:
     None
     """
-    docs = split_documents(pd.DataFrame(article))
+    df = pd.DataFrame(article)
+    df = df[['id','site','title','titleCN','contentCN','category','author','content','subtitle','publishDate','link','attachment','sentimentScore','sentimentLabel']]
+    docs = split_documents(df)
     documents_list = documents_to_list_of_dicts(docs)
     for doc in documents_list:
         upsert(doc, collection)