Skip to content

Commit 44bd8e1

Browse files
authored
Added labels proxy for results (#157)
1 parent 0195b60 commit 44bd8e1

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

taskiq/receiver/receiver.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -265,6 +265,7 @@ async def run_task( # noqa: C901, WPS210
265265
return_value=returned,
266266
execution_time=round(execution_time, 2),
267267
error=found_exception,
268+
labels=message.labels,
268269
)
269270
# If exception is found we execute middlewares.
270271
if found_exception is not None:

taskiq/result.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
from functools import partial
44
from typing import Any, Callable, Dict, Generic, Optional, TypeVar
55

6-
from pydantic import validator
6+
from pydantic import Field, validator
77
from pydantic.generics import GenericModel
88
from typing_extensions import Self
99

@@ -33,6 +33,7 @@ class TaskiqResult(GenericModel, Generic[_ReturnType]):
3333
log: Optional[str] = None
3434
return_value: _ReturnType
3535
execution_time: float
36+
labels: Dict[str, str] = Field(default_factory=dict)
3637

3738
error: Optional[BaseException] = None
3839

0 commit comments

Comments
 (0)