@wrenautomation/models 0.0.0-stage → 0.7.0

package/dist/spec.d.ts

@@ -0,0 +1,11 @@
+import { type EnvMap, type RotateOptions } from "./keys.js";
+import type { Model } from "./types.js";
+export declare const SPEC_NAMES: readonly ["fake", "claude-code", "anthropic", "gateway", "gemini", "cohere", "command", ...string[]];
+/**
+ * Keys come from the environment by vendor (`ANTHROPIC_API_KEY`, `GROQ_API_KEY_1`...); the
+ * gateway reads `WREN_LLM_GATEWAY_URL` and `WREN_LLM_GATEWAY_TOKEN`. `rotate` sets how a key ring
+ * waits when every key is busy.
+ */
+export declare function modelFromSpec(spec: string, env?: EnvMap, o?: {
+    rotate?: RotateOptions;
+}): Model;

package/dist/spec.js

@@ -0,0 +1,70 @@
+/**
+ * A model from one string and the environment, for a CLI flag or a setting (Node only: it can
+ * start `claude -p`). Grammar: `fake` | `claude-code[:model]` | `anthropic[:haiku|sonnet|opus|id]` |
+ * `gateway[:alias]` | `gemini[:id]` | `cohere[:id]` | `<groq|openrouter|mistral|cerebras|openai>[:id]` |
+ * `command:<shell command>` (the prompt on stdin, the answer on stdout).
+ */
+import { claudeCode, commandModel } from "./command.js";
+import { fakeModel } from "./fake.js";
+import { fleetKeys } from "./keys.js";
+import { anthropic, COMPATIBLE, cohere, compatible, gateway, gemini, isCompatible, } from "./providers.js";
+export const SPEC_NAMES = [
+    "fake",
+    "claude-code",
+    "anthropic",
+    "gateway",
+    "gemini",
+    "cohere",
+    "command",
+    ...Object.keys(COMPATIBLE),
+];
+const keysFor = (env, vendor) => {
+    const keys = fleetKeys(env, vendor);
+    if (keys.length)
+        return keys;
+    const upper = vendor.toUpperCase();
+    throw new Error(`${vendor} needs ${upper}_API_KEY, or NUM_${upper} and ${upper}_API_KEY_1..N`);
+};
+/**
+ * Keys come from the environment by vendor (`ANTHROPIC_API_KEY`, `GROQ_API_KEY_1`...); the
+ * gateway reads `WREN_LLM_GATEWAY_URL` and `WREN_LLM_GATEWAY_TOKEN`. `rotate` sets how a key ring
+ * waits when every key is busy.
+ */
+export function modelFromSpec(spec, env = process.env, o = {}) {
+    const at = spec.indexOf(":");
+    const name = at < 0 ? spec : spec.slice(0, at);
+    const rest = at < 0 ? undefined : spec.slice(at + 1) || undefined;
+    const vendor = (v) => ({
+        key: keysFor(env, v),
+        ...(rest ? { model: rest } : {}),
+        ...(o.rotate ? { rotate: o.rotate } : {}),
+    });
+    switch (name) {
+        case "fake":
+            return fakeModel();
+        case "command":
+            if (!rest)
+                throw new Error("command needs the command after `command:`");
+            return commandModel(rest);
+        case "claude-code":
+            return claudeCode(rest);
+        case "anthropic":
+            return anthropic(vendor("anthropic"));
+        case "gemini":
+            return gemini(vendor("gemini"));
+        case "cohere":
+            return cohere(vendor("cohere"));
+        case "gateway": {
+            const url = env.WREN_LLM_GATEWAY_URL;
+            const token = env.WREN_LLM_GATEWAY_TOKEN;
+            if (!url || !token)
+                throw new Error("gateway needs WREN_LLM_GATEWAY_URL and WREN_LLM_GATEWAY_TOKEN");
+            return gateway({ url, token, ...(rest ? { model: rest } : {}) });
+        }
+        default:
+            if (isCompatible(name))
+                return compatible(name, vendor(name));
+            throw new Error(`unknown model '${spec}'; expected one of ${SPEC_NAMES.join(", ")} [:model]`);
+    }
+}
+//# sourceMappingURL=spec.js.map

package/dist/spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"spec.js","sourceRoot":"","sources":["../src/spec.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AACH,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AACxD,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACtC,OAAO,EAAe,SAAS,EAAsB,MAAM,WAAW,CAAC;AACvE,OAAO,EACL,SAAS,EACT,UAAU,EACV,MAAM,EACN,UAAU,EACV,OAAO,EACP,MAAM,EACN,YAAY,GACb,MAAM,gBAAgB,CAAC;AAGxB,MAAM,CAAC,MAAM,UAAU,GAAG;IACxB,MAAM;IACN,aAAa;IACb,WAAW;IACX,SAAS;IACT,QAAQ;IACR,QAAQ;IACR,SAAS;IACT,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC;CAClB,CAAC;AAEX,MAAM,OAAO,GAAG,CAAC,GAAW,EAAE,MAAc,EAAY,EAAE;IACxD,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;IACpC,IAAI,IAAI,CAAC,MAAM;QAAE,OAAO,IAAI,CAAC;IAC7B,MAAM,KAAK,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC;IACnC,MAAM,IAAI,KAAK,CAAC,GAAG,MAAM,UAAU,KAAK,oBAAoB,KAAK,QAAQ,KAAK,eAAe,CAAC,CAAC;AACjG,CAAC,CAAC;AAEF;;;;GAIG;AACH,MAAM,UAAU,aAAa,CAC3B,IAAY,EACZ,MAAc,OAAO,CAAC,GAAG,EACzB,IAAgC,EAAE;IAElC,MAAM,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IAC7B,MAAM,IAAI,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAC/C,MAAM,IAAI,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,GAAG,CAAC,CAAC,IAAI,SAAS,CAAC;IAClE,MAAM,MAAM,GAAG,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC;QAC7B,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;QACpB,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAChC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC1C,CAAC,CAAC;IACH,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,MAAM;YACT,OAAO,SAAS,EAAE,CAAC;QACrB,KAAK,SAAS;YACZ,IAAI,CAAC,IAAI;gBAAE,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;YACzE,OAAO,YAAY,CAAC,IAAI,CAAC,CAAC;QAC5B,KAAK,aAAa;YAChB,OAAO,UAAU,CAAC,IAAI,CAAC,CAAC;QAC1B,KAAK,WAAW;YACd,OAAO,SAAS,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC;QACxC,KAAK,QAAQ;YACX,OAAO,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;QAClC,KAAK,QAAQ;YACX,OAAO,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;QAClC,KAAK,SAAS,CAAC,CAAC,CAAC;YACf,MAAM,GAAG,GAAG,GAAG,CAAC,oBAAoB,CAAC;YACrC,MAAM,KAAK,GAAG,GAAG,CAAC,sBAAsB,CAAC;YACzC,IAAI,CAAC,GAAG,IAAI,CAAC,KAAK;gBAChB,MAAM,IAAI,KAAK,CAAC,+DAA+D,CAAC,CAAC;YACnF,OAAO,OAAO,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;QACnE,CAAC;QACD;YACE,IAAI,YAAY,CAAC,IAAI,CAAC;gBAAE,OAAO,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;YAC9D,MAAM,IAAI,KAAK,CAAC,kBAAkB,IAAI,sBAAsB,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAClG,CAAC;AACH,CAAC"}

package/dist/types.d.ts

@@ -0,0 +1,97 @@
+/**
+ * The one shape every model call has. A prompt is a string or a list of parts; a reply is text,
+ * tokens, USD and why it stopped. `ask` parses the reply to a zod shape. raven's and falcon's
+ * `Model` (`id` plus `ask(prompt, shape)`) are met by structure: a `Model` from here is one.
+ */
+import type { ZodType } from "zod";
+/** What a part holds: bytes (base64 or raw) or a URL the vendor fetches itself. */
+export type Source = Uint8Array | string | URL;
+/** A stretch of a video, in seconds from its start; `fps` samples frames (Gemini's default is 1). */
+export interface Clip {
+    startS: number;
+    endS: number;
+    fps?: number;
+}
+export type Part = {
+    type: "text";
+    text: string;
+} | {
+    type: "image";
+    data: Source;
+    mediaType: string;
+} | {
+    type: "video";
+    data: Source;
+    mediaType: string;
+    clip?: Clip;
+} | {
+    type: "audio";
+    data: Source;
+    mediaType: string;
+} | {
+    type: "pdf";
+    data: Source;
+};
+export type PartKind = Part["type"];
+/** How hard a model thinks. Anthropic and Gemini read it; others ignore it. */
+export type Effort = "low" | "medium" | "high" | "xhigh" | "max";
+export interface Request {
+    prompt: string | readonly Part[];
+    /** The standing instructions. On Anthropic it carries a cache breakpoint. */
+    system?: string;
+    /** A ceiling against a runaway answer, never the length: the prompt says how long. */
+    maxTokens?: number;
+    effort?: Effort;
+    /** How finely images and video are sampled. Gemini reads it (low is a quarter of the tokens); others ignore it. */
+    resolution?: "low" | "medium" | "high";
+    /** What the call is for (`triage`, `drafts`). No vendor sees it; meters group by it. */
+    purpose?: string;
+    signal?: AbortSignal;
+}
+export interface Usage {
+    /** Every input token, cache reads and writes included. */
+    input?: number;
+    output?: number;
+    /** Of `input`: read from the vendor's prompt cache. */
+    cacheRead?: number;
+    /** Of `input`: written to the cache (5 minutes unless `cacheWrite1h` says). */
+    cacheWrite?: number;
+    /** Of `cacheWrite`: the share that lives 1 hour. */
+    cacheWrite1h?: number;
+    /** What the call cost. 0 on a free key; absent only when the model has no price. */
+    usd?: number;
+    /** The call ran on a free tier: `usd` is 0 and `listUsd` is what it would cost. */
+    free?: boolean;
+    listUsd?: number;
+}
+export interface Reply {
+    text: string;
+    usage: Usage;
+    /** The model that answered, as the vendor names it. */
+    model: string;
+    /** The vendor's own stop reason (`end_turn`, `max_tokens`, `STOP`, `length`). */
+    stop: string | undefined;
+    /** Wall-clock milliseconds for the round trip. */
+    ms: number;
+    /** The vendor's response metadata, kept for the caller to store. */
+    raw: Record<string, unknown>;
+}
+/** A parsed answer and what it cost. */
+export interface Answer<T> {
+    value: T;
+    usage?: Usage;
+}
+export interface AskOptions extends Omit<Request, "prompt"> {
+    /** More calls after an answer that does not parse, each told why. Default 1. */
+    repairs?: number;
+}
+export interface Model {
+    /** The model id, as spends and records name it (`claude-haiku-5-5`, `groq:openai/gpt-oss-120b`). */
+    readonly id: string;
+    /** The vendor alone (`anthropic`, `groq`), for per-vendor sums. */
+    readonly provider: string;
+    /** The part kinds it takes. A part outside this set throws `Unsupported`, never dropped. */
+    readonly takes: ReadonlySet<PartKind>;
+    complete(request: Request | string): Promise<Reply>;
+    ask<T>(prompt: string | readonly Part[], shape: ZodType<T>, o?: AskOptions): Promise<Answer<T>>;
+}

package/dist/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":""}

package/package.json

@@ -1,6 +1,58 @@
 {
   "name": "@wrenautomation/models",
-  "version": "0.0.0-stage",
-  "stub": true,
-  "description": "Temporary package placeholder for staged publishing"
+  "version": "0.7.0",
+  "description": "One door to every model: a Model takes text, image, video, audio and PDF parts, answers with text, tokens and USD, and parses an answer to a zod shape. Adapters for Anthropic, Gemini, Cohere, any OpenAI-compatible server and a shell command (claude -p); ears for speech to text; one price table; one key ring.",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/wrenautomation/engine.git",
+    "directory": "packages/models"
+  },
+  "license": "PolyForm-Shield-1.0.0",
+  "files": [
+    "dist",
+    "!dist/**/*.test.*",
+    "README.md",
+    "LICENSE.md"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "type": "module",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    },
+    "./prices": {
+      "types": "./dist/prices.d.ts",
+      "import": "./dist/prices.js"
+    },
+    "./hear": {
+      "types": "./dist/hear.d.ts",
+      "import": "./dist/hear.js"
+    },
+    "./command": {
+      "types": "./dist/command.d.ts",
+      "import": "./dist/command.js"
+    },
+    "./spec": {
+      "types": "./dist/spec.d.ts",
+      "import": "./dist/spec.js"
+    }
+  },
+  "engines": {
+    "node": ">=24"
+  },
+  "dependencies": {
+    "@ai-sdk/anthropic": "4.0.90",
+    "@ai-sdk/cohere": "4.0.71",
+    "@ai-sdk/google": "4.0.107",
+    "@ai-sdk/openai-compatible": "3.0.80",
+    "ai": "7.0.155",
+    "zod": "4.6.5"
+  },
+  "scripts": {
+    "typecheck": "tsc --noEmit",
+    "build": "rm -rf dist && tsc"
+  }
 }