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

fix tests

Browse source
This commit is contained in:
Adil Hafeez 2025-09-10 17:10:15 -07:00
parent 33ef00f997
commit 9120e5b7ba
No known key found for this signature in database
GPG key ID: 9B18EF7691369645
6 changed files with 56 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -113,15 +113,21 @@ def validate_and_render_schema():
"port": ingress_traffic.get("port", 10000),
"address": ingress_traffic.get("address", "0.0.0.0"),
"timeout": ingress_traffic.get("timeout", "30s"),
"protocol": "openai",
}
config_yaml["listeners"].append(prompt_gateway_listener)
if egress_traffic:
llm_providers = []
if config_yaml.get("llm_providers"):
llm_providers = config_yaml["llm_providers"]
del config_yaml["llm_providers"]
llm_gateway_listener = {
"name": "egress_traffic",
"port": egress_traffic.get("port", 12000),
"address": egress_traffic.get("address", "0.0.0.0"),
"timeout": egress_traffic.get("timeout", "30s"),
"llm_providers": config_yaml.get("llm_providers", []),
"llm_providers": llm_providers,
"protocol": "openai",
}
config_yaml["listeners"].append(llm_gateway_listener)
@ -237,7 +243,9 @@ def validate_and_render_schema():
}
)
config_yaml["llm_providers"] = updated_llm_providers
for listener in config_yaml["listeners"]:
if listener.get("name") == "egress_traffic":
listener["llm_providers"] = updated_llm_providers
arch_config_string = yaml.dump(config_yaml)
arch_llm_config_string = yaml.dump(config_yaml)
@ -279,7 +287,7 @@ def validate_and_render_schema():
"arch_config": arch_config_string,
"arch_llm_config": arch_llm_config_string,
"arch_clusters": inferred_clusters,
"arch_llm_providers": config_yaml["llm_providers"],
"arch_llm_providers": updated_llm_providers,
"arch_tracing": arch_tracing,
"local_llms": llms_with_endpoint,
"agent_orchestrator": agent_orchestrator,