Update func signature

airtai · May 24, 2024 · d76b1b2 · d76b1b2
1 parent 2bb6092
commit d76b1b2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/faststream/confluent/client.py b/faststream/confluent/client.py
@@ -1,4 +1,5 @@
 import asyncio
+import logging
 from ssl import SSLContext
 from time import time
 from typing import (
@@ -111,7 +112,7 @@ def __init__(
         sasl_kerberos_domain_name: Optional[str] = None,
         sasl_oauth_token_provider: Optional[str] = None,
         logger: Annotated[
-            Union["LoggerProto", None, object],
+            Union["LoggerProto", logging.Logger, None, object],
             Doc("User specified logger to pass into Context and log service messages."),
         ] = logger,
     ) -> None:
@@ -306,7 +307,7 @@ def __init__(
         sasl_kerberos_domain_name: Optional[str] = None,
         sasl_oauth_token_provider: Optional[str] = None,
         logger: Annotated[
-            Union["LoggerProto", None, object],
+            Union["LoggerProto", logging.Logger, None, object],
             Doc("User specified logger to pass into Context and log service messages."),
         ] = logger,
     ) -> None: