rename hf_token name
Browse files
climateqa/engine/talk_to_data/query.py
CHANGED
|
@@ -41,10 +41,10 @@ async def execute_sql_query(sql_query: str) -> pd.DataFrame:
|
|
| 41 |
def _execute_query():
|
| 42 |
# Execute the query
|
| 43 |
con = duckdb.connect()
|
| 44 |
-
|
| 45 |
con.execute(f"""CREATE SECRET hf_token (
|
| 46 |
TYPE huggingface,
|
| 47 |
-
TOKEN '{
|
| 48 |
);""")
|
| 49 |
results = con.execute(sql_query).fetchdf()
|
| 50 |
# return fetched data
|
|
|
|
| 41 |
def _execute_query():
|
| 42 |
# Execute the query
|
| 43 |
con = duckdb.connect()
|
| 44 |
+
HF_TTD_TOKEN = os.getenv("HF_TTD_TOKEN")
|
| 45 |
con.execute(f"""CREATE SECRET hf_token (
|
| 46 |
TYPE huggingface,
|
| 47 |
+
TOKEN '{HF_TTD_TOKEN}'
|
| 48 |
);""")
|
| 49 |
results = con.execute(sql_query).fetchdf()
|
| 50 |
# return fetched data
|