WIP reflexion service

src/text_generation/domain/guidelines_result.py

@@ -7,12 +7,14 @@ class GuidelinesResult(
     AbstractGuidelinesProcessedCompletion):
     def __init__(
             self,
+            user_prompt: str,
             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.user_prompt = user_prompt
         self.completion_text = completion_text
         self.full_prompt = full_prompt
         self.llm_config = llm_config

src/text_generation/domain/text_generation_completion_result.py

@@ -16,11 +16,13 @@ class TextGenerationCompletionResult(
     def __init__(
             self, 
             llm_config: dict,
+            original_user_prompt: str,
             original_completion: str,
             guidelines_result: Optional[GuidelinesResult] = None,
             guardrails_result: Optional[GuardrailsResult] = None):
 
         self.llm_config = llm_config
+        self.original_user_prompt = original_user_prompt
         self.original_completion = original_completion
         self.guidelines_processed_completion = guidelines_result
         self.guardrails_processed_completion = guardrails_result