Fix ChatMessage/Role → Message/string roles in workflow_hitl.py (English + Spanish)

Author: pamelafox

examples/spanish/workflow_hitl.py

@@ -12,11 +12,10 @@
     AgentResponse,
     AgentRunUpdateEvent,
     Agent,
-    ChatMessage,
+    Message,
     Content,
     Executor,
     RequestInfoEvent,
-    Role,
     WorkflowBuilder,
     WorkflowContext,
     WorkflowOutputEvent,
@@ -128,7 +127,7 @@ class DraftFeedbackRequest:
 
     prompt: str = ""
     draft_text: str = ""
-    conversation: list[ChatMessage] = field(default_factory=list)  # type: ignore[reportUnknownVariableType]
+    conversation: list[Message] = field(default_factory=list)  # type: ignore[reportUnknownVariableType]
 
 
 class Coordinator(Executor):
@@ -154,7 +153,7 @@ async def on_writer_response(
         # Respuesta del agente escritor; solicitar retroalimentación humana.
         # Preservar la conversación completa para que el editor final
         # pueda ver los rastros de herramientas y el prompt inicial.
-        conversation: list[ChatMessage]
+        conversation: list[Message]
         if draft.full_conversation is not None:
             conversation = list(draft.full_conversation)
         else:
@@ -186,22 +185,22 @@ async def on_human_feedback(
             await ctx.send_message(
                 AgentExecutorRequest(
                     messages=original_request.conversation
-                    + [ChatMessage(Role.USER, text="La versión preliminar está aprobada tal como está.")],
+                    + [Message(role="user", text="La versión preliminar está aprobada tal como está.")],
                     should_respond=True,
                 ),
                 target_id=self.final_editor_id,
             )
             return
 
         # El humano proporcionó retroalimentación; indicar al escritor que revise.
-        conversation: list[ChatMessage] = list(original_request.conversation)
+        conversation: list[Message] = list(original_request.conversation)
         instruction = (
             "Un revisor humano compartió la siguiente guía:\n"
             f"{note or 'No se proporcionó guía específica.'}\n\n"
             "Reescribe la versión preliminar del mensaje anterior del asistente en una versión final pulida. "
             "Mantén la respuesta en menos de 120 palabras y refleja los ajustes de tono solicitados."
         )
-        conversation.append(ChatMessage(Role.USER, text=instruction))
+        conversation.append(Message(role="user", text=instruction))
         await ctx.send_message(
             AgentExecutorRequest(messages=conversation, should_respond=True), target_id=self.writer_id
         )

examples/workflow_hitl.py

@@ -12,11 +12,10 @@
     AgentResponse,
     AgentRunUpdateEvent,
     Agent,
-    ChatMessage,
+    Message,
     Content,
     Executor,
     RequestInfoEvent,
-    Role,
     WorkflowBuilder,
     WorkflowContext,
     WorkflowOutputEvent,
@@ -115,7 +114,7 @@ class DraftFeedbackRequest:
 
     prompt: str = ""
     draft_text: str = ""
-    conversation: list[ChatMessage] = field(default_factory=list)  # type: ignore[reportUnknownVariableType]
+    conversation: list[Message] = field(default_factory=list)  # type: ignore[reportUnknownVariableType]
 
 
 class Coordinator(Executor):
@@ -141,7 +140,7 @@ async def on_writer_response(
         # Writer agent response; request human feedback.
         # Preserve the full conversation so the final editor
         # can see tool traces and the initial prompt.
-        conversation: list[ChatMessage]
+        conversation: list[Message]
         if draft.full_conversation is not None:
             conversation = list(draft.full_conversation)
         else:
@@ -173,22 +172,22 @@ async def on_human_feedback(
             await ctx.send_message(
                 AgentExecutorRequest(
                     messages=original_request.conversation
-                    + [ChatMessage(Role.USER, text="The draft is approved as-is.")],
+                    + [Message(role="user", text="The draft is approved as-is.")],
                     should_respond=True,
                 ),
                 target_id=self.final_editor_id,
             )
             return
 
         # Human provided feedback; prompt the writer to revise.
-        conversation: list[ChatMessage] = list(original_request.conversation)
+        conversation: list[Message] = list(original_request.conversation)
         instruction = (
             "A human reviewer shared the following guidance:\n"
             f"{note or 'No specific guidance provided.'}\n\n"
             "Rewrite the draft from the previous assistant message into a polished final version. "
             "Keep the response under 120 words and reflect any requested tone adjustments."
         )
-        conversation.append(ChatMessage(Role.USER, text=instruction))
+        conversation.append(Message(role="user", text=instruction))
         await ctx.send_message(
             AgentExecutorRequest(messages=conversation, should_respond=True), target_id=self.writer_id
         )