diff --git a/studio/backend/loggers/handlers.py b/studio/backend/loggers/handlers.py index 9ffd0b0bd8..9cade6178a 100644 --- a/studio/backend/loggers/handlers.py +++ b/studio/backend/loggers/handlers.py @@ -22,12 +22,14 @@ import structlog logger = structlog.get_logger(__name__) # Paths and suffixes excluded from request logging. -_EXCLUDED_PATHS = frozenset({ - "/api/train/status", - "/api/train/metrics", - "/api/train/hardware", - "/api/system", -}) +_EXCLUDED_PATHS = frozenset( + { + "/api/train/status", + "/api/train/metrics", + "/api/train/hardware", + "/api/system", + } +) _EXCLUDED_SUFFIXES = (".png", ".jpg", ".jpeg", ".ico", ".woff", ".woff2", ".ttf") @@ -64,10 +66,10 @@ class LoggingMiddleware: method = scope.get("method", "") logger.error( "request_failed", - path=path, - method=method, - error=str(e), - exc_info=True, + path = path, + method = method, + error = str(e), + exc_info = True, ) raise @@ -83,10 +85,10 @@ class LoggingMiddleware: method = scope.get("method", "") logger.info( "request_completed", - method=method, - path=path, - status_code=status_code, - process_time_ms=round(process_time, 2), + method = method, + path = path, + status_code = status_code, + process_time_ms = round(process_time, 2), ) diff --git a/studio/backend/routes/inference.py b/studio/backend/routes/inference.py index 9e8b359b7e..3d59a70b6f 100644 --- a/studio/backend/routes/inference.py +++ b/studio/backend/routes/inference.py @@ -28,6 +28,7 @@ import re as _re _queue_sentinel = object() _queue_error = object() + async def _run_gen_to_queue(gen, cancel_event): """Run a synchronous generator in a background thread, yield items via asyncio.Queue.""" loop = asyncio.get_running_loop() @@ -44,7 +45,7 @@ async def _run_gen_to_queue(gen, cancel_event): finally: loop.call_soon_threadsafe(q.put_nowait, _queue_sentinel) - thread = threading.Thread(target=_producer, daemon=True, name="sse-gen") + thread = threading.Thread(target = _producer, daemon = True, name = "sse-gen") thread.start() while True: