CalvinBackup/llmsecops-research
1
0
Fork 0
mirror of https://github.com/lightbroker/llmsecops-research.git synced 2026-03-17 15:56:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

guardrail analyzed response, etc.

Browse Source
This commit is contained in:
Adam Wilson
2025-07-06 15:15:59 -06:00
parent a1d3a8c1b7
commit ffa2d73ae0
8 changed files with 84 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/text_generation/entrypoints/http_api_controller.py
View File

@@ -69,7 +69,7 @@ class HttpApiController:
return [response_body]
response_text = self.text_generation_response_service.invoke(user_prompt=prompt)
score = self.generated_text_guardrail_service.analyze(response_text)
score = self.generated_text_guardrail_service.is_text_malicious(response_text)
response_body = self.format_response(response_text)
http_status_code = 200 # make enum
@@ -96,7 +96,7 @@ class HttpApiController:
return [response_body]
response_text = self.rag_response_service.invoke(user_prompt=prompt)
score = self.generated_text_guardrail_service.analyze(response_text)
score = self.generated_text_guardrail_service.is_text_malicious(response_text)
response_body = self.format_response(response_text)
http_status_code = 200 # make enum