@ycodium-ai/agent-pi 0.2.3598 → 0.2.3608

package/README.md

@@ -21,8 +21,8 @@ in `/v1`, without `/chat/completions` or `/responses`. For Anthropic Messages,
 supply the service root or a base URL ending in `/v1`; the runtime resolves the
 request route to `/v1/messages`. URLs must not contain embedded credentials,
 query parameters or a fragment. The model ID must be accepted by that service. The instance exposes
-that one configured model; its readiness check does not make a model request or
-verify API credentials. The first request checks access to the service.
+the configured model plus the instance's custom models, all on the same endpoint and key;
+its readiness check does not make a model request or verify API credentials. The first request checks access to the service.
 
 ## Install
 
@@ -48,7 +48,8 @@ is shown as invalid and must be explicitly replaced. Advanced configurations
 can also supply `YCODIUM_PI_API_BASE_URL`.
 
 The plugin starts `pi --mode rpc` through Ycodium's managed process API. It
-provides temporary model and MCP configuration, and enables Pi's built-in MCP and tool-search
+provides temporary model and MCP configuration (every listed model under one `ycodium`
+provider), and enables Pi's built-in MCP and tool-search
 extensions. MCP tools are discovered on demand rather than all advertised in the first request.
 Native session files live in a directory specific
 to the instance under `~/.pi/ycodium/`, unless a session directory is configured
@@ -60,9 +61,11 @@ history and require a new thread. Missing saved history is reported as an error;
 does not silently replace it with an empty conversation.
 
 Pi runs with full process permissions. It provides no sandbox or native per-tool
-approval flow. Context compaction and steering use native RPC operations. This
+approval flow. Context compaction and steering use native RPC operations. Choosing another listed model
+sends `set_model` before the next prompt in the same process; steering keeps the current model.
+Models are declared without reasoning support, so no thinking-level option is exposed. This
 integration does not offer OAuth, an auxiliary text-generation assistant,
-mid-session model switching, inline image attachments, or a guarantee that an
+inline image attachments, or a guarantee that an
 interrupted tool's external side effects have stopped. MCP transports are stdio
 and HTTP; legacy SSE is unsupported.
 

package/dist/index.js

@@ -5,6 +5,7 @@ import { definePlugin } from "@ycodium-ai/plugin-api";
 import * as NodeCrypto from "node:crypto";
 import * as NodeOS from "node:os";
 import * as NodePath from "node:path";
+import { matchCatalogModel } from "@ycodium-ai/plugin-api/executor";
 
 // src/protocol.ts
 var PI_API_PROTOCOLS = [
@@ -37,11 +38,16 @@ function parseConfig(value, instanceId) {
   );
   const configuredHome = string(config.homePath).trim();
   const homePath = configuredHome.startsWith("~/") || configuredHome.startsWith("~\\") ? NodePath.join(NodeOS.homedir(), configuredHome.slice(2)) : configuredHome;
+  const model = string(config.model).trim();
+  const custom = (Array.isArray(config.customModels) ? config.customModels : []).map(
+    (entry) => string(typeof entry === "string" ? entry : record(entry).slug).trim()
+  );
   return {
     binaryPath: string(config.binaryPath).trim() || "pi",
     apiBaseUrl: string(config.apiBaseUrl).trim() || environment.YCODIUM_PI_API_BASE_URL?.trim() || "",
     apiProtocol: readPiApiProtocol(config.apiProtocol),
-    model: string(config.model).trim(),
+    model,
+    models: [...new Set([model, ...custom].filter(Boolean))],
     environment,
     homePath: homePath ? NodePath.resolve(homePath) : NodePath.join(NodeOS.homedir(), ".pi", "ycodium", identifier(instanceId))
   };
@@ -65,6 +71,18 @@ function validateConnection(config) {
   if (!config.model) throw new Error("Pi model ID is required.");
   if (!config.environment[PI_API_KEY_ENV]?.trim()) throw new Error("Pi API key is required.");
 }
+function requestedModel(config, selection) {
+  const raw = string(record(selection).model).trim();
+  if (!raw) return void 0;
+  const id = matchCatalogModel(raw, config.models) ?? matchCatalogModel(raw.replace(/^ycodium\//, ""), config.models);
+  if (id) return id;
+  throw Object.assign(
+    new Error(
+      `Pi instance does not list model "${raw}". Choose one of: ${config.models.join(", ")}. Provider setting changes apply after the session restarts.`
+    ),
+    { code: "provider-rejected" }
+  );
+}
 function redact(config, value) {
   let text = value instanceof Error ? value.message : String(value);
   for (const secret of Object.values(config.environment).filter(Boolean).sort((a, b) => b.length - a.length))
@@ -97,7 +115,7 @@ var PI_CARD = {
   verifiedSkills: false,
   everydayHarness: false,
   capabilities: {
-    sessionModelSwitch: "unsupported",
+    sessionModelSwitch: "in-session",
     runtimeModeSwitch: "next-turn",
     contextCompaction: "native",
     turnSteering: true
@@ -147,7 +165,8 @@ import {
   describeExecutorSpawnFailure,
   ExecutorSpawnAbortedError,
   ExecutorSpawnRefusedError,
-  isExecutorSpawnFailure
+  isExecutorSpawnFailure,
+  raceAbort
 } from "@ycodium-ai/plugin-api/executor";
 import { runProcessOperation } from "@ycodium-ai/plugin-api/process-operation";
 
@@ -349,7 +368,7 @@ async function createLaunch(config, input, cursor) {
             baseUrl,
             api: apiProtocol,
             apiKey: `\${${PI_API_KEY_ENV}}`,
-            models: [{ id: config.model }]
+            models: config.models.map((id) => ({ id }))
           }
         }
       }),
@@ -381,7 +400,7 @@ async function createLaunch(config, input, cursor) {
       "rpc",
       "--no-approve",
       "--model",
-      `ycodium/${config.model}`,
+      `ycodium/${requestedModel(config, input?.modelSelection) ?? config.model}`,
       "--no-extensions",
       "--no-skills"
     ];
@@ -862,13 +881,6 @@ function createPiInstance(value, host) {
   const disposed = new AbortController();
   const now = () => (/* @__PURE__ */ new Date()).toISOString();
   const nativeMode = PI_NATIVE_MODE_REPORT;
-  const validateModel = (selection) => {
-    const model = string(record(selection).model).trim();
-    if (model && model !== config.model && model !== `ycodium/${config.model}`)
-      throw Object.assign(new Error("Pi can only run the model configured for this instance."), {
-        code: "provider-rejected"
-      });
-  };
   const safeError = (cause) => {
     if (cause instanceof ExecutorSpawnRefusedError || cause instanceof ExecutorSpawnAbortedError)
       return cause;
@@ -992,6 +1004,8 @@ function createPiInstance(value, host) {
   const abortTurn = async (booked, signal) => {
     const turn = booked.turn, conn = booked.connection;
     if (!turn || turn.completed || !conn) return;
+    if (booked.dispatching) await raceAbort(booked.dispatching, signal);
+    if (turn.completed) return;
     await conn.rpc.request("clear_queue", {}, signal, 1e4);
     await conn.rpc.request("abort", {}, signal, 1e4);
     await waitForSettled(booked, turn, signal);
@@ -1019,7 +1033,7 @@ function createPiInstance(value, host) {
   const startOne = async (input, signal) => {
     ensureActive();
     validateConnection(config);
-    validateModel(input.modelSelection);
+    const model = requestedModel(config, input.modelSelection) ?? config.model;
     signal?.throwIfAborted();
     const nativeModeId = piNativeModeFor(input.runtimeMode, input.nativeModeId);
     const existing = threads.get(input.threadId);
@@ -1037,7 +1051,7 @@ function createPiInstance(value, host) {
       runtimeMode: input.runtimeMode,
       nativeModeId,
       status: "starting",
-      model: config.model,
+      model,
       createdAt: now(),
       updatedAt: now(),
       resumeCursor: cursor,
@@ -1199,10 +1213,13 @@ function createPiInstance(value, host) {
     startSession: start,
     sendTurn: async (input, signal) => {
       signal?.throwIfAborted();
-      validateModel(input.modelSelection);
+      const requested = requestedModel(config, input.modelSelection);
       const booked = requireThread(input.threadId), conn = connection(booked);
       if (booked.stopping) throw new Error("Pi is stopping the current turn.");
       if (input.expectedTurnId !== void 0) {
+        if (booked.dispatching) await raceAbort(booked.dispatching, signal);
+        signal?.throwIfAborted();
+        if (booked.stopping || booked.closing) throw new Error("Pi is stopping the current turn.");
         if (!booked.turn || booked.turn.id !== input.expectedTurnId || booked.turn.completed)
           throw Object.assign(new Error("The target Pi turn has ended."), {
             code: "provider-rejected"
@@ -1221,7 +1238,8 @@ function createPiInstance(value, host) {
       }
       if (booked.turn) throw new Error("Pi already has an active turn.");
       const cursor = record(booked.session.resumeCursor);
-      const turn = new PiTurn(booked.emit, config.model, string(cursor.sessionId), () => {
+      const model = requested ?? booked.session.model ?? config.model;
+      const turn = new PiTurn(booked.emit, model, string(cursor.sessionId), () => {
         if (booked.turn === turn) booked.turn = void 0;
         booked.session.status = "idle";
         booked.session.updatedAt = now();
@@ -1229,11 +1247,26 @@ function createPiInstance(value, host) {
       booked.turn = turn;
       booked.session.status = "running";
       turn.start();
+      const issued = Promise.withResolvers();
+      const dispatch = (async () => {
+        try {
+          if (model !== booked.session.model) {
+            await conn.rpc.request("set_model", { provider: "ycodium", modelId: model }, signal);
+            booked.session.model = model;
+          }
+          if (booked.stopping || booked.closing) return turn.finish("interrupted");
+          const prompting = conn.rpc.request("prompt", { message: input.input ?? "" }, signal);
+          issued.resolve();
+          const response = record(await prompting);
+          if (response.disposition === "handled") turn.finish("completed");
+        } finally {
+          issued.resolve();
+        }
+      })();
+      const dispatched = issued.promise;
+      booked.dispatching = dispatched;
       try {
-        const response = record(
-          await conn.rpc.request("prompt", { message: input.input ?? "" }, signal)
-        );
-        if (response.disposition === "handled") turn.finish("completed");
+        await dispatch;
         return {
           threadId: input.threadId,
           turnId: turn.id,
@@ -1243,6 +1276,8 @@ function createPiInstance(value, host) {
         turn.finish("failed", redact(config, cause));
         if (!(cause instanceof PiResponseError)) await close(booked);
         throw safeError(cause);
+      } finally {
+        if (booked.dispatching === dispatched) booked.dispatching = void 0;
       }
     },
     interruptTurn: async (threadId, turnId, signal) => {
@@ -1375,7 +1410,7 @@ function createPiInstance(value, host) {
 // src/index.ts
 var index_default = definePlugin({
   id: "agent-pi",
-  apiVersion: 48,
+  apiVersion: 49,
   executors: [
     { card: PI_CARD, createInstance: async (config, host) => createPiInstance(config, host) }
   ],

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ycodium-ai/agent-pi",
-  "version": "0.2.3598",
+  "version": "0.2.3608",
   "description": "Pi executor over the native RPC protocol.",
   "keywords": [
     "ycodium",
@@ -30,9 +30,9 @@
     "tailwindcss": "^4.0.0",
     "typescript": "~6.0.3",
     "vite-plus": "0.2.2",
-    "@ycodium-ai/plugin-api": "0.2.3598",
-    "ycodium-executor-testing": "0.0.0",
-    "@ycodium/contracts": "0.0.33"
+    "@ycodium-ai/plugin-api": "0.2.3608",
+    "@ycodium/contracts": "0.0.33",
+    "ycodium-executor-testing": "0.0.0"
   },
   "ycodium": {
     "plugin": "./dist/index.js",