more progress on fluent service call

src/text_generation/domain/abstract_guardrail_analyzed_response.py

@@ -1,4 +0,0 @@
-import abc
-
-class AbstractGuardrailAnalyzedResponse(abc.ABC):
-    pass

src/text_generation/domain/abstract_guardrail_processed_completion.py

@@ -0,0 +1,4 @@
+import abc
+
+class AbstractGuardrailProcessedCompletion(abc.ABC):
+    pass

src/text_generation/domain/guardrail_analyzed_response.py

@@ -1,15 +0,0 @@
-from src.text_generation.domain.abstract_guardrail_analyzed_response import AbstractGuardrailAnalyzedResponse
-
-
-class GuardrailAnalyzedResponse(AbstractGuardrailAnalyzedResponse):
-    def __init__(
-            self, 
-            score: float,
-            cosine_similarity_risk_threshold: float,
-            original: str):
-        is_completion_malicious = score >= cosine_similarity_risk_threshold
-
-        self.score = score
-        self.original = original
-        self.is_completion_malicious = is_completion_malicious
-        self.final = "I can't answer that." if is_completion_malicious else original

src/text_generation/domain/guardrail_processed_completion.py

@@ -0,0 +1,17 @@
+from src.text_generation.domain.abstract_guardrail_processed_completion import AbstractGuardrailProcessedCompletion
+
+
+class GuardrailProcessedCompletion(
+    AbstractGuardrailProcessedCompletion):
+    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
+        self.final = final