sync again

libs/cli/langgraph_cli/cli.py

@@ -207,6 +207,8 @@ def up(
             args.append("--watch")
         if wait:
             args.append("--wait")
+        else:
+            args.append("--abort-on-container-exit")
         # run docker compose
         set("Building...")
 

libs/cli/langgraph_cli/docker.py

@@ -16,7 +16,6 @@ DEFAULT_POSTGRES_URI = (
 DB = """
     langgraph-postgres:
         image: postgres:16
-        restart: on-failure
         ports:
             - "5433:5432"
         environment:
@@ -150,7 +149,6 @@ def compose(
 {db}
 {DEBUGGER.format(debugger_port=debugger_port) if debugger_port else ""}
     langgraph-api:
-        restart: on-failure
         ports:
             - "{port}:8000\""""
     if include_db:

libs/cli/tests/unit_tests/test_cli.py

@@ -46,7 +46,6 @@ def test_prepare_args_and_stdin():
 services:
     langgraph-postgres:
         image: postgres:16
-        restart: on-failure
         ports:
             - "5433:5432"
         environment:
@@ -71,7 +70,6 @@ services:
             langgraph-postgres:
                 condition: service_healthy
     langgraph-api:
-        restart: on-failure
         ports:
             - "8000:8000"
         depends_on:

libs/cli/tests/unit_tests/test_docker.py

@@ -21,7 +21,6 @@ def test_compose_with_no_debugger_and_custom_db():
     )
     expected_compose_str = f"""services:
     langgraph-api:
-        restart: on-failure
         ports:
             - "{port}:8000"
         environment:
@@ -39,7 +38,6 @@ def test_compose_with_no_debugger_and_custom_db_with_healthcheck():
     )
     expected_compose_str = f"""services:
     langgraph-api:
-        restart: on-failure
         ports:
             - "{port}:8000"
         environment:
@@ -62,7 +60,6 @@ def test_compose_with_debugger_and_custom_db():
     )
     expected_compose_str = f"""services:
     langgraph-api:
-        restart: on-failure
         ports:
             - "{port}:8000"
         environment:
@@ -79,7 +76,6 @@ def test_compose_with_debugger_and_default_db():
 services:
     langgraph-postgres:
         image: postgres:16
-        restart: on-failure
         ports:
             - "5433:5432"
         environment:
@@ -95,7 +91,6 @@ services:
             retries: 5
             interval: 5s
     langgraph-api:
-        restart: on-failure
         ports:
             - "{port}:8000"
         depends_on: