From 0098d24762b2e988a55589a276d0d526d8de5818 Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Sat, 27 Sep 2025 11:15:13 -0700 Subject: [PATCH] use llama_stack.log --- tests/common/mcp.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/common/mcp.py b/tests/common/mcp.py index 91012753e..f65f7c952 100644 --- a/tests/common/mcp.py +++ b/tests/common/mcp.py @@ -156,7 +156,6 @@ def make_mcp_server(required_auth_token: str | None = None, tools: dict[str, Cal :param required_auth_token: Optional auth token required for access :param tools: Dictionary of tool_name -> tool_function. If None, uses default tools. """ - import logging import threading import time @@ -168,6 +167,8 @@ def make_mcp_server(required_auth_token: str | None = None, tools: dict[str, Cal from starlette.responses import Response from starlette.routing import Mount, Route + from llama_stack.log import get_logger + server = FastMCP("FastMCP Test Server", log_level="WARNING") tools = tools or default_tools() @@ -212,7 +213,7 @@ def make_mcp_server(required_auth_token: str | None = None, tools: dict[str, Cal return sock.getsockname()[1] port = get_open_port() - logger = logging.getLogger(__name__) + logger = get_logger(__name__, category="tests::mcp") # make uvicorn logs be less verbose config = uvicorn.Config(app, host="0.0.0.0", port=port, log_level="warning")