@yolk-sdk/agent 0.1.0-canary.19 → 0.1.0-canary.21

package/README.md

@@ -22,7 +22,7 @@ Canary APIs are unstable. Keep all `@yolk-sdk/*` packages on the same version.
 | `@yolk-sdk/agent/loop`         | Stateless LLM/tool loop                                        |
 | `@yolk-sdk/agent/loop/testing` | Faux provider and tool executor test helpers                   |
 | `@yolk-sdk/agent/runtime`      | Transcript or append-backed runtime orchestration              |
-| `@yolk-sdk/agent/client`       | HTTP/NDJSON transport, HITL resume, and client state helpers   |
+| `@yolk-sdk/agent/client`       | HTTP/NDJSON transport, HITL resume, retry/error state helpers  |
 | `@yolk-sdk/agent/compaction`   | Pure host-owned context budget, planning, and transformer utils |
 | `@yolk-sdk/agent/tools`        | Tool module registry, `makeTool`, task/question tool contracts |
 | `@yolk-sdk/agent/react`        | Headless React chat hook, reducer, selectors, and render model |
@@ -43,6 +43,7 @@ Canary APIs are unstable. Keep all `@yolk-sdk/*` packages on the same version.
 import {
   hitlResponseEvent,
   makeSubagentRunId,
+  ProviderErrorInfo,
   questionResponseStructuredContent,
   UserMessage
 } from '@yolk-sdk/agent/protocol'
@@ -89,6 +90,24 @@ const program = run({
 // Provide LLM provider, loop config, context transformer, and tool executor layers in the host app.
 ```
 
+## Provider failures and retries
+
+Provider adapters classify safe failure metadata at the boundary. The loop owns bounded retry
+policy and emits protocol-visible retry/error state:
+
+- `ProviderErrorInfo` carries safe provider id, failure kind, HTTP status, provider code, and
+  optional `retryAfterMs`.
+- `AgentRetry.provider` exposes current retry metadata and chosen `delayMs`.
+- `AgentError.provider` preserves final terminal metadata.
+- `AgentErrorCode` includes `rate_limit`, `overloaded`, `context_overflow`, and generic
+  `provider_error`.
+- Client and React state keep `error: string | null` for compatibility and add typed `errorInfo` /
+  `retryInfo`.
+- `buildAgentChatItems` can project active retry state as an `AgentChatItem` with `_tag: 'Retry'`.
+
+Raw provider response bodies stay out of protocol/UI. Hosts own durable persistence and display of
+typed retry/error state.
+
 ## Context compaction
 
 `@yolk-sdk/agent/compaction` provides pure scaffolding only. It does not call a model, persist
@@ -130,6 +149,10 @@ Build sources with `inlineBase64AttachmentSource`, `urlAttachmentSource`, or
 for simple apps, or persist `Ref` values and call `resolveContentAttachmentSources` at your storage
 boundary before provider execution. Host apps own upload, auth, retention, and ref hydration policy.
 
+For text files, use `documentPartFromText`, `inferTextDocumentMimeType`, and the client helper
+`documentPartFromTextFile` to create UTF-8 inline `DocumentPart` values without trusting filename
+extensions over explicit non-text MIME types.
+
 Use model capabilities like `textOnlyModelCapabilities`, `textImageModelCapabilities`, or
 `textImageDocumentModelCapabilities` so the loop rejects unsupported inputs before provider calls.
 
@@ -216,6 +239,7 @@ defects, storage failures, or implementation bugs.
 
 - Choose models/providers and map provider streams into protocol events.
 - Persist sessions, transcripts, and append logs.
+- Persist terminal provider failures and clear active run ids where applicable.
 - Provide tools, approval policy, auth, storage, and observability.
 - Compact context and decide memory/search policy.
 

package/dist/client/attachments.d.mts

@@ -0,0 +1,8 @@
+//#region src/client/attachments.d.ts
+declare const textFromBlob: (blob: Blob) => Promise<string>;
+declare const documentPartFromTextFile: (file: File, options?: {
+  readonly title?: string;
+}) => Promise<import("@yolk-sdk/agent/protocol").DocumentPart | undefined>;
+//#endregion
+export { documentPartFromTextFile, textFromBlob };
+//# sourceMappingURL=attachments.d.mts.map

package/dist/client/attachments.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"attachments.d.mts","names":[],"sources":["../../src/client/attachments.ts"],"mappings":";cAEa,YAAA,GAAgB,IAAA,EAAM,IAAA,KAAI,OAAA;AAAA,cAqB1B,wBAAA,GACX,IAAA,EAAM,IAAA,EACN,OAAA;EAAA,SACW,KAAA;AAAA,MACV,OAAA,oCAAA,YAAA"}

package/dist/client/attachments.mjs

@@ -0,0 +1,36 @@
+import { documentPartFromText, inferTextDocumentMimeType } from "@yolk-sdk/agent/protocol";
+//#region src/client/attachments.ts
+const textFromBlob = (blob) => {
+	if (typeof blob.text === "function") return blob.text();
+	if (typeof FileReader === "undefined") return Promise.reject(/* @__PURE__ */ new Error("Could not read text"));
+	return new Promise((resolve, reject) => {
+		const reader = new FileReader();
+		reader.addEventListener("load", () => {
+			if (typeof reader.result === "string") {
+				resolve(reader.result);
+				return;
+			}
+			reject(/* @__PURE__ */ new Error("Could not read text"));
+		});
+		reader.addEventListener("error", () => reject(/* @__PURE__ */ new Error("Could not read text")));
+		reader.addEventListener("abort", () => reject(/* @__PURE__ */ new Error("Could not read text")));
+		reader.readAsText(blob);
+	});
+};
+const documentPartFromTextFile = async (file, options) => {
+	const mimeType = inferTextDocumentMimeType({
+		filename: file.name,
+		mimeType: file.type
+	});
+	if (mimeType === void 0) return void 0;
+	return documentPartFromText({
+		text: await textFromBlob(file),
+		filename: file.name,
+		mimeType,
+		title: options?.title
+	});
+};
+//#endregion
+export { documentPartFromTextFile, textFromBlob };
+
+//# sourceMappingURL=attachments.mjs.map

package/dist/client/attachments.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"attachments.mjs","names":[],"sources":["../../src/client/attachments.ts"],"sourcesContent":["import { documentPartFromText, inferTextDocumentMimeType } from '@yolk-sdk/agent/protocol'\n\nexport const textFromBlob = (blob: Blob) => {\n  if (typeof blob.text === 'function') return blob.text()\n  if (typeof FileReader === 'undefined') return Promise.reject(new Error('Could not read text'))\n\n  return new Promise<string>((resolve, reject) => {\n    const reader = new FileReader()\n\n    reader.addEventListener('load', () => {\n      if (typeof reader.result === 'string') {\n        resolve(reader.result)\n        return\n      }\n\n      reject(new Error('Could not read text'))\n    })\n    reader.addEventListener('error', () => reject(new Error('Could not read text')))\n    reader.addEventListener('abort', () => reject(new Error('Could not read text')))\n    reader.readAsText(blob)\n  })\n}\n\nexport const documentPartFromTextFile = async (\n  file: File,\n  options?: {\n    readonly title?: string\n  }\n) => {\n  const mimeType = inferTextDocumentMimeType({\n    filename: file.name,\n    mimeType: file.type\n  })\n\n  if (mimeType === undefined) return undefined\n\n  const text = await textFromBlob(file)\n\n  return documentPartFromText({\n    text,\n    filename: file.name,\n    mimeType,\n    title: options?.title\n  })\n}\n"],"mappings":";;AAEA,MAAa,gBAAgB,SAAe;CAC1C,IAAI,OAAO,KAAK,SAAS,YAAY,OAAO,KAAK,KAAK;CACtD,IAAI,OAAO,eAAe,aAAa,OAAO,QAAQ,uBAAO,IAAI,MAAM,qBAAqB,CAAC;CAE7F,OAAO,IAAI,SAAiB,SAAS,WAAW;EAC9C,MAAM,SAAS,IAAI,WAAW;EAE9B,OAAO,iBAAiB,cAAc;GACpC,IAAI,OAAO,OAAO,WAAW,UAAU;IACrC,QAAQ,OAAO,MAAM;IACrB;GACF;GAEA,uBAAO,IAAI,MAAM,qBAAqB,CAAC;EACzC,CAAC;EACD,OAAO,iBAAiB,eAAe,uBAAO,IAAI,MAAM,qBAAqB,CAAC,CAAC;EAC/E,OAAO,iBAAiB,eAAe,uBAAO,IAAI,MAAM,qBAAqB,CAAC,CAAC;EAC/E,OAAO,WAAW,IAAI;CACxB,CAAC;AACH;AAEA,MAAa,2BAA2B,OACtC,MACA,YAGG;CACH,MAAM,WAAW,0BAA0B;EACzC,UAAU,KAAK;EACf,UAAU,KAAK;CACjB,CAAC;CAED,IAAI,aAAa,KAAA,GAAW,OAAO,KAAA;CAInC,OAAO,qBAAqB;EAC1B,MAAA,MAHiB,aAAa,IAAI;EAIlC,UAAU,KAAK;EACf;EACA,OAAO,SAAS;CAClB,CAAC;AACH"}

package/dist/client/index.d.mts

@@ -1,3 +1,4 @@
+import { documentPartFromTextFile, textFromBlob } from "./attachments.mjs";
 import { AgentClientState, AgentRunStatus, AgentToolRun, AgentTranscript, ApplyAgentEventOptions, appendAgentMessage, applyAgentEvent, completedToolRuns, initialAgentClientState, isActiveToolRun, markAgentAborted, markAgentError, reduceAgentEvents, submitAgentUserMessage, toolRunsFromHitlRequests } from "./state.mjs";
 import { AgentHttpResponseInfo, AgentTransportError, CancelAgentRunRequest, StreamAgentEventsRequest, StreamAgentRunEventsRequest, StreamCloudflareAgentEventsRequest, SubmitQuestionResponseRequest, SubmitToolApprovalResponseRequest, cancelAgentRun, collectAgentEvents, collectAgentEventsEffect, streamAgentEventStream, streamAgentEvents, streamAgentRunEventStream, streamAgentRunEvents, streamCloudflareAgentEventStream, streamCloudflareAgentEvents, streamQuestionResponseEventStream, streamToolApprovalResponseEventStream, submitQuestionResponse, submitToolApprovalResponse } from "./transport.mjs";
-export { type AgentClientState, type AgentHttpResponseInfo, type AgentRunStatus, type AgentToolRun, type AgentTranscript, AgentTransportError, type ApplyAgentEventOptions, type CancelAgentRunRequest, type StreamAgentEventsRequest, type StreamAgentRunEventsRequest, type StreamCloudflareAgentEventsRequest, type SubmitQuestionResponseRequest, type SubmitToolApprovalResponseRequest, appendAgentMessage, applyAgentEvent, cancelAgentRun, collectAgentEvents, collectAgentEventsEffect, completedToolRuns, initialAgentClientState, isActiveToolRun, markAgentAborted, markAgentError, reduceAgentEvents, streamAgentEventStream, streamAgentEvents, streamAgentRunEventStream, streamAgentRunEvents, streamCloudflareAgentEventStream, streamCloudflareAgentEvents, streamQuestionResponseEventStream, streamToolApprovalResponseEventStream, submitAgentUserMessage, submitQuestionResponse, submitToolApprovalResponse, toolRunsFromHitlRequests };
+export { type AgentClientState, type AgentHttpResponseInfo, type AgentRunStatus, type AgentToolRun, type AgentTranscript, AgentTransportError, type ApplyAgentEventOptions, type CancelAgentRunRequest, type StreamAgentEventsRequest, type StreamAgentRunEventsRequest, type StreamCloudflareAgentEventsRequest, type SubmitQuestionResponseRequest, type SubmitToolApprovalResponseRequest, appendAgentMessage, applyAgentEvent, cancelAgentRun, collectAgentEvents, collectAgentEventsEffect, completedToolRuns, documentPartFromTextFile, initialAgentClientState, isActiveToolRun, markAgentAborted, markAgentError, reduceAgentEvents, streamAgentEventStream, streamAgentEvents, streamAgentRunEventStream, streamAgentRunEvents, streamCloudflareAgentEventStream, streamCloudflareAgentEvents, streamQuestionResponseEventStream, streamToolApprovalResponseEventStream, submitAgentUserMessage, submitQuestionResponse, submitToolApprovalResponse, textFromBlob, toolRunsFromHitlRequests };

package/dist/client/index.mjs

@@ -1,3 +1,4 @@
+import { documentPartFromTextFile, textFromBlob } from "./attachments.mjs";
 import { appendAgentMessage, applyAgentEvent, completedToolRuns, initialAgentClientState, isActiveToolRun, markAgentAborted, markAgentError, reduceAgentEvents, submitAgentUserMessage, toolRunsFromHitlRequests } from "./state.mjs";
 import { AgentTransportError, cancelAgentRun, collectAgentEvents, collectAgentEventsEffect, streamAgentEventStream, streamAgentEvents, streamAgentRunEventStream, streamAgentRunEvents, streamCloudflareAgentEventStream, streamCloudflareAgentEvents, streamQuestionResponseEventStream, streamToolApprovalResponseEventStream, submitQuestionResponse, submitToolApprovalResponse } from "./transport.mjs";
-export { AgentTransportError, appendAgentMessage, applyAgentEvent, cancelAgentRun, collectAgentEvents, collectAgentEventsEffect, completedToolRuns, initialAgentClientState, isActiveToolRun, markAgentAborted, markAgentError, reduceAgentEvents, streamAgentEventStream, streamAgentEvents, streamAgentRunEventStream, streamAgentRunEvents, streamCloudflareAgentEventStream, streamCloudflareAgentEvents, streamQuestionResponseEventStream, streamToolApprovalResponseEventStream, submitAgentUserMessage, submitQuestionResponse, submitToolApprovalResponse, toolRunsFromHitlRequests };
+export { AgentTransportError, appendAgentMessage, applyAgentEvent, cancelAgentRun, collectAgentEvents, collectAgentEventsEffect, completedToolRuns, documentPartFromTextFile, initialAgentClientState, isActiveToolRun, markAgentAborted, markAgentError, reduceAgentEvents, streamAgentEventStream, streamAgentEvents, streamAgentRunEventStream, streamAgentRunEvents, streamCloudflareAgentEventStream, streamCloudflareAgentEvents, streamQuestionResponseEventStream, streamToolApprovalResponseEventStream, submitAgentUserMessage, submitQuestionResponse, submitToolApprovalResponse, textFromBlob, toolRunsFromHitlRequests };

package/dist/client/state.d.mts

@@ -1,4 +1,4 @@
-import { AgentEvent, AgentMessage, HitlRequest, QuestionRequest, QuestionResponse, ToolApprovalRequest, ToolCall, ToolResult, UserMessage } from "@yolk-sdk/agent/protocol";
+import { AgentError, AgentEvent, AgentMessage, AgentRetry, HitlRequest, QuestionRequest, QuestionResponse, ToolApprovalRequest, ToolCall, ToolResult, UserMessage } from "@yolk-sdk/agent/protocol";
 
 //#region src/client/state.d.ts
 type AgentRunStatus = 'idle' | 'running' | 'waiting' | 'done' | 'error' | 'aborted';
@@ -57,6 +57,8 @@ type AgentClientState = {
   readonly reasoning: string;
   readonly toolRuns: ReadonlyArray<AgentToolRun>;
   readonly error: string | null;
+  readonly errorInfo: AgentError | null;
+  readonly retryInfo: AgentRetry | null;
   readonly seenEventIds: ReadonlyArray<string>;
 };
 type ApplyAgentEventOptions = {
@@ -96,7 +98,7 @@ declare const appendAgentMessage: (messages: ReadonlyArray<AgentMessage>, messag
 declare const applyAgentEvent: (state: AgentClientState, event: AgentEvent) => AgentClientState;
 declare const applyAgentEventWithOptions: (state: AgentClientState, event: AgentEvent, options?: ApplyAgentEventOptions) => AgentClientState;
 declare const submitAgentUserMessage: (state: AgentClientState, message: UserMessage) => AgentClientState;
-declare const markAgentError: (state: AgentClientState, message?: string) => AgentClientState;
+declare const markAgentError: (state: AgentClientState, message?: string, errorInfo?: AgentError | null) => AgentClientState;
 declare const markAgentAborted: (state: AgentClientState) => AgentClientState;
 declare const reduceAgentEvents: (events: ReadonlyArray<AgentEvent>, initialState?: AgentClientState, options?: ApplyAgentEventOptions) => AgentClientState;
 //#endregion

package/dist/client/state.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"state.d.mts","names":[],"sources":["../../src/client/state.ts"],"mappings":";;;KAYY,cAAA;AAAA,KAEA,YAAA;EAAA,SAEG,IAAA;EAAA,SACA,EAAA;EAAA,SACA,IAAA;EAAA,SACA,KAAA;AAAA;EAAA,SAEA,IAAA;EAAA,SAA6B,IAAA,EAAM,QAAA;AAAA;EAAA,SAEnC,IAAA;EAAA,SACA,IAAA,EAAM,QAAA;EAAA,SACN,OAAA,GAAU,mBAAA;AAAA;EAAA,SAEV,IAAA;EAAA,SAAyB,UAAA;EAAA,SAA6B,MAAA;AAAA;EAAA,SACtD,IAAA;EAAA,SAAoC,OAAA,EAAS,eAAA;AAAA;EAAA,SAE7C,IAAA;EAAA,SACA,QAAA,EAAU,gBAAA;EAAA,SACV,OAAA,GAAU,eAAA;AAAA;EAAA,SAGV,IAAA;EAAA,SACA,QAAA,EAAU,gBAAA;EAAA,SACV,OAAA,GAAU,eAAA;AAAA;EAAA,SAEV,IAAA;EAAA,SAA4B,IAAA,EAAM,QAAA;EAAA,SAAmB,WAAA;AAAA;EAAA,SAErD,IAAA;EAAA,SACA,IAAA,EAAM,QAAA;EAAA,SACN,MAAA,EAAQ,UAAA;EAAA,SACR,WAAA;EAAA,SACA,SAAA;AAAA;EAAA,SAGA,IAAA;EAAA,SACA,IAAA,EAAM,QAAA;EAAA,SACN,OAAA;EAAA,SACA,SAAA;AAAA;EAAA,SAEA,IAAA;EAAA,SAAoC,IAAA,EAAM,QAAA;EAAA,SAAmB,MAAA,EAAQ,UAAA;AAAA;AAAA,KAIxE,gBAAA;EAAA,SACD,MAAA,EAAQ,cAAA;EAAA,SACR,QAAA,EAAU,aAAA,CAAc,YAAA;EAAA,SACxB,YAAA,EAAc,aAAA,CAAc,YAAA;EAAA,SAC5B,IAAA;EAAA,SACA,SAAA;EAAA,SACA,QAAA,EAAU,aAAA,CAAc,YAAA;EAAA,SACxB,KAAA;EAAA,SACA,YAAA,EAAc,aAAA;AAAA;AAAA,KAGb,sBAAA;EAAA,SACD,KAAK;AAAA;AAAA,KAGJ,eAAA,aAA4B,YAAA,KAAiB,KAAA,CAAM,YAAA;AAAA,cAElD,uBAAA,EAAyB,gBASrC;AAAA,cA+BY,eAAA,GAAmB,GAAA,EAAK,YAAA,KAAY,GAAA;EAAA;;;;;;iBA7FC,QAAA;AAAA;EAAA;iBAG7B,QAAA;EAAA,mBACI,mBAAA;AAAA;EAAA;oBAGmC,eAAA;AAAA;EAAA;iBAWX,QAAA;EAAA;;cAmFpC,iBAAA,GAAqB,IAAA,EAAM,aAAA,CAAc,YAAA;EAAA;iBAhFjC,QAAA;EAAA,iBACE,UAAA;EAAA;;;cAkFV,wBAAA,GACX,QAAA,EAAU,aAAA,CAAc,WAAA,MACvB,aAAA,CAAc,YAAA;AAAA,cAwFJ,kBAAA,GACX,QAAA,EAAU,aAAA,CAAc,YAAA,GACxB,OAAA,EAAS,YAAA,KACR,eAAA;AAAA,cAUU,eAAA,GAAmB,KAAA,EAAO,gBAAA,EAAkB,KAAA,EAAO,UAAA,KAAa,gBAAA;AAAA,cAMhE,0BAAA,GACX,KAAA,EAAO,gBAAA,EACP,KAAA,EAAO,UAAA,EACP,OAAA,GAAS,sBAAA,KACR,gBAAA;AAAA,cAuLU,sBAAA,GACX,KAAA,EAAO,gBAAA,EACP,OAAA,EAAS,WAAA,KACR,gBAAA;AAAA,cAYU,cAAA,GACX,KAAA,EAAO,gBAAA,EACP,OAAA,cACC,gBAKD;AAAA,cAEW,gBAAA,GAAoB,KAAA,EAAO,gBAAA,KAAmB,gBAKzD;AAAA,cAEW,iBAAA,GACX,MAAA,EAAQ,aAAA,CAAc,UAAA,GACtB,YAAA,GAAc,gBAAA,EACd,OAAA,GAAS,sBAAA,KAA2B,gBAAA"}
+{"version":3,"file":"state.d.mts","names":[],"sources":["../../src/client/state.ts"],"mappings":";;;KAcY,cAAA;AAAA,KAEA,YAAA;EAAA,SAEG,IAAA;EAAA,SACA,EAAA;EAAA,SACA,IAAA;EAAA,SACA,KAAA;AAAA;EAAA,SAEA,IAAA;EAAA,SAA6B,IAAA,EAAM,QAAA;AAAA;EAAA,SAEnC,IAAA;EAAA,SACA,IAAA,EAAM,QAAA;EAAA,SACN,OAAA,GAAU,mBAAA;AAAA;EAAA,SAEV,IAAA;EAAA,SAAyB,UAAA;EAAA,SAA6B,MAAA;AAAA;EAAA,SACtD,IAAA;EAAA,SAAoC,OAAA,EAAS,eAAA;AAAA;EAAA,SAE7C,IAAA;EAAA,SACA,QAAA,EAAU,gBAAA;EAAA,SACV,OAAA,GAAU,eAAA;AAAA;EAAA,SAGV,IAAA;EAAA,SACA,QAAA,EAAU,gBAAA;EAAA,SACV,OAAA,GAAU,eAAA;AAAA;EAAA,SAEV,IAAA;EAAA,SAA4B,IAAA,EAAM,QAAA;EAAA,SAAmB,WAAA;AAAA;EAAA,SAErD,IAAA;EAAA,SACA,IAAA,EAAM,QAAA;EAAA,SACN,MAAA,EAAQ,UAAA;EAAA,SACR,WAAA;EAAA,SACA,SAAA;AAAA;EAAA,SAGA,IAAA;EAAA,SACA,IAAA,EAAM,QAAA;EAAA,SACN,OAAA;EAAA,SACA,SAAA;AAAA;EAAA,SAEA,IAAA;EAAA,SAAoC,IAAA,EAAM,QAAA;EAAA,SAAmB,MAAA,EAAQ,UAAA;AAAA;AAAA,KAIxE,gBAAA;EAAA,SACD,MAAA,EAAQ,cAAA;EAAA,SACR,QAAA,EAAU,aAAA,CAAc,YAAA;EAAA,SACxB,YAAA,EAAc,aAAA,CAAc,YAAA;EAAA,SAC5B,IAAA;EAAA,SACA,SAAA;EAAA,SACA,QAAA,EAAU,aAAA,CAAc,YAAA;EAAA,SACxB,KAAA;EAAA,SACA,SAAA,EAAW,UAAA;EAAA,SACX,SAAA,EAAW,UAAA;EAAA,SACX,YAAA,EAAc,aAAA;AAAA;AAAA,KAGb,sBAAA;EAAA,SACD,KAAK;AAAA;AAAA,KAGJ,eAAA,aAA4B,YAAA,KAAiB,KAAA,CAAM,YAAA;AAAA,cAElD,uBAAA,EAAyB,gBAWrC;AAAA,cAkCY,eAAA,GAAmB,GAAA,EAAK,YAAA,KAAY,GAAA;EAAA;;;;;;iBApGC,QAAA;AAAA;EAAA;iBAG7B,QAAA;EAAA,mBACI,mBAAA;AAAA;EAAA;oBAGmC,eAAA;AAAA;EAAA;iBAWX,QAAA;EAAA;;cA0FpC,iBAAA,GAAqB,IAAA,EAAM,aAAA,CAAc,YAAA;EAAA;iBAvFjC,QAAA;EAAA,iBACE,UAAA;EAAA;;;cAyFV,wBAAA,GACX,QAAA,EAAU,aAAA,CAAc,WAAA,MACvB,aAAA,CAAc,YAAA;AAAA,cAwFJ,kBAAA,GACX,QAAA,EAAU,aAAA,CAAc,YAAA,GACxB,OAAA,EAAS,YAAA,KACR,eAAA;AAAA,cAUU,eAAA,GAAmB,KAAA,EAAO,gBAAA,EAAkB,KAAA,EAAO,UAAA,KAAa,gBAAA;AAAA,cAMhE,0BAAA,GACX,KAAA,EAAO,gBAAA,EACP,KAAA,EAAO,UAAA,EACP,OAAA,GAAS,sBAAA,KACR,gBAAA;AAAA,cAgMU,sBAAA,GACX,KAAA,EAAO,gBAAA,EACP,OAAA,EAAS,WAAA,KACR,gBAAA;AAAA,cAcU,cAAA,GACX,KAAA,EAAO,gBAAA,EACP,OAAA,WACA,SAAA,GAAW,UAAA,YACV,gBAAA;AAAA,cASU,gBAAA,GAAoB,KAAA,EAAO,gBAAA,KAAmB,gBAOzD;AAAA,cAEW,iBAAA,GACX,MAAA,EAAQ,aAAA,CAAc,UAAA,GACtB,YAAA,GAAc,gBAAA,EACd,OAAA,GAAS,sBAAA,KAA2B,gBAAA"}

package/dist/client/state.mjs

@@ -7,8 +7,14 @@ const initialAgentClientState = {
 	reasoning: "",
 	toolRuns: [],
 	error: null,
+	errorInfo: null,
+	retryInfo: null,
 	seenEventIds: []
 };
+const clearRetryInfo = (state) => state.retryInfo === null ? state : {
+	...state,
+	retryInfo: null
+};
 const hasSeenEvent = (state, event) => event.eventId !== void 0 && state.seenEventIds.includes(event.eventId);
 const rememberEvent = (state, event) => event.eventId === void 0 ? state : {
 	...state,
@@ -117,18 +123,20 @@ const applyAgentEventUnchecked = (state, event, nowMs) => {
 			reasoning: "",
 			liveMessages: [],
 			toolRuns: completedToolRuns(state.toolRuns),
-			error: null
+			error: null,
+			errorInfo: null,
+			retryInfo: null
 		};
-		case "AgentError": return markAgentError(state, event.message);
-		case "LLMTextDelta": return {
+		case "AgentError": return markAgentError(state, event.message, event);
+		case "LLMTextDelta": return clearRetryInfo({
 			...state,
 			text: `${state.text}${event.text}`
-		};
-		case "LLMReasoningDelta": return {
+		});
+		case "LLMReasoningDelta": return clearRetryInfo({
 			...state,
 			reasoning: `${state.reasoning}${event.text}`
-		};
-		case "ToolInputStart": return {
+		});
+		case "ToolInputStart": return clearRetryInfo({
 			...state,
 			toolRuns: replaceToolRun(state.toolRuns, {
 				_tag: "InputStreaming",
@@ -136,18 +144,18 @@ const applyAgentEventUnchecked = (state, event, nowMs) => {
 				name: event.name,
 				input: ""
 			})
-		};
-		case "ToolInputDelta": return {
+		});
+		case "ToolInputDelta": return clearRetryInfo({
 			...state,
 			toolRuns: appendToolInputDelta(state.toolRuns, event.id, event.delta)
-		};
-		case "ToolInputEnd": return {
+		});
+		case "ToolInputEnd": return clearRetryInfo({
 			...state,
 			toolRuns: replaceToolRun(state.toolRuns, {
 				_tag: "InputReady",
 				call: event.call
 			})
-		};
+		});
 		case "ToolApprovalRequested": return {
 			...state,
 			toolRuns: replaceToolRun(state.toolRuns, {
@@ -211,20 +219,20 @@ const applyAgentEventUnchecked = (state, event, nowMs) => {
 				endedAtMs: nowMs
 			})
 		};
-		case "ProviderToolResult": return {
+		case "ProviderToolResult": return clearRetryInfo({
 			...state,
 			toolRuns: replaceToolRun(state.toolRuns, {
 				_tag: "ProviderCompleted",
 				call: event.call,
 				result: event.result
 			})
-		};
-		case "AssistantMessage": return {
+		});
+		case "AssistantMessage": return clearRetryInfo({
 			...state,
 			liveMessages: [...state.liveMessages, event.message],
 			text: "",
 			reasoning: ""
-		};
+		});
 		case "AgentEnd": return {
 			...state,
 			status: "done",
@@ -232,7 +240,8 @@ const applyAgentEventUnchecked = (state, event, nowMs) => {
 			liveMessages: [],
 			text: "",
 			reasoning: "",
-			toolRuns: completedToolRuns(state.toolRuns)
+			toolRuns: completedToolRuns(state.toolRuns),
+			retryInfo: null
 		};
 		case "AgentAwaitingInput": return {
 			...state,
@@ -241,9 +250,14 @@ const applyAgentEventUnchecked = (state, event, nowMs) => {
 			liveMessages: [],
 			text: "",
 			reasoning: "",
-			error: null
+			error: null,
+			errorInfo: null,
+			retryInfo: null
+		};
+		case "AgentRetry": return {
+			...state,
+			retryInfo: event
 		};
-		case "AgentRetry":
 		case "CompactionEnd":
 		case "CompactionStart":
 		case "LLMStreamEnd":
@@ -264,19 +278,25 @@ const submitAgentUserMessage = (state, message) => ({
 	reasoning: "",
 	toolRuns: completedToolRuns(state.toolRuns),
 	error: null,
+	errorInfo: null,
+	retryInfo: null,
 	seenEventIds: []
 });
-const markAgentError = (state, message = "Agent request failed") => ({
+const markAgentError = (state, message = "Agent request failed", errorInfo = null) => ({
 	...state,
 	status: "error",
 	toolRuns: completedToolRuns(state.toolRuns),
-	error: message
+	error: message,
+	errorInfo,
+	retryInfo: null
 });
 const markAgentAborted = (state) => ({
 	...state,
 	status: "aborted",
 	toolRuns: completedToolRuns(state.toolRuns),
-	error: null
+	error: null,
+	errorInfo: null,
+	retryInfo: null
 });
 const reduceAgentEvents = (events, initialState = initialAgentClientState, options = {}) => events.reduce((state, event) => applyAgentEventWithOptions(state, event, options), initialState);
 //#endregion

package/dist/client/state.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"state.mjs","names":[],"sources":["../../src/client/state.ts"],"sourcesContent":["import {\n  type AgentEvent,\n  type AgentMessage,\n  type HitlRequest,\n  type QuestionRequest,\n  type QuestionResponse,\n  type ToolCall,\n  type ToolApprovalRequest,\n  type ToolResult,\n  type UserMessage\n} from '@yolk-sdk/agent/protocol'\n\nexport type AgentRunStatus = 'idle' | 'running' | 'waiting' | 'done' | 'error' | 'aborted'\n\nexport type AgentToolRun =\n  | {\n      readonly _tag: 'InputStreaming'\n      readonly id: string\n      readonly name?: string\n      readonly input: string\n    }\n  | { readonly _tag: 'InputReady'; readonly call: ToolCall }\n  | {\n      readonly _tag: 'ApprovalRequested'\n      readonly call: ToolCall\n      readonly request?: ToolApprovalRequest\n    }\n  | { readonly _tag: 'Denied'; readonly toolCallId: string; readonly reason: string }\n  | { readonly _tag: 'QuestionRequested'; readonly request: QuestionRequest }\n  | {\n      readonly _tag: 'QuestionAnswered'\n      readonly response: QuestionResponse\n      readonly request?: QuestionRequest\n    }\n  | {\n      readonly _tag: 'QuestionCancelled'\n      readonly response: QuestionResponse\n      readonly request?: QuestionRequest\n    }\n  | { readonly _tag: 'Executing'; readonly call: ToolCall; readonly startedAtMs: number }\n  | {\n      readonly _tag: 'Completed'\n      readonly call: ToolCall\n      readonly result: ToolResult\n      readonly startedAtMs: number\n      readonly endedAtMs: number\n    }\n  | {\n      readonly _tag: 'Errored'\n      readonly call: ToolCall\n      readonly message: string\n      readonly endedAtMs: number\n    }\n  | { readonly _tag: 'ProviderCompleted'; readonly call: ToolCall; readonly result: ToolResult }\n\ntype StartedAgentToolRun = Extract<AgentToolRun, { readonly _tag: 'Executing' | 'Completed' }>\n\nexport type AgentClientState = {\n  readonly status: AgentRunStatus\n  readonly messages: ReadonlyArray<AgentMessage>\n  readonly liveMessages: ReadonlyArray<AgentMessage>\n  readonly text: string\n  readonly reasoning: string\n  readonly toolRuns: ReadonlyArray<AgentToolRun>\n  readonly error: string | null\n  readonly seenEventIds: ReadonlyArray<string>\n}\n\nexport type ApplyAgentEventOptions = {\n  readonly nowMs?: number\n}\n\nexport type AgentTranscript = readonly [AgentMessage, ...Array<AgentMessage>]\n\nexport const initialAgentClientState: AgentClientState = {\n  status: 'idle',\n  messages: [],\n  liveMessages: [],\n  text: '',\n  reasoning: '',\n  toolRuns: [],\n  error: null,\n  seenEventIds: []\n}\n\nconst hasSeenEvent = (state: AgentClientState, event: AgentEvent) =>\n  event.eventId !== undefined && state.seenEventIds.includes(event.eventId)\n\nconst rememberEvent = (state: AgentClientState, event: AgentEvent): AgentClientState =>\n  event.eventId === undefined\n    ? state\n    : { ...state, seenEventIds: [...state.seenEventIds, event.eventId] }\n\nconst toolRunId = (run: AgentToolRun) => {\n  switch (run._tag) {\n    case 'InputStreaming':\n      return run.id\n    case 'Denied':\n      return run.toolCallId\n    case 'QuestionRequested':\n      return run.request.toolCallId\n    case 'QuestionAnswered':\n    case 'QuestionCancelled':\n      return run.response.toolCallId\n    case 'InputReady':\n    case 'ApprovalRequested':\n    case 'Executing':\n    case 'Completed':\n    case 'Errored':\n    case 'ProviderCompleted':\n      return run.call.id\n  }\n}\n\nexport const isActiveToolRun = (run: AgentToolRun) =>\n  run._tag !== 'Completed' &&\n  run._tag !== 'Errored' &&\n  run._tag !== 'Denied' &&\n  run._tag !== 'QuestionAnswered' &&\n  run._tag !== 'QuestionCancelled' &&\n  run._tag !== 'ProviderCompleted'\n\nexport const completedToolRuns = (runs: ReadonlyArray<AgentToolRun>) =>\n  runs.filter(run => run._tag === 'Completed')\n\nexport const toolRunsFromHitlRequests = (\n  requests: ReadonlyArray<HitlRequest>\n): ReadonlyArray<AgentToolRun> =>\n  requests.map(request => {\n    switch (request._tag) {\n      case 'QuestionRequest':\n        return { _tag: 'QuestionRequested', request }\n      case 'ToolApprovalRequest':\n        return { _tag: 'ApprovalRequested', call: request.call, request }\n    }\n  })\n\nconst replaceToolRun = (\n  runs: ReadonlyArray<AgentToolRun>,\n  run: AgentToolRun\n): ReadonlyArray<AgentToolRun> => {\n  const id = toolRunId(run)\n  const replaceIndex = runs.findIndex(current => toolRunId(current) === id)\n\n  if (replaceIndex === -1) {\n    return [...runs, run]\n  }\n\n  return runs.flatMap((current, index) => {\n    if (toolRunId(current) !== id) {\n      return [current]\n    }\n\n    return index === replaceIndex ? [run] : []\n  })\n}\n\nconst isStartedToolRun = (run: AgentToolRun): run is StartedAgentToolRun =>\n  run._tag === 'Executing' || run._tag === 'Completed'\n\nconst startedAtMsFor = (runs: ReadonlyArray<AgentToolRun>, toolCallId: string) =>\n  runs.filter(isStartedToolRun).find(run => run.call.id === toolCallId)?.startedAtMs\n\nconst appendToolInputDelta = (\n  runs: ReadonlyArray<AgentToolRun>,\n  id: string,\n  delta: string\n): ReadonlyArray<AgentToolRun> =>\n  runs.map(run =>\n    run._tag === 'InputStreaming' && run.id === id ? { ...run, input: `${run.input}${delta}` } : run\n  )\n\nconst questionRequestForToolCall = (\n  runs: ReadonlyArray<AgentToolRun>,\n  toolCallId: string\n): QuestionRequest | undefined =>\n  runs.flatMap(run => {\n    if (toolRunId(run) !== toolCallId) {\n      return []\n    }\n\n    switch (run._tag) {\n      case 'QuestionRequested':\n        return [run.request]\n      case 'QuestionAnswered':\n      case 'QuestionCancelled':\n        return run.request === undefined ? [] : [run.request]\n      case 'InputStreaming':\n      case 'InputReady':\n      case 'ApprovalRequested':\n      case 'Denied':\n      case 'Executing':\n      case 'Completed':\n      case 'Errored':\n      case 'ProviderCompleted':\n        return []\n    }\n  })[0]\n\nconst questionAnsweredRun = (\n  response: QuestionResponse,\n  request: QuestionRequest | undefined\n): AgentToolRun =>\n  request === undefined\n    ? { _tag: 'QuestionAnswered', response }\n    : { _tag: 'QuestionAnswered', response, request }\n\nconst questionCancelledRun = (\n  response: QuestionResponse,\n  request: QuestionRequest | undefined\n): AgentToolRun =>\n  request === undefined\n    ? { _tag: 'QuestionCancelled', response }\n    : { _tag: 'QuestionCancelled', response, request }\n\nexport const appendAgentMessage = (\n  messages: ReadonlyArray<AgentMessage>,\n  message: AgentMessage\n): AgentTranscript => {\n  const first = messages[0]\n\n  if (first === undefined) {\n    return [message]\n  }\n\n  return [first, ...messages.slice(1), message]\n}\n\nexport const applyAgentEvent = (state: AgentClientState, event: AgentEvent): AgentClientState => {\n  const nowMs = 0\n\n  return applyAgentEventWithOptions(state, event, { nowMs })\n}\n\nexport const applyAgentEventWithOptions = (\n  state: AgentClientState,\n  event: AgentEvent,\n  options: ApplyAgentEventOptions = {}\n): AgentClientState => {\n  const nowMs = options.nowMs ?? 0\n\n  if (hasSeenEvent(state, event)) {\n    return state\n  }\n\n  return rememberEvent(applyAgentEventUnchecked(state, event, nowMs), event)\n}\n\nconst applyAgentEventUnchecked = (\n  state: AgentClientState,\n  event: AgentEvent,\n  nowMs: number\n): AgentClientState => {\n  switch (event._tag) {\n    case 'AgentStart':\n      return {\n        ...state,\n        status: 'running',\n        text: '',\n        reasoning: '',\n        liveMessages: [],\n        toolRuns: completedToolRuns(state.toolRuns),\n        error: null\n      }\n    case 'AgentError':\n      return markAgentError(state, event.message)\n    case 'LLMTextDelta':\n      return { ...state, text: `${state.text}${event.text}` }\n    case 'LLMReasoningDelta':\n      return { ...state, reasoning: `${state.reasoning}${event.text}` }\n    case 'ToolInputStart':\n      return {\n        ...state,\n        toolRuns: replaceToolRun(state.toolRuns, {\n          _tag: 'InputStreaming',\n          id: event.id,\n          name: event.name,\n          input: ''\n        })\n      }\n    case 'ToolInputDelta':\n      return { ...state, toolRuns: appendToolInputDelta(state.toolRuns, event.id, event.delta) }\n    case 'ToolInputEnd':\n      return {\n        ...state,\n        toolRuns: replaceToolRun(state.toolRuns, { _tag: 'InputReady', call: event.call })\n      }\n    case 'ToolApprovalRequested':\n      return {\n        ...state,\n        toolRuns: replaceToolRun(state.toolRuns, {\n          _tag: 'ApprovalRequested',\n          call: event.call,\n          request: event.request\n        })\n      }\n    case 'ToolApprovalGranted':\n      return state\n    case 'ToolApprovalDenied':\n      return {\n        ...state,\n        toolRuns: replaceToolRun(state.toolRuns, {\n          _tag: 'Denied',\n          toolCallId: event.toolCallId,\n          reason: event.reason\n        })\n      }\n    case 'QuestionRequested':\n      return {\n        ...state,\n        toolRuns: replaceToolRun(state.toolRuns, {\n          _tag: 'QuestionRequested',\n          request: event.request\n        })\n      }\n    case 'QuestionAnswered':\n      return {\n        ...state,\n        toolRuns: replaceToolRun(\n          state.toolRuns,\n          questionAnsweredRun(\n            event.response,\n            questionRequestForToolCall(state.toolRuns, event.response.toolCallId)\n          )\n        )\n      }\n    case 'QuestionCancelled':\n      return {\n        ...state,\n        toolRuns: replaceToolRun(\n          state.toolRuns,\n          questionCancelledRun(\n            event.response,\n            questionRequestForToolCall(state.toolRuns, event.response.toolCallId)\n          )\n        )\n      }\n    case 'ToolExecutionStarted':\n      return {\n        ...state,\n        toolRuns: replaceToolRun(state.toolRuns, {\n          _tag: 'Executing',\n          call: event.call,\n          startedAtMs: nowMs\n        })\n      }\n    case 'ToolExecutionCompleted': {\n      const endedAtMs = nowMs\n      const startedAtMs = startedAtMsFor(state.toolRuns, event.call.id) ?? endedAtMs\n\n      return {\n        ...state,\n        toolRuns: replaceToolRun(state.toolRuns, {\n          _tag: 'Completed',\n          call: event.call,\n          result: event.result,\n          startedAtMs,\n          endedAtMs\n        })\n      }\n    }\n    case 'ToolExecutionError':\n      return {\n        ...state,\n        toolRuns: replaceToolRun(state.toolRuns, {\n          _tag: 'Errored',\n          call: event.call,\n          message: event.message,\n          endedAtMs: nowMs\n        })\n      }\n    case 'ProviderToolResult':\n      return {\n        ...state,\n        toolRuns: replaceToolRun(state.toolRuns, {\n          _tag: 'ProviderCompleted',\n          call: event.call,\n          result: event.result\n        })\n      }\n    case 'AssistantMessage':\n      return {\n        ...state,\n        liveMessages: [...state.liveMessages, event.message],\n        text: '',\n        reasoning: ''\n      }\n    case 'AgentEnd':\n      return {\n        ...state,\n        status: 'done',\n        messages: [...state.messages, ...event.messages],\n        liveMessages: [],\n        text: '',\n        reasoning: '',\n        toolRuns: completedToolRuns(state.toolRuns)\n      }\n    case 'AgentAwaitingInput':\n      return {\n        ...state,\n        status: 'waiting',\n        messages: [...state.messages, ...event.messages],\n        liveMessages: [],\n        text: '',\n        reasoning: '',\n        error: null\n      }\n    case 'AgentRetry':\n    case 'CompactionEnd':\n    case 'CompactionStart':\n    case 'LLMStreamEnd':\n    case 'LLMStreamStart':\n    case 'SubagentCompleted':\n    case 'SubagentStarted':\n    case 'TurnEnd':\n    case 'TurnStart':\n    case 'UsageUpdate':\n      return state\n  }\n}\n\nexport const submitAgentUserMessage = (\n  state: AgentClientState,\n  message: UserMessage\n): AgentClientState => ({\n  ...state,\n  status: 'running',\n  messages: appendAgentMessage(state.messages, message),\n  liveMessages: [],\n  text: '',\n  reasoning: '',\n  toolRuns: completedToolRuns(state.toolRuns),\n  error: null,\n  seenEventIds: []\n})\n\nexport const markAgentError = (\n  state: AgentClientState,\n  message = 'Agent request failed'\n): AgentClientState => ({\n  ...state,\n  status: 'error',\n  toolRuns: completedToolRuns(state.toolRuns),\n  error: message\n})\n\nexport const markAgentAborted = (state: AgentClientState): AgentClientState => ({\n  ...state,\n  status: 'aborted',\n  toolRuns: completedToolRuns(state.toolRuns),\n  error: null\n})\n\nexport const reduceAgentEvents = (\n  events: ReadonlyArray<AgentEvent>,\n  initialState: AgentClientState = initialAgentClientState,\n  options: ApplyAgentEventOptions = {}\n) =>\n  events.reduce((state, event) => applyAgentEventWithOptions(state, event, options), initialState)\n"],"mappings":";AA0EA,MAAa,0BAA4C;CACvD,QAAQ;CACR,UAAU,CAAC;CACX,cAAc,CAAC;CACf,MAAM;CACN,WAAW;CACX,UAAU,CAAC;CACX,OAAO;CACP,cAAc,CAAC;AACjB;AAEA,MAAM,gBAAgB,OAAyB,UAC7C,MAAM,YAAY,KAAA,KAAa,MAAM,aAAa,SAAS,MAAM,OAAO;AAE1E,MAAM,iBAAiB,OAAyB,UAC9C,MAAM,YAAY,KAAA,IACd,QACA;CAAE,GAAG;CAAO,cAAc,CAAC,GAAG,MAAM,cAAc,MAAM,OAAO;AAAE;AAEvE,MAAM,aAAa,QAAsB;CACvC,QAAQ,IAAI,MAAZ;EACE,KAAK,kBACH,OAAO,IAAI;EACb,KAAK,UACH,OAAO,IAAI;EACb,KAAK,qBACH,OAAO,IAAI,QAAQ;EACrB,KAAK;EACL,KAAK,qBACH,OAAO,IAAI,SAAS;EACtB,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK,qBACH,OAAO,IAAI,KAAK;CACpB;AACF;AAEA,MAAa,mBAAmB,QAC9B,IAAI,SAAS,eACb,IAAI,SAAS,aACb,IAAI,SAAS,YACb,IAAI,SAAS,sBACb,IAAI,SAAS,uBACb,IAAI,SAAS;AAEf,MAAa,qBAAqB,SAChC,KAAK,QAAO,QAAO,IAAI,SAAS,WAAW;AAE7C,MAAa,4BACX,aAEA,SAAS,KAAI,YAAW;CACtB,QAAQ,QAAQ,MAAhB;EACE,KAAK,mBACH,OAAO;GAAE,MAAM;GAAqB;EAAQ;EAC9C,KAAK,uBACH,OAAO;GAAE,MAAM;GAAqB,MAAM,QAAQ;GAAM;EAAQ;CACpE;AACF,CAAC;AAEH,MAAM,kBACJ,MACA,QACgC;CAChC,MAAM,KAAK,UAAU,GAAG;CACxB,MAAM,eAAe,KAAK,WAAU,YAAW,UAAU,OAAO,MAAM,EAAE;CAExE,IAAI,iBAAiB,IACnB,OAAO,CAAC,GAAG,MAAM,GAAG;CAGtB,OAAO,KAAK,SAAS,SAAS,UAAU;EACtC,IAAI,UAAU,OAAO,MAAM,IACzB,OAAO,CAAC,OAAO;EAGjB,OAAO,UAAU,eAAe,CAAC,GAAG,IAAI,CAAC;CAC3C,CAAC;AACH;AAEA,MAAM,oBAAoB,QACxB,IAAI,SAAS,eAAe,IAAI,SAAS;AAE3C,MAAM,kBAAkB,MAAmC,eACzD,KAAK,OAAO,gBAAgB,EAAE,MAAK,QAAO,IAAI,KAAK,OAAO,UAAU,GAAG;AAEzE,MAAM,wBACJ,MACA,IACA,UAEA,KAAK,KAAI,QACP,IAAI,SAAS,oBAAoB,IAAI,OAAO,KAAK;CAAE,GAAG;CAAK,OAAO,GAAG,IAAI,QAAQ;AAAQ,IAAI,GAC/F;AAEF,MAAM,8BACJ,MACA,eAEA,KAAK,SAAQ,QAAO;CAClB,IAAI,UAAU,GAAG,MAAM,YACrB,OAAO,CAAC;CAGV,QAAQ,IAAI,MAAZ;EACE,KAAK,qBACH,OAAO,CAAC,IAAI,OAAO;EACrB,KAAK;EACL,KAAK,qBACH,OAAO,IAAI,YAAY,KAAA,IAAY,CAAC,IAAI,CAAC,IAAI,OAAO;EACtD,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK,qBACH,OAAO,CAAC;CACZ;AACF,CAAC,EAAE;AAEL,MAAM,uBACJ,UACA,YAEA,YAAY,KAAA,IACR;CAAE,MAAM;CAAoB;AAAS,IACrC;CAAE,MAAM;CAAoB;CAAU;AAAQ;AAEpD,MAAM,wBACJ,UACA,YAEA,YAAY,KAAA,IACR;CAAE,MAAM;CAAqB;AAAS,IACtC;CAAE,MAAM;CAAqB;CAAU;AAAQ;AAErD,MAAa,sBACX,UACA,YACoB;CACpB,MAAM,QAAQ,SAAS;CAEvB,IAAI,UAAU,KAAA,GACZ,OAAO,CAAC,OAAO;CAGjB,OAAO;EAAC;EAAO,GAAG,SAAS,MAAM,CAAC;EAAG;CAAO;AAC9C;AAEA,MAAa,mBAAmB,OAAyB,UAAwC;CAG/F,OAAO,2BAA2B,OAAO,OAAO,EAAE,OAAA,EAAM,CAAC;AAC3D;AAEA,MAAa,8BACX,OACA,OACA,UAAkC,CAAC,MACd;CACrB,MAAM,QAAQ,QAAQ,SAAS;CAE/B,IAAI,aAAa,OAAO,KAAK,GAC3B,OAAO;CAGT,OAAO,cAAc,yBAAyB,OAAO,OAAO,KAAK,GAAG,KAAK;AAC3E;AAEA,MAAM,4BACJ,OACA,OACA,UACqB;CACrB,QAAQ,MAAM,MAAd;EACE,KAAK,cACH,OAAO;GACL,GAAG;GACH,QAAQ;GACR,MAAM;GACN,WAAW;GACX,cAAc,CAAC;GACf,UAAU,kBAAkB,MAAM,QAAQ;GAC1C,OAAO;EACT;EACF,KAAK,cACH,OAAO,eAAe,OAAO,MAAM,OAAO;EAC5C,KAAK,gBACH,OAAO;GAAE,GAAG;GAAO,MAAM,GAAG,MAAM,OAAO,MAAM;EAAO;EACxD,KAAK,qBACH,OAAO;GAAE,GAAG;GAAO,WAAW,GAAG,MAAM,YAAY,MAAM;EAAO;EAClE,KAAK,kBACH,OAAO;GACL,GAAG;GACH,UAAU,eAAe,MAAM,UAAU;IACvC,MAAM;IACN,IAAI,MAAM;IACV,MAAM,MAAM;IACZ,OAAO;GACT,CAAC;EACH;EACF,KAAK,kBACH,OAAO;GAAE,GAAG;GAAO,UAAU,qBAAqB,MAAM,UAAU,MAAM,IAAI,MAAM,KAAK;EAAE;EAC3F,KAAK,gBACH,OAAO;GACL,GAAG;GACH,UAAU,eAAe,MAAM,UAAU;IAAE,MAAM;IAAc,MAAM,MAAM;GAAK,CAAC;EACnF;EACF,KAAK,yBACH,OAAO;GACL,GAAG;GACH,UAAU,eAAe,MAAM,UAAU;IACvC,MAAM;IACN,MAAM,MAAM;IACZ,SAAS,MAAM;GACjB,CAAC;EACH;EACF,KAAK,uBACH,OAAO;EACT,KAAK,sBACH,OAAO;GACL,GAAG;GACH,UAAU,eAAe,MAAM,UAAU;IACvC,MAAM;IACN,YAAY,MAAM;IAClB,QAAQ,MAAM;GAChB,CAAC;EACH;EACF,KAAK,qBACH,OAAO;GACL,GAAG;GACH,UAAU,eAAe,MAAM,UAAU;IACvC,MAAM;IACN,SAAS,MAAM;GACjB,CAAC;EACH;EACF,KAAK,oBACH,OAAO;GACL,GAAG;GACH,UAAU,eACR,MAAM,UACN,oBACE,MAAM,UACN,2BAA2B,MAAM,UAAU,MAAM,SAAS,UAAU,CACtE,CACF;EACF;EACF,KAAK,qBACH,OAAO;GACL,GAAG;GACH,UAAU,eACR,MAAM,UACN,qBACE,MAAM,UACN,2BAA2B,MAAM,UAAU,MAAM,SAAS,UAAU,CACtE,CACF;EACF;EACF,KAAK,wBACH,OAAO;GACL,GAAG;GACH,UAAU,eAAe,MAAM,UAAU;IACvC,MAAM;IACN,MAAM,MAAM;IACZ,aAAa;GACf,CAAC;EACH;EACF,KAAK,0BAA0B;GAC7B,MAAM,YAAY;GAClB,MAAM,cAAc,eAAe,MAAM,UAAU,MAAM,KAAK,EAAE,KAAK;GAErE,OAAO;IACL,GAAG;IACH,UAAU,eAAe,MAAM,UAAU;KACvC,MAAM;KACN,MAAM,MAAM;KACZ,QAAQ,MAAM;KACd;KACA;IACF,CAAC;GACH;EACF;EACA,KAAK,sBACH,OAAO;GACL,GAAG;GACH,UAAU,eAAe,MAAM,UAAU;IACvC,MAAM;IACN,MAAM,MAAM;IACZ,SAAS,MAAM;IACf,WAAW;GACb,CAAC;EACH;EACF,KAAK,sBACH,OAAO;GACL,GAAG;GACH,UAAU,eAAe,MAAM,UAAU;IACvC,MAAM;IACN,MAAM,MAAM;IACZ,QAAQ,MAAM;GAChB,CAAC;EACH;EACF,KAAK,oBACH,OAAO;GACL,GAAG;GACH,cAAc,CAAC,GAAG,MAAM,cAAc,MAAM,OAAO;GACnD,MAAM;GACN,WAAW;EACb;EACF,KAAK,YACH,OAAO;GACL,GAAG;GACH,QAAQ;GACR,UAAU,CAAC,GAAG,MAAM,UAAU,GAAG,MAAM,QAAQ;GAC/C,cAAc,CAAC;GACf,MAAM;GACN,WAAW;GACX,UAAU,kBAAkB,MAAM,QAAQ;EAC5C;EACF,KAAK,sBACH,OAAO;GACL,GAAG;GACH,QAAQ;GACR,UAAU,CAAC,GAAG,MAAM,UAAU,GAAG,MAAM,QAAQ;GAC/C,cAAc,CAAC;GACf,MAAM;GACN,WAAW;GACX,OAAO;EACT;EACF,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK,eACH,OAAO;CACX;AACF;AAEA,MAAa,0BACX,OACA,aACsB;CACtB,GAAG;CACH,QAAQ;CACR,UAAU,mBAAmB,MAAM,UAAU,OAAO;CACpD,cAAc,CAAC;CACf,MAAM;CACN,WAAW;CACX,UAAU,kBAAkB,MAAM,QAAQ;CAC1C,OAAO;CACP,cAAc,CAAC;AACjB;AAEA,MAAa,kBACX,OACA,UAAU,4BACY;CACtB,GAAG;CACH,QAAQ;CACR,UAAU,kBAAkB,MAAM,QAAQ;CAC1C,OAAO;AACT;AAEA,MAAa,oBAAoB,WAA+C;CAC9E,GAAG;CACH,QAAQ;CACR,UAAU,kBAAkB,MAAM,QAAQ;CAC1C,OAAO;AACT;AAEA,MAAa,qBACX,QACA,eAAiC,yBACjC,UAAkC,CAAC,MAEnC,OAAO,QAAQ,OAAO,UAAU,2BAA2B,OAAO,OAAO,OAAO,GAAG,YAAY"}
+{"version":3,"file":"state.mjs","names":[],"sources":["../../src/client/state.ts"],"sourcesContent":["import {\n  type AgentError,\n  type AgentEvent,\n  type AgentMessage,\n  type AgentRetry,\n  type HitlRequest,\n  type QuestionRequest,\n  type QuestionResponse,\n  type ToolCall,\n  type ToolApprovalRequest,\n  type ToolResult,\n  type UserMessage\n} from '@yolk-sdk/agent/protocol'\n\nexport type AgentRunStatus = 'idle' | 'running' | 'waiting' | 'done' | 'error' | 'aborted'\n\nexport type AgentToolRun =\n  | {\n      readonly _tag: 'InputStreaming'\n      readonly id: string\n      readonly name?: string\n      readonly input: string\n    }\n  | { readonly _tag: 'InputReady'; readonly call: ToolCall }\n  | {\n      readonly _tag: 'ApprovalRequested'\n      readonly call: ToolCall\n      readonly request?: ToolApprovalRequest\n    }\n  | { readonly _tag: 'Denied'; readonly toolCallId: string; readonly reason: string }\n  | { readonly _tag: 'QuestionRequested'; readonly request: QuestionRequest }\n  | {\n      readonly _tag: 'QuestionAnswered'\n      readonly response: QuestionResponse\n      readonly request?: QuestionRequest\n    }\n  | {\n      readonly _tag: 'QuestionCancelled'\n      readonly response: QuestionResponse\n      readonly request?: QuestionRequest\n    }\n  | { readonly _tag: 'Executing'; readonly call: ToolCall; readonly startedAtMs: number }\n  | {\n      readonly _tag: 'Completed'\n      readonly call: ToolCall\n      readonly result: ToolResult\n      readonly startedAtMs: number\n      readonly endedAtMs: number\n    }\n  | {\n      readonly _tag: 'Errored'\n      readonly call: ToolCall\n      readonly message: string\n      readonly endedAtMs: number\n    }\n  | { readonly _tag: 'ProviderCompleted'; readonly call: ToolCall; readonly result: ToolResult }\n\ntype StartedAgentToolRun = Extract<AgentToolRun, { readonly _tag: 'Executing' | 'Completed' }>\n\nexport type AgentClientState = {\n  readonly status: AgentRunStatus\n  readonly messages: ReadonlyArray<AgentMessage>\n  readonly liveMessages: ReadonlyArray<AgentMessage>\n  readonly text: string\n  readonly reasoning: string\n  readonly toolRuns: ReadonlyArray<AgentToolRun>\n  readonly error: string | null\n  readonly errorInfo: AgentError | null\n  readonly retryInfo: AgentRetry | null\n  readonly seenEventIds: ReadonlyArray<string>\n}\n\nexport type ApplyAgentEventOptions = {\n  readonly nowMs?: number\n}\n\nexport type AgentTranscript = readonly [AgentMessage, ...Array<AgentMessage>]\n\nexport const initialAgentClientState: AgentClientState = {\n  status: 'idle',\n  messages: [],\n  liveMessages: [],\n  text: '',\n  reasoning: '',\n  toolRuns: [],\n  error: null,\n  errorInfo: null,\n  retryInfo: null,\n  seenEventIds: []\n}\n\nconst clearRetryInfo = (state: AgentClientState): AgentClientState =>\n  state.retryInfo === null ? state : { ...state, retryInfo: null }\n\nconst hasSeenEvent = (state: AgentClientState, event: AgentEvent) =>\n  event.eventId !== undefined && state.seenEventIds.includes(event.eventId)\n\nconst rememberEvent = (state: AgentClientState, event: AgentEvent): AgentClientState =>\n  event.eventId === undefined\n    ? state\n    : { ...state, seenEventIds: [...state.seenEventIds, event.eventId] }\n\nconst toolRunId = (run: AgentToolRun) => {\n  switch (run._tag) {\n    case 'InputStreaming':\n      return run.id\n    case 'Denied':\n      return run.toolCallId\n    case 'QuestionRequested':\n      return run.request.toolCallId\n    case 'QuestionAnswered':\n    case 'QuestionCancelled':\n      return run.response.toolCallId\n    case 'InputReady':\n    case 'ApprovalRequested':\n    case 'Executing':\n    case 'Completed':\n    case 'Errored':\n    case 'ProviderCompleted':\n      return run.call.id\n  }\n}\n\nexport const isActiveToolRun = (run: AgentToolRun) =>\n  run._tag !== 'Completed' &&\n  run._tag !== 'Errored' &&\n  run._tag !== 'Denied' &&\n  run._tag !== 'QuestionAnswered' &&\n  run._tag !== 'QuestionCancelled' &&\n  run._tag !== 'ProviderCompleted'\n\nexport const completedToolRuns = (runs: ReadonlyArray<AgentToolRun>) =>\n  runs.filter(run => run._tag === 'Completed')\n\nexport const toolRunsFromHitlRequests = (\n  requests: ReadonlyArray<HitlRequest>\n): ReadonlyArray<AgentToolRun> =>\n  requests.map(request => {\n    switch (request._tag) {\n      case 'QuestionRequest':\n        return { _tag: 'QuestionRequested', request }\n      case 'ToolApprovalRequest':\n        return { _tag: 'ApprovalRequested', call: request.call, request }\n    }\n  })\n\nconst replaceToolRun = (\n  runs: ReadonlyArray<AgentToolRun>,\n  run: AgentToolRun\n): ReadonlyArray<AgentToolRun> => {\n  const id = toolRunId(run)\n  const replaceIndex = runs.findIndex(current => toolRunId(current) === id)\n\n  if (replaceIndex === -1) {\n    return [...runs, run]\n  }\n\n  return runs.flatMap((current, index) => {\n    if (toolRunId(current) !== id) {\n      return [current]\n    }\n\n    return index === replaceIndex ? [run] : []\n  })\n}\n\nconst isStartedToolRun = (run: AgentToolRun): run is StartedAgentToolRun =>\n  run._tag === 'Executing' || run._tag === 'Completed'\n\nconst startedAtMsFor = (runs: ReadonlyArray<AgentToolRun>, toolCallId: string) =>\n  runs.filter(isStartedToolRun).find(run => run.call.id === toolCallId)?.startedAtMs\n\nconst appendToolInputDelta = (\n  runs: ReadonlyArray<AgentToolRun>,\n  id: string,\n  delta: string\n): ReadonlyArray<AgentToolRun> =>\n  runs.map(run =>\n    run._tag === 'InputStreaming' && run.id === id ? { ...run, input: `${run.input}${delta}` } : run\n  )\n\nconst questionRequestForToolCall = (\n  runs: ReadonlyArray<AgentToolRun>,\n  toolCallId: string\n): QuestionRequest | undefined =>\n  runs.flatMap(run => {\n    if (toolRunId(run) !== toolCallId) {\n      return []\n    }\n\n    switch (run._tag) {\n      case 'QuestionRequested':\n        return [run.request]\n      case 'QuestionAnswered':\n      case 'QuestionCancelled':\n        return run.request === undefined ? [] : [run.request]\n      case 'InputStreaming':\n      case 'InputReady':\n      case 'ApprovalRequested':\n      case 'Denied':\n      case 'Executing':\n      case 'Completed':\n      case 'Errored':\n      case 'ProviderCompleted':\n        return []\n    }\n  })[0]\n\nconst questionAnsweredRun = (\n  response: QuestionResponse,\n  request: QuestionRequest | undefined\n): AgentToolRun =>\n  request === undefined\n    ? { _tag: 'QuestionAnswered', response }\n    : { _tag: 'QuestionAnswered', response, request }\n\nconst questionCancelledRun = (\n  response: QuestionResponse,\n  request: QuestionRequest | undefined\n): AgentToolRun =>\n  request === undefined\n    ? { _tag: 'QuestionCancelled', response }\n    : { _tag: 'QuestionCancelled', response, request }\n\nexport const appendAgentMessage = (\n  messages: ReadonlyArray<AgentMessage>,\n  message: AgentMessage\n): AgentTranscript => {\n  const first = messages[0]\n\n  if (first === undefined) {\n    return [message]\n  }\n\n  return [first, ...messages.slice(1), message]\n}\n\nexport const applyAgentEvent = (state: AgentClientState, event: AgentEvent): AgentClientState => {\n  const nowMs = 0\n\n  return applyAgentEventWithOptions(state, event, { nowMs })\n}\n\nexport const applyAgentEventWithOptions = (\n  state: AgentClientState,\n  event: AgentEvent,\n  options: ApplyAgentEventOptions = {}\n): AgentClientState => {\n  const nowMs = options.nowMs ?? 0\n\n  if (hasSeenEvent(state, event)) {\n    return state\n  }\n\n  return rememberEvent(applyAgentEventUnchecked(state, event, nowMs), event)\n}\n\nconst applyAgentEventUnchecked = (\n  state: AgentClientState,\n  event: AgentEvent,\n  nowMs: number\n): AgentClientState => {\n  switch (event._tag) {\n    case 'AgentStart':\n      return {\n        ...state,\n        status: 'running',\n        text: '',\n        reasoning: '',\n        liveMessages: [],\n        toolRuns: completedToolRuns(state.toolRuns),\n        error: null,\n        errorInfo: null,\n        retryInfo: null\n      }\n    case 'AgentError':\n      return markAgentError(state, event.message, event)\n    case 'LLMTextDelta':\n      return clearRetryInfo({ ...state, text: `${state.text}${event.text}` })\n    case 'LLMReasoningDelta':\n      return clearRetryInfo({ ...state, reasoning: `${state.reasoning}${event.text}` })\n    case 'ToolInputStart':\n      return clearRetryInfo({\n        ...state,\n        toolRuns: replaceToolRun(state.toolRuns, {\n          _tag: 'InputStreaming',\n          id: event.id,\n          name: event.name,\n          input: ''\n        })\n      })\n    case 'ToolInputDelta':\n      return clearRetryInfo({\n        ...state,\n        toolRuns: appendToolInputDelta(state.toolRuns, event.id, event.delta)\n      })\n    case 'ToolInputEnd':\n      return clearRetryInfo({\n        ...state,\n        toolRuns: replaceToolRun(state.toolRuns, { _tag: 'InputReady', call: event.call })\n      })\n    case 'ToolApprovalRequested':\n      return {\n        ...state,\n        toolRuns: replaceToolRun(state.toolRuns, {\n          _tag: 'ApprovalRequested',\n          call: event.call,\n          request: event.request\n        })\n      }\n    case 'ToolApprovalGranted':\n      return state\n    case 'ToolApprovalDenied':\n      return {\n        ...state,\n        toolRuns: replaceToolRun(state.toolRuns, {\n          _tag: 'Denied',\n          toolCallId: event.toolCallId,\n          reason: event.reason\n        })\n      }\n    case 'QuestionRequested':\n      return {\n        ...state,\n        toolRuns: replaceToolRun(state.toolRuns, {\n          _tag: 'QuestionRequested',\n          request: event.request\n        })\n      }\n    case 'QuestionAnswered':\n      return {\n        ...state,\n        toolRuns: replaceToolRun(\n          state.toolRuns,\n          questionAnsweredRun(\n            event.response,\n            questionRequestForToolCall(state.toolRuns, event.response.toolCallId)\n          )\n        )\n      }\n    case 'QuestionCancelled':\n      return {\n        ...state,\n        toolRuns: replaceToolRun(\n          state.toolRuns,\n          questionCancelledRun(\n            event.response,\n            questionRequestForToolCall(state.toolRuns, event.response.toolCallId)\n          )\n        )\n      }\n    case 'ToolExecutionStarted':\n      return {\n        ...state,\n        toolRuns: replaceToolRun(state.toolRuns, {\n          _tag: 'Executing',\n          call: event.call,\n          startedAtMs: nowMs\n        })\n      }\n    case 'ToolExecutionCompleted': {\n      const endedAtMs = nowMs\n      const startedAtMs = startedAtMsFor(state.toolRuns, event.call.id) ?? endedAtMs\n\n      return {\n        ...state,\n        toolRuns: replaceToolRun(state.toolRuns, {\n          _tag: 'Completed',\n          call: event.call,\n          result: event.result,\n          startedAtMs,\n          endedAtMs\n        })\n      }\n    }\n    case 'ToolExecutionError':\n      return {\n        ...state,\n        toolRuns: replaceToolRun(state.toolRuns, {\n          _tag: 'Errored',\n          call: event.call,\n          message: event.message,\n          endedAtMs: nowMs\n        })\n      }\n    case 'ProviderToolResult':\n      return clearRetryInfo({\n        ...state,\n        toolRuns: replaceToolRun(state.toolRuns, {\n          _tag: 'ProviderCompleted',\n          call: event.call,\n          result: event.result\n        })\n      })\n    case 'AssistantMessage':\n      return clearRetryInfo({\n        ...state,\n        liveMessages: [...state.liveMessages, event.message],\n        text: '',\n        reasoning: ''\n      })\n    case 'AgentEnd':\n      return {\n        ...state,\n        status: 'done',\n        messages: [...state.messages, ...event.messages],\n        liveMessages: [],\n        text: '',\n        reasoning: '',\n        toolRuns: completedToolRuns(state.toolRuns),\n        retryInfo: null\n      }\n    case 'AgentAwaitingInput':\n      return {\n        ...state,\n        status: 'waiting',\n        messages: [...state.messages, ...event.messages],\n        liveMessages: [],\n        text: '',\n        reasoning: '',\n        error: null,\n        errorInfo: null,\n        retryInfo: null\n      }\n    case 'AgentRetry':\n      return { ...state, retryInfo: event }\n    case 'CompactionEnd':\n    case 'CompactionStart':\n    case 'LLMStreamEnd':\n    case 'LLMStreamStart':\n    case 'SubagentCompleted':\n    case 'SubagentStarted':\n    case 'TurnEnd':\n    case 'TurnStart':\n    case 'UsageUpdate':\n      return state\n  }\n}\n\nexport const submitAgentUserMessage = (\n  state: AgentClientState,\n  message: UserMessage\n): AgentClientState => ({\n  ...state,\n  status: 'running',\n  messages: appendAgentMessage(state.messages, message),\n  liveMessages: [],\n  text: '',\n  reasoning: '',\n  toolRuns: completedToolRuns(state.toolRuns),\n  error: null,\n  errorInfo: null,\n  retryInfo: null,\n  seenEventIds: []\n})\n\nexport const markAgentError = (\n  state: AgentClientState,\n  message = 'Agent request failed',\n  errorInfo: AgentError | null = null\n): AgentClientState => ({\n  ...state,\n  status: 'error',\n  toolRuns: completedToolRuns(state.toolRuns),\n  error: message,\n  errorInfo,\n  retryInfo: null\n})\n\nexport const markAgentAborted = (state: AgentClientState): AgentClientState => ({\n  ...state,\n  status: 'aborted',\n  toolRuns: completedToolRuns(state.toolRuns),\n  error: null,\n  errorInfo: null,\n  retryInfo: null\n})\n\nexport const reduceAgentEvents = (\n  events: ReadonlyArray<AgentEvent>,\n  initialState: AgentClientState = initialAgentClientState,\n  options: ApplyAgentEventOptions = {}\n) =>\n  events.reduce((state, event) => applyAgentEventWithOptions(state, event, options), initialState)\n"],"mappings":";AA8EA,MAAa,0BAA4C;CACvD,QAAQ;CACR,UAAU,CAAC;CACX,cAAc,CAAC;CACf,MAAM;CACN,WAAW;CACX,UAAU,CAAC;CACX,OAAO;CACP,WAAW;CACX,WAAW;CACX,cAAc,CAAC;AACjB;AAEA,MAAM,kBAAkB,UACtB,MAAM,cAAc,OAAO,QAAQ;CAAE,GAAG;CAAO,WAAW;AAAK;AAEjE,MAAM,gBAAgB,OAAyB,UAC7C,MAAM,YAAY,KAAA,KAAa,MAAM,aAAa,SAAS,MAAM,OAAO;AAE1E,MAAM,iBAAiB,OAAyB,UAC9C,MAAM,YAAY,KAAA,IACd,QACA;CAAE,GAAG;CAAO,cAAc,CAAC,GAAG,MAAM,cAAc,MAAM,OAAO;AAAE;AAEvE,MAAM,aAAa,QAAsB;CACvC,QAAQ,IAAI,MAAZ;EACE,KAAK,kBACH,OAAO,IAAI;EACb,KAAK,UACH,OAAO,IAAI;EACb,KAAK,qBACH,OAAO,IAAI,QAAQ;EACrB,KAAK;EACL,KAAK,qBACH,OAAO,IAAI,SAAS;EACtB,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK,qBACH,OAAO,IAAI,KAAK;CACpB;AACF;AAEA,MAAa,mBAAmB,QAC9B,IAAI,SAAS,eACb,IAAI,SAAS,aACb,IAAI,SAAS,YACb,IAAI,SAAS,sBACb,IAAI,SAAS,uBACb,IAAI,SAAS;AAEf,MAAa,qBAAqB,SAChC,KAAK,QAAO,QAAO,IAAI,SAAS,WAAW;AAE7C,MAAa,4BACX,aAEA,SAAS,KAAI,YAAW;CACtB,QAAQ,QAAQ,MAAhB;EACE,KAAK,mBACH,OAAO;GAAE,MAAM;GAAqB;EAAQ;EAC9C,KAAK,uBACH,OAAO;GAAE,MAAM;GAAqB,MAAM,QAAQ;GAAM;EAAQ;CACpE;AACF,CAAC;AAEH,MAAM,kBACJ,MACA,QACgC;CAChC,MAAM,KAAK,UAAU,GAAG;CACxB,MAAM,eAAe,KAAK,WAAU,YAAW,UAAU,OAAO,MAAM,EAAE;CAExE,IAAI,iBAAiB,IACnB,OAAO,CAAC,GAAG,MAAM,GAAG;CAGtB,OAAO,KAAK,SAAS,SAAS,UAAU;EACtC,IAAI,UAAU,OAAO,MAAM,IACzB,OAAO,CAAC,OAAO;EAGjB,OAAO,UAAU,eAAe,CAAC,GAAG,IAAI,CAAC;CAC3C,CAAC;AACH;AAEA,MAAM,oBAAoB,QACxB,IAAI,SAAS,eAAe,IAAI,SAAS;AAE3C,MAAM,kBAAkB,MAAmC,eACzD,KAAK,OAAO,gBAAgB,EAAE,MAAK,QAAO,IAAI,KAAK,OAAO,UAAU,GAAG;AAEzE,MAAM,wBACJ,MACA,IACA,UAEA,KAAK,KAAI,QACP,IAAI,SAAS,oBAAoB,IAAI,OAAO,KAAK;CAAE,GAAG;CAAK,OAAO,GAAG,IAAI,QAAQ;AAAQ,IAAI,GAC/F;AAEF,MAAM,8BACJ,MACA,eAEA,KAAK,SAAQ,QAAO;CAClB,IAAI,UAAU,GAAG,MAAM,YACrB,OAAO,CAAC;CAGV,QAAQ,IAAI,MAAZ;EACE,KAAK,qBACH,OAAO,CAAC,IAAI,OAAO;EACrB,KAAK;EACL,KAAK,qBACH,OAAO,IAAI,YAAY,KAAA,IAAY,CAAC,IAAI,CAAC,IAAI,OAAO;EACtD,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK,qBACH,OAAO,CAAC;CACZ;AACF,CAAC,EAAE;AAEL,MAAM,uBACJ,UACA,YAEA,YAAY,KAAA,IACR;CAAE,MAAM;CAAoB;AAAS,IACrC;CAAE,MAAM;CAAoB;CAAU;AAAQ;AAEpD,MAAM,wBACJ,UACA,YAEA,YAAY,KAAA,IACR;CAAE,MAAM;CAAqB;AAAS,IACtC;CAAE,MAAM;CAAqB;CAAU;AAAQ;AAErD,MAAa,sBACX,UACA,YACoB;CACpB,MAAM,QAAQ,SAAS;CAEvB,IAAI,UAAU,KAAA,GACZ,OAAO,CAAC,OAAO;CAGjB,OAAO;EAAC;EAAO,GAAG,SAAS,MAAM,CAAC;EAAG;CAAO;AAC9C;AAEA,MAAa,mBAAmB,OAAyB,UAAwC;CAG/F,OAAO,2BAA2B,OAAO,OAAO,EAAE,OAAA,EAAM,CAAC;AAC3D;AAEA,MAAa,8BACX,OACA,OACA,UAAkC,CAAC,MACd;CACrB,MAAM,QAAQ,QAAQ,SAAS;CAE/B,IAAI,aAAa,OAAO,KAAK,GAC3B,OAAO;CAGT,OAAO,cAAc,yBAAyB,OAAO,OAAO,KAAK,GAAG,KAAK;AAC3E;AAEA,MAAM,4BACJ,OACA,OACA,UACqB;CACrB,QAAQ,MAAM,MAAd;EACE,KAAK,cACH,OAAO;GACL,GAAG;GACH,QAAQ;GACR,MAAM;GACN,WAAW;GACX,cAAc,CAAC;GACf,UAAU,kBAAkB,MAAM,QAAQ;GAC1C,OAAO;GACP,WAAW;GACX,WAAW;EACb;EACF,KAAK,cACH,OAAO,eAAe,OAAO,MAAM,SAAS,KAAK;EACnD,KAAK,gBACH,OAAO,eAAe;GAAE,GAAG;GAAO,MAAM,GAAG,MAAM,OAAO,MAAM;EAAO,CAAC;EACxE,KAAK,qBACH,OAAO,eAAe;GAAE,GAAG;GAAO,WAAW,GAAG,MAAM,YAAY,MAAM;EAAO,CAAC;EAClF,KAAK,kBACH,OAAO,eAAe;GACpB,GAAG;GACH,UAAU,eAAe,MAAM,UAAU;IACvC,MAAM;IACN,IAAI,MAAM;IACV,MAAM,MAAM;IACZ,OAAO;GACT,CAAC;EACH,CAAC;EACH,KAAK,kBACH,OAAO,eAAe;GACpB,GAAG;GACH,UAAU,qBAAqB,MAAM,UAAU,MAAM,IAAI,MAAM,KAAK;EACtE,CAAC;EACH,KAAK,gBACH,OAAO,eAAe;GACpB,GAAG;GACH,UAAU,eAAe,MAAM,UAAU;IAAE,MAAM;IAAc,MAAM,MAAM;GAAK,CAAC;EACnF,CAAC;EACH,KAAK,yBACH,OAAO;GACL,GAAG;GACH,UAAU,eAAe,MAAM,UAAU;IACvC,MAAM;IACN,MAAM,MAAM;IACZ,SAAS,MAAM;GACjB,CAAC;EACH;EACF,KAAK,uBACH,OAAO;EACT,KAAK,sBACH,OAAO;GACL,GAAG;GACH,UAAU,eAAe,MAAM,UAAU;IACvC,MAAM;IACN,YAAY,MAAM;IAClB,QAAQ,MAAM;GAChB,CAAC;EACH;EACF,KAAK,qBACH,OAAO;GACL,GAAG;GACH,UAAU,eAAe,MAAM,UAAU;IACvC,MAAM;IACN,SAAS,MAAM;GACjB,CAAC;EACH;EACF,KAAK,oBACH,OAAO;GACL,GAAG;GACH,UAAU,eACR,MAAM,UACN,oBACE,MAAM,UACN,2BAA2B,MAAM,UAAU,MAAM,SAAS,UAAU,CACtE,CACF;EACF;EACF,KAAK,qBACH,OAAO;GACL,GAAG;GACH,UAAU,eACR,MAAM,UACN,qBACE,MAAM,UACN,2BAA2B,MAAM,UAAU,MAAM,SAAS,UAAU,CACtE,CACF;EACF;EACF,KAAK,wBACH,OAAO;GACL,GAAG;GACH,UAAU,eAAe,MAAM,UAAU;IACvC,MAAM;IACN,MAAM,MAAM;IACZ,aAAa;GACf,CAAC;EACH;EACF,KAAK,0BAA0B;GAC7B,MAAM,YAAY;GAClB,MAAM,cAAc,eAAe,MAAM,UAAU,MAAM,KAAK,EAAE,KAAK;GAErE,OAAO;IACL,GAAG;IACH,UAAU,eAAe,MAAM,UAAU;KACvC,MAAM;KACN,MAAM,MAAM;KACZ,QAAQ,MAAM;KACd;KACA;IACF,CAAC;GACH;EACF;EACA,KAAK,sBACH,OAAO;GACL,GAAG;GACH,UAAU,eAAe,MAAM,UAAU;IACvC,MAAM;IACN,MAAM,MAAM;IACZ,SAAS,MAAM;IACf,WAAW;GACb,CAAC;EACH;EACF,KAAK,sBACH,OAAO,eAAe;GACpB,GAAG;GACH,UAAU,eAAe,MAAM,UAAU;IACvC,MAAM;IACN,MAAM,MAAM;IACZ,QAAQ,MAAM;GAChB,CAAC;EACH,CAAC;EACH,KAAK,oBACH,OAAO,eAAe;GACpB,GAAG;GACH,cAAc,CAAC,GAAG,MAAM,cAAc,MAAM,OAAO;GACnD,MAAM;GACN,WAAW;EACb,CAAC;EACH,KAAK,YACH,OAAO;GACL,GAAG;GACH,QAAQ;GACR,UAAU,CAAC,GAAG,MAAM,UAAU,GAAG,MAAM,QAAQ;GAC/C,cAAc,CAAC;GACf,MAAM;GACN,WAAW;GACX,UAAU,kBAAkB,MAAM,QAAQ;GAC1C,WAAW;EACb;EACF,KAAK,sBACH,OAAO;GACL,GAAG;GACH,QAAQ;GACR,UAAU,CAAC,GAAG,MAAM,UAAU,GAAG,MAAM,QAAQ;GAC/C,cAAc,CAAC;GACf,MAAM;GACN,WAAW;GACX,OAAO;GACP,WAAW;GACX,WAAW;EACb;EACF,KAAK,cACH,OAAO;GAAE,GAAG;GAAO,WAAW;EAAM;EACtC,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK,eACH,OAAO;CACX;AACF;AAEA,MAAa,0BACX,OACA,aACsB;CACtB,GAAG;CACH,QAAQ;CACR,UAAU,mBAAmB,MAAM,UAAU,OAAO;CACpD,cAAc,CAAC;CACf,MAAM;CACN,WAAW;CACX,UAAU,kBAAkB,MAAM,QAAQ;CAC1C,OAAO;CACP,WAAW;CACX,WAAW;CACX,cAAc,CAAC;AACjB;AAEA,MAAa,kBACX,OACA,UAAU,wBACV,YAA+B,UACT;CACtB,GAAG;CACH,QAAQ;CACR,UAAU,kBAAkB,MAAM,QAAQ;CAC1C,OAAO;CACP;CACA,WAAW;AACb;AAEA,MAAa,oBAAoB,WAA+C;CAC9E,GAAG;CACH,QAAQ;CACR,UAAU,kBAAkB,MAAM,QAAQ;CAC1C,OAAO;CACP,WAAW;CACX,WAAW;AACb;AAEA,MAAa,qBACX,QACA,eAAiC,yBACjC,UAAkC,CAAC,MAEnC,OAAO,QAAQ,OAAO,UAAU,2BAA2B,OAAO,OAAO,OAAO,GAAG,YAAY"}

package/dist/client/transport.d.mts

@@ -1,8 +1,8 @@
 import { AgentTranscript } from "./state.mjs";
+import { AgentEvent, AgentReasoningEffort, HitlResponse, QuestionResponse, ToolApprovalResponse } from "@yolk-sdk/agent/protocol";
 import { Cause, Effect, Layer, Stream } from "effect";
 import { HttpClient } from "effect/unstable/http";
 import * as Schema from "effect/Schema";
-import { AgentEvent, AgentReasoningEffort, HitlResponse, QuestionResponse, ToolApprovalResponse } from "@yolk-sdk/agent/protocol";
 
 //#region src/client/transport.d.ts
 declare const AgentTransportError_base: Schema.Class<AgentTransportError, Schema.TaggedStruct<"AgentTransportError", {

package/dist/client/transport.mjs

@@ -1,7 +1,7 @@
+import { AgentEvent, AgentWebSocketServerMessage, QuestionResponseInput, ToolApprovalResponseInput, UserInput } from "@yolk-sdk/agent/protocol";
 import { Cause, Channel, Effect, Exit, Pull, Queue, Ref, Scope, Stream } from "effect";
 import { FetchHttpClient, HttpClient, HttpClientRequest } from "effect/unstable/http";
 import * as Schema from "effect/Schema";
-import { AgentEvent, AgentWebSocketServerMessage, QuestionResponseInput, ToolApprovalResponseInput, UserInput } from "@yolk-sdk/agent/protocol";
 //#region src/client/transport.ts
 var AgentTransportError = class extends Schema.TaggedErrorClass()("AgentTransportError", {
 	message: Schema.String,

package/dist/compaction/transformer.d.mts

@@ -1,7 +1,7 @@
 import { TranscriptTokenEstimator } from "./estimator.mjs";
 import { CompactionResult } from "./window.mjs";
-import { Layer } from "effect";
 import { AgentMessage } from "@yolk-sdk/agent/protocol";
+import { Layer } from "effect";
 import { ContextTransformer } from "@yolk-sdk/agent/loop";
 
 //#region src/compaction/transformer.d.ts

package/dist/loop/error.d.mts

@@ -1,11 +1,12 @@
+import { AgentError, ProviderErrorInfo } from "@yolk-sdk/agent/protocol";
 import * as Schema from "effect/Schema";
-import { AgentError } from "@yolk-sdk/agent/protocol";
 
 //#region src/loop/error.d.ts
 declare const LLMError_base: Schema.Class<LLMError, Schema.TaggedStruct<"LLMError", {
-  readonly cause: Schema.Literals<readonly ["validation_error", "provider_error", "rate_limit", "context_overflow", "invalid_response"]>;
+  readonly cause: Schema.Literals<readonly ["validation_error", "provider_error", "rate_limit", "overloaded", "context_overflow", "invalid_response"]>;
   readonly message: Schema.String;
   readonly retryable: Schema.Boolean;
+  readonly provider: Schema.optional<typeof ProviderErrorInfo>;
 }>, import("effect/Cause").YieldableError>;
 declare class LLMError extends LLMError_base {}
 declare const FauxExhaustedError_base: Schema.Class<FauxExhaustedError, Schema.TaggedStruct<"FauxExhaustedError", {

package/dist/loop/error.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"error.d.mts","names":[],"sources":["../../src/loop/error.ts"],"mappings":";;;;cAC0E,aAAA;;;;;cAE7D,QAAA,SAAiB,aAU5B;AAAA,cAAG,uBAAA;;;cAEQ,kBAAA,SAA2B,uBAKvC;AAAA,cAAG,cAAA;;;;;cAES,SAAA,SAAkB,cAa7B;AAAA,cAAG,0BAAA;;;;;cAEQ,qBAAA,SAA8B,0BAO1C;AAAA,cAAG,eAAA;;;cAES,UAAA,SAAmB,eAE9B;AAAA,KAEU,gBAAA,GAAmB,QAAA,GAAW,kBAAA,GAAqB,UAAA;AAAA,KAEnD,cAAA,GAAiB,gBAAA,GAAmB,SAAA,GAAY,qBAAA;AAAA,cAmB/C,0BAAA,GAA8B,KAAA,EAAO,cAAA,KAAiB,UAiClE"}
+{"version":3,"file":"error.d.mts","names":[],"sources":["../../src/loop/error.ts"],"mappings":";;;;cAC6F,aAAA;;;;;;cAEhF,QAAA,SAAiB,aAY5B;AAAA,cAAG,uBAAA;;;cAEQ,kBAAA,SAA2B,uBAKvC;AAAA,cAAG,cAAA;;;;;cAES,SAAA,SAAkB,cAa7B;AAAA,cAAG,0BAAA;;;;;cAEQ,qBAAA,SAA8B,0BAO1C;AAAA,cAAG,eAAA;;;cAES,UAAA,SAAmB,eAE9B;AAAA,KAEU,gBAAA,GAAmB,QAAA,GAAW,kBAAA,GAAqB,UAAA;AAAA,KAEnD,cAAA,GAAiB,gBAAA,GAAmB,SAAA,GAAY,qBAAA;AAAA,cAmB/C,0BAAA,GAA8B,KAAA,EAAO,cAAA,KAAiB,UAkClE"}

package/dist/loop/error.mjs

@@ -1,16 +1,18 @@
+import { AgentError, ProviderErrorInfo } from "@yolk-sdk/agent/protocol";
 import * as Schema from "effect/Schema";
-import { AgentError } from "@yolk-sdk/agent/protocol";
 //#region src/loop/error.ts
 var LLMError = class extends Schema.TaggedErrorClass()("LLMError", {
 	cause: Schema.Literals([
 		"validation_error",
 		"provider_error",
 		"rate_limit",
+		"overloaded",
 		"context_overflow",
 		"invalid_response"
 	]),
 	message: Schema.String,
-	retryable: Schema.Boolean
+	retryable: Schema.Boolean,
+	provider: Schema.optional(ProviderErrorInfo)
 }) {};
 var FauxExhaustedError = class extends Schema.TaggedErrorClass()("FauxExhaustedError", { message: Schema.String }) {};
 var ToolError = class extends Schema.TaggedErrorClass()("ToolError", {
@@ -54,7 +56,8 @@ const agentLoopErrorToAgentError = (error) => {
 		case "LLMError": return AgentError.make({
 			code: error.cause,
 			message: error.message,
-			retryable: error.retryable
+			retryable: error.retryable,
+			...error.provider === void 0 ? {} : { provider: error.provider }
 		});
 		case "ToolError": return AgentError.make({
 			code: toolErrorCode(error),

package/dist/loop/error.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"error.mjs","names":[],"sources":["../../src/loop/error.ts"],"sourcesContent":["import * as Schema from 'effect/Schema'\nimport { AgentError, type AgentErrorCode } from '@yolk-sdk/agent/protocol'\n\nexport class LLMError extends Schema.TaggedErrorClass<LLMError>()('LLMError', {\n  cause: Schema.Literals([\n    'validation_error',\n    'provider_error',\n    'rate_limit',\n    'context_overflow',\n    'invalid_response'\n  ]),\n  message: Schema.String,\n  retryable: Schema.Boolean\n}) {}\n\nexport class FauxExhaustedError extends Schema.TaggedErrorClass<FauxExhaustedError>()(\n  'FauxExhaustedError',\n  {\n    message: Schema.String\n  }\n) {}\n\nexport class ToolError extends Schema.TaggedErrorClass<ToolError>()('ToolError', {\n  tool: Schema.String,\n  message: Schema.String,\n  cause: Schema.Literals([\n    'validation',\n    'invalid_input',\n    'execution',\n    'timeout',\n    'permission',\n    'denied',\n    'not_found',\n    'unavailable'\n  ])\n}) {}\n\nexport class ContextTransformError extends Schema.TaggedErrorClass<ContextTransformError>()(\n  'ContextTransformError',\n  {\n    cause: Schema.Literals(['context_overflow', 'invalid_response']),\n    message: Schema.String,\n    retryable: Schema.Boolean\n  }\n) {}\n\nexport class AbortError extends Schema.TaggedErrorClass<AbortError>()('AbortError', {\n  reason: Schema.Literals(['user', 'system', 'max_turns'])\n}) {}\n\nexport type LLMProviderError = LLMError | FauxExhaustedError | AbortError\n\nexport type AgentLoopError = LLMProviderError | ToolError | ContextTransformError\n\nconst toolErrorCode = (error: ToolError): AgentErrorCode => {\n  switch (error.cause) {\n    case 'validation':\n    case 'invalid_input':\n      return 'validation_error'\n    case 'timeout':\n      return 'tool_timeout'\n    case 'permission':\n    case 'denied':\n      return 'tool_denied'\n    case 'execution':\n    case 'not_found':\n    case 'unavailable':\n      return 'tool_error'\n  }\n}\n\nexport const agentLoopErrorToAgentError = (error: AgentLoopError): AgentError => {\n  switch (error._tag) {\n    case 'LLMError':\n      return AgentError.make({\n        code: error.cause,\n        message: error.message,\n        retryable: error.retryable\n      })\n    case 'ToolError':\n      return AgentError.make({\n        code: toolErrorCode(error),\n        message: error.message,\n        retryable: error.cause === 'timeout'\n      })\n    case 'ContextTransformError':\n      return AgentError.make({\n        code: error.cause,\n        message: error.message,\n        retryable: error.retryable\n      })\n    case 'AbortError':\n      return AgentError.make({\n        code: 'aborted',\n        message: `Agent run aborted: ${error.reason}`,\n        retryable: error.reason === 'system'\n      })\n    case 'FauxExhaustedError':\n      return AgentError.make({\n        code: 'provider_error',\n        message: error.message,\n        retryable: false\n      })\n  }\n}\n"],"mappings":";;;AAGA,IAAa,WAAb,cAA8B,OAAO,iBAA2B,EAAE,YAAY;CAC5E,OAAO,OAAO,SAAS;EACrB;EACA;EACA;EACA;EACA;CACF,CAAC;CACD,SAAS,OAAO;CAChB,WAAW,OAAO;AACpB,CAAC,EAAE,CAAC;AAEJ,IAAa,qBAAb,cAAwC,OAAO,iBAAqC,EAClF,sBACA,EACE,SAAS,OAAO,OAClB,CACF,EAAE,CAAC;AAEH,IAAa,YAAb,cAA+B,OAAO,iBAA4B,EAAE,aAAa;CAC/E,MAAM,OAAO;CACb,SAAS,OAAO;CAChB,OAAO,OAAO,SAAS;EACrB;EACA;EACA;EACA;EACA;EACA;EACA;EACA;CACF,CAAC;AACH,CAAC,EAAE,CAAC;AAEJ,IAAa,wBAAb,cAA2C,OAAO,iBAAwC,EACxF,yBACA;CACE,OAAO,OAAO,SAAS,CAAC,oBAAoB,kBAAkB,CAAC;CAC/D,SAAS,OAAO;CAChB,WAAW,OAAO;AACpB,CACF,EAAE,CAAC;AAEH,IAAa,aAAb,cAAgC,OAAO,iBAA6B,EAAE,cAAc,EAClF,QAAQ,OAAO,SAAS;CAAC;CAAQ;CAAU;AAAW,CAAC,EACzD,CAAC,EAAE,CAAC;AAMJ,MAAM,iBAAiB,UAAqC;CAC1D,QAAQ,MAAM,OAAd;EACE,KAAK;EACL,KAAK,iBACH,OAAO;EACT,KAAK,WACH,OAAO;EACT,KAAK;EACL,KAAK,UACH,OAAO;EACT,KAAK;EACL,KAAK;EACL,KAAK,eACH,OAAO;CACX;AACF;AAEA,MAAa,8BAA8B,UAAsC;CAC/E,QAAQ,MAAM,MAAd;EACE,KAAK,YACH,OAAO,WAAW,KAAK;GACrB,MAAM,MAAM;GACZ,SAAS,MAAM;GACf,WAAW,MAAM;EACnB,CAAC;EACH,KAAK,aACH,OAAO,WAAW,KAAK;GACrB,MAAM,cAAc,KAAK;GACzB,SAAS,MAAM;GACf,WAAW,MAAM,UAAU;EAC7B,CAAC;EACH,KAAK,yBACH,OAAO,WAAW,KAAK;GACrB,MAAM,MAAM;GACZ,SAAS,MAAM;GACf,WAAW,MAAM;EACnB,CAAC;EACH,KAAK,cACH,OAAO,WAAW,KAAK;GACrB,MAAM;GACN,SAAS,sBAAsB,MAAM;GACrC,WAAW,MAAM,WAAW;EAC9B,CAAC;EACH,KAAK,sBACH,OAAO,WAAW,KAAK;GACrB,MAAM;GACN,SAAS,MAAM;GACf,WAAW;EACb,CAAC;CACL;AACF"}
+{"version":3,"file":"error.mjs","names":[],"sources":["../../src/loop/error.ts"],"sourcesContent":["import * as Schema from 'effect/Schema'\nimport { AgentError, ProviderErrorInfo, type AgentErrorCode } from '@yolk-sdk/agent/protocol'\n\nexport class LLMError extends Schema.TaggedErrorClass<LLMError>()('LLMError', {\n  cause: Schema.Literals([\n    'validation_error',\n    'provider_error',\n    'rate_limit',\n    'overloaded',\n    'context_overflow',\n    'invalid_response'\n  ]),\n  message: Schema.String,\n  retryable: Schema.Boolean,\n  provider: Schema.optional(ProviderErrorInfo)\n}) {}\n\nexport class FauxExhaustedError extends Schema.TaggedErrorClass<FauxExhaustedError>()(\n  'FauxExhaustedError',\n  {\n    message: Schema.String\n  }\n) {}\n\nexport class ToolError extends Schema.TaggedErrorClass<ToolError>()('ToolError', {\n  tool: Schema.String,\n  message: Schema.String,\n  cause: Schema.Literals([\n    'validation',\n    'invalid_input',\n    'execution',\n    'timeout',\n    'permission',\n    'denied',\n    'not_found',\n    'unavailable'\n  ])\n}) {}\n\nexport class ContextTransformError extends Schema.TaggedErrorClass<ContextTransformError>()(\n  'ContextTransformError',\n  {\n    cause: Schema.Literals(['context_overflow', 'invalid_response']),\n    message: Schema.String,\n    retryable: Schema.Boolean\n  }\n) {}\n\nexport class AbortError extends Schema.TaggedErrorClass<AbortError>()('AbortError', {\n  reason: Schema.Literals(['user', 'system', 'max_turns'])\n}) {}\n\nexport type LLMProviderError = LLMError | FauxExhaustedError | AbortError\n\nexport type AgentLoopError = LLMProviderError | ToolError | ContextTransformError\n\nconst toolErrorCode = (error: ToolError): AgentErrorCode => {\n  switch (error.cause) {\n    case 'validation':\n    case 'invalid_input':\n      return 'validation_error'\n    case 'timeout':\n      return 'tool_timeout'\n    case 'permission':\n    case 'denied':\n      return 'tool_denied'\n    case 'execution':\n    case 'not_found':\n    case 'unavailable':\n      return 'tool_error'\n  }\n}\n\nexport const agentLoopErrorToAgentError = (error: AgentLoopError): AgentError => {\n  switch (error._tag) {\n    case 'LLMError':\n      return AgentError.make({\n        code: error.cause,\n        message: error.message,\n        retryable: error.retryable,\n        ...(error.provider === undefined ? {} : { provider: error.provider })\n      })\n    case 'ToolError':\n      return AgentError.make({\n        code: toolErrorCode(error),\n        message: error.message,\n        retryable: error.cause === 'timeout'\n      })\n    case 'ContextTransformError':\n      return AgentError.make({\n        code: error.cause,\n        message: error.message,\n        retryable: error.retryable\n      })\n    case 'AbortError':\n      return AgentError.make({\n        code: 'aborted',\n        message: `Agent run aborted: ${error.reason}`,\n        retryable: error.reason === 'system'\n      })\n    case 'FauxExhaustedError':\n      return AgentError.make({\n        code: 'provider_error',\n        message: error.message,\n        retryable: false\n      })\n  }\n}\n"],"mappings":";;;AAGA,IAAa,WAAb,cAA8B,OAAO,iBAA2B,EAAE,YAAY;CAC5E,OAAO,OAAO,SAAS;EACrB;EACA;EACA;EACA;EACA;EACA;CACF,CAAC;CACD,SAAS,OAAO;CAChB,WAAW,OAAO;CAClB,UAAU,OAAO,SAAS,iBAAiB;AAC7C,CAAC,EAAE,CAAC;AAEJ,IAAa,qBAAb,cAAwC,OAAO,iBAAqC,EAClF,sBACA,EACE,SAAS,OAAO,OAClB,CACF,EAAE,CAAC;AAEH,IAAa,YAAb,cAA+B,OAAO,iBAA4B,EAAE,aAAa;CAC/E,MAAM,OAAO;CACb,SAAS,OAAO;CAChB,OAAO,OAAO,SAAS;EACrB;EACA;EACA;EACA;EACA;EACA;EACA;EACA;CACF,CAAC;AACH,CAAC,EAAE,CAAC;AAEJ,IAAa,wBAAb,cAA2C,OAAO,iBAAwC,EACxF,yBACA;CACE,OAAO,OAAO,SAAS,CAAC,oBAAoB,kBAAkB,CAAC;CAC/D,SAAS,OAAO;CAChB,WAAW,OAAO;AACpB,CACF,EAAE,CAAC;AAEH,IAAa,aAAb,cAAgC,OAAO,iBAA6B,EAAE,cAAc,EAClF,QAAQ,OAAO,SAAS;CAAC;CAAQ;CAAU;AAAW,CAAC,EACzD,CAAC,EAAE,CAAC;AAMJ,MAAM,iBAAiB,UAAqC;CAC1D,QAAQ,MAAM,OAAd;EACE,KAAK;EACL,KAAK,iBACH,OAAO;EACT,KAAK,WACH,OAAO;EACT,KAAK;EACL,KAAK,UACH,OAAO;EACT,KAAK;EACL,KAAK;EACL,KAAK,eACH,OAAO;CACX;AACF;AAEA,MAAa,8BAA8B,UAAsC;CAC/E,QAAQ,MAAM,MAAd;EACE,KAAK,YACH,OAAO,WAAW,KAAK;GACrB,MAAM,MAAM;GACZ,SAAS,MAAM;GACf,WAAW,MAAM;GACjB,GAAI,MAAM,aAAa,KAAA,IAAY,CAAC,IAAI,EAAE,UAAU,MAAM,SAAS;EACrE,CAAC;EACH,KAAK,aACH,OAAO,WAAW,KAAK;GACrB,MAAM,cAAc,KAAK;GACzB,SAAS,MAAM;GACf,WAAW,MAAM,UAAU;EAC7B,CAAC;EACH,KAAK,yBACH,OAAO,WAAW,KAAK;GACrB,MAAM,MAAM;GACZ,SAAS,MAAM;GACf,WAAW,MAAM;EACnB,CAAC;EACH,KAAK,cACH,OAAO,WAAW,KAAK;GACrB,MAAM;GACN,SAAS,sBAAsB,MAAM;GACrC,WAAW,MAAM,WAAW;EAC9B,CAAC;EACH,KAAK,sBACH,OAAO,WAAW,KAAK;GACrB,MAAM;GACN,SAAS,MAAM;GACf,WAAW;EACb,CAAC;CACL;AACF"}

package/dist/loop/llm-event.d.mts

@@ -1,5 +1,5 @@
-import * as Schema from "effect/Schema";
 import { AgentUsage, ToolCall, ToolResult } from "@yolk-sdk/agent/protocol";
+import * as Schema from "effect/Schema";
 
 //#region src/loop/llm-event.d.ts
 declare const LLMTextDelta_base: Schema.Class<LLMTextDelta, Schema.TaggedStruct<"TextDelta", {

package/dist/loop/llm-event.mjs

@@ -1,5 +1,5 @@
-import * as Schema from "effect/Schema";
 import { AgentUsage, ToolCall, ToolResult } from "@yolk-sdk/agent/protocol";
+import * as Schema from "effect/Schema";
 //#region src/loop/llm-event.ts
 var LLMTextDelta = class extends Schema.TaggedClass()("TextDelta", { text: Schema.String }) {};
 var LLMReasoningDelta = class extends Schema.TaggedClass()("ReasoningDelta", { text: Schema.String }) {};

package/dist/loop/run.d.mts

@@ -3,8 +3,8 @@ import { ContextTransformer } from "./services/context-transformer.mjs";
 import { LLMProvider } from "./services/llm-provider.mjs";
 import { LoopConfig } from "./services/loop-config.mjs";
 import { ToolExecutor } from "./services/tool-executor.mjs";
-import { Stream } from "effect";
 import { AgentEvent, AgentMessage, AgentModelCapabilities, AgentReasoningEffort, AgentUsage, HitlResponse, ToolCall, ToolDef } from "@yolk-sdk/agent/protocol";
+import { Stream } from "effect";
 
 //#region src/loop/run.d.ts
 type AgentLoopRunId = string;

package/dist/loop/run.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"run.d.mts","names":[],"sources":["../../src/loop/run.ts"],"mappings":";;;;;;;;;KAoEY,cAAA;AAAA,KAEA,SAAA;EAAA,SACD,QAAA,EAAU,aAAA,CAAc,YAAA;EAAA,SACxB,YAAA;EAAA,SACA,KAAA,EAAO,aAAA,CAAc,OAAA;EAAA,SACrB,aAAA,GAAgB,aAAA,CAAc,YAAA;EAAA,SAC9B,KAAA;EAAA,SACA,eAAA,GAAkB,oBAAA;EAAA,SAClB,YAAA,GAAe,sBAAA;AAAA;AAAA,KAGd,eAAA,GAAkB,SAAS;EAAA,SAC5B,IAAI;AAAA;AAAA,KAGH,eAAA;EAAA,SACD,KAAA,EAAO,aAAA,CAAc,QAAA;EAAA,SACrB,KAAA,GAAQ,aAAA,CAAc,OAAA;EAAA,SACtB,aAAA,GAAgB,aAAA,CAAc,YAAA;EAAA,SAC9B,KAAA;EAAA,SACA,eAAA,GAAkB,aAAA,CAAc,YAAA;EAAA,SAChC,IAAA;EAAA,SACA,KAAA,GAAQ,UAAA;AAAA;AAAA,cAqhCN,YAAA,GACX,MAAA,EAAQ,eAAA,KACP,MAAA,CAAO,MAAA,CAAO,UAAA,EAAY,cAAA,EAAgB,kBAAA,GAAqB,WAAA,GAAc,UAAA;AAAA,cAuBnE,YAAA,GACX,MAAA,EAAQ,eAAA,KACP,MAAA,CAAO,MAAA,CAAO,UAAA,EAAY,cAAA,EAAgB,UAAA,GAAa,YAAA;AAAA,cA6C7C,GAAA,GACX,MAAA,EAAQ,SAAA,KACP,MAAA,CAAO,MAAA,CACR,UAAA,EACA,cAAA,EACA,kBAAA,GAAqB,WAAA,GAAc,UAAA,GAAa,YAAA"}
+{"version":3,"file":"run.d.mts","names":[],"sources":["../../src/loop/run.ts"],"mappings":";;;;;;;;;KAoEY,cAAA;AAAA,KAEA,SAAA;EAAA,SACD,QAAA,EAAU,aAAA,CAAc,YAAA;EAAA,SACxB,YAAA;EAAA,SACA,KAAA,EAAO,aAAA,CAAc,OAAA;EAAA,SACrB,aAAA,GAAgB,aAAA,CAAc,YAAA;EAAA,SAC9B,KAAA;EAAA,SACA,eAAA,GAAkB,oBAAA;EAAA,SAClB,YAAA,GAAe,sBAAA;AAAA;AAAA,KAGd,eAAA,GAAkB,SAAS;EAAA,SAC5B,IAAI;AAAA;AAAA,KAGH,eAAA;EAAA,SACD,KAAA,EAAO,aAAA,CAAc,QAAA;EAAA,SACrB,KAAA,GAAQ,aAAA,CAAc,OAAA;EAAA,SACtB,aAAA,GAAgB,aAAA,CAAc,YAAA;EAAA,SAC9B,KAAA;EAAA,SACA,eAAA,GAAkB,aAAA,CAAc,YAAA;EAAA,SAChC,IAAA;EAAA,SACA,KAAA,GAAQ,UAAA;AAAA;AAAA,cA4iCN,YAAA,GACX,MAAA,EAAQ,eAAA,KACP,MAAA,CAAO,MAAA,CAAO,UAAA,EAAY,cAAA,EAAgB,kBAAA,GAAqB,WAAA,GAAc,UAAA;AAAA,cAuBnE,YAAA,GACX,MAAA,EAAQ,eAAA,KACP,MAAA,CAAO,MAAA,CAAO,UAAA,EAAY,cAAA,EAAgB,UAAA,GAAa,YAAA;AAAA,cA6C7C,GAAA,GACX,MAAA,EAAQ,SAAA,KACP,MAAA,CAAO,MAAA,CACR,UAAA,EACA,cAAA,EACA,kBAAA,GAAqB,WAAA,GAAc,UAAA,GAAa,YAAA"}

package/dist/loop/run.mjs

@@ -4,9 +4,9 @@ import { ContextTransformer } from "./services/context-transformer.mjs";
 import { LLMProvider } from "./services/llm-provider.mjs";
 import { LoopConfig } from "./services/loop-config.mjs";
 import { ToolExecutor } from "./services/tool-executor.mjs";
+import { AgentAwaitingInput, AgentEnd, AgentRetry, AgentStart, AssistantMessageEvent, LLMReasoningDelta, LLMStreamEnd, LLMStreamStart, LLMTextDelta, ProviderToolResult, QuestionRequest, QuestionRequested, QuestionToolParams, SubagentCompleted, SubagentStarted, ToolApprovalRequest, ToolApprovalRequested, ToolExecutionCompleted, ToolExecutionError, ToolExecutionStarted, ToolInputDelta, ToolInputEnd, ToolInputStart, ToolResult, ToolResultMessage, TurnEnd, TurnStart, UsageUpdate, addAgentUsage, assistantHostToolCalls, contentParts, contentPreview, formatQuestionResponseContent, hitlResponseEvent, makeSubagentRunId, questionResponseStructuredContent, zeroAgentUsage } from "@yolk-sdk/agent/protocol";
 import { Clock, Effect, Ref, Stream } from "effect";
 import * as Schema from "effect/Schema";
-import { AgentAwaitingInput, AgentEnd, AgentRetry, AgentStart, AssistantMessageEvent, LLMReasoningDelta, LLMStreamEnd, LLMStreamStart, LLMTextDelta, ProviderToolResult, QuestionRequest, QuestionRequested, QuestionToolParams, SubagentCompleted, SubagentStarted, ToolApprovalRequest, ToolApprovalRequested, ToolExecutionCompleted, ToolExecutionError, ToolExecutionStarted, ToolInputDelta, ToolInputEnd, ToolInputStart, ToolResult, ToolResultMessage, TurnEnd, TurnStart, UsageUpdate, addAgentUsage, assistantHostToolCalls, contentParts, contentPreview, formatQuestionResponseContent, hitlResponseEvent, makeSubagentRunId, questionResponseStructuredContent, zeroAgentUsage } from "@yolk-sdk/agent/protocol";
 //#region src/loop/run.ts
 const questionToolName = "question";
 const objectField = (input, key) => input !== null && typeof input === "object" ? Object.getOwnPropertyDescriptor(input, key)?.value : void 0;
@@ -111,7 +111,20 @@ const isLlmEvent = (event) => {
 		default: return false;
 	}
 };
-const retryDelayMs = (baseDelayMs, attempt) => Math.max(0, Math.floor(baseDelayMs * 2 ** Math.max(0, attempt - 1)));
+const maxUnhintedRetryDelayMs = 3e4;
+const maxHintedRetryDelayMs = 2147483647;
+const validDelayMs = (delayMs) => Number.isFinite(delayMs) && delayMs >= 0 ? Math.floor(delayMs) : void 0;
+const hintedRetryDelayMs = (error) => {
+	const delayMs = error.provider?.retryAfterMs;
+	const validDelay = delayMs === void 0 ? void 0 : validDelayMs(delayMs);
+	return validDelay === void 0 ? void 0 : Math.min(validDelay, maxHintedRetryDelayMs);
+};
+const retryDelayMs = (baseDelayMs, attempt, error) => {
+	const hintedDelay = hintedRetryDelayMs(error);
+	if (hintedDelay !== void 0) return hintedDelay;
+	const delayMs = validDelayMs(baseDelayMs * 2 ** Math.max(0, attempt - 1)) ?? 0;
+	return Math.min(delayMs, maxUnhintedRetryDelayMs);
+};
 const retryReason = (error) => error.cause;
 const retrySleep = (delayMs) => delayMs === 0 ? Effect.void : Effect.sleep(`${delayMs} millis`);
 const failAgentLoopError = (error) => Stream.fail(error);
@@ -119,12 +132,13 @@ const sleepStream = (delayMs) => Stream.fromEffect(retrySleep(delayMs)).pipe(Str
 const withProviderRetries = (stream, loopConfig, makeStream, attempt) => Stream.unwrap(Ref.make(false).pipe(Effect.map((emittedProviderEvent) => stream.pipe(Stream.tap(() => Ref.set(emittedProviderEvent, true)), Stream.catchTags({
 	LLMError: (error) => Stream.unwrap(Ref.get(emittedProviderEvent).pipe(Effect.map((emitted) => {
 		if (emitted || !error.retryable || error.cause === "context_overflow" || attempt > loopConfig.maxRetries) return failAgentLoopError(error);
-		const delayMs = retryDelayMs(loopConfig.retryBaseDelayMs, attempt);
+		const delayMs = retryDelayMs(loopConfig.retryBaseDelayMs, attempt, error);
 		return Stream.make(AgentRetry.make({
 			attempt,
 			reason: retryReason(error),
 			delayMs,
-			message: error.message
+			message: error.message,
+			...error.provider === void 0 ? {} : { provider: error.provider }
 		})).pipe(Stream.concat(sleepStream(delayMs)), Stream.concat(withProviderRetries(makeStream(), loopConfig, makeStream, attempt + 1)));
 	}))),
 	AbortError: failAgentLoopError,