Enabled passing postgres connection details (#100)

qdrant · Mar 7, 2024 · 20b9cf4 · 20b9cf4
1 parent 0d20796
commit 20b9cf4
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/engine/clients/pgvector/config.py b/engine/clients/pgvector/config.py
@@ -1,7 +1,9 @@
-PGVECTOR_PORT = 9200
-PGVECTOR_DB = "postgres"
-PGVECTOR_USER = "postgres"
-PGVECTOR_PASSWORD = "passwd"
+import os
+
+PGVECTOR_PORT = int(os.getenv("PGVECTOR_PORT", 9200))
+PGVECTOR_DB = os.getenv("PGVECTOR_DB", "postgres")
+PGVECTOR_USER = os.getenv("PGVECTOR_USER", "postgres")
+PGVECTOR_PASSWORD = os.getenv("PGVECTOR_PASSWORD", "passwd")
 
 
 def get_db_config(host, connection_params):