Fix runtime type introspection for datastar_response decorator (#2)

Author: gazpachoking

src/datastar_py/litestar.py

@@ -77,7 +77,7 @@ async def wrapper(*args: P.args, **kwargs: P.kwargs) -> DatastarResponse:
             return DatastarResponse(await r)
         return DatastarResponse(r)
 
-    wrapper.__annotations__["return"] = "DatastarResponse"
+    wrapper.__annotations__["return"] = DatastarResponse
     return wrapper
 
 

src/datastar_py/quart.py

@@ -55,7 +55,7 @@ async def wrapper(*args: P.args, **kwargs: P.kwargs) -> DatastarResponse:
             return DatastarResponse(stream_with_context(func)(*args, **kwargs))
         return DatastarResponse(await copy_current_request_context(func)(*args, **kwargs))
 
-    wrapper.__annotations__["return"] = "DatastarResponse"
+    wrapper.__annotations__["return"] = DatastarResponse
     return wrapper
 
 

src/datastar_py/sanic.py

@@ -3,7 +3,7 @@
 from collections.abc import Awaitable, Collection, Mapping
 from functools import wraps
 from inspect import isasyncgen, isgenerator
-from typing import Any, Callable, ParamSpec
+from typing import Any, Callable, ParamSpec, Union
 
 from sanic import HTTPResponse, Request
 
@@ -87,7 +87,7 @@ async def wrapper(*args: P.args, **kwargs: P.kwargs) -> DatastarResponse | None:
             return None
         return DatastarResponse(r)
 
-    wrapper.__annotations__["return"] = "DatastarResponse"
+    wrapper.__annotations__["return"] = Union[DatastarResponse, None]
     return wrapper
 
 

src/datastar_py/starlette.py

@@ -67,7 +67,7 @@ async def wrapper(*args: P.args, **kwargs: P.kwargs) -> DatastarResponse:
             return DatastarResponse(await r)
         return DatastarResponse(r)
 
-    wrapper.__annotations__["return"] = "DatastarResponse"
+    wrapper.__annotations__["return"] = DatastarResponse
     return wrapper