ndurner commited on
Commit
18ccfa1
·
1 Parent(s): 7dab3e1

add gpt-5-pro, clean up model list

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -350,7 +350,7 @@ async def bot(message, history, history_openai_format, oai_key, system_prompt, t
350
  if log_to_console:
351
  print(f"br_prompt: {str(history_openai_format)}")
352
 
353
- reasoner_models = {"o1", "o1-pro", "o1-2024-12-17", "o3-mini", "o4-mini", "o3", "o3-pro", "gpt-5", "gpt-5-mini"}
354
  reasoner = model in reasoner_models
355
 
356
  assistant_msgs = []
@@ -722,7 +722,7 @@ with gr.Blocks(delete_cache=(86400, 86400)) as demo:
722
 
723
  oai_key = gr.Textbox(label="OpenAI API Key", elem_id="oai_key", value=os.environ.get("OPENAI_API_KEY"))
724
  model = gr.Dropdown(label="Model", value="gpt-5-mini", allow_custom_value=True, elem_id="model",
725
- choices=["gpt-5", "gpt-5-mini", "gpt-5-chat-latest", "gpt-4o", "gpt-4.1", "o3", "o3-pro", "o4-mini", "chatgpt-4o-latest", "gpt-4o-mini", "gpt-4-turbo", "whisper", "gpt-image-1"])
726
  reasoning_effort = gr.Dropdown(label="Reasoning Effort", value="medium", choices=["low", "medium", "high"], elem_id="reasoning_effort")
727
  verbosity = gr.Dropdown(label="Verbosity (GPT-5)", value="medium", choices=["low", "medium", "high"], elem_id="verbosity")
728
  system_prompt = gr.TextArea("You are a helpful yet diligent AI assistant. Answer faithfully and factually correct. Respond with 'I do not know' if uncertain.", label="System/Developer Prompt", lines=3, max_lines=250, elem_id="system_prompt")
 
350
  if log_to_console:
351
  print(f"br_prompt: {str(history_openai_format)}")
352
 
353
+ reasoner_models = {"o4-mini", "o3", "o3-pro", "gpt-5", "gpt-5-mini", "gpt-5-pro"}
354
  reasoner = model in reasoner_models
355
 
356
  assistant_msgs = []
 
722
 
723
  oai_key = gr.Textbox(label="OpenAI API Key", elem_id="oai_key", value=os.environ.get("OPENAI_API_KEY"))
724
  model = gr.Dropdown(label="Model", value="gpt-5-mini", allow_custom_value=True, elem_id="model",
725
+ choices=["gpt-5", "gpt-5-mini", "gpt-5-chat-latest", "gpt-5-pro", "gpt-4o", "gpt-4.1", "o3", "o3-pro", "o4-mini", "chatgpt-4o-latest", "gpt-4o-mini", "gpt-4-turbo", "whisper", "gpt-image-1"])
726
  reasoning_effort = gr.Dropdown(label="Reasoning Effort", value="medium", choices=["low", "medium", "high"], elem_id="reasoning_effort")
727
  verbosity = gr.Dropdown(label="Verbosity (GPT-5)", value="medium", choices=["low", "medium", "high"], elem_id="verbosity")
728
  system_prompt = gr.TextArea("You are a helpful yet diligent AI assistant. Answer faithfully and factually correct. Respond with 'I do not know' if uncertain.", label="System/Developer Prompt", lines=3, max_lines=250, elem_id="system_prompt")