@@ -663,6 +663,14 @@ def init_command(project_directory, minimal: bool, template: str):
    show_default=True,
    help="Timeout for keep alive",
)
@click.pass_context
def serve_command(
    ctx: click.Context,
@@ -673,6 +681,7 @@ def serve_command(
    no_feature_log: bool,
    workers: int,
    keep_alive_timeout: int,
):
    """Start a feature server locally on a given port."""
    store = create_feature_store(ctx)
@@ -685,6 +694,7 @@ def serve_command(
        no_feature_log=no_feature_log,
        workers=workers,
        keep_alive_timeout=keep_alive_timeout,
    )



@@ -1,4 +1,5 @@
import json
import traceback
import warnings
from typing import List, Optional
@@ -44,14 +45,37 @@ class MaterializeIncrementalRequest(BaseModel):
    feature_views: Optional[List[str]] = None


-def get_app(store: "feast.FeatureStore"):
    proto_json.patch()

    app = FastAPI()

    async def get_body(request: Request):
        return await request.body()

    @app.post("/get-online-features")
    def get_online_features(body=Depends(get_body)):
        try:
@@ -180,7 +204,10 @@ def materialize_incremental(body=Depends(get_body)):

class FeastServeApplication(gunicorn.app.base.BaseApplication):
    def __init__(self, store: "feast.FeatureStore", **options):
-        self._app = get_app(store=store)
        self._options = options
        super().__init__()

@@ -202,11 +229,13 @@ def start_server(
    no_access_log: bool,
    workers: int,
    keep_alive_timeout: int,
):
    FeastServeApplication(
        store=store,
        bind=f"{host}:{port}",
        accesslog=None if no_access_log else "-",
        workers=workers,
        keepalive=keep_alive_timeout,
    ).run()
@@ -2228,6 +2228,7 @@ def serve(
        no_feature_log: bool,
        workers: int,
        keep_alive_timeout: int,
    ) -> None:
        """Start the feature consumption server locally on a given port."""
        type_ = type_.lower()
@@ -2243,6 +2244,7 @@ def serve(
            no_access_log=no_access_log,
            workers=workers,
            keep_alive_timeout=keep_alive_timeout,
        )

    @log_exceptions_and_usage