Refactor responses to leverage hosting for server and route registration

sdk/agentserver/azure-ai-agentserver-hosting/azure/ai/agentserver/hosting/_base.py

@@ -8,6 +8,8 @@
 from typing import Any, Optional
 
 from starlette.applications import Starlette
+from starlette.middleware import Middleware
+from starlette.middleware.base import BaseHTTPMiddleware
 from starlette.requests import Request
 from starlette.responses import Response
 from starlette.routing import Route
@@ -29,6 +31,15 @@
 _CONSOLE_HANDLER_ATTR = "_agentserver_console"
 
 
+class _PlatformHeaderMiddleware(BaseHTTPMiddleware):
+    """Middleware that adds x-platform-server identity header to all responses."""
+
+    async def dispatch(self, request: Request, call_next):  # type: ignore[no-untyped-def, override]
+        response = await call_next(request)
+        response.headers["x-platform-server"] = _PLATFORM_SERVER_VALUE
+        return response
+
+
 class AgentServer:
     """Agent server host framework with built-in protocol endpoints.
 
@@ -284,14 +295,11 @@ async def _lifespan(_app: Starlette) -> AsyncGenerator[None, None]:  # noqa: RUF
             Route("/healthy", self._healthy_endpoint, methods=["GET"], name="healthy"),
         )
 
-        self._app = Starlette(routes=routes, lifespan=_lifespan)
-
-        # Add x-platform-server identity header to all responses.
-        @self._app.middleware("http")
-        async def _add_platform_header(request: Request, call_next):  # type: ignore[no-untyped-def]
-            response = await call_next(request)
-            response.headers["x-platform-server"] = _PLATFORM_SERVER_VALUE
-            return response
+        self._app = Starlette(
+            routes=routes,
+            lifespan=_lifespan,
+            middleware=[Middleware(_PlatformHeaderMiddleware)],
+        )
 
     # ------------------------------------------------------------------
     # Health endpoint

sdk/agentserver/azure-ai-agentserver-responses/azure/ai/agentserver/responses/hosting/__init__.py

@@ -2,7 +2,7 @@
 # Licensed under the MIT license.
 """HTTP hosting, routing, and request orchestration for the Responses server."""
 
-from ._routing import map_responses_server
+from ._routing import ResponseHandler
 from ._observability import (
     CreateSpan,
     CreateSpanHook,
@@ -23,7 +23,7 @@
 )
 
 __all__ = [
-    "map_responses_server",
+    "ResponseHandler",
     "CreateSpan",
     "CreateSpanHook",
     "InMemoryCreateSpanHook",