From 6dd969129df247a2feed47bf67a242fe7e9c77bc Mon Sep 17 00:00:00 2001
From: Rodrigo Agundez <rodrigo.agundez@dyson.com>
Date: Thu, 5 Jun 2025 23:21:04 +0800
Subject: [PATCH] Add option to blacklist modules in code interpreter

Co-authored-by: KG <kahghi@users.noreply.github.com>
---
 backend/open_webui/config.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/backend/open_webui/config.py b/backend/open_webui/config.py
index 0c7dc3d52..ad4570a89 100644
--- a/backend/open_webui/config.py
+++ b/backend/open_webui/config.py
@@ -1711,6 +1711,16 @@ CODE_INTERPRETER_JUPYTER_TIMEOUT = PersistentConfig(
     ),
 )
 
+CODE_INTERPRETER_BLACKLISTED_MODULES = PersistentConfig(
+    "CODE_INTERPRETER_BLACKLISTED_MODULES",
+    "code_interpreter.blacklisted_modules",
+    [
+        library.strip()
+        for library in os.environ.get("CODE_INTERPRETER_BLACKLISTED_MODULES", "").split(",")
+        if library.strip()
+    ],
+)
+
 
 DEFAULT_CODE_INTERPRETER_PROMPT = """
 #### Tools Available