apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-04-25 08:46:24 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

make arch-router cluster optional (#518)

Browse source
This commit is contained in:
Adil Hafeez 2025-07-08 00:33:40 -07:00 • committed by GitHub
parent a212dd79da
commit 147908ba7e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
9 changed files with 44 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -140,14 +140,20 @@ def validate_and_render_schema():
llm_provider["protocol"] = protocol
llms_with_endpoint.append(llm_provider)
if (
len(llms_with_usage) > 0
and config_yaml.get("routing", {}).get("model", None) == None
):
llms_with_usage_names = ", ".join(llms_with_usage)
raise Exception(
f"LLMs with usage found ({llms_with_usage_names}), please provide model in routing section in your arch_config.yaml file"
)
if len(llms_with_usage) > 0:
routing_llm_provider = config_yaml.get("routing", {}).get("llm_provider", None)
if routing_llm_provider and routing_llm_provider not in llm_provider_name_set:
raise Exception(
f"Routing llm_provider {routing_llm_provider} is not defined in llm_providers"
)
if routing_llm_provider is None and "arch-router" not in llm_provider_name_set:
updated_llm_providers.append(
{
"name": "arch-router",
"provider_interface": "arch",
"model": config_yaml.get("routing", {}).get("model", "Arch-Router"),
}
)
config_yaml["llm_providers"] = updated_llm_providers