fix

Author: felixs8696

src/agentex/lib/cli/templates/sync/project/acp.py.j2

@@ -3,7 +3,8 @@ from agentex.lib.sdk.fastacp.fastacp import FastACP
 from agentex.lib.types.acp import SendMessageParams
 
 from agentex.lib.types.task_message_updates import TaskMessageUpdate
-from agentex.types.task_messages import TaskMessageContent, TextContent
+from agentex.types.task_message_content import TaskMessageContent
+from agentex.types.text_content import TextContent
 from agentex.lib.utils.logging import make_logger
 
 logger = make_logger(__name__)
@@ -14,14 +15,12 @@ acp = FastACP.create(
     acp_type="sync",
 )
 
-
 @acp.on_message_send
 async def handle_message_send(
     params: SendMessageParams
-) -> Union[TaskMessageContent, AsyncGenerator[TaskMessageUpdate, None]]:
+) -> TaskMessageContent | list[TaskMessageContent] | AsyncGenerator[TaskMessageUpdate, None]:
     """Default message handler with streaming support"""
     return TextContent(
         author="agent",
         content=f"Hello! I've received your message. Here's a generic response, but in future tutorials we'll see how you can get me to intelligently respond to your message. This is what I heard you say: {params.content.content}",
     )
-

src/agentex/lib/sdk/fastacp/base/base_acp_server.py

@@ -294,7 +294,7 @@ def on_message_send(
         self,
         fn: Callable[
             [SendMessageParams],
-            Awaitable[list[TaskMessageContent] | AsyncGenerator[TaskMessageUpdate, None]],
+            Awaitable[TaskMessageContent | list[TaskMessageContent] | AsyncGenerator[TaskMessageUpdate, None]],
         ],
     ):
         """Handle message/send method - supports both single and streaming responses