CalvinBackup/llmsecops-research
1
0
Fork 0
mirror of https://github.com/lightbroker/llmsecops-research.git synced 2026-08-19 01:27:17 +02:00
Code Issues Packages Projects Releases Wiki Activity

more test cases and notes

Browse Source
This commit is contained in:
Adam Wilson
2025-07-06 17:47:49 -06:00
parent ffa2d73ae0
commit 7d49d160b9
7 changed files with 73 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/text_generation/entrypoints/http_api_controller.py
+2 -2
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.is_text_malicious(response_text)
score = self.generated_text_guardrail_service.process_generated_text(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.is_text_malicious(response_text)
score = self.generated_text_guardrail_service.process_generated_text(response_text)
response_body = self.format_response(response_text)
http_status_code = 200 # make enum