From b35bb19b31625b819ff5455fedc31505f4b47bf5 Mon Sep 17 00:00:00 2001
From: vbarda <vadym@langchain.dev>
Date: Tue, 18 Jun 2024 20:38:40 -0400
Subject: [PATCH] sync again

---
 libs/cli/langgraph_cli/cli.py            | 2 ++
 libs/cli/langgraph_cli/docker.py         | 2 --
 libs/cli/tests/unit_tests/test_cli.py    | 2 --
 libs/cli/tests/unit_tests/test_docker.py | 5 -----
 4 files changed, 2 insertions(+), 9 deletions(-)

diff --git a/libs/cli/langgraph_cli/cli.py b/libs/cli/langgraph_cli/cli.py
index 94b1d8a6a..cc4cd6b73 100644
--- a/libs/cli/langgraph_cli/cli.py
+++ b/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...")
 
diff --git a/libs/cli/langgraph_cli/docker.py b/libs/cli/langgraph_cli/docker.py
index d7627015f..658e1268e 100644
--- a/libs/cli/langgraph_cli/docker.py
+++ b/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:
diff --git a/libs/cli/tests/unit_tests/test_cli.py b/libs/cli/tests/unit_tests/test_cli.py
index 4d605443e..d316228a9 100644
--- a/libs/cli/tests/unit_tests/test_cli.py
+++ b/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:
diff --git a/libs/cli/tests/unit_tests/test_docker.py b/libs/cli/tests/unit_tests/test_docker.py
index 416c22889..4d5e09b94 100644
--- a/libs/cli/tests/unit_tests/test_docker.py
+++ b/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: