bugfix

defog-ai · Dec 9, 2024 · eba77e9 · eba77e9
1 parent 815b8d5
commit eba77e9
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/defog/generate_schema.py b/defog/generate_schema.py
@@ -72,11 +72,18 @@ def generate_postgres_schema(
         else:
             schema = "public"
         cur.execute(
-            """SELECT
-                CAST(column_name AS TEXT),
-                CAST(data_type AS TEXT),
-                col_descripton(format('%s.%s', table_schema, table_name)::regclass::oid, ordinal_position) AS column_description
-            FROM information_schema.columns WHERE table_name::text = %s AND table_schema = %s;""",
+            """
+            SELECT 
+                CAST(column_name AS TEXT), 
+                CAST(data_type AS TEXT), 
+                col_description(
+                    FORMAT('%%s.%%s', table_schema, table_name)::regclass::oid, 
+                    ordinal_position
+                ) AS column_description 
+            FROM information_schema.columns 
+            WHERE table_name::text = %s 
+            AND table_schema = %s;
+            """,
             (
                 table_name,
                 schema,