From 2eedd2ed846642f584380890bde49d2baef1e08a Mon Sep 17 00:00:00 2001 From: Shubham <25881429+shoebham@users.noreply.github.com> Date: Wed, 28 Aug 2024 19:59:54 +0530 Subject: [PATCH] fix bug in malicious code scan --- pandasai/pipelines/chat/code_cleaning.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pandasai/pipelines/chat/code_cleaning.py b/pandasai/pipelines/chat/code_cleaning.py index a6e047b18..759813b14 100644 --- a/pandasai/pipelines/chat/code_cleaning.py +++ b/pandasai/pipelines/chat/code_cleaning.py @@ -176,7 +176,7 @@ def _is_malicious_code(self, code) -> bool: "(chr", "b64decode", ] - return any(module in code for module in dangerous_modules) + return any(re.search(r'\b'+re.escape(module)+r'\b',code) for module in dangerous_modules) def _is_jailbreak(self, node: ast.stmt) -> bool: """