wllama-service 2.0.0 → 2.0.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wllama-service",
-  "version": "2.0.0",
+  "version": "2.0.1",
   "description": "Framework-agnostic browser LLM service wrapper with WebGPU and WebAssembly support",
   "main": "./dist/index.js",
   "bin": {

package/scripts/postinstall.js

@@ -1,26 +1,63 @@
 #!/usr/bin/env node
 'use strict';
 
+const fs = require('fs');
 const path = require('path');
-const { copyWasm } = require('./copyWllamaWasm');
 
-// This file is wired up as the "postinstall" script in the CONSUMING
-// APP's package.json (see setup below) — npm runs it automatically
-// right after `npm install` finishes. No command to type, no step
-// to remember.
-//
-// Destination defaults to <app>/public/wllama, but can be overridden
-// with an env var if the app's public folder lives somewhere else:
-//   WLLAMA_WASM_DEST=./static/wllama npm install
+/**
+ * Resolve @wllama/wllama's wasm file using Node's real module resolution
+ * instead of a fixed relative path. This is the part that breaks in
+ * yarn workspaces monorepos: @wllama/wllama gets hoisted to the
+ * monorepo ROOT node_modules, not nested under wllama-service's own
+ * node_modules, so `__dirname + '../node_modules/@wllama/wllama/...'`
+ * never finds it. require.resolve walks up node_modules the same way
+ * Node itself would, so it finds the package wherever it actually is.
+ */
+function resolveWasmSrc() {
+  const pkgJsonPath = require.resolve('@wllama/wllama/package.json');
+  const pkgRoot = path.dirname(pkgJsonPath);
+  return path.join(pkgRoot, 'esm', 'wasm', 'wllama.wasm');
+}
 
-const destArg = process.env.WLLAMA_WASM_DEST || 'public/wllama';
-const destDir = path.resolve(process.cwd(), destArg);
+/**
+ * Find the consuming app's public directory. Tries common monorepo
+ * layouts, then falls back to walking up from cwd looking for a
+ * package.json + public/ pair.
+ */
+function findPublicDir() {
+  const candidates = [
+    path.resolve(process.cwd(), 'public'),
+    path.resolve(process.cwd(), '../../public'),
+    path.resolve(process.cwd(), '../../../public'),
+    path.resolve(process.cwd(), '../../../../public'),
+  ];
+  return candidates.find((p) => fs.existsSync(p));
+}
 
-const result = copyWasm(destDir);
+let wasmSrc;
+try {
+  wasmSrc = resolveWasmSrc();
+} catch (err) {
+  console.warn('[wllama-service] Could not resolve @wllama/wllama:', err.message);
+  process.exit(0);
+}
+
+if (!fs.existsSync(wasmSrc)) {
+  console.warn('[wllama-service] wllama.wasm not found at', wasmSrc);
+  process.exit(0);
+}
 
-if (!result) {
-  // Non-fatal: don't break `npm install` for the app if something's
-  // off. Worst case, loadModel() fails clearly at runtime with a 404,
-  // which is easier to debug than a broken install.
+const publicDir = findPublicDir();
+if (!publicDir) {
+  console.warn(
+    '[wllama-service] Could not find a public/ directory near',
+    process.cwd(),
+    '\n  Set WLLAMA_PUBLIC_DIR env var to specify it explicitly.',
+  );
   process.exit(0);
-}
+}
+
+const dest = path.join(publicDir, 'wllama', 'wllama.wasm');
+fs.mkdirSync(path.dirname(dest), { recursive: true });
+fs.copyFileSync(wasmSrc, dest);
+console.log(`[wllama-service] ✓ Copied wllama.wasm to ${dest}`);