[Feat]: TaskManager.update_with_message should save updated task in TaskStore?

[jerry-junjie-xu]

Is your feature request related to a problem? Please describe.

Suppose we have AgentA deligating part of its job to AgentB, and want to share information about the task AgentA is executing, so that AgentB has more context and can could do a better job.

From the API it seems that could be achieved if we:

• set should_populate_referred_tasks=True when using SimpleRequestContextBuilder,
• pass AgentA's task Id to AgentB in Message.referenceTaskIds when calling AgentB,
• read RequestContext.related_tasks in AgentB.

However, today TaskManager only saves task to TaskStore on update event received, which could be AFTER AgentB is executed.

Describe the solution you'd like

The issue could be solved if we save the task early, for example, in TaskManager.update_with_message.

Describe alternatives you've considered

We can also yield some unimportant status updates to trigger task saving before doing anything in AgentB, but that might be too hacky.

Additional context

No response

Code of Conduct

• I agree to follow this project's Code of Conduct

[lkawka]

Hi jerry-junjie-xu,

This issue appears to have been resolved. The TaskManager.update_with_message method already exists.

I will be closing this issue now. Please feel free to reopen it if you believe the problem persists.

Thanks,
Lukasz