diff --git a/guardrails_api/app.py b/guardrails_api/app.py index eb1dd8a..29e2d44 100644 --- a/guardrails_api/app.py +++ b/guardrails_api/app.py @@ -20,6 +20,7 @@ from starlette.middleware.base import BaseHTTPMiddleware + class RequestInfoMiddleware(BaseHTTPMiddleware): async def dispatch(self, request: Request, call_next): tracer = trace.get_tracer(__name__) diff --git a/tests/cli/test_start.py b/tests/cli/test_start.py index befe21a..3138a66 100644 --- a/tests/cli/test_start.py +++ b/tests/cli/test_start.py @@ -10,6 +10,8 @@ def test_start(mocker): "guardrails_api.cli.start.create_app", return_value=mock_app ) + mocker.patch("uvicorn.run") + from guardrails_api.cli.start import start # pg enabled