Small refactor for run_with_pty

This commit is contained in:
Ashwin Bharambe 2025-01-28 09:32:33 -08:00
parent 8332ea23ad
commit aee6237685
3 changed files with 13 additions and 10 deletions

View file

@ -21,7 +21,7 @@ from llama_stack.distribution.distribution import get_provider_registry
from llama_stack.distribution.utils.config_dirs import BUILDS_BASE_DIR
from llama_stack.distribution.utils.exec import run_command, run_with_pty, run_with_win
from llama_stack.distribution.utils.exec import run_command, run_with_pty
from llama_stack.providers.datatypes import Api
log = logging.getLogger(__name__)
@ -157,9 +157,7 @@ def build_image(
is_terminal = sys.stdin.isatty()
if is_terminal:
return_code = (
run_with_win(args) if sys.platform.startswith("win") else run_with_pty(args)
)
return_code = run_with_pty(args)
else:
return_code = run_command(args)