[Log] Add setLogLevel to Engine Interface

Author: Neet-Nestor

src/engine.ts

@@ -63,8 +63,8 @@ export async function CreateMLCEngine(
   modelId: string,
   engineConfig?: MLCEngineConfig,
 ): Promise<MLCEngine> {
-  log.setLevel(engineConfig?.logLevel || DefaultLogLevel);
   const engine = new MLCEngine();
+  engine.setLogLevel(engineConfig?.logLevel || DefaultLogLevel);
   engine.setInitProgressCallback(engineConfig?.initProgressCallback);
   engine.setLogitProcessorRegistry(engineConfig?.logitProcessorRegistry);
   await engine.reload(modelId, engineConfig?.chatOpts, engineConfig?.appConfig);
@@ -640,6 +640,15 @@ export class MLCEngine implements MLCEngineInterface {
     return this.getPipeline().getMessage();
   }
 
+  /**
+   * Set MLCEngine logging output level
+   *
+   * @param logLevel The new log level
+   */
+  setLogLevel(logLevel: LogLevel) {
+    log.setLevel(logLevel);
+  }
+
   /**
    * Get a new Conversation object based on the chat completion request.
    *

src/extension_service_worker.ts

@@ -140,10 +140,10 @@ export async function CreateServiceWorkerMLCEngine(
   engineConfig?: MLCEngineConfig,
   keepAliveMs = 10000,
 ): Promise<ServiceWorkerMLCEngine> {
-  const serviceWorkerMLCEngine = new ServiceWorkerMLCEngine(
-    keepAliveMs,
-    engineConfig?.logLevel,
-  );
+  const serviceWorkerMLCEngine = new ServiceWorkerMLCEngine(keepAliveMs);
+  if (engineConfig?.logLevel) {
+    serviceWorkerMLCEngine.setLogLevel(engineConfig.logLevel);
+  }
   serviceWorkerMLCEngine.setInitProgressCallback(
     engineConfig?.initProgressCallback,
   );
@@ -192,10 +192,10 @@ class PortAdapter implements ChatWorker {
 export class ServiceWorkerMLCEngine extends WebWorkerMLCEngine {
   port: chrome.runtime.Port;
 
-  constructor(keepAliveMs = 10000, logLevel: LogLevel = "WARN") {
+  constructor(keepAliveMs = 10000) {
     const port = chrome.runtime.connect({ name: "web_llm_service_worker" });
     const chatWorker = new PortAdapter(port);
-    super(chatWorker, logLevel);
+    super(chatWorker);
     this.port = port;
     setInterval(() => {
       this.keepAlive();

src/index.ts

@@ -14,6 +14,7 @@ export {
   InitProgressReport,
   MLCEngineInterface,
   LogitProcessor,
+  LogLevel,
 } from "./types";
 
 export { MLCEngine, CreateMLCEngine } from "./engine";

src/service_worker.ts

@@ -206,11 +206,10 @@ export async function CreateServiceWorkerMLCEngine(
         "Please refresh the page to retry initializing the service worker.",
     );
   }
-  const serviceWorkerMLCEngine = new ServiceWorkerMLCEngine(
-    serviceWorker,
-    undefined,
-    engineConfig?.logLevel,
-  );
+  const serviceWorkerMLCEngine = new ServiceWorkerMLCEngine(serviceWorker);
+  if (engineConfig?.logLevel) {
+    serviceWorkerMLCEngine.setLogLevel(engineConfig.logLevel);
+  }
   serviceWorkerMLCEngine.setInitProgressCallback(
     engineConfig?.initProgressCallback,
   );
@@ -228,15 +227,11 @@ export async function CreateServiceWorkerMLCEngine(
 export class ServiceWorkerMLCEngine extends WebWorkerMLCEngine {
   missedHeatbeat = 0;
 
-  constructor(
-    worker: IServiceWorker,
-    keepAliveMs = 10000,
-    logLevel: LogLevel = "WARN",
-  ) {
+  constructor(worker: IServiceWorker, keepAliveMs = 10000) {
     if (!("serviceWorker" in navigator)) {
       throw new Error("Service worker API is not available");
     }
-    super(new ServiceWorker(worker), logLevel);
+    super(new ServiceWorker(worker));
     const onmessage = this.onmessage.bind(this);
 
     (navigator.serviceWorker as ServiceWorkerContainer).addEventListener(

src/types.ts

@@ -194,6 +194,21 @@ export interface MLCEngineInterface {
     inputIds: Array<number>,
     isPrefill: boolean,
   ): Promise<number>;
+
+  /**
+   * Set MLCEngine logging output level
+   *
+   * @param logLevel The new log level
+   */
+  setLogLevel(logLevel: LogLevel): void;
 }
 
-export type LogLevel = "TRACE" | "DEBUG" | "INFO" | "WARN" | "ERROR" | "SILENT";
+export const LOG_LEVELS = {
+  TRACE: 0,
+  DEBUG: 1,
+  INFO: 2,
+  WARN: 3,
+  ERROR: 4,
+  SILENT: 5,
+};
+export type LogLevel = keyof typeof LOG_LEVELS;

src/web_worker.ts

@@ -350,8 +350,7 @@ export class WebWorkerMLCEngine implements MLCEngineInterface {
   >();
   private pendingPromise = new Map<string, (msg: WorkerResponse) => void>();
 
-  constructor(worker: ChatWorker, logLevel: LogLevel = "WARN") {
-    log.setLevel(logLevel);
+  constructor(worker: ChatWorker) {
     this.worker = worker;
     worker.onmessage = (event: any) => {
       this.onmessage.bind(this)(event);
@@ -627,4 +626,8 @@ export class WebWorkerMLCEngine implements MLCEngineInterface {
       }
     }
   }
+
+  setLogLevel(logLevel: LogLevel) {
+    log.setLevel(logLevel);
+  }
 }