From 38473f8ee06ac5b6f2074d8bc1f84f88ee7a9da0 Mon Sep 17 00:00:00 2001
From: Bingxu Chen <Bingxu.Chen@amd.com>
Date: Sat, 14 Feb 2026 16:07:26 +0800
Subject: [PATCH] [AMD] Fix sgl-model-gateway Build Errors in ROCm Docker
 Release  (#18836)

---
 .../workflows/release-docker-amd-nightly.yml  |  2 +-
 .../release-docker-amd-rocm720-nightly.yml    |  2 +-
 sgl-model-gateway/Cargo.toml                  | 22 +++++++++----------
 3 files changed, 13 insertions(+), 13 deletions(-)

diff --git a/.github/workflows/release-docker-amd-nightly.yml b/.github/workflows/release-docker-amd-nightly.yml
index 81454fb3e..b07d473e9 100644
--- a/.github/workflows/release-docker-amd-nightly.yml
+++ b/.github/workflows/release-docker-amd-nightly.yml
@@ -78,7 +78,7 @@ jobs:
 
           tag=v${version}-${rocm_tag}
 
-          docker build . -f docker/rocm.Dockerfile --build-arg BUILD_TYPE=${{ matrix.build_type }} --build-arg GPU_ARCH=${{ matrix.gpu_arch }} --build-arg ENABLE_MORI=1 --build-arg NIC_BACKEND=ainic --build-arg SETUPTOOLS_SCM_PRETEND_VERSION=${pretend_version} -t rocm/sgl-dev:${tag}-${{ env.DATE }} --no-cache
+          docker build . -f docker/rocm.Dockerfile --build-arg SGL_BRANCH=${{ github.ref_name }} --build-arg BUILD_TYPE=${{ matrix.build_type }} --build-arg GPU_ARCH=${{ matrix.gpu_arch }} --build-arg ENABLE_MORI=1 --build-arg NIC_BACKEND=ainic --build-arg SETUPTOOLS_SCM_PRETEND_VERSION=${pretend_version} -t rocm/sgl-dev:${tag}-${{ env.DATE }} --no-cache
           docker push rocm/sgl-dev:${tag}-${{ env.DATE }}
 
   # Temporarily disable docker cache seeding until performant storage is in place
diff --git a/.github/workflows/release-docker-amd-rocm720-nightly.yml b/.github/workflows/release-docker-amd-rocm720-nightly.yml
index cf9ce21a2..b3da54bae 100644
--- a/.github/workflows/release-docker-amd-rocm720-nightly.yml
+++ b/.github/workflows/release-docker-amd-rocm720-nightly.yml
@@ -78,5 +78,5 @@ jobs:
 
           tag=v${version}-${rocm_tag}
 
-          docker build . -f docker/rocm720.Dockerfile --build-arg BUILD_TYPE=${{ matrix.build_type }} --build-arg GPU_ARCH=${{ matrix.gpu_arch }} --build-arg ENABLE_MORI=1 --build-arg NIC_BACKEND=ainic --build-arg SETUPTOOLS_SCM_PRETEND_VERSION=${pretend_version} -t rocm/sgl-dev:${tag}-${{ env.DATE }} --no-cache
+          docker build . -f docker/rocm720.Dockerfile --build-arg SGL_BRANCH=${{ github.ref_name }} --build-arg BUILD_TYPE=${{ matrix.build_type }} --build-arg GPU_ARCH=${{ matrix.gpu_arch }} --build-arg ENABLE_MORI=1 --build-arg NIC_BACKEND=ainic --build-arg SETUPTOOLS_SCM_PRETEND_VERSION=${pretend_version} -t rocm/sgl-dev:${tag}-${{ env.DATE }} --no-cache
           docker push rocm/sgl-dev:${tag}-${{ env.DATE }}
diff --git a/sgl-model-gateway/Cargo.toml b/sgl-model-gateway/Cargo.toml
index 4578a5bb1..b05a7c9fc 100644
--- a/sgl-model-gateway/Cargo.toml
+++ b/sgl-model-gateway/Cargo.toml
@@ -75,16 +75,16 @@ url = "2.5.4"
 validator = { version = "0.20.0", features = ["derive"] }
 tokio-stream = { version = "0.1", features = ["sync"] }
 anyhow = "1.0"
-reasoning-parser = "1.0.0"
-openai-protocol = { version = "1.0.0", features = ["axum"] }
-tool-parser = "1.0.0"
-llm-tokenizer = "1.0.0"
-smg-auth = "1.0.0"
-wfaas = "1.0.0"
-data-connector = "1.0.0"
-smg-mcp = "1.0.0"
-smg-wasm = "1.0.0"
-smg-mesh = "1.0.0"
+reasoning-parser = "~1.0.0"
+openai-protocol = { version = "~1.0.0", features = ["axum"] }
+tool-parser = "~1.0.0"
+llm-tokenizer = "~1.0.0"
+smg-auth = "~1.0.0"
+wfaas = "~1.0.0"
+data-connector = "~1.0.0"
+smg-mcp = "~1.0.0"
+smg-wasm = "~1.0.0"
+smg-mesh = "~1.0.0"
 rustls = { version = "0.23", default-features = false, features = ["ring", "std"] }
 rustls-pemfile = "2.2"
 openssl = "0.10.73"
@@ -107,7 +107,7 @@ openmetrics-parser = "0.4.4"
 arc-swap = "1.7.1"
 
 # gRPC and Protobuf dependencies
-smg-grpc-client = "1.0.0"
+smg-grpc-client = "~1.0.0"
 tonic = { version = "0.14.2", features = ["gzip", "transport"] }
 prost = "0.14.1"
 prost-types = "0.14.1"