From 83de0c799551cb594c14eb4d38d9561b5a1964de Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Fri, 6 Dec 2024 11:25:53 -0800 Subject: [PATCH] disable fastapi access logs --- llama_stack/distribution/server/server.py | 2 +- .../meta_reference/console_span_processor.py | 12 +++++++----- 2 files changed, 8 insertions(+), 6 deletions(-) diff --git a/llama_stack/distribution/server/server.py b/llama_stack/distribution/server/server.py index 43e9c0706..a0c0e5423 100644 --- a/llama_stack/distribution/server/server.py +++ b/llama_stack/distribution/server/server.py @@ -347,7 +347,7 @@ def main(): listen_host = ["::", "0.0.0.0"] if not args.disable_ipv6 else "0.0.0.0" print(f"Listening on {listen_host}:{args.port}") - uvicorn.run(app, host=listen_host, port=args.port) + uvicorn.run(app, host=listen_host, port=args.port, access_log=False) if __name__ == "__main__": diff --git a/llama_stack/providers/inline/telemetry/meta_reference/console_span_processor.py b/llama_stack/providers/inline/telemetry/meta_reference/console_span_processor.py index 9a7661469..2031fc88e 100644 --- a/llama_stack/providers/inline/telemetry/meta_reference/console_span_processor.py +++ b/llama_stack/providers/inline/telemetry/meta_reference/console_span_processor.py @@ -39,7 +39,9 @@ class ConsoleSpanProcessor(SpanProcessor): )[:-3] print( - f"{COLORS['dim']}{timestamp} " f"[START] " f"{span.name}{COLORS['reset']}" + f"{timestamp} " + f"{COLORS['bold']}[START]{COLORS['reset']} " + f"{COLORS['dim']}{span.name}{COLORS['reset']}" ) def on_end(self, span: ReadableSpan) -> None: @@ -50,15 +52,15 @@ class ConsoleSpanProcessor(SpanProcessor): "%H:%M:%S.%f" )[:-3] - span_context = f"{COLORS['dim']}{timestamp} [END] {span.name}" + span_context = f"{timestamp} {COLORS['bold']}[END]{COLORS['reset']} {COLORS['dim']}{span.name}" if span.status.status_code == 2: # ERROR - span_context += f" {COLORS['red']}[ERROR]{COLORS['reset']}" + span_context += f"{COLORS['reset']} {COLORS['red']}[ERROR]{COLORS['reset']}" elif span.status.status_code != 0: # UNSET - span_context += f" {COLORS['dim']}[{span.status.status_code}]" + span_context += f"{COLORS['reset']} [{span.status.status_code}]" duration_ms = (span.end_time - span.start_time) / 1e6 - span_context += f" ({duration_ms:.2f}ms){COLORS['reset']}" + span_context += f"{COLORS['reset']} ({duration_ms:.2f}ms)" print(span_context)