diff --git a/simplemind/__init__.py b/simplemind/__init__.py
index b9e4d1a..53f6a7e 100644
--- a/simplemind/__init__.py
+++ b/simplemind/__init__.py
@@ -94,6 +94,7 @@ def generate_data(
         prompt=prompt,
         llm_model=llm_model,
         response_model=response_model,
+        **kwargs,
     )