update output model for mosaic

src/titiler/core/titiler/core/factory.py

@@ -170,7 +170,7 @@ def register_routes(self):
 
         Because we wrap the endpoints in a class we cannot define the routes as
         methods (because of the self argument). The HACK is to define routes inside
-        the class method and register them after the class initialisation.
+        the class method and register them after the class initialization.
 
         """
         # Default Routes

src/titiler/core/titiler/core/models/responses.py

@@ -3,7 +3,7 @@
 from typing import Dict, List, Union
 
 from geojson_pydantic.features import Feature, FeatureCollection
-from geojson_pydantic.geometries import Geometry
+from geojson_pydantic.geometries import Geometry, Polygon
 from pydantic import BaseModel
 from rio_tiler.models import BandStatistics, Info
 
@@ -27,8 +27,7 @@ class MultiBasePoint(BaseModel):
     values: List[List[float]]
 
 
-InfoGeoJSON = Feature[Geometry, Info]
-
+InfoGeoJSON = Feature[Polygon, Info]
 Statistics = Dict[str, BandStatistics]
 
 
@@ -50,7 +49,7 @@ class Config:
 
 # MultiBase Models
 MultiBaseInfo = Dict[str, Info]
-MultiBaseInfoGeoJSON = Feature[Geometry, MultiBaseInfo]
+MultiBaseInfoGeoJSON = Feature[Polygon, MultiBaseInfo]
 
 MultiBaseStatistics = Dict[str, Statistics]
 MultiBaseStatisticsGeoJSON = Union[

src/titiler/mosaic/titiler/mosaic/factory.py

@@ -2,7 +2,7 @@
 
 import os
 from dataclasses import dataclass, field
-from typing import Callable, Dict, Optional, Type
+from typing import Callable, Dict, Optional, Type, Union
 from urllib.parse import urlencode, urlparse
 
 import rasterio
@@ -19,8 +19,9 @@
 from titiler.core.dependencies import WebMercatorTMSParams
 from titiler.core.factory import BaseTilerFactory, img_endpoint_params, templates
 from titiler.core.models.mapbox import TileJSON
+from titiler.core.models.responses import MultiBasePoint, Point
 from titiler.core.resources.enums import ImageType, MediaType, OptionalHeader
-from titiler.core.resources.responses import GeoJSONResponse, XMLResponse
+from titiler.core.resources.responses import GeoJSONResponse, JSONResponse, XMLResponse
 from titiler.core.utils import Timer
 from titiler.mosaic.resources.enums import PixelSelectionMethod
 
@@ -126,7 +127,7 @@ def info(src_path=Depends(self.path_dependency)):
 
         @self.router.get(
             "/info.geojson",
-            response_model=Feature,
+            response_model=Feature[Polygon, mosaicInfo],
             response_model_exclude_none=True,
             response_class=GeoJSONResponse,
             responses={
@@ -438,6 +439,8 @@ def point(self):
 
         @self.router.get(
             r"/point/{lon},{lat}",
+            response_model=Union[Point, MultiBasePoint],
+            response_class=JSONResponse,
             responses={200: {"description": "Return a value for a point"}},
         )
         def point(