TaskList is disconnected from messages

[hayescode]

Describe the bug
In previous versions before 2.0.0 making a TaskList with Tasks assigned to messages via forId would let you click on the Task and it would highlight the attached message.

Documentation: https://docs.chainlit.io/api-reference/elements/tasklist

Perhaps TaskList should be re-thought because this is a great capability for multi-agents (my use case) and it would be great to have "Agent Name" (author) above each task, integrate with the state of the Step/Message (if end is not null it's "processing") and things like that.

To Reproduce

import chainlit as cl


@cl.on_chat_start
async def main():
    # Create the TaskList
    task_list = cl.TaskList()
    task_list.status = "Running..."

    # Create a task and put it in the running state
    message1 = await cl.Message(content="Started processing data").send()
    task1 = cl.Task(title="Processing data", status=cl.TaskStatus.RUNNING, forId=message1.id)
    await task_list.add_task(task1)
    # Create another task that is in the ready state
    message2 = await cl.Message(content="Analyzing data").send()
    task2 = cl.Task(title="Analyzing data", status=cl.TaskStatus.READY, forId=message2.id)
    await task_list.add_task(task2)

    # Update the task list in the interface
    await task_list.send()

    # Perform some action on your end
    await cl.sleep(3)

    # Update the task statuses
    task1.status = cl.TaskStatus.DONE
    task2.status = cl.TaskStatus.RUNNING
    await task_list.send()
    await cl.sleep(3)

    task2.status = cl.TaskStatus.FAILED
    task_list.status = "Failed"
    await task_list.send()

[github-actions]

This issue is stale because it has been open for 14 days with no activity.

[s0l4r]

I encountered the same issue. I updated this in element.py -> send, to get forId back, not sure if it's the correct approach:

@dataclass
class TaskList(Element):
    type: ClassVar[ElementType] = "tasklist"
    tasks: List[Task] = Field(default_factory=list, exclude=True)
    status: str = "Ready"
    name: str = "tasklist"
    content: str = "dummy content to pass validation"

    def __post_init__(self) -> None:
        super().__post_init__()
        self.updatable = True

    async def add_task(self, task: Task):
        self.tasks.append(task)

    async def update(self):
        await self.send()

    async def send(self):
        await self.preprocess_content()
        await super().send(for_id="")  # <-- always "", so UI shows "forId": ""

Fix:

    async def send(self, for_id: Optional[str] = None, persist: bool = True):
        await self.preprocess_content()
        if for_id is not None:
            self.for_id = for_id
        if not self.for_id:
            raise ValueError("TaskList.send requires a for_id (message id).")
        await super().send(for_id=self.for_id, persist=persist)

Before fix:

[
    "element",
    {
        "id": "0f12cd5f-7eae-458c-ba19-162f7c0681fa",
        "threadId": "1bb26afa-861a-46a5-b99e-6c0e97db89eb",
        "type": "tasklist",
        "url": null,
        "chainlitKey": "7a27fde4-c87b-4822-871e-825d1badb493",
        "name": "tasklist",
        "display": "inline",
        "objectKey": null,
        "size": null,
        "props": null,
        "page": null,
        "autoPlay": null,
        "playerConfig": null,
        "language": null,
        "forId": "",
        "mime": "application/json"
    }
]

After fix:

[
    "element",
    {
        "id": "0f12cd5f-7eae-458c-ba19-162f7c0681fa",
        "threadId": "1bb26afa-861a-46a5-b99e-6c0e97db89eb",
        "type": "tasklist",
        "url": null,
        "chainlitKey": "7a27fde4-c87b-4822-871e-825d1badb493",
        "name": "tasklist",
        "display": "inline",
        "objectKey": null,
        "size": null,
        "props": null,
        "page": null,
        "autoPlay": null,
        "playerConfig": null,
        "language": null,
        "forId": "f8fba65f-1b12-458c-8552-f0a883e289b6",
        "mime": "application/json"
    }
]

Example:

tl = cl.TaskList(status="Running...")

   parse_task = cl.Task(title="Task 1", status=cl.TaskStatus.RUNNING)
   plan_task  = cl.Task(title="Task 2",  status=cl.TaskStatus.READY)
   tool_task  = cl.Task(title="Task 3", status=cl.TaskStatus.READY)
   final_task = cl.Task(title="Task 4", status=cl.TaskStatus.READY)

   message = await cl.Message(content="Started processing data").send()

   parse_task.forId = message.id
   plan_task.forId  = message.id
   tool_task.forId  = message.id
   final_task.forId = message.id

   await tl.add_task(parse_task)
   await tl.add_task(plan_task)
   await tl.add_task(tool_task)
   await tl.add_task(final_task)

   # Associate the TaskList element with the message
   await tl.send(for_id=message.id)

   # Later updates
   tl.status = "Executing plan..."
   await tl.update()

I haven't verified it to work fully yet with my use case.

[hayescode]

@s0l4r if you verify this works please submit a PR for this this would be a great contribution!