diff --git a/pyproject.toml b/pyproject.toml
index e147980ba2..a414a9e234 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -10,6 +10,9 @@ requires-python = ">=3.11"
 dependencies = [
     "huanzhi-utils",
     "scikit-learn",
+    "datasets",
+    "python-dotenv",
+    "openai",
     "torch",
     "setuptools",
     "deepspeed==0.16.3",