RedisRouter asyncapi_version argument

airtai · Aug 2, 2024 · 2f01e6b · 2f01e6b
1 parent f635f0d
commit 2f01e6b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/faststream/redis/fastapi/fastapi.py b/faststream/redis/fastapi/fastapi.py
@@ -28,6 +28,7 @@
 from typing_extensions import Annotated, Doc, deprecated, override
 
 from faststream.__about__ import SERVICE_NAME
+from faststream.asyncapi.version import AsyncAPIVersion
 from faststream.broker.fastapi.router import StreamRouter
 from faststream.broker.utils import default_filter
 from faststream.redis.broker.broker import RedisBroker as RB
@@ -130,6 +131,10 @@ def __init__(
             Optional[str],
             Doc("AsyncAPI server description."),
         ] = None,
+        asyncapi_version: Annotated[
+            AsyncAPIVersion,
+            Doc("Version of AsyncAPI for schema generation")
+        ] = AsyncAPIVersion.v2_6,
         asyncapi_tags: Annotated[
             Optional[Iterable[Union["asyncapi.Tag", "asyncapi.TagDict"]]],
             Doc("AsyncAPI server tags."),
@@ -412,6 +417,7 @@ def __init__(
             protocol=protocol,
             description=description,
             protocol_version=protocol_version,
+            asyncapi_version=asyncapi_version,
             asyncapi_tags=asyncapi_tags,
             asyncapi_url=asyncapi_url,
             # FastAPI kwargs

diff --git a/tests/asyncapi/redis/v3_0_0/test_fastapi.py b/tests/asyncapi/redis/v3_0_0/test_fastapi.py