post_send fires after task execution with await_inplace=True

[mr-tabasco]

Taskiq version

0.11.20

Python version

Python 3.13

OS

macos 26.2

What happened?

When using InMemoryBroker(await_inplace=True), the post_send hook fires after the task completes instead of after the message is sent.

Current behavior:
pre_send → pre_execute → task → post_execute → post_send

Expected behavior:
pre_send → post_send → pre_execute → task → post_execute

This breaks the semantic meaning of "post send". If I'm logging "task queued" in post_send, it ends up with a later timestamp than "task started" in pre_execute.

Is this intentional? If so, might be worth documenting. If not, the broker could call post_send before invoking the task inline.

Relevant log output

Broker initialization code

from taskiq import InMemoryBroker, TaskiqMessage
  from taskiq.middlewares import SimpleRetryMiddleware
  from taskiq.abc.middleware import TaskiqMiddleware

  class TaskLoggingMiddleware(TaskiqMiddleware):
      async def _log_event(self, message: TaskiqMessage, status: str) -> None:
          # Inserts row to DB with created_at = now()
          # Problem: SENT rows end up with later timestamps than STARTED rows
          # because post_send fires after task execution with await_inplace=True
          pass

      async def post_send(self, message: TaskiqMessage) -> None:
          await self._log_event(message, "SENT")

      async def pre_execute(self, message: TaskiqMessage) -> TaskiqMessage:
          await self._log_event(message, "STARTED")
          return message

  broker = InMemoryBroker(await_inplace=True).with_middlewares(
      TaskLoggingMiddleware(),
      SimpleRetryMiddleware(default_retry_count=3),
  )