INNER CODE UNIT · Python

set_provider_registry

stacklok/codegate · src/codegate/server.py:40

    def set_provider_registry(self, registry: ProviderRegistry):
        self.provider_registry = registry


def init_app(pipeline_factory: PipelineFactory) -> CodeGateServer:
    """Create the FastAPI application."""
    app = CodeGateServer(
        title="CodeGate",
        description=__description__,
        version=__version__,
    )

    @app.middleware("http")
    async def log_user_agent(request: Request, call_next):
        user_agent = request.headers.get("user-agent")
        client_host = request.client.host if request.client else "unknown"
        logger.debug(f"User-Agent header received: {user_agent} from {client_host}")
        response = await call_next(request)

View source record →