mirror of
https://github.com/lightbroker/llmsecops-research.git
synced 2026-10-11 02:19:03 +02:00
updates for RAG + CoT tests
This commit is contained in:
1 parent
72785c6420
commit
741629908c
10 files changed
+6603
-61
No files matched your search
@@ -1,4 +1,4 @@
|
||||
from typing import List
|
||||
from typing import Any, List
|
||||
from src.text_generation.domain.abstract_guidelines_processed_completion import AbstractGuidelinesProcessedCompletion
|
||||
from src.text_generation.domain.abstract_text_generation_completion_result import AbstractTextGenerationCompletionResult
|
||||
|
||||
@@ -8,11 +8,13 @@ class GuidelinesResult(
|
||||
def __init__(
|
||||
self,
|
||||
completion_text: str,
|
||||
full_prompt: dict[str, Any],
|
||||
llm_config: dict,
|
||||
cosine_similarity_score: float = 0.0,
|
||||
cosine_similarity_risk_threshold: float = 0.0):
|
||||
|
||||
self.completion_text = completion_text
|
||||
self.full_prompt = full_prompt
|
||||
self.llm_config = llm_config
|
||||
self.cosine_similarity_score = cosine_similarity_score
|
||||
self.cosine_similarity_risk_threshold = cosine_similarity_risk_threshold
|
||||
|
||||
Reference in new issue
Block a user