Skip to content

Commit 6e8ac73

Browse files
authored
Merge pull request #1144 from guardrails-ai/dont_pass_gr_args_to_llms
update for litellm dont pass reask messages to llms
2 parents af9f898 + 76f6acb commit 6e8ac73

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

guardrails/llm_providers.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -204,8 +204,7 @@ def _invoke_llm(
204204
)
205205

206206
# these are gr only and should not be getting passed to llms
207-
kwargs.pop("reask_prompt", None)
208-
kwargs.pop("reask_instructions", None)
207+
kwargs.pop("reask_messages", None)
209208

210209
response = completion(
211210
model=model,
@@ -663,8 +662,7 @@ async def invoke_llm(
663662
)
664663

665664
# these are gr only and should not be getting passed to llms
666-
kwargs.pop("reask_prompt", None)
667-
kwargs.pop("reask_instructions", None)
665+
kwargs.pop("reask_messages", None)
668666

669667
response = await acompletion(
670668
*args,

0 commit comments

Comments
 (0)