retrunnig route while adding api route for extandtion

fastapi/routing.py

@@ -911,7 +911,7 @@ class APIRouter(routing.Router):
         generate_unique_id_function: Union[
             Callable[[APIRoute], str], DefaultPlaceholder
         ] = Default(generate_unique_id),
-    ) -> None:
+    ) -> "APIRoute":
         route_class = route_class_override or self.route_class
         responses = responses or {}
         combined_responses = {**self.responses, **responses}
@@ -959,6 +959,7 @@ class APIRouter(routing.Router):
             generate_unique_id_function=current_generate_unique_id,
         )
         self.routes.append(route)
+        return route
 
     def api_route(
         self,
@@ -1029,7 +1030,7 @@ class APIRouter(routing.Router):
         name: Optional[str] = None,
         *,
         dependencies: Optional[Sequence[params.Depends]] = None,
-    ) -> None:
+    ) -> "APIWebSocketRoute":
         current_dependencies = self.dependencies.copy()
         if dependencies:
             current_dependencies.extend(dependencies)
@@ -1042,6 +1043,7 @@ class APIRouter(routing.Router):
             dependency_overrides_provider=self.dependency_overrides_provider,
         )
         self.routes.append(route)
+        return route
 
     def websocket(
         self,