@yolk-sdk/agent 0.1.0-canary.20 → 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
@@ -220,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/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/loop/error.d.mts

@@ -1,11 +1,12 @@
-import { AgentError } from "@yolk-sdk/agent/protocol";
+import { AgentError, ProviderErrorInfo } from "@yolk-sdk/agent/protocol";
 import * as Schema from "effect/Schema";
 
 //#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,4 +1,4 @@
-import { AgentError } from "@yolk-sdk/agent/protocol";
+import { AgentError, ProviderErrorInfo } from "@yolk-sdk/agent/protocol";
 import * as Schema from "effect/Schema";
 //#region src/loop/error.ts
 var LLMError = class extends Schema.TaggedErrorClass()("LLMError", {
@@ -6,11 +6,13 @@ var LLMError = class extends Schema.TaggedErrorClass()("LLMError", {
 		"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/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

@@ -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,