ofermend commited on
Commit
1388aa0
1 Parent(s): 2a89f81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,10 +22,10 @@ def launch_bot():
22
  'title': os.environ['title'],
23
  'description': os.environ['description'],
24
  'source_data_desc': os.environ['source_data_desc'],
25
- 'prompt_name': os.environ['prompt_name']
26
  })
27
  st.session_state.cfg = cfg
28
- st.session_state.vq = VectaraQuery(cfg.api_key, cfg.customer_id, cfg.corpus_ids)
29
 
30
  cfg = st.session_state.cfg
31
  vq = st.session_state.vq
 
22
  'title': os.environ['title'],
23
  'description': os.environ['description'],
24
  'source_data_desc': os.environ['source_data_desc'],
25
+ 'prompt_name': os.environ.get('prompt_name', None)
26
  })
27
  st.session_state.cfg = cfg
28
+ st.session_state.vq = VectaraQuery(cfg.api_key, cfg.customer_id, cfg.corpus_ids, cfg.prompt_name)
29
 
30
  cfg = st.session_state.cfg
31
  vq = st.session_state.vq