diff --git a/src/promptum/providers/provider.py b/src/promptum/providers/provider.py
index b2eded0..670ee15 100644
--- a/src/promptum/providers/provider.py
+++ b/src/promptum/providers/provider.py
@@ -2,6 +2,7 @@
 from typing import Any
 
 from promptum.providers.metrics import Metrics
+from promptum.providers.retry import RetryConfig
 
 
 class LLMProvider(ABC):
@@ -13,6 +14,7 @@ async def generate(
         system_prompt: str | None = None,
         temperature: float = 1.0,
         max_tokens: int | None = None,
+        retry_config: RetryConfig | None = None,
         **kwargs: Any,
     ) -> tuple[str, Metrics]:
         """