diff --git a/stubs/psycopg2/psycopg2/extras.pyi b/stubs/psycopg2/psycopg2/extras.pyi index f732bc5ad..b3f953cdb 100644 --- a/stubs/psycopg2/psycopg2/extras.pyi +++ b/stubs/psycopg2/psycopg2/extras.pyi @@ -196,6 +196,9 @@ class ReplicationCursor(_replicationCursor): status_interval: int = 10, ) -> None: ... def fileno(self): ... + def consume_stream( + self, consume: Callable[[ReplicationMessage], object], keepalive_interval: float | None = None + ) -> None: ... class UUID_adapter: def __init__(self, uuid) -> None: ...