Zekun Wu commited on
Commit
5f064a8
·
1 Parent(s): 0a48925
Files changed (1) hide show
  1. pages/1_Injection.py +1 -1
pages/1_Injection.py CHANGED
@@ -54,10 +54,10 @@ else:
54
  # Model selection and configuration
55
  model_type = st.sidebar.radio("Select the type of agent", ('GPTAgent', 'AzureAgent','Claude3Agent'))
56
  st.session_state.api_key = st.sidebar.text_input("API Key", type="password", value=st.session_state.api_key)
57
- st.session_state.endpoint_url = st.sidebar.text_input("Endpoint URL", value=st.session_state.endpoint_url)
58
  st.session_state.deployment_name = st.sidebar.text_input("Model Name", value=st.session_state.deployment_name)
59
 
60
  if model_type == 'GPTAgent' or model_type == 'AzureAgent':
 
61
  api_version = '2024-02-15-preview' if model_type == 'GPTAgent' else ''
62
  st.session_state.temperature = st.sidebar.slider("Temperature", 0.0, 1.0, st.session_state.temperature, 0.01)
63
  st.session_state.max_tokens = st.sidebar.number_input("Max Tokens", 1, 1000, st.session_state.max_tokens)
 
54
  # Model selection and configuration
55
  model_type = st.sidebar.radio("Select the type of agent", ('GPTAgent', 'AzureAgent','Claude3Agent'))
56
  st.session_state.api_key = st.sidebar.text_input("API Key", type="password", value=st.session_state.api_key)
 
57
  st.session_state.deployment_name = st.sidebar.text_input("Model Name", value=st.session_state.deployment_name)
58
 
59
  if model_type == 'GPTAgent' or model_type == 'AzureAgent':
60
+ st.session_state.endpoint_url = st.sidebar.text_input("Endpoint URL", value=st.session_state.endpoint_url)
61
  api_version = '2024-02-15-preview' if model_type == 'GPTAgent' else ''
62
  st.session_state.temperature = st.sidebar.slider("Temperature", 0.0, 1.0, st.session_state.temperature, 0.01)
63
  st.session_state.max_tokens = st.sidebar.number_input("Max Tokens", 1, 1000, st.session_state.max_tokens)