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)