make task id not required

Author: felixs8696

pyproject.toml

@@ -39,7 +39,7 @@ dependencies = [
     "pytest-asyncio>=1.0.0",
     "scale-gp-beta==0.1.0a20",
     "ipykernel>=6.29.5",
-    "openai==1.99.9", # anything higher than 1.99.9 breaks litellm - https://github.com/BerriAI/litellm/issues/13711 
+    "openai==1.99.9", # anything higher than 1.99.9 breaks litellm - https://github.com/BerriAI/litellm/issues/13711
     "cloudpickle>=3.1.1",
 ]
 requires-python = ">= 3.12,<4"

src/agentex/lib/adk/_modules/acp.py

@@ -1,6 +1,7 @@
 from datetime import timedelta
-from typing import Any
+from typing import Any, List
 
+from agentex.types import Event
 from temporalio.common import RetryPolicy
 
 from agentex import AsyncAgentex
@@ -58,7 +59,7 @@ async def create_task(
         start_to_close_timeout: timedelta = timedelta(seconds=5),
         heartbeat_timeout: timedelta = timedelta(seconds=5),
         retry_policy: RetryPolicy = DEFAULT_RETRY_POLICY,
-    ):
+    ) -> Task:
         """
         Create a new task.
 
@@ -111,7 +112,7 @@ async def send_event(
         start_to_close_timeout: timedelta = timedelta(seconds=5),
         heartbeat_timeout: timedelta = timedelta(seconds=5),
         retry_policy: RetryPolicy = DEFAULT_RETRY_POLICY,
-    ):
+    ) -> Event:
         """
         Send an event to a task.
 
@@ -152,16 +153,16 @@ async def send_event(
 
     async def send_message(
         self,
-        task_id: str,
         content: TaskMessageContent,
+        task_id: str | None = None,
         agent_id: str | None = None,
         agent_name: str | None = None,
         trace_id: str | None = None,
         parent_span_id: str | None = None,
         start_to_close_timeout: timedelta = timedelta(seconds=5),
         heartbeat_timeout: timedelta = timedelta(seconds=5),
         retry_policy: RetryPolicy = DEFAULT_RETRY_POLICY,
-    ):
+    ) -> List[TaskMessage]:
         """
         Send a message to a task.
 
@@ -209,7 +210,7 @@ async def cancel_task(
         start_to_close_timeout: timedelta = timedelta(seconds=5),
         heartbeat_timeout: timedelta = timedelta(seconds=5),
         retry_policy: RetryPolicy = DEFAULT_RETRY_POLICY,
-    ):
+    ) -> Task:
         """
         Cancel a task.
 

src/agentex/lib/core/services/adk/acp/acp.py

@@ -116,6 +116,7 @@ async def message_send(
                 raise ValueError("Either agent_name or agent_id must be provided")
 
             task_messages: List[TaskMessage] = []
+            logger.info(f"json_rpc_response: {json_rpc_response}")
             if isinstance(json_rpc_response.result, list):
                 for message in json_rpc_response.result:
                     task_message = TaskMessage.model_validate(message)

src/agentex/lib/core/temporal/activities/__init__.py

@@ -1,3 +1,4 @@
+import httpx
 from agentex.lib.adk.utils._modules.client import create_async_agentex_client
 from scale_gp import SGPClient, SGPClientError
 
@@ -59,7 +60,9 @@ def get_all_activities(sgp_client=None):
 
     llm_gateway = LiteLLMGateway()
     stream_repository = RedisStreamRepository()
-    agentex_client = create_async_agentex_client()
+    agentex_client = create_async_agentex_client(
+        timeout=httpx.Timeout(timeout=1000),
+    )
     tracer = AsyncTracer(agentex_client)
 
     # Services

src/agentex/lib/core/temporal/activities/adk/acp/acp_activities.py

@@ -1,5 +1,5 @@
 from enum import Enum
-from typing import Any
+from typing import Any, List
 
 from temporalio import activity
 
@@ -61,7 +61,7 @@ async def task_create(self, params: TaskCreateParams) -> Task:
         )
 
     @activity.defn(name=ACPActivityName.MESSAGE_SEND)
-    async def message_send(self, params: MessageSendParams) -> TaskMessage:
+    async def message_send(self, params: MessageSendParams) -> List[TaskMessage]:
         return await self._acp_service.message_send(
             agent_id=params.agent_id,
             agent_name=params.agent_name,