@wayfield/ember 0.1.1 → 0.1.2

package/dist/engine.d.ts

@@ -1,12 +1,11 @@
 import type { ChatMessage, EmberModelId } from "./types";
-export declare function createEmber(opts?: {
-    model?: EmberModelId;
-}): {
-    chat(m: ChatMessage[]): Promise<any>;
-    rewrite(t: string, tone?: string): Promise<any>;
-    summarize(t: string): Promise<any>;
-    models: {
-        name: string;
-        id: string;
-    }[];
-};
+
+export interface EmberInstance {
+  init(): Promise<any>;
+  chat(m: ChatMessage[]): Promise<string>;
+  rewrite(t: string, tone?: string): Promise<string>;
+  summarize(t: string): Promise<string>;
+  models: { name: string; id: string }[];
+}
+
+export declare function createEmber(modelId?: EmberModelId): EmberInstance;

package/dist/engine.js

@@ -1,5 +1,43 @@
 import { CreateWebWorkerMLCEngine } from "@mlc-ai/web-llm";
 import MODELS from "./models.json";
 import { prompts } from "./prompts";
-export function createEmber(opts = {}) { const model = opts.model ?? "lightning"; const cfg = MODELS[model]; let engine = null; async function get() { if (engine)
-    return engine; const w = new Worker(new URL("./worker.js", import.meta.url), { type: "module" }); engine = await CreateWebWorkerMLCEngine(w, cfg.id); return engine; } return { async chat(m) { const e = await get(); const r = await e.chat.completions.create({ messages: m }); return r.choices[0].message.content; }, async rewrite(t, tone = "friendly") { return this.chat([{ role: "user", content: prompts.rewrite(t, tone) }]); }, async summarize(t) { return this.chat([{ role: "user", content: prompts.summarize(t) }]); }, models: MODELS }; }
+
+export function createEmber(modelId = "ember-balance") {
+  const cfg = MODELS[modelId];
+  if (!cfg) throw new Error(`Unknown model: ${modelId}`);
+
+  let engine = null;
+  let initPromise = null;
+
+  async function init() {
+    if (engine) return engine;
+    if (initPromise) return initPromise;
+    initPromise = (async () => {
+      const w = new Worker(new URL("./worker.js", import.meta.url), { type: "module" });
+      engine = await CreateWebWorkerMLCEngine(w, cfg.id);
+      return engine;
+    })().catch(e => {
+      initPromise = null;  // allow retry on failure
+      throw e;
+    });
+    return initPromise;
+  }
+
+  return {
+    async init() {
+      return init();
+    },
+    async chat(m) {
+      const e = await init();
+      const r = await e.chat.completions.create({ messages: m });
+      return r.choices[0].message.content;
+    },
+    async rewrite(t, tone = "friendly") {
+      return this.chat([{ role: "user", content: prompts.rewrite(t, tone) }]);
+    },
+    async summarize(t) {
+      return this.chat([{ role: "user", content: prompts.summarize(t) }]);
+    },
+    models: MODELS
+  };
+}

package/dist/index.d.ts

@@ -1,3 +1,4 @@
 export { createEmber } from "./engine";
+export type { EmberInstance } from "./engine";
 export { prompts } from "./prompts";
 export type { EmberModelId } from "./types";

package/dist/models.json

@@ -1,7 +1,7 @@
 [
-    { "name": "lightning", "id": "Llama-3.2-1B-Instruct-q4f16_1-MLC" },
-    { "name": "nano", "id": "Phi-3.5-mini-instruct-q4f16_1-MLC" },
-    { "name": "standard", "id": "gemma-2-2b-it-q4f16_1-MLC" },
-    { "name": "balance", "id": "Qwen2.5-Coder-1.5B-Instruct-q4f16_1-MLC" },
-    { "name": "titan", "id": "Llama-3.2-3B-Instruct-q4f16_1-MLC" }
+    { "name": "ember-lightning", "id": "Llama-3.2-1B-Instruct-q4f16_1-MLC" },
+    { "name": "ember-nano", "id": "Qwen2.5-Coder-1.5B-Instruct-q4f16_1-MLC" },
+    { "name": "ember-standard", "id": "gemma-2-2b-it-q4f16_1-MLC" },
+    { "name": "ember-balance", "id": "Phi-3.5-mini-instruct-q4f16_1-MLC" },
+    { "name": "ember-titan", "id": "Llama-3.2-3B-Instruct-q4f16_1-MLC" }
 ]

package/dist/types.d.ts

@@ -1,5 +1,11 @@
-export type EmberModelId = "lightning" | "nano" | "balanced" | "standard" | "large";
+export type EmberModelId =
+  | "ember-lightning"
+  | "ember-nano"
+  | "ember-standard"
+  | "ember-balance"
+  | "ember-titan";
+
 export interface ChatMessage {
-    role: "system" | "user" | "assistant";
-    content: string;
+  role: "system" | "user" | "assistant";
+  content: string;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wayfield/ember",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "description": "Local-first AI engine for the web. Runs entirely in browser via WebGPU. No API keys.",
   "type": "module",
   "main": "./dist/index.js",