sync changes

libs/cli/tests/unit_tests/test_cli.py

@@ -3,8 +3,7 @@ import pathlib
 from langgraph_cli.cli import prepare_args_and_stdin
 from langgraph_cli.config import Config, validate_config
 from langgraph_cli.docker import DEFAULT_POSTGRES_URI, DockerCapabilities, Version
-
-from .helpers import clean_empty_lines
+from langgraph_cli.util import clean_empty_lines
 
 DEFAULT_DOCKER_CAPABILITIES = DockerCapabilities(
     version_docker=Version(26, 1, 1),
@@ -81,6 +80,7 @@ services:
         environment:
             POSTGRES_URI: {DEFAULT_POSTGRES_URI}
         healthcheck:
+            test: python /api/healthcheck.py
             interval: 60s
             start_interval: 1s
             start_period: 10s

libs/cli/tests/unit_tests/test_config.py

@@ -5,8 +5,7 @@ import click
 import pytest
 
 from langgraph_cli.config import config_to_compose, config_to_docker, validate_config
-
-from .helpers import clean_empty_lines
+from langgraph_cli.util import clean_empty_lines
 
 PATH_TO_CONFIG = pathlib.Path("tests/unit_tests/test_config.json")
 

libs/cli/tests/unit_tests/test_docker.py

@@ -4,7 +4,7 @@ from langgraph_cli.docker import (
     Version,
     compose,
 )
-from tests.unit_tests.helpers import clean_empty_lines
+from langgraph_cli.util import clean_empty_lines
 
 DEFAULT_DOCKER_CAPABILITIES = DockerCapabilities(
     version_docker=Version(26, 1, 1),
@@ -24,9 +24,6 @@ def test_compose_with_no_debugger_and_custom_db():
         restart: on-failure
         ports:
             - "{port}:8000"
-        depends_on:
-            langgraph-postgres:
-                condition: service_healthy
         environment:
             POSTGRES_URI: {custom_postgres_uri}"""
     assert clean_empty_lines(actual_compose_str) == expected_compose_str
@@ -45,12 +42,10 @@ def test_compose_with_no_debugger_and_custom_db_with_healthcheck():
         restart: on-failure
         ports:
             - "{port}:8000"
-        depends_on:
-            langgraph-postgres:
-                condition: service_healthy
         environment:
             POSTGRES_URI: {custom_postgres_uri}
         healthcheck:
+            test: python /api/healthcheck.py
             interval: 60s
             start_interval: 1s
             start_period: 10s"""
@@ -70,9 +65,6 @@ def test_compose_with_debugger_and_custom_db():
         restart: on-failure
         ports:
             - "{port}:8000"
-        depends_on:
-            langgraph-postgres:
-                condition: service_healthy
         environment:
             POSTGRES_URI: {custom_postgres_uri}"""
     assert clean_empty_lines(actual_compose_str) == expected_compose_str