agents 0.17.3 → 0.18.0

package/dist/chat/index.js

@@ -1,7 +1,7 @@
 import { i as normalizeToolInput, n as getPartialStreamText, r as isReplayChunk, t as applyChunkToParts } from "../message-builder-BymO4N_D.js";
 import { i as interceptAgentToolBroadcast, n as applyAgentToolEvent, r as createAgentToolEventState, t as AgentToolProgressEmitter } from "../agent-tools-y7zLfw4Q.js";
 import { t as truncateToolOutput } from "../tool-output-truncation-CNnnGZQ3.js";
-import { n as transition, r as StreamAccumulator, t as MessageType } from "../wire-types-nflOzNuU.js";
+import { a as StreamAccumulator, i as transition, n as CHAT_MESSAGE_TYPES, r as STREAM_RESUME_NONE_REASONS, t as MessageType } from "../wire-types-CU9rLoeS.js";
 import { jsonSchema, tool } from "ai";
 import { nanoid } from "nanoid";
 //#region src/chat/sanitize.ts
@@ -335,31 +335,6 @@ var SubmitConcurrencyController = class {
 	}
 };
 //#endregion
-//#region src/chat/protocol.ts
-/**
-* Wire protocol message type constants for the cf_agent_chat_* protocol.
-*
-* These are the string values used on the wire between agent servers and
-* clients. Both @cloudflare/ai-chat (via its MessageType enum) and
-* @cloudflare/think use these values.
-*/
-const CHAT_MESSAGE_TYPES = {
-	CHAT_MESSAGES: "cf_agent_chat_messages",
-	USE_CHAT_REQUEST: "cf_agent_use_chat_request",
-	USE_CHAT_RESPONSE: "cf_agent_use_chat_response",
-	CHAT_CLEAR: "cf_agent_chat_clear",
-	CHAT_REQUEST_CANCEL: "cf_agent_chat_request_cancel",
-	STREAM_RESUMING: "cf_agent_stream_resuming",
-	STREAM_RESUME_ACK: "cf_agent_stream_resume_ack",
-	STREAM_RESUME_REQUEST: "cf_agent_stream_resume_request",
-	STREAM_RESUME_NONE: "cf_agent_stream_resume_none",
-	STREAM_PENDING: "cf_agent_stream_pending",
-	TOOL_RESULT: "cf_agent_tool_result",
-	TOOL_APPROVAL: "cf_agent_tool_approval",
-	MESSAGE_UPDATED: "cf_agent_message_updated",
-	CHAT_RECOVERING: "cf_agent_chat_recovering"
-};
-//#endregion
 //#region src/chat/connection.ts
 /**
 * Send a message on a connection, swallowing the specific
@@ -1245,12 +1220,13 @@ var PreStreamTurns = class {
 	* `pendingResumeConnections` — they must keep receiving any live broadcast —
 	* until the host flushes them through `notifyStreamResuming` on stream start.
 	*/
-	park(connection) {
+	park(connection, probeId) {
 		if (!this.hasInFlight()) return false;
 		this.awaitingConnections.set(connection.id, connection);
 		sendIfOpen(connection, JSON.stringify({
 			type: MSG_STREAM_PENDING,
-			...this._latestRequestId ? { id: this._latestRequestId } : {}
+			...this._latestRequestId ? { id: this._latestRequestId } : {},
+			...probeId ? { probeId } : {}
 		}));
 		return true;
 	}
@@ -1870,7 +1846,10 @@ function parseProtocolMessage(raw) {
 			approved: data.approved,
 			autoContinue: data.autoContinue
 		};
-		case CHAT_MESSAGE_TYPES.STREAM_RESUME_REQUEST: return { type: "stream-resume-request" };
+		case CHAT_MESSAGE_TYPES.STREAM_RESUME_REQUEST: return {
+			type: "stream-resume-request",
+			...typeof data.probeId === "string" ? { probeId: data.probeId } : {}
+		};
 		case CHAT_MESSAGE_TYPES.STREAM_RESUME_ACK: return {
 			type: "stream-resume-ack",
 			id: data.id
@@ -2327,13 +2306,15 @@ var ResumeHandshake = class {
 	* `reconnectToStream` hangs to its timeout with no replay), and the proactive
 	* notify is required for clients that never send a request. The notify is one
 	* tiny frame; the client dedupes its ACK so the buffer is not replayed twice.
+	* Direct responses echo the opaque probe id; proactive notifications omit it.
 	*/
-	notifyStreamResuming(connection) {
+	notifyStreamResuming(connection, probeId) {
 		const { resumableStream, pendingResumeConnections } = this.host;
 		if (!resumableStream.hasActiveStream()) return;
 		if (sendIfOpen(connection, JSON.stringify({
 			type: CHAT_MESSAGE_TYPES.STREAM_RESUMING,
-			id: resumableStream.activeRequestId
+			id: resumableStream.activeRequestId,
+			...probeId ? { probeId } : {}
 		}))) pendingResumeConnections.add(connection.id);
 	}
 	/**
@@ -2341,20 +2322,28 @@ var ResumeHandshake = class {
 	* message handler is registered, avoiding the race where a proactive
 	* `STREAM_RESUMING` from onConnect arrives before the handler is ready.
 	*/
-	async handleResumeRequest(connection) {
+	async handleResumeRequest(connection, probeId) {
 		const { resumableStream, continuation, preStream } = this.host;
-		if (resumableStream.hasActiveStream()) if (continuation.activeRequestId === resumableStream.activeRequestId && continuation.activeConnectionId !== null && continuation.activeConnectionId !== connection.id && this._ownerStillPresent(continuation.activeConnectionId)) sendIfOpen(connection, JSON.stringify({ type: CHAT_MESSAGE_TYPES.STREAM_RESUME_NONE }));
-		else this.notifyStreamResuming(connection);
+		if (resumableStream.hasActiveStream()) if (continuation.activeRequestId === resumableStream.activeRequestId && continuation.activeConnectionId !== null && continuation.activeConnectionId !== connection.id && this._ownerStillPresent(continuation.activeConnectionId)) this._sendResumeNone(connection, STREAM_RESUME_NONE_REASONS.CONTINUATION_OWNED, probeId);
+		else this.notifyStreamResuming(connection, probeId);
 		else if (continuation.pending !== null && (continuation.pending.connectionId === null || continuation.pending.connectionId === connection.id)) {
 			continuation.awaitingConnections.set(connection.id, connection);
-			this._sendStreamPending(connection, continuation.pending.requestId);
-		} else if (await this._replayTerminalOnResume(connection)) {} else if (preStream?.park(connection)) {} else sendIfOpen(connection, JSON.stringify({ type: CHAT_MESSAGE_TYPES.STREAM_RESUME_NONE }));
+			this._sendStreamPending(connection, continuation.pending.requestId, probeId);
+		} else if (await this._replayTerminalOnResume(connection, probeId)) {} else if (preStream?.park(connection, probeId)) {} else this._sendResumeNone(connection, STREAM_RESUME_NONE_REASONS.IDLE, probeId);
+	}
+	_sendResumeNone(connection, reason, probeId) {
+		sendIfOpen(connection, JSON.stringify({
+			type: CHAT_MESSAGE_TYPES.STREAM_RESUME_NONE,
+			reason,
+			...probeId ? { probeId } : {}
+		}));
 	}
 	/** Send a keep-waiting `STREAM_PENDING` frame (#1784). */
-	_sendStreamPending(connection, requestId) {
+	_sendStreamPending(connection, requestId, probeId) {
 		sendIfOpen(connection, JSON.stringify({
 			type: CHAT_MESSAGE_TYPES.STREAM_PENDING,
-			...requestId ? { id: requestId } : {}
+			...requestId ? { id: requestId } : {},
+			...probeId ? { probeId } : {}
 		}));
 	}
 	/** Whether the active continuation's owner connection is still present. */
@@ -2385,12 +2374,13 @@ var ResumeHandshake = class {
 	* error frame once the client ACKs (see {@link _replayTerminalOnAck}). Returns
 	* `true` if a terminal was pending (and `STREAM_RESUMING` was sent).
 	*/
-	async _replayTerminalOnResume(connection) {
+	async _replayTerminalOnResume(connection, probeId) {
 		const pending = await this.host.pendingChatTerminal();
 		if (!pending) return false;
 		sendIfOpen(connection, JSON.stringify({
 			type: CHAT_MESSAGE_TYPES.STREAM_RESUMING,
-			id: pending.requestId
+			id: pending.requestId,
+			...probeId ? { probeId } : {}
 		}));
 		return true;
 	}
@@ -3403,6 +3393,6 @@ async function* iterateWithStallWatchdog(source, timeoutMs, onStall) {
 	}
 }
 //#endregion
-export { AGENT_TOOL_STREAM_PROGRESS_BUMP_THROTTLE_MS, AbortRegistry, AgentToolProgressEmitter, AgentToolStreamProgressThrottle, AutoContinuationController, CHAT_LAST_TERMINAL_KEY, CHAT_MESSAGE_TYPES, CHAT_RECOVERING_FLAG_TTL_MS, CHAT_RECOVERING_KEY, CHAT_RECOVERY_ALARM_DEBOUNCE_MS, CHAT_RECOVERY_INCIDENT_KEY_PREFIX, CHAT_RECOVERY_INCIDENT_TTL_MS, CHAT_RECOVERY_PROGRESS_KEY, CHAT_RECOVERY_STABLE_RETRY_DELAY_SECONDS, CHAT_STREAM_PROGRESS_CREDIT_THROTTLE_MS, ChatRecoveryEngine, ChatStreamStalledError, ContinuationState, DEFAULT_CHAT_RECOVERY_MAX_ATTEMPTS, DEFAULT_CHAT_RECOVERY_MAX_OOM_RETRIES, DEFAULT_CHAT_RECOVERY_MAX_WORK, DEFAULT_CHAT_RECOVERY_NO_PROGRESS_TIMEOUT_MS, DEFAULT_CHAT_RECOVERY_STABLE_TIMEOUT_MS, DEFAULT_CHAT_RECOVERY_TERMINAL_MESSAGE, KV_DELETE_MAX_KEYS, MAX_BOUND_PARAMS, MessageType, PreStreamTurns, ROW_MAX_BYTES, ResumableStream, ResumeHandshake, STREAM_CLEANUP_DELAY_SECONDS, StreamAccumulator, StreamProgressCreditThrottle, SubmitConcurrencyController, TIMED_OUT, TurnQueue, aiSdkRecoveryCodec, applyAgentToolEvent, applyChunkToParts, applyToolUpdate, awaitWithDeadline, transition as broadcastTransition, buildChatRecoveringFrame, buildInClauseStrings, bumpChatRecoveryProgress, byteLength, chatRecoverySchedulePolicy, classifyAgentToolChildRecovery, cleanupStreamBuffers, clearChatTerminal, clientResolvableToolNames, createAgentToolEventState, createChatFiberSnapshot, createToolsFromClientSchemas, crossMessageToolResultUpdate, drainInteractionApplies, enforceRowSizeLimit, hasIncompleteToolBatch, interceptAgentToolBroadcast, isReplayChunk, iterateWithStallWatchdog, listActiveChatRecoveryIncidents, normalizeToolInput, parseProtocolMessage, partAwaitsClientInteraction, pausedExecutionUpdate, pendingChatTerminal, persistReconstructedOrphan, readChatRecoveryProgress, reconcileMessages, reconcileOrphanPartial, recordChatTerminal, repairInterruptedToolParts, resolveChatRecoveryConfig, resolveToolMergeId, runChatRecoveryExhaustion, sanitizeMessage, sendIfOpen, setChatRecovering, shouldCreditStreamProgress, sweepStaleChatRecoveryIncidents, toolApprovalUpdate, toolPartHasSettledResult, toolResultUpdate, unwrapChatFiberSnapshot, wrapChatFiberSnapshot };
+export { AGENT_TOOL_STREAM_PROGRESS_BUMP_THROTTLE_MS, AbortRegistry, AgentToolProgressEmitter, AgentToolStreamProgressThrottle, AutoContinuationController, CHAT_LAST_TERMINAL_KEY, CHAT_MESSAGE_TYPES, CHAT_RECOVERING_FLAG_TTL_MS, CHAT_RECOVERING_KEY, CHAT_RECOVERY_ALARM_DEBOUNCE_MS, CHAT_RECOVERY_INCIDENT_KEY_PREFIX, CHAT_RECOVERY_INCIDENT_TTL_MS, CHAT_RECOVERY_PROGRESS_KEY, CHAT_RECOVERY_STABLE_RETRY_DELAY_SECONDS, CHAT_STREAM_PROGRESS_CREDIT_THROTTLE_MS, ChatRecoveryEngine, ChatStreamStalledError, ContinuationState, DEFAULT_CHAT_RECOVERY_MAX_ATTEMPTS, DEFAULT_CHAT_RECOVERY_MAX_OOM_RETRIES, DEFAULT_CHAT_RECOVERY_MAX_WORK, DEFAULT_CHAT_RECOVERY_NO_PROGRESS_TIMEOUT_MS, DEFAULT_CHAT_RECOVERY_STABLE_TIMEOUT_MS, DEFAULT_CHAT_RECOVERY_TERMINAL_MESSAGE, KV_DELETE_MAX_KEYS, MAX_BOUND_PARAMS, MessageType, PreStreamTurns, ROW_MAX_BYTES, ResumableStream, ResumeHandshake, STREAM_CLEANUP_DELAY_SECONDS, STREAM_RESUME_NONE_REASONS, StreamAccumulator, StreamProgressCreditThrottle, SubmitConcurrencyController, TIMED_OUT, TurnQueue, aiSdkRecoveryCodec, applyAgentToolEvent, applyChunkToParts, applyToolUpdate, awaitWithDeadline, transition as broadcastTransition, buildChatRecoveringFrame, buildInClauseStrings, bumpChatRecoveryProgress, byteLength, chatRecoverySchedulePolicy, classifyAgentToolChildRecovery, cleanupStreamBuffers, clearChatTerminal, clientResolvableToolNames, createAgentToolEventState, createChatFiberSnapshot, createToolsFromClientSchemas, crossMessageToolResultUpdate, drainInteractionApplies, enforceRowSizeLimit, hasIncompleteToolBatch, interceptAgentToolBroadcast, isReplayChunk, iterateWithStallWatchdog, listActiveChatRecoveryIncidents, normalizeToolInput, parseProtocolMessage, partAwaitsClientInteraction, pausedExecutionUpdate, pendingChatTerminal, persistReconstructedOrphan, readChatRecoveryProgress, reconcileMessages, reconcileOrphanPartial, recordChatTerminal, repairInterruptedToolParts, resolveChatRecoveryConfig, resolveToolMergeId, runChatRecoveryExhaustion, sanitizeMessage, sendIfOpen, setChatRecovering, shouldCreditStreamProgress, sweepStaleChatRecoveryIncidents, toolApprovalUpdate, toolPartHasSettledResult, toolResultUpdate, unwrapChatFiberSnapshot, wrapChatFiberSnapshot };
 
 //# sourceMappingURL=index.js.map