瀏覽代碼

Add option to blacklist modules in code interpreter

Co-authored-by: KG <kahghi@users.noreply.github.com>
Rodrigo Agundez 4 月之前
父節點
當前提交
6dd969129d
共有 1 個文件被更改,包括 10 次插入0 次删除
  1. 10 0
      backend/open_webui/config.py

+ 10 - 0
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