Update prompts/code_agent.yaml

prompts/code_agent.yaml

@@ -20,12 +20,12 @@ system_prompt: |-
 
   Example:
   ---
-  Task: "You're clueless."
+  Task: "You're a roly poly."
   Thought: I will call polite_guard once and return the raw result.
-  "{{code_block_opening_tag}}"
-  result = polite_guard("You're clueless.")
+  {{code_block_opening_tag}}
+  result = polite_guard("You're a roly poly.")
   final_answer(result)
-  "{{code_block_closing_tag}}"
+  {{code_block_closing_tag}}
 
   Rules:
   1. Always provide a 'Thought:' and at least one Code block.
@@ -46,13 +46,15 @@ planning:
   update_plan_pre_messages: |-
     You have been given this task:
     ```
-    "{{task}}"
+    {{task}}
     ```
-    Create a one-step plan that calls polite_guard only once.
+    Create a two-step plan that calls polite_guard only once.
 
   update_plan_post_messages: |-
     ## Plan
     1. Call polite_guard once and return result with final_answer(result).
+    2. If the content is impolite then provide two possible better pieces of content that are more polite than the original input.
+    
     <end_plan>
 
 managed_agent: