From 75f757f8072ad17ffba9e533d4b4e0549ad35f77 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E5=BE=B7=E5=AE=87?= <kvtodev@outlook.com>
Date: Thu, 24 Sep 2026 20:40:14 +0800
Subject: [PATCH] =?UTF-8?q?=E5=9B=BE=E7=94=9F=203D=20=E5=9C=A8=E7=9C=9F?=
 =?UTF-8?q?=E6=AD=A3=E6=8F=90=E4=BA=A4=E5=89=8D=E5=86=8D=E7=A1=AE=E8=AE=A4?=
 =?UTF-8?q?=E4=B8=80=E6=AC=A1=E5=89=A9=E4=BD=99=E9=A2=84=E7=AE=97?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- worker：`submit_and_checkpoint` 接收 provider 截止时间，图生 3D 在下载源图、上传给 provider 之后、真正 submit 之前再调一次 `ensure_submit_budget`
- 原因写进注释：解析源图这两步都可能耗时到把预算用光，而 submit 不可撤销且计费，宁可本次尝试判失败，也不要提交上去立刻被截止时间判失败、白花一次外部调用
- `ensure_submit_budget` 的说明补上两处调用点
---
 server-rs/crates/api-server/src/tripo3d/worker.rs | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/server-rs/crates/api-server/src/tripo3d/worker.rs b/server-rs/crates/api-server/src/tripo3d/worker.rs
index 24311a203..081f5129b 100644
--- a/server-rs/crates/api-server/src/tripo3d/worker.rs
+++ b/server-rs/crates/api-server/src/tripo3d/worker.rs
@@ -94,7 +94,7 @@ async fn run_model3d_job(
             // 大图、任务排在其他 job 之后），这时提交上去也只会立刻被 deadline 判失败，
             // 白花一次 provider 调用。只有还轮询得起才提交。
             ensure_submit_budget(provider_deadline)?;
-            submit_and_checkpoint(state, caller, job, &client, &request).await?
+            submit_and_checkpoint(state, caller, job, &client, &request, provider_deadline).await?
         }
     };
     let snapshot = poll_until_terminal(&client, &handle, provider_deadline).await?;
@@ -154,6 +154,9 @@ fn existing_checkpoint(job: &ExternalGenerationJobRecord) -> Option<String> {
 }
 
 /// 提交前的预算检查：本次尝试已经没有轮询预算时，不发起不可逆且计费的 provider submit。
+///
+/// 调用点有两处：进入提交流程之前，以及图生 3D 在下载 / 上传源图之后、真正 submit 之前
+/// （那两步都可能耗时到把预算用光）。
 fn ensure_submit_budget(provider_deadline: Instant) -> Result<(), AppError> {
     if Instant::now() >= provider_deadline {
         return Err(provider_deadline_error());
@@ -167,12 +170,17 @@ async fn submit_and_checkpoint(
     job: &ExternalGenerationJobRecord,
     client: &TripoProviderClient,
     request: &Model3dJobRequest,
+    provider_deadline: Instant,
 ) -> Result<TripoTaskHandle, AppError> {
     let handle = match request {
         Model3dJobRequest::TextToModel(request) => {
+            ensure_submit_budget(provider_deadline)?;
             client.submit_text_to_model(&request.generation).await
         }
         Model3dJobRequest::ImageToModel(request) => {
+            // 解析源图要先下载站内图片、再上传给 provider，这一步可能把剩下的预算用光。
+            // 提交不可撤销且计费，所以真正发起之前再确认一次：宁可本次尝试判失败，
+            // 也不要提交上去紧接着被 deadline 判失败，白花一次外部调用。
             let input = resolve_image_input(
                 state,
                 client,
@@ -180,6 +188,7 @@ async fn submit_and_checkpoint(
                 &request.source,
             )
             .await?;
+            ensure_submit_budget(provider_deadline)?;
             client
                 .submit_image_to_model(&input, &request.generation)
                 .await