nuabase 2.3.5 → 2.3.7

package/README.md

@@ -172,7 +172,7 @@ Every call returns a discriminated union. When `success` is `false`, you get an
 | `engine`            | `'http'` \| `'local-agent'`  | Whether an LLM provider's API or a coding agent on this machine answered.                                 |
 | `schemaEnforcement` | `'in-prompt'` \| `'native'`  | Whether the schema was described in the prompt or enforced by the engine. Output is validated either way. |
 | `provider`          | `string`                     | `engine: 'http'` only. The LLM provider.                                                                  |
-| `agent`             | `'claude-code'` \| `'codex'` | `engine: 'local-agent'` only. Which coding agent answered.                                                |
+| `agent`             | `'claude-code'` \| `'codex'` \| `'pi'` | `engine: 'local-agent'` only. Which coding agent answered.                                                |
 | `costUsdEstimate`   | `number \| undefined`        | `engine: 'local-agent'` only. The agent's own estimate. On a subscription this is not what you pay.       |
 
 **Error**
@@ -233,13 +233,13 @@ const agents = await detectLocalAgents();
 // [{ agent: 'claude-code', installed: true, loggedIn: true, version: '...', binaryPath: '...' }, ...]
 ```
 
-This runs a coding-agent CLI already installed and logged in on this machine, in its headless mode (`claude -p` or `codex exec`). Calls use the user's own Claude or ChatGPT subscription, so no API key is needed. `get()` and `list()` behave as with providers: the output is validated against your schema and typed the same way. Queue operations are not supported. `nuabase/local-agent` is a separate entry point so that browser bundles of `nuabase` never include it.
+This runs a coding-agent CLI already installed and logged in on this machine, in its headless mode (`claude -p`, `codex exec`, or `pi --mode json`). Calls use the user's own logged-in subscription or provider login, so no API key is needed. `get()` and `list()` behave as with providers: the output is validated against your schema and typed the same way. Queue operations are not supported. `nuabase/local-agent` is a separate entry point so that browser bundles of `nuabase` never include it.
 
 The agent is set up once, when you create it. If you install an agent or log in after that, create a new one.
 
 **`localAgent(config)`** sets up a named agent. It throws right away if the agent's executable cannot be found; if the agent is not logged in, the first call fails with that error.
 
-- `agent: 'claude-code' | 'codex'`
+- `agent: 'claude-code' | 'codex' | 'pi'`
 - `binaryPath?: string` – Path to the agent executable. Defaults to looking it up on `PATH`. On Windows, point this at a native executable; `.cmd` shims are not supported.
 - and the options below.
 
@@ -247,13 +247,13 @@ The agent is set up once, when you create it. If you install an agent or log in
 
 Options for both:
 
-- `model?: string` – The agent's own model name, e.g. `'haiku'` for Claude Code, for calls that name no model. Defaults to the agent's default. Per-call `model: { alias }` works for aliases the agent understands (`haiku`, `sonnet`, `opus` for Claude Code).
+- `model?: string` – The agent's own model name, e.g. `'haiku'` for Claude Code, for calls that name no model. Defaults to the agent's default. Per-call `model: { alias }` works for aliases the agent understands (`haiku`, `sonnet`, `opus` for Claude Code). Codex and Pi have no aliases: pass one of their own model names, and for Pi a `provider/id` name such as `'openrouter/z-ai/glm-5.2'`.
 - `timeoutMs?: number` – Per-call limit. Defaults to `180000`.
 - `concurrency?: number` – Maximum agent processes at once. Defaults to `2`.
 - `auth?: 'subscription' | 'inherit'` – `subscription` (default) hides `ANTHROPIC_API_KEY`, `OPENAI_API_KEY` and similar variables from the agent so it bills the logged-in plan rather than an API key.
 - `logger?` – An object with `debug`, `info`, `warn` and `error` methods, where calls and retries are logged. Defaults to the console.
 
-How it works: each call runs the agent in a fresh, empty temporary directory with its tools, project settings and MCP servers turned off. The schema is wrapped as `{ value: ... }` (the agents require an object at the top level) and passed to the agent's own schema option. Codex only accepts OpenAI strict-mode schemas, so a schema with optional properties or open-ended objects is instead described in the prompt, and the output is validated as usual. If an agent rejects a schema, the call is retried with the schema in the prompt. `maxTokens` has no CLI equivalent and is ignored. Calls count against the user's plan limits; check each vendor's terms before offering this to other users.
+How it works: each call runs the agent in a fresh, empty temporary directory with its tools, project settings and MCP servers turned off. The schema is wrapped as `{ value: ... }` (the agents require an object at the top level) and passed to the agent's own schema option. Codex only accepts OpenAI strict-mode schemas, so a schema with optional properties or open-ended objects is instead described in the prompt, and the output is validated as usual. Pi has no schema flag at all, so its schemas are always described in the prompt. If an agent rejects a schema, the call is retried with the schema in the prompt. `maxTokens` has no CLI equivalent and is ignored. Calls count against the user's plan limits; check each vendor's terms before offering this to other users.
 
 ### Methods
 

package/dist/cjs/local-agent.cjs

@@ -367,6 +367,161 @@ ${result.stderr}`.trim();
   return codex;
 }
 
+var pi = {};
+
+var hasRequiredPi;
+function requirePi() {
+  if (hasRequiredPi) return pi;
+  hasRequiredPi = 1;
+  Object.defineProperty(pi, "__esModule", { value: true });
+  pi.piAdapter = void 0;
+  const failure_classification_1 = requireFailureClassification();
+  const SYSTEM_PROMPT = [
+    "You are a data transformation function called by a program.",
+    "You have no tools. Follow the instructions in the user message exactly",
+    "and reply only with the requested JSON."
+  ].join(" ");
+  pi.piAdapter = {
+    id: "pi",
+    binaryName: "pi",
+    // Every environment credential Pi's providers recognize. Under `auth:
+    // "subscription"` these are removed so Pi bills its own `auth.json` login.
+    apiKeyEnvVars: [
+      "AI_GATEWAY_API_KEY",
+      "ANT_LING_API_KEY",
+      "ANTHROPIC_API_KEY",
+      "ANTHROPIC_AUTH_TOKEN",
+      "ANTHROPIC_OAUTH_TOKEN",
+      "AWS_ACCESS_KEY_ID",
+      "AWS_SECRET_ACCESS_KEY",
+      "AWS_SESSION_TOKEN",
+      "AZURE_OPENAI_API_KEY",
+      "BASETEN_API_KEY",
+      "CEREBRAS_API_KEY",
+      "CLOUDFLARE_API_KEY",
+      "COPILOT_GITHUB_TOKEN",
+      "DEEPSEEK_API_KEY",
+      "FIREWORKS_API_KEY",
+      "GEMINI_API_KEY",
+      "GOOGLE_API_KEY",
+      "GOOGLE_CLOUD_API_KEY",
+      "GROQ_API_KEY",
+      "HF_TOKEN",
+      "KIMI_API_KEY",
+      "META_API_KEY",
+      "MINIMAX_API_KEY",
+      "MINIMAX_CN_API_KEY",
+      "MISTRAL_API_KEY",
+      "MOONSHOT_API_KEY",
+      "NVIDIA_API_KEY",
+      "OPENAI_API_KEY",
+      "OPENCODE_API_KEY",
+      "OPENROUTER_API_KEY",
+      "QWEN_TOKEN_PLAN_API_KEY",
+      "QWEN_TOKEN_PLAN_CN_API_KEY",
+      "RADIUS_API_KEY",
+      "TOGETHER_API_KEY",
+      "XAI_API_KEY",
+      "XIAOMI_API_KEY",
+      "XIAOMI_TOKEN_PLAN_AMS_API_KEY",
+      "XIAOMI_TOKEN_PLAN_CN_API_KEY",
+      "XIAOMI_TOKEN_PLAN_SGP_API_KEY",
+      "ZAI_API_KEY",
+      "ZAI_CODING_CN_API_KEY"
+    ],
+    // Pi uses its own `provider/id` model names; set one with localAgent({ model }).
+    modelAliases: {},
+    nativeSchema() {
+      return null;
+    },
+    buildInvocation(request, model) {
+      const args = [
+        "--mode",
+        "json",
+        "--no-session",
+        "--no-tools",
+        "--no-builtin-tools",
+        "--no-extensions",
+        "--no-skills",
+        "--no-prompt-templates",
+        "--no-themes",
+        "--no-context-files",
+        "--system-prompt",
+        SYSTEM_PROMPT,
+        "--offline"
+      ];
+      if (model !== void 0) {
+        args.push("--model", model);
+      }
+      return { args, stdin: request.prompt };
+    },
+    readResult(result, request) {
+      const events = parseEvents(result.stdout);
+      const assistant = events.filter((event) => event.type === "message_end" && event.message?.role === "assistant").map((event) => event.message).pop();
+      if (assistant && (assistant.stopReason === "error" || assistant.errorMessage)) {
+        return failed((0, failure_classification_1.classifyAgentFailure)("pi", assistant.errorMessage ?? "pi stopped with an error"));
+      }
+      const retryFailed = events.find((event) => event.type === "auto_retry_end" && event.success === false);
+      if (retryFailed) {
+        return failed((0, failure_classification_1.classifyAgentFailure)("pi", retryFailed.finalError ?? "auto retry failed"));
+      }
+      if (!assistant) {
+        return failed((0, failure_classification_1.classifyAgentFailure)("pi", `exited with code ${result.exitCode} without a reply: ${(0, failure_classification_1.stderrTail)(result)}`));
+      }
+      const text = (assistant.content ?? []).filter((block) => block.type === "text").map((block) => block.text ?? "").join("");
+      const usage = assistant.usage ?? {};
+      const promptTokens = (usage.input ?? 0) + (usage.cacheRead ?? 0) + (usage.cacheWrite ?? 0);
+      const completionTokens = usage.output ?? 0;
+      const reported = {
+        usage: { promptTokens, completionTokens, totalTokens: promptTokens + completionTokens },
+        model: assistant.model,
+        costUsdEstimate: usage.cost?.total
+      };
+      if (request.enforcement.kind === "in-prompt") {
+        return { kind: "answered", answer: { kind: "text", text }, ...reported };
+      }
+      let envelope;
+      try {
+        envelope = JSON.parse(text);
+      } catch {
+        return failed({ kind: "transient", message: `pi reply is not JSON: ${text}` });
+      }
+      return { kind: "answered", answer: { kind: "structured", envelope }, ...reported };
+    },
+    // Pi has no provider-agnostic login check: `pi auth check` requires --provider
+    // or --model, and the provider differs per user (it is not known until Pi's
+    // settings are read). Report installed Pi as logged in; the first real call
+    // surfaces any auth error, which classifyAgentFailure marks fatal.
+    loginStatusArgs: ["--version"],
+    readLoginStatus(result) {
+      return {
+        loggedIn: result.exitCode === 0,
+        detail: "installed; auth is verified on the first call"
+      };
+    }
+  };
+  function failed(failure) {
+    return { kind: "failed", failure };
+  }
+  function parseEvents(stdout) {
+    const events = [];
+    for (const line of stdout.split("\n")) {
+      const trimmed = line.replace(/\r$/, "").trim();
+      if (!trimmed.startsWith("{"))
+        continue;
+      try {
+        const parsed = JSON.parse(trimmed);
+        if (typeof parsed === "object" && parsed !== null && !Array.isArray(parsed)) {
+          events.push(parsed);
+        }
+      } catch {
+      }
+    }
+    return events;
+  }
+  return pi;
+}
+
 var hasRequiredAdapters;
 function requireAdapters() {
   if (hasRequiredAdapters) return adapters;
@@ -375,9 +530,11 @@ function requireAdapters() {
   adapters.AGENT_ADAPTERS = void 0;
   const claude_code_1 = requireClaudeCode();
   const codex_1 = requireCodex();
+  const pi_1 = requirePi();
   adapters.AGENT_ADAPTERS = {
     "claude-code": claude_code_1.claudeCodeAdapter,
-    codex: codex_1.codexAdapter
+    codex: codex_1.codexAdapter,
+    pi: pi_1.piAdapter
   };
   return adapters;
 }
@@ -392,7 +549,7 @@ function requireAgentId() {
     Object.defineProperty(exports$1, "__esModule", { value: true });
     exports$1.LOCAL_AGENT_IDS = void 0;
     exports$1.isLocalAgentId = isLocalAgentId;
-    exports$1.LOCAL_AGENT_IDS = ["claude-code", "codex"];
+    exports$1.LOCAL_AGENT_IDS = ["claude-code", "codex", "pi"];
     const LOCAL_AGENT_ID_SET = new Set(exports$1.LOCAL_AGENT_IDS);
     function isLocalAgentId(value) {
       return typeof value === "string" && LOCAL_AGENT_ID_SET.has(value);

package/dist/esm/local-agent.mjs

@@ -364,6 +364,161 @@ ${result.stderr}`.trim();
   return codex;
 }
 
+var pi = {};
+
+var hasRequiredPi;
+function requirePi() {
+  if (hasRequiredPi) return pi;
+  hasRequiredPi = 1;
+  Object.defineProperty(pi, "__esModule", { value: true });
+  pi.piAdapter = void 0;
+  const failure_classification_1 = requireFailureClassification();
+  const SYSTEM_PROMPT = [
+    "You are a data transformation function called by a program.",
+    "You have no tools. Follow the instructions in the user message exactly",
+    "and reply only with the requested JSON."
+  ].join(" ");
+  pi.piAdapter = {
+    id: "pi",
+    binaryName: "pi",
+    // Every environment credential Pi's providers recognize. Under `auth:
+    // "subscription"` these are removed so Pi bills its own `auth.json` login.
+    apiKeyEnvVars: [
+      "AI_GATEWAY_API_KEY",
+      "ANT_LING_API_KEY",
+      "ANTHROPIC_API_KEY",
+      "ANTHROPIC_AUTH_TOKEN",
+      "ANTHROPIC_OAUTH_TOKEN",
+      "AWS_ACCESS_KEY_ID",
+      "AWS_SECRET_ACCESS_KEY",
+      "AWS_SESSION_TOKEN",
+      "AZURE_OPENAI_API_KEY",
+      "BASETEN_API_KEY",
+      "CEREBRAS_API_KEY",
+      "CLOUDFLARE_API_KEY",
+      "COPILOT_GITHUB_TOKEN",
+      "DEEPSEEK_API_KEY",
+      "FIREWORKS_API_KEY",
+      "GEMINI_API_KEY",
+      "GOOGLE_API_KEY",
+      "GOOGLE_CLOUD_API_KEY",
+      "GROQ_API_KEY",
+      "HF_TOKEN",
+      "KIMI_API_KEY",
+      "META_API_KEY",
+      "MINIMAX_API_KEY",
+      "MINIMAX_CN_API_KEY",
+      "MISTRAL_API_KEY",
+      "MOONSHOT_API_KEY",
+      "NVIDIA_API_KEY",
+      "OPENAI_API_KEY",
+      "OPENCODE_API_KEY",
+      "OPENROUTER_API_KEY",
+      "QWEN_TOKEN_PLAN_API_KEY",
+      "QWEN_TOKEN_PLAN_CN_API_KEY",
+      "RADIUS_API_KEY",
+      "TOGETHER_API_KEY",
+      "XAI_API_KEY",
+      "XIAOMI_API_KEY",
+      "XIAOMI_TOKEN_PLAN_AMS_API_KEY",
+      "XIAOMI_TOKEN_PLAN_CN_API_KEY",
+      "XIAOMI_TOKEN_PLAN_SGP_API_KEY",
+      "ZAI_API_KEY",
+      "ZAI_CODING_CN_API_KEY"
+    ],
+    // Pi uses its own `provider/id` model names; set one with localAgent({ model }).
+    modelAliases: {},
+    nativeSchema() {
+      return null;
+    },
+    buildInvocation(request, model) {
+      const args = [
+        "--mode",
+        "json",
+        "--no-session",
+        "--no-tools",
+        "--no-builtin-tools",
+        "--no-extensions",
+        "--no-skills",
+        "--no-prompt-templates",
+        "--no-themes",
+        "--no-context-files",
+        "--system-prompt",
+        SYSTEM_PROMPT,
+        "--offline"
+      ];
+      if (model !== void 0) {
+        args.push("--model", model);
+      }
+      return { args, stdin: request.prompt };
+    },
+    readResult(result, request) {
+      const events = parseEvents(result.stdout);
+      const assistant = events.filter((event) => event.type === "message_end" && event.message?.role === "assistant").map((event) => event.message).pop();
+      if (assistant && (assistant.stopReason === "error" || assistant.errorMessage)) {
+        return failed((0, failure_classification_1.classifyAgentFailure)("pi", assistant.errorMessage ?? "pi stopped with an error"));
+      }
+      const retryFailed = events.find((event) => event.type === "auto_retry_end" && event.success === false);
+      if (retryFailed) {
+        return failed((0, failure_classification_1.classifyAgentFailure)("pi", retryFailed.finalError ?? "auto retry failed"));
+      }
+      if (!assistant) {
+        return failed((0, failure_classification_1.classifyAgentFailure)("pi", `exited with code ${result.exitCode} without a reply: ${(0, failure_classification_1.stderrTail)(result)}`));
+      }
+      const text = (assistant.content ?? []).filter((block) => block.type === "text").map((block) => block.text ?? "").join("");
+      const usage = assistant.usage ?? {};
+      const promptTokens = (usage.input ?? 0) + (usage.cacheRead ?? 0) + (usage.cacheWrite ?? 0);
+      const completionTokens = usage.output ?? 0;
+      const reported = {
+        usage: { promptTokens, completionTokens, totalTokens: promptTokens + completionTokens },
+        model: assistant.model,
+        costUsdEstimate: usage.cost?.total
+      };
+      if (request.enforcement.kind === "in-prompt") {
+        return { kind: "answered", answer: { kind: "text", text }, ...reported };
+      }
+      let envelope;
+      try {
+        envelope = JSON.parse(text);
+      } catch {
+        return failed({ kind: "transient", message: `pi reply is not JSON: ${text}` });
+      }
+      return { kind: "answered", answer: { kind: "structured", envelope }, ...reported };
+    },
+    // Pi has no provider-agnostic login check: `pi auth check` requires --provider
+    // or --model, and the provider differs per user (it is not known until Pi's
+    // settings are read). Report installed Pi as logged in; the first real call
+    // surfaces any auth error, which classifyAgentFailure marks fatal.
+    loginStatusArgs: ["--version"],
+    readLoginStatus(result) {
+      return {
+        loggedIn: result.exitCode === 0,
+        detail: "installed; auth is verified on the first call"
+      };
+    }
+  };
+  function failed(failure) {
+    return { kind: "failed", failure };
+  }
+  function parseEvents(stdout) {
+    const events = [];
+    for (const line of stdout.split("\n")) {
+      const trimmed = line.replace(/\r$/, "").trim();
+      if (!trimmed.startsWith("{"))
+        continue;
+      try {
+        const parsed = JSON.parse(trimmed);
+        if (typeof parsed === "object" && parsed !== null && !Array.isArray(parsed)) {
+          events.push(parsed);
+        }
+      } catch {
+      }
+    }
+    return events;
+  }
+  return pi;
+}
+
 var hasRequiredAdapters;
 function requireAdapters() {
   if (hasRequiredAdapters) return adapters;
@@ -372,9 +527,11 @@ function requireAdapters() {
   adapters.AGENT_ADAPTERS = void 0;
   const claude_code_1 = requireClaudeCode();
   const codex_1 = requireCodex();
+  const pi_1 = requirePi();
   adapters.AGENT_ADAPTERS = {
     "claude-code": claude_code_1.claudeCodeAdapter,
-    codex: codex_1.codexAdapter
+    codex: codex_1.codexAdapter,
+    pi: pi_1.piAdapter
   };
   return adapters;
 }
@@ -389,7 +546,7 @@ function requireAgentId() {
     Object.defineProperty(exports$1, "__esModule", { value: true });
     exports$1.LOCAL_AGENT_IDS = void 0;
     exports$1.isLocalAgentId = isLocalAgentId;
-    exports$1.LOCAL_AGENT_IDS = ["claude-code", "codex"];
+    exports$1.LOCAL_AGENT_IDS = ["claude-code", "codex", "pi"];
     const LOCAL_AGENT_ID_SET = new Set(exports$1.LOCAL_AGENT_IDS);
     function isLocalAgentId(value) {
       return typeof value === "string" && LOCAL_AGENT_ID_SET.has(value);

package/dist/types/nua-llm-core/modules/engine/local-agent/adapters/pi.d.ts

@@ -0,0 +1,2 @@
+import { AgentAdapter } from "./types";
+export declare const piAdapter: AgentAdapter;

package/dist/types/nua-llm-core/modules/engine/local-agent/agent-id.d.ts

@@ -1,4 +1,4 @@
 /** Supported coding-agent CLIs, in the order findLocalAgent() tries them. */
-export declare const LOCAL_AGENT_IDS: readonly ["claude-code", "codex"];
+export declare const LOCAL_AGENT_IDS: readonly ["claude-code", "codex", "pi"];
 export type LocalAgentId = (typeof LOCAL_AGENT_IDS)[number];
 export declare function isLocalAgentId(value: unknown): value is LocalAgentId;

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "nuabase",
   "description": "Nuabase TypeScript client to create typed functions from LLM prompts",
-  "version": "2.3.5",
+  "version": "2.3.7",
   "type": "module",
   "author": "Jasim A Basheer <jasim@nuabase.com>",
   "engines": {