diff --git a/tests/remote_server/remote_server.py b/tests/remote_server/remote_server.py index 4ac4fd6c..c7655671 100644 --- a/tests/remote_server/remote_server.py +++ b/tests/remote_server/remote_server.py @@ -13,6 +13,9 @@ app = FastAPI() +# Configure logging for the remote server (required for INFO-level logs to be emitted) +logging.basicConfig(level=logging.INFO, format="%(name)s - %(levelname)s - %(message)s") + # Attach Fireworks tracing handler to root logger fireworks_handler = FireworksTracingHttpHandler() logging.getLogger().addHandler(fireworks_handler) diff --git a/tests/remote_server/remote_server_multi_turn.py b/tests/remote_server/remote_server_multi_turn.py index 88815f09..971b6251 100644 --- a/tests/remote_server/remote_server_multi_turn.py +++ b/tests/remote_server/remote_server_multi_turn.py @@ -12,6 +12,9 @@ app = FastAPI() +# Configure logging for the remote server (required for INFO-level logs to be emitted) +logging.basicConfig(level=logging.INFO, format="%(name)s - %(levelname)s - %(message)s") + # Attach Fireworks tracing handler to root logger fireworks_handler = FireworksTracingHttpHandler() logging.getLogger().addHandler(fireworks_handler)