From 319747605187d033d569caf3c267c8f6d9428ca3 Mon Sep 17 00:00:00 2001
From: Raoul Scalise <36519284+scaliseraoul@users.noreply.github.com>
Date: Wed, 26 Feb 2025 16:02:36 +0100
Subject: [PATCH] fix(expression): modulo operation issues (#1637)

---
 pandasai/data_loader/sql_loader.py  | 2 ++
 pandasai/data_loader/view_loader.py | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/pandasai/data_loader/sql_loader.py b/pandasai/data_loader/sql_loader.py
index fc50afd45..aa7b0845d 100644
--- 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
index 5a778462b..5dbf45d53 100644
--- 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: