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)
|