From 112e29b3a65fb8164a903e44101afee8ade260ee Mon Sep 17 00:00:00 2001 From: cpacker Date: Mon, 15 Jan 2024 12:55:51 -0800 Subject: [PATCH] Added logging option for server process, fixed bug on agent list with bad type --- memgpt/cli/cli.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/memgpt/cli/cli.py b/memgpt/cli/cli.py index 794f5ecb..f31b871e 100644 --- a/memgpt/cli/cli.py +++ b/memgpt/cli/cli.py @@ -219,9 +219,23 @@ def server( type: ServerChoice = typer.Option("rest", help="Server to run"), port: int = typer.Option(None, help="Port to run the server on"), host: str = typer.Option(None, help="Host to run the server on (default to localhost)"), + debug: bool = typer.Option(True, help="Turn debugging output on"), ): """Launch a MemGPT server process""" + if debug: + from memgpt.server.server import logger as server_logger + + # Set the logging level + server_logger.setLevel(logging.DEBUG) + # Create a StreamHandler + stream_handler = logging.StreamHandler() + # Set the formatter (optional) + formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s") + stream_handler.setFormatter(formatter) + # Add the handler to the logger + server_logger.addHandler(stream_handler) + if type == ServerChoice.rest_api: import uvicorn from memgpt.server.rest_api.server import app