Fix for test database port

vanna-ai · Oct 1, 2024 · 85586ac · 85586ac
1 parent 8cdfe58
commit 85586ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_pgvector.py b/tests/test_pgvector.py
@@ -13,7 +13,7 @@
 def get_vanna_connection_string():
     server = os.environ.get("PG_SERVER")
     driver = "psycopg"
-    port = 5434
+    port = os.environ.get("PG_PORT", 5432)
     database = os.environ.get("PG_DATABASE")
     username = os.environ.get("PG_USERNAME")
     password = os.environ.get("PG_PASSWORD")