diff --git a/src/agentlab/llm/llm_utils.py b/src/agentlab/llm/llm_utils.py
index 708d68e8..31e8c137 100644
--- a/src/agentlab/llm/llm_utils.py
+++ b/src/agentlab/llm/llm_utils.py
@@ -116,6 +116,7 @@ def retry_multiple(
         parser (callable): a function taking a message and retruning a parsed value,
             or raising a ParseError
         log (bool): whether to log the retry messages.
+        num_samples (int): the number of samples to generate from the model.
 
     Returns:
         dict: the parsed value, with a string at key "action".