Added logging option for server process, fixed bug on agent list with bad type

This commit is contained in:
cpacker
2024-01-15 12:55:51 -08:00
parent 434a4b2f99
commit 112e29b3a6

View File

@@ -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