1st iter

backend/src/core/database.py

@@ -12,8 +12,9 @@
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker, Session
 from backend.src.models.mapping import Base
-# Import TaskRecord to ensure it's registered with Base
+# Import models to ensure they're registered with Base
 from backend.src.models.task import TaskRecord
+from backend.src.models.connection import ConnectionConfig
 import os
 # [/SECTION]