timeki commited on
Commit
d78ea76
·
1 Parent(s): 11ab5fb

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
- HF_TOKEN = os.getenv("HF_TOKEN")
45
  con.execute(f"""CREATE SECRET hf_token (
46
  TYPE huggingface,
47
- TOKEN '{HF_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