react-cai 1.0.0

package/src/worker.ts

@@ -0,0 +1,63 @@
+// src/worker.ts
+import { CreateMLCEngine, MLCEngineInterface } from "@mlc-ai/web-llm";
+import { WorkerMessage, WorkerPayload } from "./types";
+
+let engine: MLCEngineInterface | null = null;
+
+self.onmessage = async (e: MessageEvent<any>) => {
+    // FIX: Destructure from payload because the hook wraps it
+    const { type, payload } = e.data;
+    
+    try {
+        if (type === "LOAD" && payload) {
+            const { model, options } = payload;
+            
+            const chatOpts = options?.context_window_size 
+                ? { context_window_size: options.context_window_size }
+                : undefined;
+
+            const engineConfig: any = {
+                initProgressCallback: (report: any) => {
+                    postMsg({ type: "PROGRESS", payload: report });
+                },
+            };
+
+            let targetId = "";
+
+            // Handle Custom Records vs String IDs
+            if (typeof model === "object") {
+                targetId = model.model_id;
+                engineConfig.appConfig = {
+                    model_list: [model] 
+                };
+            } else {
+                targetId = model;
+            }
+
+            // Create engine
+            engine = await CreateMLCEngine(targetId, engineConfig, chatOpts);
+            postMsg({ type: "READY" });
+        }
+
+        if (type === "GENERATE" && engine) {
+            const { messages } = e.data; // Note: chat() sends 'messages' at top level
+            
+            const chunks = await engine.chat.completions.create({
+                messages,
+                stream: true,
+            });
+
+            for await (const chunk of chunks) {
+                const content = chunk.choices[0]?.delta?.content || "";
+                if (content) postMsg({ type: "TOKEN", payload: content });
+            }
+            postMsg({ type: "DONE" });
+        }
+    } catch (err: any) {
+        postMsg({ type: "ERROR", payload: err?.message || String(err) });
+    }
+};
+
+function postMsg(msg: WorkerMessage) {
+    self.postMessage(msg);
+}

package/tsconfig.json

@@ -0,0 +1,20 @@
+{
+  "compilerOptions": {
+    "target": "ES2017",
+    "module": "ESNext",
+    "lib": ["DOM", "DOM.Iterable", "ESNext"],
+    "declaration": true,
+    "sourceMap": true,
+    "strict": true,
+    "moduleResolution": "node",
+    "allowSyntheticDefaultImports": true,
+    "esModuleInterop": true,
+    "skipLibCheck": true,
+    "forceConsistentCasingInFileNames": true,
+    "incremental": false,  // <--- THIS IS THE FIX
+    "jsx": "react",
+    "types":["@webgpu/types"]
+  },
+  "include": ["src"],
+  "exclude": ["node_modules", "dist"]
+}

package/tsup.config.ts

@@ -0,0 +1,37 @@
+import { defineConfig } from "tsup";
+// ✅ CORRECT IMPORT for the named export
+import { ObfuscatorPlugin } from "esbuild-plugin-obfuscator";
+
+export default defineConfig({
+    entry: ["src/index.ts", "src/worker.ts"],
+    format: ["cjs", "esm"],
+    dts: true,
+    clean: true,
+    external: ["react"], 
+    noExternal: ["@mlc-ai/web-llm"], 
+    
+    // Protection Config
+    splitting: false, 
+    sourcemap: false, 
+    minify: true, 
+    
+    // ✅ FIX: Move 'drop' inside esbuildOptions
+    esbuildOptions(options) {
+        options.drop = ['console', 'debugger'];
+    },
+
+    esbuildPlugins: [
+        ObfuscatorPlugin({
+            compact: true,
+            controlFlowFlattening: true, 
+            deadCodeInjection: true,     
+            deadCodeInjectionThreshold: 0.2,
+            identifierNamesGenerator: 'hexadecimal', 
+            
+            // KEEP THIS FALSE so your Worker can still talk to React
+            renameProperties: false, 
+            
+            splitStrings: true, 
+        }),
+    ],
+});