From bdce669e7d5c14e62e2370567d357c37ac9366c2 Mon Sep 17 00:00:00 2001
From: Michael Sun <michael.sun@scale.com>
Date: Tue, 29 Jul 2025 15:28:01 -0700
Subject: [PATCH] Revert sync template

---
 .../lib/cli/templates/sync/project/acp.py.j2  | 79 ++++---------------
 1 file changed, 15 insertions(+), 64 deletions(-)

diff --git a/src/agentex/lib/cli/templates/sync/project/acp.py.j2 b/src/agentex/lib/cli/templates/sync/project/acp.py.j2
index 069dc0b3..3d5b59db 100644
--- a/src/agentex/lib/cli/templates/sync/project/acp.py.j2
+++ b/src/agentex/lib/cli/templates/sync/project/acp.py.j2
@@ -1,75 +1,26 @@
-import json
-from agentex.lib import adk
+from typing import AsyncGenerator, Union
 from agentex.lib.sdk.fastacp.fastacp import FastACP
-from agentex.lib.types.fastacp import AgenticACPConfig
-from agentex.lib.types.acp import CancelTaskParams, CreateTaskParams, SendEventParams
+from agentex.lib.types.acp import SendMessageParams
 
+from agentex.lib.types.task_message_updates import TaskMessageUpdate
+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__)
 
 
-# Create an ACP server with base configuration
-# This sets up the core server that will handle task creation, events, and cancellation
+# Create an ACP server
 acp = FastACP.create(
-    acp_type="agentic",
-    config=AgenticACPConfig(
-        type="base",
-    ),
+    acp_type="sync",
 )
 
-@acp.on_task_create
-async def handle_task_create(params: CreateTaskParams):
-    # This handler is called first whenever a new task is created.
-    # It's a good place to initialize any state or resources needed for the task.
-
-    #########################################################
-    # 1. (👋) Do task initialization here.
-    #########################################################
-
-    # Acknowledge that the task has been created.
-    await adk.messages.create(
-        task_id=params.task.id,
-        content=TextContent(
-            author="agent",
-            content=f"Hello! I've received your task. Normally you can do some state initialization here, or just pass and do nothing until you get your first event. For now I'm just acknowledging that I've received a task with the following params:\n\n{json.dumps(params.params, indent=2)}.\n\nYou should only see this message once, when the task is created. All subsequent events will be handled by the `on_task_event_send` handler.",
-        ),
-    )
-
-@acp.on_task_event_send
-async def handle_event_send(params: SendEventParams):
-    # This handler is called whenever a new event (like a message) is sent to the task
-    
-    #########################################################
-    # 2. (👋) Echo back the client's message to show it in the UI.
-    #########################################################
-    
-    # This is not done by default so the agent developer has full control over what is shown to the user.
-    if params.event.content:
-        await adk.messages.create(task_id=params.task.id, content=params.event.content)
-
-    #########################################################
-    # 3. (👋) Send a simple response message.
-    #########################################################
-
-    # In future tutorials, this is where we'll add more sophisticated response logic.
-    await adk.messages.create(
-        task_id=params.task.id,
-        content=TextContent(
-            author="agent",
-            content=f"Hello! I've received your message. I can't respond right now, but in future tutorials we'll see how you can get me to intelligently respond to your message.",
-        ),
-    )
-
-@acp.on_task_cancel
-async def handle_task_cancel(params: CancelTaskParams):
-    # This handler is called when a task is cancelled.
-    # It's useful for cleaning up any resources or state associated with the task.
-
-    #########################################################
-    # 4. (👋) Do task cleanup here.
-    #########################################################
-
-    # This is mostly for durable workflows that are cancellable like Temporal, but we will leave it here for demonstration purposes.
-    logger.info(f"Hello! I've received task cancel for task {params.task.id}: {params.task}. This isn't necessary for this example, but it's good to know that it's available.")
+@acp.on_message_send
+async def handle_message_send(
+    params: SendMessageParams
+) -> 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}",
+    )
\ No newline at end of file