fix: Correct WAV conversion to accept EITHER task_id OR audio_id

Per Suno API documentation: 'You can provide either taskId or audioId
to identify the source track'

Changes:
- Update convert_to_wav() signature: callback_url required, task_id and audio_id optional
- Validate that at least one ID (task_id or audio_id) is provided
- Build payload dynamically based on which ID(s) are provided
- Update MCP tool definition to reflect optional parameters
- Update tool handler to support flexible ID parameters
- Add comprehensive docstring with both usage examples

This resolves the 'record does not exist' error by allowing users to
provide just the audio_id (track ID) without requiring the task_id.

Recommended usage: Provide audio_id for precise track conversion,
especially when generation task has multiple tracks.

Author: CodeKeanu

server.py

@@ -150,20 +150,24 @@ async def handle_list_tools() -> list[Tool]:
         ),
         Tool(
             name="convert_to_wav",
-            description="Convert a generated audio track to WAV format. Use the track ID (audioId) from a previously generated music track. Returns a task ID that can be used to check conversion status.",
+            description="Convert a generated audio track to WAV format. Provide EITHER task_id OR audio_id (or both). Using audio_id is recommended for precision when a generation task has multiple tracks. Returns a conversion task ID that can be used to check conversion status.",
             inputSchema={
                 "type": "object",
                 "properties": {
-                    "audio_id": {
-                        "type": "string",
-                        "description": "Track ID (audioId) of the generated music to convert to WAV format"
-                    },
                     "callback_url": {
                         "type": "string",
                         "description": "Webhook URL for conversion completion notification (e.g., 'https://example.com/webhook')"
+                    },
+                    "task_id": {
+                        "type": "string",
+                        "description": "Original generation task ID (taskId) from generate_music - converts all tracks from that generation (optional if audio_id provided)"
+                    },
+                    "audio_id": {
+                        "type": "string",
+                        "description": "Track ID (audioId) of the specific generated music track to convert - more precise, recommended (optional if task_id provided)"
                     }
                 },
-                "required": ["audio_id", "callback_url"]
+                "required": ["callback_url"]
             }
         ),
         Tool(
@@ -403,16 +407,22 @@ async def handle_call_tool(name: str, arguments: dict | None) -> list[TextConten
 
         elif name == "convert_to_wav":
             # Extract arguments
-            audio_id = arguments.get("audio_id")
             callback_url = arguments.get("callback_url")
+            task_id = arguments.get("task_id")
+            audio_id = arguments.get("audio_id")
 
-            if not audio_id:
-                raise ValueError("audio_id is required")
             if not callback_url:
                 raise ValueError("callback_url is required")
 
-            # Convert to WAV
-            result = await suno_client.convert_to_wav(audio_id, callback_url)
+            if not task_id and not audio_id:
+                raise ValueError("Either task_id or audio_id must be provided")
+
+            # Convert to WAV (pass whichever IDs were provided)
+            result = await suno_client.convert_to_wav(
+                callback_url=callback_url,
+                task_id=task_id,
+                audio_id=audio_id
+            )
 
             # Format response
             response_text = "WAV Conversion Started!\n\n"

suno_client.py

@@ -240,34 +240,60 @@ async def get_credits(self) -> Dict[str, Any]:
 
     async def convert_to_wav(
         self,
-        audio_id: str,
-        callback_url: str
+        callback_url: str,
+        task_id: str = None,
+        audio_id: str = None
     ) -> Dict[str, Any]:
         """
         Convert a generated audio track to WAV format.
 
+        Per Suno API docs: You can provide EITHER taskId OR audioId to identify the source track.
+        Providing audioId ensures the exact track is converted (recommended when a task has multiple tracks).
+
         Args:
-            audio_id: Track ID (audioId) of the generated music to convert
-            callback_url: Webhook URL for conversion completion notification
+            callback_url: Webhook URL for conversion completion notification (required)
+            task_id: Original generation task ID (taskId) from generate_music (optional)
+            audio_id: Track ID (audioId) of the specific track to convert (optional)
 
         Returns:
             Dictionary containing WAV conversion task information with taskId
 
         Raises:
             SunoAPIError: If the API request fails
-            ValueError: If required parameters are missing or invalid
+            ValueError: If neither task_id nor audio_id is provided, or if callback_url is missing
+
+        Example:
+            # Option 1: Convert using audio_id (recommended - more precise)
+            music = await client.generate_music(prompt="Epic soundtrack", wait_audio=True)
+            track_id = music['data']['sunoData'][0]['id']
+            conversion = await client.convert_to_wav(
+                callback_url="https://example.com/webhook",
+                audio_id=track_id
+            )
+
+            # Option 2: Convert using task_id (converts all tracks from that generation)
+            generation_task_id = music['data']['taskId']
+            conversion = await client.convert_to_wav(
+                callback_url="https://example.com/webhook",
+                task_id=generation_task_id
+            )
         """
-        if not audio_id:
-            raise ValueError("audio_id is required for WAV conversion")
         if not callback_url:
             raise ValueError("callback_url is required for WAV conversion")
 
-        # Build payload with required parameters
+        if not task_id and not audio_id:
+            raise ValueError("Either task_id or audio_id must be provided for WAV conversion")
+
+        # Build payload - include whichever ID was provided
         payload: Dict[str, Any] = {
-            "audioId": audio_id,
             "callBackUrl": callback_url
         }
 
+        if task_id:
+            payload["taskId"] = task_id
+        if audio_id:
+            payload["audioId"] = audio_id
+
         try:
             response = await self.client.post("/api/v1/wav/generate", json=payload)
             response.raise_for_status()