diff --git a/guardrails/llm_providers.py b/guardrails/llm_providers.py index fda296d03..1c826191c 100644 --- a/guardrails/llm_providers.py +++ b/guardrails/llm_providers.py @@ -204,8 +204,7 @@ def _invoke_llm( ) # these are gr only and should not be getting passed to llms - kwargs.pop("reask_prompt", None) - kwargs.pop("reask_instructions", None) + kwargs.pop("reask_messages", None) response = completion( model=model, @@ -663,8 +662,7 @@ async def invoke_llm( ) # these are gr only and should not be getting passed to llms - kwargs.pop("reask_prompt", None) - kwargs.pop("reask_instructions", None) + kwargs.pop("reask_messages", None) response = await acompletion( *args,