refac

Author: tjbck

backend/open_webui/main.py

@@ -92,6 +92,7 @@
     ENABLE_SCIM,
     ENABLE_SIGNUP_PASSWORD_CONFIRMATION,
     ENABLE_STAR_SESSIONS_MIDDLEWARE,
+    ENABLE_PYODIDE_FILE_PERSISTENCE,
     ENABLE_VERSION_UPDATE_CHECK,
     ENABLE_WEBSOCKET_SUPPORT,
     GLOBAL_LOG_LEVEL,
@@ -271,6 +272,13 @@ async def get_response(self, path: str, scope):
                 raise ex
 
 
+class CORSStaticFiles(StaticFiles):
+    async def get_response(self, path: str, scope):
+        response = await super().get_response(path, scope)
+        response.headers['Access-Control-Allow-Origin'] = '*'
+        return response
+
+
 if LOG_FORMAT != 'json':
     banner = rf"""
  ██████╗ ██████╗ ███████╗███╗   ██╗    ██╗    ██╗███████╗██████╗ ██╗   ██╗██╗
@@ -2580,6 +2588,10 @@ def swagger_ui_html(*args, **kwargs):
 
 if os.path.exists(FRONTEND_BUILD_DIR):
     mimetypes.add_type('text/javascript', '.js')
+    pyodide_dir = FRONTEND_BUILD_DIR / 'pyodide'
+    if os.path.exists(pyodide_dir):
+        app.mount('/pyodide', CORSStaticFiles(directory=pyodide_dir), name='pyodide')
+
     app.mount(
         '/',
         SPAStaticFiles(directory=FRONTEND_BUILD_DIR, html=True),

src/lib/pyodide/createPyodideWorker.ts

@@ -0,0 +1,10 @@
+import { get } from 'svelte/store';
+
+import { config } from '$lib/stores';
+import { PyodideSandboxHost } from '$lib/pyodide/pyodideSandboxHost';
+import PyodideWorker from '$lib/workers/pyodide.worker?worker';
+
+export const createPyodideWorker = (): Worker =>
+	get(config)?.features?.enable_pyodide_file_persistence
+		? new PyodideWorker()
+		: (new PyodideSandboxHost() as unknown as Worker);