add validation for unique llm names

This commit is contained in:
Adil Hafeez 2025-05-22 23:37:57 -07:00
parent eaf90be387
commit bd43b1793e
No known key found for this signature in database
GPG key ID: 9B18EF7691369645
2 changed files with 11 additions and 6 deletions

View file

@ -80,7 +80,17 @@ def validate_and_render_schema():
llms_with_endpoint = []
updated_llm_providers = []
llm_provider_name_set = set()
for llm_provider in config_yaml["llm_providers"]:
if llm_provider.get("name") in llm_provider_name_set:
raise Exception(
f"Duplicate llm_provider name {llm_provider.get('name')}, please provide unique name for each llm_provider"
)
if llm_provider.get("name") is None:
raise Exception(
f"llm_provider name is required, please provide name for llm_provider"
)
llm_provider_name_set.add(llm_provider.get("name"))
provider = None
if llm_provider.get("provider") and llm_provider.get("provider_interface"):
raise Exception(

View file

@ -17,17 +17,12 @@ llm_providers:
model: cotran2/llama-4-epoch
base_url: http://34.46.85.85:8000/v1
- name: gpt-4o
provider_interface: openai
access_key: $OPENAI_API_KEY
model: gpt-4o
default: true
- name: gpt-4o
provider_interface: openai
access_key: $OPENAI_API_KEY
model: gpt-4o
usage: Generating original content such as scripts, articles, or creative materials.
default: true
- name: o4-mini
provider_interface: openai