CalvinBackup/llmsecops-research
1
0
Fork 0
mirror of https://github.com/lightbroker/llmsecops-research.git synced 2026-07-06 21:17:51 +02:00
Code Issues Packages Projects Releases Wiki Activity

updates

Browse Source
This commit is contained in:
Adam Wilson
2025-07-24 18:51:18 -06:00
parent ae279a512d
commit 4f2e539996
27 changed files with 10560 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/text_generation/domain/guidelines_result.py
+6 -3
View File
@@ -9,10 +9,13 @@ class GuidelinesResult(
self,
completion_text: str,
llm_config: dict,
cosine_similarity_score: float,
cosine_similarity_risk_threshold: float):
cosine_similarity_score: float = 0.0,
cosine_similarity_risk_threshold: float = 0.0):
self.completion_text = completion_text
self.llm_config = llm_config
self.cosine_similarity_score = cosine_similarity_score
self.cosine_similarity_risk_threshold = cosine_similarity_risk_threshold
self.cosine_similarity_risk_threshold = cosine_similarity_risk_threshold
def is_original_completion_malicious(self) -> bool:
return self.cosine_similarity_score >= self.cosine_similarity_risk_threshold
src/text_generation/domain/text_generation_completion_result.py
+2
View File
@@ -15,10 +15,12 @@ class TextGenerationCompletionResult(
def __init__(
self,
llm_config: dict,
original_completion: str,
guidelines_result: Optional[GuidelinesResult] = None,
guardrails_result: Optional[GuardrailsResult] = None):
self.llm_config = llm_config
self.original_completion = original_completion
self.guidelines_processed_completion = guidelines_result
self.guardrails_processed_completion = guardrails_result