release: 0.9.7

.release-please-manifest.json

@@ -1,3 +1,3 @@
 {
-  ".": "0.9.6"
+  ".": "0.9.7"
 }

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 45
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/sgp%2Fagentex-sdk-484a34db630cbb844d4496b9eada50771ded02db3f8ef71ec5316ce14d5470e4.yml
-openapi_spec_hash: aba2cc1906c8b07dc66f3b290d6d176f
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/sgp%2Fagentex-sdk-5fa3cb3c867281c804913c7c3e6d2143b5606d4924d42119f4b2b246f33e3db3.yml
+openapi_spec_hash: 8ec711692f3ed7cd34a7a3b9d3e33f7c
 config_hash: fb079ef7936611b032568661b8165f19

CHANGELOG.md

@@ -1,5 +1,13 @@
 # Changelog
 
+## 0.9.7 (2026-03-30)
+
+Full Changelog: [v0.9.6...v0.9.7](https://github.com/scaleapi/scale-agentex-python/compare/v0.9.6...v0.9.7)
+
+### Features
+
+* **lib:** Add task updates to adk ([ff12ae1](https://github.com/scaleapi/scale-agentex-python/commit/ff12ae199b38223c7c71b703fc8b11d5de99b0d8))
+
 ## 0.9.6 (2026-03-30)
 
 Full Changelog: [v0.9.5...v0.9.6](https://github.com/scaleapi/scale-agentex-python/compare/v0.9.5...v0.9.6)

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentex-sdk"
-version = "0.9.6"
+version = "0.9.7"
 description = "The official Python library for the agentex API"
 dynamic = ["readme"]
 license = "Apache-2.0"

src/agentex/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "agentex"
-__version__ = "0.9.6"  # x-release-please-version
+__version__ = "0.9.7"  # x-release-please-version

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

@@ -12,7 +12,10 @@
 from agentex.lib.core.temporal.activities.adk.tasks_activities import (
     DeleteTaskParams,
     GetTaskParams,
+    QueryWorkflowParams,
     TasksActivityName,
+    TaskStatusTransitionParams,
+    UpdateTaskParams,
 )
 from agentex.lib.core.tracing.tracer import AsyncTracer
 from agentex.types.task import Task
@@ -128,3 +131,301 @@ async def delete(
                 trace_id=trace_id,
                 parent_span_id=parent_span_id,
             )
+
+    async def cancel(
+        self,
+        *,
+        task_id: str,
+        reason: 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,
+    ) -> Task:
+        """
+        Mark a running task as canceled.
+        Args:
+            task_id: The ID of the task to cancel.
+            reason: Optional reason for cancellation.
+        Returns:
+            The updated task entry.
+        """
+        params = TaskStatusTransitionParams(
+            task_id=task_id,
+            reason=reason,
+            trace_id=trace_id,
+            parent_span_id=parent_span_id,
+        )
+        if in_temporal_workflow():
+            return await ActivityHelpers.execute_activity(
+                activity_name=TasksActivityName.CANCEL_TASK,
+                request=params,
+                response_type=Task,
+                start_to_close_timeout=start_to_close_timeout,
+                retry_policy=retry_policy,
+                heartbeat_timeout=heartbeat_timeout,
+            )
+        else:
+            return await self._tasks_service.cancel_task(
+                task_id=task_id,
+                reason=reason,
+                trace_id=trace_id,
+                parent_span_id=parent_span_id,
+            )
+
+    async def complete(
+        self,
+        *,
+        task_id: str,
+        reason: 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,
+    ) -> Task:
+        """
+        Mark a running task as completed.
+        Args:
+            task_id: The ID of the task to complete.
+            reason: Optional reason for completion.
+        Returns:
+            The updated task entry.
+        """
+        params = TaskStatusTransitionParams(
+            task_id=task_id,
+            reason=reason,
+            trace_id=trace_id,
+            parent_span_id=parent_span_id,
+        )
+        if in_temporal_workflow():
+            return await ActivityHelpers.execute_activity(
+                activity_name=TasksActivityName.COMPLETE_TASK,
+                request=params,
+                response_type=Task,
+                start_to_close_timeout=start_to_close_timeout,
+                retry_policy=retry_policy,
+                heartbeat_timeout=heartbeat_timeout,
+            )
+        else:
+            return await self._tasks_service.complete_task(
+                task_id=task_id,
+                reason=reason,
+                trace_id=trace_id,
+                parent_span_id=parent_span_id,
+            )
+
+    async def fail(
+        self,
+        *,
+        task_id: str,
+        reason: 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,
+    ) -> Task:
+        """
+        Mark a running task as failed.
+        Args:
+            task_id: The ID of the task to fail.
+            reason: Optional reason for failure.
+        Returns:
+            The updated task entry.
+        """
+        params = TaskStatusTransitionParams(
+            task_id=task_id,
+            reason=reason,
+            trace_id=trace_id,
+            parent_span_id=parent_span_id,
+        )
+        if in_temporal_workflow():
+            return await ActivityHelpers.execute_activity(
+                activity_name=TasksActivityName.FAIL_TASK,
+                request=params,
+                response_type=Task,
+                start_to_close_timeout=start_to_close_timeout,
+                retry_policy=retry_policy,
+                heartbeat_timeout=heartbeat_timeout,
+            )
+        else:
+            return await self._tasks_service.fail_task(
+                task_id=task_id,
+                reason=reason,
+                trace_id=trace_id,
+                parent_span_id=parent_span_id,
+            )
+
+    async def terminate(
+        self,
+        *,
+        task_id: str,
+        reason: 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,
+    ) -> Task:
+        """
+        Mark a running task as terminated.
+        Args:
+            task_id: The ID of the task to terminate.
+            reason: Optional reason for termination.
+        Returns:
+            The updated task entry.
+        """
+        params = TaskStatusTransitionParams(
+            task_id=task_id,
+            reason=reason,
+            trace_id=trace_id,
+            parent_span_id=parent_span_id,
+        )
+        if in_temporal_workflow():
+            return await ActivityHelpers.execute_activity(
+                activity_name=TasksActivityName.TERMINATE_TASK,
+                request=params,
+                response_type=Task,
+                start_to_close_timeout=start_to_close_timeout,
+                retry_policy=retry_policy,
+                heartbeat_timeout=heartbeat_timeout,
+            )
+        else:
+            return await self._tasks_service.terminate_task(
+                task_id=task_id,
+                reason=reason,
+                trace_id=trace_id,
+                parent_span_id=parent_span_id,
+            )
+
+    async def timeout(
+        self,
+        *,
+        task_id: str,
+        reason: 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,
+    ) -> Task:
+        """
+        Mark a running task as timed out.
+        Args:
+            task_id: The ID of the task to time out.
+            reason: Optional reason for timeout.
+        Returns:
+            The updated task entry.
+        """
+        params = TaskStatusTransitionParams(
+            task_id=task_id,
+            reason=reason,
+            trace_id=trace_id,
+            parent_span_id=parent_span_id,
+        )
+        if in_temporal_workflow():
+            return await ActivityHelpers.execute_activity(
+                activity_name=TasksActivityName.TIMEOUT_TASK,
+                request=params,
+                response_type=Task,
+                start_to_close_timeout=start_to_close_timeout,
+                retry_policy=retry_policy,
+                heartbeat_timeout=heartbeat_timeout,
+            )
+        else:
+            return await self._tasks_service.timeout_task(
+                task_id=task_id,
+                reason=reason,
+                trace_id=trace_id,
+                parent_span_id=parent_span_id,
+            )
+
+    async def update(
+        self,
+        *,
+        task_id: str | None = None,
+        task_name: str | None = None,
+        task_metadata: dict[str, object] | 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,
+    ) -> Task:
+        """
+        Update mutable fields for a task by ID or name.
+        Args:
+            task_id: The ID of the task to update.
+            task_name: The name of the task to update.
+            task_metadata: Metadata to update on the task.
+        Returns:
+            The updated task entry.
+        """
+        params = UpdateTaskParams(
+            task_id=task_id,
+            task_name=task_name,
+            task_metadata=task_metadata,
+            trace_id=trace_id,
+            parent_span_id=parent_span_id,
+        )
+        if in_temporal_workflow():
+            return await ActivityHelpers.execute_activity(
+                activity_name=TasksActivityName.UPDATE_TASK,
+                request=params,
+                response_type=Task,
+                start_to_close_timeout=start_to_close_timeout,
+                retry_policy=retry_policy,
+                heartbeat_timeout=heartbeat_timeout,
+            )
+        else:
+            return await self._tasks_service.update_task(
+                task_id=task_id,
+                task_name=task_name,
+                task_metadata=task_metadata,
+                trace_id=trace_id,
+                parent_span_id=parent_span_id,
+            )
+
+    async def query_workflow(
+        self,
+        *,
+        task_id: str,
+        query_name: str,
+        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,
+    ) -> dict[str, object]:
+        """
+        Query a Temporal workflow associated with a task for its current state.
+        Args:
+            task_id: The ID of the task whose workflow to query.
+            query_name: The name of the query to execute.
+        Returns:
+            The query result.
+        """
+        params = QueryWorkflowParams(
+            task_id=task_id,
+            query_name=query_name,
+            trace_id=trace_id,
+            parent_span_id=parent_span_id,
+        )
+        if in_temporal_workflow():
+            return await ActivityHelpers.execute_activity(
+                activity_name=TasksActivityName.QUERY_WORKFLOW,
+                request=params,
+                response_type=dict,
+                start_to_close_timeout=start_to_close_timeout,
+                retry_policy=retry_policy,
+                heartbeat_timeout=heartbeat_timeout,
+            )
+        else:
+            return await self._tasks_service.query_workflow(
+                task_id=task_id,
+                query_name=query_name,
+                trace_id=trace_id,
+                parent_span_id=parent_span_id,
+            )