Ravi theja K commited on
Commit
c60cfa3
·
verified ·
1 Parent(s): 1457a2f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
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)