mirror of
https://github.com/katanemo/plano.git
synced 2026-06-17 15:25:17 +02:00
adding support for model aliases in archgw
This commit is contained in:
parent
1e8c81d8f6
commit
f13b420146
13 changed files with 1438 additions and 8 deletions
|
|
@ -7,6 +7,7 @@ properties:
|
|||
- v0.1
|
||||
- v0.1.0
|
||||
- 0.1-beta
|
||||
- 0.2.0
|
||||
listeners:
|
||||
type: object
|
||||
additionalProperties: false
|
||||
|
|
@ -102,6 +103,19 @@ properties:
|
|||
additionalProperties: false
|
||||
required:
|
||||
- model
|
||||
|
||||
model_aliases:
|
||||
type: object
|
||||
patternProperties:
|
||||
"^.*$":
|
||||
type: object
|
||||
properties:
|
||||
target:
|
||||
type: string
|
||||
additionalProperties: false
|
||||
required:
|
||||
- target
|
||||
|
||||
overrides:
|
||||
type: object
|
||||
properties:
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue