mpsk commited on
Commit
e4853cf
β€’
1 Parent(s): eb820e1

update to experimental

Browse files
Files changed (2) hide show
  1. app.py +6 -4
  2. requirements.txt +2 -1
app.py CHANGED
@@ -17,11 +17,12 @@ from langchain.prompts.prompt import PromptTemplate
17
  from langchain.prompts import PromptTemplate, ChatPromptTemplate, \
18
  SystemMessagePromptTemplate, HumanMessagePromptTemplate
19
  from sqlalchemy import create_engine, MetaData
20
- from langchain.chains.sql_database.base import SQLDatabaseChain
21
- from langchain.chains.sql_database.parser import VectorSQLRetrieveAllOutputParser
22
  from langchain.chains import LLMChain
23
- from langchain.sql_database import SQLDatabase
24
- from langchain.retrievers import SQLDatabaseChainRetriever
 
 
 
25
 
26
  from chains.arxiv_chains import ArXivQAwithSourcesChain, ArXivStuffDocumentChain
27
  from callbacks.arxiv_callbacks import ChatDataSelfSearchCallBackHandler, \
@@ -29,6 +30,7 @@ from callbacks.arxiv_callbacks import ChatDataSelfSearchCallBackHandler, \
29
  ChatDataSQLAskCallBackHandler
30
  from prompts.arxiv_prompt import combine_prompt_template, _myscale_prompt
31
 
 
32
  st.set_page_config(page_title="ChatData")
33
 
34
  st.header("ChatData")
 
17
  from langchain.prompts import PromptTemplate, ChatPromptTemplate, \
18
  SystemMessagePromptTemplate, HumanMessagePromptTemplate
19
  from sqlalchemy import create_engine, MetaData
 
 
20
  from langchain.chains import LLMChain
21
+
22
+ from langchain_experimental.utilities.sql_database import SQLDatabase
23
+ from langchain_experimental.retrievers.sql_database import SQLDatabaseChainRetriever
24
+ from langchain_experimental.sql.base import SQLDatabaseChain
25
+ from langchain_experimental.sql.parser import VectorSQLRetrieveAllOutputParser
26
 
27
  from chains.arxiv_chains import ArXivQAwithSourcesChain, ArXivStuffDocumentChain
28
  from callbacks.arxiv_callbacks import ChatDataSelfSearchCallBackHandler, \
 
30
  ChatDataSQLAskCallBackHandler
31
  from prompts.arxiv_prompt import combine_prompt_template, _myscale_prompt
32
 
33
+
34
  st.set_page_config(page_title="ChatData")
35
 
36
  st.header("ChatData")
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
- langchain @ git+https://github.com/myscale/langchain.git@preview
 
2
  InstructorEmbedding
3
  pandas
4
  sentence_transformers
 
1
+ langchain @ git+https://github.com/myscale/langchain.git#egg=langchain&subdirectory=libs/langchain
2
+ langchain-experimental @ git+https://github.com/myscale/langchain.git#egg=langchain-experimental&subdirectory=libs/experimental
3
  InstructorEmbedding
4
  pandas
5
  sentence_transformers