service implementations

src/text_generation/domain/abstract_text_generation_completion_result.py

@@ -1,13 +1,4 @@
 import abc
 
 class AbstractTextGenerationCompletionResult(abc.ABC):
-    
-    @abc.abstractmethod
-    def get_text(self) -> str:
-        """Return the generated text."""
-        pass
-    
-    @abc.abstractmethod
-    def get_metadata(self) -> dict:
-        """Return metadata about the generation."""
-        pass
+    pass

src/text_generation/domain/guidelines_processed_completion.py

@@ -1,4 +1,5 @@
 from src.text_generation.domain.abstract_guidelines_processed_completion import AbstractGuidelinesProcessedCompletion
+from src.text_generation.domain.abstract_text_generation_completion_result import AbstractTextGenerationCompletionResult
 
 
 class GuidelinesProcessedCompletion(
@@ -11,6 +12,22 @@ class GuidelinesProcessedCompletion(
             final: str):
         is_original_completion_malicious = score >= cosine_similarity_risk_threshold
 
+        self.score = score
+        self.original_completion = original_completion
+        self.is_original_completion_malicious = is_original_completion_malicious
+        self.final = final
+
+
+class TextGenerationCompletionResult(AbstractTextGenerationCompletionResult):
+    # TODO - implement
+    def __init__(
+            self, 
+            score: float,
+            cosine_similarity_risk_threshold: float,
+            original_completion: str,
+            final: str):
+        is_original_completion_malicious = score >= cosine_similarity_risk_threshold
+
         self.score = score
         self.original_completion = original_completion
         self.is_original_completion_malicious = is_original_completion_malicious