feat(cli): add debugger host CLI flag to specify default base URL

libs/cli/langgraph_cli/cli.py

@@ -134,6 +134,12 @@
     type=int,
     help="Pull the debugger image locally and serve the UI on specified port",
 )
+OPT_DEBUGGER_HOST = click.option(
+    "--debugger-host",
+    type=str,
+    help="Host for the debugger UI. Defaults to localhost",
+)
+
 OPT_POSTGRES_URI = click.option(
     "--postgres-uri",
     help="Postgres URI to use for the database. Defaults to launching a local database",
@@ -152,6 +158,7 @@ def cli():
 @OPT_CONFIG
 @OPT_VERBOSE
 @OPT_DEBUGGER_PORT
+@OPT_DEBUGGER_HOST
 @OPT_WATCH
 @OPT_POSTGRES_URI
 @click.option(
@@ -173,6 +180,7 @@ def up(
     wait: bool,
     verbose: bool,
     debugger_port: Optional[int],
+    debugger_host: Optional[str],
     postgres_uri: Optional[str],
 ):
     click.secho("Starting LangGraph API server...", fg="green")
@@ -193,6 +201,7 @@ def up(
             watch=watch,
             verbose=verbose,
             debugger_port=debugger_port,
+            debugger_host=debugger_host,
             postgres_uri=postgres_uri,
         )
         # add up + options
@@ -448,13 +457,15 @@ def prepare_args_and_stdin(
     port: int,
     watch: bool,
     debugger_port: Optional[int] = None,
+    debugger_host: Optional[str] = None,
     postgres_uri: Optional[str] = None,
 ):
     # prepare args
     stdin = langgraph_cli.docker.compose(
         capabilities,
         port=port,
         debugger_port=debugger_port,
+        debugger_host=debugger_host,
         postgres_uri=postgres_uri,
     )
     args = [
@@ -486,6 +497,7 @@ def prepare(
     watch: bool,
     verbose: bool,
     debugger_port: Optional[int] = None,
+    debugger_host: Optional[str] = None,
     postgres_uri: Optional[str] = None,
 ):
     with open(config_path) as f:
@@ -509,6 +521,7 @@ def prepare(
         port=port,
         watch=watch,
         debugger_port=debugger_port,
+        debugger_host=debugger_host or f"http://127.0.0.1:{DEFAULT_PORT}",
         postgres_uri=postgres_uri,
     )
     return args, stdin

libs/cli/langgraph_cli/docker.py

@@ -32,18 +32,6 @@
 """
 
 
-DEBUGGER = """
-    langgraph-debugger:
-        image: langchain/langgraph-debugger
-        restart: on-failure
-        ports:
-            - "{debugger_port}:3968"
-        depends_on:
-            langgraph-postgres:
-                condition: service_healthy
-"""
-
-
 class Version(NamedTuple):
     major: int
     minor: int
@@ -117,11 +105,36 @@ def check_capabilities(runner) -> DockerCapabilities:
     )
 
 
+def debugger_compose(*, port: Optional[int] = None, host: Optional[str] = None) -> str:
+    if port is None:
+        return ""
+
+    compose_str = """
+    langgraph-debugger:
+        image: langchain/langgraph-debugger
+        restart: on-failure
+        depends_on:
+            langgraph-postgres:
+                condition: service_healthy
+        ports:
+            - "{port}:3968"
+"""
+
+    if host:
+        compose_str += """
+        environment:
+            VITE_STUDIO_LOCAL_GRAPH_URL: {host}
+"""
+
+    return compose_str.format(port=port, host=host)
+
+
 def compose(
     capabilities: DockerCapabilities,
     *,
     port: int,
     debugger_port: Optional[int] = None,
+    debugger_host: Optional[str] = None,
     # postgres://user:password@host:port/database?option=value
     postgres_uri: Optional[str] = None,
 ) -> str:
@@ -151,7 +164,7 @@ def compose(
 
     compose_str = f"""{volumes}services:
 {db}
-{DEBUGGER.format(debugger_port=debugger_port) if debugger_port else ""}
+{debugger_compose(port=debugger_port, host=debugger_host)}
     langgraph-api:
         ports:
             - "{port}:8000\""""

libs/cli/tests/unit_tests/test_cli.py

@@ -20,6 +20,7 @@ def test_prepare_args_and_stdin():
     )
     port = 8000
     debugger_port = 8001
+    debugger_host = "http://127.0.0.1:8000"
 
     actual_args, actual_stdin = prepare_args_and_stdin(
         capabilities=DEFAULT_DOCKER_CAPABILITIES,
@@ -28,6 +29,7 @@ def test_prepare_args_and_stdin():
         docker_compose="custom-docker-compose.yml",
         port=port,
         debugger_port=debugger_port,
+        debugger_host=debugger_host,
         watch=True,
     )
 
@@ -63,11 +65,13 @@ def test_prepare_args_and_stdin():
     langgraph-debugger:
         image: langchain/langgraph-debugger
         restart: on-failure
-        ports:
-            - "{debugger_port}:3968"
         depends_on:
             langgraph-postgres:
                 condition: service_healthy
+        ports:
+            - "{debugger_port}:3968"
+        environment:
+            VITE_STUDIO_LOCAL_GRAPH_URL: {debugger_host}
     langgraph-api:
         ports:
             - "8000:8000"