@mono-agent/agent-runtime 0.11.2 → 0.11.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mono-agent/agent-runtime",
-  "version": "0.11.2",
+  "version": "0.11.3",
   "description": "Agent runtime supporting Claude SDK/CLI, Codex, OpenCode, and Pi SDK bridges out of the box",
   "type": "module",
   "license": "GPL-3.0-only",

package/src/ai/providers/pi-native/result-builder.js

@@ -94,9 +94,9 @@ export function emitCapabilitiesResolved(onEvent, { sdk, model, capabilitiesUsed
 
 /**
  * The aborted-run result (entry pre-check / pre-request abort). Pure factory.
- * @param {{resolved: any, options: any, events: any[], runtimeWarnings: any[], start: number, providerSessionId: string}} params
+ * @param {{resolved: any, options: any, events: any[], runtimeWarnings: any[], start: number, providerSessionId: string, piTransport: string}} params
  */
-export function abortedResult({ resolved, options, events, runtimeWarnings, start, providerSessionId }) {
+export function abortedResult({ resolved, options, events, runtimeWarnings, start, providerSessionId, piTransport }) {
   return {
     text: null,
     thinking: "",
@@ -116,6 +116,7 @@ export function abortedResult({ resolved, options, events, runtimeWarnings, star
       provider_session_id: providerSessionId,
       pi_stop_reason: "aborted",
       pi_engine: "native",
+      pi_transport_requested: piTransport,
       external_abort: true,
     },
   };
@@ -201,6 +202,7 @@ export function buildErrorResult(params) {
     providerSessionId,
     runtimeWarnings,
     isRetryable,
+    piTransport,
   } = params;
   return {
     text: assistantTexts.join("") || null,
@@ -221,6 +223,7 @@ export function buildErrorResult(params) {
       max_turns_hit: maxTurnsHit,
       provider_session_id: providerSessionId,
       pi_engine: "native",
+      pi_transport_requested: piTransport,
       pi_error_retryable: isRetryable,
     },
     failureKind: externalAbort ? null : failureKindForPiError(errorMessage, {}, { maxTurnsHit }),
@@ -230,6 +233,7 @@ export function buildErrorResult(params) {
       provider_session_id: providerSessionId,
       pi_stop_reason: externalAbort ? "aborted" : "error",
       pi_engine: "native",
+      pi_transport_requested: piTransport,
       max_turns_hit: maxTurnsHit,
       turn_count: turnCount,
       external_abort: externalAbort,
@@ -252,6 +256,7 @@ export function buildDiagnostics(params) {
     runAssistantCount,
     externalAbort,
     maxRetries,
+    piTransport,
     lastToolName,
     structuredRetry,
     contextCompactionDiagnostics,
@@ -265,6 +270,7 @@ export function buildDiagnostics(params) {
     turn_count: turnCount || runAssistantCount,
     external_abort: externalAbort,
     pi_max_retries: maxRetries,
+    pi_transport_requested: piTransport,
     ...(lastToolName ? { last_tool_name: lastToolName } : {}),
     ...structuredOutputRetryDiagnostics(
       structuredRetry.attempts,

package/src/ai/providers/pi-native/session-lifecycle.js

@@ -177,6 +177,7 @@ function sessionUnavailableResult({
   errorMessage,
   failureKind,
   piErrorCode,
+  piTransport,
 }) {
   return {
     text: null,
@@ -196,6 +197,7 @@ function sessionUnavailableResult({
       provider_session_id: sessionId,
       pi_error_code: piErrorCode,
       pi_engine: "native",
+      pi_transport_requested: piTransport,
     },
   };
 }
@@ -221,6 +223,7 @@ export async function resolveSession(runState, {
   events,
   runtimeWarnings,
   start,
+  piTransport,
 }) {
   // Resume check first: a session miss must stay cheap (no tool/MCP/harness
   // init). This mirrors the legacy bridge's fail-fast contract.
@@ -307,6 +310,7 @@ export async function resolveSession(runState, {
             errorMessage: `Pi session ${requestedSessionId} is not live`,
             failureKind: "session_not_found",
             piErrorCode: "pi_session_not_found",
+            piTransport,
           }),
         };
       }
@@ -344,6 +348,7 @@ export async function resolveSession(runState, {
               : `Pi session ${requestedSessionId} is busy with another turn`,
             failureKind: missing ? "session_not_found" : "session_busy",
             piErrorCode: missing ? "pi_session_not_found" : "pi_session_busy",
+            piTransport,
           }),
         };
       }

package/src/ai/providers/pi-native/transport.js

@@ -0,0 +1,16 @@
+// @ts-check
+
+const PI_TRANSPORTS = new Set(["auto", "sse", "websocket", "websocket-cached"]);
+
+/**
+ * Normalize the public runtime option at the untyped JS boundary. Typed/config
+ * callers already validate it; direct JavaScript callers fail safely to Pi's
+ * compatibility-preserving automatic transport selection.
+ * @param {unknown} value
+ * @returns {"auto" | "sse" | "websocket" | "websocket-cached"}
+ */
+export function resolvePiTransport(value) {
+  return typeof value === "string" && PI_TRANSPORTS.has(value)
+    ? /** @type {"auto" | "sse" | "websocket" | "websocket-cached"} */ (value)
+    : "auto";
+}

package/src/ai/providers/pi-native/turn-runner.js

@@ -172,6 +172,7 @@ export function buildTurnHarness(runState, {
   systemPrompt,
   outputSchema,
   tools,
+  transport,
   maxRetries,
   maxRetryDelayMs,
   steeringMode,
@@ -187,7 +188,7 @@ export function buildTurnHarness(runState, {
     thinkingLevel,
     systemPrompt: appendStructuredOutputInstruction(systemPrompt, outputSchema, options.prompts),
     tools,
-    streamOptions: { maxRetries, maxRetryDelayMs },
+    streamOptions: { transport, maxRetries, maxRetryDelayMs },
     steeringMode,
     followUpMode: steeringMode,
   });

package/src/ai/providers/pi-native.js

@@ -76,6 +76,7 @@ import {
   startLiveInput,
   thinkingLevelForEffort,
 } from "./pi-native/turn-runner.js";
+import { resolvePiTransport } from "./pi-native/transport.js";
 
 async function resolveApiKey(provider, { apiKeys, resolvePiApiKey, runtimeWarnings }) {
   if (apiKeys?.has(provider)) return apiKeys.get(provider);
@@ -302,6 +303,7 @@ export async function generatePiNativeResponse(systemPrompt, options = {}) {
   let structuredOutputFinalizationRetryAttempts = 0;
   let structuredOutputFinalizationRetryReason = null;
   let structuredOutputFinalizationRetryFailed = false;
+  const piTransport = resolvePiTransport(options.piTransport);
 
   const onEvent = (event) => emitCaptured(events, options.onEvent, event);
   const approvalManager = options.onToolApprovalRequest
@@ -316,7 +318,7 @@ export async function generatePiNativeResponse(systemPrompt, options = {}) {
     : null;
 
   if (options.abortSignal?.aborted) {
-    return abortedResult({ resolved, options, events, runtimeWarnings, start, providerSessionId });
+    return abortedResult({ resolved, options, events, runtimeWarnings, start, providerSessionId, piTransport });
   }
 
   const durableRepo = resolveDurableNativeSessionRepo(options.piSessionsRoot);
@@ -341,6 +343,7 @@ export async function generatePiNativeResponse(systemPrompt, options = {}) {
       events,
       runtimeWarnings,
       start,
+      piTransport,
     });
     if (resolvedSession.done) return resolvedSession.result;
 
@@ -432,6 +435,7 @@ export async function generatePiNativeResponse(systemPrompt, options = {}) {
       systemPrompt,
       outputSchema: options.outputSchema,
       tools,
+      transport: piTransport,
       maxRetries,
       maxRetryDelayMs,
       steeringMode: toolSteeringMode,
@@ -484,7 +488,7 @@ export async function generatePiNativeResponse(systemPrompt, options = {}) {
       // and closes MCP clients. For a resume no transcript was appended yet
       // (prompt never ran), so the live session needs no rollback.
       await discardUncommittedSession(runState, { durableRepo });
-      return abortedResult({ resolved, options, events, runtimeWarnings, start, providerSessionId });
+      return abortedResult({ resolved, options, events, runtimeWarnings, start, providerSessionId, piTransport });
     }
 
     // Compaction policy against the LIVE model's context window, then proactive
@@ -619,6 +623,7 @@ export async function generatePiNativeResponse(systemPrompt, options = {}) {
       runAssistantCount,
       externalAbort: runState.externalAbort,
       maxRetries,
+      piTransport,
       lastToolName: runState.lastToolName,
       structuredRetry,
       contextCompactionDiagnostics: runState.compaction.diagnostics,
@@ -737,6 +742,7 @@ export async function generatePiNativeResponse(systemPrompt, options = {}) {
       providerSessionId,
       runtimeWarnings,
       isRetryable,
+      piTransport,
     });
   } finally {
     if (runState.sessionEntry) runState.sessionEntry.busy = false;

package/types/ai/providers/pi-native/result-builder.d.ts

@@ -53,15 +53,16 @@ export function emitCapabilitiesResolved(onEvent: (event: any) => void, { sdk, m
 }): void;
 /**
  * The aborted-run result (entry pre-check / pre-request abort). Pure factory.
- * @param {{resolved: any, options: any, events: any[], runtimeWarnings: any[], start: number, providerSessionId: string}} params
+ * @param {{resolved: any, options: any, events: any[], runtimeWarnings: any[], start: number, providerSessionId: string, piTransport: string}} params
  */
-export function abortedResult({ resolved, options, events, runtimeWarnings, start, providerSessionId }: {
+export function abortedResult({ resolved, options, events, runtimeWarnings, start, providerSessionId, piTransport }: {
     resolved: any;
     options: any;
     events: any[];
     runtimeWarnings: any[];
     start: number;
     providerSessionId: string;
+    piTransport: string;
 }): {
     text: any;
     thinking: string;
@@ -81,6 +82,7 @@ export function abortedResult({ resolved, options, events, runtimeWarnings, star
         provider_session_id: string;
         pi_stop_reason: string;
         pi_engine: string;
+        pi_transport_requested: string;
         external_abort: boolean;
     };
 };
@@ -140,6 +142,7 @@ export function buildErrorResult(params: object): {
         max_turns_hit: any;
         provider_session_id: any;
         pi_engine: string;
+        pi_transport_requested: any;
         pi_error_retryable: any;
     };
     failureKind: string;
@@ -149,6 +152,7 @@ export function buildErrorResult(params: object): {
         provider_session_id: any;
         pi_stop_reason: string;
         pi_engine: string;
+        pi_transport_requested: any;
         max_turns_hit: any;
         turn_count: any;
         external_abort: any;

package/types/ai/providers/pi-native/session-lifecycle.d.ts

@@ -17,7 +17,7 @@ export function retireDurableNativeSession(providerSessionId: any, piSessionsRoo
  * @param {any} params
  * @returns {Promise<{done: true, result: any} | {done: false}>}
  */
-export function resolveSession(runState: any, { requestedSessionId, providerSessionId, durableRepo, sessionTtlMs, cwd, resolved, options, events, runtimeWarnings, start, }: any): Promise<{
+export function resolveSession(runState: any, { requestedSessionId, providerSessionId, durableRepo, sessionTtlMs, cwd, resolved, options, events, runtimeWarnings, start, piTransport, }: any): Promise<{
     done: true;
     result: any;
 } | {

package/types/ai/providers/pi-native/transport.d.ts

@@ -0,0 +1,8 @@
+/**
+ * Normalize the public runtime option at the untyped JS boundary. Typed/config
+ * callers already validate it; direct JavaScript callers fail safely to Pi's
+ * compatibility-preserving automatic transport selection.
+ * @param {unknown} value
+ * @returns {"auto" | "sse" | "websocket" | "websocket-cached"}
+ */
+export function resolvePiTransport(value: unknown): "auto" | "sse" | "websocket" | "websocket-cached";

package/types/ai/providers/pi-native/turn-runner.d.ts

@@ -30,7 +30,7 @@ export function thinkingLevelForEffort(effort: string, capabilities: any): strin
  * @param {any} params
  * @returns {any}
  */
-export function buildTurnHarness(runState: any, { cwd, session, piModels, model, thinkingLevel, systemPrompt, outputSchema, tools, maxRetries, maxRetryDelayMs, steeringMode, onEvent, options, toolLimits, }: any): any;
+export function buildTurnHarness(runState: any, { cwd, session, piModels, model, thinkingLevel, systemPrompt, outputSchema, tools, transport, maxRetries, maxRetryDelayMs, steeringMode, onEvent, options, toolLimits, }: any): any;
 /**
  * Start the live-input steering consumer. Consumes follow-up messages and steers
  * the harness mid-run; the consumer is tied to run completion (an internal