Ravi theja K
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -6,9 +6,17 @@ from langchain_community.utilities import SQLDatabase
|
|
6 |
from langchain.llms import HuggingFaceHub
|
7 |
from langchain.chains import create_sql_query_chain
|
8 |
|
|
|
|
|
|
|
9 |
@st.cache_resource(show_spinner="Connecting...")
|
10 |
def getSession():
|
11 |
pars = SnowflakeLoginOptions("test_conn")
|
|
|
|
|
|
|
|
|
|
|
12 |
pars["database"] = "SNOWFLAKE_SAMPLE_DATA"
|
13 |
pars["schema"] = "TPCH_SF1"
|
14 |
session = Session.builder.configs(pars).create()
|
@@ -17,7 +25,7 @@ def getSession():
|
|
17 |
+ f"/{pars['database']}/{pars['schema']}"
|
18 |
+ f"?warehouse={pars['warehouse']}&role={pars['role']}")
|
19 |
db = SQLDatabase.from_uri(url)
|
20 |
-
|
21 |
os.environ["HUGGINGFACEHUB_API_TOKEN"]
|
22 |
llm = HuggingFaceHub(repo_id= "ravithejakandi/Sf-Arctic-Demo-Enu")
|
23 |
chain = create_sql_query_chain(llm, db)
|
|
|
6 |
from langchain.llms import HuggingFaceHub
|
7 |
from langchain.chains import create_sql_query_chain
|
8 |
|
9 |
+
|
10 |
+
|
11 |
+
|
12 |
@st.cache_resource(show_spinner="Connecting...")
|
13 |
def getSession():
|
14 |
pars = SnowflakeLoginOptions("test_conn")
|
15 |
+
pars["account"] ="ap20346.ap-south-1"
|
16 |
+
pars["user"] = "Vassist"
|
17 |
+
pars["password"]= "Vassist@123"
|
18 |
+
pars["role"]= "ACCOUNTADMIN"
|
19 |
+
pars["warehouse"] = "COMPUTE_WH"
|
20 |
pars["database"] = "SNOWFLAKE_SAMPLE_DATA"
|
21 |
pars["schema"] = "TPCH_SF1"
|
22 |
session = Session.builder.configs(pars).create()
|
|
|
25 |
+ f"/{pars['database']}/{pars['schema']}"
|
26 |
+ f"?warehouse={pars['warehouse']}&role={pars['role']}")
|
27 |
db = SQLDatabase.from_uri(url)
|
28 |
+
|
29 |
os.environ["HUGGINGFACEHUB_API_TOKEN"]
|
30 |
llm = HuggingFaceHub(repo_id= "ravithejakandi/Sf-Arctic-Demo-Enu")
|
31 |
chain = create_sql_query_chain(llm, db)
|