Add model options

app.py

@@ -96,6 +96,7 @@ def write_data(new_chat_name=current_chat):
     # 防止高频创建时组件尚未渲染完成，不影响正常写入
     if "frequency_penalty" + current_chat in st.session_state:
         st.session_state["paras"] = {
+            "model": st.session_state["model" + current_chat],
             "temperature": st.session_state["temperature" + current_chat],
             "top_p": st.session_state["top_p" + current_chat],
             "presence_penalty": st.session_state["presence_penalty" + current_chat],
@@ -152,6 +153,9 @@ with tap_set:
               key='context_level' + current_chat, help="表示每次会话中包含的历史对话次数，预设内容不计算在内。")
 
     st.markdown("模型参数：")
+    st.selectbox("Model", ["gpt-3.5-turbo", "gpt-3.5-turbo-16k", "gpt-4", "gpt-4-32k"], index=0, 
+                 help="[模型选择参考](https://platform.openai.com/docs/models)", 
+                 on_change=write_data, key='model' + current_chat)
     st.slider("Temperature", 0.0, 2.0, st.session_state["temperature" + current_chat + "default"], 0.1,
               help="""在0和2之间，应该使用什么样的采样温度？较高的值（如0.8）会使输出更随机，而较低的值（如0.2）则会使其更加集中和确定性。
               我们一般建议只更改这个参数或top_p参数中的一个，而不要同时更改两个。""",
@@ -211,6 +215,7 @@ with tap_input:
                                   "content": st.session_state['context_input' + current_chat]}]
                               + history_tem)
         paras_need_input = {
+            "model": st.session_state["model" + current_chat],
             "temperature": st.session_state["temperature" + current_chat],
             "top_p": st.session_state["top_p" + current_chat],
             "presence_penalty": st.session_state["presence_penalty" + current_chat],
@@ -222,7 +227,7 @@ with tap_input:
                     openai.api_key = apikey
                 else:
                     openai.api_key = st.secrets["apikey"]
-                r = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=history_need_input, stream=True,
+                r = openai.ChatCompletion.create(messages=history_need_input, stream=True,
                                                  **paras_need_input)
             except (FileNotFoundError, KeyError):
                 area_error.error("缺失 OpenAI API Key，请在复制项目后配置Secrets，或者在设置中进行临时配置。"