Update database/remote_db.py

database/remote_db.py

@@ -1,17 +1,27 @@
-from sqlalchemy import create_engine
+from __future__ import annotations
+
 import os
 
+from sqlalchemy import create_engine
+
 
 DATABASE_URL = os.getenv("DATABASE_URL")
 
 if not DATABASE_URL:
     raise RuntimeError("DATABASE_URL is not set")
 
+DATABASE_URL = DATABASE_URL.strip()
+
+# Cockroach needs the cockroachdb dialect for SQLAlchemy
+if DATABASE_URL.startswith("postgresql://"):
+    DATABASE_URL = "cockroachdb://" + DATABASE_URL[len("postgresql://"):]
+
 engine = create_engine(
     DATABASE_URL,
     pool_pre_ping=True,
     pool_recycle=300,
 )
 
+
 def get_connection():
     return engine.connect()