fixed issue with extracting columns for tables with schema names pre-…

…prended
defog-ai · May 9, 2024 · 11206c5 · 11206c5
1 parent 491a7ca
commit 11206c5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/defog/generate_schema.py b/defog/generate_schema.py
@@ -49,6 +49,8 @@ def generate_postgres_schema(
     # get the columns for each table
     for schema in schemas:
         for table_name in tables:
+            if "." in table_name:
+                _, table_name = table_name.split(".", 1)
             cur.execute(
                 "SELECT CAST(column_name AS TEXT), CAST(data_type AS TEXT) FROM information_schema.columns WHERE table_name::text = %s AND table_schema = %s;",
                 (