Pass auth_options to op.update

Author: r4victor

pyproject.toml

@@ -190,7 +190,7 @@ oci = [
     "dstack[server]",
 ]
 nebius = [
-    "nebius>=0.3.0,<0.4; python_version >= '3.10'",
+    "nebius>=0.3.4,<0.4; python_version >= '3.10'",
     "dstack[server]",
 ]
 all = [

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

@@ -379,7 +379,12 @@ def _wait_for_instance(sdk: SDK, op: SDKOperation[Operation]) -> None:
             op.status(),
         )
         time.sleep(WAIT_FOR_INSTANCE_UPDATE_INTERVAL)
-        resources.LOOP.await_(op.update(per_retry_timeout=resources.REQUEST_TIMEOUT))
+        resources.LOOP.await_(
+            op.update(
+                per_retry_timeout=resources.REQUEST_TIMEOUT,
+                auth_options=resources.REQUEST_AUTH_OPTIONS,
+            )
+        )
 
 
 def _supported_instances(offer: InstanceOffer) -> bool:

src/dstack/_internal/core/backends/nebius/resources.py

@@ -117,7 +117,9 @@ def wait_for_operation(
         if time.monotonic() + interval > deadline:
             raise TimeoutError(f"Operation {op.id} wait timeout")
         time.sleep(interval)
-        LOOP.await_(op.update(per_retry_timeout=REQUEST_TIMEOUT))
+        LOOP.await_(
+            op.update(per_retry_timeout=REQUEST_TIMEOUT, auth_options=REQUEST_AUTH_OPTIONS)
+        )
 
 
 def get_region_to_project_id_map(