|
7 | 7 |
|
8 | 8 | from typing_extensions import Annotated, Doc, deprecated, override
|
9 | 9 |
|
10 |
| -from faststream.broker.message import gen_cor_id |
| 10 | +from faststream.broker.message import SourceType, gen_cor_id |
11 | 11 | from faststream.broker.publisher.usecase import PublisherUsecase
|
12 | 12 | from faststream.exceptions import NOT_CONNECTED_YET
|
13 | 13 | from faststream.redis.message import UnifyRedisDict
|
@@ -268,6 +268,7 @@ async def request(
|
268 | 268 |
|
269 | 269 | parsed_msg = await self._producer._parser(published_msg)
|
270 | 270 | parsed_msg._decoded_body = await self._producer._decoder(parsed_msg)
|
| 271 | + parsed_msg._source_type = SourceType.Response |
271 | 272 | return await return_msg(parsed_msg)
|
272 | 273 |
|
273 | 274 | raise AssertionError("unreachable")
|
@@ -481,6 +482,7 @@ async def request(
|
481 | 482 |
|
482 | 483 | parsed_msg = await self._producer._parser(published_msg)
|
483 | 484 | parsed_msg._decoded_body = await self._producer._decoder(parsed_msg)
|
| 485 | + parsed_msg._source_type = SourceType.Response |
484 | 486 | return await return_msg(parsed_msg)
|
485 | 487 |
|
486 | 488 | raise AssertionError("unreachable")
|
@@ -762,6 +764,7 @@ async def request(
|
762 | 764 |
|
763 | 765 | parsed_msg = await self._producer._parser(published_msg)
|
764 | 766 | parsed_msg._decoded_body = await self._producer._decoder(parsed_msg)
|
| 767 | + parsed_msg._source_type = SourceType.Response |
765 | 768 | return await return_msg(parsed_msg)
|
766 | 769 |
|
767 | 770 | raise AssertionError("unreachable")
|
0 commit comments