Update chains/arxiv_chains.py

chains/arxiv_chains.py

@@ -73,7 +73,7 @@ class VectorSQLRetrieveCustomOutputParser(VectorSQLOutputParser):
         start = text.upper().find("SELECT")
         if start >= 0:
             end = text.upper().find("FROM")
-            text = text.replace(text[start + len("SELECT") + 1 : end - 1], "title, abstract, authors, pubdate, categories, id")
+            text = text.replace(text[start + len("SELECT") + 1 : end - 1], ", ".join(self.must_have_columns))
         return super().parse(text)
 
 class ArXivStuffDocumentChain(StuffDocumentsChain):