diff --git a/openllm-python/pyproject.toml b/openllm-python/pyproject.toml
index 254a90db..656cce58 100644
--- a/openllm-python/pyproject.toml
+++ b/openllm-python/pyproject.toml
@@ -46,6 +46,7 @@ dependencies = [
     "optimum>=1.12.0",
     "accelerate",
     "ghapi",
+    "build[virtualenv]<1",
     "click>=8.1.3",
     "cuda-python;platform_system!=\"Darwin\"",
     "bitsandbytes<0.42",
diff --git a/tools/dependencies.py b/tools/dependencies.py
index f0ed6e09..c3b66f35 100755
--- a/tools/dependencies.py
+++ b/tools/dependencies.py
@@ -7,20 +7,16 @@ import typing as t
 
 import inflection
 import tomlkit
-
 from ghapi.all import GhApi
 
-
 if t.TYPE_CHECKING:
-  from tomlkit.items import Array
-  from tomlkit.items import Table
+  from tomlkit.items import Array, Table
 
 ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 sys.path.insert(0, os.path.join(ROOT, 'openllm-python', 'src'))
 
 import openllm
 
-
 _OWNER, _REPO = 'bentoml', 'openllm'
 
 
@@ -157,6 +153,7 @@ _BASE_DEPENDENCIES = [
   Dependencies(name='optimum', lower_constraint='1.12.0'),
   Dependencies(name='accelerate'),
   Dependencies(name='ghapi'),
+  Dependencies(name='build', upper_constraint='1', extensions=['virtualenv']),
   Dependencies(name='click', lower_constraint='8.1.3'),
   Dependencies(name='cuda-python', platform=('Darwin', 'ne')),
   Dependencies(name='bitsandbytes', upper_constraint='0.42'),  # 0.41 works with CUDA 11.8