✨ Add security_protocol all possible values (#22)

gabrielmbmb · Apr 11, 2023 · 6abe86d · 6abe86d
1 parent 3edf959
commit 6abe86d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/kaflow/applications.py b/kaflow/applications.py
@@ -87,7 +87,9 @@ def __init__(
         consumer_timeout_ms: int = 200,
         max_poll_records: int | None = None,
         kafka_api_version: str = "auto",
-        security_protocol: Literal["PLAINTEXT", "SSL"] = "PLAINTEXT",
+        security_protocol: Literal[
+            "PLAINTEXT", "SSL", "SASL_PLAINTEXT", "SASL_SSL"
+        ] = "PLAINTEXT",
         exclude_internal_topics: bool = True,
         connection_max_idle_ms: int = 540000,
         isolation_level: Literal[