bugfix: Handle return annotations

litestar-org · Dec 12, 2021 · 6724cbf · 6724cbf
1 parent cb334fc
commit 6724cbf
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/starlite/request.py b/starlite/request.py
@@ -74,6 +74,11 @@ class Config(BaseConfig):
         signature = Signature.from_callable(fn)
         field_definitions: Dict[str, Tuple[Any, Any]] = {}
         for key, value in getfullargspec(fn).annotations.items():
+
+            # discard return annotations
+            if key == "return":
+                continue
+
             parameter = signature.parameters[key]
             if parameter.default is not signature.empty:
                 field_definitions[key] = (value, parameter.default)

diff --git a/tests/test_request.py b/tests/test_request.py
@@ -7,7 +7,7 @@
 from pydantic.fields import ModelField
 from starlette.requests import Request
 
-from starlite import HttpMethod, ImproperlyConfiguredException, Provide, get, route, Starlite
+from starlite import HttpMethod, ImproperlyConfiguredException, Provide, get, route
 from starlite.request import (
     create_function_signature_model,
     get_kwargs_from_request,
@@ -97,7 +97,7 @@ async def test_function(data: Person):
 
 @pytest.mark.asyncio
 async def test_handle_return_annotation():
-    @get(path='/health', status_code=204)
+    @get(path="/health", status_code=204)
     async def health_check() -> None:
         return