Added logging option for server process, fixed bug on agent list with bad type
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user