diff --git a/libs/cli/langgraph_cli/docker.py b/libs/cli/langgraph_cli/docker.py index 5d4c1cddd..1b8d542f7 100644 --- a/libs/cli/langgraph_cli/docker.py +++ b/libs/cli/langgraph_cli/docker.py @@ -170,13 +170,14 @@ def compose_as_dict( # Add Postgres service before langgraph-api if it is needed if include_db: services["langgraph-postgres"] = { - "image": "postgres:16", + "image": "pgvector/pgvector:pg16", "ports": ['"5433:5432"'], "environment": { "POSTGRES_DB": "postgres", "POSTGRES_USER": "postgres", "POSTGRES_PASSWORD": "postgres", }, + "command": ["postgres", "-c", "shared_preload_libraries=vector"], "volumes": ["langgraph-data:/var/lib/postgresql/data"], "healthcheck": { "test": "pg_isready -U postgres", diff --git a/libs/cli/pyproject.toml b/libs/cli/pyproject.toml index bcb314066..d9f26b493 100644 --- a/libs/cli/pyproject.toml +++ b/libs/cli/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "langgraph-cli" -version = "0.1.68" +version = "0.1.69" description = "CLI for interacting with LangGraph API" authors = [] license = "MIT" diff --git a/libs/cli/tests/unit_tests/cli/test_cli.py b/libs/cli/tests/unit_tests/cli/test_cli.py index 52e2b682f..3b0e0dae6 100644 --- a/libs/cli/tests/unit_tests/cli/test_cli.py +++ b/libs/cli/tests/unit_tests/cli/test_cli.py @@ -79,13 +79,17 @@ services: timeout: 1s retries: 5 langgraph-postgres: - image: postgres:16 + image: pgvector/pgvector:pg16 ports: - "5433:5432" environment: POSTGRES_DB: postgres POSTGRES_USER: postgres POSTGRES_PASSWORD: postgres + command: + - postgres + - -c + - shared_preload_libraries=vector volumes: - langgraph-data:/var/lib/postgresql/data healthcheck: diff --git a/libs/cli/tests/unit_tests/test_docker.py b/libs/cli/tests/unit_tests/test_docker.py index 781c049c1..355efeb4d 100644 --- a/libs/cli/tests/unit_tests/test_docker.py +++ b/libs/cli/tests/unit_tests/test_docker.py @@ -115,13 +115,17 @@ services: timeout: 1s retries: 5 langgraph-postgres: - image: postgres:16 + image: pgvector/pgvector:pg16 ports: - "5433:5432" environment: POSTGRES_DB: postgres POSTGRES_USER: postgres POSTGRES_PASSWORD: postgres + command: + - postgres + - -c + - shared_preload_libraries=vector volumes: - langgraph-data:/var/lib/postgresql/data healthcheck: