207207export LAMBDA_REGION LAMBDA_INSTANCE_TYPE LAMBDA_SSH_KEY_NAME TOOL_NAME RUN_ID
208208
209209LAUNCH_RESP=" $DISPATCH_OUT_DIR /${TOOL_NAME} -${RUN_ID} -launch-response.json"
210+ LAMBDA_AUTH_HEADER=$( printf ' %s:' " $LAMBDA_API_KEY " | base64 | tr -d ' \n' )
210211
211212set +e
212213HTTP_CODE=$( curl -sS \
213214 -o " $LAUNCH_RESP " -w " %{http_code}" \
214215 -X POST " https://cloud.lambdalabs.com/api/v1/instance-operations/launch" \
215- -u " ${LAMBDA_API_KEY} : " \
216+ -H " Authorization: Basic $LAMBDA_AUTH_HEADER " \
216217 -H " Content-Type: application/json" \
217218 --data-binary @" $PAYLOAD_FILE " )
218219CURL_RC=$?
@@ -259,7 +260,7 @@ for i in $(seq 1 60); do
259260 set +e
260261 STATUS_RESP=$( curl -sS \
261262 -X GET " https://cloud.lambdalabs.com/api/v1/instances/${INSTANCE_ID} " \
262- -u " ${LAMBDA_API_KEY} : " )
263+ -H " Authorization: Basic $LAMBDA_AUTH_HEADER " )
263264 set -e
264265 INSTANCE_IP=$( python3 -c "
265266import json,sys
@@ -334,7 +335,7 @@ print(json.dumps({
334335 "status_push_prefix": "$STATUS_PUSH_PREFIX ",
335336 "ssh_command": "ssh -i <ssh-private-key> ubuntu@$INSTANCE_IP ",
336337 "monitor_command": "ssh -i <ssh-private-key> ubuntu@$INSTANCE_IP cat ${MOUNT_PATH} /${TOOL_NAME} /${RUN_ID} /STATUS",
337- "terminate_command": "operator-side: curl -sS -X POST -u '<lambda-api-key>: ' -H 'Content-Type: application/json' -d '{\"instance_ids\":[\"$INSTANCE_ID \"]}' https://cloud.lambdalabs.com/api/v1/instance-operations/terminate",
338+ "terminate_command": "operator-side: set LAMBDA_AUTH_HEADER from LAMBDA_API_KEY, then curl -sS -X POST -H 'Authorization: Basic $LAMBDA_AUTH_HEADER ' -H 'Content-Type: application/json' -d '{\"instance_ids\":[\"$INSTANCE_ID \"]}' https://cloud.lambdalabs.com/api/v1/instance-operations/terminate",
338339}, indent=2))
339340PY
340341
@@ -358,7 +359,8 @@ Tail logs:
358359 "tail -f ${MOUNT_PATH} /${TOOL_NAME} /${RUN_ID} /logs/boot.log"
359360
360361Terminate (no preserve, Lambda has no stop/hibernate):
361- curl -sS -X POST -u '${LAMBDA_API_KEY} :' \\
362+ curl -sS -X POST \\
363+ -H "Authorization: Basic $LAMBDA_AUTH_HEADER " \\
362364 -H 'Content-Type: application/json' \\
363365 -d '{"instance_ids":["$INSTANCE_ID "]}' \\
364366 https://cloud.lambdalabs.com/api/v1/instance-operations/terminate
0 commit comments