From 90fca993f30b09a815580f50db201325cddaf721 Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <tim@openwebui.com>
Date: Wed, 19 Feb 2025 16:41:54 -0800
Subject: [PATCH] wip: jupyter code excution timeout
---
backend/open_webui/config.py | 16 ++++++++++++++++
1 file changed, 16 insertions(+)
diff --git a/backend/open_webui/config.py b/backend/open_webui/config.py
index f1a55fdce..325ba486d 100644
--- a/backend/open_webui/config.py
+++ b/backend/open_webui/config.py
@@ -1402,6 +1402,11 @@ CODE_EXECUTION_JUPYTER_AUTH_PASSWORD = PersistentConfig(
os.environ.get("CODE_EXECUTION_JUPYTER_AUTH_PASSWORD", ""),
)
+CODE_EXECUTION_JUPYTER_TIMEOUT = PersistentConfig(
+ "CODE_EXECUTION_JUPYTER_TIMEOUT",
+ "code_execution.jupyter.timeout",
+ int(os.environ.get("CODE_EXECUTION_JUPYTER_TIMEOUT", "60")),
+)
ENABLE_CODE_INTERPRETER = PersistentConfig(
"ENABLE_CODE_INTERPRETER",
@@ -1457,6 +1462,17 @@ CODE_INTERPRETER_JUPYTER_AUTH_PASSWORD = PersistentConfig(
),
)
+CODE_INTERPRETER_JUPYTER_TIMEOUT = PersistentConfig(
+ "CODE_INTERPRETER_JUPYTER_TIMEOUT",
+ "code_interpreter.jupyter.timeout",
+ int(
+ os.environ.get(
+ "CODE_INTERPRETER_JUPYTER_TIMEOUT",
+ os.environ.get("CODE_EXECUTION_JUPYTER_TIMEOUT", "60"),
+ )
+ ),
+)
+
DEFAULT_CODE_INTERPRETER_PROMPT = """
#### Tools Available