fix(view_loader): avoid lower casing the column name (#1703)

* fix(view_loader): avoid lower casing the column name

* fix: remove print statements

Author: ArslanSaleem

pandasai/helpers/sql_sanitizer.py

@@ -2,10 +2,18 @@
 import re
 
 import sqlglot
+from sqlglot import parse_one
+from sqlglot.optimizer.qualify_columns import quote_identifiers
 
 
 def sanitize_view_column_name(relation_name: str) -> str:
-    return ".".join(list(map(sanitize_sql_table_name, relation_name.split("."))))
+    return (
+        parse_one(
+            ".".join(list(map(sanitize_sql_table_name, relation_name.split("."))))
+        )
+        .transform(quote_identifiers)
+        .sql()
+    )
 
 
 def sanitize_sql_table_name(table_name: str) -> str:

pandasai/query_builders/view_query_builder.py

@@ -24,13 +24,12 @@ def __init__(
 
     @staticmethod
     def normalize_view_column_name(name: str) -> str:
-        return normalize_identifiers(parse_one(sanitize_view_column_name(name))).sql()
+        return sanitize_view_column_name(name)
 
     @staticmethod
     def normalize_view_column_alias(name: str) -> str:
-        return normalize_identifiers(
-            sanitize_view_column_name(name).replace(".", "_")
-        ).sql()
+        col_name = name.replace(".", "_")
+        return sanitize_view_column_name(col_name)
 
     def _get_group_by_columns(self) -> list[str]:
         """Get the group by columns with proper view column aliasing."""

tests/unit_tests/helpers/test_sql_sanitizer.py

@@ -25,7 +25,7 @@ def test_sanitize_file_name_long_name(self):
 
     def test_sanitize_relation_name_valid(self):
         relation = "dataset-name.column"
-        expected = "dataset_name.column"
+        expected = '"dataset_name"."column"'
         assert sanitize_view_column_name(relation) == expected
 
     def test_safe_select_query(self):