Setup: fix key error (#84)

cohere-ai · May 1, 2024 · 3dd9411 · 3dd9411
1 parent 44845aa
commit 3dd9411
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/backend/models/database.py b/src/backend/models/database.py
@@ -8,8 +8,9 @@
 
 load_dotenv()
 
-SQLALCHEMY_DATABASE_URL = os.environ["DATABASE_URL"]
-
+SQLALCHEMY_DATABASE_URL = os.environ.get(
+    "DATABASE_URL", "postgresql+psycopg2://postgres:postgres@db:5432"
+)
 engine = create_engine(
     SQLALCHEMY_DATABASE_URL, pool_size=5, max_overflow=10, pool_timeout=30
 )