1
1
import * as tvmjs from "tvmjs" ;
2
- import { AppConfig , ChatOptions , MLCEngineConfig , ModelRecord } from "./config" ;
2
+ import { AppConfig , ChatOptions , MLCEngineConfig } from "./config" ;
3
3
import { ReloadParams , WorkerRequest , WorkerResponse } from "./message" ;
4
4
import { MLCEngineInterface , InitProgressReport } from "./types" ;
5
5
import {
@@ -40,7 +40,7 @@ export class ServiceWorkerMLCEngineHandler extends MLCEngineWorkerHandler {
40
40
> ( ) ;
41
41
private initReuqestUuid ?: string ;
42
42
43
- constructor ( engine : MLCEngineInterface , verbose = false ) {
43
+ constructor ( engine : MLCEngineInterface ) {
44
44
if ( ! self || ! ( "addEventListener" in self ) ) {
45
45
throw new Error (
46
46
"ServiceWorkerGlobalScope is not defined. ServiceWorkerMLCEngineHandler must be created in service worker script." ,
@@ -90,6 +90,9 @@ export class ServiceWorkerMLCEngineHandler extends MLCEngineWorkerHandler {
90
90
onError ?: ( ) => void ,
91
91
) : void {
92
92
const msg = event . data as WorkerRequest ;
93
+ console . debug (
94
+ `ServiceWorker message: [${ msg . kind } ] ${ JSON . stringify ( msg . content ) } ` ,
95
+ ) ;
93
96
94
97
if ( msg . kind === "keepAlive" ) {
95
98
const reply : WorkerRequest = {
@@ -212,7 +215,7 @@ export async function CreateServiceWorkerMLCEngine(
212
215
export class ServiceWorkerMLCEngine extends WebWorkerMLCEngine {
213
216
missedHeatbeat = 0 ;
214
217
215
- constructor ( worker : IServiceWorker , keepAliveMs = 10000 , verbose = false ) {
218
+ constructor ( worker : IServiceWorker , keepAliveMs = 10000 ) {
216
219
if ( ! ( "serviceWorker" in navigator ) ) {
217
220
throw new Error ( "Service worker API is not available" ) ;
218
221
}
@@ -223,6 +226,9 @@ export class ServiceWorkerMLCEngine extends WebWorkerMLCEngine {
223
226
"message" ,
224
227
( event : MessageEvent ) => {
225
228
const msg = event . data ;
229
+ console . debug (
230
+ `MLC client message: [${ msg . kind } ] ${ JSON . stringify ( msg . content ) } ` ,
231
+ ) ;
226
232
try {
227
233
if ( msg . kind === "heartbeat" ) {
228
234
this . missedHeatbeat = 0 ;
@@ -241,6 +247,7 @@ export class ServiceWorkerMLCEngine extends WebWorkerMLCEngine {
241
247
setInterval ( ( ) => {
242
248
this . worker . postMessage ( { kind : "keepAlive" , uuid : crypto . randomUUID ( ) } ) ;
243
249
this . missedHeatbeat += 1 ;
250
+ console . debug ( "missedHeatbeat" , this . missedHeatbeat ) ;
244
251
} , keepAliveMs ) ;
245
252
}
246
253
0 commit comments