fix random

Gradio_UI.py

@@ -40,17 +40,15 @@ def pull_messages_from_step(
         if hasattr(step_log, "model_output") and step_log.model_output is not None:
             # Clean up the LLM output
             model_output = step_log.model_output.strip()
-            # Remove any trailing <end_code> and extra backticks, handling multiple possible formats
-            model_output = re.sub(r"```\s*<end_code>", "```", model_output)  # handles ```<end_code>
-            model_output = re.sub(r"<end_code>\s*```", "```", model_output)  # handles <end_code>```
-            model_output = re.sub(r"```\s*\n\s*<end_code>", "```", model_output)  # handles ```\n<end_code>
+            # Remove any code-related formatting
+            model_output = re.sub(r"```.*?\n", "", model_output)  # Remove code blocks
+            model_output = re.sub(r"\s*<end_code>\s*", "", model_output)  # Remove end_code tags
             model_output = model_output.strip()
             yield gr.ChatMessage(role="assistant", content=model_output)
 
         # For tool calls, create a parent message
         if hasattr(step_log, "tool_calls") and step_log.tool_calls is not None:
             first_tool_call = step_log.tool_calls[0]
-            used_code = first_tool_call.name == "python_interpreter"
             parent_id = f"call_{len(step_log.tool_calls)}"
 
             # Tool call becomes the parent message with timing info
@@ -61,14 +59,6 @@ def pull_messages_from_step(
             else:
                 content = str(args).strip()
 
-            if used_code:
-                # Clean up the content by removing any end code tags
-                content = re.sub(r"```.*?\n", "", content)  # Remove existing code blocks
-                content = re.sub(r"\s*<end_code>\s*", "", content)  # Remove end_code tags
-                content = content.strip()
-                if not content.startswith("```python"):
-                    content = f"```python\n{content}\n```"
-
             parent_message_tool = gr.ChatMessage(
                 role="assistant",
                 content=content,