refactor(llm): standardize access to parsed task output text

- Updated all instances to use `result.text` for accessing parsed
  task output text.
- Replaced direct usage of `result` with `text` where applicable
  for clarity and consistency.
- Adjusted related logic to ensure proper handling of stripped
  and formatted text.

Author: Pouyanpi

nemoguardrails/actions/llm/generation.py

@@ -442,8 +442,9 @@ async def generate_user_intent(
             result = self.llm_task_manager.parse_task_output(
                 Task.GENERATE_USER_INTENT, output=result
             )
+            text = result.text
 
-            user_intent = get_first_nonempty_line(result)
+            user_intent = get_first_nonempty_line(text)
             if user_intent is None:
                 user_intent = "unknown message"
 
@@ -527,9 +528,11 @@ async def generate_user_intent(
                             prompt,
                             custom_callback_handlers=[streaming_handler_var.get()],
                         )
-                    text = self.llm_task_manager.parse_task_output(
+                    result = self.llm_task_manager.parse_task_output(
                         Task.GENERAL, output=text
                     )
+                    text = result.text
+                    text = text.strip()
             else:
                 # Initialize the LLMCallInfo object
                 llm_call_info_var.set(LLMCallInfo(task=Task.GENERAL.value))
@@ -562,9 +565,10 @@ async def generate_user_intent(
                         stop=["User:"],
                     )
 
-                text = self.llm_task_manager.parse_task_output(
+                result = self.llm_task_manager.parse_task_output(
                     Task.GENERAL, output=result
                 )
+                text = result.text
                 text = text.strip()
                 if text.startswith('"'):
                     text = text[1:-1]
@@ -646,10 +650,11 @@ async def generate_next_step(
             result = self.llm_task_manager.parse_task_output(
                 Task.GENERATE_NEXT_STEPS, output=result
             )
+            text = result.text
 
             # If we don't have multi-step generation enabled, we only look at the first line.
             if not self.config.enable_multi_step_generation:
-                result = get_first_nonempty_line(result)
+                result = get_first_nonempty_line(text)
 
                 if result and result.startswith("bot "):
                     bot_intent = result[4:]
@@ -687,7 +692,7 @@ async def generate_next_step(
                 # Otherwise, we parse the output as a single flow.
                 # If we have a parsing error, we try to reduce size of the flow, potentially
                 # up to a single step.
-                lines = result.split("\n")
+                lines = text.split("\n")
                 while True:
                     try:
                         parse_colang_file("dynamic.co", content="\n".join(lines))
@@ -896,10 +901,15 @@ async def generate_bot_message(
                             llm, prompt, custom_callback_handlers=[streaming_handler]
                         )
 
+                        # it seems that removing the reasoning traces is llm_call responsibility
+                        #
+
                         result = self.llm_task_manager.parse_task_output(
                             Task.GENERAL, output=result
                         )
 
+                        result = result.text
+
                     log.info(
                         "--- :: LLM Bot Message Generation passthrough call took %.2f seconds",
                         time() - t0,
@@ -963,6 +973,8 @@ async def generate_bot_message(
                     Task.GENERATE_BOT_MESSAGE, output=result
                 )
 
+                result = result.text
+
                 # TODO: catch openai.error.InvalidRequestError from exceeding max token length
 
                 result = get_multiline_response(result)
@@ -1055,10 +1067,11 @@ async def generate_value(
         result = self.llm_task_manager.parse_task_output(
             Task.GENERATE_VALUE, output=result
         )
+        text = result.text
 
         # We only use the first line for now
         # TODO: support multi-line values?
-        value = result.strip().split("\n")[0]
+        value = text.strip().split("\n")[0]
 
         # Because of conventions from other languages, sometimes the LLM might add
         # a ";" at the end of the line. We remove that
@@ -1266,22 +1279,23 @@ async def generate_intent_steps_message(
             result = self.llm_task_manager.parse_task_output(
                 Task.GENERATE_INTENT_STEPS_MESSAGE, output=result
             )
+            text = result.text
 
             # TODO: Implement logic for generating more complex Colang next steps (multi-step),
             #  not just a single bot intent.
 
             # Get the next 2 non-empty lines, these should contain:
             # line 1 - user intent, line 2 - bot intent.
             # Afterwards we have the bot message.
-            next_three_lines = get_top_k_nonempty_lines(result, k=2)
+            next_three_lines = get_top_k_nonempty_lines(text, k=2)
             user_intent = next_three_lines[0] if len(next_three_lines) > 0 else None
             bot_intent = next_three_lines[1] if len(next_three_lines) > 1 else None
             bot_message = None
             if bot_intent:
-                pos = result.find(bot_intent)
+                pos = text.find(bot_intent)
                 if pos != -1:
                     # The bot message could be multiline
-                    bot_message = result[pos + len(bot_intent) :]
+                    bot_message = text[pos + len(bot_intent) :]
                     bot_message = get_multiline_response(bot_message)
                     bot_message = strip_quotes(bot_message)
                     # Quick hack for degenerated / empty bot messages
@@ -1348,7 +1362,8 @@ async def generate_intent_steps_message(
             result = self.llm_task_manager.parse_task_output(
                 Task.GENERAL, output=result
             )
-            text = result.strip()
+            text = result.text
+            text = text.strip()
             if text.startswith('"'):
                 text = text[1:-1]
 

nemoguardrails/actions/v2_x/generation.py

@@ -197,7 +197,7 @@ async def _collect_user_intent_and_examples(
 
             # We add these in reverse order so the most relevant is towards the end.
             for result in reversed(results):
-                examples += f"user action: user said \"{result.text}\"\nuser intent: {result.meta['intent']}\n\n"
+                examples += f'user action: user said "{result.text}"\nuser intent: {result.meta["intent"]}\n\n'
                 if result.meta["intent"] not in potential_user_intents:
                     potential_user_intents.append(result.meta["intent"])
 
@@ -302,6 +302,8 @@ async def generate_user_intent(
             Task.GENERATE_USER_INTENT_FROM_USER_ACTION, output=result
         )
 
+        result = result.text
+
         user_intent = get_first_nonempty_line(result)
         # GTP-4o often adds 'user intent: ' in front
         if user_intent and ":" in user_intent:
@@ -378,6 +380,8 @@ async def generate_user_intent_and_bot_action(
             Task.GENERATE_USER_INTENT_AND_BOT_ACTION_FROM_USER_ACTION, output=result
         )
 
+        result = result.text
+
         user_intent = get_first_nonempty_line(result)
 
         if user_intent and ":" in user_intent:
@@ -458,6 +462,8 @@ async def passthrough_llm_action(
 
             text = self.llm_task_manager.parse_task_output(Task.GENERAL, output=text)
 
+            text = result.text
+
         return text
 
     @action(name="CheckValidFlowExistsAction", is_system_action=True)
@@ -541,6 +547,8 @@ async def generate_flow_from_instructions(
             task=Task.GENERATE_FLOW_FROM_INSTRUCTIONS, output=result
         )
 
+        result = result.text
+
         # TODO: why this is not part of a filter or output_parser?
         #
         lines = _remove_leading_empty_lines(result).split("\n")
@@ -613,6 +621,8 @@ async def generate_flow_from_name(
             task=Task.GENERATE_FLOW_FROM_NAME, output=result
         )
 
+        result = result.text
+
         lines = _remove_leading_empty_lines(result).split("\n")
 
         if lines[0].startswith("flow"):
@@ -680,6 +690,8 @@ async def generate_flow_continuation(
             task=Task.GENERATE_FLOW_CONTINUATION, output=result
         )
 
+        result = result.text
+
         lines = _remove_leading_empty_lines(result).split("\n")
 
         if len(lines) == 0 or (len(lines) == 1 and lines[0] == ""):
@@ -806,6 +818,8 @@ async def generate_value(
             Task.GENERATE_VALUE_FROM_INSTRUCTION, output=result
         )
 
+        result = result.text
+
         # We only use the first line for now
         # TODO: support multi-line values?
         value = result.strip().split("\n")[0]
@@ -913,6 +927,8 @@ async def generate_flow(
             Task.GENERATE_FLOW_CONTINUATION_FROM_NLD, output=result
         )
 
+        result = result.text
+
         result = _remove_leading_empty_lines(result)
         lines = result.split("\n")
         if "codeblock" in lines[0]:

nemoguardrails/library/content_safety/actions.py

@@ -80,6 +80,7 @@ async def content_safety_check_input(
         result = await llm_call(llm, check_input_prompt, stop=stop)
 
     result = llm_task_manager.parse_task_output(task, output=result)
+    result = result.text
 
     try:
         is_safe, violated_policies = result
@@ -162,6 +163,8 @@ async def content_safety_check_output(
 
     result = llm_task_manager.parse_task_output(task, output=result)
 
+    result = result.text
+
     try:
         is_safe, violated_policies = result
     except TypeError:

nemoguardrails/library/self_check/facts/actions.py

@@ -82,6 +82,7 @@ async def self_check_facts(
             task, output=response, forced_output_parser="is_content_safe"
         )
 
+    result = result.text
     is_not_safe, _ = result
 
     result = float(not is_not_safe)

nemoguardrails/library/self_check/input_check/actions.py

@@ -83,6 +83,7 @@ async def self_check_input(
                 task, output=response, forced_output_parser="is_content_safe"
             )
 
+        result = result.text
         is_safe, _ = result
 
         if not is_safe:

nemoguardrails/library/self_check/output_check/actions.py

@@ -87,6 +87,7 @@ async def self_check_output(
                 task, output=response, forced_output_parser="is_content_safe"
             )
 
+        result = result.text
         is_safe, _ = result
 
         return is_safe

tests/test_configs/with_custom_llm_prompt_action_v2_x/actions.py

@@ -50,6 +50,7 @@ async def custom_llm_request(
         result = await llm_call(llm, prompt, stop=stop)
 
     result = llm_task_manager.parse_task_output(prompt_template_name, output=result)
+    result = result.text
 
     # Any additional parsing of the output
     value = result.strip().split("\n")[0]