Skip to content

Commit dab1693

Browse files
committed
refactor(chore): read llm provider and model from dictionary
1 parent a877812 commit dab1693

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

webui.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -385,14 +385,14 @@ def create_ui(theme_name="Ocean"):
385385
with gr.TabItem("🔧 LLM Configuration", id=2):
386386
with gr.Group():
387387
llm_provider = gr.Dropdown(
388-
["anthropic", "openai", "deepseek", "gemini", "ollama", "azure_openai"],
388+
choices=[provider for provider,model in utils.model_names.items()],
389389
label="LLM Provider",
390390
value="openai",
391391
info="Select your preferred language model provider"
392392
)
393393
llm_model_name = gr.Dropdown(
394394
label="Model Name",
395-
choices=["gpt-4o", "gpt-4o-mini", "gpt-4", "gpt-3.5-turbo",],
395+
choices=utils.model_names['openai'],
396396
value="gpt-4o",
397397
interactive=True,
398398
allow_custom_value=True, # Allow users to input custom model names

0 commit comments

Comments
 (0)