Skip to content

Commit fa7800d

Browse files
committed
fix: #90
1 parent 11bcebd commit fa7800d

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

taskiq_faststream/kicker.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,13 @@
1+
from typing import Any
2+
13
from taskiq.kicker import AsyncKicker, _FuncParams, _ReturnType
4+
from taskiq.message import TaskiqMessage
25

36

47
class LabelRespectKicker(AsyncKicker[_FuncParams, _ReturnType]):
58
"""Patched kicker doesn't cast labels to str."""
9+
10+
def _prepare_message(self, *args: Any, **kwargs: Any) -> TaskiqMessage:
11+
msg = super()._prepare_message(*args, **kwargs)
12+
msg.labels["message"] = self.labels["message"]
13+
return msg

0 commit comments

Comments
 (0)