From 76f6acbc03521056b627c6a5e88b9ca85555e639 Mon Sep 17 00:00:00 2001 From: David Tam Date: Fri, 25 Oct 2024 16:18:20 -0700 Subject: [PATCH] update for litellm dont pass reask messages to llms --- guardrails/llm_providers.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) 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,