fix breaking changes in web-llm api

midudev · Jun 13, 2024 · 95ed3c6 · 95ed3c6
1 parent a1c7d18
commit 95ed3c6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/04-chatgpt-local/worker.js b/04-chatgpt-local/worker.js
@@ -1,8 +1,7 @@
-import { MLCEngineWorkerHandler, MLCEngine } from "https://esm.run/@mlc-ai/web-llm"
+import { WebWorkerMLCEngineHandler } from "https://esm.run/@mlc-ai/web-llm"
 
-const engine = new MLCEngine()
-const handler = new MLCEngineWorkerHandler(engine)
+const handler = new WebWorkerMLCEngineHandler()
 
-onmessage = msg => {
+self.onmessage = (msg) => {
  handler.onmessage(msg)
 }
diff --git a/web/public/projects/04-chatgpt-local/worker.js b/web/public/projects/04-chatgpt-local/worker.js
@@ -1,8 +1,7 @@
-import { MLCEngineWorkerHandler, MLCEngine } from "https://esm.run/@mlc-ai/web-llm"
+import { WebWorkerMLCEngineHandler } from "https://esm.run/@mlc-ai/web-llm"
 
-const engine = new MLCEngine()
-const handler = new MLCEngineWorkerHandler(engine)
+const handler = new WebWorkerMLCEngineHandler()
 
-onmessage = msg => {
+self.onmessage = (msg) => {
  handler.onmessage(msg)
 }