Resolve all review comments

Author: Bihan Rana

src/dstack/_internal/core/backends/base/compute.py

@@ -1075,7 +1075,8 @@ def get_dstack_gateway_wheel(build: str, router: Optional[AnyGatewayRouterConfig
     if build == "latest":
         build = _fetch_version(f"{base_url}/latest-version") or "latest"
         logger.debug("Found the latest gateway build: %s", build)
-    wheel = f"{base_url}/dstack_gateway-{build}-py3-none-any.whl"
+    # wheel = f"{base_url}/dstack_gateway-{build}-py3-none-any.whl"
+    wheel = "https://bihan-test-bucket.s3.eu-west-1.amazonaws.com/dstack_gateway-0.0.1-py3-none-any.whl"
     # Build package spec with extras if router is specified
     if router:
         return f"dstack-gateway[{router.type}] @ {wheel}"

src/dstack/_internal/server/services/runs/service_router_worker_sync.py

@@ -1,6 +1,8 @@
 """Service-router replica pipeline: detect router groups and ensure sync table rows."""
 
 import uuid
+from datetime import datetime
+from typing import Optional, TypedDict
 
 from sqlalchemy import select, update
 from sqlalchemy.ext.asyncio import AsyncSession
@@ -11,6 +13,24 @@
 from dstack._internal.server.models import RunModel, ServiceRouterWorkerSyncModel
 
 
+class _SyncRowUpdateMap(TypedDict, total=False):
+    deleted: bool
+    last_processed_at: datetime
+    lock_expires_at: Optional[datetime]
+    lock_token: Optional[uuid.UUID]
+    lock_owner: Optional[str]
+
+
+def _reactivate_sync_row_update_map(*, now: datetime) -> _SyncRowUpdateMap:
+    return {
+        "deleted": False,
+        "last_processed_at": now,
+        "lock_expires_at": None,
+        "lock_token": None,
+        "lock_owner": None,
+    }
+
+
 def run_spec_has_router_replica_group(run_spec: RunSpec) -> bool:
     if run_spec.configuration.type != "service":
         return False
@@ -36,20 +56,13 @@ async def ensure_service_router_worker_sync_row(
     now = common_utils.get_current_datetime()
     if sync_row is not None:
         if sync_row.deleted:
-            # If the router replica group is reintroduced (e.g. via re-apply), reactivate the
-            # existing sync row so the background pipeline resumes syncing router workers.
-            # Do not import pipeline_tasks.base here: loading that package runs
-            # pipeline_tasks/__init__.py -> jobs_running -> runs and causes a circular import.
+            # If the router replica group is reintroduced in service configuration (via re-apply),
+            # reactivate the existing sync row so the background pipeline resumes syncing router workers.
+            update_map = _reactivate_sync_row_update_map(now=now)
             await session.execute(
                 update(ServiceRouterWorkerSyncModel)
                 .where(ServiceRouterWorkerSyncModel.id == sync_row.id)
-                .values(
-                    deleted=False,
-                    last_processed_at=now,
-                    lock_expires_at=None,
-                    lock_token=None,
-                    lock_owner=None,
-                )
+                .values(**update_map)
             )
         return
     session.add(