refac: include packages

8766d269 · Timothy J. Baek · 7f90d813 · 8766d269
Commit 8766d269 authored May 18, 2024 by Timothy J. Baek
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 2 deletions

src/lib/components/chat/Messages/CodeBlock.svelte src/lib/components/chat/Messages/CodeBlock.svelte +7 -2

No files found.
--- 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();