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

package/dist/react/chat-core.mjs

@@ -5,6 +5,8 @@ import { hitlResponseEvent } from "@yolk-sdk/agent/protocol";
 const initialAgentChatState = {
 	status: "idle",
 	error: null,
+	errorInfo: null,
+	retryInfo: null,
 	chatMessages: [],
 	sessionEvents: [],
 	seenEventIds: []
@@ -14,17 +16,25 @@ const rememberEvent = (state, event) => event.eventId === void 0 ? state : {
 	...state,
 	seenEventIds: [...state.seenEventIds, event.eventId]
 };
+const clearRetryInfo = (state) => state.retryInfo === null ? state : {
+	...state,
+	retryInfo: null
+};
 const reduceAgentChatState = (state, action) => {
 	switch (action._tag) {
 		case "HydrateMessage": return {
 			...state,
 			chatMessages: appendProtocolMessage(state.chatMessages, action.message),
-			error: null
+			error: null,
+			errorInfo: null,
+			retryInfo: null
 		};
 		case "Submit": return {
 			...state,
 			status: "running",
 			error: null,
+			errorInfo: null,
+			retryInfo: null,
 			seenEventIds: [],
 			chatMessages: appendProtocolMessage(state.chatMessages, action.message),
 			sessionEvents: [...state.sessionEvents, UserMessageSubmitted.make({ message: action.message })]
@@ -33,12 +43,16 @@ const reduceAgentChatState = (state, action) => {
 			...state,
 			chatMessages: appendProtocolMessage(state.chatMessages, action.message),
 			error: null,
+			errorInfo: null,
+			retryInfo: null,
 			sessionEvents: [...state.sessionEvents, ProtocolMessageAppended.make({ message: action.message })]
 		};
 		case "SubmitHitlResponse": return {
 			...state,
 			status: "running",
 			error: null,
+			errorInfo: null,
+			retryInfo: null,
 			seenEventIds: [],
 			chatMessages: applyAgentEventToChatMessages(state.chatMessages, hitlResponseEvent(action.response))
 		};
@@ -48,6 +62,8 @@ const reduceAgentChatState = (state, action) => {
 			return {
 				...state,
 				error: null,
+				errorInfo: null,
+				retryInfo: null,
 				chatMessages: next.messages,
 				sessionEvents: [...state.sessionEvents, TurnDeleted.make({
 					turnStartMessageId: next.turnStartMessageId,
@@ -62,6 +78,8 @@ const reduceAgentChatState = (state, action) => {
 				...state,
 				status: "running",
 				error: null,
+				errorInfo: null,
+				retryInfo: null,
 				seenEventIds: [],
 				chatMessages: next.messages,
 				sessionEvents: [...state.sessionEvents, MessagesRegenerated.make({
@@ -77,6 +95,8 @@ const reduceAgentChatState = (state, action) => {
 				...state,
 				status: "running",
 				error: null,
+				errorInfo: null,
+				retryInfo: null,
 				seenEventIds: [],
 				chatMessages: next.messages,
 				sessionEvents: [...state.sessionEvents, UserMessageEdited.make({
@@ -93,29 +113,41 @@ const reduceAgentChatState = (state, action) => {
 					...state,
 					status: "running",
 					error: null,
+					errorInfo: null,
+					retryInfo: null,
 					chatMessages: applyAgentEventToChatMessages(state.chatMessages, action.event, action)
 				}, action.event);
 				case "AgentError": return rememberEvent({
 					...state,
 					status: "error",
 					error: action.event.message,
+					errorInfo: action.event,
+					retryInfo: null,
 					chatMessages: applyAgentEventToChatMessages(state.chatMessages, action.event, action)
 				}, action.event);
 				case "AgentEnd": return rememberEvent({
 					...state,
 					status: "done",
 					error: null,
+					errorInfo: null,
+					retryInfo: null,
 					chatMessages: applyAgentEventToChatMessages(state.chatMessages, action.event, action)
 				}, action.event);
 				case "AgentAwaitingInput": return rememberEvent({
 					...state,
 					status: "waiting",
 					error: null,
+					errorInfo: null,
+					retryInfo: null,
+					chatMessages: applyAgentEventToChatMessages(state.chatMessages, action.event, action)
+				}, action.event);
+				case "AgentRetry": return rememberEvent({
+					...state,
+					retryInfo: action.event,
 					chatMessages: applyAgentEventToChatMessages(state.chatMessages, action.event, action)
 				}, action.event);
-				case "AssistantMessage":
-				case "AgentRetry":
 				case "CompactionEnd":
+				case "AssistantMessage":
 				case "CompactionStart":
 				case "LLMReasoningDelta":
 				case "LLMStreamEnd":
@@ -138,21 +170,25 @@ const reduceAgentChatState = (state, action) => {
 				case "ToolInputStart":
 				case "TurnEnd":
 				case "TurnStart":
-				case "UsageUpdate": return rememberEvent({
+				case "UsageUpdate": return rememberEvent(clearRetryInfo({
 					...state,
 					chatMessages: applyAgentEventToChatMessages(state.chatMessages, action.event, action)
-				}, action.event);
+				}), action.event);
 			}
 		case "Error": return {
 			...state,
 			status: "error",
 			error: action.message,
+			errorInfo: null,
+			retryInfo: null,
 			chatMessages: markChatError(state.chatMessages, action.message)
 		};
 		case "Abort": return {
 			...state,
 			status: "aborted",
-			error: null
+			error: null,
+			errorInfo: null,
+			retryInfo: null
 		};
 	}
 };

package/dist/react/chat-core.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"chat-core.mjs","names":[],"sources":["../../src/react/chat-core.ts"],"sourcesContent":["import {\n  hitlResponseEvent,\n  type AgentEvent,\n  type AgentMessage,\n  type Content,\n  type HitlResponse,\n  type UserMessage\n} from '@yolk-sdk/agent/protocol'\nimport {\n  appendProtocolMessage,\n  applyAgentEventToChatMessages,\n  deleteChatTurn,\n  editChatUserMessage,\n  markChatError,\n  regenerateChatMessagesFrom,\n  type AgentChatMessage,\n  type AgentChatPart,\n  type ApplyAgentEventToChatMessagesOptions\n} from './chat-messages.ts'\nimport {\n  MessagesRegenerated,\n  ProtocolMessageAppended,\n  TurnDeleted,\n  UserMessageEdited,\n  UserMessageSubmitted,\n  type AgentChatSessionEvent\n} from './chat-session-events.ts'\n\nexport type AgentRunStatus = 'idle' | 'running' | 'waiting' | 'done' | 'error' | 'aborted'\n\nexport type AgentChatState = {\n  readonly status: AgentRunStatus\n  readonly error: string | null\n  readonly chatMessages: ReadonlyArray<AgentChatMessage>\n  readonly sessionEvents: ReadonlyArray<AgentChatSessionEvent>\n  readonly seenEventIds: ReadonlyArray<string>\n}\n\nexport const initialAgentChatState: AgentChatState = {\n  status: 'idle',\n  error: null,\n  chatMessages: [],\n  sessionEvents: [],\n  seenEventIds: []\n}\n\nconst hasSeenEvent = (state: AgentChatState, event: AgentEvent) =>\n  event.eventId !== undefined && state.seenEventIds.includes(event.eventId)\n\nconst rememberEvent = (state: AgentChatState, event: AgentEvent): AgentChatState =>\n  event.eventId === undefined\n    ? state\n    : { ...state, seenEventIds: [...state.seenEventIds, event.eventId] }\n\nexport type AgentChatAction =\n  | { readonly _tag: 'HydrateMessage'; readonly message: AgentMessage }\n  | { readonly _tag: 'Submit'; readonly message: UserMessage }\n  | { readonly _tag: 'AppendMessage'; readonly message: AgentMessage }\n  | { readonly _tag: 'SubmitHitlResponse'; readonly response: HitlResponse }\n  | { readonly _tag: 'DeleteTurn'; readonly messageId: string }\n  | { readonly _tag: 'RegenerateFrom'; readonly messageId: string }\n  | { readonly _tag: 'EditUserMessage'; readonly messageId: string; readonly content: Content }\n  | ({ readonly _tag: 'Event'; readonly event: AgentEvent } & ApplyAgentEventToChatMessagesOptions)\n  | { readonly _tag: 'Error'; readonly message: string }\n  | { readonly _tag: 'Abort' }\n\nexport const reduceAgentChatState = (\n  state: AgentChatState,\n  action: AgentChatAction\n): AgentChatState => {\n  switch (action._tag) {\n    case 'HydrateMessage':\n      return {\n        ...state,\n        chatMessages: appendProtocolMessage(state.chatMessages, action.message),\n        error: null\n      }\n    case 'Submit':\n      return {\n        ...state,\n        status: 'running',\n        error: null,\n        seenEventIds: [],\n        chatMessages: appendProtocolMessage(state.chatMessages, action.message),\n        sessionEvents: [\n          ...state.sessionEvents,\n          UserMessageSubmitted.make({ message: action.message })\n        ]\n      }\n    case 'AppendMessage':\n      return {\n        ...state,\n        chatMessages: appendProtocolMessage(state.chatMessages, action.message),\n        error: null,\n        sessionEvents: [\n          ...state.sessionEvents,\n          ProtocolMessageAppended.make({ message: action.message })\n        ]\n      }\n    case 'SubmitHitlResponse':\n      return {\n        ...state,\n        status: 'running',\n        error: null,\n        seenEventIds: [],\n        chatMessages: applyAgentEventToChatMessages(\n          state.chatMessages,\n          hitlResponseEvent(action.response)\n        )\n      }\n    case 'DeleteTurn': {\n      const next = deleteChatTurn(state.chatMessages, action.messageId)\n\n      if (next._tag === 'NotFound') {\n        return state\n      }\n\n      return {\n        ...state,\n        error: null,\n        chatMessages: next.messages,\n        sessionEvents: [\n          ...state.sessionEvents,\n          TurnDeleted.make({\n            turnStartMessageId: next.turnStartMessageId,\n            deletedMessageIds: next.deletedMessageIds\n          })\n        ]\n      }\n    }\n    case 'RegenerateFrom': {\n      const next = regenerateChatMessagesFrom(state.chatMessages, action.messageId)\n\n      if (next._tag === 'NotFound') {\n        return state\n      }\n\n      return {\n        ...state,\n        status: 'running',\n        error: null,\n        seenEventIds: [],\n        chatMessages: next.messages,\n        sessionEvents: [\n          ...state.sessionEvents,\n          MessagesRegenerated.make({\n            fromMessageId: action.messageId,\n            keptMessageIds: next.messages.map(message => message.id)\n          })\n        ]\n      }\n    }\n    case 'EditUserMessage': {\n      const next = editChatUserMessage(state.chatMessages, action.messageId, action.content)\n\n      if (next._tag !== 'Edited') {\n        return state\n      }\n\n      return {\n        ...state,\n        status: 'running',\n        error: null,\n        seenEventIds: [],\n        chatMessages: next.messages,\n        sessionEvents: [\n          ...state.sessionEvents,\n          UserMessageEdited.make({\n            messageId: next.messageId,\n            content: action.content,\n            keptMessageIds: next.messages.map(message => message.id)\n          })\n        ]\n      }\n    }\n    case 'Event': {\n      if (hasSeenEvent(state, action.event)) {\n        return state\n      }\n\n      switch (action.event._tag) {\n        case 'AgentStart':\n          return rememberEvent(\n            {\n              ...state,\n              status: 'running',\n              error: null,\n              chatMessages: applyAgentEventToChatMessages(state.chatMessages, action.event, action)\n            },\n            action.event\n          )\n        case 'AgentError':\n          return rememberEvent(\n            {\n              ...state,\n              status: 'error',\n              error: action.event.message,\n              chatMessages: applyAgentEventToChatMessages(state.chatMessages, action.event, action)\n            },\n            action.event\n          )\n        case 'AgentEnd':\n          return rememberEvent(\n            {\n              ...state,\n              status: 'done',\n              error: null,\n              chatMessages: applyAgentEventToChatMessages(state.chatMessages, action.event, action)\n            },\n            action.event\n          )\n        case 'AgentAwaitingInput':\n          return rememberEvent(\n            {\n              ...state,\n              status: 'waiting',\n              error: null,\n              chatMessages: applyAgentEventToChatMessages(state.chatMessages, action.event, action)\n            },\n            action.event\n          )\n        case 'AssistantMessage':\n        case 'AgentRetry':\n        case 'CompactionEnd':\n        case 'CompactionStart':\n        case 'LLMReasoningDelta':\n        case 'LLMStreamEnd':\n        case 'LLMStreamStart':\n        case 'LLMTextDelta':\n        case 'ProviderToolResult':\n        case 'QuestionAnswered':\n        case 'QuestionCancelled':\n        case 'QuestionRequested':\n        case 'SubagentCompleted':\n        case 'SubagentStarted':\n        case 'ToolApprovalDenied':\n        case 'ToolApprovalGranted':\n        case 'ToolApprovalRequested':\n        case 'ToolExecutionCompleted':\n        case 'ToolExecutionError':\n        case 'ToolExecutionStarted':\n        case 'ToolInputDelta':\n        case 'ToolInputEnd':\n        case 'ToolInputStart':\n        case 'TurnEnd':\n        case 'TurnStart':\n        case 'UsageUpdate':\n          return rememberEvent(\n            {\n              ...state,\n              chatMessages: applyAgentEventToChatMessages(state.chatMessages, action.event, action)\n            },\n            action.event\n          )\n      }\n    }\n    case 'Error': {\n      return {\n        ...state,\n        status: 'error',\n        error: action.message,\n        chatMessages: markChatError(state.chatMessages, action.message)\n      }\n    }\n    case 'Abort':\n      return { ...state, status: 'aborted', error: null }\n  }\n}\n\nexport const hasAgentMessageReasoning = (message: AgentMessage) =>\n  message._tag === 'Assistant' &&\n  message.parts.some(part => part._tag === 'Reasoning' && part.text.length > 0)\n\nexport const hasAgentChatReasoningSummary = (state: AgentChatState) =>\n  state.chatMessages.some(message =>\n    message.parts.some(part => part._tag === 'Reasoning' && part.text.length > 0)\n  )\n\nexport const isActiveChatToolPart = (part: AgentChatPart) =>\n  part._tag === 'ToolCall' &&\n  part.state._tag !== 'Completed' &&\n  part.state._tag !== 'ProviderCompleted' &&\n  part.state._tag !== 'QuestionAnswered' &&\n  part.state._tag !== 'QuestionCancelled' &&\n  part.state._tag !== 'Errored' &&\n  part.state._tag !== 'Denied'\nexport type ActiveChatToolPart = Extract<AgentChatPart, { readonly _tag: 'ToolCall' }>\n\nexport const isCompletedChatToolPart = (part: AgentChatPart) =>\n  part._tag === 'ToolCall' && part.state._tag === 'Completed'\nexport type CompletedChatToolPart = Extract<AgentChatPart, { readonly _tag: 'ToolCall' }>\n\nexport const getActiveChatToolParts = (messages: ReadonlyArray<AgentChatMessage>) =>\n  messages.flatMap(message =>\n    message.parts.filter((part): part is ActiveChatToolPart => isActiveChatToolPart(part))\n  )\n\nexport const getCompletedChatToolParts = (messages: ReadonlyArray<AgentChatMessage>) =>\n  messages.flatMap(message =>\n    message.parts.filter((part): part is CompletedChatToolPart => isCompletedChatToolPart(part))\n  )\n\ntype AgentChatLiveActivityInput = {\n  readonly isTextRunning: boolean\n  readonly activeToolCallCount: number\n  readonly isVoiceActive: boolean\n}\n\nexport const getAgentChatLiveActivityCount = ({\n  isTextRunning,\n  activeToolCallCount,\n  isVoiceActive\n}: AgentChatLiveActivityInput) =>\n  (isTextRunning ? 1 : 0) + activeToolCallCount + (isVoiceActive ? 1 : 0)\n"],"mappings":";;;;AAsCA,MAAa,wBAAwC;CACnD,QAAQ;CACR,OAAO;CACP,cAAc,CAAC;CACf,eAAe,CAAC;CAChB,cAAc,CAAC;AACjB;AAEA,MAAM,gBAAgB,OAAuB,UAC3C,MAAM,YAAY,KAAA,KAAa,MAAM,aAAa,SAAS,MAAM,OAAO;AAE1E,MAAM,iBAAiB,OAAuB,UAC5C,MAAM,YAAY,KAAA,IACd,QACA;CAAE,GAAG;CAAO,cAAc,CAAC,GAAG,MAAM,cAAc,MAAM,OAAO;AAAE;AAcvE,MAAa,wBACX,OACA,WACmB;CACnB,QAAQ,OAAO,MAAf;EACE,KAAK,kBACH,OAAO;GACL,GAAG;GACH,cAAc,sBAAsB,MAAM,cAAc,OAAO,OAAO;GACtE,OAAO;EACT;EACF,KAAK,UACH,OAAO;GACL,GAAG;GACH,QAAQ;GACR,OAAO;GACP,cAAc,CAAC;GACf,cAAc,sBAAsB,MAAM,cAAc,OAAO,OAAO;GACtE,eAAe,CACb,GAAG,MAAM,eACT,qBAAqB,KAAK,EAAE,SAAS,OAAO,QAAQ,CAAC,CACvD;EACF;EACF,KAAK,iBACH,OAAO;GACL,GAAG;GACH,cAAc,sBAAsB,MAAM,cAAc,OAAO,OAAO;GACtE,OAAO;GACP,eAAe,CACb,GAAG,MAAM,eACT,wBAAwB,KAAK,EAAE,SAAS,OAAO,QAAQ,CAAC,CAC1D;EACF;EACF,KAAK,sBACH,OAAO;GACL,GAAG;GACH,QAAQ;GACR,OAAO;GACP,cAAc,CAAC;GACf,cAAc,8BACZ,MAAM,cACN,kBAAkB,OAAO,QAAQ,CACnC;EACF;EACF,KAAK,cAAc;GACjB,MAAM,OAAO,eAAe,MAAM,cAAc,OAAO,SAAS;GAEhE,IAAI,KAAK,SAAS,YAChB,OAAO;GAGT,OAAO;IACL,GAAG;IACH,OAAO;IACP,cAAc,KAAK;IACnB,eAAe,CACb,GAAG,MAAM,eACT,YAAY,KAAK;KACf,oBAAoB,KAAK;KACzB,mBAAmB,KAAK;IAC1B,CAAC,CACH;GACF;EACF;EACA,KAAK,kBAAkB;GACrB,MAAM,OAAO,2BAA2B,MAAM,cAAc,OAAO,SAAS;GAE5E,IAAI,KAAK,SAAS,YAChB,OAAO;GAGT,OAAO;IACL,GAAG;IACH,QAAQ;IACR,OAAO;IACP,cAAc,CAAC;IACf,cAAc,KAAK;IACnB,eAAe,CACb,GAAG,MAAM,eACT,oBAAoB,KAAK;KACvB,eAAe,OAAO;KACtB,gBAAgB,KAAK,SAAS,KAAI,YAAW,QAAQ,EAAE;IACzD,CAAC,CACH;GACF;EACF;EACA,KAAK,mBAAmB;GACtB,MAAM,OAAO,oBAAoB,MAAM,cAAc,OAAO,WAAW,OAAO,OAAO;GAErF,IAAI,KAAK,SAAS,UAChB,OAAO;GAGT,OAAO;IACL,GAAG;IACH,QAAQ;IACR,OAAO;IACP,cAAc,CAAC;IACf,cAAc,KAAK;IACnB,eAAe,CACb,GAAG,MAAM,eACT,kBAAkB,KAAK;KACrB,WAAW,KAAK;KAChB,SAAS,OAAO;KAChB,gBAAgB,KAAK,SAAS,KAAI,YAAW,QAAQ,EAAE;IACzD,CAAC,CACH;GACF;EACF;EACA,KAAK;GACH,IAAI,aAAa,OAAO,OAAO,KAAK,GAClC,OAAO;GAGT,QAAQ,OAAO,MAAM,MAArB;IACE,KAAK,cACH,OAAO,cACL;KACE,GAAG;KACH,QAAQ;KACR,OAAO;KACP,cAAc,8BAA8B,MAAM,cAAc,OAAO,OAAO,MAAM;IACtF,GACA,OAAO,KACT;IACF,KAAK,cACH,OAAO,cACL;KACE,GAAG;KACH,QAAQ;KACR,OAAO,OAAO,MAAM;KACpB,cAAc,8BAA8B,MAAM,cAAc,OAAO,OAAO,MAAM;IACtF,GACA,OAAO,KACT;IACF,KAAK,YACH,OAAO,cACL;KACE,GAAG;KACH,QAAQ;KACR,OAAO;KACP,cAAc,8BAA8B,MAAM,cAAc,OAAO,OAAO,MAAM;IACtF,GACA,OAAO,KACT;IACF,KAAK,sBACH,OAAO,cACL;KACE,GAAG;KACH,QAAQ;KACR,OAAO;KACP,cAAc,8BAA8B,MAAM,cAAc,OAAO,OAAO,MAAM;IACtF,GACA,OAAO,KACT;IACF,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK,eACH,OAAO,cACL;KACE,GAAG;KACH,cAAc,8BAA8B,MAAM,cAAc,OAAO,OAAO,MAAM;IACtF,GACA,OAAO,KACT;GACJ;EAEF,KAAK,SACH,OAAO;GACL,GAAG;GACH,QAAQ;GACR,OAAO,OAAO;GACd,cAAc,cAAc,MAAM,cAAc,OAAO,OAAO;EAChE;EAEF,KAAK,SACH,OAAO;GAAE,GAAG;GAAO,QAAQ;GAAW,OAAO;EAAK;CACtD;AACF;AAEA,MAAa,4BAA4B,YACvC,QAAQ,SAAS,eACjB,QAAQ,MAAM,MAAK,SAAQ,KAAK,SAAS,eAAe,KAAK,KAAK,SAAS,CAAC;AAE9E,MAAa,gCAAgC,UAC3C,MAAM,aAAa,MAAK,YACtB,QAAQ,MAAM,MAAK,SAAQ,KAAK,SAAS,eAAe,KAAK,KAAK,SAAS,CAAC,CAC9E;AAEF,MAAa,wBAAwB,SACnC,KAAK,SAAS,cACd,KAAK,MAAM,SAAS,eACpB,KAAK,MAAM,SAAS,uBACpB,KAAK,MAAM,SAAS,sBACpB,KAAK,MAAM,SAAS,uBACpB,KAAK,MAAM,SAAS,aACpB,KAAK,MAAM,SAAS;AAGtB,MAAa,2BAA2B,SACtC,KAAK,SAAS,cAAc,KAAK,MAAM,SAAS;AAGlD,MAAa,0BAA0B,aACrC,SAAS,SAAQ,YACf,QAAQ,MAAM,QAAQ,SAAqC,qBAAqB,IAAI,CAAC,CACvF;AAEF,MAAa,6BAA6B,aACxC,SAAS,SAAQ,YACf,QAAQ,MAAM,QAAQ,SAAwC,wBAAwB,IAAI,CAAC,CAC7F;AAQF,MAAa,iCAAiC,EAC5C,eACA,qBACA,qBAEC,gBAAgB,IAAI,KAAK,uBAAuB,gBAAgB,IAAI"}
+{"version":3,"file":"chat-core.mjs","names":[],"sources":["../../src/react/chat-core.ts"],"sourcesContent":["import {\n  hitlResponseEvent,\n  type AgentError,\n  type AgentEvent,\n  type AgentMessage,\n  type AgentRetry,\n  type Content,\n  type HitlResponse,\n  type UserMessage\n} from '@yolk-sdk/agent/protocol'\nimport {\n  appendProtocolMessage,\n  applyAgentEventToChatMessages,\n  deleteChatTurn,\n  editChatUserMessage,\n  markChatError,\n  regenerateChatMessagesFrom,\n  type AgentChatMessage,\n  type AgentChatPart,\n  type ApplyAgentEventToChatMessagesOptions\n} from './chat-messages.ts'\nimport {\n  MessagesRegenerated,\n  ProtocolMessageAppended,\n  TurnDeleted,\n  UserMessageEdited,\n  UserMessageSubmitted,\n  type AgentChatSessionEvent\n} from './chat-session-events.ts'\n\nexport type AgentRunStatus = 'idle' | 'running' | 'waiting' | 'done' | 'error' | 'aborted'\n\nexport type AgentChatState = {\n  readonly status: AgentRunStatus\n  readonly error: string | null\n  readonly errorInfo: AgentError | null\n  readonly retryInfo: AgentRetry | null\n  readonly chatMessages: ReadonlyArray<AgentChatMessage>\n  readonly sessionEvents: ReadonlyArray<AgentChatSessionEvent>\n  readonly seenEventIds: ReadonlyArray<string>\n}\n\nexport const initialAgentChatState: AgentChatState = {\n  status: 'idle',\n  error: null,\n  errorInfo: null,\n  retryInfo: null,\n  chatMessages: [],\n  sessionEvents: [],\n  seenEventIds: []\n}\n\nconst hasSeenEvent = (state: AgentChatState, event: AgentEvent) =>\n  event.eventId !== undefined && state.seenEventIds.includes(event.eventId)\n\nconst rememberEvent = (state: AgentChatState, event: AgentEvent): AgentChatState =>\n  event.eventId === undefined\n    ? state\n    : { ...state, seenEventIds: [...state.seenEventIds, event.eventId] }\n\nconst clearRetryInfo = (state: AgentChatState): AgentChatState =>\n  state.retryInfo === null ? state : { ...state, retryInfo: null }\n\nexport type AgentChatAction =\n  | { readonly _tag: 'HydrateMessage'; readonly message: AgentMessage }\n  | { readonly _tag: 'Submit'; readonly message: UserMessage }\n  | { readonly _tag: 'AppendMessage'; readonly message: AgentMessage }\n  | { readonly _tag: 'SubmitHitlResponse'; readonly response: HitlResponse }\n  | { readonly _tag: 'DeleteTurn'; readonly messageId: string }\n  | { readonly _tag: 'RegenerateFrom'; readonly messageId: string }\n  | { readonly _tag: 'EditUserMessage'; readonly messageId: string; readonly content: Content }\n  | ({ readonly _tag: 'Event'; readonly event: AgentEvent } & ApplyAgentEventToChatMessagesOptions)\n  | { readonly _tag: 'Error'; readonly message: string }\n  | { readonly _tag: 'Abort' }\n\nexport const reduceAgentChatState = (\n  state: AgentChatState,\n  action: AgentChatAction\n): AgentChatState => {\n  switch (action._tag) {\n    case 'HydrateMessage':\n      return {\n        ...state,\n        chatMessages: appendProtocolMessage(state.chatMessages, action.message),\n        error: null,\n        errorInfo: null,\n        retryInfo: null\n      }\n    case 'Submit':\n      return {\n        ...state,\n        status: 'running',\n        error: null,\n        errorInfo: null,\n        retryInfo: null,\n        seenEventIds: [],\n        chatMessages: appendProtocolMessage(state.chatMessages, action.message),\n        sessionEvents: [\n          ...state.sessionEvents,\n          UserMessageSubmitted.make({ message: action.message })\n        ]\n      }\n    case 'AppendMessage':\n      return {\n        ...state,\n        chatMessages: appendProtocolMessage(state.chatMessages, action.message),\n        error: null,\n        errorInfo: null,\n        retryInfo: null,\n        sessionEvents: [\n          ...state.sessionEvents,\n          ProtocolMessageAppended.make({ message: action.message })\n        ]\n      }\n    case 'SubmitHitlResponse':\n      return {\n        ...state,\n        status: 'running',\n        error: null,\n        errorInfo: null,\n        retryInfo: null,\n        seenEventIds: [],\n        chatMessages: applyAgentEventToChatMessages(\n          state.chatMessages,\n          hitlResponseEvent(action.response)\n        )\n      }\n    case 'DeleteTurn': {\n      const next = deleteChatTurn(state.chatMessages, action.messageId)\n\n      if (next._tag === 'NotFound') {\n        return state\n      }\n\n      return {\n        ...state,\n        error: null,\n        errorInfo: null,\n        retryInfo: null,\n        chatMessages: next.messages,\n        sessionEvents: [\n          ...state.sessionEvents,\n          TurnDeleted.make({\n            turnStartMessageId: next.turnStartMessageId,\n            deletedMessageIds: next.deletedMessageIds\n          })\n        ]\n      }\n    }\n    case 'RegenerateFrom': {\n      const next = regenerateChatMessagesFrom(state.chatMessages, action.messageId)\n\n      if (next._tag === 'NotFound') {\n        return state\n      }\n\n      return {\n        ...state,\n        status: 'running',\n        error: null,\n        errorInfo: null,\n        retryInfo: null,\n        seenEventIds: [],\n        chatMessages: next.messages,\n        sessionEvents: [\n          ...state.sessionEvents,\n          MessagesRegenerated.make({\n            fromMessageId: action.messageId,\n            keptMessageIds: next.messages.map(message => message.id)\n          })\n        ]\n      }\n    }\n    case 'EditUserMessage': {\n      const next = editChatUserMessage(state.chatMessages, action.messageId, action.content)\n\n      if (next._tag !== 'Edited') {\n        return state\n      }\n\n      return {\n        ...state,\n        status: 'running',\n        error: null,\n        errorInfo: null,\n        retryInfo: null,\n        seenEventIds: [],\n        chatMessages: next.messages,\n        sessionEvents: [\n          ...state.sessionEvents,\n          UserMessageEdited.make({\n            messageId: next.messageId,\n            content: action.content,\n            keptMessageIds: next.messages.map(message => message.id)\n          })\n        ]\n      }\n    }\n    case 'Event': {\n      if (hasSeenEvent(state, action.event)) {\n        return state\n      }\n\n      switch (action.event._tag) {\n        case 'AgentStart':\n          return rememberEvent(\n            {\n              ...state,\n              status: 'running',\n              error: null,\n              errorInfo: null,\n              retryInfo: null,\n              chatMessages: applyAgentEventToChatMessages(state.chatMessages, action.event, action)\n            },\n            action.event\n          )\n        case 'AgentError':\n          return rememberEvent(\n            {\n              ...state,\n              status: 'error',\n              error: action.event.message,\n              errorInfo: action.event,\n              retryInfo: null,\n              chatMessages: applyAgentEventToChatMessages(state.chatMessages, action.event, action)\n            },\n            action.event\n          )\n        case 'AgentEnd':\n          return rememberEvent(\n            {\n              ...state,\n              status: 'done',\n              error: null,\n              errorInfo: null,\n              retryInfo: null,\n              chatMessages: applyAgentEventToChatMessages(state.chatMessages, action.event, action)\n            },\n            action.event\n          )\n        case 'AgentAwaitingInput':\n          return rememberEvent(\n            {\n              ...state,\n              status: 'waiting',\n              error: null,\n              errorInfo: null,\n              retryInfo: null,\n              chatMessages: applyAgentEventToChatMessages(state.chatMessages, action.event, action)\n            },\n            action.event\n          )\n        case 'AgentRetry':\n          return rememberEvent(\n            {\n              ...state,\n              retryInfo: action.event,\n              chatMessages: applyAgentEventToChatMessages(state.chatMessages, action.event, action)\n            },\n            action.event\n          )\n        case 'CompactionEnd':\n        case 'AssistantMessage':\n        case 'CompactionStart':\n        case 'LLMReasoningDelta':\n        case 'LLMStreamEnd':\n        case 'LLMStreamStart':\n        case 'LLMTextDelta':\n        case 'ProviderToolResult':\n        case 'QuestionAnswered':\n        case 'QuestionCancelled':\n        case 'QuestionRequested':\n        case 'SubagentCompleted':\n        case 'SubagentStarted':\n        case 'ToolApprovalDenied':\n        case 'ToolApprovalGranted':\n        case 'ToolApprovalRequested':\n        case 'ToolExecutionCompleted':\n        case 'ToolExecutionError':\n        case 'ToolExecutionStarted':\n        case 'ToolInputDelta':\n        case 'ToolInputEnd':\n        case 'ToolInputStart':\n        case 'TurnEnd':\n        case 'TurnStart':\n        case 'UsageUpdate':\n          return rememberEvent(\n            clearRetryInfo({\n              ...state,\n              chatMessages: applyAgentEventToChatMessages(state.chatMessages, action.event, action)\n            }),\n            action.event\n          )\n      }\n    }\n    case 'Error': {\n      return {\n        ...state,\n        status: 'error',\n        error: action.message,\n        errorInfo: null,\n        retryInfo: null,\n        chatMessages: markChatError(state.chatMessages, action.message)\n      }\n    }\n    case 'Abort':\n      return { ...state, status: 'aborted', error: null, errorInfo: null, retryInfo: null }\n  }\n}\n\nexport const hasAgentMessageReasoning = (message: AgentMessage) =>\n  message._tag === 'Assistant' &&\n  message.parts.some(part => part._tag === 'Reasoning' && part.text.length > 0)\n\nexport const hasAgentChatReasoningSummary = (state: AgentChatState) =>\n  state.chatMessages.some(message =>\n    message.parts.some(part => part._tag === 'Reasoning' && part.text.length > 0)\n  )\n\nexport const isActiveChatToolPart = (part: AgentChatPart) =>\n  part._tag === 'ToolCall' &&\n  part.state._tag !== 'Completed' &&\n  part.state._tag !== 'ProviderCompleted' &&\n  part.state._tag !== 'QuestionAnswered' &&\n  part.state._tag !== 'QuestionCancelled' &&\n  part.state._tag !== 'Errored' &&\n  part.state._tag !== 'Denied'\nexport type ActiveChatToolPart = Extract<AgentChatPart, { readonly _tag: 'ToolCall' }>\n\nexport const isCompletedChatToolPart = (part: AgentChatPart) =>\n  part._tag === 'ToolCall' && part.state._tag === 'Completed'\nexport type CompletedChatToolPart = Extract<AgentChatPart, { readonly _tag: 'ToolCall' }>\n\nexport const getActiveChatToolParts = (messages: ReadonlyArray<AgentChatMessage>) =>\n  messages.flatMap(message =>\n    message.parts.filter((part): part is ActiveChatToolPart => isActiveChatToolPart(part))\n  )\n\nexport const getCompletedChatToolParts = (messages: ReadonlyArray<AgentChatMessage>) =>\n  messages.flatMap(message =>\n    message.parts.filter((part): part is CompletedChatToolPart => isCompletedChatToolPart(part))\n  )\n\ntype AgentChatLiveActivityInput = {\n  readonly isTextRunning: boolean\n  readonly activeToolCallCount: number\n  readonly isVoiceActive: boolean\n}\n\nexport const getAgentChatLiveActivityCount = ({\n  isTextRunning,\n  activeToolCallCount,\n  isVoiceActive\n}: AgentChatLiveActivityInput) =>\n  (isTextRunning ? 1 : 0) + activeToolCallCount + (isVoiceActive ? 1 : 0)\n"],"mappings":";;;;AA0CA,MAAa,wBAAwC;CACnD,QAAQ;CACR,OAAO;CACP,WAAW;CACX,WAAW;CACX,cAAc,CAAC;CACf,eAAe,CAAC;CAChB,cAAc,CAAC;AACjB;AAEA,MAAM,gBAAgB,OAAuB,UAC3C,MAAM,YAAY,KAAA,KAAa,MAAM,aAAa,SAAS,MAAM,OAAO;AAE1E,MAAM,iBAAiB,OAAuB,UAC5C,MAAM,YAAY,KAAA,IACd,QACA;CAAE,GAAG;CAAO,cAAc,CAAC,GAAG,MAAM,cAAc,MAAM,OAAO;AAAE;AAEvE,MAAM,kBAAkB,UACtB,MAAM,cAAc,OAAO,QAAQ;CAAE,GAAG;CAAO,WAAW;AAAK;AAcjE,MAAa,wBACX,OACA,WACmB;CACnB,QAAQ,OAAO,MAAf;EACE,KAAK,kBACH,OAAO;GACL,GAAG;GACH,cAAc,sBAAsB,MAAM,cAAc,OAAO,OAAO;GACtE,OAAO;GACP,WAAW;GACX,WAAW;EACb;EACF,KAAK,UACH,OAAO;GACL,GAAG;GACH,QAAQ;GACR,OAAO;GACP,WAAW;GACX,WAAW;GACX,cAAc,CAAC;GACf,cAAc,sBAAsB,MAAM,cAAc,OAAO,OAAO;GACtE,eAAe,CACb,GAAG,MAAM,eACT,qBAAqB,KAAK,EAAE,SAAS,OAAO,QAAQ,CAAC,CACvD;EACF;EACF,KAAK,iBACH,OAAO;GACL,GAAG;GACH,cAAc,sBAAsB,MAAM,cAAc,OAAO,OAAO;GACtE,OAAO;GACP,WAAW;GACX,WAAW;GACX,eAAe,CACb,GAAG,MAAM,eACT,wBAAwB,KAAK,EAAE,SAAS,OAAO,QAAQ,CAAC,CAC1D;EACF;EACF,KAAK,sBACH,OAAO;GACL,GAAG;GACH,QAAQ;GACR,OAAO;GACP,WAAW;GACX,WAAW;GACX,cAAc,CAAC;GACf,cAAc,8BACZ,MAAM,cACN,kBAAkB,OAAO,QAAQ,CACnC;EACF;EACF,KAAK,cAAc;GACjB,MAAM,OAAO,eAAe,MAAM,cAAc,OAAO,SAAS;GAEhE,IAAI,KAAK,SAAS,YAChB,OAAO;GAGT,OAAO;IACL,GAAG;IACH,OAAO;IACP,WAAW;IACX,WAAW;IACX,cAAc,KAAK;IACnB,eAAe,CACb,GAAG,MAAM,eACT,YAAY,KAAK;KACf,oBAAoB,KAAK;KACzB,mBAAmB,KAAK;IAC1B,CAAC,CACH;GACF;EACF;EACA,KAAK,kBAAkB;GACrB,MAAM,OAAO,2BAA2B,MAAM,cAAc,OAAO,SAAS;GAE5E,IAAI,KAAK,SAAS,YAChB,OAAO;GAGT,OAAO;IACL,GAAG;IACH,QAAQ;IACR,OAAO;IACP,WAAW;IACX,WAAW;IACX,cAAc,CAAC;IACf,cAAc,KAAK;IACnB,eAAe,CACb,GAAG,MAAM,eACT,oBAAoB,KAAK;KACvB,eAAe,OAAO;KACtB,gBAAgB,KAAK,SAAS,KAAI,YAAW,QAAQ,EAAE;IACzD,CAAC,CACH;GACF;EACF;EACA,KAAK,mBAAmB;GACtB,MAAM,OAAO,oBAAoB,MAAM,cAAc,OAAO,WAAW,OAAO,OAAO;GAErF,IAAI,KAAK,SAAS,UAChB,OAAO;GAGT,OAAO;IACL,GAAG;IACH,QAAQ;IACR,OAAO;IACP,WAAW;IACX,WAAW;IACX,cAAc,CAAC;IACf,cAAc,KAAK;IACnB,eAAe,CACb,GAAG,MAAM,eACT,kBAAkB,KAAK;KACrB,WAAW,KAAK;KAChB,SAAS,OAAO;KAChB,gBAAgB,KAAK,SAAS,KAAI,YAAW,QAAQ,EAAE;IACzD,CAAC,CACH;GACF;EACF;EACA,KAAK;GACH,IAAI,aAAa,OAAO,OAAO,KAAK,GAClC,OAAO;GAGT,QAAQ,OAAO,MAAM,MAArB;IACE,KAAK,cACH,OAAO,cACL;KACE,GAAG;KACH,QAAQ;KACR,OAAO;KACP,WAAW;KACX,WAAW;KACX,cAAc,8BAA8B,MAAM,cAAc,OAAO,OAAO,MAAM;IACtF,GACA,OAAO,KACT;IACF,KAAK,cACH,OAAO,cACL;KACE,GAAG;KACH,QAAQ;KACR,OAAO,OAAO,MAAM;KACpB,WAAW,OAAO;KAClB,WAAW;KACX,cAAc,8BAA8B,MAAM,cAAc,OAAO,OAAO,MAAM;IACtF,GACA,OAAO,KACT;IACF,KAAK,YACH,OAAO,cACL;KACE,GAAG;KACH,QAAQ;KACR,OAAO;KACP,WAAW;KACX,WAAW;KACX,cAAc,8BAA8B,MAAM,cAAc,OAAO,OAAO,MAAM;IACtF,GACA,OAAO,KACT;IACF,KAAK,sBACH,OAAO,cACL;KACE,GAAG;KACH,QAAQ;KACR,OAAO;KACP,WAAW;KACX,WAAW;KACX,cAAc,8BAA8B,MAAM,cAAc,OAAO,OAAO,MAAM;IACtF,GACA,OAAO,KACT;IACF,KAAK,cACH,OAAO,cACL;KACE,GAAG;KACH,WAAW,OAAO;KAClB,cAAc,8BAA8B,MAAM,cAAc,OAAO,OAAO,MAAM;IACtF,GACA,OAAO,KACT;IACF,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK;IACL,KAAK,eACH,OAAO,cACL,eAAe;KACb,GAAG;KACH,cAAc,8BAA8B,MAAM,cAAc,OAAO,OAAO,MAAM;IACtF,CAAC,GACD,OAAO,KACT;GACJ;EAEF,KAAK,SACH,OAAO;GACL,GAAG;GACH,QAAQ;GACR,OAAO,OAAO;GACd,WAAW;GACX,WAAW;GACX,cAAc,cAAc,MAAM,cAAc,OAAO,OAAO;EAChE;EAEF,KAAK,SACH,OAAO;GAAE,GAAG;GAAO,QAAQ;GAAW,OAAO;GAAM,WAAW;GAAM,WAAW;EAAK;CACxF;AACF;AAEA,MAAa,4BAA4B,YACvC,QAAQ,SAAS,eACjB,QAAQ,MAAM,MAAK,SAAQ,KAAK,SAAS,eAAe,KAAK,KAAK,SAAS,CAAC;AAE9E,MAAa,gCAAgC,UAC3C,MAAM,aAAa,MAAK,YACtB,QAAQ,MAAM,MAAK,SAAQ,KAAK,SAAS,eAAe,KAAK,KAAK,SAAS,CAAC,CAC9E;AAEF,MAAa,wBAAwB,SACnC,KAAK,SAAS,cACd,KAAK,MAAM,SAAS,eACpB,KAAK,MAAM,SAAS,uBACpB,KAAK,MAAM,SAAS,sBACpB,KAAK,MAAM,SAAS,uBACpB,KAAK,MAAM,SAAS,aACpB,KAAK,MAAM,SAAS;AAGtB,MAAa,2BAA2B,SACtC,KAAK,SAAS,cAAc,KAAK,MAAM,SAAS;AAGlD,MAAa,0BAA0B,aACrC,SAAS,SAAQ,YACf,QAAQ,MAAM,QAAQ,SAAqC,qBAAqB,IAAI,CAAC,CACvF;AAEF,MAAa,6BAA6B,aACxC,SAAS,SAAQ,YACf,QAAQ,MAAM,QAAQ,SAAwC,wBAAwB,IAAI,CAAC,CAC7F;AAQF,MAAa,iCAAiC,EAC5C,eACA,qBACA,qBAEC,gBAAgB,IAAI,KAAK,uBAAuB,gBAAgB,IAAI"}

package/dist/react/chat-items.d.mts

@@ -1,5 +1,5 @@
 import { AgentChatMessage } from "./chat-messages.mjs";
-import { Content, QuestionRequest, QuestionResponse, ToolApprovalRequest, ToolCall, ToolResult } from "@yolk-sdk/agent/protocol";
+import { AgentRetry, Content, QuestionRequest, QuestionResponse, ToolApprovalRequest, ToolCall, ToolResult } from "@yolk-sdk/agent/protocol";
 import { Option } from "effect";
 
 //#region src/react/chat-items.d.ts
@@ -104,6 +104,10 @@ type AgentChatItem = {
   readonly _tag: 'AssistantDraft';
   readonly id: string;
   readonly text: string;
+} | {
+  readonly _tag: 'Retry';
+  readonly id: string;
+  readonly retry: AgentRetry;
 } | {
   readonly _tag: 'AssistantStatus';
   readonly id: string;
@@ -117,12 +121,14 @@ type BuildAgentChatItemsInput = {
   readonly messages: ReadonlyArray<AgentChatMessage>;
   readonly isRunning: boolean;
   readonly activeToolLabel: Option.Option<string>;
+  readonly retryInfo?: AgentRetry | null;
 };
 declare const dedupeAgentChatToolRunItems: (items: ReadonlyArray<AgentChatItem>) => ReadonlyArray<AgentChatItem>;
 declare const buildAgentChatItems: ({
   messages,
   isRunning,
-  activeToolLabel
+  activeToolLabel,
+  retryInfo
 }: BuildAgentChatItemsInput) => ReadonlyArray<AgentChatItem>;
 //#endregion
 export { AgentChatItem, BuildAgentChatItemsInput, ToolDuration, ToolRunState, ToolRunTiming, buildAgentChatItems, dedupeAgentChatToolRunItems };

package/dist/react/chat-items.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"chat-items.d.mts","names":[],"sources":["../../src/react/chat-items.ts"],"mappings":";;;;;KAYY,YAAA;EAAA,SACG,IAAA;EAAA,SAAwB,YAAA;AAAA;EAAA,SACxB,IAAA;AAAA;AAAA,KAEV,eAAA;EAAA,SACM,QAAA;IAAA,SAAqB,IAAA;EAAA;EAAA,SACrB,WAAA;EAAA,SACA,SAAA;AAAA;AAAA,KAGN,oBAAA;EAAA,SACM,QAAA;IAAA,SAAqB,IAAA;EAAA;EAAA,SACrB,WAAA;EAAA,SACA,SAAA;AAAA;AAAA,KAGN,kBAAA;EAAA,SACM,QAAA;IAAA,SAAqB,IAAA;IAAA,SAAwB,YAAA;EAAA;EAAA,SAC7C,WAAA;EAAA,SACA,SAAA;AAAA;AAAA,KAGC,aAAA,GAAgB,eAAA,GAAkB,oBAAA,GAAuB,kBAAA;AAAA,KAEhE,qBAAA,GAAwB,aAAa;AAAA,KAE9B,YAAA;EAAA,SACI,IAAA;AAAA,IAAoB,oBAAA;EAAA,SACpB,IAAA;AAAA,IAAmB,eAAA;EAAA,SACnB,IAAA;EAAA,SAAiC,KAAA;AAAA,IAAkB,eAAA;EAAA,SAEpD,IAAA;EAAA,SACA,OAAA,GAAU,mBAAA;AAAA,IACjB,eAAA;EAAA,SACQ,IAAA;EAAA,SAAyB,MAAA;AAAA,IAAmB,eAAA;EAAA,SAC5C,IAAA;EAAA,SAAoC,OAAA,EAAS,eAAA;AAAA,IAAoB,eAAA;EAAA,SAElE,IAAA;EAAA,SACA,QAAA,EAAU,gBAAA;EAAA,SACV,OAAA,GAAU,eAAA;AAAA,IACjB,eAAA;EAAA,SAEO,IAAA;EAAA,SACA,QAAA,EAAU,gBAAA;EAAA,SACV,OAAA,GAAU,eAAA;AAAA,IACjB,eAAA;EAAA,SACQ,IAAA;EAAA,SAA4B,MAAA,EAAQ,UAAA;AAAA,IAAe,qBAAA;EAAA,SACnD,IAAA;EAAA,SAA0B,OAAA;AAAA,IAAoB,qBAAA;EAAA,SAE/C,IAAA;EAAA,SACA,MAAA,EAAQ,UAAA;AAAA,IACf,eAAA;AAAA,KAEI,aAAA;EAAA,SAEG,IAAA;EAAA,SACA,EAAA;EAAA,SACA,SAAA;EAAA,SACA,OAAA,EAAS,OAAA;AAAA;EAAA,SAGT,IAAA;EAAA,SACA,EAAA;EAAA,SACA,SAAA;EAAA,SACA,OAAA,EAAS,OAAA;AAAA;EAAA,SAGT,IAAA;EAAA,SACA,EAAA;EAAA,SACA,SAAA;EAAA,SACA,IAAA;AAAA;EAAA,SAGA,IAAA;EAAA,SACA,EAAA;EAAA,SACA,SAAA;EAAA,SACA,IAAA,EAAM,QAAA;EAAA,SACN,KAAA,EAAO,YAAA;AAAA;EAAA,SAGP,IAAA;EAAA,SACA,EAAA;EAAA,SACA,SAAA;EAAA,SACA,UAAA;EAAA,SACA,IAAA;EAAA,SACA,OAAA,EAAS,OAAA;EAAA,SACT,OAAA;AAAA;EAAA,SAEA,IAAA;EAAA,SAA4B,EAAA;EAAA,SAAqB,IAAA;AAAA;EAAA,SACjD,IAAA;EAAA,SAAiC,EAAA;EAAA,SAAqB,IAAA;AAAA;EAAA,SACtD,IAAA;EAAA,SAAkC,EAAA;EAAA,SAAqB,KAAA;AAAA;EAAA,SACvD,IAAA;EAAA,SAAwB,EAAA;EAAA,SAAqB,OAAA;AAAA;AAAA,KAEhD,wBAAA;EAAA,SACD,QAAA,EAAU,aAAA,CAAc,gBAAA;EAAA,SACxB,SAAA;EAAA,SACA,eAAA,EAAiB,MAAA,CAAO,MAAA;AAAA;AAAA,cAGtB,2BAAA,GACX,KAAA,EAAO,aAAA,CAAc,aAAA,MACpB,aAAA,CAAc,aAAA;AAAA,cAuMJ,mBAAA;EAAuB,QAAA;EAAA,SAAA;EAAA;AAAA,GAIjC,wBAAA,KAA2B,aAAA,CAAc,aAAA"}
+{"version":3,"file":"chat-items.d.mts","names":[],"sources":["../../src/react/chat-items.ts"],"mappings":";;;;;KAaY,YAAA;EAAA,SACG,IAAA;EAAA,SAAwB,YAAA;AAAA;EAAA,SACxB,IAAA;AAAA;AAAA,KAEV,eAAA;EAAA,SACM,QAAA;IAAA,SAAqB,IAAA;EAAA;EAAA,SACrB,WAAA;EAAA,SACA,SAAA;AAAA;AAAA,KAGN,oBAAA;EAAA,SACM,QAAA;IAAA,SAAqB,IAAA;EAAA;EAAA,SACrB,WAAA;EAAA,SACA,SAAA;AAAA;AAAA,KAGN,kBAAA;EAAA,SACM,QAAA;IAAA,SAAqB,IAAA;IAAA,SAAwB,YAAA;EAAA;EAAA,SAC7C,WAAA;EAAA,SACA,SAAA;AAAA;AAAA,KAGC,aAAA,GAAgB,eAAA,GAAkB,oBAAA,GAAuB,kBAAA;AAAA,KAEhE,qBAAA,GAAwB,aAAa;AAAA,KAE9B,YAAA;EAAA,SACI,IAAA;AAAA,IAAoB,oBAAA;EAAA,SACpB,IAAA;AAAA,IAAmB,eAAA;EAAA,SACnB,IAAA;EAAA,SAAiC,KAAA;AAAA,IAAkB,eAAA;EAAA,SAEpD,IAAA;EAAA,SACA,OAAA,GAAU,mBAAA;AAAA,IACjB,eAAA;EAAA,SACQ,IAAA;EAAA,SAAyB,MAAA;AAAA,IAAmB,eAAA;EAAA,SAC5C,IAAA;EAAA,SAAoC,OAAA,EAAS,eAAA;AAAA,IAAoB,eAAA;EAAA,SAElE,IAAA;EAAA,SACA,QAAA,EAAU,gBAAA;EAAA,SACV,OAAA,GAAU,eAAA;AAAA,IACjB,eAAA;EAAA,SAEO,IAAA;EAAA,SACA,QAAA,EAAU,gBAAA;EAAA,SACV,OAAA,GAAU,eAAA;AAAA,IACjB,eAAA;EAAA,SACQ,IAAA;EAAA,SAA4B,MAAA,EAAQ,UAAA;AAAA,IAAe,qBAAA;EAAA,SACnD,IAAA;EAAA,SAA0B,OAAA;AAAA,IAAoB,qBAAA;EAAA,SAE/C,IAAA;EAAA,SACA,MAAA,EAAQ,UAAA;AAAA,IACf,eAAA;AAAA,KAEI,aAAA;EAAA,SAEG,IAAA;EAAA,SACA,EAAA;EAAA,SACA,SAAA;EAAA,SACA,OAAA,EAAS,OAAA;AAAA;EAAA,SAGT,IAAA;EAAA,SACA,EAAA;EAAA,SACA,SAAA;EAAA,SACA,OAAA,EAAS,OAAA;AAAA;EAAA,SAGT,IAAA;EAAA,SACA,EAAA;EAAA,SACA,SAAA;EAAA,SACA,IAAA;AAAA;EAAA,SAGA,IAAA;EAAA,SACA,EAAA;EAAA,SACA,SAAA;EAAA,SACA,IAAA,EAAM,QAAA;EAAA,SACN,KAAA,EAAO,YAAA;AAAA;EAAA,SAGP,IAAA;EAAA,SACA,EAAA;EAAA,SACA,SAAA;EAAA,SACA,UAAA;EAAA,SACA,IAAA;EAAA,SACA,OAAA,EAAS,OAAA;EAAA,SACT,OAAA;AAAA;EAAA,SAEA,IAAA;EAAA,SAA4B,EAAA;EAAA,SAAqB,IAAA;AAAA;EAAA,SACjD,IAAA;EAAA,SAAiC,EAAA;EAAA,SAAqB,IAAA;AAAA;EAAA,SACtD,IAAA;EAAA,SAAwB,EAAA;EAAA,SAAqB,KAAA,EAAO,UAAA;AAAA;EAAA,SACpD,IAAA;EAAA,SAAkC,EAAA;EAAA,SAAqB,KAAA;AAAA;EAAA,SACvD,IAAA;EAAA,SAAwB,EAAA;EAAA,SAAqB,OAAA;AAAA;AAAA,KAEhD,wBAAA;EAAA,SACD,QAAA,EAAU,aAAA,CAAc,gBAAA;EAAA,SACxB,SAAA;EAAA,SACA,eAAA,EAAiB,MAAA,CAAO,MAAA;EAAA,SACxB,SAAA,GAAY,UAAA;AAAA;AAAA,cAGV,2BAAA,GACX,KAAA,EAAO,aAAA,CAAc,aAAA,MACpB,aAAA,CAAc,aAAA;AAAA,cAsMJ,mBAAA;EAAuB,QAAA;EAAA,SAAA;EAAA,eAAA;EAAA;AAAA,GAKjC,wBAAA,KAA2B,aAAA,CAAc,aAAA"}

package/dist/react/chat-items.mjs

@@ -153,16 +153,23 @@ const itemFromPart = (message, part) => {
 		});
 	}
 };
-const buildAgentChatItems = ({ messages, isRunning, activeToolLabel }) => {
+const buildAgentChatItems = ({ messages, isRunning, activeToolLabel, retryInfo }) => {
 	const items = dedupeAgentChatToolRunItems(Array.getSomes(Array.flatMap(messages, (message) => Array.map(message.parts, (part) => itemFromPart(message, part)))));
-	if (isRunning) return [...items, {
-		_tag: "AssistantStatus",
-		id: "assistant-status",
-		label: activeStatusLabel({
-			messages,
-			activeToolLabel
-		})
-	}];
+	if (isRunning) {
+		if (retryInfo !== void 0 && retryInfo !== null) return [...items, {
+			_tag: "Retry",
+			id: "agent-retry",
+			retry: retryInfo
+		}];
+		return [...items, {
+			_tag: "AssistantStatus",
+			id: "assistant-status",
+			label: activeStatusLabel({
+				messages,
+				activeToolLabel
+			})
+		}];
+	}
 	return items;
 };
 //#endregion

package/dist/react/chat-items.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"chat-items.mjs","names":["Arr"],"sources":["../../src/react/chat-items.ts"],"sourcesContent":["import { Array as Arr, Option } from 'effect'\nimport {\n  contentText,\n  type QuestionRequest,\n  type QuestionResponse,\n  type ToolApprovalRequest,\n  type ToolResult,\n  type Content,\n  type ToolCall\n} from '@yolk-sdk/agent/protocol'\nimport type { AgentChatMessage, ChatToolState } from './chat-messages.ts'\n\nexport type ToolDuration =\n  | { readonly _tag: 'Known'; readonly milliseconds: number }\n  | { readonly _tag: 'Unknown' }\n\ntype ToolRunNoTiming = {\n  readonly duration: { readonly _tag: 'Unknown' }\n  readonly startedAtMs?: undefined\n  readonly endedAtMs?: undefined\n}\n\ntype ToolRunStartedTiming = {\n  readonly duration: { readonly _tag: 'Unknown' }\n  readonly startedAtMs: number\n  readonly endedAtMs?: undefined\n}\n\ntype ToolRunKnownTiming = {\n  readonly duration: { readonly _tag: 'Known'; readonly milliseconds: number }\n  readonly startedAtMs: number\n  readonly endedAtMs: number\n}\n\nexport type ToolRunTiming = ToolRunNoTiming | ToolRunStartedTiming | ToolRunKnownTiming\n\ntype ToolRunTerminalTiming = ToolRunTiming\n\nexport type ToolRunState =\n  | ({ readonly _tag: 'Running' } & ToolRunStartedTiming)\n  | ({ readonly _tag: 'Called' } & ToolRunNoTiming)\n  | ({ readonly _tag: 'InputStreaming'; readonly input: string } & ToolRunNoTiming)\n  | ({\n      readonly _tag: 'ApprovalRequested'\n      readonly request?: ToolApprovalRequest\n    } & ToolRunNoTiming)\n  | ({ readonly _tag: 'Denied'; readonly reason: string } & ToolRunNoTiming)\n  | ({ readonly _tag: 'QuestionRequested'; readonly request: QuestionRequest } & ToolRunNoTiming)\n  | ({\n      readonly _tag: 'QuestionAnswered'\n      readonly response: QuestionResponse\n      readonly request?: QuestionRequest\n    } & ToolRunNoTiming)\n  | ({\n      readonly _tag: 'QuestionCancelled'\n      readonly response: QuestionResponse\n      readonly request?: QuestionRequest\n    } & ToolRunNoTiming)\n  | ({ readonly _tag: 'Completed'; readonly result: ToolResult } & ToolRunTerminalTiming)\n  | ({ readonly _tag: 'Errored'; readonly message: string } & ToolRunTerminalTiming)\n  | ({\n      readonly _tag: 'ProviderCompleted'\n      readonly result: ToolResult\n    } & ToolRunNoTiming)\n\nexport type AgentChatItem =\n  | {\n      readonly _tag: 'UserMessage'\n      readonly id: string\n      readonly messageId: string\n      readonly content: Content\n    }\n  | {\n      readonly _tag: 'AssistantMessage'\n      readonly id: string\n      readonly messageId: string\n      readonly content: Content\n    }\n  | {\n      readonly _tag: 'Reasoning'\n      readonly id: string\n      readonly messageId: string\n      readonly text: string\n    }\n  | {\n      readonly _tag: 'ToolRun'\n      readonly id: string\n      readonly messageId: string\n      readonly call: ToolCall\n      readonly state: ToolRunState\n    }\n  | {\n      readonly _tag: 'ToolResult'\n      readonly id: string\n      readonly messageId: string\n      readonly toolCallId: string\n      readonly name: string\n      readonly content: Content\n      readonly isError?: boolean\n    }\n  | { readonly _tag: 'UserDraft'; readonly id: string; readonly text: string }\n  | { readonly _tag: 'AssistantDraft'; readonly id: string; readonly text: string }\n  | { readonly _tag: 'AssistantStatus'; readonly id: string; readonly label: string }\n  | { readonly _tag: 'Error'; readonly id: string; readonly message: string }\n\nexport type BuildAgentChatItemsInput = {\n  readonly messages: ReadonlyArray<AgentChatMessage>\n  readonly isRunning: boolean\n  readonly activeToolLabel: Option.Option<string>\n}\n\nexport const dedupeAgentChatToolRunItems = (\n  items: ReadonlyArray<AgentChatItem>\n): ReadonlyArray<AgentChatItem> => {\n  // Tool call ids are transcript-global; latest wins when live and persisted projections overlap.\n  const latestIndexByToolCallId = new Map<string, number>()\n\n  for (const [index, item] of items.entries()) {\n    if (item._tag === 'ToolRun') {\n      latestIndexByToolCallId.set(item.call.id, index)\n    }\n  }\n\n  return items.filter(\n    (item, index) =>\n      item._tag !== 'ToolRun' || latestIndexByToolCallId.get(item.call.id) === index\n  )\n}\n\nconst activeStatusLabel = ({\n  messages,\n  activeToolLabel\n}: Pick<BuildAgentChatItemsInput, 'messages' | 'activeToolLabel'>) => {\n  if (Option.isSome(activeToolLabel)) {\n    return activeToolLabel.value\n  }\n\n  const parts = messages.flatMap(message =>\n    message.parts.map(part => ({ messageRole: message.role, part }))\n  )\n\n  if (\n    parts.some(\n      ({ messageRole, part }) =>\n        part._tag === 'Text' && part.state === 'streaming' && messageRole === 'assistant'\n    )\n  ) {\n    return 'Responding'\n  }\n\n  if (parts.some(({ part }) => part._tag === 'Reasoning' && part.state === 'streaming')) {\n    return 'Thinking'\n  }\n\n  return 'Thinking'\n}\n\nconst noTiming = (): ToolRunNoTiming => ({ duration: { _tag: 'Unknown' } })\n\nconst startedTiming = (startedAtMs: number): ToolRunStartedTiming => ({\n  duration: { _tag: 'Unknown' },\n  startedAtMs\n})\n\nconst unknownTerminalTiming = (startedAtMs?: number): ToolRunNoTiming | ToolRunStartedTiming =>\n  startedAtMs === undefined ? noTiming() : startedTiming(startedAtMs)\n\nconst knownTiming = (startedAtMs: number, endedAtMs: number): ToolRunTiming => ({\n  duration: { _tag: 'Known', milliseconds: Math.max(0, endedAtMs - startedAtMs) },\n  startedAtMs,\n  endedAtMs\n})\n\nconst terminalTimingFromState = (\n  state: Extract<ChatToolState, { readonly _tag: 'Completed' | 'Errored' }>\n): ToolRunTerminalTiming => {\n  if (state.startedAtMs !== undefined && state.endedAtMs !== undefined) {\n    return knownTiming(state.startedAtMs, state.endedAtMs)\n  }\n\n  return unknownTerminalTiming(state.startedAtMs)\n}\n\nconst toolRunStateFor = (state: ChatToolState): ToolRunState => {\n  if (state._tag === 'Running') {\n    return { _tag: 'Running', ...startedTiming(state.startedAtMs) }\n  }\n\n  if (state._tag === 'Completed') {\n    return { _tag: 'Completed', ...terminalTimingFromState(state), result: state.result }\n  }\n\n  if (state._tag === 'InputStreaming') {\n    return { _tag: 'InputStreaming', ...noTiming(), input: state.input }\n  }\n\n  if (state._tag === 'ApprovalRequested') {\n    return { _tag: 'ApprovalRequested', ...noTiming(), request: state.request }\n  }\n\n  if (state._tag === 'Denied') {\n    return { _tag: 'Denied', ...noTiming(), reason: state.reason }\n  }\n\n  if (state._tag === 'QuestionRequested') {\n    return { _tag: 'QuestionRequested', ...noTiming(), request: state.request }\n  }\n\n  if (state._tag === 'QuestionAnswered') {\n    return {\n      _tag: 'QuestionAnswered',\n      ...noTiming(),\n      response: state.response,\n      request: state.request\n    }\n  }\n\n  if (state._tag === 'QuestionCancelled') {\n    return {\n      _tag: 'QuestionCancelled',\n      ...noTiming(),\n      response: state.response,\n      request: state.request\n    }\n  }\n\n  if (state._tag === 'Errored') {\n    return { _tag: 'Errored', ...terminalTimingFromState(state), message: state.message }\n  }\n\n  if (state._tag === 'ProviderCompleted') {\n    return { _tag: 'ProviderCompleted', ...noTiming(), result: state.result }\n  }\n\n  return { _tag: 'Called', ...noTiming() }\n}\n\nconst textItemFromPart = (\n  message: AgentChatMessage,\n  part: Extract<AgentChatMessage['parts'][number], { readonly _tag: 'Text' }>\n): Option.Option<AgentChatItem> => {\n  if (part.state === 'streaming') {\n    switch (message.role) {\n      case 'user':\n        return Option.some({\n          _tag: 'UserDraft',\n          id: part.id,\n          text: contentText(part.content)\n        })\n      case 'assistant':\n        return Option.some({\n          _tag: 'AssistantDraft',\n          id: part.id,\n          text: contentText(part.content)\n        })\n      case 'system':\n        return Option.none()\n    }\n  }\n\n  switch (message.role) {\n    case 'user':\n      return Option.some({\n        _tag: 'UserMessage',\n        id: part.id,\n        messageId: message.id,\n        content: part.content\n      })\n    case 'assistant':\n      return Option.some({\n        _tag: 'AssistantMessage',\n        id: part.id,\n        messageId: message.id,\n        content: part.content\n      })\n    case 'system':\n      return Option.none()\n  }\n}\n\nconst itemFromPart = (\n  message: AgentChatMessage,\n  part: AgentChatMessage['parts'][number]\n): Option.Option<AgentChatItem> => {\n  switch (part._tag) {\n    case 'Text':\n      return textItemFromPart(message, part)\n    case 'Reasoning':\n      return Option.some({ _tag: 'Reasoning', id: part.id, messageId: message.id, text: part.text })\n    case 'ToolCall':\n      return Option.some({\n        _tag: 'ToolRun',\n        id: part.id,\n        messageId: message.id,\n        call: part.call,\n        state: toolRunStateFor(part.state)\n      })\n    case 'ToolResult':\n      return Option.some({\n        _tag: 'ToolResult',\n        id: part.id,\n        messageId: message.id,\n        toolCallId: part.toolCallId,\n        name: part.name,\n        content: part.content,\n        isError: part.isError\n      })\n    case 'Error':\n      return Option.some({ _tag: 'Error', id: part.id, message: part.message })\n  }\n}\n\nexport const buildAgentChatItems = ({\n  messages,\n  isRunning,\n  activeToolLabel\n}: BuildAgentChatItemsInput): ReadonlyArray<AgentChatItem> => {\n  const items = dedupeAgentChatToolRunItems(\n    Arr.getSomes(\n      Arr.flatMap(messages, message => Arr.map(message.parts, part => itemFromPart(message, part)))\n    )\n  )\n\n  if (isRunning) {\n    return [\n      ...items,\n      {\n        _tag: 'AssistantStatus',\n        id: 'assistant-status',\n        label: activeStatusLabel({ messages, activeToolLabel })\n      }\n    ]\n  }\n\n  return items\n}\n"],"mappings":";;;AA+GA,MAAa,+BACX,UACiC;CAEjC,MAAM,0CAA0B,IAAI,IAAoB;CAExD,KAAK,MAAM,CAAC,OAAO,SAAS,MAAM,QAAQ,GACxC,IAAI,KAAK,SAAS,WAChB,wBAAwB,IAAI,KAAK,KAAK,IAAI,KAAK;CAInD,OAAO,MAAM,QACV,MAAM,UACL,KAAK,SAAS,aAAa,wBAAwB,IAAI,KAAK,KAAK,EAAE,MAAM,KAC7E;AACF;AAEA,MAAM,qBAAqB,EACzB,UACA,sBACoE;CACpE,IAAI,OAAO,OAAO,eAAe,GAC/B,OAAO,gBAAgB;CAGzB,MAAM,QAAQ,SAAS,SAAQ,YAC7B,QAAQ,MAAM,KAAI,UAAS;EAAE,aAAa,QAAQ;EAAM;CAAK,EAAE,CACjE;CAEA,IACE,MAAM,MACH,EAAE,aAAa,WACd,KAAK,SAAS,UAAU,KAAK,UAAU,eAAe,gBAAgB,WAC1E,GAEA,OAAO;CAGT,IAAI,MAAM,MAAM,EAAE,WAAW,KAAK,SAAS,eAAe,KAAK,UAAU,WAAW,GAClF,OAAO;CAGT,OAAO;AACT;AAEA,MAAM,kBAAmC,EAAE,UAAU,EAAE,MAAM,UAAU,EAAE;AAEzE,MAAM,iBAAiB,iBAA+C;CACpE,UAAU,EAAE,MAAM,UAAU;CAC5B;AACF;AAEA,MAAM,yBAAyB,gBAC7B,gBAAgB,KAAA,IAAY,SAAS,IAAI,cAAc,WAAW;AAEpE,MAAM,eAAe,aAAqB,eAAsC;CAC9E,UAAU;EAAE,MAAM;EAAS,cAAc,KAAK,IAAI,GAAG,YAAY,WAAW;CAAE;CAC9E;CACA;AACF;AAEA,MAAM,2BACJ,UAC0B;CAC1B,IAAI,MAAM,gBAAgB,KAAA,KAAa,MAAM,cAAc,KAAA,GACzD,OAAO,YAAY,MAAM,aAAa,MAAM,SAAS;CAGvD,OAAO,sBAAsB,MAAM,WAAW;AAChD;AAEA,MAAM,mBAAmB,UAAuC;CAC9D,IAAI,MAAM,SAAS,WACjB,OAAO;EAAE,MAAM;EAAW,GAAG,cAAc,MAAM,WAAW;CAAE;CAGhE,IAAI,MAAM,SAAS,aACjB,OAAO;EAAE,MAAM;EAAa,GAAG,wBAAwB,KAAK;EAAG,QAAQ,MAAM;CAAO;CAGtF,IAAI,MAAM,SAAS,kBACjB,OAAO;EAAE,MAAM;EAAkB,GAAG,SAAS;EAAG,OAAO,MAAM;CAAM;CAGrE,IAAI,MAAM,SAAS,qBACjB,OAAO;EAAE,MAAM;EAAqB,GAAG,SAAS;EAAG,SAAS,MAAM;CAAQ;CAG5E,IAAI,MAAM,SAAS,UACjB,OAAO;EAAE,MAAM;EAAU,GAAG,SAAS;EAAG,QAAQ,MAAM;CAAO;CAG/D,IAAI,MAAM,SAAS,qBACjB,OAAO;EAAE,MAAM;EAAqB,GAAG,SAAS;EAAG,SAAS,MAAM;CAAQ;CAG5E,IAAI,MAAM,SAAS,oBACjB,OAAO;EACL,MAAM;EACN,GAAG,SAAS;EACZ,UAAU,MAAM;EAChB,SAAS,MAAM;CACjB;CAGF,IAAI,MAAM,SAAS,qBACjB,OAAO;EACL,MAAM;EACN,GAAG,SAAS;EACZ,UAAU,MAAM;EAChB,SAAS,MAAM;CACjB;CAGF,IAAI,MAAM,SAAS,WACjB,OAAO;EAAE,MAAM;EAAW,GAAG,wBAAwB,KAAK;EAAG,SAAS,MAAM;CAAQ;CAGtF,IAAI,MAAM,SAAS,qBACjB,OAAO;EAAE,MAAM;EAAqB,GAAG,SAAS;EAAG,QAAQ,MAAM;CAAO;CAG1E,OAAO;EAAE,MAAM;EAAU,GAAG,SAAS;CAAE;AACzC;AAEA,MAAM,oBACJ,SACA,SACiC;CACjC,IAAI,KAAK,UAAU,aACjB,QAAQ,QAAQ,MAAhB;EACE,KAAK,QACH,OAAO,OAAO,KAAK;GACjB,MAAM;GACN,IAAI,KAAK;GACT,MAAM,YAAY,KAAK,OAAO;EAChC,CAAC;EACH,KAAK,aACH,OAAO,OAAO,KAAK;GACjB,MAAM;GACN,IAAI,KAAK;GACT,MAAM,YAAY,KAAK,OAAO;EAChC,CAAC;EACH,KAAK,UACH,OAAO,OAAO,KAAK;CACvB;CAGF,QAAQ,QAAQ,MAAhB;EACE,KAAK,QACH,OAAO,OAAO,KAAK;GACjB,MAAM;GACN,IAAI,KAAK;GACT,WAAW,QAAQ;GACnB,SAAS,KAAK;EAChB,CAAC;EACH,KAAK,aACH,OAAO,OAAO,KAAK;GACjB,MAAM;GACN,IAAI,KAAK;GACT,WAAW,QAAQ;GACnB,SAAS,KAAK;EAChB,CAAC;EACH,KAAK,UACH,OAAO,OAAO,KAAK;CACvB;AACF;AAEA,MAAM,gBACJ,SACA,SACiC;CACjC,QAAQ,KAAK,MAAb;EACE,KAAK,QACH,OAAO,iBAAiB,SAAS,IAAI;EACvC,KAAK,aACH,OAAO,OAAO,KAAK;GAAE,MAAM;GAAa,IAAI,KAAK;GAAI,WAAW,QAAQ;GAAI,MAAM,KAAK;EAAK,CAAC;EAC/F,KAAK,YACH,OAAO,OAAO,KAAK;GACjB,MAAM;GACN,IAAI,KAAK;GACT,WAAW,QAAQ;GACnB,MAAM,KAAK;GACX,OAAO,gBAAgB,KAAK,KAAK;EACnC,CAAC;EACH,KAAK,cACH,OAAO,OAAO,KAAK;GACjB,MAAM;GACN,IAAI,KAAK;GACT,WAAW,QAAQ;GACnB,YAAY,KAAK;GACjB,MAAM,KAAK;GACX,SAAS,KAAK;GACd,SAAS,KAAK;EAChB,CAAC;EACH,KAAK,SACH,OAAO,OAAO,KAAK;GAAE,MAAM;GAAS,IAAI,KAAK;GAAI,SAAS,KAAK;EAAQ,CAAC;CAC5E;AACF;AAEA,MAAa,uBAAuB,EAClC,UACA,WACA,sBAC4D;CAC5D,MAAM,QAAQ,4BACZA,MAAI,SACFA,MAAI,QAAQ,WAAU,YAAWA,MAAI,IAAI,QAAQ,QAAO,SAAQ,aAAa,SAAS,IAAI,CAAC,CAAC,CAC9F,CACF;CAEA,IAAI,WACF,OAAO,CACL,GAAG,OACH;EACE,MAAM;EACN,IAAI;EACJ,OAAO,kBAAkB;GAAE;GAAU;EAAgB,CAAC;CACxD,CACF;CAGF,OAAO;AACT"}
+{"version":3,"file":"chat-items.mjs","names":["Arr"],"sources":["../../src/react/chat-items.ts"],"sourcesContent":["import { Array as Arr, Option } from 'effect'\nimport {\n  contentText,\n  type AgentRetry,\n  type QuestionRequest,\n  type QuestionResponse,\n  type ToolApprovalRequest,\n  type ToolResult,\n  type Content,\n  type ToolCall\n} from '@yolk-sdk/agent/protocol'\nimport type { AgentChatMessage, ChatToolState } from './chat-messages.ts'\n\nexport type ToolDuration =\n  | { readonly _tag: 'Known'; readonly milliseconds: number }\n  | { readonly _tag: 'Unknown' }\n\ntype ToolRunNoTiming = {\n  readonly duration: { readonly _tag: 'Unknown' }\n  readonly startedAtMs?: undefined\n  readonly endedAtMs?: undefined\n}\n\ntype ToolRunStartedTiming = {\n  readonly duration: { readonly _tag: 'Unknown' }\n  readonly startedAtMs: number\n  readonly endedAtMs?: undefined\n}\n\ntype ToolRunKnownTiming = {\n  readonly duration: { readonly _tag: 'Known'; readonly milliseconds: number }\n  readonly startedAtMs: number\n  readonly endedAtMs: number\n}\n\nexport type ToolRunTiming = ToolRunNoTiming | ToolRunStartedTiming | ToolRunKnownTiming\n\ntype ToolRunTerminalTiming = ToolRunTiming\n\nexport type ToolRunState =\n  | ({ readonly _tag: 'Running' } & ToolRunStartedTiming)\n  | ({ readonly _tag: 'Called' } & ToolRunNoTiming)\n  | ({ readonly _tag: 'InputStreaming'; readonly input: string } & ToolRunNoTiming)\n  | ({\n      readonly _tag: 'ApprovalRequested'\n      readonly request?: ToolApprovalRequest\n    } & ToolRunNoTiming)\n  | ({ readonly _tag: 'Denied'; readonly reason: string } & ToolRunNoTiming)\n  | ({ readonly _tag: 'QuestionRequested'; readonly request: QuestionRequest } & ToolRunNoTiming)\n  | ({\n      readonly _tag: 'QuestionAnswered'\n      readonly response: QuestionResponse\n      readonly request?: QuestionRequest\n    } & ToolRunNoTiming)\n  | ({\n      readonly _tag: 'QuestionCancelled'\n      readonly response: QuestionResponse\n      readonly request?: QuestionRequest\n    } & ToolRunNoTiming)\n  | ({ readonly _tag: 'Completed'; readonly result: ToolResult } & ToolRunTerminalTiming)\n  | ({ readonly _tag: 'Errored'; readonly message: string } & ToolRunTerminalTiming)\n  | ({\n      readonly _tag: 'ProviderCompleted'\n      readonly result: ToolResult\n    } & ToolRunNoTiming)\n\nexport type AgentChatItem =\n  | {\n      readonly _tag: 'UserMessage'\n      readonly id: string\n      readonly messageId: string\n      readonly content: Content\n    }\n  | {\n      readonly _tag: 'AssistantMessage'\n      readonly id: string\n      readonly messageId: string\n      readonly content: Content\n    }\n  | {\n      readonly _tag: 'Reasoning'\n      readonly id: string\n      readonly messageId: string\n      readonly text: string\n    }\n  | {\n      readonly _tag: 'ToolRun'\n      readonly id: string\n      readonly messageId: string\n      readonly call: ToolCall\n      readonly state: ToolRunState\n    }\n  | {\n      readonly _tag: 'ToolResult'\n      readonly id: string\n      readonly messageId: string\n      readonly toolCallId: string\n      readonly name: string\n      readonly content: Content\n      readonly isError?: boolean\n    }\n  | { readonly _tag: 'UserDraft'; readonly id: string; readonly text: string }\n  | { readonly _tag: 'AssistantDraft'; readonly id: string; readonly text: string }\n  | { readonly _tag: 'Retry'; readonly id: string; readonly retry: AgentRetry }\n  | { readonly _tag: 'AssistantStatus'; readonly id: string; readonly label: string }\n  | { readonly _tag: 'Error'; readonly id: string; readonly message: string }\n\nexport type BuildAgentChatItemsInput = {\n  readonly messages: ReadonlyArray<AgentChatMessage>\n  readonly isRunning: boolean\n  readonly activeToolLabel: Option.Option<string>\n  readonly retryInfo?: AgentRetry | null\n}\n\nexport const dedupeAgentChatToolRunItems = (\n  items: ReadonlyArray<AgentChatItem>\n): ReadonlyArray<AgentChatItem> => {\n  // Tool call ids are transcript-global; latest wins when live and persisted projections overlap.\n  const latestIndexByToolCallId = new Map<string, number>()\n\n  for (const [index, item] of items.entries()) {\n    if (item._tag === 'ToolRun') {\n      latestIndexByToolCallId.set(item.call.id, index)\n    }\n  }\n\n  return items.filter(\n    (item, index) => item._tag !== 'ToolRun' || latestIndexByToolCallId.get(item.call.id) === index\n  )\n}\n\nconst activeStatusLabel = ({\n  messages,\n  activeToolLabel\n}: Pick<BuildAgentChatItemsInput, 'messages' | 'activeToolLabel'>) => {\n  if (Option.isSome(activeToolLabel)) {\n    return activeToolLabel.value\n  }\n\n  const parts = messages.flatMap(message =>\n    message.parts.map(part => ({ messageRole: message.role, part }))\n  )\n\n  if (\n    parts.some(\n      ({ messageRole, part }) =>\n        part._tag === 'Text' && part.state === 'streaming' && messageRole === 'assistant'\n    )\n  ) {\n    return 'Responding'\n  }\n\n  if (parts.some(({ part }) => part._tag === 'Reasoning' && part.state === 'streaming')) {\n    return 'Thinking'\n  }\n\n  return 'Thinking'\n}\n\nconst noTiming = (): ToolRunNoTiming => ({ duration: { _tag: 'Unknown' } })\n\nconst startedTiming = (startedAtMs: number): ToolRunStartedTiming => ({\n  duration: { _tag: 'Unknown' },\n  startedAtMs\n})\n\nconst unknownTerminalTiming = (startedAtMs?: number): ToolRunNoTiming | ToolRunStartedTiming =>\n  startedAtMs === undefined ? noTiming() : startedTiming(startedAtMs)\n\nconst knownTiming = (startedAtMs: number, endedAtMs: number): ToolRunTiming => ({\n  duration: { _tag: 'Known', milliseconds: Math.max(0, endedAtMs - startedAtMs) },\n  startedAtMs,\n  endedAtMs\n})\n\nconst terminalTimingFromState = (\n  state: Extract<ChatToolState, { readonly _tag: 'Completed' | 'Errored' }>\n): ToolRunTerminalTiming => {\n  if (state.startedAtMs !== undefined && state.endedAtMs !== undefined) {\n    return knownTiming(state.startedAtMs, state.endedAtMs)\n  }\n\n  return unknownTerminalTiming(state.startedAtMs)\n}\n\nconst toolRunStateFor = (state: ChatToolState): ToolRunState => {\n  if (state._tag === 'Running') {\n    return { _tag: 'Running', ...startedTiming(state.startedAtMs) }\n  }\n\n  if (state._tag === 'Completed') {\n    return { _tag: 'Completed', ...terminalTimingFromState(state), result: state.result }\n  }\n\n  if (state._tag === 'InputStreaming') {\n    return { _tag: 'InputStreaming', ...noTiming(), input: state.input }\n  }\n\n  if (state._tag === 'ApprovalRequested') {\n    return { _tag: 'ApprovalRequested', ...noTiming(), request: state.request }\n  }\n\n  if (state._tag === 'Denied') {\n    return { _tag: 'Denied', ...noTiming(), reason: state.reason }\n  }\n\n  if (state._tag === 'QuestionRequested') {\n    return { _tag: 'QuestionRequested', ...noTiming(), request: state.request }\n  }\n\n  if (state._tag === 'QuestionAnswered') {\n    return {\n      _tag: 'QuestionAnswered',\n      ...noTiming(),\n      response: state.response,\n      request: state.request\n    }\n  }\n\n  if (state._tag === 'QuestionCancelled') {\n    return {\n      _tag: 'QuestionCancelled',\n      ...noTiming(),\n      response: state.response,\n      request: state.request\n    }\n  }\n\n  if (state._tag === 'Errored') {\n    return { _tag: 'Errored', ...terminalTimingFromState(state), message: state.message }\n  }\n\n  if (state._tag === 'ProviderCompleted') {\n    return { _tag: 'ProviderCompleted', ...noTiming(), result: state.result }\n  }\n\n  return { _tag: 'Called', ...noTiming() }\n}\n\nconst textItemFromPart = (\n  message: AgentChatMessage,\n  part: Extract<AgentChatMessage['parts'][number], { readonly _tag: 'Text' }>\n): Option.Option<AgentChatItem> => {\n  if (part.state === 'streaming') {\n    switch (message.role) {\n      case 'user':\n        return Option.some({\n          _tag: 'UserDraft',\n          id: part.id,\n          text: contentText(part.content)\n        })\n      case 'assistant':\n        return Option.some({\n          _tag: 'AssistantDraft',\n          id: part.id,\n          text: contentText(part.content)\n        })\n      case 'system':\n        return Option.none()\n    }\n  }\n\n  switch (message.role) {\n    case 'user':\n      return Option.some({\n        _tag: 'UserMessage',\n        id: part.id,\n        messageId: message.id,\n        content: part.content\n      })\n    case 'assistant':\n      return Option.some({\n        _tag: 'AssistantMessage',\n        id: part.id,\n        messageId: message.id,\n        content: part.content\n      })\n    case 'system':\n      return Option.none()\n  }\n}\n\nconst itemFromPart = (\n  message: AgentChatMessage,\n  part: AgentChatMessage['parts'][number]\n): Option.Option<AgentChatItem> => {\n  switch (part._tag) {\n    case 'Text':\n      return textItemFromPart(message, part)\n    case 'Reasoning':\n      return Option.some({ _tag: 'Reasoning', id: part.id, messageId: message.id, text: part.text })\n    case 'ToolCall':\n      return Option.some({\n        _tag: 'ToolRun',\n        id: part.id,\n        messageId: message.id,\n        call: part.call,\n        state: toolRunStateFor(part.state)\n      })\n    case 'ToolResult':\n      return Option.some({\n        _tag: 'ToolResult',\n        id: part.id,\n        messageId: message.id,\n        toolCallId: part.toolCallId,\n        name: part.name,\n        content: part.content,\n        isError: part.isError\n      })\n    case 'Error':\n      return Option.some({ _tag: 'Error', id: part.id, message: part.message })\n  }\n}\n\nexport const buildAgentChatItems = ({\n  messages,\n  isRunning,\n  activeToolLabel,\n  retryInfo\n}: BuildAgentChatItemsInput): ReadonlyArray<AgentChatItem> => {\n  const items = dedupeAgentChatToolRunItems(\n    Arr.getSomes(\n      Arr.flatMap(messages, message => Arr.map(message.parts, part => itemFromPart(message, part)))\n    )\n  )\n\n  if (isRunning) {\n    if (retryInfo !== undefined && retryInfo !== null) {\n      return [...items, { _tag: 'Retry', id: 'agent-retry', retry: retryInfo }]\n    }\n\n    return [\n      ...items,\n      {\n        _tag: 'AssistantStatus',\n        id: 'assistant-status',\n        label: activeStatusLabel({ messages, activeToolLabel })\n      }\n    ]\n  }\n\n  return items\n}\n"],"mappings":";;;AAkHA,MAAa,+BACX,UACiC;CAEjC,MAAM,0CAA0B,IAAI,IAAoB;CAExD,KAAK,MAAM,CAAC,OAAO,SAAS,MAAM,QAAQ,GACxC,IAAI,KAAK,SAAS,WAChB,wBAAwB,IAAI,KAAK,KAAK,IAAI,KAAK;CAInD,OAAO,MAAM,QACV,MAAM,UAAU,KAAK,SAAS,aAAa,wBAAwB,IAAI,KAAK,KAAK,EAAE,MAAM,KAC5F;AACF;AAEA,MAAM,qBAAqB,EACzB,UACA,sBACoE;CACpE,IAAI,OAAO,OAAO,eAAe,GAC/B,OAAO,gBAAgB;CAGzB,MAAM,QAAQ,SAAS,SAAQ,YAC7B,QAAQ,MAAM,KAAI,UAAS;EAAE,aAAa,QAAQ;EAAM;CAAK,EAAE,CACjE;CAEA,IACE,MAAM,MACH,EAAE,aAAa,WACd,KAAK,SAAS,UAAU,KAAK,UAAU,eAAe,gBAAgB,WAC1E,GAEA,OAAO;CAGT,IAAI,MAAM,MAAM,EAAE,WAAW,KAAK,SAAS,eAAe,KAAK,UAAU,WAAW,GAClF,OAAO;CAGT,OAAO;AACT;AAEA,MAAM,kBAAmC,EAAE,UAAU,EAAE,MAAM,UAAU,EAAE;AAEzE,MAAM,iBAAiB,iBAA+C;CACpE,UAAU,EAAE,MAAM,UAAU;CAC5B;AACF;AAEA,MAAM,yBAAyB,gBAC7B,gBAAgB,KAAA,IAAY,SAAS,IAAI,cAAc,WAAW;AAEpE,MAAM,eAAe,aAAqB,eAAsC;CAC9E,UAAU;EAAE,MAAM;EAAS,cAAc,KAAK,IAAI,GAAG,YAAY,WAAW;CAAE;CAC9E;CACA;AACF;AAEA,MAAM,2BACJ,UAC0B;CAC1B,IAAI,MAAM,gBAAgB,KAAA,KAAa,MAAM,cAAc,KAAA,GACzD,OAAO,YAAY,MAAM,aAAa,MAAM,SAAS;CAGvD,OAAO,sBAAsB,MAAM,WAAW;AAChD;AAEA,MAAM,mBAAmB,UAAuC;CAC9D,IAAI,MAAM,SAAS,WACjB,OAAO;EAAE,MAAM;EAAW,GAAG,cAAc,MAAM,WAAW;CAAE;CAGhE,IAAI,MAAM,SAAS,aACjB,OAAO;EAAE,MAAM;EAAa,GAAG,wBAAwB,KAAK;EAAG,QAAQ,MAAM;CAAO;CAGtF,IAAI,MAAM,SAAS,kBACjB,OAAO;EAAE,MAAM;EAAkB,GAAG,SAAS;EAAG,OAAO,MAAM;CAAM;CAGrE,IAAI,MAAM,SAAS,qBACjB,OAAO;EAAE,MAAM;EAAqB,GAAG,SAAS;EAAG,SAAS,MAAM;CAAQ;CAG5E,IAAI,MAAM,SAAS,UACjB,OAAO;EAAE,MAAM;EAAU,GAAG,SAAS;EAAG,QAAQ,MAAM;CAAO;CAG/D,IAAI,MAAM,SAAS,qBACjB,OAAO;EAAE,MAAM;EAAqB,GAAG,SAAS;EAAG,SAAS,MAAM;CAAQ;CAG5E,IAAI,MAAM,SAAS,oBACjB,OAAO;EACL,MAAM;EACN,GAAG,SAAS;EACZ,UAAU,MAAM;EAChB,SAAS,MAAM;CACjB;CAGF,IAAI,MAAM,SAAS,qBACjB,OAAO;EACL,MAAM;EACN,GAAG,SAAS;EACZ,UAAU,MAAM;EAChB,SAAS,MAAM;CACjB;CAGF,IAAI,MAAM,SAAS,WACjB,OAAO;EAAE,MAAM;EAAW,GAAG,wBAAwB,KAAK;EAAG,SAAS,MAAM;CAAQ;CAGtF,IAAI,MAAM,SAAS,qBACjB,OAAO;EAAE,MAAM;EAAqB,GAAG,SAAS;EAAG,QAAQ,MAAM;CAAO;CAG1E,OAAO;EAAE,MAAM;EAAU,GAAG,SAAS;CAAE;AACzC;AAEA,MAAM,oBACJ,SACA,SACiC;CACjC,IAAI,KAAK,UAAU,aACjB,QAAQ,QAAQ,MAAhB;EACE,KAAK,QACH,OAAO,OAAO,KAAK;GACjB,MAAM;GACN,IAAI,KAAK;GACT,MAAM,YAAY,KAAK,OAAO;EAChC,CAAC;EACH,KAAK,aACH,OAAO,OAAO,KAAK;GACjB,MAAM;GACN,IAAI,KAAK;GACT,MAAM,YAAY,KAAK,OAAO;EAChC,CAAC;EACH,KAAK,UACH,OAAO,OAAO,KAAK;CACvB;CAGF,QAAQ,QAAQ,MAAhB;EACE,KAAK,QACH,OAAO,OAAO,KAAK;GACjB,MAAM;GACN,IAAI,KAAK;GACT,WAAW,QAAQ;GACnB,SAAS,KAAK;EAChB,CAAC;EACH,KAAK,aACH,OAAO,OAAO,KAAK;GACjB,MAAM;GACN,IAAI,KAAK;GACT,WAAW,QAAQ;GACnB,SAAS,KAAK;EAChB,CAAC;EACH,KAAK,UACH,OAAO,OAAO,KAAK;CACvB;AACF;AAEA,MAAM,gBACJ,SACA,SACiC;CACjC,QAAQ,KAAK,MAAb;EACE,KAAK,QACH,OAAO,iBAAiB,SAAS,IAAI;EACvC,KAAK,aACH,OAAO,OAAO,KAAK;GAAE,MAAM;GAAa,IAAI,KAAK;GAAI,WAAW,QAAQ;GAAI,MAAM,KAAK;EAAK,CAAC;EAC/F,KAAK,YACH,OAAO,OAAO,KAAK;GACjB,MAAM;GACN,IAAI,KAAK;GACT,WAAW,QAAQ;GACnB,MAAM,KAAK;GACX,OAAO,gBAAgB,KAAK,KAAK;EACnC,CAAC;EACH,KAAK,cACH,OAAO,OAAO,KAAK;GACjB,MAAM;GACN,IAAI,KAAK;GACT,WAAW,QAAQ;GACnB,YAAY,KAAK;GACjB,MAAM,KAAK;GACX,SAAS,KAAK;GACd,SAAS,KAAK;EAChB,CAAC;EACH,KAAK,SACH,OAAO,OAAO,KAAK;GAAE,MAAM;GAAS,IAAI,KAAK;GAAI,SAAS,KAAK;EAAQ,CAAC;CAC5E;AACF;AAEA,MAAa,uBAAuB,EAClC,UACA,WACA,iBACA,gBAC4D;CAC5D,MAAM,QAAQ,4BACZA,MAAI,SACFA,MAAI,QAAQ,WAAU,YAAWA,MAAI,IAAI,QAAQ,QAAO,SAAQ,aAAa,SAAS,IAAI,CAAC,CAAC,CAC9F,CACF;CAEA,IAAI,WAAW;EACb,IAAI,cAAc,KAAA,KAAa,cAAc,MAC3C,OAAO,CAAC,GAAG,OAAO;GAAE,MAAM;GAAS,IAAI;GAAe,OAAO;EAAU,CAAC;EAG1E,OAAO,CACL,GAAG,OACH;GACE,MAAM;GACN,IAAI;GACJ,OAAO,kBAAkB;IAAE;IAAU;GAAgB,CAAC;EACxD,CACF;CACF;CAEA,OAAO;AACT"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yolk-sdk/agent",
-  "version": "0.1.0-canary.20",
+  "version": "0.1.0-canary.21",
   "description": "Protocol, loop, runtime, client, compaction, tools, React, providers, OAuth, skillset, and voice primitives for building Yolk agents.",
   "license": "MIT",
   "type": "module",

package/src/client/state.ts

@@ -1,6 +1,8 @@
 import {
+  type AgentError,
   type AgentEvent,
   type AgentMessage,
+  type AgentRetry,
   type HitlRequest,
   type QuestionRequest,
   type QuestionResponse,
@@ -63,6 +65,8 @@ export 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>
 }
 
@@ -80,9 +84,14 @@ export const initialAgentClientState: AgentClientState = {
   reasoning: '',
   toolRuns: [],
   error: null,
+  errorInfo: null,
+  retryInfo: null,
   seenEventIds: []
 }
 
+const clearRetryInfo = (state: AgentClientState): AgentClientState =>
+  state.retryInfo === null ? state : { ...state, retryInfo: null }
+
 const hasSeenEvent = (state: AgentClientState, event: AgentEvent) =>
   event.eventId !== undefined && state.seenEventIds.includes(event.eventId)
 
@@ -260,16 +269,18 @@ const applyAgentEventUnchecked = (
         reasoning: '',
         liveMessages: [],
         toolRuns: completedToolRuns(state.toolRuns),
-        error: null
+        error: null,
+        errorInfo: null,
+        retryInfo: null
       }
     case 'AgentError':
-      return markAgentError(state, event.message)
+      return markAgentError(state, event.message, event)
     case 'LLMTextDelta':
-      return { ...state, text: `${state.text}${event.text}` }
+      return clearRetryInfo({ ...state, text: `${state.text}${event.text}` })
     case 'LLMReasoningDelta':
-      return { ...state, reasoning: `${state.reasoning}${event.text}` }
+      return clearRetryInfo({ ...state, reasoning: `${state.reasoning}${event.text}` })
     case 'ToolInputStart':
-      return {
+      return clearRetryInfo({
         ...state,
         toolRuns: replaceToolRun(state.toolRuns, {
           _tag: 'InputStreaming',
@@ -277,14 +288,17 @@ const applyAgentEventUnchecked = (
           name: event.name,
           input: ''
         })
-      }
+      })
     case 'ToolInputDelta':
-      return { ...state, toolRuns: appendToolInputDelta(state.toolRuns, event.id, event.delta) }
+      return clearRetryInfo({
+        ...state,
+        toolRuns: appendToolInputDelta(state.toolRuns, event.id, event.delta)
+      })
     case 'ToolInputEnd':
-      return {
+      return clearRetryInfo({
         ...state,
         toolRuns: replaceToolRun(state.toolRuns, { _tag: 'InputReady', call: event.call })
-      }
+      })
     case 'ToolApprovalRequested':
       return {
         ...state,
@@ -370,21 +384,21 @@ const applyAgentEventUnchecked = (
         })
       }
     case 'ProviderToolResult':
-      return {
+      return clearRetryInfo({
         ...state,
         toolRuns: replaceToolRun(state.toolRuns, {
           _tag: 'ProviderCompleted',
           call: event.call,
           result: event.result
         })
-      }
+      })
     case 'AssistantMessage':
-      return {
+      return clearRetryInfo({
         ...state,
         liveMessages: [...state.liveMessages, event.message],
         text: '',
         reasoning: ''
-      }
+      })
     case 'AgentEnd':
       return {
         ...state,
@@ -393,7 +407,8 @@ const applyAgentEventUnchecked = (
         liveMessages: [],
         text: '',
         reasoning: '',
-        toolRuns: completedToolRuns(state.toolRuns)
+        toolRuns: completedToolRuns(state.toolRuns),
+        retryInfo: null
       }
     case 'AgentAwaitingInput':
       return {
@@ -403,9 +418,12 @@ const applyAgentEventUnchecked = (
         liveMessages: [],
         text: '',
         reasoning: '',
-        error: null
+        error: null,
+        errorInfo: null,
+        retryInfo: null
       }
     case 'AgentRetry':
+      return { ...state, retryInfo: event }
     case 'CompactionEnd':
     case 'CompactionStart':
     case 'LLMStreamEnd':
@@ -431,24 +449,31 @@ export const submitAgentUserMessage = (
   reasoning: '',
   toolRuns: completedToolRuns(state.toolRuns),
   error: null,
+  errorInfo: null,
+  retryInfo: null,
   seenEventIds: []
 })
 
 export const markAgentError = (
   state: AgentClientState,
-  message = 'Agent request failed'
+  message = 'Agent request failed',
+  errorInfo: AgentError | null = null
 ): AgentClientState => ({
   ...state,
   status: 'error',
   toolRuns: completedToolRuns(state.toolRuns),
-  error: message
+  error: message,
+  errorInfo,
+  retryInfo: null
 })
 
 export const markAgentAborted = (state: AgentClientState): AgentClientState => ({
   ...state,
   status: 'aborted',
   toolRuns: completedToolRuns(state.toolRuns),
-  error: null
+  error: null,
+  errorInfo: null,
+  retryInfo: null
 })
 
 export const reduceAgentEvents = (

package/src/loop/error.ts

@@ -1,16 +1,18 @@
 import * as Schema from 'effect/Schema'
-import { AgentError, type AgentErrorCode } from '@yolk-sdk/agent/protocol'
+import { AgentError, ProviderErrorInfo, type AgentErrorCode } from '@yolk-sdk/agent/protocol'
 
 export class LLMError extends Schema.TaggedErrorClass<LLMError>()('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)
 }) {}
 
 export class FauxExhaustedError extends Schema.TaggedErrorClass<FauxExhaustedError>()(
@@ -75,7 +77,8 @@ export const agentLoopErrorToAgentError = (error: AgentLoopError): AgentError =>
       return AgentError.make({
         code: error.cause,
         message: error.message,
-        retryable: error.retryable
+        retryable: error.retryable,
+        ...(error.provider === undefined ? {} : { provider: error.provider })
       })
     case 'ToolError':
       return AgentError.make({

package/src/loop/run.ts

@@ -292,8 +292,30 @@ type TurnStreamInput = {
   readonly turn: number
 }
 
-const retryDelayMs = (baseDelayMs: number, attempt: number) =>
-  Math.max(0, Math.floor(baseDelayMs * 2 ** Math.max(0, attempt - 1)))
+const maxUnhintedRetryDelayMs = 30_000
+const maxHintedRetryDelayMs = 2_147_483_647
+
+const validDelayMs = (delayMs: number) =>
+  Number.isFinite(delayMs) && delayMs >= 0 ? Math.floor(delayMs) : undefined
+
+const hintedRetryDelayMs = (error: LLMError) => {
+  const delayMs = error.provider?.retryAfterMs
+  const validDelay = delayMs === undefined ? undefined : validDelayMs(delayMs)
+
+  return validDelay === undefined ? undefined : Math.min(validDelay, maxHintedRetryDelayMs)
+}
+
+const retryDelayMs = (baseDelayMs: number, attempt: number, error: LLMError) => {
+  const hintedDelay = hintedRetryDelayMs(error)
+
+  if (hintedDelay !== undefined) {
+    return hintedDelay
+  }
+
+  const delayMs = validDelayMs(baseDelayMs * 2 ** Math.max(0, attempt - 1)) ?? 0
+
+  return Math.min(delayMs, maxUnhintedRetryDelayMs)
+}
 
 const retryReason = (error: LLMError): AgentErrorCode => error.cause
 
@@ -332,13 +354,14 @@ const withProviderRetries = (
                       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 === undefined ? {} : { provider: error.provider })
                       })
                     ).pipe(
                       Stream.concat(sleepStream(delayMs)),

package/src/loop/services/loop-config.ts

@@ -14,7 +14,7 @@ export class LoopConfig extends Context.Service<LoopConfig, LoopConfigShape>()(
   static defaultLayer = this.layer({
     maxTurns: 500,
     maxRetries: 2,
-    retryBaseDelayMs: 1000,
+    retryBaseDelayMs: 2000,
     toolConcurrency: 4
   })
 }