mirror of
https://github.com/katanemo/plano.git
synced 2026-06-17 15:25:17 +02:00
add validation for unique llm names
This commit is contained in:
parent
eaf90be387
commit
bd43b1793e
2 changed files with 11 additions and 6 deletions
|
|
@ -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(
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue