Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

allow passing custom host to uvicorn (#618)

Browse Source
This commit is contained in:
Charles Packer
2023-12-14 13:01:33 -08:00
committed by GitHub
parent 490c0ccd4a
commit d51c61e91c
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
memgpt/cli/cli.py
View File

@@ -31,7 +31,9 @@ class ServerChoice(Enum):
def server(
type: ServerChoice = typer.Option("rest", help="Server to run"), port: int = typer.Option(None, help="Port to run the server on")
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)"),
):
"""Launch a MemGPT server process"""
@@ -44,7 +46,10 @@ def server(
script_dir = script_path.parent
server_directory = os.path.join(script_dir.parent, "server", "rest_api")
command = f"uvicorn server:app --reload --port {port}"
if host is None:
command = f"uvicorn server:app --reload --port {port}"
else:
command = f"uvicorn server:app --reload --port {port} --host {host}"
# Run the command
print(f"Running REST server: {command} (inside {server_directory})")