diff --git a/src/datastar_py/sse.py b/src/datastar_py/sse.py index c5f7577..d8e50a1 100644 --- a/src/datastar_py/sse.py +++ b/src/datastar_py/sse.py @@ -6,7 +6,7 @@ from typing import Literal, Protocol, TypeAlias, overload, runtime_checkable import datastar_py.consts as consts -from datastar_py.attributes import _escape +from datastar_py.attributes import _escape, SignalValue SSE_HEADERS: dict[str, str] = { "Cache-Control": "no-cache", @@ -136,7 +136,7 @@ def remove_elements( @classmethod def patch_signals( cls, - signals: dict | str, + signals: dict[str, SignalValue] | str, event_id: str | None = None, only_if_missing: bool | None = None, retry_duration: int | None = None,