Optimize dstack-gateway installation time (#2146)

Add the `dstack[gateway]` extra, so that the
gateway doesn't install `dstack[server]`.
`dstack[gateway]` still has some unnecessary
dependencies from BASE_DEPS, but the current
optimization is enough for now.

Also increase the gateway installation timeout
just in case.

Author: jvstme

gateway/pyproject.toml

@@ -11,7 +11,7 @@ requires-python = ">=3.10"
 dynamic = ["version"]
 dependencies = [
     # release builds of dstack-gateway depend on a PyPI version of dstack instead
-    "dstack[server] @ https://github.com/dstackai/dstack/archive/refs/heads/master.zip",
+    "dstack[gateway] @ https://github.com/dstackai/dstack/archive/refs/heads/master.zip",
 ]
 
 [tool.setuptools.package-data]

setup.py

@@ -51,10 +51,17 @@ def get_long_description():
     "gpuhunt>=0.0.17,<0.1.0",
 ]
 
-SERVER_DEPS = [
+GATEWAY_AND_SERVER_COMMON_DEPS = [
     "fastapi",
     "starlette>=0.26.0",
     "uvicorn",
+    "aiorwlock",
+    "aiocache",
+    "httpx",
+    "jinja2",
+]
+
+SERVER_DEPS = GATEWAY_AND_SERVER_COMMON_DEPS + [
     "watchfiles",
     "sqlalchemy[asyncio]>=2.0.0",
     "sqlalchemy_utils>=0.40.0",
@@ -66,12 +73,8 @@ def get_long_description():
     "sentry-sdk[fastapi]",
     "alembic-postgresql-enum",
     "asyncpg",
-    "aiorwlock",
-    "aiocache",
     "cachetools",
-    "httpx",
     "python-json-logger",
-    "jinja2",
     "grpcio>=1.50",  # indirect
 ]
 
@@ -138,6 +141,7 @@ def get_long_description():
     install_requires=BASE_DEPS,
     extras_require={
         "all": ALL_DEPS,
+        "gateway": GATEWAY_AND_SERVER_COMMON_DEPS,
         "server": SERVER_DEPS,
         "aws": SERVER_DEPS + AWS_DEPS,
         "azure": SERVER_DEPS + AZURE_DEPS,

src/dstack/_internal/server/services/gateways/__init__.py

@@ -78,7 +78,7 @@
 
 GATEWAY_CONNECT_ATTEMPTS = 30
 GATEWAY_CONNECT_DELAY = 10
-GATEWAY_CONFIGURE_ATTEMPTS = 40
+GATEWAY_CONFIGURE_ATTEMPTS = 50
 GATEWAY_CONFIGURE_DELAY = 3