From 9d5166a0d3be19132bd6e30637032d559409dfda Mon Sep 17 00:00:00 2001
From: Tat Dat Duong <david@duong.cz>
Date: Sat, 25 Jan 2025 04:16:52 +0100
Subject: [PATCH] Fix tests

---
 libs/cli/tests/unit_tests/cli/test_cli.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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: