apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-06-17 15:25:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

adding support for model aliases in archgw

Browse source
This commit is contained in:
Salman Paracha 2025-09-14 22:30:57 -07:00
parent 1e8c81d8f6
commit f13b420146
13 changed files with 1438 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

14
arch/tools/cli/config_generator.py
View file

@ -208,6 +208,20 @@ def validate_and_render_schema():
config_yaml["llm_providers"] = updated_llm_providers
# Validate model aliases if present
if "model_aliases" in config_yaml:
model_aliases = config_yaml["model_aliases"]
for alias_name, alias_config in model_aliases.items():
target = alias_config.get("target")
if not target:
raise Exception(
f"Model alias '{alias_name}' must have a target specified"
)
if target not in model_name_keys:
raise Exception(
f"Model alias '{alias_name}' targets '{target}' which is not defined as a model. Available models: {', '.join(sorted(model_name_keys))}"
)
arch_config_string = yaml.dump(config_yaml)
arch_llm_config_string = yaml.dump(config_yaml)