diff --git a/scripts/cluster/cross-node-file-e2e.mjs b/scripts/cluster/cross-node-file-e2e.mjs index 4e9093f..c1d8b0e 100755 --- a/scripts/cluster/cross-node-file-e2e.mjs +++ b/scripts/cluster/cross-node-file-e2e.mjs @@ -56,6 +56,7 @@ const account = Buffer.from(process.env.E2E_ACCOUNT_B64 || '', 'base64').toStrin const password = Buffer.from(process.env.E2E_PASSWORD_B64 || '', 'base64').toString('utf8') const model = process.env.AI_GATEWAY_E2E_IMAGE_MODEL || 'openai:gpt-image-1' const imageSize = process.env.AI_GATEWAY_E2E_IMAGE_SIZE || '2048x2048' +const platformModelId = process.env.AI_GATEWAY_E2E_PLATFORM_MODEL_ID || '' assert(account && password, 'E2E login credentials are unavailable') assert(/^\d+x\d+$/.test(imageSize), 'E2E image size is invalid') const [imageWidth, imageHeight] = imageSize.split('x').map(Number) @@ -137,7 +138,8 @@ try { (item) => [item?.modelName, item?.modelAlias].includes(model) && Array.isArray(item?.modelType) && - item.modelType.includes('image_edit'), + item.modelType.includes('image_edit') && + (!platformModelId || item.id === platformModelId), ) assert(targetModels.length > 0, `no assignable image_edit platform model found for ${model}`) const resourceMap = new Map() diff --git a/scripts/cluster/run-cross-node-file-e2e.sh b/scripts/cluster/run-cross-node-file-e2e.sh index 8c9ed90..1901a8b 100755 --- a/scripts/cluster/run-cross-node-file-e2e.sh +++ b/scripts/cluster/run-cross-node-file-e2e.sh @@ -70,6 +70,8 @@ chmod 0600 "$credentials_file" "${AI_GATEWAY_E2E_IMAGE_MODEL:-openai:gpt-image-1}" printf 'AI_GATEWAY_E2E_IMAGE_SIZE=%s\n' \ "${AI_GATEWAY_E2E_IMAGE_SIZE:-2048x2048}" + printf 'AI_GATEWAY_E2E_PLATFORM_MODEL_ID=%s\n' \ + "${AI_GATEWAY_E2E_PLATFORM_MODEL_ID:-}" } >"$credentials_file" cluster_scp "$credentials_file" "$CLUSTER_NINGBO_HOST:/root/easyai-cross-node-e2e.env" cluster_scp "$script_dir/cross-node-file-e2e.mjs" \