diff --git a/llama_stack/distribution/server/server.py b/llama_stack/distribution/server/server.py index f4a6c9559..8c1e41dc0 100644 --- a/llama_stack/distribution/server/server.py +++ b/llama_stack/distribution/server/server.py @@ -242,7 +242,7 @@ def main(): parser.add_argument( "--port", type=int, - default=int(os.getenv("LLAMA_STACK_PORT", 5000)), + default=int(os.getenv("LLAMASTACK_PORT", 5000)), help="Port to listen on", ) parser.add_argument( diff --git a/llama_stack/distribution/start_container.sh b/llama_stack/distribution/start_container.sh index 3b49a22f8..3b7b55b97 100755 --- a/llama_stack/distribution/start_container.sh +++ b/llama_stack/distribution/start_container.sh @@ -90,6 +90,6 @@ $DOCKER_BINARY run $DOCKER_OPTS -it \ $env_vars \ -v "$yaml_config:/app/config.yaml" \ $mounts \ - --env LLAMA_STACK_PORT=$port \ + --env LLAMASTACK_PORT=$port \ --entrypoint='["python", "-m", "llama_stack.distribution.server.server", "--yaml-config", "/app/config.yaml"]' \ $docker_image:$version_tag