From 3e749369efb702ef806373fed3d1399fbc9118d1 Mon Sep 17 00:00:00 2001 From: Jamsheed Mistri Date: Tue, 30 Apr 2024 11:55:26 -0700 Subject: [PATCH] community[minor]: bump version of LayerupSecurity, add support for untrusted_input parameter (#19985) **Description:** update version of LayerupSecurity package for the Layerup Security integration. Add untrusted_input parameter. --- libs/community/langchain_community/llms/layerup_security.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/community/langchain_community/llms/layerup_security.py b/libs/community/langchain_community/llms/layerup_security.py index 2a383cd2eeb..55b228cc5c7 100644 --- a/libs/community/langchain_community/llms/layerup_security.py +++ b/libs/community/langchain_community/llms/layerup_security.py @@ -82,7 +82,7 @@ class LayerupSecurity(LLM): if self.prompt_guardrails: security_response = self.client.execute_guardrails( - self.prompt_guardrails, messages, self.metadata + self.prompt_guardrails, messages, prompt, self.metadata ) if not security_response["all_safe"]: return self.handle_prompt_guardrail_violation(security_response) @@ -98,7 +98,7 @@ class LayerupSecurity(LLM): if self.response_guardrails: security_response = self.client.execute_guardrails( - self.response_guardrails, messages, self.metadata + self.response_guardrails, messages, result, self.metadata ) if not security_response["all_safe"]: return self.handle_response_guardrail_violation(security_response)