Skip to content

Commit 6a56fc0

Browse files
committed
Get requestID and streamID from request body
1 parent e2ad7cf commit 6a56fc0

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

runner/app/live/log.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ def config_logging(*, log_level: int = 0, request_id: str = "", stream_id: str =
3030

3131
def config_logging_fields(handler: logging.Handler, request_id: str, stream_id: str):
3232
formatter = logging.Formatter(
33-
"timestamp=%(asctime)s level=%(levelname)s location=%(filename)s:%(lineno)d:%(funcName)s request_id=%(request_id)s stream_id=%(stream_id)s message=%(message)s",
33+
"timestamp=%(asctime)s level=%(levelname)s location=%(filename)s:%(lineno)d:%(funcName)s gateway_request_id=%(request_id)s stream_id=%(stream_id)s message=%(message)s",
3434
defaults={"request_id": request_id, "stream_id": stream_id},
3535
datefmt="%Y-%m-%d %H:%M:%S",
3636
)

runner/app/routes/live_video_to_video.py

+5
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,11 @@ async def live_video_to_video(
144144
)
145145

146146
try:
147+
if requestID is None:
148+
requestID = params.gateway_request_id
149+
if streamID is None:
150+
streamID = params.stream_id
151+
logger.error(f"AAA live_video_to_video: {requestID} {streamID}")
147152
pipeline(**params.model_dump(), request_id=requestID, stream_id=streamID)
148153
except Exception as e:
149154
if isinstance(e, torch.cuda.OutOfMemoryError):

0 commit comments

Comments
 (0)