fix(expression): modulo operation issues (#1637)

sinaptik-ai · Feb 26, 2025 · 3197476 · 3197476
1 parent 3cb04f1
commit 3197476
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/pandasai/data_loader/sql_loader.py b/pandasai/data_loader/sql_loader.py
@@ -48,6 +48,8 @@ def execute_query(self, query: str, params: Optional[list] = None) -> pd.DataFra
                 "The SQL query is deemed unsafe and will not be executed."
             )
         try:
+            if params:
+                query = query.replace(" % ", " %% ")
             return load_function(connection_info, query, params)
 
         except ModuleNotFoundError as e:

diff --git a/pandasai/data_loader/view_loader.py b/pandasai/data_loader/view_loader.py
@@ -103,6 +103,8 @@ def execute_query(self, query: str, params: Optional[list] = None) -> pd.DataFra
                 "The SQL query is deemed unsafe and will not be executed."
             )
         try:
+            if params:
+                query = query.replace(" % ", " %% ")
             return load_function(connection_info, query, params)
 
         except ModuleNotFoundError as e: