Преглед на файлове

wip: jupyter code excution timeout

Timothy J. Baek преди 2 месеца
родител
ревизия
90fca993f3
променени са 1 файла, в които са добавени 16 реда и са изтрити 0 реда
  1. 16 0
      backend/open_webui/config.py

+ 16 - 0
backend/open_webui/config.py

@@ -1402,6 +1402,11 @@ CODE_EXECUTION_JUPYTER_AUTH_PASSWORD = PersistentConfig(
     os.environ.get("CODE_EXECUTION_JUPYTER_AUTH_PASSWORD", ""),
     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 = PersistentConfig(
     "ENABLE_CODE_INTERPRETER",
     "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 = """
 DEFAULT_CODE_INTERPRETER_PROMPT = """
 #### Tools Available
 #### Tools Available