Add Remote Python Sidecar

AI-Hypercomputer · Jan 14, 2025 · 39e0b04 · 39e0b04
1 parent 072f7de
commit 39e0b04
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/src/xpk/commands/workload.py b/src/xpk/commands/workload.py
@@ -254,6 +254,26 @@
               volumeMounts:
               - mountPath: /tmp
                 name: shared-tmp
+            initContainers:
+            - args:
+              name: remote-python-sidecar
+              image: {args.remote_python_sidecar_image}
+              imagePullPolicy: Always
+              command:
+              - "bash"
+              - "-c"
+              - |
+                python start_remote_python.py
+              securityContext:
+                privileged: true
+              volumeMounts:
+              - mountPath: /tmp
+                name: shared-tmp
+              ports:
+                - containerPort: 50051
+                env:
+                - name: GRPC_SERVER_ADDRESS
+                  value: "0.0.0.0:50051"
             nodeSelector:
               {accelerator_label}
               {machine_label}

diff --git a/src/xpk/parser/workload.py b/src/xpk/parser/workload.py
@@ -503,6 +503,14 @@ def add_shared_workload_create_optional_arguments(args_parsers):
             ' event or deletion request.Defaults to 30 seconds.'
         ),
     )
+    custom_parser.add_argument(
+        '--remote-python-sidecar-image',
+        type=str,
+        default='TempImageLocation',
+        help=(
+            'Remote Python sidecar image to use.'
+        ),
+    )
     custom_parser.add_argument(
         '--enable-debug-logs',
         action='store_true',