John Graham Reynolds commited on
Commit
0e29746
·
1 Parent(s): 8bb66b9

rename token env var

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -4,13 +4,13 @@ from langchain_huggingface import HuggingFaceEmbeddings
4
  from langchain_databricks.vectorstores import DatabricksVectorSearch
5
 
6
  DATABRICKS_HOST = os.environ.get("DATABRICKS_HOST")
7
- DATABRICKS_API_TOKEN = os.environ.get("DATABRICKS_API_TOKEN")
8
  VS_ENDPOINT_NAME = os.environ.get("VS_ENDPOINT_NAME")
9
  VS_INDEX_NAME = os.environ.get("VS_INDEX_NAME")
10
 
11
  if DATABRICKS_HOST is None:
12
  raise ValueError("DATABRICKS_HOST environment variable must be set")
13
- if DATABRICKS_API_TOKEN is None:
14
  raise ValueError("DATABRICKS_API_TOKEN environment variable must be set")
15
 
16
  TITLE = "VUMC Chatbot"
@@ -32,6 +32,8 @@ st.markdown(DESCRIPTION)
32
  st.markdown("\n")
33
  st.write(DATABRICKS_HOST)
34
  st.markdown("\n")
 
 
35
  st.write(VS_ENDPOINT_NAME)
36
  st.markdown("\n")
37
 
 
4
  from langchain_databricks.vectorstores import DatabricksVectorSearch
5
 
6
  DATABRICKS_HOST = os.environ.get("DATABRICKS_HOST")
7
+ DATABRICKS_TOKEN = os.environ.get("DATABRICKS_TOKEN")
8
  VS_ENDPOINT_NAME = os.environ.get("VS_ENDPOINT_NAME")
9
  VS_INDEX_NAME = os.environ.get("VS_INDEX_NAME")
10
 
11
  if DATABRICKS_HOST is None:
12
  raise ValueError("DATABRICKS_HOST environment variable must be set")
13
+ if DATABRICKS_TOKEN is None:
14
  raise ValueError("DATABRICKS_API_TOKEN environment variable must be set")
15
 
16
  TITLE = "VUMC Chatbot"
 
32
  st.markdown("\n")
33
  st.write(DATABRICKS_HOST)
34
  st.markdown("\n")
35
+ st.write(DATABRICKS_TOKEN)
36
+ st.markdown("\n")
37
  st.write(VS_ENDPOINT_NAME)
38
  st.markdown("\n")
39