[log] Fix setLogLevel() in workers

Author: Neet-Nestor

src/message.ts

@@ -1,5 +1,5 @@
 import { AppConfig, ChatOptions, GenerationConfig } from "./config";
-import { InitProgressReport } from "./types";
+import { InitProgressReport, LogLevel } from "./types";
 import {
   ChatCompletionRequestStreaming,
   ChatCompletionRequestNonStreaming,
@@ -27,7 +27,8 @@ type RequestKind =
   | "chatCompletionStreamNextChunk"
   | "customRequest"
   | "keepAlive"
-  | "heartbeat";
+  | "heartbeat"
+  | "setLogLevel";
 
 // eslint-disable-next-line @typescript-eslint/no-unused-vars
 type ResponseKind =
@@ -78,6 +79,7 @@ export type MessageContent =
   | ChatCompletionStreamInitParams
   | CustomRequestParams
   | InitProgressReport
+  | LogLevel
   | string
   | null
   | number

src/service_worker.ts

@@ -2,7 +2,7 @@ import * as tvmjs from "tvmjs";
 import log from "loglevel";
 import { AppConfig, ChatOptions, MLCEngineConfig } from "./config";
 import { ReloadParams, WorkerRequest, WorkerResponse } from "./message";
-import { MLCEngineInterface, InitProgressReport } from "./types";
+import { MLCEngineInterface, InitProgressReport, LogLevel } from "./types";
 import {
   MLCEngineWorkerHandler,
   WebWorkerMLCEngine,

src/web_worker.ts

@@ -275,6 +275,13 @@ export class MLCEngineWorkerHandler {
         });
         return;
       }
+      case "setLogLevel": {
+        const logLevel = msg.content as LogLevel;
+        this.engine.setLogLevel(logLevel);
+        log.setLevel(logLevel);
+        onComplete?.(null);
+        return;
+      }
       case "customRequest": {
         onComplete?.(null);
         return;
@@ -575,6 +582,16 @@ export class WebWorkerMLCEngine implements MLCEngineInterface {
     return await this.getPromise<ChatCompletion>(msg);
   }
 
+  async setLogLevel(logLevel: LogLevel) {
+    log.setLevel(logLevel);
+    const msg: WorkerRequest = {
+      kind: "setLogLevel",
+      uuid: crypto.randomUUID(),
+      content: logLevel,
+    };
+    return await this.getPromise<ChatCompletion>(msg);
+  }
+
   onmessage(event: any) {
     let msg: WorkerResponse;
     if (event instanceof MessageEvent) {
@@ -626,8 +643,4 @@ export class WebWorkerMLCEngine implements MLCEngineInterface {
       }
     }
   }
-
-  setLogLevel(logLevel: LogLevel) {
-    log.setLevel(logLevel);
-  }
 }