Embabel Agent Documentation/com.embabel.agent.a2a.server.support/TaskStateManager

TaskStateManager

class TaskStateManager

Manages task state and history for streaming and resubscription support. Tracks active tasks, completed tasks, and their event history.

Constructors

TaskStateManager

constructor()

Types

data class TaskInfo(val taskId: String, val contextId: String, val streamId: String, val events: MutableList<<Error class: unknown class>> = mutableListOf(), val createdAt: Instant = Instant.now(), var completedAt: Instant? = null, var currentState: <Error class: unknown class> = TaskState.WORKING)

Stores information about a task

Functions

cleanupOldTasks

fun cleanupOldTasks(olderThan: Instant)

Removes old completed tasks (basic cleanup)

fun getStreamId(taskId: String): String?

Gets the stream ID associated with a task

fun getTaskEvents(taskId: String): List<<Error class: unknown class>>

Gets all events for a task

fun getTaskInfo(taskId: String): TaskStateManager.TaskInfo?

Gets the task info for a given task ID

fun isTaskActive(taskId: String): Boolean

Checks if a task is active

fun recordEvent(taskId: String, event: <Error class: unknown class>)

Records an event for a task

fun registerTask(taskId: String, contextId: String, streamId: String)

Registers a new task with its stream ID

fun taskExists(taskId: String): Boolean

Checks if a task exists (active or completed)

fun updateStreamId(taskId: String, newStreamId: String)

Updates the stream ID for a task (used during resubscription)