feat: add modelling layer

backend/app/connectors/postgres.py

@@ -22,6 +22,9 @@ class PostgresConnector:
             return [dict(row._mapping) for row in result]
 
     def get_schema(self):
+        if self.engine.dialect.name == "sqlite":
+            return self._get_sqlite_schema()
+
         query = """
         SELECT table_name, column_name, data_type
         FROM information_schema.columns
@@ -35,12 +38,27 @@ class PostgresConnector:
                 table = row['table_name']
                 if table not in schema:
                     schema[table] = []
-                schema[table].append(f"{row['column_name']} ({row['data_type']})")
+                schema[table].append({"name": row['column_name'], "type": row['data_type']})
             return schema
         except Exception as e:
             print(f"Error getting schema: {e}")
             return {}
 
+    def _get_sqlite_schema(self):
+        try:
+            from sqlalchemy import inspect
+            inspector = inspect(self.engine)
+            schema = {}
+            for table_name in inspector.get_table_names():
+                columns = []
+                for col in inspector.get_columns(table_name):
+                    columns.append({"name": col['name'], "type": str(col['type'])})
+                schema[table_name] = columns
+            return schema
+        except Exception as e:
+            print(f"Error getting SQLite schema: {e}")
+            return {}
+
     def test_connection(self) -> bool:
         try:
             with self.engine.connect() as connection: