added raw dada logs (#2736)
This commit is contained in:
committed by
GitHub
parent
fe38488a40
commit
3e3522fc4c
@@ -15,6 +15,7 @@ from starlette_context.plugins.base import Plugin
|
||||
from skyvern.config import settings
|
||||
from skyvern.exceptions import SkyvernHTTPException
|
||||
from skyvern.forge import app as forge_app
|
||||
from skyvern.forge.request_logging import log_raw_request_middleware
|
||||
from skyvern.forge.sdk.core import skyvern_context
|
||||
from skyvern.forge.sdk.core.skyvern_context import SkyvernContext
|
||||
from skyvern.forge.sdk.db.exceptions import NotFoundError
|
||||
@@ -113,6 +114,10 @@ def get_agent_app() -> FastAPI:
|
||||
finally:
|
||||
skyvern_context.reset()
|
||||
|
||||
@app.middleware("http")
|
||||
async def raw_request_logging(request: Request, call_next: Callable[[Request], Awaitable[Response]]) -> Response:
|
||||
return await log_raw_request_middleware(request, call_next)
|
||||
|
||||
if settings.ADDITIONAL_MODULES:
|
||||
for module in settings.ADDITIONAL_MODULES:
|
||||
LOG.info("Loading additional module to set up api app", module=module)
|
||||
|
||||
Reference in New Issue
Block a user