Refactor: Improve replicate retry logic and remove unused import

Co-authored-by: nicholas.tindle <[email protected]>

Author: cursoragent

autogpt_platform/backend/backend/blocks/ai_music_generator.py

@@ -1,11 +1,9 @@
-import asyncio
 import logging
 from enum import Enum
 from typing import Literal
 
 from pydantic import SecretStr
 from replicate.client import Client as ReplicateClient
-
 from replicate.helpers import FileOutput
 
 from backend.blocks.replicate._helper import run_replicate_with_retry

autogpt_platform/backend/backend/blocks/replicate/_helper.py

@@ -85,7 +85,7 @@ async def run_replicate_with_retry(
         except Exception as e:
             last_error = e
             if attempt < max_retries - 1:
-                wait_time = retry_delay * (2 ** attempt)
+                wait_time = retry_delay * (2**attempt)
                 logger.warning(
                     f"Replicate attempt {attempt + 1} failed: {str(e)}. Retrying in {wait_time}s..."
                 )