Skip to content

Commit 4f13faf

Browse files
committed
Fix temporal as well
1 parent ec24827 commit 4f13faf

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

src/agentex/lib/core/temporal/services/temporal_task_service.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
from typing import Any
12
from agentex.lib.core.clients.temporal.temporal_client import TemporalClient
23
from agentex.lib.core.clients.temporal.types import WorkflowState
34
from agentex.lib.core.temporal.types.workflow import SignalName
@@ -22,7 +23,7 @@ def __init__(
2223
self._temporal_client = temporal_client
2324
self._env_vars = env_vars
2425

25-
async def submit_task(self, agent: Agent, task: Task) -> str:
26+
async def submit_task(self, agent: Agent, task: Task, params: dict[str, Any] | None) -> str:
2627
"""
2728
Submit a task to the async runtime for execution.
2829
@@ -33,7 +34,7 @@ async def submit_task(self, agent: Agent, task: Task) -> str:
3334
arg=CreateTaskParams(
3435
agent=agent,
3536
task=task,
36-
params=None,
37+
params=params,
3738
),
3839
id=task.id,
3940
task_queue=self._env_vars.WORKFLOW_TASK_QUEUE,

src/agentex/lib/sdk/fastacp/impl/temporal_acp.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ def _setup_handlers(self):
7171
async def handle_task_create(params: CreateTaskParams) -> None:
7272
"""Default create task handler - logs the task"""
7373
logger.info(f"TemporalACP received task create rpc call for task {params.task.id}")
74-
await self._temporal_task_service.submit_task(agent=params.agent, task=params.task)
74+
await self._temporal_task_service.submit_task(agent=params.agent, task=params.task, params=params.params)
7575

7676
@self.on_task_event_send
7777
async def handle_event_send(params: SendEventParams) -> None:

0 commit comments

Comments
 (0)