Update app.py

app.py

@@ -22,8 +22,8 @@ from langchain.chains import LLMChain
 from langchain_experimental.utilities.sql_database import SQLDatabase
 from langchain_experimental.retrievers.sql_database import SQLDatabaseChainRetriever
 from langchain_experimental.sql.base import SQLDatabaseChain
-from langchain_experimental.sql.parser import VectorSQLRetrieveAllOutputParser
 
+from chains.arxiv_chains import VectorSQLRetrieveCustomOutputParser
 from chains.arxiv_chains import ArXivQAwithSourcesChain, ArXivStuffDocumentChain
 from callbacks.arxiv_callbacks import ChatDataSelfSearchCallBackHandler, \
     ChatDataSelfAskCallBackHandler, ChatDataSQLSearchCallBackHandler, \
@@ -155,7 +155,7 @@ def build_retriever():
             template=_myscale_prompt,
         )
 
-        output_parser = VectorSQLRetrieveAllOutputParser.from_embeddings(
+        output_parser = VectorSQLRetrieveCustomOutputParser.from_embeddings(
             model=embeddings)
         sql_query_chain = SQLDatabaseChain.from_llm(
             llm=OpenAI(openai_api_key=OPENAI_API_KEY, temperature=0),