sync: Merge pull request #36 from AceDataCloud/fix/improve-polling-guidance

Synced from monorepo directory: suno

Author: acedatacloud-dev

core/utils.py

@@ -17,7 +17,16 @@ def _with_submission_guidance(
         "task_id": task_id,
         "poll_tool": poll_tool,
         "batch_poll_tool": batch_poll_tool,
-        "next_step": f'Call {poll_tool}(task_id="{task_id}") to poll until the task completes and the final media URLs are available.',
+        "polling_interval_seconds": 15,
+        "max_poll_attempts": 20,
+        "expected_wait_seconds": 300,
+        "next_step": (
+            f'Call {poll_tool}(task_id="{task_id}") to poll until the task completes '
+            f"and the final media URLs are available. "
+            f"IMPORTANT: Media generation typically takes 1-5 minutes. "
+            f"Wait at least 15 seconds between each poll. "
+            f"Keep polling for up to 20 attempts. Do NOT stop early — the task is still running."
+        ),
     }
     return payload
 
@@ -50,11 +59,17 @@ def _with_task_guidance(
             "batch_poll_tool": batch_poll_tool,
             "state": state,
             "is_complete": False,
-            "next_step": f'Task is NOT complete yet (state: "{state}"). '
-            f'IMPORTANT: Only state="complete" with success=true means the task is finished. '
-            f"Ignore any intermediate audio_url values (e.g. audiopipe.suno.ai URLs) — "
-            f"these are streaming previews, NOT final results. "
-            f'Call {poll_tool}(task_id="{task_id}") again after a few seconds.',
+            "polling_interval_seconds": 15,
+            "max_poll_attempts": 20,
+            "next_step": (
+                f'Task is NOT complete yet (state: "{state}"). '
+                f'IMPORTANT: Only state="complete" with success=true means the task is finished. '
+                f"Ignore any intermediate audio_url values — "
+                f"these are streaming previews, NOT final results. "
+                f'Wait 15 seconds, then call {poll_tool}(task_id="{task_id}") again. '
+                f"Media generation typically takes 1-5 minutes. "
+                f"Keep polling for up to 20 attempts. Do NOT stop early."
+            ),
         }
     return payload