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

fix listener type names to match Rust enum (model/prompt not model_listener/prompt_listener)

Browse source
This commit is contained in:
Adil Hafeez 2026-03-12 14:38:41 -07:00
parent efa677683a
commit 4e290fb715
No known key found for this signature in database
GPG key ID: 9B18EF7691369645
3 changed files with 9 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

2
cli/planoai/core.py
View file

@ -165,7 +165,7 @@ def _resolve_cli_agent_endpoint(plano_config_yaml: dict) -> tuple[str, int]:
if isinstance(listeners, list): if isinstance(listeners, list):
for listener in listeners: for listener in listeners:
if listener.get("type") in ["model", "model_listener"]: if listener.get("type") == "model":
host = listener.get("host") or listener.get("address") or "0.0.0.0" host = listener.get("host") or listener.get("address") or "0.0.0.0"
port = listener.get("port", 12000) port = listener.get("port", 12000)
return host, port return host, port

4
cli/planoai/utils.py
View file

@ -89,7 +89,7 @@ def convert_legacy_listeners(
) -> tuple[list, dict | None, dict | None]: ) -> tuple[list, dict | None, dict | None]:
llm_gateway_listener = { llm_gateway_listener = {
"name": "egress_traffic", "name": "egress_traffic",
"type": "model_listener", "type": "model",
"port": 12000, "port": 12000,
"address": "0.0.0.0", "address": "0.0.0.0",
"timeout": "30s", "timeout": "30s",
@ -98,7 +98,7 @@ def convert_legacy_listeners(
prompt_gateway_listener = { prompt_gateway_listener = {
"name": "ingress_traffic", "name": "ingress_traffic",
"type": "prompt_listener", "type": "prompt",
"port": 10000, "port": 10000,
"address": "0.0.0.0", "address": "0.0.0.0",
"timeout": "30s", "timeout": "30s",

12
cli/test/test_config_generator.py
View file

@ -376,7 +376,7 @@ def test_convert_legacy_llm_providers():
assert updated_providers == [ assert updated_providers == [
{ {
"name": "egress_traffic", "name": "egress_traffic",
"type": "model_listener", "type": "model",
"port": 12000, "port": 12000,
"address": "0.0.0.0", "address": "0.0.0.0",
"timeout": "30s", "timeout": "30s",
@ -384,7 +384,7 @@ def test_convert_legacy_llm_providers():
}, },
{ {
"name": "ingress_traffic", "name": "ingress_traffic",
"type": "prompt_listener", "type": "prompt",
"port": 10000, "port": 10000,
"address": "0.0.0.0", "address": "0.0.0.0",
"timeout": "30s", "timeout": "30s",
@ -400,7 +400,7 @@ def test_convert_legacy_llm_providers():
}, },
], ],
"name": "egress_traffic", "name": "egress_traffic",
"type": "model_listener", "type": "model",
"port": 12000, "port": 12000,
"timeout": "30s", "timeout": "30s",
} }
@ -410,7 +410,7 @@ def test_convert_legacy_llm_providers():
"name": "ingress_traffic", "name": "ingress_traffic",
"port": 10000, "port": 10000,
"timeout": "30s", "timeout": "30s",
"type": "prompt_listener", "type": "prompt",
} }
@ -449,7 +449,7 @@ def test_convert_legacy_llm_providers_no_prompt_gateway():
"name": "egress_traffic", "name": "egress_traffic",
"port": 12000, "port": 12000,
"timeout": "30s", "timeout": "30s",
"type": "model_listener", "type": "model",
} }
] ]
assert llm_gateway == { assert llm_gateway == {
@ -461,7 +461,7 @@ def test_convert_legacy_llm_providers_no_prompt_gateway():
}, },
], ],
"name": "egress_traffic", "name": "egress_traffic",
"type": "model_listener", "type": "model",
"port": 12000, "port": 12000,
"timeout": "30s", "timeout": "30s",
} }