Make useAlternativeEndpoints configurable

Author: NickCao

packages/jumpstarter/jumpstarter/client/client.py

@@ -26,7 +26,7 @@ async def client_from_channel(
     stack: ExitStack,
     allow: list[str],
     unsafe: bool,
-    use_alternative_endpoints: bool = True,
+    use_alternative_endpoints: bool = False,
 ) -> DriverClient:
     topo = defaultdict(list)
     last_seen = {}

packages/jumpstarter/jumpstarter/client/lease.py

@@ -41,6 +41,7 @@ class Lease(AbstractContextManager, AbstractAsyncContextManager):
     controller: jumpstarter_pb2_grpc.ControllerServiceStub = field(init=False)
     tls_config: TLSConfigV1Alpha1 = field(default_factory=TLSConfigV1Alpha1)
     grpc_options: dict[str, Any] = field(default_factory=dict)
+    use_alternative_endpoints: bool = False
 
     def __post_init__(self):
         if hasattr(super(), "__post_init__"):
@@ -184,7 +185,14 @@ async def _monitor():
     @asynccontextmanager
     async def connect_async(self, stack):
         async with self.serve_unix_async() as path:
-            async with client_from_path(path, self.portal, stack, allow=self.allow, unsafe=self.unsafe) as client:
+            async with client_from_path(
+                path,
+                self.portal,
+                stack,
+                allow=self.allow,
+                unsafe=self.unsafe,
+                use_alternative_endpoints=self.use_alternative_endpoints,
+            ) as client:
                 yield client
 
     @contextmanager

packages/jumpstarter/jumpstarter/config/client.py

@@ -50,6 +50,8 @@ class ClientConfigV1Alpha1(BaseModel):
     token: str
     grpcOptions: dict[str, str | int] | None = Field(default_factory=dict)
 
+    useAlternativeEndpoints: bool = False
+
     drivers: ClientConfigV1Alpha1Drivers
 
     async def channel(self):
@@ -172,6 +174,7 @@ async def lease_async(
             release=release_lease,
             tls_config=self.tls,
             grpc_options=self.grpcOptions,
+            use_alternative_endpoints=self.useAlternativeEndpoints,
         ) as lease:
             yield lease