summaryrefslogtreecommitdiffstats
path: root/gui/streamlit_chat_app.py
diff options
context:
space:
mode:
Diffstat (limited to 'gui/streamlit_chat_app.py')
-rw-r--r--gui/streamlit_chat_app.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/gui/streamlit_chat_app.py b/gui/streamlit_chat_app.py
index ed3474fd..3ced571a 100644
--- a/gui/streamlit_chat_app.py
+++ b/gui/streamlit_chat_app.py
@@ -7,7 +7,7 @@ import streamlit as st
from streamlit_chat import message
from query_methods import query, avail_query_methods
import pickle
-
+import openai_rev
conversations_file = "conversations.pkl"
@@ -71,9 +71,10 @@ if st.sidebar.button("New Conversation"):
st.session_state['current_conversation'] = {'user_inputs': [], 'generated_responses': []}
st.session_state['input_field_key'] += 1
-
+print(openai_rev.Provider.__methods__.keys())
st.session_state['query_method'] = st.sidebar.selectbox(
"Select API:",
+
options=avail_query_methods.keys(),
index=0
)