From 2a50e0b3b68a84e22f6b06fec83286664b64258b Mon Sep 17 00:00:00 2001 From: Raoul Scalise <36519284+scaliseraoul@users.noreply.github.com> Date: Thu, 27 Feb 2025 16:51:15 +0100 Subject: [PATCH 1/2] Update pandasai/helpers/sql_sanitizer.py Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com> --- pandasai/helpers/sql_sanitizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pandasai/helpers/sql_sanitizer.py b/pandasai/helpers/sql_sanitizer.py index 1787921e9..089761eaa 100644 --- a/pandasai/helpers/sql_sanitizer.py +++ b/pandasai/helpers/sql_sanitizer.py @@ -99,7 +99,7 @@ def is_sql_query_safe(query: str, dialect: str = "postgres") -> bool: return False -def is_sql_query(query): +def is_sql_query(query: str) -> bool: # Define SQL patterns with context to avoid standalone keyword matches sql_patterns = [ r"\bSELECT\b.*\bFROM\b", From 077ac9ccd1a170b0282392dbb8e26f88998c29cf Mon Sep 17 00:00:00 2001 From: Raoul Scalise <36519284+scaliseraoul@users.noreply.github.com> Date: Thu, 27 Feb 2025 16:53:18 +0100 Subject: [PATCH 2/2] Update pandasai/query_builders/sql_parser.py Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com> --- pandasai/query_builders/sql_parser.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pandasai/query_builders/sql_parser.py b/pandasai/query_builders/sql_parser.py index 9a70dc442..1b1c2f586 100644 --- a/pandasai/query_builders/sql_parser.py +++ b/pandasai/query_builders/sql_parser.py @@ -57,7 +57,7 @@ def transform_node(node): return transformed.sql(pretty=True) @staticmethod - def transpile_sql_dialect(query: str, to_dialect, from_dialect=None): + def transpile_sql_dialect(query: str, to_dialect: str, from_dialect: Optional[str] = None): placeholder = "___PLACEHOLDER___" query = query.replace("%s", placeholder) query = (