github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-21 12:01:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[minor]: bump version of LayerupSecurity, add support for untrusted_input parameter (#19985)

Browse Source 
**Description:** update version of LayerupSecurity package for the
Layerup Security integration. Add untrusted_input parameter.
This commit is contained in:
Jamsheed Mistri 2024-04-30 11:55:26 -07:00 committed by GitHub
parent f1c3687aa5
commit 3e749369ef
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libs/community/langchain_community/llms/layerup_security.py
View File

@ -82,7 +82,7 @@ class LayerupSecurity(LLM):
if self.prompt_guardrails: if self.prompt_guardrails:
security_response = self.client.execute_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"]: if not security_response["all_safe"]:
return self.handle_prompt_guardrail_violation(security_response) return self.handle_prompt_guardrail_violation(security_response)
@ -98,7 +98,7 @@ class LayerupSecurity(LLM):
if self.response_guardrails: if self.response_guardrails:
security_response = self.client.execute_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"]: if not security_response["all_safe"]:
return self.handle_response_guardrail_violation(security_response) return self.handle_response_guardrail_violation(security_response)