@runanywhere/web 0.1.0-beta.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +795 -0
- package/dist/Foundation/ErrorTypes.d.ts +61 -0
- package/dist/Foundation/ErrorTypes.d.ts.map +1 -0
- package/dist/Foundation/ErrorTypes.js +90 -0
- package/dist/Foundation/ErrorTypes.js.map +1 -0
- package/dist/Foundation/EventBus.d.ts +171 -0
- package/dist/Foundation/EventBus.d.ts.map +1 -0
- package/dist/Foundation/EventBus.js +113 -0
- package/dist/Foundation/EventBus.js.map +1 -0
- package/dist/Foundation/PlatformAdapter.d.ts +101 -0
- package/dist/Foundation/PlatformAdapter.d.ts.map +1 -0
- package/dist/Foundation/PlatformAdapter.js +417 -0
- package/dist/Foundation/PlatformAdapter.js.map +1 -0
- package/dist/Foundation/SDKLogger.d.ts +33 -0
- package/dist/Foundation/SDKLogger.d.ts.map +1 -0
- package/dist/Foundation/SDKLogger.js +82 -0
- package/dist/Foundation/SDKLogger.js.map +1 -0
- package/dist/Foundation/SherpaONNXBridge.d.ts +134 -0
- package/dist/Foundation/SherpaONNXBridge.d.ts.map +1 -0
- package/dist/Foundation/SherpaONNXBridge.js +332 -0
- package/dist/Foundation/SherpaONNXBridge.js.map +1 -0
- package/dist/Foundation/StructOffsets.d.ts +147 -0
- package/dist/Foundation/StructOffsets.d.ts.map +1 -0
- package/dist/Foundation/StructOffsets.js +161 -0
- package/dist/Foundation/StructOffsets.js.map +1 -0
- package/dist/Foundation/WASMBridge.d.ts +241 -0
- package/dist/Foundation/WASMBridge.d.ts.map +1 -0
- package/dist/Foundation/WASMBridge.js +393 -0
- package/dist/Foundation/WASMBridge.js.map +1 -0
- package/dist/Infrastructure/ArchiveUtility.d.ts +25 -0
- package/dist/Infrastructure/ArchiveUtility.d.ts.map +1 -0
- package/dist/Infrastructure/ArchiveUtility.js +139 -0
- package/dist/Infrastructure/ArchiveUtility.js.map +1 -0
- package/dist/Infrastructure/AudioCapture.d.ts +87 -0
- package/dist/Infrastructure/AudioCapture.d.ts.map +1 -0
- package/dist/Infrastructure/AudioCapture.js +231 -0
- package/dist/Infrastructure/AudioCapture.js.map +1 -0
- package/dist/Infrastructure/AudioPlayback.d.ts +53 -0
- package/dist/Infrastructure/AudioPlayback.d.ts.map +1 -0
- package/dist/Infrastructure/AudioPlayback.js +117 -0
- package/dist/Infrastructure/AudioPlayback.js.map +1 -0
- package/dist/Infrastructure/DeviceCapabilities.d.ts +39 -0
- package/dist/Infrastructure/DeviceCapabilities.d.ts.map +1 -0
- package/dist/Infrastructure/DeviceCapabilities.js +111 -0
- package/dist/Infrastructure/DeviceCapabilities.js.map +1 -0
- package/dist/Infrastructure/ExtensionRegistry.d.ts +30 -0
- package/dist/Infrastructure/ExtensionRegistry.d.ts.map +1 -0
- package/dist/Infrastructure/ExtensionRegistry.js +41 -0
- package/dist/Infrastructure/ExtensionRegistry.js.map +1 -0
- package/dist/Infrastructure/ModelDownloader.d.ts +98 -0
- package/dist/Infrastructure/ModelDownloader.d.ts.map +1 -0
- package/dist/Infrastructure/ModelDownloader.js +431 -0
- package/dist/Infrastructure/ModelDownloader.js.map +1 -0
- package/dist/Infrastructure/ModelLoaderTypes.d.ts +34 -0
- package/dist/Infrastructure/ModelLoaderTypes.d.ts.map +1 -0
- package/dist/Infrastructure/ModelLoaderTypes.js +12 -0
- package/dist/Infrastructure/ModelLoaderTypes.js.map +1 -0
- package/dist/Infrastructure/ModelManager.d.ts +219 -0
- package/dist/Infrastructure/ModelManager.d.ts.map +1 -0
- package/dist/Infrastructure/ModelManager.js +885 -0
- package/dist/Infrastructure/ModelManager.js.map +1 -0
- package/dist/Infrastructure/ModelRegistry.d.ts +131 -0
- package/dist/Infrastructure/ModelRegistry.d.ts.map +1 -0
- package/dist/Infrastructure/ModelRegistry.js +122 -0
- package/dist/Infrastructure/ModelRegistry.js.map +1 -0
- package/dist/Infrastructure/OPFSStorage.d.ts +143 -0
- package/dist/Infrastructure/OPFSStorage.d.ts.map +1 -0
- package/dist/Infrastructure/OPFSStorage.js +330 -0
- package/dist/Infrastructure/OPFSStorage.js.map +1 -0
- package/dist/Infrastructure/VLMWorkerBridge.d.ts +211 -0
- package/dist/Infrastructure/VLMWorkerBridge.d.ts.map +1 -0
- package/dist/Infrastructure/VLMWorkerBridge.js +264 -0
- package/dist/Infrastructure/VLMWorkerBridge.js.map +1 -0
- package/dist/Infrastructure/VLMWorkerRuntime.d.ts +38 -0
- package/dist/Infrastructure/VLMWorkerRuntime.d.ts.map +1 -0
- package/dist/Infrastructure/VLMWorkerRuntime.js +503 -0
- package/dist/Infrastructure/VLMWorkerRuntime.js.map +1 -0
- package/dist/Infrastructure/VideoCapture.d.ts +118 -0
- package/dist/Infrastructure/VideoCapture.d.ts.map +1 -0
- package/dist/Infrastructure/VideoCapture.js +207 -0
- package/dist/Infrastructure/VideoCapture.js.map +1 -0
- package/dist/Public/Extensions/DiffusionTypes.d.ts +64 -0
- package/dist/Public/Extensions/DiffusionTypes.d.ts.map +1 -0
- package/dist/Public/Extensions/DiffusionTypes.js +28 -0
- package/dist/Public/Extensions/DiffusionTypes.js.map +1 -0
- package/dist/Public/Extensions/EmbeddingsTypes.d.ts +33 -0
- package/dist/Public/Extensions/EmbeddingsTypes.d.ts.map +1 -0
- package/dist/Public/Extensions/EmbeddingsTypes.js +13 -0
- package/dist/Public/Extensions/EmbeddingsTypes.js.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+Diffusion.d.ts +44 -0
- package/dist/Public/Extensions/RunAnywhere+Diffusion.d.ts.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+Diffusion.js +189 -0
- package/dist/Public/Extensions/RunAnywhere+Diffusion.js.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+Embeddings.d.ts +56 -0
- package/dist/Public/Extensions/RunAnywhere+Embeddings.d.ts.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+Embeddings.js +240 -0
- package/dist/Public/Extensions/RunAnywhere+Embeddings.js.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+ModelManagement.d.ts +53 -0
- package/dist/Public/Extensions/RunAnywhere+ModelManagement.d.ts.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+ModelManagement.js +153 -0
- package/dist/Public/Extensions/RunAnywhere+ModelManagement.js.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+STT.d.ts +95 -0
- package/dist/Public/Extensions/RunAnywhere+STT.d.ts.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+STT.js +417 -0
- package/dist/Public/Extensions/RunAnywhere+STT.js.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+StructuredOutput.d.ts +69 -0
- package/dist/Public/Extensions/RunAnywhere+StructuredOutput.d.ts.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+StructuredOutput.js +196 -0
- package/dist/Public/Extensions/RunAnywhere+StructuredOutput.js.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+TTS.d.ts +55 -0
- package/dist/Public/Extensions/RunAnywhere+TTS.d.ts.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+TTS.js +253 -0
- package/dist/Public/Extensions/RunAnywhere+TTS.js.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+TextGeneration.d.ts +80 -0
- package/dist/Public/Extensions/RunAnywhere+TextGeneration.d.ts.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+TextGeneration.js +436 -0
- package/dist/Public/Extensions/RunAnywhere+TextGeneration.js.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+ToolCalling.d.ts +82 -0
- package/dist/Public/Extensions/RunAnywhere+ToolCalling.d.ts.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+ToolCalling.js +576 -0
- package/dist/Public/Extensions/RunAnywhere+ToolCalling.js.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+VAD.d.ts +70 -0
- package/dist/Public/Extensions/RunAnywhere+VAD.d.ts.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+VAD.js +231 -0
- package/dist/Public/Extensions/RunAnywhere+VAD.js.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+VLM.d.ts +58 -0
- package/dist/Public/Extensions/RunAnywhere+VLM.d.ts.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+VLM.js +262 -0
- package/dist/Public/Extensions/RunAnywhere+VLM.js.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+VoiceAgent.d.ts +49 -0
- package/dist/Public/Extensions/RunAnywhere+VoiceAgent.d.ts.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+VoiceAgent.js +222 -0
- package/dist/Public/Extensions/RunAnywhere+VoiceAgent.js.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+VoicePipeline.d.ts +63 -0
- package/dist/Public/Extensions/RunAnywhere+VoicePipeline.d.ts.map +1 -0
- package/dist/Public/Extensions/RunAnywhere+VoicePipeline.js +168 -0
- package/dist/Public/Extensions/RunAnywhere+VoicePipeline.js.map +1 -0
- package/dist/Public/Extensions/STTTypes.d.ts +53 -0
- package/dist/Public/Extensions/STTTypes.d.ts.map +1 -0
- package/dist/Public/Extensions/STTTypes.js +16 -0
- package/dist/Public/Extensions/STTTypes.js.map +1 -0
- package/dist/Public/Extensions/TTSTypes.d.ts +31 -0
- package/dist/Public/Extensions/TTSTypes.d.ts.map +1 -0
- package/dist/Public/Extensions/TTSTypes.js +3 -0
- package/dist/Public/Extensions/TTSTypes.js.map +1 -0
- package/dist/Public/Extensions/ToolCallingTypes.d.ts +78 -0
- package/dist/Public/Extensions/ToolCallingTypes.d.ts.map +1 -0
- package/dist/Public/Extensions/ToolCallingTypes.js +8 -0
- package/dist/Public/Extensions/ToolCallingTypes.js.map +1 -0
- package/dist/Public/Extensions/VADTypes.d.ts +30 -0
- package/dist/Public/Extensions/VADTypes.d.ts.map +1 -0
- package/dist/Public/Extensions/VADTypes.js +8 -0
- package/dist/Public/Extensions/VADTypes.js.map +1 -0
- package/dist/Public/Extensions/VLMTypes.d.ts +56 -0
- package/dist/Public/Extensions/VLMTypes.d.ts.map +1 -0
- package/dist/Public/Extensions/VLMTypes.js +24 -0
- package/dist/Public/Extensions/VLMTypes.js.map +1 -0
- package/dist/Public/Extensions/VoiceAgentTypes.d.ts +42 -0
- package/dist/Public/Extensions/VoiceAgentTypes.d.ts.map +1 -0
- package/dist/Public/Extensions/VoiceAgentTypes.js +12 -0
- package/dist/Public/Extensions/VoiceAgentTypes.js.map +1 -0
- package/dist/Public/Extensions/VoicePipelineTypes.d.ts +69 -0
- package/dist/Public/Extensions/VoicePipelineTypes.d.ts.map +1 -0
- package/dist/Public/Extensions/VoicePipelineTypes.js +9 -0
- package/dist/Public/Extensions/VoicePipelineTypes.js.map +1 -0
- package/dist/Public/RunAnywhere.d.ts +121 -0
- package/dist/Public/RunAnywhere.d.ts.map +1 -0
- package/dist/Public/RunAnywhere.js +332 -0
- package/dist/Public/RunAnywhere.js.map +1 -0
- package/dist/index.d.ts +67 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +53 -0
- package/dist/index.js.map +1 -0
- package/dist/types/LLMTypes.d.ts +48 -0
- package/dist/types/LLMTypes.d.ts.map +1 -0
- package/dist/types/LLMTypes.js +8 -0
- package/dist/types/LLMTypes.js.map +1 -0
- package/dist/types/enums.d.ts +144 -0
- package/dist/types/enums.d.ts.map +1 -0
- package/dist/types/enums.js +159 -0
- package/dist/types/enums.js.map +1 -0
- package/dist/types/index.d.ts +9 -0
- package/dist/types/index.d.ts.map +1 -0
- package/dist/types/index.js +8 -0
- package/dist/types/index.js.map +1 -0
- package/dist/types/models.d.ts +154 -0
- package/dist/types/models.d.ts.map +1 -0
- package/dist/types/models.js +8 -0
- package/dist/types/models.js.map +1 -0
- package/dist/workers/vlm-worker.d.ts +9 -0
- package/dist/workers/vlm-worker.d.ts.map +1 -0
- package/dist/workers/vlm-worker.js +10 -0
- package/dist/workers/vlm-worker.js.map +1 -0
- package/package.json +62 -0
- package/wasm/racommons-webgpu.js +156 -0
- package/wasm/racommons-webgpu.wasm +0 -0
- package/wasm/racommons.js +126 -0
- package/wasm/racommons.wasm +0 -0
- package/wasm/sherpa/sherpa-onnx-asr.js +1538 -0
- package/wasm/sherpa/sherpa-onnx-glue-original.js +19 -0
- package/wasm/sherpa/sherpa-onnx-glue.js +17 -0
- package/wasm/sherpa/sherpa-onnx-tts.js +657 -0
- package/wasm/sherpa/sherpa-onnx-vad.js +337 -0
- package/wasm/sherpa/sherpa-onnx-wave.js +88 -0
- package/wasm/sherpa/sherpa-onnx.wasm +0 -0
|
@@ -0,0 +1,264 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* RunAnywhere Web SDK - VLM Worker Bridge
|
|
3
|
+
*
|
|
4
|
+
* Main-thread proxy for the VLM Web Worker. All VLM inference runs off the
|
|
5
|
+
* main thread so the camera, UI animations, and event loop stay responsive.
|
|
6
|
+
*
|
|
7
|
+
* Usage:
|
|
8
|
+
* ```typescript
|
|
9
|
+
* import { VLMWorkerBridge } from '@runanywhere/web';
|
|
10
|
+
*
|
|
11
|
+
* const vlm = VLMWorkerBridge.shared;
|
|
12
|
+
* await vlm.init();
|
|
13
|
+
* await vlm.loadModel({ ... });
|
|
14
|
+
* const result = await vlm.process(rgbPixels, width, height, prompt, { maxTokens: 100 });
|
|
15
|
+
* ```
|
|
16
|
+
*/
|
|
17
|
+
import { WASMBridge } from '../Foundation/WASMBridge';
|
|
18
|
+
import { SDKLogger } from '../Foundation/SDKLogger';
|
|
19
|
+
// ---------------------------------------------------------------------------
|
|
20
|
+
// Bridge
|
|
21
|
+
// ---------------------------------------------------------------------------
|
|
22
|
+
const logger = new SDKLogger('VLMWorkerBridge');
|
|
23
|
+
/**
|
|
24
|
+
* VLMWorkerBridge - Main-thread proxy for VLM Web Worker inference.
|
|
25
|
+
*
|
|
26
|
+
* Manages the lifecycle of a dedicated Web Worker that runs VLM inference
|
|
27
|
+
* in its own WASM instance. Provides:
|
|
28
|
+
* - RPC protocol with message ID tracking and promise correlation
|
|
29
|
+
* - Auto-recovery from WASM crashes (OOB, stack overflow, etc.)
|
|
30
|
+
* - Progress listeners for model loading stages
|
|
31
|
+
* - Transferable pixel data for zero-copy image transfer
|
|
32
|
+
*/
|
|
33
|
+
export class VLMWorkerBridge {
|
|
34
|
+
// ---- Singleton ----
|
|
35
|
+
static _instance = null;
|
|
36
|
+
static get shared() {
|
|
37
|
+
if (!VLMWorkerBridge._instance) {
|
|
38
|
+
VLMWorkerBridge._instance = new VLMWorkerBridge();
|
|
39
|
+
}
|
|
40
|
+
return VLMWorkerBridge._instance;
|
|
41
|
+
}
|
|
42
|
+
// ---- State ----
|
|
43
|
+
worker = null;
|
|
44
|
+
nextId = 0;
|
|
45
|
+
pending = new Map();
|
|
46
|
+
_isInitialized = false;
|
|
47
|
+
_isModelLoaded = false;
|
|
48
|
+
_progressListeners = [];
|
|
49
|
+
/** Saved for auto-recovery after WASM crash */
|
|
50
|
+
_lastModelParams = null;
|
|
51
|
+
_needsRecovery = false;
|
|
52
|
+
/**
|
|
53
|
+
* Optional: the app can provide a custom Worker URL or factory.
|
|
54
|
+
* If not set, the bridge uses the bundled SDK worker entry point.
|
|
55
|
+
*/
|
|
56
|
+
_workerUrl = null;
|
|
57
|
+
get isInitialized() { return this._isInitialized; }
|
|
58
|
+
get isModelLoaded() { return this._isModelLoaded; }
|
|
59
|
+
// ---- Configuration ----
|
|
60
|
+
/**
|
|
61
|
+
* Set a custom Worker URL.
|
|
62
|
+
*
|
|
63
|
+
* By default the bridge creates a Worker using the SDK's bundled entry point
|
|
64
|
+
* (`workers/vlm-worker.js`). Apps that need to customise the Worker location
|
|
65
|
+
* (e.g. different deploy path, or a worker that wraps the runtime) can call
|
|
66
|
+
* this before `init()`.
|
|
67
|
+
*/
|
|
68
|
+
set workerUrl(url) {
|
|
69
|
+
this._workerUrl = url;
|
|
70
|
+
}
|
|
71
|
+
// ---- Progress ----
|
|
72
|
+
/**
|
|
73
|
+
* Subscribe to progress updates from the Worker.
|
|
74
|
+
* Returns an unsubscribe function.
|
|
75
|
+
*/
|
|
76
|
+
onProgress(fn) {
|
|
77
|
+
this._progressListeners.push(fn);
|
|
78
|
+
return () => {
|
|
79
|
+
this._progressListeners = this._progressListeners.filter((l) => l !== fn);
|
|
80
|
+
};
|
|
81
|
+
}
|
|
82
|
+
emitProgress(stage) {
|
|
83
|
+
for (const fn of this._progressListeners)
|
|
84
|
+
fn(stage);
|
|
85
|
+
}
|
|
86
|
+
// ---- Lifecycle ----
|
|
87
|
+
/**
|
|
88
|
+
* Initialize the Worker and its WASM instance.
|
|
89
|
+
* Must be called once before loadModel/process.
|
|
90
|
+
*
|
|
91
|
+
* Reads the WASM URL and acceleration mode from WASMBridge internally — the
|
|
92
|
+
* app does not need to pass these.
|
|
93
|
+
*
|
|
94
|
+
* @param wasmJsUrl - Optional explicit URL for the WASM glue JS.
|
|
95
|
+
* When omitted, the SDK's WASMBridge.workerWasmUrl is used
|
|
96
|
+
* so the worker loads the exact same variant (WebGPU or CPU)
|
|
97
|
+
* that the main thread successfully loaded.
|
|
98
|
+
*/
|
|
99
|
+
async init(wasmJsUrl) {
|
|
100
|
+
if (this._isInitialized)
|
|
101
|
+
return;
|
|
102
|
+
// All acceleration logic lives in the SDK's WASMBridge.
|
|
103
|
+
// We just read the decision it already made.
|
|
104
|
+
const bridge = WASMBridge.shared;
|
|
105
|
+
const useWebGPU = bridge.accelerationMode === 'webgpu';
|
|
106
|
+
const resolvedUrl = wasmJsUrl ?? bridge.workerWasmUrl ?? '';
|
|
107
|
+
if (!resolvedUrl) {
|
|
108
|
+
throw new Error('[VLMWorkerBridge] SDK not initialized — no WASM URL available');
|
|
109
|
+
}
|
|
110
|
+
// Create the Worker
|
|
111
|
+
const workerUrl = this._workerUrl ?? new URL('../workers/vlm-worker.js', import.meta.url);
|
|
112
|
+
this.worker = new Worker(workerUrl, { type: 'module' });
|
|
113
|
+
this.worker.onmessage = this.handleMessage.bind(this);
|
|
114
|
+
this.worker.onerror = (e) => {
|
|
115
|
+
logger.error(`Worker error: ${e.message ?? e}`);
|
|
116
|
+
};
|
|
117
|
+
await this.send('init', { wasmJsUrl: resolvedUrl, useWebGPU });
|
|
118
|
+
this._isInitialized = true;
|
|
119
|
+
logger.info(`Worker initialized (${useWebGPU ? 'WebGPU' : 'CPU'})`);
|
|
120
|
+
}
|
|
121
|
+
/**
|
|
122
|
+
* Load a VLM model in the Worker's WASM instance.
|
|
123
|
+
*
|
|
124
|
+
* Normally the Worker reads model files directly from OPFS (zero-copy).
|
|
125
|
+
* When OPFS quota is exceeded and models are only in the main-thread memory
|
|
126
|
+
* cache, the data is transferred via postMessage (still zero-copy via
|
|
127
|
+
* Transferable ArrayBuffers).
|
|
128
|
+
*/
|
|
129
|
+
async loadModel(params) {
|
|
130
|
+
if (!this._isInitialized) {
|
|
131
|
+
await this.init();
|
|
132
|
+
}
|
|
133
|
+
// Transfer data buffers when provided (zero-copy to Worker)
|
|
134
|
+
const transferables = [];
|
|
135
|
+
if (params.modelData)
|
|
136
|
+
transferables.push(params.modelData);
|
|
137
|
+
if (params.mmprojData)
|
|
138
|
+
transferables.push(params.mmprojData);
|
|
139
|
+
await this.send('load-model', params, transferables);
|
|
140
|
+
this._isModelLoaded = true;
|
|
141
|
+
this._lastModelParams = params;
|
|
142
|
+
this._needsRecovery = false;
|
|
143
|
+
logger.info(`Model loaded: ${params.modelId}`);
|
|
144
|
+
}
|
|
145
|
+
/**
|
|
146
|
+
* Process an image with the VLM.
|
|
147
|
+
* Returns a promise that resolves when inference is complete.
|
|
148
|
+
* The main thread stays responsive during processing.
|
|
149
|
+
*
|
|
150
|
+
* The pixel buffer is transferred (zero-copy) to the Worker.
|
|
151
|
+
*/
|
|
152
|
+
async process(rgbPixels, width, height, prompt, options = {}) {
|
|
153
|
+
// Auto-recover from previous WASM crash (OOB, etc.)
|
|
154
|
+
if (this._needsRecovery) {
|
|
155
|
+
await this.recover();
|
|
156
|
+
}
|
|
157
|
+
if (!this._isModelLoaded) {
|
|
158
|
+
throw new Error('No VLM model loaded in Worker. Call loadModel() first.');
|
|
159
|
+
}
|
|
160
|
+
// Transfer the pixel buffer (zero-copy to Worker)
|
|
161
|
+
const buffer = rgbPixels.buffer.slice(rgbPixels.byteOffset, rgbPixels.byteOffset + rgbPixels.byteLength);
|
|
162
|
+
try {
|
|
163
|
+
return await this.send('process', {
|
|
164
|
+
rgbPixels: buffer,
|
|
165
|
+
width,
|
|
166
|
+
height,
|
|
167
|
+
prompt,
|
|
168
|
+
maxTokens: options.maxTokens ?? 200,
|
|
169
|
+
temperature: options.temperature ?? 0.7,
|
|
170
|
+
}, [buffer]);
|
|
171
|
+
}
|
|
172
|
+
catch (err) {
|
|
173
|
+
const msg = err instanceof Error ? err.message : String(err);
|
|
174
|
+
// WASM runtime errors (OOB, stack overflow) leave the instance corrupted.
|
|
175
|
+
// Mark for recovery so the next call creates a fresh Worker.
|
|
176
|
+
if (msg.includes('memory access out of bounds') ||
|
|
177
|
+
msg.includes('unreachable') ||
|
|
178
|
+
msg.includes('RuntimeError')) {
|
|
179
|
+
logger.warning('WASM crash detected, will recover on next call');
|
|
180
|
+
this._needsRecovery = true;
|
|
181
|
+
}
|
|
182
|
+
throw err;
|
|
183
|
+
}
|
|
184
|
+
}
|
|
185
|
+
/**
|
|
186
|
+
* Recover from a WASM crash by terminating the old Worker,
|
|
187
|
+
* creating a fresh one, and reloading the model.
|
|
188
|
+
*/
|
|
189
|
+
async recover() {
|
|
190
|
+
if (!this._lastModelParams) {
|
|
191
|
+
throw new Error('Cannot recover: no model params saved');
|
|
192
|
+
}
|
|
193
|
+
logger.info('Recovering from WASM crash...');
|
|
194
|
+
const params = this._lastModelParams;
|
|
195
|
+
// Destroy old worker completely
|
|
196
|
+
this.terminate();
|
|
197
|
+
// Reinitialize fresh worker + reload model
|
|
198
|
+
await this.init();
|
|
199
|
+
await this.loadModel(params);
|
|
200
|
+
logger.info('Recovery complete');
|
|
201
|
+
}
|
|
202
|
+
/** Cancel in-progress VLM generation. */
|
|
203
|
+
cancel() {
|
|
204
|
+
if (this.worker) {
|
|
205
|
+
this.worker.postMessage({ type: 'cancel', id: -2 });
|
|
206
|
+
}
|
|
207
|
+
}
|
|
208
|
+
/** Unload the VLM model. */
|
|
209
|
+
async unloadModel() {
|
|
210
|
+
if (!this._isModelLoaded)
|
|
211
|
+
return;
|
|
212
|
+
await this.send('unload', {});
|
|
213
|
+
this._isModelLoaded = false;
|
|
214
|
+
}
|
|
215
|
+
/**
|
|
216
|
+
* Terminate the Worker entirely.
|
|
217
|
+
*
|
|
218
|
+
* Rejects any in-flight RPC promises so callers aren't left hanging,
|
|
219
|
+
* then terminates the underlying Web Worker.
|
|
220
|
+
*/
|
|
221
|
+
terminate() {
|
|
222
|
+
// Reject all pending RPC calls so callers don't hang forever
|
|
223
|
+
for (const [, { reject }] of this.pending) {
|
|
224
|
+
reject(new Error('VLM Worker terminated'));
|
|
225
|
+
}
|
|
226
|
+
this.pending.clear();
|
|
227
|
+
this.worker?.terminate();
|
|
228
|
+
this.worker = null;
|
|
229
|
+
this._isInitialized = false;
|
|
230
|
+
this._isModelLoaded = false;
|
|
231
|
+
}
|
|
232
|
+
// ---- Internal RPC ----
|
|
233
|
+
/* eslint-disable @typescript-eslint/no-explicit-any */
|
|
234
|
+
send(type, payload, transferables) {
|
|
235
|
+
return new Promise((resolve, reject) => {
|
|
236
|
+
if (!this.worker) {
|
|
237
|
+
reject(new Error('VLM Worker not initialized'));
|
|
238
|
+
return;
|
|
239
|
+
}
|
|
240
|
+
const id = this.nextId++;
|
|
241
|
+
this.pending.set(id, { resolve, reject });
|
|
242
|
+
this.worker.postMessage({ type, id, payload }, transferables ?? []);
|
|
243
|
+
});
|
|
244
|
+
}
|
|
245
|
+
handleMessage(e) {
|
|
246
|
+
const { id, type, payload } = e.data;
|
|
247
|
+
// Progress messages (id=-1) are not RPC responses
|
|
248
|
+
if (type === 'progress') {
|
|
249
|
+
this.emitProgress(payload.stage);
|
|
250
|
+
return;
|
|
251
|
+
}
|
|
252
|
+
const pending = this.pending.get(id);
|
|
253
|
+
if (!pending)
|
|
254
|
+
return;
|
|
255
|
+
this.pending.delete(id);
|
|
256
|
+
if (type === 'error') {
|
|
257
|
+
pending.reject(new Error(payload.message));
|
|
258
|
+
}
|
|
259
|
+
else {
|
|
260
|
+
pending.resolve(payload);
|
|
261
|
+
}
|
|
262
|
+
}
|
|
263
|
+
}
|
|
264
|
+
//# sourceMappingURL=VLMWorkerBridge.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"VLMWorkerBridge.js","sourceRoot":"","sources":["../../src/Infrastructure/VLMWorkerBridge.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AACtD,OAAO,EAAE,SAAS,EAAE,MAAM,yBAAyB,CAAC;AAsFpD,8EAA8E;AAC9E,SAAS;AACT,8EAA8E;AAE9E,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,iBAAiB,CAAC,CAAC;AAEhD;;;;;;;;;GASG;AACH,MAAM,OAAO,eAAe;IAC1B,sBAAsB;IACd,MAAM,CAAC,SAAS,GAA2B,IAAI,CAAC;IAExD,MAAM,KAAK,MAAM;QACf,IAAI,CAAC,eAAe,CAAC,SAAS,EAAE,CAAC;YAC/B,eAAe,CAAC,SAAS,GAAG,IAAI,eAAe,EAAE,CAAC;QACpD,CAAC;QACD,OAAO,eAAe,CAAC,SAAS,CAAC;IACnC,CAAC;IAED,kBAAkB;IACV,MAAM,GAAkB,IAAI,CAAC;IAC7B,MAAM,GAAG,CAAC,CAAC;IACX,OAAO,GAAG,IAAI,GAAG,EAAqE,CAAC;IACvF,cAAc,GAAG,KAAK,CAAC;IACvB,cAAc,GAAG,KAAK,CAAC;IACvB,kBAAkB,GAAuB,EAAE,CAAC;IACpD,+CAA+C;IACvC,gBAAgB,GAA8B,IAAI,CAAC;IACnD,cAAc,GAAG,KAAK,CAAC;IAE/B;;;OAGG;IACK,UAAU,GAAwB,IAAI,CAAC;IAE/C,IAAI,aAAa,KAAc,OAAO,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC;IAC5D,IAAI,aAAa,KAAc,OAAO,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC;IAE5D,0BAA0B;IAE1B;;;;;;;OAOG;IACH,IAAI,SAAS,CAAC,GAAiB;QAC7B,IAAI,CAAC,UAAU,GAAG,GAAG,CAAC;IACxB,CAAC;IAED,qBAAqB;IAErB;;;OAGG;IACH,UAAU,CAAC,EAAoB;QAC7B,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjC,OAAO,GAAG,EAAE;YACV,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC;QAC5E,CAAC,CAAC;IACJ,CAAC;IAEO,YAAY,CAAC,KAAa;QAChC,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,kBAAkB;YAAE,EAAE,CAAC,KAAK,CAAC,CAAC;IACtD,CAAC;IAED,sBAAsB;IAEtB;;;;;;;;;;;OAWG;IACH,KAAK,CAAC,IAAI,CAAC,SAAkB;QAC3B,IAAI,IAAI,CAAC,cAAc;YAAE,OAAO;QAEhC,wDAAwD;QACxD,6CAA6C;QAC7C,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;QACjC,MAAM,SAAS,GAAG,MAAM,CAAC,gBAAgB,KAAK,QAAQ,CAAC;QACvD,MAAM,WAAW,GAAG,SAAS,IAAI,MAAM,CAAC,aAAa,IAAI,EAAE,CAAC;QAE5D,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,+DAA+D,CAAC,CAAC;QACnF,CAAC;QAED,oBAAoB;QACpB,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,IAAI,IAAI,GAAG,CAAC,0BAA0B,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC1F,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;QAExD,IAAI,CAAC,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtD,IAAI,CAAC,MAAM,CAAC,OAAO,GAAG,CAAC,CAAC,EAAE,EAAE;YAC1B,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,OAAO,IAAI,CAAC,EAAE,CAAC,CAAC;QAClD,CAAC,CAAC;QAEF,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC,CAAC;QAC/D,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC3B,MAAM,CAAC,IAAI,CAAC,uBAAuB,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC;IACtE,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,SAAS,CAAC,MAA0B;QACxC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACzB,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QACpB,CAAC;QAED,4DAA4D;QAC5D,MAAM,aAAa,GAAmB,EAAE,CAAC;QACzC,IAAI,MAAM,CAAC,SAAS;YAAE,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC3D,IAAI,MAAM,CAAC,UAAU;YAAE,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAE7D,MAAM,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC;QACrD,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC3B,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAC;QAC/B,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC;QAC5B,MAAM,CAAC,IAAI,CAAC,iBAAiB,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;IACjD,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,OAAO,CACX,SAAqB,EACrB,KAAa,EACb,MAAc,EACd,MAAc,EACd,UAA6B,EAAE;QAE/B,oDAAoD;QACpD,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACvB,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACzB,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAC;QAC5E,CAAC;QAED,kDAAkD;QAClD,MAAM,MAAM,GAAG,SAAS,CAAC,MAAM,CAAC,KAAK,CACnC,SAAS,CAAC,UAAU,EACpB,SAAS,CAAC,UAAU,GAAG,SAAS,CAAC,UAAU,CAC5C,CAAC;QAEF,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,CAAC,IAAI,CACpB,SAAS,EACT;gBACE,SAAS,EAAE,MAAM;gBACjB,KAAK;gBACL,MAAM;gBACN,MAAM;gBACN,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,GAAG;gBACnC,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;aACxC,EACD,CAAC,MAAM,CAAC,CACT,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC7D,0EAA0E;YAC1E,6DAA6D;YAC7D,IAAI,GAAG,CAAC,QAAQ,CAAC,6BAA6B,CAAC;gBAC3C,GAAG,CAAC,QAAQ,CAAC,aAAa,CAAC;gBAC3B,GAAG,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;gBACjC,MAAM,CAAC,OAAO,CAAC,gDAAgD,CAAC,CAAC;gBACjE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;YAC7B,CAAC;YACD,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,OAAO;QACnB,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;QAC3D,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;QAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,CAAC;QAErC,gCAAgC;QAChC,IAAI,CAAC,SAAS,EAAE,CAAC;QAEjB,2CAA2C;QAC3C,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAClB,MAAM,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC7B,MAAM,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IACnC,CAAC;IAED,yCAAyC;IACzC,MAAM;QACJ,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;IAED,4BAA4B;IAC5B,KAAK,CAAC,WAAW;QACf,IAAI,CAAC,IAAI,CAAC,cAAc;YAAE,OAAO;QACjC,MAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QAC9B,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC;IAC9B,CAAC;IAED;;;;;OAKG;IACH,SAAS;QACP,6DAA6D;QAC7D,KAAK,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAC1C,MAAM,CAAC,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC,CAAC;QAC7C,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QAErB,IAAI,CAAC,MAAM,EAAE,SAAS,EAAE,CAAC;QACzB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC;QAC5B,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC;IAC9B,CAAC;IAED,yBAAyB;IAEzB,uDAAuD;IAC/C,IAAI,CAAC,IAAY,EAAE,OAAY,EAAE,aAA8B;QACrE,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACjB,MAAM,CAAC,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC,CAAC;gBAChD,OAAO;YACT,CAAC;YAED,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;YACzB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;YAC1C,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,aAAa,IAAI,EAAE,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,aAAa,CAAC,CAAkC;QACtD,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC,IAAI,CAAC;QAErC,kDAAkD;QAClD,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;YACxB,IAAI,CAAC,YAAY,CAAE,OAAe,CAAC,KAAK,CAAC,CAAC;YAC1C,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACrC,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAExB,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;YACrB,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAE,OAAe,CAAC,OAAO,CAAC,CAAC,CAAC;QACtD,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC"}
|
|
@@ -0,0 +1,38 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* RunAnywhere Web SDK - VLM Worker Runtime
|
|
3
|
+
*
|
|
4
|
+
* Encapsulates the Worker-side logic for VLM inference. This module runs
|
|
5
|
+
* inside a dedicated Web Worker and manages its own WASM instance
|
|
6
|
+
* (separate from the main-thread SDK).
|
|
7
|
+
*
|
|
8
|
+
* Architecture:
|
|
9
|
+
* - Loads its OWN WASM instance (separate from the main thread SDK)
|
|
10
|
+
* - Reads model files from OPFS directly (no large postMessage transfers)
|
|
11
|
+
* - Communicates via typed postMessage RPC
|
|
12
|
+
*
|
|
13
|
+
* Why a separate WASM instance?
|
|
14
|
+
* The C function `rac_vlm_component_process` is synchronous and blocks for
|
|
15
|
+
* ~100s (2B model in WASM). Running it on the main thread freezes the entire UI.
|
|
16
|
+
* A Worker with its own WASM instance allows inference to happen concurrently.
|
|
17
|
+
*
|
|
18
|
+
* IMPORTANT: This file must NOT import from WASMBridge.ts or other SDK modules
|
|
19
|
+
* that assume a main-thread context. The Worker has its own WASM instance and
|
|
20
|
+
* should be self-contained. Only `type`-only imports are safe.
|
|
21
|
+
*/
|
|
22
|
+
import type { VLMWorkerResult } from './VLMWorkerBridge';
|
|
23
|
+
export type { VLMWorkerResult };
|
|
24
|
+
/**
|
|
25
|
+
* Start the VLM Worker runtime.
|
|
26
|
+
*
|
|
27
|
+
* Call this once from the Worker entry point. It sets up the `self.onmessage`
|
|
28
|
+
* handler that processes RPC commands from the main-thread VLMWorkerBridge.
|
|
29
|
+
*
|
|
30
|
+
* @example
|
|
31
|
+
* ```typescript
|
|
32
|
+
* // workers/vlm-worker.ts
|
|
33
|
+
* import { startVLMWorkerRuntime } from '../Infrastructure/VLMWorkerRuntime';
|
|
34
|
+
* startVLMWorkerRuntime();
|
|
35
|
+
* ```
|
|
36
|
+
*/
|
|
37
|
+
export declare function startVLMWorkerRuntime(): void;
|
|
38
|
+
//# sourceMappingURL=VLMWorkerRuntime.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"VLMWorkerRuntime.d.ts","sourceRoot":"","sources":["../../src/Infrastructure/VLMWorkerRuntime.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;GAoBG;AAMH,OAAO,KAAK,EAAuC,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAK9F,YAAY,EAAE,eAAe,EAAE,CAAC;AAmjBhC;;;;;;;;;;;;GAYG;AACH,wBAAgB,qBAAqB,IAAI,IAAI,CAG5C"}
|