refac: include packages

open-webui · May 18, 2024 · 8766d26 · 8766d26
1 parent 7f90d81
commit 8766d26
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/lib/components/chat/Messages/CodeBlock.svelte b/src/lib/components/chat/Messages/CodeBlock.svelte
@@ -183,7 +183,10 @@
 					code.includes('bs4') ? 'beautifulsoup4' : null,
 					code.includes('numpy') ? 'numpy' : null,
 					code.includes('pandas') ? 'pandas' : null,
-					code.includes('matplotlib') ? 'matplotlib' : null
+					code.includes('matplotlib') ? 'matplotlib' : null,
+					code.includes('scikit-learn') ? 'sklearn' : null,
+					code.includes('scipy') ? 'scipy' : null,
+					code.includes('re') ? 'regex' : null
 				].filter(Boolean);
 
 				console.log(packages);
@@ -230,7 +233,9 @@ __builtins__.input = input`);
 			code.includes('bs4') ? 'beautifulsoup4' : null,
 			code.includes('numpy') ? 'numpy' : null,
 			code.includes('pandas') ? 'pandas' : null,
-			code.includes('matplotlib') ? 'matplotlib' : null
+			code.includes('scikit-learn') ? 'sklearn' : null,
+			code.includes('scipy') ? 'scipy' : null,
+			code.includes('re') ? 'regex' : null
 		].filter(Boolean);
 
 		const pyodideWorker = new PyodideWorker();