pybind/mgr: extend return types valid for Responder wrapped functions

One wonders why this wasn't found by the test code, but oh well.

Signed-off-by: John Mulligan <jmulligan@redhat.com>

diff --git a/src/pybind/mgr/object_format.py b/src/pybind/mgr/object_format.py
index ab3e0c707624a0b1be887b8d2a0bce5fea2928fa..d4c04fd91447eb98699dcfbc44e3534f74696ee5 100644 (file)
--- a/src/pybind/mgr/object_format.py
+++ b/src/pybind/mgr/object_format.py
@@ -190,12 +190,6 @@ class Format(str, enum.Enum):
 SimpleData = Any
 
 
-ObjectResponseFuncType = Union[
-    Callable[..., Dict[Any, Any]],
-    Callable[..., List[Any]],
-]
-
-
 class SimpleDataProvider(Protocol):
     def to_simplified(self) -> SimpleData:
         """Return a simplified representation of the current object.
@@ -443,6 +437,16 @@ class ErrorResponse(ErrorResponseBase):
         return err
 
 
+ObjectResponseFuncType = Union[
+    Callable[..., Dict[Any, Any]],
+    Callable[..., List[Any]],
+    Callable[..., SimpleDataProvider],
+    Callable[..., JSONDataProvider],
+    Callable[..., YAMLDataProvider],
+    Callable[..., ReturnValueProvider],
+]
+
+
 def _get_requested_format(f: ObjectResponseFuncType, kw: Dict[str, Any]) -> str:
     # todo: leave 'format' in kw dict iff its part of f's signature
     return kw.pop("format", None)