Add sglang optional dependencies to pyproject.toml

Author: mukesh reddy

pyproject.toml

@@ -41,6 +41,34 @@ backend = [
     "vllm==0.15.1 ; sys_platform == 'linux'",
 ]
 
+# SGLang training environment (main env - NO sglang here, just training deps)
+# SGLang server runs in COMPLETELY SEPARATE venv (just: pip install sglang[srt])
+# Communication between envs is via HTTP (localhost), not Python imports
+sglang = [
+    "peft>=0.14.0",
+    "hf-xet>=1.1.0",
+    "bitsandbytes>=0.45.2",
+    "unsloth==2025.12.9",
+    "unsloth-zoo==2025.12.7",
+    "torch>=2.8.0",
+    "torchao==0.14.1",
+    "accelerate==1.7.0",
+    "awscli>=1.38.1",
+    "setuptools>=78.1.0",
+    "wandb==0.23.1",
+    "transformers>=4.55.2,<=4.57.3",
+    "duckdb>=1.0.0",
+    "pyarrow>=15.0.0",
+    "trl==0.20.0",
+    "nbclient>=0.10.1",
+    "pytest>=8.4.1",
+    "nbmake>=1.5.5",
+    "gql<4",
+    "aiohttp>=3.9.0",
+]
+# NOTE: SGLang server venv is created separately with JUST: pip install "sglang[srt]"
+# Do NOT install ART in the server venv - they communicate via HTTP only
+
 langgraph = [
     "langchain-core>=0.3.51",
     "langgraph>=0.6.2",
@@ -145,15 +173,17 @@ allowed-unresolved-imports = [
     "uvicorn.**",
     "vllm.**",
     "wandb.**",
+    # sglang deps
+    "sglang.**",
+    "flashinfer.**",
+    "flashinfer_python.**",
     # langgraph deps
     "langchain_core.**",
     "langchain_openai.**",
     "langgraph.**",
     # plotting deps
     "matplotlib.**",
     "seaborn.**",
-    # megatron deps
-    "megatron.**",
 ]
 
 [dependency-groups]