@ably/ai-transport 0.0.1 → 0.2.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +114 -116
- package/dist/ably-ai-transport.js +1743 -961
- package/dist/ably-ai-transport.js.map +1 -1
- package/dist/ably-ai-transport.umd.cjs +1 -1
- package/dist/ably-ai-transport.umd.cjs.map +1 -1
- package/dist/constants.d.ts +117 -39
- package/dist/core/agent.d.ts +29 -0
- package/dist/core/codec/decoder.d.ts +20 -23
- package/dist/core/codec/encoder.d.ts +11 -8
- package/dist/core/codec/index.d.ts +1 -2
- package/dist/core/codec/lifecycle-tracker.d.ts +10 -9
- package/dist/core/codec/types.d.ts +410 -101
- package/dist/core/transport/agent-session.d.ts +10 -0
- package/dist/core/transport/branch-chain.d.ts +43 -0
- package/dist/core/transport/client-session.d.ts +13 -0
- package/dist/core/transport/decode-fold.d.ts +47 -0
- package/dist/core/transport/headers.d.ts +97 -17
- package/dist/core/transport/index.d.ts +5 -3
- package/dist/core/transport/internal/bounded-map.d.ts +20 -0
- package/dist/core/transport/invocation.d.ts +74 -0
- package/dist/core/transport/load-conversation.d.ts +128 -0
- package/dist/core/transport/load-history.d.ts +39 -0
- package/dist/core/transport/pipe-stream.d.ts +9 -8
- package/dist/core/transport/run-manager.d.ts +78 -0
- package/dist/core/transport/tree.d.ts +435 -0
- package/dist/core/transport/types/agent.d.ts +353 -0
- package/dist/core/transport/types/client.d.ts +168 -0
- package/dist/core/transport/types/shared.d.ts +24 -0
- package/dist/core/transport/types/tree.d.ts +315 -0
- package/dist/core/transport/types/view.d.ts +222 -0
- package/dist/core/transport/types.d.ts +13 -402
- package/dist/core/transport/view.d.ts +354 -0
- package/dist/errors.d.ts +37 -9
- package/dist/index.d.ts +6 -6
- package/dist/logger.d.ts +12 -0
- package/dist/react/ably-ai-transport-react.js +1164 -645
- package/dist/react/ably-ai-transport-react.js.map +1 -1
- package/dist/react/ably-ai-transport-react.umd.cjs +1 -1
- package/dist/react/ably-ai-transport-react.umd.cjs.map +1 -1
- package/dist/react/contexts/client-session-context.d.ts +36 -0
- package/dist/react/contexts/client-session-provider.d.ts +53 -0
- package/dist/react/create-session-hooks.d.ts +116 -0
- package/dist/react/index.d.ts +16 -10
- package/dist/react/internal/use-resolved-session.d.ts +36 -0
- package/dist/react/use-ably-messages.d.ts +20 -11
- package/dist/react/use-client-session.d.ts +81 -0
- package/dist/react/use-create-view.d.ts +23 -0
- package/dist/react/use-tree.d.ts +35 -0
- package/dist/react/use-view.d.ts +110 -0
- package/dist/utils.d.ts +32 -23
- package/dist/vercel/ably-ai-transport-vercel.js +2748 -1625
- package/dist/vercel/ably-ai-transport-vercel.js.map +1 -1
- package/dist/vercel/ably-ai-transport-vercel.umd.cjs +1 -1
- package/dist/vercel/ably-ai-transport-vercel.umd.cjs.map +1 -1
- package/dist/vercel/codec/decoder.d.ts +5 -18
- package/dist/vercel/codec/encoder.d.ts +6 -36
- package/dist/vercel/codec/events.d.ts +51 -0
- package/dist/vercel/codec/index.d.ts +24 -12
- package/dist/vercel/codec/reducer.d.ts +144 -0
- package/dist/vercel/codec/tool-transitions.d.ts +50 -0
- package/dist/vercel/index.d.ts +4 -2
- package/dist/vercel/react/ably-ai-transport-vercel-react.js +10298 -1410
- package/dist/vercel/react/ably-ai-transport-vercel-react.js.map +1 -1
- package/dist/vercel/react/ably-ai-transport-vercel-react.umd.cjs +70 -1
- package/dist/vercel/react/ably-ai-transport-vercel-react.umd.cjs.map +1 -1
- package/dist/vercel/react/contexts/chat-transport-context.d.ts +33 -0
- package/dist/vercel/react/contexts/chat-transport-provider.d.ts +96 -0
- package/dist/vercel/react/index.d.ts +4 -0
- package/dist/vercel/react/use-chat-transport.d.ts +66 -21
- package/dist/vercel/react/use-message-sync.d.ts +31 -12
- package/dist/vercel/run-end-reason.d.ts +29 -0
- package/dist/vercel/transport/chat-transport.d.ts +71 -30
- package/dist/vercel/transport/index.d.ts +25 -18
- package/dist/vercel/transport/run-output-stream.d.ts +56 -0
- package/dist/version.d.ts +2 -0
- package/package.json +47 -34
- package/src/constants.ts +126 -47
- package/src/core/agent.ts +68 -0
- package/src/core/codec/decoder.ts +71 -98
- package/src/core/codec/encoder.ts +115 -58
- package/src/core/codec/index.ts +13 -6
- package/src/core/codec/lifecycle-tracker.ts +10 -9
- package/src/core/codec/types.ts +438 -106
- package/src/core/transport/agent-session.ts +1344 -0
- package/src/core/transport/branch-chain.ts +58 -0
- package/src/core/transport/client-session.ts +775 -0
- package/src/core/transport/decode-fold.ts +91 -0
- package/src/core/transport/headers.ts +182 -19
- package/src/core/transport/index.ts +29 -22
- package/src/core/transport/internal/bounded-map.ts +27 -0
- package/src/core/transport/invocation.ts +98 -0
- package/src/core/transport/load-conversation.ts +355 -0
- package/src/core/transport/load-history.ts +269 -0
- package/src/core/transport/pipe-stream.ts +58 -40
- package/src/core/transport/run-manager.ts +249 -0
- package/src/core/transport/tree.ts +1167 -0
- package/src/core/transport/types/agent.ts +407 -0
- package/src/core/transport/types/client.ts +211 -0
- package/src/core/transport/types/shared.ts +27 -0
- package/src/core/transport/types/tree.ts +344 -0
- package/src/core/transport/types/view.ts +259 -0
- package/src/core/transport/types.ts +13 -527
- package/src/core/transport/view.ts +1271 -0
- package/src/errors.ts +42 -9
- package/src/event-emitter.ts +3 -2
- package/src/index.ts +55 -39
- package/src/logger.ts +14 -1
- package/src/react/contexts/client-session-context.ts +41 -0
- package/src/react/contexts/client-session-provider.tsx +186 -0
- package/src/react/create-session-hooks.ts +141 -0
- package/src/react/index.ts +27 -10
- package/src/react/internal/use-resolved-session.ts +63 -0
- package/src/react/use-ably-messages.ts +47 -19
- package/src/react/use-client-session.ts +201 -0
- package/src/react/use-create-view.ts +72 -0
- package/src/react/use-tree.ts +84 -0
- package/src/react/use-view.ts +275 -0
- package/src/react/vite.config.ts +4 -1
- package/src/utils.ts +63 -45
- package/src/vercel/codec/decoder.ts +336 -255
- package/src/vercel/codec/encoder.ts +348 -196
- package/src/vercel/codec/events.ts +87 -0
- package/src/vercel/codec/index.ts +59 -14
- package/src/vercel/codec/reducer.ts +977 -0
- package/src/vercel/codec/tool-transitions.ts +122 -0
- package/src/vercel/index.ts +7 -3
- package/src/vercel/react/contexts/chat-transport-context.ts +41 -0
- package/src/vercel/react/contexts/chat-transport-provider.tsx +150 -0
- package/src/vercel/react/index.ts +13 -1
- package/src/vercel/react/use-chat-transport.ts +162 -42
- package/src/vercel/react/use-message-sync.ts +121 -22
- package/src/vercel/react/vite.config.ts +4 -2
- package/src/vercel/run-end-reason.ts +78 -0
- package/src/vercel/transport/chat-transport.ts +553 -113
- package/src/vercel/transport/index.ts +40 -28
- package/src/vercel/transport/run-output-stream.ts +170 -0
- package/src/version.ts +2 -0
- package/dist/core/transport/client-transport.d.ts +0 -10
- package/dist/core/transport/conversation-tree.d.ts +0 -9
- package/dist/core/transport/decode-history.d.ts +0 -41
- package/dist/core/transport/server-transport.d.ts +0 -7
- package/dist/core/transport/stream-router.d.ts +0 -19
- package/dist/core/transport/turn-manager.d.ts +0 -34
- package/dist/react/use-active-turns.d.ts +0 -8
- package/dist/react/use-client-transport.d.ts +0 -7
- package/dist/react/use-conversation-tree.d.ts +0 -20
- package/dist/react/use-edit.d.ts +0 -7
- package/dist/react/use-history.d.ts +0 -19
- package/dist/react/use-messages.d.ts +0 -7
- package/dist/react/use-regenerate.d.ts +0 -7
- package/dist/react/use-send.d.ts +0 -7
- package/dist/vercel/codec/accumulator.d.ts +0 -21
- package/src/core/transport/client-transport.ts +0 -959
- package/src/core/transport/conversation-tree.ts +0 -434
- package/src/core/transport/decode-history.ts +0 -337
- package/src/core/transport/server-transport.ts +0 -458
- package/src/core/transport/stream-router.ts +0 -118
- package/src/core/transport/turn-manager.ts +0 -147
- package/src/react/use-active-turns.ts +0 -61
- package/src/react/use-client-transport.ts +0 -37
- package/src/react/use-conversation-tree.ts +0 -71
- package/src/react/use-edit.ts +0 -24
- package/src/react/use-history.ts +0 -111
- package/src/react/use-messages.ts +0 -32
- package/src/react/use-regenerate.ts +0 -24
- package/src/react/use-send.ts +0 -25
- package/src/vercel/codec/accumulator.ts +0 -603
|
@@ -1,2 +1,2 @@
|
|
|
1
|
-
(function(e,t){typeof exports==`object`&&typeof module<`u`?t(exports,require(`ably`),require(`ai`)):typeof define==`function`&&define.amd?define([`exports`,`ably`,`ai`],t):(e=typeof globalThis<`u`?globalThis:e||self,t(e.AblyAiTransportVercel={},e.Ably,e.AI))})(this,function(e,t,n){Object.defineProperty(e,Symbol.toStringTag,{value:`Module`});var r=Object.create,i=Object.defineProperty,a=Object.getOwnPropertyDescriptor,o=Object.getOwnPropertyNames,s=Object.getPrototypeOf,c=Object.prototype.hasOwnProperty,l=(e,t,n,r)=>{if(t&&typeof t==`object`||typeof t==`function`)for(var s=o(t),l=0,u=s.length,d;l<u;l++)d=s[l],!c.call(e,d)&&d!==n&&i(e,d,{get:(e=>t[e]).bind(null,d),enumerable:!(r=a(t,d))||r.enumerable});return e};t=((e,t,n)=>(n=e==null?{}:r(s(e)),l(t||!e||!e.__esModule?i(n,`default`,{value:e,enumerable:!0}):n,e)))(t);var u=`x-ably-stream`,d=`x-ably-status`,f=`x-ably-stream-id`,p=`x-ably-turn-id`,m=`x-ably-msg-id`,h=`x-ably-turn-client-id`,g=`x-ably-role`,_=`x-ably-cancel-turn-id`,v=`x-ably-cancel-own`,y=`x-ably-cancel-all`,b=`x-ably-cancel-client-id`,x=`x-ably-parent`,ee=`x-ably-fork-of`,S=`x-ably-turn-reason`,C=`x-ably-cancel`,w=`x-ably-turn-start`,T=`x-ably-turn-end`,E=`x-domain-`,D=e=>{let t=e.extras;if(!t||typeof t!=`object`)return{};let n=t.headers;return!n||typeof n!=`object`?{}:n},te=e=>{if(e!==void 0)try{return JSON.parse(e)}catch{return}},O=(e,t)=>({...e,...t}),k=e=>{if(e!==void 0)return e===`true`},A=(e,t)=>e[E+t],j=e=>{let t={};for(let n in e)Object.prototype.hasOwnProperty.call(e,n)&&e[n]!==void 0&&(t[n]=e[n]);return t},ne=e=>({str:t=>A(e,t),strOr:(t,n)=>A(e,t)??n,bool:t=>k(A(e,t)),json:t=>te(A(e,t))}),M=()=>{let e={},t={str:(n,r)=>(r!==void 0&&(e[E+n]=r),t),bool:(n,r)=>(r!==void 0&&(e[E+n]=String(r)),t),json:(n,r)=>(r!=null&&(e[E+n]=JSON.stringify(r)),t),build:()=>e};return t},N=e=>j({type:`dynamic-tool`,toolCallId:e.toolCallId,toolName:e.toolName,title:e.title,providerExecuted:e.providerExecuted}),P=class{constructor(e){this._activeIndex=new Map,this._partType=e}start(e,t,n){this._activeIndex.set(e,t.parts.length),t.parts.push({type:this._partType,text:``}),n.set(e,`streaming`)}delta(e,t,n){let r=this._activeIndex.get(e);if(r===void 0)return;let i=t.parts[r];i?.type===this._partType&&(i.text+=n)}end(e,t){t.set(e,`finished`),this._activeIndex.delete(e)}reset(){this._activeIndex=new Map}},re=class{constructor(){this._messageList=[],this._activeMessages=new Map}get messages(){return this._messageList}get completedMessages(){let e=new Set;for(let t of this._activeMessages.values())e.add(t.message);return this._messageList.filter(t=>!e.has(t))}get hasActiveStream(){for(let e of this._activeMessages.values())for(let t of e.streamStatus.values())if(t===`streaming`)return!0;return!1}processOutputs(e){for(let t of e)t.kind===`message`?this._messageList.push(t.message):t.messageId!==void 0&&this._processEvent(t.event,t.messageId)}updateMessage(e){let t=this._messageList.findIndex(t=>t.id===e.id);t!==-1&&(this._messageList[t]=e)}_ensureActiveMessage(e){let t=this._activeMessages.get(e);if(t)return t;let n={message:{id:e,role:`assistant`,parts:[]},textStreams:new P(`text`),reasoningStreams:new P(`reasoning`),toolTrackers:{},streamStatus:new Map};return this._activeMessages.set(e,n),this._messageList.push(n.message),n}_getToolPart(e,t){let n=t.toolTrackers[e];if(!n)return;let r=t.message.parts[n.partIndex];if(r?.type===`dynamic-tool`)return{tracker:n,part:r}}_processEvent(e,t){switch(e.type){case`start`:case`start-step`:case`finish-step`:case`finish`:case`abort`:case`error`:case`message-metadata`:this._processLifecycle(e,t);break;case`text-start`:case`text-delta`:case`text-end`:case`reasoning-start`:case`reasoning-delta`:case`reasoning-end`:this._processTextOrReasoning(e,t);break;case`tool-input-start`:case`tool-input-delta`:case`tool-input-available`:case`tool-input-error`:this._processToolInput(e,t);break;case`tool-output-available`:case`tool-output-error`:case`tool-output-denied`:case`tool-approval-request`:this._processToolOutput(e,t);break;case`file`:case`source-url`:case`source-document`:this._processContentPart(e,t);break;default:if(e.type.startsWith(`data-`)){if(e.transient)break;let n=this._ensureActiveMessage(t),r=j({type:e.type,id:e.id,data:e.data});if(e.id!==void 0){let t=n.message.parts.findIndex(t=>t.type===e.type&&`id`in t&&t.id===e.id);if(t!==-1){n.message.parts[t]=r;break}}n.message.parts.push(r)}break}}_processLifecycle(e,t){switch(e.type){case`start`:{let n=this._ensureActiveMessage(t);e.messageId&&(n.message.id=e.messageId),e.messageMetadata!==void 0&&(n.message.metadata=e.messageMetadata);break}case`start-step`:this._ensureActiveMessage(t).message.parts.push({type:`step-start`});break;case`finish-step`:{let e=this._activeMessages.get(t);e&&(e.textStreams.reset(),e.reasoningStreams.reset());break}case`finish`:{let n=this._activeMessages.get(t);n&&e.messageMetadata!==void 0&&(n.message.metadata=e.messageMetadata),this._activeMessages.delete(t);break}case`abort`:{let e=this._activeMessages.get(t);if(e)for(let[t,n]of e.streamStatus)n===`streaming`&&e.streamStatus.set(t,`aborted`);this._activeMessages.delete(t);break}case`error`:break;case`message-metadata`:{let n=this._activeMessages.get(t);n&&e.messageMetadata!==void 0&&(n.message.metadata=e.messageMetadata);break}}}_processTextOrReasoning(e,t){let n=this._ensureActiveMessage(t);switch(e.type){case`text-start`:n.textStreams.start(e.id,n.message,n.streamStatus);break;case`text-delta`:n.textStreams.delta(e.id,n.message,e.delta);break;case`text-end`:n.textStreams.end(e.id,n.streamStatus);break;case`reasoning-start`:n.reasoningStreams.start(e.id,n.message,n.streamStatus);break;case`reasoning-delta`:n.reasoningStreams.delta(e.id,n.message,e.delta);break;case`reasoning-end`:n.reasoningStreams.end(e.id,n.streamStatus);break}}_processToolInput(e,t){switch(e.type){case`tool-input-start`:{let n=this._ensureActiveMessage(t),r=n.message.parts.length;n.message.parts.push({...N(e),state:`input-streaming`,input:void 0}),n.toolTrackers[e.toolCallId]={partIndex:r,inputText:``},n.streamStatus.set(e.toolCallId,`streaming`);break}case`tool-input-delta`:{let n=this._ensureActiveMessage(t),r=n.toolTrackers[e.toolCallId];if(!r)break;r.inputText+=e.inputTextDelta;let i;try{i=JSON.parse(r.inputText)}catch{i=void 0}let a=this._getToolPart(e.toolCallId,n);if(!a)break;n.message.parts[a.tracker.partIndex]={...N(a.part),state:`input-streaming`,input:i};break}case`tool-input-available`:{let n=this._ensureActiveMessage(t),r=this._getToolPart(e.toolCallId,n);if(!r)break;n.message.parts[r.tracker.partIndex]={...N(r.part),state:`input-available`,input:e.input},n.streamStatus.set(e.toolCallId,`finished`);break}case`tool-input-error`:{let n=this._ensureActiveMessage(t),r=this._getToolPart(e.toolCallId,n);if(r)n.message.parts[r.tracker.partIndex]={...N(r.part),state:`output-error`,input:e.input,errorText:e.errorText};else{let t=n.message.parts.length;n.message.parts.push({...N(e),state:`output-error`,input:e.input,errorText:e.errorText}),n.toolTrackers[e.toolCallId]={partIndex:t,inputText:``}}n.streamStatus.set(e.toolCallId,`finished`);break}}}_processToolOutput(e,t){let n=this._ensureActiveMessage(t),r=this._getToolPart(e.toolCallId,n);if(r)switch(e.type){case`tool-output-available`:n.message.parts[r.tracker.partIndex]=j({...N(r.part),state:`output-available`,input:r.part.input,output:e.output,preliminary:e.preliminary});break;case`tool-output-error`:n.message.parts[r.tracker.partIndex]={...N(r.part),state:`output-error`,input:r.part.input,errorText:e.errorText};break;case`tool-output-denied`:n.message.parts[r.tracker.partIndex]={...N(r.part),state:`output-denied`,input:r.part.input,approval:{id:``,approved:!1}};break;case`tool-approval-request`:n.message.parts[r.tracker.partIndex]={...N(r.part),state:`approval-requested`,input:r.part.input,approval:{id:e.approvalId}};break}}_processContentPart(e,t){let n=this._ensureActiveMessage(t);switch(e.type){case`file`:n.message.parts.push({type:`file`,mediaType:e.mediaType,url:e.url});break;case`source-url`:n.message.parts.push(j({type:`source-url`,sourceId:e.sourceId,url:e.url,title:e.title}));break;case`source-document`:n.message.parts.push(j({type:`source-document`,sourceId:e.sourceId,mediaType:e.mediaType,title:e.title,filename:e.filename}));break}}},ie=()=>new re,ae=e=>[{kind:`event`,event:e}],oe=class{constructor(e,t={}){this._serialState=new Map,this._hooks=e,this._onStreamUpdate=t.onStreamUpdate,this._onStreamDelete=t.onStreamDelete,this._logger=t.logger?.withContext({component:`DecoderCore`})}decode(e){let t=e.action;this._logger?.trace(`DefaultDecoderCore.decode();`,{action:t,serial:e.serial,name:e.name});let n;switch(t){case`message.create`:{let t=this._toPayload(e);n=t.headers?.[`x-ably-stream`]===`true`?this._decodeStreamedCreate(t,e.serial):this._hooks.decodeDiscrete(t);break}case`message.append`:n=this._decodeAppend(e);break;case`message.update`:n=this._decodeUpdate(e);break;case`message.delete`:n=this._decodeDelete(e);break;default:return[]}let r=D(e)[m];if(r)for(let e of n)e.kind===`event`&&(e.messageId=r);return n}_toPayload(e){return{name:e.name??``,data:e.data,headers:D(e)}}_stringData(e){return typeof e.data==`string`?e.data:``}_invokeOnStreamUpdate(e){if(this._onStreamUpdate)try{this._onStreamUpdate(e)}catch(e){this._logger?.error(`DefaultDecoderCore._invokeOnStreamUpdate(); callback threw`,{error:e})}}_invokeOnStreamDelete(e,t){if(this._onStreamDelete)try{this._onStreamDelete(e,t)}catch(e){this._logger?.error(`DefaultDecoderCore._invokeOnStreamDelete(); callback threw`,{error:e})}}_decodeStreamedCreate(e,t){if(!t)return[];let n=e.headers?.[`x-ably-stream-id`]??``,r=e.headers??{},i={name:e.name,streamId:n,accumulated:``,headers:{...r},closed:!1};return this._serialState.set(t,i),this._logger?.debug(`DefaultDecoderCore._decodeStreamedCreate(); new stream`,{name:e.name,streamId:n,serial:t}),this._hooks.buildStartEvents(i)}_decodeAppend(e){let t=e.serial;if(!t)return[];let n=this._serialState.get(t);if(!n)return this._decodeUpdate(e);let r=D(e),i=typeof e.data==`string`?e.data:``,a=r[d],o=[];return i.length>0&&(n.accumulated+=i,o.push(...this._hooks.buildDeltaEvents(n,i))),a===`finished`&&!n.closed?(n.closed=!0,o.push(...this._hooks.buildEndEvents(n,r)),this._logger?.debug(`DefaultDecoderCore._decodeAppend(); stream finished`,{streamId:n.streamId})):a===`aborted`&&!n.closed&&(n.closed=!0,this._logger?.debug(`DefaultDecoderCore._decodeAppend(); stream aborted`,{streamId:n.streamId})),o}_decodeUpdate(e){let t=e.serial;if(!t)return[];let n=this._toPayload(e),r=n.headers??{},i=r[u]===`true`,a=r[d],o=this._serialState.get(t);if(!o)return this._decodeFirstContact(n,i,a,t);let s=this._stringData(e);if(s.startsWith(o.accumulated)){let e=s.slice(o.accumulated.length),t=[];return e.length>0&&(o.accumulated=s,t.push(...this._hooks.buildDeltaEvents(o,e))),a===`finished`&&!o.closed?(o.closed=!0,t.push(...this._hooks.buildEndEvents(o,r))):a===`aborted`&&!o.closed&&(o.closed=!0),t}return o.accumulated=s,o.headers={...r},this._invokeOnStreamUpdate(o),[]}_decodeFirstContact(e,t,n,r){if(!t)return this._hooks.decodeDiscrete(e);let i=e.headers?.[`x-ably-stream-id`]??``,a=e.headers??{},o=typeof e.data==`string`?e.data:``;this._logger?.debug(`DefaultDecoderCore._decodeFirstContact(); first-contact stream`,{name:e.name,streamId:i,serial:r});let s={name:e.name,streamId:i,accumulated:o,headers:{...a},closed:n===`finished`||n===`aborted`};this._serialState.set(r,s);let c=this._hooks.buildStartEvents(s);return o.length>0&&c.push(...this._hooks.buildDeltaEvents(s,o)),n===`finished`&&c.push(...this._hooks.buildEndEvents(s,a)),c}_decodeDelete(e){let t=e.serial;if(!t)return[];let n=this._serialState.get(t);return this._invokeOnStreamDelete(t,n),n&&(n.accumulated=``,n.closed=!0),this._logger?.debug(`DefaultDecoderCore._decodeDelete();`,{serial:t}),[]}},se=(e,t={})=>new oe(e,t),ce=class{constructor(e){this._emitted=new Map,this._phases=e}ensurePhases(e,t){let n=this._getOrCreate(e),r=[];for(let e of this._phases)n.has(e.key)||(n.add(e.key),r.push(...e.build(t)));return r}markEmitted(e,t){this._getOrCreate(e).add(t)}resetPhase(e,t){this._emitted.get(e)?.delete(t)}clearScope(e){this._emitted.delete(e)}_getOrCreate(e){let t=this._emitted.get(e);return t||(t=new Set,this._emitted.set(e,t)),t}},le=e=>new ce(e),F=e=>{let t=ne(e);return{...t,providerMetadata:()=>t.json(`providerMetadata`)}},I=e=>ae(e),ue=(e,t)=>e===`stop`||e===`length`||e===`content-filter`||e===`tool-calls`||e===`error`||e===`other`?e:t,L=e=>e.startsWith(`data-`),R=e=>{if(e)try{return JSON.parse(e)}catch{return e}},z=e=>{let t=F(e.headers);switch(e.name){case`text`:return j({type:`text-start`,id:e.streamId,providerMetadata:t.providerMetadata()});case`reasoning`:return j({type:`reasoning-start`,id:e.streamId,providerMetadata:t.providerMetadata()});case`tool-input`:return j({type:`tool-input-start`,toolCallId:e.streamId,toolName:t.strOr(`toolName`,``),dynamic:t.bool(`dynamic`),title:t.str(`title`),providerExecuted:t.bool(`providerExecuted`),providerMetadata:t.providerMetadata()});default:return{type:`text-start`,id:e.streamId}}},B=(e,t)=>{switch(e.name){case`text`:return{type:`text-delta`,id:e.streamId,delta:t};case`reasoning`:return{type:`reasoning-delta`,id:e.streamId,delta:t};case`tool-input`:return{type:`tool-input-delta`,toolCallId:e.streamId,inputTextDelta:t};default:return{type:`text-delta`,id:e.streamId,delta:t}}},V=(e,t)=>{let n=F(t);switch(e.name){case`text`:return j({type:`text-end`,id:e.streamId,providerMetadata:n.providerMetadata()});case`reasoning`:return j({type:`reasoning-end`,id:e.streamId,providerMetadata:n.providerMetadata()});case`tool-input`:return j({type:`tool-input-available`,toolCallId:e.streamId,toolName:n.strOr(`toolName`,F(e.headers).strOr(`toolName`,``)),input:R(e.accumulated),providerMetadata:n.providerMetadata()});default:return{type:`text-end`,id:e.streamId}}},H=()=>le([{key:`start`,build:e=>[j({type:`start`,messageId:e.messageId})]},{key:`start-step`,build:()=>[{type:`start-step`}]}]),U=(e,t,n)=>e.ensurePhases(t,n).map(e=>({kind:`event`,event:e})),de=(e,t,n)=>(n.markEmitted(t,`start`),I(j({type:`start`,messageId:e.str(`messageId`),messageMetadata:e.json(`messageMetadata`)}))),fe=(e,t)=>(t.markEmitted(e,`start-step`),I({type:`start-step`})),pe=(e,t)=>(t.resetPhase(e,`start-step`),I({type:`finish-step`})),me=(e,t,n)=>(n.clearScope(t),I(j({type:`finish`,finishReason:ue(e.str(`finishReason`),`stop`),messageMetadata:e.json(`messageMetadata`)}))),he=e=>I({type:`error`,errorText:typeof e==`string`?e:``}),ge=(e,t,n)=>(n.clearScope(t),I(j({type:`abort`,reason:typeof e==`string`&&e?e:void 0}))),_e=e=>I({type:`message-metadata`,messageMetadata:e.json(`messageMetadata`)}),ve=(e,t)=>I(j({type:`file`,url:typeof t==`string`?t:``,mediaType:e.strOr(`mediaType`,``),providerMetadata:e.providerMetadata()})),ye=(e,t)=>I(j({type:`source-url`,sourceId:e.strOr(`sourceId`,``),url:typeof t==`string`?t:``,title:e.str(`title`),providerMetadata:e.providerMetadata()})),be=e=>I(j({type:`source-document`,sourceId:e.strOr(`sourceId`,``),mediaType:e.strOr(`mediaType`,``),title:e.strOr(`title`,``),filename:e.str(`filename`),providerMetadata:e.providerMetadata()})),xe=(e,t)=>{let n=t;return I(j({type:`tool-input-error`,toolCallId:e.strOr(`toolCallId`,``),toolName:e.strOr(`toolName`,``),errorText:n?.errorText??``,input:n?.input,dynamic:e.bool(`dynamic`),title:e.str(`title`),providerExecuted:e.bool(`providerExecuted`),providerMetadata:e.providerMetadata()}))},Se=(e,t)=>{let n=t;return I(j({type:`tool-output-available`,toolCallId:e.strOr(`toolCallId`,``),output:n?.output,dynamic:e.bool(`dynamic`),providerExecuted:e.bool(`providerExecuted`),preliminary:e.bool(`preliminary`)}))},Ce=(e,t)=>{let n=t;return I(j({type:`tool-output-error`,toolCallId:e.strOr(`toolCallId`,``),errorText:n?.errorText??``,dynamic:e.bool(`dynamic`),providerExecuted:e.bool(`providerExecuted`)}))},we=e=>I({type:`tool-approval-request`,toolCallId:e.strOr(`toolCallId`,``),approvalId:e.strOr(`approvalId`,``)}),Te=e=>I({type:`tool-output-denied`,toolCallId:e.strOr(`toolCallId`,``)}),Ee=(e,t,n)=>I(j({type:e,data:n,id:t.str(`id`),transient:t.bool(`transient`)})),De=(e,t,n,r)=>{let i=U(r,n,{messageId:e.str(`messageId`)});return i.push({kind:`event`,event:j({type:`tool-input-start`,toolCallId:e.strOr(`toolCallId`,``),toolName:e.strOr(`toolName`,``),dynamic:e.bool(`dynamic`),title:e.str(`title`),providerExecuted:e.bool(`providerExecuted`),providerMetadata:e.providerMetadata()})},{kind:`event`,event:j({type:`tool-input-available`,toolCallId:e.strOr(`toolCallId`,``),toolName:e.strOr(`toolName`,``),input:t,providerMetadata:e.providerMetadata()})}),i},Oe=e=>{let t=e.headers??{},n=F(t),r=t[`x-ably-role`]??`user`,i=n.str(`messageId`)??``,a;switch(e.name){case`text`:a={type:`text`,text:typeof e.data==`string`?e.data:``};break;case`file`:a={type:`file`,mediaType:n.strOr(`mediaType`,``),url:typeof e.data==`string`?e.data:``};break;default:L(e.name)&&(a=j({type:e.name,id:n.str(`id`),data:e.data}));break}return a?[{kind:`message`,message:{id:i,role:r,parts:[a]}}]:[]},ke=(e,t)=>(e===`text`||e===`file`||L(e))&&`x-ably-role`in t,Ae=(e,t)=>{let n=e.headers??{},r=F(n),i=n[`x-ably-turn-id`]??``;if(ke(e.name,n))return Oe(e);if(e.name===`tool-input`)return De(r,e.data,i,t);switch(e.name){case`start`:return de(r,i,t);case`start-step`:return fe(i,t);case`finish-step`:return pe(i,t);case`finish`:return me(r,i,t);case`error`:return he(e.data);case`abort`:return ge(e.data,i,t);case`message-metadata`:return _e(r);case`file`:return ve(r,e.data);case`source-url`:return ye(r,e.data);case`source-document`:return be(r);case`tool-input-error`:return xe(r,e.data);case`tool-output-available`:return Se(r,e.data);case`tool-output-error`:return Ce(r,e.data);case`tool-approval-request`:return we(r);case`tool-output-denied`:return Te(r);default:return L(e.name)?Ee(e.name,r,e.data):[]}},je=e=>({buildStartEvents:t=>{let n=U(e,t.headers[`x-ably-turn-id`]??``,{messageId:F(t.headers).str(`messageId`)});return n.push({kind:`event`,event:z(t)}),n},buildDeltaEvents:(e,t)=>I(B(e,t)),buildEndEvents:(e,t)=>I(V(e,t)),decodeDiscrete:t=>Ae(t,e)}),Me=class{constructor(e={}){this._core=se(je(H()),e)}decode(e){return this._core.decode(e)}},Ne=(e={})=>new Me(e),W=function(e){return e[e.BadRequest=4e4]=`BadRequest`,e[e.InvalidArgument=40003]=`InvalidArgument`,e[e.EncoderRecoveryFailed=104e3]=`EncoderRecoveryFailed`,e[e.TransportSubscriptionError=104001]=`TransportSubscriptionError`,e[e.CancelListenerError=104002]=`CancelListenerError`,e[e.TurnLifecycleError=104003]=`TurnLifecycleError`,e[e.TransportClosed=104004]=`TransportClosed`,e[e.TransportSendFailed=104005]=`TransportSendFailed`,e}({}),Pe=(e,t)=>e.code===t,Fe=class{constructor(e,t={}){this._trackers=new Map,this._pending=[],this._closed=!1,this._writer=e,this._defaultClientId=t.clientId,this._defaultExtras=t.extras,this._onMessageHook=t.onMessage??(()=>{}),this._logger=t.logger?.withContext({component:`EncoderCore`})}async publishDiscrete(e,t){this._assertNotClosed(),this._logger?.trace(`DefaultEncoderCore.publishDiscrete();`,{name:e.name});let n=this._buildDiscreteMessage(e,t);return this._writer.publish(n)}async publishDiscreteBatch(e,t){this._assertNotClosed(),this._logger?.trace(`DefaultEncoderCore.publishDiscreteBatch();`,{count:e.length});let n=e.map(e=>this._buildDiscreteMessage(e,t));return this._writer.publish(n)}async startStream(e,n,r){this._assertNotClosed(),this._logger?.trace(`DefaultEncoderCore.startStream();`,{name:n.name,streamId:e});let i=this._buildHeaders(n.headers??{},r);i[u]=`true`,i[d]=`streaming`,i[f]=e;let a=this._resolveClientId(r),o={name:n.name,data:n.data,extras:{headers:i},...a?{clientId:a}:{}};this._invokeOnMessage(o);let s=(await this._writer.publish(o)).serials[0];if(!s)throw new t.ErrorInfo(`unable to start stream; no serial returned for stream '${n.name}' (streamId: ${e})`,W.BadRequest,400);this._trackers.set(e,{serial:s,name:n.name,streamId:e,accumulated:n.data,persistentHeaders:i,aborted:!1}),this._logger?.debug(`DefaultEncoderCore.startStream(); stream started`,{name:n.name,streamId:e,serial:s})}appendStream(e,n){this._assertNotClosed();let r=this._trackers.get(e);if(!r)throw new t.ErrorInfo(`unable to append to stream; no active stream for streamId '${e}'`,W.InvalidArgument,400);r.accumulated+=n;let i={serial:r.serial,data:n,extras:{headers:{...r.persistentHeaders}}};this._invokeOnMessage(i);let a=this._writer.appendMessage(i);this._pending.push({promise:a,streamId:e})}async closeStream(e,n){this._assertNotClosed(),this._logger?.trace(`DefaultEncoderCore.closeStream();`,{streamId:e});let r=this._trackers.get(e);if(!r)throw new t.ErrorInfo(`unable to close stream; no active stream for streamId '${e}'`,W.InvalidArgument,400);r.accumulated+=n.data;let i=this._buildClosingHeaders(r,n.headers??{});i[d]=`finished`;let a={serial:r.serial,data:n.data,extras:{headers:i}};this._invokeOnMessage(a);let o=this._writer.appendMessage(a);this._pending.push({promise:o,streamId:e}),await this._flushPending(),this._logger?.debug(`DefaultEncoderCore.closeStream(); stream closed`,{streamId:e})}async abortStream(e,n){this._assertNotClosed(),this._logger?.trace(`DefaultEncoderCore.abortStream();`,{streamId:e});let r=this._trackers.get(e);if(!r)throw new t.ErrorInfo(`unable to abort stream; no active stream for streamId '${e}'`,W.InvalidArgument,400);r.aborted=!0;let i=this._buildClosingHeaders(r,{},n);i[d]=`aborted`;let a={serial:r.serial,data:``,extras:{headers:i}};this._invokeOnMessage(a);let o=this._writer.appendMessage(a);this._pending.push({promise:o,streamId:e}),await this._flushPending(),this._logger?.debug(`DefaultEncoderCore.abortStream(); stream aborted`,{streamId:e})}async abortAllStreams(e){this._assertNotClosed(),this._logger?.trace(`DefaultEncoderCore.abortAllStreams();`,{streamCount:this._trackers.size});for(let t of this._trackers.values()){t.aborted=!0;let n=this._buildClosingHeaders(t,{},e);n[d]=`aborted`;let r={serial:t.serial,data:``,extras:{headers:n}};this._invokeOnMessage(r);let i=this._writer.appendMessage(r);this._pending.push({promise:i,streamId:t.streamId})}await this._flushPending()}async _flushPending(){if(this._flushPromise)return this._flushPromise;let e=this._pending;if(this._pending=[],e.length!==0){this._logger?.trace(`DefaultEncoderCore._flushPending();`,{count:e.length}),this._flushPromise=this._doFlush(e);try{await this._flushPromise}finally{this._flushPromise=void 0}}}async _doFlush(e){let n=await Promise.allSettled(e.map(async e=>e.promise)),r=new Set;for(let[t,i]of n.entries()){let n=e[t];n&&i.status===`rejected`&&r.add(n.streamId)}if(r.size===0){this._logger?.debug(`DefaultEncoderCore._flushPending(); all appends succeeded`);return}this._logger?.warn(`DefaultEncoderCore._flushPending(); recovering failed appends`,{failedStreams:[...r]});let i=[];for(let e of r){let t=this._trackers.get(e);if(!t)continue;let n=t.aborted?`aborted`:`finished`,r={serial:t.serial,data:t.accumulated,extras:{headers:{...t.persistentHeaders,[d]:n}}};try{await this._writer.updateMessage(r)}catch(t){i.push({streamId:e,error:t})}}if(i.length>0){let e=i.map(e=>e.streamId).join(`, `);throw this._logger?.error(`DefaultEncoderCore._flushPending(); recovery failed`,{failedStreams:e}),new t.ErrorInfo(`unable to flush pending appends; recovery failed for stream(s): ${e}`,W.EncoderRecoveryFailed,500)}}async close(){if(!this._closed){this._logger?.trace(`DefaultEncoderCore.close();`),this._closed=!0;try{await this._flushPending()}finally{this._trackers.clear()}this._logger?.debug(`DefaultEncoderCore.close(); encoder closed`)}}_invokeOnMessage(e){try{this._onMessageHook(e)}catch(e){this._logger?.error(`DefaultEncoderCore._invokeOnMessage(); hook threw`,{error:e})}}_assertNotClosed(){if(this._closed)throw new t.ErrorInfo(`unable to write to encoder; encoder has been closed`,W.InvalidArgument,400)}_resolveClientId(e){return e?.clientId??this._defaultClientId}_buildHeaders(e,t){let n={...O(this._defaultExtras?.headers,t?.extras?.headers),...e};return t?.messageId!==void 0&&(n[m]=t.messageId),n}_buildDiscreteMessage(e,t){let n=this._buildHeaders(e.headers??{},t);n[u]=`false`;let r=this._resolveClientId(t),i={name:e.name,data:e.data,extras:{headers:n,...e.ephemeral?{ephemeral:!0}:{}},...r?{clientId:r}:{}};return this._invokeOnMessage(i),i}_buildClosingHeaders(e,t,n){let r={...e.persistentHeaders},i=O(this._defaultExtras?.headers,n?.extras?.headers);return Object.assign(r,i),Object.assign(r,t),r}},Ie=(e,t={})=>new Fe(e,t),Le=class{constructor(e,t={}){this._aborted=!1,this._core=Ie(e,t)}async appendEvent(e,n){switch(e.type){case`text-start`:{let t=M().str(`id`,e.id).json(`providerMetadata`,e.providerMetadata).build();await this._core.startStream(e.id,{name:`text`,data:``,headers:t},n);break}case`reasoning-start`:{let t=M().str(`id`,e.id).json(`providerMetadata`,e.providerMetadata).build();await this._core.startStream(e.id,{name:`reasoning`,data:``,headers:t},n);break}case`tool-input-start`:{let t=M().str(`toolCallId`,e.toolCallId).str(`toolName`,e.toolName).bool(`dynamic`,e.dynamic).str(`title`,e.title).bool(`providerExecuted`,e.providerExecuted).json(`providerMetadata`,e.providerMetadata).build();await this._core.startStream(e.toolCallId,{name:`tool-input`,data:``,headers:t},n);break}case`text-delta`:this._core.appendStream(e.id,e.delta);break;case`reasoning-delta`:this._core.appendStream(e.id,e.delta);break;case`tool-input-delta`:this._core.appendStream(e.toolCallId,e.inputTextDelta);break;case`text-end`:{let t=M().str(`id`,e.id).json(`providerMetadata`,e.providerMetadata).build();await this._core.closeStream(e.id,{name:`text`,data:``,headers:t});break}case`reasoning-end`:{let t=M().str(`id`,e.id).json(`providerMetadata`,e.providerMetadata).build();await this._core.closeStream(e.id,{name:`reasoning`,data:``,headers:t});break}case`tool-input-available`:try{let t=M().str(`toolCallId`,e.toolCallId).str(`toolName`,e.toolName).json(`providerMetadata`,e.providerMetadata).build();await this._core.closeStream(e.toolCallId,{name:`tool-input`,data:``,headers:t})}catch(n){if(!(n instanceof t.ErrorInfo&&Pe(n,W.InvalidArgument)))throw n;let r=M().str(`toolCallId`,e.toolCallId).str(`toolName`,e.toolName).bool(`dynamic`,e.dynamic).str(`title`,e.title).bool(`providerExecuted`,e.providerExecuted).json(`providerMetadata`,e.providerMetadata).build();await this._core.publishDiscrete({name:`tool-input`,data:e.input,headers:r})}break;case`start`:{let t=M().str(`messageId`,e.messageId).json(`messageMetadata`,e.messageMetadata).build();await this._core.publishDiscrete({name:`start`,data:``,headers:t},n);break}case`start-step`:await this._core.publishDiscrete({name:`start-step`,data:``},n);break;case`finish-step`:await this._core.publishDiscrete({name:`finish-step`,data:``},n);break;case`finish`:{let t=M().str(`finishReason`,e.finishReason).json(`messageMetadata`,e.messageMetadata).build();await this._core.publishDiscrete({name:`finish`,data:``,headers:t},n);break}case`error`:await this._core.publishDiscrete({name:`error`,data:e.errorText},n);break;case`abort`:this._aborted=!0,await this._core.abortAllStreams(n),await this._core.publishDiscrete({name:`abort`,data:e.reason??``,headers:{[d]:`aborted`}},n);break;case`tool-input-error`:{let t=M().str(`toolCallId`,e.toolCallId).str(`toolName`,e.toolName).bool(`dynamic`,e.dynamic).str(`title`,e.title).bool(`providerExecuted`,e.providerExecuted).json(`providerMetadata`,e.providerMetadata).build();await this._core.publishDiscrete({name:`tool-input-error`,data:{errorText:e.errorText,input:e.input},headers:t});break}case`tool-output-available`:{let t=M().str(`toolCallId`,e.toolCallId).bool(`dynamic`,e.dynamic).bool(`providerExecuted`,e.providerExecuted).bool(`preliminary`,e.preliminary).build();await this._core.publishDiscrete({name:`tool-output-available`,data:{output:e.output},headers:t});break}case`tool-output-error`:{let t=M().str(`toolCallId`,e.toolCallId).bool(`dynamic`,e.dynamic).bool(`providerExecuted`,e.providerExecuted).build();await this._core.publishDiscrete({name:`tool-output-error`,data:{errorText:e.errorText},headers:t});break}case`tool-approval-request`:{let t=M().str(`toolCallId`,e.toolCallId).str(`approvalId`,e.approvalId).build();await this._core.publishDiscrete({name:`tool-approval-request`,data:``,headers:t},n);break}case`tool-output-denied`:{let t=M().str(`toolCallId`,e.toolCallId).build();await this._core.publishDiscrete({name:`tool-output-denied`,data:``,headers:t},n);break}case`file`:{let t=M().str(`mediaType`,e.mediaType).json(`providerMetadata`,e.providerMetadata).build();await this._core.publishDiscrete({name:`file`,data:e.url,headers:t},n);break}case`source-url`:{let t=M().str(`sourceId`,e.sourceId).str(`title`,e.title).json(`providerMetadata`,e.providerMetadata).build();await this._core.publishDiscrete({name:`source-url`,data:e.url,headers:t},n);break}case`source-document`:{let t=M().str(`sourceId`,e.sourceId).str(`mediaType`,e.mediaType).str(`title`,e.title).str(`filename`,e.filename).json(`providerMetadata`,e.providerMetadata).build();await this._core.publishDiscrete({name:`source-document`,data:``,headers:t},n);break}case`message-metadata`:{let t=M().json(`messageMetadata`,e.messageMetadata).build();await this._core.publishDiscrete({name:`message-metadata`,data:``,headers:t},n);break}default:if(e.type.startsWith(`data-`)){let t=M().str(`id`,e.id).bool(`transient`,e.transient).build(),r=e.transient===!0;await this._core.publishDiscrete({name:e.type,data:e.data,headers:t,ephemeral:r},n)}break}}async writeEvent(e,n){if(!e.type.startsWith(`data-`))throw new t.ErrorInfo(`unable to write event; only data-* chunk types are supported, got '${e.type}'`,W.InvalidArgument,400);let r=M().str(`id`,`id`in e?e.id:void 0).bool(`transient`,`transient`in e?e.transient:void 0).build(),i=`transient`in e&&e.transient===!0;return this._core.publishDiscrete({name:e.type,data:`data`in e?e.data:void 0,headers:r,ephemeral:i},n)}async writeMessages(e,t){let n=e.flatMap(e=>Re(e));return this._core.publishDiscreteBatch(n,t)}async abort(e){this._aborted||(this._aborted=!0,await this._core.abortAllStreams(),await this._core.publishDiscrete({name:`abort`,data:e??``,headers:{[d]:`aborted`}}))}async close(){await this._core.close()}},Re=e=>{let t=e.id,r=[];for(let i of e.parts)switch(i.type){case`text`:r.push({name:`text`,data:i.text,headers:M().str(`messageId`,t).build()});break;case`file`:r.push({name:`file`,data:i.url,headers:M().str(`messageId`,t).str(`mediaType`,i.mediaType).build()});break;default:(0,n.isDataUIPart)(i)&&r.push({name:i.type,data:i.data,headers:M().str(`messageId`,t).str(`id`,i.id).build()});break}return r.length===0&&r.push({name:`text`,data:``,headers:M().str(`messageId`,t).build()}),r},G={createEncoder:(e,t={})=>new Le(e,t),createDecoder:Ne,createAccumulator:ie,getMessageKey:e=>e.id,isTerminal:e=>e.type===`finish`||e.type===`error`||e.type===`abort`},ze=(e,n)=>({sendMessages:async r=>{let{messages:i,abortSignal:a,trigger:o,messageId:s}=r,c,l;if(o===`regenerate-message`)c=[],l=i;else{if(i.length===0)throw new t.ErrorInfo(`unable to send messages; messages array is empty for submit-message trigger`,W.InvalidArgument,400);c=[i.at(-1)],l=i.slice(0,-1)}let u,d;if(o===`regenerate-message`&&s){u=s;let t=e.getTree().getNodeByKey(s);t&&(u=t.msgId,d=t.parentId)}let f,p;if(n?.prepareSendMessagesRequest){let e=n.prepareSendMessagesRequest({id:r.chatId,trigger:o,messageId:s,history:l,messages:c,forkOf:u,parent:d});f=e.body??{},p=e.headers}else f={history:l.map(t=>({message:t,headers:e.getMessageHeaders(t)})),id:r.chatId,trigger:o,...s!==void 0&&{messageId:s},...u!==void 0&&{forkOf:u},...d!==void 0&&{parent:d}},p=void 0;let m={body:f,headers:p};u!==void 0&&(m.forkOf=u),d!==void 0&&(m.parent=d);let h=await e.send(c,m);a&&a.addEventListener(`abort`,()=>void e.cancel({all:!0}),{once:!0});let{readable:g,writable:_}=new TransformStream,v=_.getWriter();return h.stream.pipeTo(new WritableStream({close:()=>{v.close().catch(()=>{})},abort:()=>{v.close().catch(()=>{})}})).catch(()=>{v.close().catch(()=>{})}),g},reconnectToStream:()=>Promise.resolve(null),close:async t=>e.close(t)}),Be=e=>({logAction:(t,n,r)=>{e.error(n,{detail:r})},shouldLog:()=>!0}),Ve=t.Realtime.EventEmitter,He=class extends Ve{constructor(e){super(Be(e))}},K=function(e){return e.Trace=`trace`,e.Debug=`debug`,e.Info=`info`,e.Warn=`warn`,e.Error=`error`,e.Silent=`silent`,e}({}),Ue=(e,t,n)=>{let r=n?`, context: ${JSON.stringify(n)}`:``,i=`[${new Date().toISOString()}] ${t.valueOf().toUpperCase()} ably-ai-transport: ${e}${r}`;switch(t){case K.Trace:case K.Debug:console.log(i);break;case K.Info:console.info(i);break;case K.Warn:console.warn(i);break;case K.Error:console.error(i);break;case K.Silent:break}},We=e=>new Ge(e.logHandler??Ue,e.logLevel),q=function(e){return e[e.Trace=0]=`Trace`,e[e.Debug=1]=`Debug`,e[e.Info=2]=`Info`,e[e.Warn=3]=`Warn`,e[e.Error=4]=`Error`,e[e.Silent=5]=`Silent`,e}(q||{}),J=new Map([[K.Trace,q.Trace],[K.Debug,q.Debug],[K.Info,q.Info],[K.Warn,q.Warn],[K.Error,q.Error],[K.Silent,q.Silent]]),Ge=class e{constructor(e,n,r){this._handler=e,this._context=r;let i=J.get(n);if(i===void 0)throw new t.ErrorInfo(`unable to create logger; invalid log level: ${n}`,W.InvalidArgument,400);this._levelNumber=i}trace(e,t){this._write(e,K.Trace,q.Trace,t)}debug(e,t){this._write(e,K.Debug,q.Debug,t)}info(e,t){this._write(e,K.Info,q.Info,t)}warn(e,t){this._write(e,K.Warn,q.Warn,t)}error(e,t){this._write(e,K.Error,q.Error,t)}withContext(t){let n=[...J.entries()].find(([,e])=>e===this._levelNumber)?.[0]??K.Error;return new e(this._handler,n,this._mergeContext(t))}_write(e,t,n,r){n>=this._levelNumber&&this._handler(e,t,this._mergeContext(r))}_mergeContext(e){return this._context?e?{...this._context,...e}:this._context:e??void 0}},Ke=class{constructor(e,t){this._nodeIndex=new Map,this._codecKeyIndex=new Map,this._sortedList=[],this._parentIndex=new Map,this._selections=new Map,this._seqCounter=0,this._getKey=e,this._logger=t}_compareNodes(e,t){let n=e.node.serial,r=t.node.serial;return n===void 0&&r===void 0?e.insertSeq-t.insertSeq:n===void 0?1:r===void 0||n<r?-1:n>r?1:e.insertSeq-t.insertSeq}_insertSorted(e){if(e.node.serial===void 0){this._sortedList.push(e);return}let t=0,n=this._sortedList.length;for(;t<n;){let r=t+n>>>1,i=this._sortedList[r];if(!i)break;this._compareNodes(i,e)<=0?t=r+1:n=r}this._sortedList.splice(t,0,e)}_removeSorted(e){let t=this._sortedList.indexOf(e);t!==-1&&this._sortedList.splice(t,1)}_addToParentIndex(e,t){let n=this._parentIndex.get(e);n||(n=new Set,this._parentIndex.set(e,n)),n.add(t)}_removeFromParentIndex(e,t){let n=this._parentIndex.get(e);n&&(n.delete(t),n.size===0&&this._parentIndex.delete(e))}_getSiblingGroup(e){let t=this._nodeIndex.get(e);if(!t)return[];let n=t.node,r=new Set([n.msgId]);for(;n.forkOf&&!r.has(n.forkOf);){let e=this._nodeIndex.get(n.forkOf);if(!e||e.node.parentId!==n.parentId)break;n=e.node,r.add(n.msgId)}let i=n.parentId,a=n.msgId,o=[],s=this._parentIndex.get(i);if(s)for(let e of s){let t=this._nodeIndex.get(e);t&&this._isSiblingOf(t.node,a)&&o.push(t)}return o.sort((e,t)=>this._compareNodes(e,t)),o.map(e=>e.node)}_isSiblingOf(e,t){if(e.msgId===t)return!0;let n=e,r=new Set([n.msgId]);for(;n.forkOf;){if(n.forkOf===t)return!0;if(r.has(n.forkOf))break;let e=this._nodeIndex.get(n.forkOf);if(!e)break;n=e.node,r.add(n.msgId)}return!1}_getGroupRoot(e){let t=this._nodeIndex.get(e);if(!t)return e;let n=t.node,r=new Set([n.msgId]);for(;n.forkOf&&!r.has(n.forkOf);){let e=this._nodeIndex.get(n.forkOf);if(!e||e.node.parentId!==n.parentId)break;n=e.node,r.add(n.msgId)}return n.msgId}flatten(){let e=[],t=new Set,n=new Map;for(let r of this._sortedList){let i=r.node,{msgId:a,parentId:o}=i;if(o!==void 0&&!t.has(o))continue;let s=this._getSiblingGroup(a);if(s.length>1){let e=this._getGroupRoot(a),t=n.get(e);if(t===void 0){let r=this._selections.get(e)??s.length-1,i=s[Math.max(0,Math.min(r,s.length-1))];if(!i)break;t=i.msgId,n.set(e,t)}if(a!==t)continue}t.add(a),e.push(i.message)}return e}getSiblings(e){return this._getSiblingGroup(e).map(e=>e.message)}hasSiblings(e){return this._getSiblingGroup(e).length>1}getSelectedIndex(e){let t=this._getSiblingGroup(e);if(t.length<=1)return 0;let n=this._getGroupRoot(e),r=this._selections.get(n);return r===void 0?t.length-1:Math.max(0,Math.min(r,t.length-1))}select(e,t){this._logger.debug(`ConversationTree.select();`,{msgId:e,index:t});let n=this._getSiblingGroup(e);if(n.length<=1)return;let r=this._getGroupRoot(e);this._selections.set(r,Math.max(0,Math.min(t,n.length-1)))}getNode(e){return this._nodeIndex.get(e)?.node}getNodeByKey(e){let t=this._codecKeyIndex.get(e);if(t)return this._nodeIndex.get(t)?.node}getHeaders(e){return this._nodeIndex.get(e)?.node.headers}upsert(e,t,n,r){let i=n[`x-ably-parent`]??void 0,a=n[`x-ably-fork-of`]??void 0;this._codecKeyIndex.set(this._getKey(t),e);let o=this._nodeIndex.get(e);if(o){o.node.message=t,Object.keys(n).length>0&&(o.node.headers={...n}),r&&!o.node.serial&&(this._logger.debug(`ConversationTree.upsert(); promoting serial`,{msgId:e,serial:r}),o.node.serial=r,this._removeSorted(o),this._insertSorted(o));return}this._logger.trace(`ConversationTree.upsert(); inserting new node`,{msgId:e,parentId:i,forkOf:a});let s={node:{message:t,msgId:e,parentId:i,forkOf:a,headers:{...n},serial:r},insertSeq:this._seqCounter++};this._nodeIndex.set(e,s),this._addToParentIndex(i,e),this._insertSorted(s)}delete(e){let t=this._nodeIndex.get(e);if(!t)return;this._logger.debug(`ConversationTree.delete();`,{msgId:e});let{node:n}=t,r=this._getKey(n.message);this._codecKeyIndex.get(r)===e&&this._codecKeyIndex.delete(r),this._removeFromParentIndex(n.parentId,e),this._removeSorted(t),this._nodeIndex.delete(e),this._selections.delete(e)}},qe=(e,t)=>new Ke(e,t),Y=e=>{let t=[...e.rawMessages].toReversed(),n=e.codec.createDecoder(),r=new Map,i=e.codec.createAccumulator(),a=0,o=new Map,s=new Map;for(let c of t){let t=n.decode(c),l=D(c),u=l[p],d=l[m],f=c.serial;if(u){let n=r.get(u);if(n||(n={accumulator:e.codec.createAccumulator(),firstSeen:a++,msgHeaders:new Map,msgSerials:new Map},r.set(u,n)),d){let e=n.msgHeaders.get(d);e?Object.keys(l).length>0&&Object.assign(e,l):(n.msgHeaders.set(d,{...l}),f&&n.msgSerials.set(d,f))}n.accumulator.processOutputs(t)}else i.processOutputs(t);for(let n of t)if(n.kind===`message`){let t=e.getMessageKey(n.message),r=o.get(t);r?Object.keys(l).length>0&&Object.assign(r,l):(o.set(t,{...l}),f&&s.set(t,f))}}let c=[];for(let t of i.completedMessages){let n=e.getMessageKey(t);c.push({message:t,headers:o.get(n)??{},serial:s.get(n)??``})}let l=[...r.values()].toSorted((e,t)=>e.firstSeen-t.firstSeen);for(let t of l){let n=new Set,r=new Map,i=new Map;for(let a of t.accumulator.completedMessages){let t=e.getMessageKey(a),c=o.get(t);if(c){r.set(t,c);let e=s.get(t);e&&i.set(t,e);let a=c[m];a&&n.add(a)}}let a=[...t.msgHeaders.entries()].filter(([e])=>!n.has(e)),l=0;for(let n of t.accumulator.completedMessages){let o=e.getMessageKey(n),s=a[l];if(!r.has(o)&&s){let[e,n]=s;r.set(o,n);let a=t.msgSerials.get(e);a&&i.set(o,a),l++}}for(let n of t.accumulator.completedMessages){let t=e.getMessageKey(n);c.push({message:n,headers:r.get(t)??{},serial:i.get(t)??``})}}return c.toReversed()},X=async(e,t,n)=>{e.rawMessages.push(...t.items),e.lastAblyPage=t;let r=Y(e).length;for(;r<e.returnedCount+n&&t.hasNext();){e.logger.debug(`decodeHistory.fetchUntilLimit(); fetching next page`,{collected:e.rawMessages.length,decoded:r});let n=await t.next();if(!n)break;t=n,e.rawMessages.push(...n.items),e.lastAblyPage=n,r=Y(e).length}},Z=(e,t)=>{let n=Y(e),r=n.slice(e.returnedCount,e.returnedCount+t),i=[...r].toReversed();e.returnedCount+=r.length;let a=n.length>e.returnedCount,o=e.lastAblyPage?.hasNext()??!1,s=e.rawMessages.length-e.returnedRawCount>0?e.rawMessages.slice(e.returnedRawCount).toReversed():[];return e.returnedRawCount=e.rawMessages.length,{items:i.map(e=>e.message),itemHeaders:i.map(e=>e.headers),itemSerials:i.map(e=>e.serial),rawMessages:s,hasNext:()=>a||o,next:async()=>{if(a)return Z(e,t);if(!o||!e.lastAblyPage)return;let n=await e.lastAblyPage.next();if(n)return await X(e,n,t),Z(e,t)}}},Je=async(e,t,n,r)=>{let i=n?.limit??100,a={codec:t,rawMessages:[],returnedCount:0,returnedRawCount:0,lastAblyPage:void 0,getMessageKey:t.getMessageKey.bind(t),logger:r};r.trace(`decodeHistory();`,{limit:i});let o=i*10;return await e.attach(),await X(a,await e.history({untilAttach:!0,limit:o}),i),Z(a,i)},Q=e=>{let t={[g]:e.role,[p]:e.turnId,[m]:e.msgId};return e.turnClientId!==void 0&&(t[h]=e.turnClientId),e.parent&&(t[x]=e.parent),e.forkOf&&(t[ee]=e.forkOf),t},Ye=class{constructor(e,t){this._turns=new Map,this._isTerminal=e,this._logger=t}createStream(e){this._logger.trace(`StreamRouter.createStream();`,{turnId:e});let n={},r=new ReadableStream({start(e){n.controller=e}});if(!n.controller)throw new t.ErrorInfo(`unable to create stream; ReadableStream start() was not called synchronously`,W.TransportSubscriptionError,500);return this._turns.set(e,{controller:n.controller,turnId:e}),r}closeStream(e){let t=this._turns.get(e);if(!t)return!1;this._logger.debug(`StreamRouter.closeStream(); closing stream`,{turnId:e});try{t.controller.close()}catch{}return this._turns.delete(e),!0}route(e,t){let n=this._turns.get(e);if(!n)return!1;try{n.controller.enqueue(t)}catch{return this._turns.delete(e),!1}return this._isTerminal(t)&&this.closeStream(e),!0}has(e){return this._turns.has(e)}},Xe=(e,t)=>new Ye(e,t),Ze=()=>{},$=class{constructor(e){if(this._ownMsgIds=new Set,this._ownTurnIds=new Set,this._turnClientIds=new Map,this._turnMsgIds=new Map,this._turnObservers=new Map,this._ablyMessages=[],this._withheldKeys=new Set,this._closed=!1,this._channel=e.channel,this._codec=e.codec,this._clientId=e.clientId,this._api=e.api??`/api/chat`,this._credentials=e.credentials,this._headersFn=typeof e.headers==`function`?e.headers:e.headers?()=>e.headers:void 0,this._bodyFn=typeof e.body==`function`?e.body:e.body?()=>e.body:void 0,this._fetchFn=e.fetch??globalThis.fetch.bind(globalThis),this._logger=(e.logger??We({logLevel:K.Silent})).withContext({component:`ClientTransport`}),this._emitter=new He(this._logger),this._tree=qe(this._codec.getMessageKey.bind(this._codec),this._logger),this._router=Xe(this._codec.isTerminal.bind(this._codec),this._logger),this._decoder=this._codec.createDecoder(),e.messages){let t;for(let n of e.messages){let e=this._codec.getMessageKey(n),r={};t&&(r[x]=t),this._tree.upsert(e,n,r),t=e}this._emitter.emit(`message`)}this._onMessage=e=>{this._handleMessage(e)},this._attachPromise=this._channel.subscribe(this._onMessage)}_handleMessage(e){if(!this._closed){this._ablyMessages.push(e),this._emitter.emit(`ably-message`);try{if(e.name===`x-ably-turn-start`){let t=D(e),n=t[p],r=t[`x-ably-turn-client-id`]??``;n&&(this._turnClientIds.set(n,r),this._emitter.emit(`turn`,{type:w,turnId:n,clientId:r}));return}if(e.name===`x-ably-turn-end`){let t=D(e),n=t[p],r=t[`x-ably-turn-client-id`]??``,i=t[`x-ably-turn-reason`]??`complete`;if(n){this._router.closeStream(n),this._turnObservers.delete(n),this._turnClientIds.delete(n);let e=this._turnMsgIds.get(n);if(e){for(let t of e)this._ownMsgIds.delete(t);this._turnMsgIds.delete(n)}this._ownTurnIds.delete(n),this._emitter.emit(`turn`,{type:T,turnId:n,clientId:r,reason:i})}return}let t=this._decoder.decode(e),n=D(e),r=e.serial,i=n[p];i&&this._updateTurnObserverHeaders(i,n,r);for(let i of t)i.kind===`message`?this._handleMessageOutput(i.message,n,r,e.action):this._handleEventOutput(i,n)}catch(e){let n=e instanceof t.ErrorInfo?e:void 0;this._emitter.emit(`error`,new t.ErrorInfo(`unable to process channel message; ${e instanceof Error?e.message:String(e)}`,W.TransportSubscriptionError,500,n))}}}_handleMessageOutput(e,t,n,r){let i=t[m];if(i&&this._ownMsgIds.has(i)){this._upsertAndNotify(e,t,n);return}r===`message.create`&&this._upsertAndNotify(e,t,n)}_handleEventOutput(e,t){if(e.kind!==`event`)return;let n=e.event,r=t[p];if(r){if(this._router.route(r,n)){this._accumulateAndEmit(r,e),this._codec.isTerminal(n)&&this._turnObservers.delete(r);return}this._ownTurnIds.has(r)&&!this._turnObservers.has(r)||(this._accumulateAndEmit(r,e),this._codec.isTerminal(n)&&this._turnObservers.delete(r))}}_upsertAndNotify(e,t,n){let r=this._codec.getMessageKey(e),i=t[`x-ably-msg-id`]??r;this._tree.upsert(i,e,t,n),this._emitter.emit(`message`)}_updateTurnObserverHeaders(e,t,n){let r=this._turnObservers.get(e);r?(Object.keys(t).length>0&&Object.assign(r.headers,t),n!==void 0&&(r.serial=n)):this._turnObservers.set(e,{headers:{...t},serial:n,accumulator:this._codec.createAccumulator()})}_accumulateAndEmit(e,t){let n=this._turnObservers.get(e);if(!n)return;n.accumulator.processOutputs([t]);let r=n.accumulator.messages;if(r.length===0)return;let i;try{i=structuredClone(r.at(-1))}catch{i=r.at(-1)}i&&(this._tree.upsert(n.headers[`x-ably-msg-id`]??this._codec.getMessageKey(i),i,{...n.headers},n.serial),this._emitter.emit(`message`))}async _publishCancel(e){this._logger.trace(`ClientTransport._publishCancel();`,{filter:e});let t={};e.turnId?t[_]=e.turnId:e.own?t[v]=`true`:e.clientId?t[b]=e.clientId:e.all&&(t[y]=`true`),await this._channel.publish({name:C,extras:{headers:t}})}_closeMatchingTurnStreams(e){if(e.all)for(let e of this._ownTurnIds)this._router.closeStream(e);else if(e.own)for(let e of this._ownTurnIds)this._router.closeStream(e);else if(e.clientId)for(let[t,n]of this._turnClientIds)n===e.clientId&&this._router.closeStream(t);else e.turnId&&this._router.closeStream(e.turnId)}_getMatchingTurnIds(e){let t=new Set;if(e.all)for(let e of this._turnClientIds.keys())t.add(e);else if(e.own)for(let[e,n]of this._turnClientIds)n===this._clientId&&t.add(e);else if(e.clientId)for(let[n,r]of this._turnClientIds)r===e.clientId&&t.add(n);else e.turnId&&this._turnClientIds.has(e.turnId)&&t.add(e.turnId);return t}_getMessagesWithHeaders(){return this._tree.flatten().map(e=>({message:e,headers:this.getMessageHeaders(e)}))}_getHistoryBefore(e){let t=this._getMessagesWithHeaders(),n=t.findIndex(t=>t.headers?.[m]===e);return n===-1?t:t.slice(0,n)}_processHistoryPage(e){for(let[t,n]of e.items.entries()){let r=e.itemHeaders?.[t]??{},i=e.itemSerials?.[t],a=this._codec.getMessageKey(n),o=r[`x-ably-msg-id`]??a;this._tree.upsert(o,n,r,i)}this._emitter.emit(`message`),e.rawMessages&&e.rawMessages.length>0&&(this._ablyMessages.unshift(...e.rawMessages),this._emitter.emit(`ably-message`))}async _loadUntilVisible(e,t,n){this._processHistoryPage(e);let r=e,i=()=>{let e=0;for(let t of this._tree.flatten())n.has(this._codec.getMessageKey(t))||e++;return e};for(;i()<t&&r.hasNext();){let e=await r.next();if(!e)break;this._processHistoryPage(e),r=e}return{newVisible:this._tree.flatten().filter(e=>!n.has(this._codec.getMessageKey(e))),lastPage:r}}_releaseWithheld(e){for(let t of e)this._withheldKeys.delete(this._codec.getMessageKey(t));e.length>0&&this._emitter.emit(`message`)}async send(e,n){if(this._closed||(await this._attachPromise,this._closed))throw new t.ErrorInfo(`unable to send; transport is closed`,W.TransportClosed,400);this._logger.trace(`ClientTransport.send();`);let r=Array.isArray(e)?e:[e],i=crypto.randomUUID();this._ownTurnIds.add(i);let a=new Set,o=[],s=this._getMessagesWithHeaders(),c;if(n?.parent===void 0&&!n?.forkOf){let e=this._tree.flatten();if(e.length>0){let t=e.at(-1);if(t){let e=this._codec.getMessageKey(t);c=this._tree.getNodeByKey(e)?.msgId??e}}}let l=n?.parent===void 0?c:n.parent;for(let e of r){let t=crypto.randomUUID();this._ownMsgIds.add(t),a.add(t);let r=n?.parent===void 0?c:n.parent??void 0,s=Q({role:`user`,turnId:i,msgId:t,turnClientId:this._clientId,parent:r,forkOf:n?.forkOf});this._upsertAndNotify(e,s);let l={[m]:t,[g]:`user`};r&&(l[x]=r),o.push({message:e,headers:l}),n?.parent===void 0&&!n?.forkOf&&(c=t)}this._turnMsgIds.set(i,a);let u=this._router.createStream(i),d=this._headersFn?.()??{},f={...this._bodyFn?.()??{},history:s,...n?.body,turnId:i,clientId:this._clientId,messages:o,...n?.forkOf!==void 0&&{forkOf:n.forkOf},...l!==void 0&&{parent:l}},p={...d,...n?.headers};return this._fetchFn(this._api,{method:`POST`,headers:{"Content-Type":`application/json`,...p},body:JSON.stringify(f),...this._credentials?{credentials:this._credentials}:{}}).then(e=>{e.ok||(this._emitter.emit(`error`,new t.ErrorInfo(`unable to send; HTTP POST to ${this._api} returned ${String(e.status)} ${e.statusText}`,W.TransportSendFailed,e.status)),this._router.closeStream(i))}).catch(e=>{let n=e instanceof t.ErrorInfo?e:void 0;this._emitter.emit(`error`,new t.ErrorInfo(`unable to send; HTTP POST to ${this._api} failed: ${e instanceof Error?e.message:String(e)}`,W.TransportSendFailed,500,n)),this._router.closeStream(i)}),{stream:u,turnId:i,cancel:async()=>this.cancel({turnId:i})}}async regenerate(e,t){this._logger.trace(`ClientTransport.regenerate();`,{messageId:e});let n=this._tree.getNode(e)?.parentId;return this.send([],{...t,body:{history:this._getHistoryBefore(e),...t?.body},forkOf:e,parent:n})}async edit(e,t,n){this._logger.trace(`ClientTransport.edit();`,{messageId:e});let r=this._tree.getNode(e)?.parentId;return this.send(t,{...n,body:{history:this._getHistoryBefore(e),...n?.body},forkOf:e,parent:r})}async cancel(e){if(this._closed)return;let t=e??{own:!0};this._logger.debug(`ClientTransport.cancel();`,{filter:t}),await this._publishCancel(t),this._closeMatchingTurnStreams(t)}async waitForTurn(e){if(this._closed)return;let t=e??{own:!0},n=this._getMatchingTurnIds(t);if(n.size!==0)return this._logger.debug(`ClientTransport.waitForTurn();`,{turnIds:[...n]}),new Promise(e=>{let t=r=>{r.type===`x-ably-turn-end`&&(n.delete(r.turnId),n.size===0&&(this._emitter.off(`turn`,t),e()))};this._emitter.on(`turn`,t)})}on(e,t){if(this._closed)return Ze;let n=t;return this._emitter.on(e,n),()=>{this._emitter.off(e,n)}}getTree(){return this._tree}getActiveTurnIds(){let e=new Map;for(let[t,n]of this._turnClientIds){let r=e.get(n);r||(r=new Set,e.set(n,r)),r.add(t)}return e}getMessageHeaders(e){let t=this._codec.getMessageKey(e);return this._tree.getNodeByKey(t)?.headers}getMessages(){return this._withheldKeys.size===0?this._tree.flatten():this._tree.flatten().filter(e=>!this._withheldKeys.has(this._codec.getMessageKey(e)))}getMessagesWithHeaders(){return this._getMessagesWithHeaders()}getAblyMessages(){return[...this._ablyMessages]}async history(e){if(this._closed)throw new t.ErrorInfo(`unable to load history; transport is closed`,W.TransportClosed,400);this._logger.trace(`ClientTransport.history();`,{limit:e?.limit});let n=e?.limit??100,r=new Set(this._tree.flatten().map(e=>this._codec.getMessageKey(e))),i=await Je(this._channel,this._codec,e,this._logger),a=await this._loadUntilVisible(i,n,r);i=a.lastPage;let o=a.newVisible;for(let e of o)this._withheldKeys.add(this._codec.getMessageKey(e));let s=o.slice(-n),c=o.slice(0,-n);this._releaseWithheld(s);let l=e=>({items:e,hasNext:()=>c.length>0||i.hasNext(),next:async()=>{if(c.length>0){let e=c.splice(-n,n);return this._releaseWithheld(e),l(e)}if(!i.hasNext())return;let e=await i.next();if(!e)return;let t=new Set(r);for(let e of this._tree.flatten())t.add(this._codec.getMessageKey(e));let a=await this._loadUntilVisible(e,n,t);i=a.lastPage;let o=a.newVisible;for(let e of o)this._withheldKeys.add(this._codec.getMessageKey(e));let s=o.splice(-n,n);if(c.push(...o),this._releaseWithheld(s),s.length!==0)return l(s)}});return l(s)}async close(e){if(!this._closed){if(this._closed=!0,this._logger.info(`ClientTransport.close();`),e?.cancel){try{await this._publishCancel(e.cancel)}catch{}this._closeMatchingTurnStreams(e.cancel)}this._channel.unsubscribe(this._onMessage);for(let e of this._ownTurnIds)this._router.closeStream(e);this._turnObservers.clear(),this._emitter.off(),this._ownTurnIds.clear(),this._ownMsgIds.clear(),this._turnMsgIds.clear(),this._turnClientIds.clear(),this._withheldKeys.clear(),this._ablyMessages.length=0}}},Qe=e=>new $(e),$e=async(e,t,n,r,i)=>{i?.trace(`pipeStream();`);let a=e.getReader(),o,s=n?new Promise(e=>{if(n.aborted){e();return}o=()=>{e()},n.addEventListener(`abort`,o,{once:!0})}):new Promise(()=>{}),c=`complete`;try{for(;;){let e=await Promise.race([a.read(),s.then(()=>`aborted`)]);if(e===`aborted`){c=`cancelled`,i?.debug(`pipeStream(); stream cancelled by abort signal`),r&&await r(async e=>t.appendEvent(e)),await t.abort(`cancelled`);break}let{done:n,value:o}=e;if(n){await t.close(),i?.debug(`pipeStream(); stream completed`);break}await t.appendEvent(o)}}catch(e){c=`error`;let n=e instanceof Error?e.message:String(e);i?.error(`pipeStream(); stream error`,{error:n});try{await t.close()}catch{}}finally{o&&n?.removeEventListener(`abort`,o),a.releaseLock()}return{reason:c}},et=class{constructor(e,t){this._activeTurns=new Map,this._channel=e,this._logger=t?.withContext({component:`TurnManager`})}async startTurn(e,t,n){this._logger?.trace(`DefaultTurnManager.startTurn();`,{turnId:e,clientId:t});let r=n??new AbortController,i=t??``;return this._activeTurns.set(e,{controller:r,clientId:i}),await this._channel.publish({name:w,extras:{headers:{[p]:e,[h]:i}}}),this._logger?.debug(`DefaultTurnManager.startTurn(); turn started`,{turnId:e}),r.signal}async endTurn(e,t){this._logger?.trace(`DefaultTurnManager.endTurn();`,{turnId:e,reason:t});let n=this._activeTurns.get(e)?.clientId??``;await this._channel.publish({name:T,extras:{headers:{[p]:e,[h]:n,[S]:t}}}),this._activeTurns.delete(e),this._logger?.debug(`DefaultTurnManager.endTurn(); turn ended`,{turnId:e,reason:t})}getSignal(e){return this._activeTurns.get(e)?.controller.signal}getClientId(e){return this._activeTurns.get(e)?.clientId}abort(e){this._logger?.debug(`DefaultTurnManager.abort();`,{turnId:e}),this._activeTurns.get(e)?.controller.abort()}getActiveTurnIds(){return[...this._activeTurns.keys()]}close(){this._logger?.trace(`DefaultTurnManager.close();`,{activeTurns:this._activeTurns.size});for(let e of this._activeTurns.values())e.controller.abort();this._activeTurns.clear()}},tt=(e,t)=>new et(e,t),nt=class{constructor(e){this._registeredTurns=new Map,this._channel=e.channel,this._codec=e.codec,this._logger=e.logger?.withContext({component:`ServerTransport`}),this._onError=e.onError,this._turnManager=tt(this._channel,this._logger),this._channelListener=e=>{this._handleChannelMessage(e)},this._attachPromise=this._channel.subscribe(C,this._channelListener).then(()=>{},e=>{let n=new t.ErrorInfo(`unable to subscribe to cancel messages; ${e instanceof Error?e.message:String(e)}`,W.TransportSubscriptionError,500,e instanceof t.ErrorInfo?e:void 0);this._logger?.error(`DefaultServerTransport(); subscribe failed`),this._onError?.(n)}),this._logger?.debug(`DefaultServerTransport(); transport created`)}newTurn(e){return this._logger?.trace(`DefaultServerTransport.newTurn();`,{turnId:e.turnId}),this._createTurn(e)}close(){this._logger?.trace(`DefaultServerTransport.close();`),this._channel.unsubscribe(C,this._channelListener);for(let e of this._registeredTurns.values())e.controller.abort();this._registeredTurns.clear(),this._turnManager.close(),this._logger?.debug(`DefaultServerTransport.close(); transport closed`)}_resolveFilter(e,t){let n=[...this._registeredTurns.keys()];return e.all?n:e.own&&t?n.filter(e=>this._registeredTurns.get(e)?.clientId===t):e.clientId?n.filter(t=>this._registeredTurns.get(t)?.clientId===e.clientId):e.turnId&&this._registeredTurns.has(e.turnId)?[e.turnId]:[]}async _handleCancelMessage(e){let n=D(e),r={};n[`x-ably-cancel-turn-id`]?r.turnId=n[_]:n[`x-ably-cancel-own`]===`true`?r.own=!0:n[`x-ably-cancel-client-id`]?r.clientId=n[b]:n[`x-ably-cancel-all`]===`true`&&(r.all=!0);let i=this._resolveFilter(r,e.clientId);if(i.length===0)return;this._logger?.debug(`DefaultServerTransport._handleCancelMessage(); matched turns`,{matchedTurnIds:i,filter:r});let a=new Map;for(let e of i){let t=this._registeredTurns.get(e);a.set(e,t?.clientId??``)}let o={message:e,filter:r,matchedTurnIds:i,turnOwners:a};for(let e of i){let n=this._registeredTurns.get(e);if(n)try{if(n.onCancel&&!await n.onCancel(o)){this._logger?.debug(`DefaultServerTransport._handleCancelMessage(); cancel rejected by onCancel`,{turnId:e});continue}n.controller.abort(),this._logger?.debug(`DefaultServerTransport._handleCancelMessage(); turn aborted`,{turnId:e})}catch(r){let i=new t.ErrorInfo(`unable to process cancel for turn ${e}; onCancel handler threw: ${r instanceof Error?r.message:String(r)}`,W.CancelListenerError,500,r instanceof t.ErrorInfo?r:void 0);this._logger?.error(`DefaultServerTransport._handleCancelMessage(); onCancel threw`,{turnId:e}),(n.onError??this._onError)?.(i)}}}_handleChannelMessage(e){try{e.name===`x-ably-cancel`&&this._handleCancelMessage(e).catch(e=>{let n=new t.ErrorInfo(`unable to route cancel message; ${e instanceof Error?e.message:String(e)}`,W.CancelListenerError,500,e instanceof t.ErrorInfo?e:void 0);this._logger?.error(`DefaultServerTransport._handleChannelMessage(); cancel routing error`),this._onError?.(n)})}catch(e){let n=new t.ErrorInfo(`unable to process channel message; ${e instanceof Error?e.message:String(e)}`,W.TransportSubscriptionError,500,e instanceof t.ErrorInfo?e:void 0);this._logger?.error(`DefaultServerTransport._handleChannelMessage(); subscription error`),this._onError?.(n)}}_createTurn(e){let{turnId:n,clientId:r,onMessage:i,onAbort:a,onCancel:o,onError:s,parent:c,forkOf:l}=e,u=new AbortController,d=!1,f=!1,p={turnId:n,clientId:r??``,controller:u,onCancel:o,onError:s};this._registeredTurns.set(n,p);let m=this._logger,h=this._turnManager,g=this._attachPromise,_=this._codec,v=this._channel,y=this._registeredTurns;return{get turnId(){return n},get abortSignal(){return u.signal},start:async()=>{if(m?.trace(`Turn.start();`,{turnId:n}),u.signal.aborted)throw new t.ErrorInfo(`unable to start turn; turn ${n} was cancelled before start()`,W.InvalidArgument,400);if(!d){d=!0;try{await h.startTurn(n,r,u)}catch(e){let r=new t.ErrorInfo(`unable to publish turn-start for turn ${n}; ${e instanceof Error?e.message:String(e)}`,W.TurnLifecycleError,500,e instanceof t.ErrorInfo?e:void 0);throw m?.error(`Turn.start(); failed to publish turn-start`,{turnId:n}),s?.(r),r}m?.debug(`Turn.start(); turn started`,{turnId:n})}},addMessages:async(e,r)=>{if(m?.trace(`Turn.addMessages();`,{turnId:n,count:e.length}),!d)throw new t.ErrorInfo(`unable to add messages; start() must be called before addMessages() (turn ${n})`,W.InvalidArgument,400);await g;let a=[];for(let t of e){let e=crypto.randomUUID(),o=O(Q({role:`user`,turnId:n,msgId:e,turnClientId:r?.clientId,parent:r?.parent===void 0?c??void 0:r.parent??void 0,forkOf:r?.forkOf??l}),t.headers);await _.createEncoder(v,{extras:{headers:o},onMessage:i}).writeMessages([t.message],r?.clientId?{clientId:r.clientId}:void 0),a.push(o[`x-ably-msg-id`]??e)}return m?.debug(`Turn.addMessages(); messages published`,{turnId:n,count:e.length}),{msgIds:a}},streamResponse:async(e,r)=>{if(m?.trace(`Turn.streamResponse();`,{turnId:n}),!d)throw new t.ErrorInfo(`unable to stream response; start() must be called before streamResponse() (turn ${n})`,W.InvalidArgument,400);await g;let o=h.getSignal(n),s=h.getClientId(n),u=r?.parent===void 0?c??void 0:r.parent??void 0,f=Q({role:`assistant`,turnId:n,msgId:crypto.randomUUID(),turnClientId:s,parent:u,forkOf:r?.forkOf??l}),p=await $e(e,_.createEncoder(v,{extras:{headers:f},onMessage:i}),o,a,m);return m?.debug(`Turn.streamResponse(); stream finished`,{turnId:n,reason:p.reason}),p},end:async e=>{if(m?.trace(`Turn.end();`,{turnId:n,reason:e}),!d)throw new t.ErrorInfo(`unable to end turn; start() must be called before end() (turn ${n})`,W.InvalidArgument,400);if(!f){f=!0;try{await h.endTurn(n,e)}catch(e){let r=new t.ErrorInfo(`unable to publish turn-end for turn ${n}; ${e instanceof Error?e.message:String(e)}`,W.TurnLifecycleError,500,e instanceof t.ErrorInfo?e:void 0);throw m?.error(`Turn.end(); failed to publish turn-end`,{turnId:n}),s?.(r),r}finally{y.delete(n)}m?.debug(`Turn.end(); turn ended`,{turnId:n,reason:e})}}}}},rt=e=>new nt(e);e.UIMessageCodec=G,e.createChatTransport=ze,e.createClientTransport=e=>Qe({...e,codec:G}),e.createServerTransport=e=>rt({...e,codec:G})});
|
|
1
|
+
(function(e,t){typeof exports==`object`&&typeof module<`u`?t(exports,require(`ably`),require(`ai`)):typeof define==`function`&&define.amd?define([`exports`,`ably`,`ai`],t):(e=typeof globalThis<`u`?globalThis:e||self,t(e.AblyAiTransportVercel={},e.Ably,e.AI))})(this,function(e,t,n){Object.defineProperty(e,Symbol.toStringTag,{value:`Module`});var r=Object.create,i=Object.defineProperty,a=Object.getOwnPropertyDescriptor,o=Object.getOwnPropertyNames,s=Object.getPrototypeOf,c=Object.prototype.hasOwnProperty,l=(e,t,n,r)=>{if(t&&typeof t==`object`||typeof t==`function`)for(var s=o(t),l=0,u=s.length,d;l<u;l++)d=s[l],!c.call(e,d)&&d!==n&&i(e,d,{get:(e=>t[e]).bind(null,d),enumerable:!(r=a(t,d))||r.enumerable});return e};t=((e,t,n)=>(n=e==null?{}:r(s(e)),l(t||!e||!e.__esModule?i(n,`default`,{value:e,enumerable:!0}):n,e)))(t,1);var u=`stream`,d=`status`,f=`stream-id`,p=`discrete`,m=`run-id`,h=`invocation-id`,g=`event-id`,_=`codec-message-id`,v=`run-client-id`,y=`input-client-id`,b=`role`,x=`parent`,S=`fork-of`,C=`msg-regenerate`,w=`run-reason`,T=`input-codec-message-id`,E=`error-code`,D=`ai-cancel`,O=`ai-run-start`,ee=`ai-run-suspend`,te=`ai-run-resume`,k=`ai-run-end`,A=`ai-output`,j=`ai-input`,M=(e,t)=>{let n=e.extras;if(!n||typeof n!=`object`)return{};let r=n.ai;if(!r||typeof r!=`object`)return{};let i=r[t];return!i||typeof i!=`object`?{}:i},N=e=>M(e,`transport`),P=e=>M(e,`codec`),F=e=>{if(e!==void 0)try{return JSON.parse(e)}catch{return}},I=(e,t)=>({...e,...t}),L=e=>{if(e!==void 0)return e===`true`},R=(e,t)=>e.serial===void 0&&t.serial===void 0?0:e.serial===void 0?1:t.serial===void 0||e.serial<t.serial?-1:+(e.serial>t.serial),z=(e,t)=>e[t],B=e=>{let t={};for(let n in e)Object.prototype.hasOwnProperty.call(e,n)&&e[n]!==void 0&&(t[n]=e[n]);return t},V=e=>({str:t=>z(e,t),strOr:(t,n)=>z(e,t)??n,bool:t=>L(z(e,t)),json:t=>F(z(e,t))}),H=()=>{let e={},t={str:(n,r)=>(r!==void 0&&(e[n]=r),t),bool:(n,r)=>(r!==void 0&&(e[n]=String(r)),t),json:(n,r)=>(r!=null&&(e[n]=JSON.stringify(r)),t),build:()=>e};return t},U=class{constructor(e,t={}){this._serialState=new Map,this._hooks=e,this._onStreamUpdate=t.onStreamUpdate,this._onStreamDelete=t.onStreamDelete,this._logger=t.logger?.withContext({component:`DecoderCore`})}decode(e){let t=e.action;switch(this._logger?.trace(`DefaultDecoderCore.decode();`,{action:t,serial:e.serial,name:e.name}),t){case`message.create`:{let t=this._toPayload(e);return t.transportHeaders?.stream===`true`?this._decodeStreamedCreate(t,e.serial):this._hooks.decodeDiscrete(t)}case`message.append`:return this._decodeAppend(e);case`message.update`:return this._decodeUpdate(e);case`message.delete`:return this._decodeDelete(e);default:return[]}}_toPayload(e){return{name:e.name??``,data:e.data,transportHeaders:N(e),codecHeaders:P(e)}}_stringData(e){return typeof e.data==`string`?e.data:``}_invokeOnStreamUpdate(e){if(this._onStreamUpdate)try{this._onStreamUpdate(e)}catch(e){this._logger?.error(`DefaultDecoderCore._invokeOnStreamUpdate(); callback threw`,{error:e})}}_invokeOnStreamDelete(e,t){if(this._onStreamDelete)try{this._onStreamDelete(e,t)}catch(e){this._logger?.error(`DefaultDecoderCore._invokeOnStreamDelete(); callback threw`,{error:e})}}_decodeStreamedCreate(e,t){if(!t)return[];let n=e.transportHeaders?.[`stream-id`]??``,r={name:e.name,streamId:n,accumulated:``,codecHeaders:{...e.codecHeaders},transportHeaders:{...e.transportHeaders},closed:!1};return this._serialState.set(t,r),this._logger?.debug(`DefaultDecoderCore._decodeStreamedCreate(); new stream`,{name:e.name,streamId:n,serial:t}),this._hooks.buildStartEvents(r)}_decodeAppend(e){let t=e.serial;if(!t)return[];let n=this._serialState.get(t);if(!n)return this._decodeUpdate(e);let r=N(e),i=P(e),a=typeof e.data==`string`?e.data:``,o=r[d],s=[];return a.length>0&&(n.accumulated+=a,s.push(...this._hooks.buildDeltaEvents(n,a))),o===`complete`&&!n.closed?(n.closed=!0,s.push(...this._hooks.buildEndEvents(n,i)),this._logger?.debug(`DefaultDecoderCore._decodeAppend(); stream complete`,{streamId:n.streamId})):o===`cancelled`&&!n.closed&&(n.closed=!0,this._logger?.debug(`DefaultDecoderCore._decodeAppend(); stream cancelled`,{streamId:n.streamId})),s}_decodeUpdate(e){let t=e.serial;if(!t)return[];let n=this._toPayload(e),r=n.transportHeaders??{},i=n.codecHeaders??{},a=r[u]===`true`,o=r[d],s=this._serialState.get(t);if(!s)return this._decodeFirstContact(n,a,o,t);let c=this._stringData(e);if(c.startsWith(s.accumulated)){let e=c.slice(s.accumulated.length),t=[];return e.length>0&&(s.accumulated=c,t.push(...this._hooks.buildDeltaEvents(s,e))),o===`complete`&&!s.closed?(s.closed=!0,t.push(...this._hooks.buildEndEvents(s,i))):o===`cancelled`&&!s.closed&&(s.closed=!0),t}return s.accumulated=c,s.codecHeaders={...i},s.transportHeaders={...r},this._invokeOnStreamUpdate(s),[]}_decodeFirstContact(e,t,n,r){if(!t)return this._hooks.decodeDiscrete(e);let i=e.transportHeaders?.[`stream-id`]??``,a=e.codecHeaders??{},o=typeof e.data==`string`?e.data:``;this._logger?.debug(`DefaultDecoderCore._decodeFirstContact(); first-contact stream`,{name:e.name,streamId:i,serial:r});let s={name:e.name,streamId:i,accumulated:o,codecHeaders:{...a},transportHeaders:{...e.transportHeaders},closed:n===`complete`||n===`cancelled`};this._serialState.set(r,s);let c=this._hooks.buildStartEvents(s);return o.length>0&&c.push(...this._hooks.buildDeltaEvents(s,o)),n===`complete`&&c.push(...this._hooks.buildEndEvents(s,a)),c}_decodeDelete(e){let t=e.serial;if(!t)return[];let n=this._serialState.get(t);return this._invokeOnStreamDelete(t,n),n&&(n.accumulated=``,n.closed=!0),this._logger?.debug(`DefaultDecoderCore._decodeDelete();`,{serial:t}),[]}},ne=(e,t={})=>new U(e,t),re=class{constructor(e){this._emitted=new Map,this._phases=e}ensurePhases(e,t){let n=this._getOrCreate(e),r=[];for(let e of this._phases)n.has(e.key)||(n.add(e.key),r.push(...e.build(t)));return r}markEmitted(e,t){this._getOrCreate(e).add(t)}resetPhase(e,t){this._emitted.get(e)?.delete(t)}clearScope(e){this._emitted.delete(e)}_getOrCreate(e){let t=this._emitted.get(e);return t||(t=new Set,this._emitted.set(e,t)),t}},ie=e=>new re(e),W=e=>{let t=V(e);return{...t,providerMetadata:()=>t.json(`providerMetadata`)}},ae=(e,t)=>e===`stop`||e===`length`||e===`content-filter`||e===`tool-calls`||e===`error`||e===`other`?e:t,oe=e=>e.startsWith(`data-`),se=e=>{if(e)try{return JSON.parse(e)}catch{return e}},ce=e=>W(e.codecHeaders).strOr(`type`,``),le=e=>{let t=W(e.codecHeaders);switch(ce(e)){case`text`:return B({type:`text-start`,id:e.streamId,providerMetadata:t.providerMetadata()});case`reasoning`:return B({type:`reasoning-start`,id:e.streamId,providerMetadata:t.providerMetadata()});case`tool-input`:return B({type:`tool-input-start`,toolCallId:e.streamId,toolName:t.strOr(`toolName`,``),dynamic:t.bool(`dynamic`),title:t.str(`title`),providerExecuted:t.bool(`providerExecuted`),providerMetadata:t.providerMetadata()});default:return{type:`text-start`,id:e.streamId}}},ue=(e,t)=>{switch(ce(e)){case`text`:return{type:`text-delta`,id:e.streamId,delta:t};case`reasoning`:return{type:`reasoning-delta`,id:e.streamId,delta:t};case`tool-input`:return{type:`tool-input-delta`,toolCallId:e.streamId,inputTextDelta:t};default:return{type:`text-delta`,id:e.streamId,delta:t}}},de=(e,t)=>{let n=W(t);switch(ce(e)){case`text`:return B({type:`text-end`,id:e.streamId,providerMetadata:n.providerMetadata()});case`reasoning`:return B({type:`reasoning-end`,id:e.streamId,providerMetadata:n.providerMetadata()});case`tool-input`:return B({type:`tool-input-available`,toolCallId:e.streamId,toolName:n.strOr(`toolName`,W(e.codecHeaders).strOr(`toolName`,``)),input:se(e.accumulated),providerMetadata:n.providerMetadata()});default:return{type:`text-end`,id:e.streamId}}},fe=()=>ie([{key:`start`,build:e=>[B({type:`start`,messageId:e.messageId})]},{key:`start-step`,build:()=>[{type:`start-step`}]}]),pe=(e,t,n)=>(n.markEmitted(t,`start`),[B({type:`start`,messageId:e.str(`messageId`),messageMetadata:e.json(`messageMetadata`)})]),me=(e,t)=>(t.markEmitted(e,`start-step`),[{type:`start-step`}]),he=(e,t)=>(t.resetPhase(e,`start-step`),[{type:`finish-step`}]),ge=(e,t,n)=>(n.clearScope(t),[B({type:`finish`,finishReason:ae(e.str(`finishReason`),`stop`),messageMetadata:e.json(`messageMetadata`)})]),_e=(e,t,n)=>(n.clearScope(t),[{type:`error`,errorText:typeof e==`string`?e:``}]),ve=(e,t,n)=>(n.clearScope(t),[B({type:`abort`,reason:typeof e==`string`&&e?e:void 0})]),ye=e=>[{type:`message-metadata`,messageMetadata:e.json(`messageMetadata`)}],be=(e,t)=>[B({type:`file`,url:typeof t==`string`?t:``,mediaType:e.strOr(`mediaType`,``),providerMetadata:e.providerMetadata()})],xe=(e,t)=>[B({type:`source-url`,sourceId:e.strOr(`sourceId`,``),url:typeof t==`string`?t:``,title:e.str(`title`),providerMetadata:e.providerMetadata()})],Se=e=>[B({type:`source-document`,sourceId:e.strOr(`sourceId`,``),mediaType:e.strOr(`mediaType`,``),title:e.strOr(`title`,``),filename:e.str(`filename`),providerMetadata:e.providerMetadata()})],Ce=(e,t)=>{let n=t;return[B({type:`tool-input-error`,toolCallId:e.strOr(`toolCallId`,``),toolName:e.strOr(`toolName`,``),errorText:n?.errorText??``,input:n?.input,dynamic:e.bool(`dynamic`),title:e.str(`title`),providerExecuted:e.bool(`providerExecuted`),providerMetadata:e.providerMetadata()})]},we=(e,t)=>{let n=t;return[B({type:`tool-output-available`,toolCallId:e.strOr(`toolCallId`,``),output:n?.output,dynamic:e.bool(`dynamic`),providerExecuted:e.bool(`providerExecuted`),preliminary:e.bool(`preliminary`)})]},Te=(e,t)=>{let n=t;return[B({type:`tool-output-error`,toolCallId:e.strOr(`toolCallId`,``),errorText:n?.errorText??``,dynamic:e.bool(`dynamic`),providerExecuted:e.bool(`providerExecuted`)})]},Ee=e=>[{type:`tool-approval-request`,toolCallId:e.strOr(`toolCallId`,``),approvalId:e.strOr(`approvalId`,``)}],De=e=>[{type:`tool-output-denied`,toolCallId:e.strOr(`toolCallId`,``)}],Oe=(e,t,n)=>[B({type:e,data:n,id:t.str(`id`),transient:t.bool(`transient`)})],ke=(e,t,n,r)=>[...r.ensurePhases(n,{messageId:e.str(`messageId`)}),B({type:`tool-input-start`,toolCallId:e.strOr(`toolCallId`,``),toolName:e.strOr(`toolName`,``),dynamic:e.bool(`dynamic`),title:e.str(`title`),providerExecuted:e.bool(`providerExecuted`),providerMetadata:e.providerMetadata()}),B({type:`tool-input-available`,toolCallId:e.strOr(`toolCallId`,``),toolName:e.strOr(`toolName`,``),input:t,providerMetadata:e.providerMetadata()})],Ae=e=>{let t=W(e.codecHeaders??{}),n=e.transportHeaders?.role??`user`,r=t.str(`messageId`)??``,i=t.strOr(`type`,``),a;switch(i){case`text`:a={type:`text`,text:typeof e.data==`string`?e.data:``};break;case`file`:a={type:`file`,mediaType:t.strOr(`mediaType`,``),url:typeof e.data==`string`?e.data:``};break;default:oe(i)&&(a=B({type:i,id:t.str(`id`),data:e.data}));break}return a?[{kind:`user-message`,message:{id:r,role:n,parts:[a]}}]:[]},je=(e,t)=>(e===`text`||e===`file`||oe(e))&&`discrete`in t,Me=(e,t,n)=>{let r=n;return[{kind:`tool-result`,codecMessageId:e,payload:{toolCallId:t.strOr(`toolCallId`,``),output:r?.output}}]},Ne=(e,t,n)=>{let r=n;return[{kind:`tool-result-error`,codecMessageId:e,payload:{toolCallId:t.strOr(`toolCallId`,``),message:r?.message??``}}]},Pe=(e,t)=>[{kind:`tool-approval-response`,codecMessageId:e,payload:B({toolCallId:t.strOr(`toolCallId`,``),approved:t.bool(`approved`)??!1,reason:t.str(`reason`)})}],Fe=(e,t,n,r,i)=>{switch(e){case`start`:return pe(t,r,i);case`start-step`:return me(r,i);case`finish-step`:return he(r,i);case`finish`:return ge(t,r,i);case`error`:return _e(n,r,i);case`abort`:return ve(n,r,i);case`message-metadata`:return ye(t);case`file`:return be(t,n);case`source-url`:return xe(t,n);case`source-document`:return Se(t);case`tool-input`:return ke(t,n,r,i);case`tool-input-error`:return Ce(t,n);case`tool-output-available`:return we(t,n);case`tool-output-error`:return Te(t,n);case`tool-approval-request`:return Ee(t);case`tool-output-denied`:return De(t);default:return oe(e)?Oe(e,t,n):[]}},Ie=(e,t,n)=>{if(je(e,t.transportHeaders??{}))return Ae(t);let r=t.transportHeaders?.[`codec-message-id`]??``;switch(e){case`tool-result`:return Me(r,n,t.data);case`tool-result-error`:return Ne(r,n,t.data);case`tool-approval-response`:return Pe(r,n);case`regenerate`:return[];default:return[]}},Le=(e,t)=>{let n=W(e.codecHeaders??{}),r=e.transportHeaders?.[`run-id`]??``,i=n.strOr(`type`,``);return e.name===`ai-input`?Ie(i,e,n):e.name===`ai-output`?Fe(i,n,e.data,r,t):[]},Re=e=>({buildStartEvents:t=>{let n=t.transportHeaders[`run-id`]??``,r=W(t.codecHeaders).str(`messageId`);return[...e.ensurePhases(n,{messageId:r}),le(t)]},buildDeltaEvents:(e,t)=>[ue(e,t)],buildEndEvents:(e,t)=>[de(e,t)],decodeDiscrete:t=>Le(t,e)}),ze=e=>`kind`in e,Be=class{constructor(e={}){this._core=ne(Re(fe()),e)}decode(e){let t=this._core.decode(e),n=[],r=[];for(let e of t)ze(e)?n.push(e):r.push(e);return{inputs:n,outputs:r}}},Ve=(e={})=>new Be(e),G=function(e){return e[e.BadRequest=4e4]=`BadRequest`,e[e.InvalidArgument=40003]=`InvalidArgument`,e[e.InsufficientCapability=40160]=`InsufficientCapability`,e[e.EncoderRecoveryFailed=104e3]=`EncoderRecoveryFailed`,e[e.SessionSubscriptionError=104001]=`SessionSubscriptionError`,e[e.CancelListenerError=104002]=`CancelListenerError`,e[e.RunLifecycleError=104003]=`RunLifecycleError`,e[e.SessionClosed=104004]=`SessionClosed`,e[e.SessionSendFailed=104005]=`SessionSendFailed`,e[e.ChannelContinuityLost=104006]=`ChannelContinuityLost`,e[e.ChannelNotReady=104007]=`ChannelNotReady`,e[e.StreamError=104008]=`StreamError`,e[e.InputEventNotFound=104010]=`InputEventNotFound`,e}({}),He=(e,t)=>e.code===t,Ue=class{constructor(e,t={}){this._trackers=new Map,this._pending=[],this._closed=!1,this._writer=e,this._defaultClientId=t.clientId,this._defaultExtras=t.extras,this._onMessageHook=t.onMessage??(()=>{}),this._logger=t.logger?.withContext({component:`EncoderCore`})}async publishDiscrete(e,t){this._assertNotClosed(),this._logger?.trace(`DefaultEncoderCore.publishDiscrete();`,{name:e.name});let n=this._buildDiscreteMessage(e,t);return this._writer.publish(n)}async publishDiscreteBatch(e,t){this._assertNotClosed(),this._logger?.trace(`DefaultEncoderCore.publishDiscreteBatch();`,{count:e.length});let n=e.map(e=>this._buildDiscreteMessage(e,t,!0));return this._writer.publish(n)}async startStream(e,n,r){this._assertNotClosed(),this._logger?.trace(`DefaultEncoderCore.startStream();`,{name:n.name,streamId:e});let i=this._buildTransport(n.transportHeaders,r);i[u]=`true`,i[d]=`streaming`,i[f]=e;let a=n.codecHeaders??{},o=this._resolveClientId(r),s={name:n.name,data:n.data,extras:{ai:this._aiExtras(i,a)},...o?{clientId:o}:{}};this._invokeOnMessage(s);let c=(await this._writer.publish(s)).serials[0];if(!c)throw new t.ErrorInfo(`unable to start stream; no serial returned for stream '${n.name}' (streamId: ${e})`,G.BadRequest,400);this._trackers.set(e,{serial:c,name:n.name,streamId:e,accumulated:n.data,persistentTransport:i,persistentCodec:a,cancelled:!1}),this._logger?.debug(`DefaultEncoderCore.startStream(); stream started`,{name:n.name,streamId:e,serial:c})}appendStream(e,n){this._assertNotClosed();let r=this._trackers.get(e);if(!r)throw new t.ErrorInfo(`unable to append to stream; no active stream for streamId '${e}'`,G.InvalidArgument,400);r.accumulated+=n;let i={serial:r.serial,data:n,extras:{ai:this._aiExtras({...r.persistentTransport},{...r.persistentCodec})}};this._invokeOnMessage(i);let a=this._writer.appendMessage(i);this._pending.push({promise:a,streamId:e})}async closeStream(e,n){this._assertNotClosed(),this._logger?.trace(`DefaultEncoderCore.closeStream();`,{streamId:e});let r=this._trackers.get(e);if(!r)throw new t.ErrorInfo(`unable to close stream; no active stream for streamId '${e}'`,G.InvalidArgument,400);r.accumulated+=n.data;let{transport:i,codec:a}=this._buildClosing(r,n);i[d]=`complete`;let o={serial:r.serial,data:n.data,extras:{ai:this._aiExtras(i,a)}};this._invokeOnMessage(o);let s=this._writer.appendMessage(o);this._pending.push({promise:s,streamId:e}),await this._flushPending(),this._logger?.debug(`DefaultEncoderCore.closeStream(); stream closed`,{streamId:e})}async cancelStream(e,n){this._assertNotClosed(),this._logger?.trace(`DefaultEncoderCore.cancelStream();`,{streamId:e});let r=this._trackers.get(e);if(!r)throw new t.ErrorInfo(`unable to cancel stream; no active stream for streamId '${e}'`,G.InvalidArgument,400);r.cancelled=!0;let{transport:i,codec:a}=this._buildClosing(r,void 0,n);i[d]=`cancelled`;let o={serial:r.serial,data:``,extras:{ai:this._aiExtras(i,a)}};this._invokeOnMessage(o);let s=this._writer.appendMessage(o);this._pending.push({promise:s,streamId:e}),await this._flushPending(),this._logger?.debug(`DefaultEncoderCore.cancelStream(); stream cancelled`,{streamId:e})}async cancelAllStreams(e){this._assertNotClosed(),this._logger?.trace(`DefaultEncoderCore.cancelAllStreams();`,{streamCount:this._trackers.size});for(let t of this._trackers.values()){t.cancelled=!0;let{transport:n,codec:r}=this._buildClosing(t,void 0,e);n[d]=`cancelled`;let i={serial:t.serial,data:``,extras:{ai:this._aiExtras(n,r)}};this._invokeOnMessage(i);let a=this._writer.appendMessage(i);this._pending.push({promise:a,streamId:t.streamId})}await this._flushPending()}async _flushPending(){if(this._flushPromise)return this._flushPromise;let e=this._pending;if(this._pending=[],e.length!==0){this._logger?.trace(`DefaultEncoderCore._flushPending();`,{count:e.length}),this._flushPromise=this._doFlush(e);try{await this._flushPromise}finally{this._flushPromise=void 0}}}async _doFlush(e){let n=await Promise.allSettled(e.map(async e=>e.promise)),r=new Set;for(let[t,i]of n.entries()){let n=e[t];n&&i.status===`rejected`&&r.add(n.streamId)}if(r.size===0){this._logger?.debug(`DefaultEncoderCore._flushPending(); all appends succeeded`);return}this._logger?.warn(`DefaultEncoderCore._flushPending(); recovering failed appends`,{failedStreams:[...r]});let i=[];for(let e of r){let t=this._trackers.get(e);if(!t)continue;let n=t.cancelled?`cancelled`:`complete`,r={serial:t.serial,data:t.accumulated,extras:{ai:this._aiExtras({...t.persistentTransport,[d]:n},{...t.persistentCodec})}};try{await this._writer.updateMessage(r)}catch(t){i.push({streamId:e,error:t})}}if(i.length>0){let e=i.map(e=>e.streamId).join(`, `);throw this._logger?.error(`DefaultEncoderCore._flushPending(); recovery failed`,{failedStreams:e}),new t.ErrorInfo(`unable to flush pending appends; recovery failed for stream(s): ${e}`,G.EncoderRecoveryFailed,500)}}async close(){if(!this._closed){this._logger?.trace(`DefaultEncoderCore.close();`),this._closed=!0;try{await this._flushPending()}finally{this._trackers.clear()}this._logger?.debug(`DefaultEncoderCore.close(); encoder closed`)}}_invokeOnMessage(e){try{this._onMessageHook(e)}catch(e){this._logger?.error(`DefaultEncoderCore._invokeOnMessage(); hook threw`,{error:e})}}_assertNotClosed(){if(this._closed)throw new t.ErrorInfo(`unable to write to encoder; encoder has been closed`,G.InvalidArgument,400)}_resolveClientId(e){return e?.clientId??this._defaultClientId}_buildTransport(e,t){let n={...I(this._defaultExtras?.headers,t?.extras?.headers),...e};return t?.messageId!==void 0&&(n[_]=t.messageId),n}_aiExtras(e,t){return Object.keys(t).length>0?{transport:e,codec:t}:{transport:e}}_buildDiscreteMessage(e,t,n=!1){let r=this._buildTransport(e.transportHeaders,t);r[u]=`false`,n&&(r[p]=`true`);let i=this._resolveClientId(t),a={name:e.name,data:e.data,extras:{ai:this._aiExtras(r,e.codecHeaders??{}),...e.ephemeral?{ephemeral:!0}:{}},...i?{clientId:i}:{}};return this._invokeOnMessage(a),a}_buildClosing(e,t,n){let r=I(this._defaultExtras?.headers,n?.extras?.headers);return{transport:{...e.persistentTransport,...r,...t?.transportHeaders},codec:{...e.persistentCodec,...t?.codecHeaders}}}},We=(e,t={})=>new Ue(e,t),Ge=class{constructor(e,t={}){this._cancelled=!1,this._core=We(e,t),this._messageId=t.messageId}async publishInput(e,t){switch(e.kind){case`user-message`:await this._publishUserMessage(e,t);return;case`regenerate`:await this._publishRegenerate(t);return;case`tool-result`:await this._publishToolResult(e,t);return;case`tool-result-error`:await this._publishToolResultError(e,t);return;case`tool-approval-response`:await this._publishToolApprovalResponse(e,t);return}}async publishOutput(e,t){await this._publishChunk(e,t)}async cancel(e){this._cancelled||(this._cancelled=!0,await this._core.cancelAllStreams(),await this._core.publishDiscrete({name:A,data:e??``,codecHeaders:H().str(`type`,`abort`).build(),transportHeaders:{[d]:`cancelled`}}))}async close(){await this._core.close()}async _publishChunk(e,n){switch(e.type){case`text-start`:{let t=H().str(`type`,`text`).str(`id`,e.id).json(`providerMetadata`,e.providerMetadata).build();await this._core.startStream(e.id,{name:A,data:``,codecHeaders:t},n);return}case`reasoning-start`:{let t=H().str(`type`,`reasoning`).str(`id`,e.id).json(`providerMetadata`,e.providerMetadata).build();await this._core.startStream(e.id,{name:A,data:``,codecHeaders:t},n);return}case`tool-input-start`:{let t=H().str(`type`,`tool-input`).str(`toolCallId`,e.toolCallId).str(`toolName`,e.toolName).bool(`dynamic`,e.dynamic).str(`title`,e.title).bool(`providerExecuted`,e.providerExecuted).json(`providerMetadata`,e.providerMetadata).build();await this._core.startStream(e.toolCallId,{name:A,data:``,codecHeaders:t},n);return}case`text-delta`:this._core.appendStream(e.id,e.delta);return;case`reasoning-delta`:this._core.appendStream(e.id,e.delta);return;case`tool-input-delta`:this._core.appendStream(e.toolCallId,e.inputTextDelta);return;case`text-end`:{let t=H().str(`type`,`text`).str(`id`,e.id).json(`providerMetadata`,e.providerMetadata).build();await this._core.closeStream(e.id,{name:A,data:``,codecHeaders:t});return}case`reasoning-end`:{let t=H().str(`type`,`reasoning`).str(`id`,e.id).json(`providerMetadata`,e.providerMetadata).build();await this._core.closeStream(e.id,{name:A,data:``,codecHeaders:t});return}case`tool-input-available`:try{let t=H().str(`type`,`tool-input`).str(`toolCallId`,e.toolCallId).str(`toolName`,e.toolName).json(`providerMetadata`,e.providerMetadata).build();await this._core.closeStream(e.toolCallId,{name:A,data:``,codecHeaders:t})}catch(n){if(!(n instanceof t.ErrorInfo&&He(n,G.InvalidArgument)))throw n;let r=H().str(`type`,`tool-input`).str(`toolCallId`,e.toolCallId).str(`toolName`,e.toolName).bool(`dynamic`,e.dynamic).str(`title`,e.title).bool(`providerExecuted`,e.providerExecuted).json(`providerMetadata`,e.providerMetadata).build();await this._core.publishDiscrete({name:A,data:e.input,codecHeaders:r})}return;case`start`:{let t=H().str(`type`,`start`).str(`messageId`,e.messageId??this._messageId).json(`messageMetadata`,e.messageMetadata).build();await this._core.publishDiscrete({name:A,data:``,codecHeaders:t},n);return}case`start-step`:{let e=H().str(`type`,`start-step`).build();await this._core.publishDiscrete({name:A,data:``,codecHeaders:e},n);return}case`finish-step`:{let e=H().str(`type`,`finish-step`).build();await this._core.publishDiscrete({name:A,data:``,codecHeaders:e},n);return}case`finish`:{let t=H().str(`type`,`finish`).str(`finishReason`,e.finishReason).json(`messageMetadata`,e.messageMetadata).build();await this._core.publishDiscrete({name:A,data:``,codecHeaders:t},n);return}case`error`:{let t=H().str(`type`,`error`).build();await this._core.publishDiscrete({name:A,data:e.errorText,codecHeaders:t},n);return}case`abort`:this._cancelled=!0,await this._core.cancelAllStreams(n),await this._core.publishDiscrete({name:A,data:e.reason??``,codecHeaders:H().str(`type`,`abort`).build(),transportHeaders:{[d]:`cancelled`}},n);return;case`message-metadata`:{let t=H().str(`type`,`message-metadata`).json(`messageMetadata`,e.messageMetadata).build();await this._core.publishDiscrete({name:A,data:``,codecHeaders:t},n);return}case`tool-input-error`:{let t=H().str(`type`,`tool-input-error`).str(`toolCallId`,e.toolCallId).str(`toolName`,e.toolName).bool(`dynamic`,e.dynamic).str(`title`,e.title).bool(`providerExecuted`,e.providerExecuted).json(`providerMetadata`,e.providerMetadata).build();await this._core.publishDiscrete({name:A,data:{errorText:e.errorText,input:e.input},codecHeaders:t},n);return}case`tool-output-available`:case`tool-output-error`:case`tool-approval-request`:case`tool-output-denied`:await this._core.publishDiscrete(Ke(e),n);return;case`file`:{let t=H().str(`type`,`file`).str(`mediaType`,e.mediaType).json(`providerMetadata`,e.providerMetadata).build();await this._core.publishDiscrete({name:A,data:e.url,codecHeaders:t},n);return}case`source-url`:{let t=H().str(`type`,`source-url`).str(`sourceId`,e.sourceId).str(`title`,e.title).json(`providerMetadata`,e.providerMetadata).build();await this._core.publishDiscrete({name:A,data:e.url,codecHeaders:t},n);return}case`source-document`:{let t=H().str(`type`,`source-document`).str(`sourceId`,e.sourceId).str(`mediaType`,e.mediaType).str(`title`,e.title).str(`filename`,e.filename).json(`providerMetadata`,e.providerMetadata).build();await this._core.publishDiscrete({name:A,data:``,codecHeaders:t},n);return}default:if(e.type.startsWith(`data-`)){let t=e,r=H().str(`type`,t.type).str(`id`,t.id).bool(`transient`,t.transient).build(),i=t.transient===!0;await this._core.publishDiscrete({name:A,data:t.data,codecHeaders:r,ephemeral:i},n);return}throw new t.ErrorInfo(`unable to publish output; unsupported chunk type '${e.type}'`,G.InvalidArgument,400)}}async _publishUserMessage(e,t){let n=qe(e.message);for(let e of n)e.transportHeaders={...e.transportHeaders,[b]:`user`};await this._core.publishDiscreteBatch(n,t)}async _publishRegenerate(e){let t=H().str(`type`,`regenerate`).build();await this._core.publishDiscrete({name:j,data:``,codecHeaders:t},e)}async _publishToolResult(e,t){let n=H().str(`type`,`tool-result`).str(`toolCallId`,e.payload.toolCallId).build();await this._core.publishDiscrete({name:j,data:{output:e.payload.output},codecHeaders:n},t)}async _publishToolResultError(e,t){let n=H().str(`type`,`tool-result-error`).str(`toolCallId`,e.payload.toolCallId).build();await this._core.publishDiscrete({name:j,data:{message:e.payload.message},codecHeaders:n},t)}async _publishToolApprovalResponse(e,t){let n=H().str(`type`,`tool-approval-response`).str(`toolCallId`,e.payload.toolCallId).bool(`approved`,e.payload.approved).str(`reason`,e.payload.reason).build();await this._core.publishDiscrete({name:j,data:``,codecHeaders:n},t)}},Ke=e=>{switch(e.type){case`tool-output-available`:{let t=H().str(`type`,`tool-output-available`).str(`toolCallId`,e.toolCallId).bool(`dynamic`,e.dynamic).bool(`providerExecuted`,e.providerExecuted).bool(`preliminary`,e.preliminary).build();return{name:A,data:{output:e.output},codecHeaders:t}}case`tool-output-error`:{let t=H().str(`type`,`tool-output-error`).str(`toolCallId`,e.toolCallId).bool(`dynamic`,e.dynamic).bool(`providerExecuted`,e.providerExecuted).build();return{name:A,data:{errorText:e.errorText},codecHeaders:t}}case`tool-approval-request`:return{name:A,data:``,codecHeaders:H().str(`type`,`tool-approval-request`).str(`toolCallId`,e.toolCallId).str(`approvalId`,e.approvalId).build()};case`tool-output-denied`:return{name:A,data:``,codecHeaders:H().str(`type`,`tool-output-denied`).str(`toolCallId`,e.toolCallId).build()}}},qe=e=>{let t=e.id,r=[];for(let i of e.parts)switch(i.type){case`text`:r.push({name:j,data:i.text,codecHeaders:H().str(`type`,`text`).str(`messageId`,t).build()});break;case`file`:r.push({name:j,data:i.url,codecHeaders:H().str(`type`,`file`).str(`messageId`,t).str(`mediaType`,i.mediaType).build()});break;default:(0,n.isDataUIPart)(i)&&r.push({name:j,data:i.data,codecHeaders:H().str(`type`,i.type).str(`messageId`,t).str(`id`,i.id).build()});break}return r.length===0&&r.push({name:j,data:``,codecHeaders:H().str(`type`,`text`).str(`messageId`,t).build()}),r},Je=(e,t={})=>new Ge(e,t),K=e=>B({type:`dynamic-tool`,toolCallId:e.toolCallId,toolName:e.toolName,title:e.title,providerExecuted:e.providerExecuted}),q=(e,t)=>{let n=K(e);switch(t.type){case`tool-output-available`:return B({...n,state:`output-available`,input:e.input,output:t.output,preliminary:t.preliminary});case`tool-output-error`:return{...n,state:`output-error`,input:e.input,errorText:t.errorText};case`tool-output-denied`:return{...n,state:`output-denied`,input:e.input,approval:{id:``,approved:!1}};case`tool-approval-request`:return{...n,state:`approval-requested`,input:e.input,approval:{id:t.approvalId}}}},Ye=()=>({messages:[],conflictSerials:new Map,trackers:new Map,pendingToolResolutions:[]}),Xe=(e,t,n)=>{if(n.serial){let r=Qe(t,n);if(r!==void 0){let t=e.conflictSerials.get(r);if(t!==void 0&&n.serial<=t)return e;e.conflictSerials.set(r,n.serial)}}if(Ze(t))switch(t.kind){case`user-message`:$e(e,t.message,n);break;case`regenerate`:break;case`tool-result`:et(e,t,n);break;case`tool-result-error`:tt(e,t,n);break;case`tool-approval-response`:nt(e,t,n);break}else ot(e,t,n);return e.pendingToolResolutions.length>0&&at(e),e},Ze=e=>`kind`in e,Qe=(e,t)=>{if(Ze(e))switch(e.kind){case`user-message`:return t.messageId===void 0?void 0:`user-msg:${t.messageId}`;case`tool-approval-response`:return`tool-approval:${e.payload.toolCallId}`;case`tool-result`:case`tool-result-error`:return`tool-output:${e.payload.toolCallId}`;case`regenerate`:return}switch(e.type){case`tool-input-start`:case`tool-input-available`:case`tool-input-error`:return`${e.type}:${e.toolCallId}`;case`tool-output-available`:case`tool-output-error`:case`tool-output-denied`:case`tool-approval-request`:return`tool-output:${e.toolCallId}`;case`text-start`:case`text-end`:case`reasoning-start`:case`reasoning-end`:return`${e.type}:${t.messageId??``}:${e.id}`;case`finish`:case`message-metadata`:return`${e.type}:${t.messageId??``}`;default:return}},$e=(e,t,n)=>{let r=n.messageId;if(r===void 0)return e.messages.push({codecMessageId:t.id,message:t}),e;let i=e.messages.findIndex(e=>e.codecMessageId===r);return i===-1?e.messages.push({codecMessageId:r,message:t}):e.messages[i]={codecMessageId:r,message:t},e},et=(e,t,n)=>{let{toolCallId:r,output:i}=t.payload,a=J(e,t.codecMessageId,r);return a?(a.message.parts[a.tracker.partIndex]=q(a.part,{type:`tool-output-available`,toolCallId:r,output:i}),e):(e.pendingToolResolutions.push({targetCodecMessageId:t.codecMessageId,toolCallId:r,serial:n.serial,resolution:{kind:`tool-result`,output:i}}),e)},tt=(e,t,n)=>{let{toolCallId:r,message:i}=t.payload,a=J(e,t.codecMessageId,r);return a?(a.message.parts[a.tracker.partIndex]=q(a.part,{type:`tool-output-error`,toolCallId:r,errorText:i}),e):(e.pendingToolResolutions.push({targetCodecMessageId:t.codecMessageId,toolCallId:r,serial:n.serial,resolution:{kind:`tool-result-error`,message:i}}),e)},nt=(e,t,n)=>{let{toolCallId:r,approved:i,reason:a}=t.payload,o=J(e,t.codecMessageId,r);return o?(o.message.parts[o.tracker.partIndex]=it(o.part,i,a),e):(e.pendingToolResolutions.push({targetCodecMessageId:t.codecMessageId,toolCallId:r,serial:n.serial,resolution:{kind:`tool-approval-response`,approved:i,...a===void 0?{}:{reason:a}}}),e)},J=(e,t,n)=>{let r=e.messages.find(e=>e.codecMessageId===t);if(!r)return;let i=X(e,t),a=Z(r.message,i,n);if(a)return{message:r.message,tracker:a.tracker,part:a.part}},rt=(e,t)=>{for(let n of e.messages){let r=e.trackers.get(n.codecMessageId);if(!r)continue;let i=Z(n.message,r,t);if(i)return{message:n.message,tracker:i.tracker,part:i.part}}},it=(e,t,n)=>t?{...K(e),state:`approval-responded`,input:`input`in e?e.input:void 0,approval:{id:`approval`in e&&e.approval?e.approval.id:``,approved:!0,...n===void 0?{}:{reason:n}}}:q(e,{type:`tool-output-denied`,toolCallId:e.toolCallId,...n===void 0?{}:{reason:n}}),at=e=>{let t=[];for(let n of e.pendingToolResolutions){let r=J(e,n.targetCodecMessageId,n.toolCallId);if(!r){t.push(n);continue}switch(n.resolution.kind){case`tool-result`:r.message.parts[r.tracker.partIndex]=q(r.part,{type:`tool-output-available`,toolCallId:n.toolCallId,output:n.resolution.output});break;case`tool-result-error`:r.message.parts[r.tracker.partIndex]=q(r.part,{type:`tool-output-error`,toolCallId:n.toolCallId,errorText:n.resolution.message});break;case`tool-approval-response`:r.message.parts[r.tracker.partIndex]=it(r.part,n.resolution.approved,n.resolution.reason);break}}e.pendingToolResolutions=t},ot=(e,t,n)=>{let r=n.messageId;if(r===void 0)return e;switch(t.type){case`start`:case`start-step`:case`finish-step`:case`finish`:case`abort`:case`error`:case`message-metadata`:return st(e,t,r);case`text-start`:case`text-delta`:case`text-end`:case`reasoning-start`:case`reasoning-delta`:case`reasoning-end`:return ct(e,t,r);case`tool-input-start`:case`tool-input-delta`:case`tool-input-available`:case`tool-input-error`:return lt(e,t,r);case`tool-output-available`:case`tool-output-error`:case`tool-output-denied`:case`tool-approval-request`:return ut(e,t,r);case`file`:case`source-url`:case`source-document`:return dt(e,t,r);default:return t.type.startsWith(`data-`)?ft(e,t,r):e}},Y=(e,t)=>{let n=e.messages.find(e=>e.codecMessageId===t);return n||(n={codecMessageId:t,message:{id:t,role:`assistant`,parts:[]}},e.messages.push(n)),n.message},X=(e,t)=>{let n=e.trackers.get(t);return n||(n={text:new Map,reasoning:new Map,tools:new Map},e.trackers.set(t,n)),n},Z=(e,t,n)=>{let r=t.tools.get(n);if(!r)return;let i=e.parts[r.partIndex];if(i?.type===`dynamic-tool`)return{tracker:r,part:i}},st=(e,t,n)=>{switch(t.type){case`start`:{let r=Y(e,n);return t.messageId!==void 0&&(r.id=t.messageId),t.messageMetadata!==void 0&&(r.metadata=t.messageMetadata),e}case`start-step`:return Y(e,n).parts.push({type:`step-start`}),e;case`finish-step`:{let t=e.trackers.get(n);return t&&(t.text.clear(),t.reasoning.clear()),e}case`finish`:{let r=e.messages.find(e=>e.codecMessageId===n)?.message;return r&&t.messageMetadata!==void 0&&(r.metadata=t.messageMetadata),e}case`abort`:case`error`:return e;case`message-metadata`:{let r=e.messages.find(e=>e.codecMessageId===n)?.message;return r&&t.messageMetadata!==void 0&&(r.metadata=t.messageMetadata),e}}},ct=(e,t,n)=>{let r=Y(e,n),i=X(e,n),a=t.type.startsWith(`text-`),o=a?`text`:`reasoning`,s=a?i.text:i.reasoning;switch(t.type){case`text-start`:case`reasoning-start`:return s.set(t.id,r.parts.length),r.parts.push({type:o,text:``}),e;case`text-delta`:case`reasoning-delta`:{let n=s.get(t.id);if(n===void 0)return e;let i=r.parts[n];return i?.type===o&&(i.text+=t.delta),e}case`text-end`:case`reasoning-end`:return s.delete(t.id),e}},lt=(e,t,n)=>{let r=Y(e,n),i=X(e,n);switch(t.type){case`tool-input-start`:{let n=r.parts.length;return r.parts.push({...K(t),state:`input-streaming`,input:void 0}),i.tools.set(t.toolCallId,{partIndex:n,inputText:``}),e}case`tool-input-delta`:{let n=i.tools.get(t.toolCallId);if(!n)return e;n.inputText+=t.inputTextDelta;let a;try{a=JSON.parse(n.inputText)}catch{a=void 0}let o=Z(r,i,t.toolCallId);return o&&(r.parts[o.tracker.partIndex]={...K(o.part),state:`input-streaming`,input:a}),e}case`tool-input-available`:{let n=Z(r,i,t.toolCallId);return n&&(r.parts[n.tracker.partIndex]={...K(n.part),state:`input-available`,input:t.input}),e}case`tool-input-error`:{let n=Z(r,i,t.toolCallId);if(n)r.parts[n.tracker.partIndex]={...K(n.part),state:`output-error`,input:t.input,errorText:t.errorText};else{let e=r.parts.length;r.parts.push({...K(t),state:`output-error`,input:t.input,errorText:t.errorText}),i.tools.set(t.toolCallId,{partIndex:e,inputText:``})}return e}}},ut=(e,t,n)=>{if(t.type===`tool-output-available`||t.type===`tool-output-error`){let n=rt(e,t.toolCallId);return n&&(n.message.parts[n.tracker.partIndex]=q(n.part,t)),e}let r=Y(e,n),i=Z(r,X(e,n),t.toolCallId);return i&&(r.parts[i.tracker.partIndex]=q(i.part,t)),e},dt=(e,t,n)=>{let r=Y(e,n);switch(t.type){case`file`:return r.parts.push({type:`file`,mediaType:t.mediaType,url:t.url}),e;case`source-url`:return r.parts.push(B({type:`source-url`,sourceId:t.sourceId,url:t.url,title:t.title})),e;case`source-document`:return r.parts.push(B({type:`source-document`,sourceId:t.sourceId,mediaType:t.mediaType,title:t.title,filename:t.filename})),e}},ft=(e,t,n)=>{if(t.transient)return e;let r=Y(e,n),i=B({type:t.type,id:t.id,data:t.data});if(t.id!==void 0){let n=r.parts.findIndex(e=>e.type===t.type&&`id`in e&&e.id===t.id);if(n!==-1)return r.parts[n]=i,e}return r.parts.push(i),e},Q={adapterTag:`vercel-ai-sdk-ui-message`,init:Ye,fold:Xe,createEncoder:Je,createDecoder:Ve,getMessages:e=>e.messages,createUserMessage:e=>({kind:`user-message`,message:e}),createRegenerate:(e,t)=>({kind:`regenerate`,target:e,parent:t}),createToolResult:(e,t)=>({kind:`tool-result`,codecMessageId:e,payload:t}),createToolResultError:(e,t)=>({kind:`tool-result-error`,codecMessageId:e,payload:t}),createToolApprovalResponse:(e,t)=>({kind:`tool-approval-response`,codecMessageId:e,payload:t})},pt=e=>({logAction:(t,n,r)=>{e.error(n,{detail:r})},shouldLog:()=>!0}),mt=t.Realtime.EventEmitter,ht=class extends mt{constructor(e){super(pt(e))}},gt=function(e){return e.Trace=`trace`,e.Debug=`debug`,e.Info=`info`,e.Warn=`warn`,e.Error=`error`,e.Silent=`silent`,e}({}),_t=(e,t,n)=>{let r=n?`, context: ${JSON.stringify(n)}`:``,i=`[${new Date().toISOString()}] ${t.valueOf().toUpperCase()} ably-ai-transport: ${e}${r}`;switch(t){case`trace`:case`debug`:console.log(i);break;case`info`:console.info(i);break;case`warn`:console.warn(i);break;case`error`:console.error(i);break;case`silent`:break}},vt=e=>new bt(e.logHandler??_t,e.logLevel),yt=new Map([[`trace`,0],[`debug`,1],[`info`,2],[`warn`,3],[`error`,4],[`silent`,5]]),bt=class e{constructor(e,n,r){this._handler=e,this._context=r;let i=yt.get(n);if(i===void 0)throw new t.ErrorInfo(`unable to create logger; invalid log level: ${n}`,G.InvalidArgument,400);this._levelNumber=i}trace(e,t){this._write(e,`trace`,0,t)}debug(e,t){this._write(e,`debug`,1,t)}info(e,t){this._write(e,`info`,2,t)}warn(e,t){this._write(e,`warn`,3,t)}error(e,t){this._write(e,`error`,4,t)}withContext(t){let n=[...yt.entries()].find(([,e])=>e===this._levelNumber)?.[0]??`error`;return new e(this._handler,n,this._mergeContext(t))}_write(e,t,n,r){n>=this._levelNumber&&this._handler(e,t,this._mergeContext(r))}_mergeContext(e){return this._context?e?{...this._context,...e}:this._context:e??void 0}},xt=e=>e.type===`finish`||e.type===`error`||e.type===`abort`,St=(e,n,r)=>{let i={},a=[],o=new ReadableStream({start:e=>{i.controller=e},cancel:()=>{u()}}),{controller:s}=i;if(!s)throw new t.ErrorInfo(`unable to create run stream; ReadableStream start() was not called synchronously`,G.SessionSubscriptionError,500);let c;n.then(e=>{c=e},()=>{});let l=!1,u=()=>{for(let e of a)e();a.length=0},d=e=>{if(!l){l=!0;try{e()}catch{}u()}},f=()=>{d(()=>{s.close()})},p=e=>{d(()=>{s.error(e)})};return a.push(e.tree.on(`output`,e=>{if(e.inputCodecMessageId===r)for(let t of e.events){try{s.enqueue(t)}catch{f();return}if(xt(t)){f();return}}}),e.tree.on(`run`,e=>{e.type===`end`&&c!==void 0&&e.runId===c&&f()}),e.on(`error`,e=>{p(e)})),{stream:o,close:f,error:p}},Ct=e=>{let t,n=new Promise(e=>{t=e}),r=new TransformStream({flush:()=>{t()}}),i=new AbortController;return e.pipeTo(r.writable,{signal:i.signal,preventCancel:!0}).catch(()=>{t()}),{stream:r.readable,done:n,fail:e=>{i.abort(e)}}},wt=e=>(e.type===`dynamic-tool`||e.type.startsWith(`tool-`))&&`toolCallId`in e&&`state`in e,Tt=e=>e.role===`assistant`&&e.parts.some(e=>wt(e)&&(e.state===`input-streaming`||e.state===`input-available`||e.state===`approval-requested`)),Et=new Set([`input-streaming`,`input-available`,`approval-requested`]),Dt=(e,t)=>{let n=[];for(let r of t){if(r.role!==`assistant`)continue;let t=e.find(e=>e.message.id===r.id);if(!t)continue;let{codecMessageId:i,message:a}=t;for(let e of r.parts){if(!wt(e))continue;let t=a.parts.find(t=>wt(t)&&t.toolCallId===e.toolCallId);if(e.state===`approval-responded`&&(!t||t.state===`approval-requested`)){n.push(Q.createToolApprovalResponse(i,{toolCallId:e.toolCallId,approved:e.approval.approved,...e.approval.reason===void 0?{}:{reason:e.approval.reason}}));continue}e.state!==`output-available`&&e.state!==`output-error`||t&&!Et.has(t.state)||(e.state===`output-available`?n.push(Q.createToolResult(i,{toolCallId:e.toolCallId,output:e.output})):n.push(Q.createToolResultError(i,{toolCallId:e.toolCallId,message:e.errorText})))}}return n},Ot=(e,t)=>{let n=e.findIndex(e=>e.message.id===t);if(!(n<=0))return e[n-1]?.codecMessageId},kt=(e,n)=>{let r=n?.api??`/api/chat`,i=n?.fetch??globalThis.fetch.bind(globalThis),a=n?.credentials,o=!1,s=new ht(vt({logLevel:gt.Silent})),c=e=>{o=e,s.emit(`streaming`,e)};return{sendMessages:async o=>{let{messages:s,abortSignal:l,trigger:u,messageId:d}=o,f=e.view.getMessages(),p=new Map(f.map(e=>[e.message.id,e.codecMessageId])),m=e=>p.get(e),h=s.at(-1),g=!!h&&p.has(h.id),_=u===`submit-message`&&h?.role===`assistant`&&g,v=u===`submit-message`&&!d&&h?.role===`user`?s.at(-2):void 0,y=v&&Tt(v)&&p.has(v.id)?v.id:void 0,b,x;if(u===`regenerate-message`||_)b=[],x=s;else{if(s.length===0)throw new t.ErrorInfo(`unable to send messages; messages array is empty for submit-message trigger`,G.InvalidArgument,400);b=[s.at(-1)],x=y?s.slice(0,-2):s.slice(0,-1)}let S,C;u===`submit-message`&&d&&!_?(S=m(d),C=Ot(f,d)):y&&(S=m(y),C=Ot(f,y));let w,T;if(n?.prepareSendMessagesRequest){let e=n.prepareSendMessagesRequest({chatId:o.chatId,trigger:u,messageId:d,history:x,messages:b,forkOf:S,parent:C});w=e.body??{},T=e.headers}else w={},T=void 0;let E={};if(S!==void 0&&(E.forkOf=S),C!==void 0&&(E.parent=C),_){let t=m(h.id),n=t===void 0?void 0:e.view.runOf(t);n&&(E.runId=n.runId)}let D;if(_){let t=Dt(f,s);D=await e.view.send(t,E)}else if(u===`regenerate-message`){if(d===void 0)throw new t.ErrorInfo(`unable to regenerate; regenerate-message trigger fired without messageId`,G.InvalidArgument,400);let n=m(d);if(n===void 0)throw new t.ErrorInfo(`unable to regenerate; message not visible: ${d}`,G.InvalidArgument,400);D=await e.view.regenerate(n,E)}else{let t=b.map(e=>Q.createUserMessage(e));D=await e.view.send(t,E)}let O=St(e,D.runId,D.inputCodecMessageId);if(l){let e=()=>{D.cancel(),O.close()};l.aborted?e():l.addEventListener(`abort`,e,{once:!0})}let{stream:ee,done:te,fail:k}=Ct(O.stream);c(!0),te.then(()=>{c(!1)});let A={...w,...D.toInvocation().toJSON()};return i(r,{method:`POST`,headers:{"Content-Type":`application/json`,...T},body:JSON.stringify(A),...a?{credentials:a}:{}}).then(e=>{e.ok||k(new t.ErrorInfo(`unable to send; HTTP POST to ${r} returned ${String(e.status)} ${e.statusText}`,G.SessionSendFailed,e.status))}).catch(e=>{let n=e instanceof t.ErrorInfo?e:void 0;k(new t.ErrorInfo(`unable to send; HTTP POST to ${r} failed: ${e instanceof Error?e.message:String(e)}`,G.SessionSendFailed,500,n))}),ee},reconnectToStream:()=>Promise.resolve(null),close:async()=>e.close(),get streaming(){return o},onStreamingChange:e=>(s.on(`streaming`,e),()=>{s.off(`streaming`,e)})}},At=`0.2.0`,jt=`ai-transport-js`,Mt=(e,t)=>{let n=e;return n.options.agents={...n.options.agents,...t},{params:{agent:Object.entries(t).map(([e,t])=>`${e}/${t}`).join(` `)}}},Nt=(e,t)=>{let n=t?.adapterTag,r={[jt]:At};return n&&(r[n]=At),Mt(e,r)},Pt=e=>{let t={[b]:e.role,[_]:e.codecMessageId};return e.runId!==void 0&&(t[m]=e.runId),e.runClientId!==void 0&&(t[v]=e.runClientId),e.parent&&(t[x]=e.parent),e.forkOf&&(t[S]=e.forkOf),e.regenerates&&(t[C]=e.regenerates),e.invocationId&&(t[h]=e.invocationId),e.inputClientId!==void 0&&(t[y]=e.inputClientId),e.inputCodecMessageId!==void 0&&(t[T]=e.inputCodecMessageId),e.inputEventId&&(t[g]=e.inputEventId),t},Ft=e=>{let t={[m]:e.runId,[v]:e.runClientId};return e.reason!==void 0&&(t[w]=e.reason),e.parent!==void 0&&(t[x]=e.parent),e.forkOf!==void 0&&(t[S]=e.forkOf),e.regenerates!==void 0&&(t[C]=e.regenerates),e.invocationId!==void 0&&(t[h]=e.invocationId),e.inputClientId!==void 0&&(t[y]=e.inputClientId),e.inputCodecMessageId!==void 0&&(t[T]=e.inputCodecMessageId),t},It=e=>e===`ai-run-start`||e===`ai-run-suspend`||e===`ai-run-resume`||e===`ai-run-end`,Lt=(e,t,n)=>{let r=t[m];if(!r)return;let i=t[`run-client-id`]??``;if(e===`ai-run-start`){let e=t[x],a=t[S],o=t[C];return{type:`start`,runId:r,clientId:i,serial:n,invocationId:t[`invocation-id`]??``,...e!==void 0&&{parent:e},...a!==void 0&&{forkOf:a},...o!==void 0&&{regenerates:o}}}if(e===`ai-run-suspend`)return{type:`suspend`,runId:r,clientId:i,serial:n,invocationId:t[`invocation-id`]??``};if(e===`ai-run-resume`)return{type:`resume`,runId:r,clientId:i,serial:n,invocationId:t[`invocation-id`]??``};if(e===`ai-run-end`){let e=t[`run-reason`]??`complete`;return{type:`end`,runId:r,clientId:i,serial:n,invocationId:t[`invocation-id`]??``,reason:e}}},Rt=(e,t,n)=>{if(e.has(t)||e.size<n)return;let r=e.keys().next().value;if(r!==void 0)return e.delete(r),r},zt=(e,t)=>{let n=[],r=new Set,i=t;for(;i!==void 0&&!r.has(i);)r.add(i),n.push(i),i=e.get(i)?.parentCodecMessageId;return n.toReversed()},Bt=(e,t,n)=>{let r=N(n),i=n.serial;if(It(n.name)){let t=Lt(n.name,r,i);return t&&e.applyRunLifecycle(t),t}let{inputs:a,outputs:o}=t.decode(n);(a.length>0||o.length>0||r[`run-id`])&&e.applyMessage({inputs:a,outputs:o},r,i)},Vt=(e,t,n,r,i)=>{let{inputs:a,outputs:o}=t.decode(r),s=n;for(let t of[...a,...o])s=e.fold(s,t,{serial:r.serial??``,messageId:i});return s},Ht=(e,t)=>{let n=new Set,r=[];for(let t of e)t.serial!==void 0&&!n.has(t.serial)&&(n.add(t.serial),r.push(t));if(t!==void 0)for(let e of t)e.serial!==void 0&&!n.has(e.serial)&&(n.add(e.serial),r.push(e));return r.toSorted(R)},Ut=async(e,t,n)=>{let r=[],i=await e.history({limit:t});for(r.push(...i.items);i.hasNext()&&r.length<n;){let e=await i.next();if(!e)break;r.push(...e.items),i=e}return r},Wt=(e,t,n,r)=>{let i=e.createDecoder(),a=e.init(),o=0;for(let s of t){let t=N(s);if(t[`run-id`]!==n||It(s.name))continue;let c=t[_];if(r!==void 0&&c===r)break;a=Vt(e,i,a,s,c??``),o++}return{projection:a,folded:o}},Gt=(e,t,n)=>{let r=e.createDecoder(),i=e.init();for(let a of t){let t=N(a);t[`run-id`]===void 0&&t[`codec-message-id`]===n&&(i=Vt(e,r,i,a,n))}return i},Kt=async e=>{let{channel:n,codec:r,runId:i,signal:a,logger:o,liveMessages:s}=e;if(a.aborted)throw new t.ErrorInfo(`unable to load run projection; run ${i} was cancelled`,G.InvalidArgument,400);await n.attach();let{projection:c,folded:l}=Wt(r,Ht(await Ut(n,200,2e3),s),i);return o?.debug(`loadRunProjection(); folded run events`,{runId:i,folded:l}),c},qt=async e=>{let{channel:n,codec:r,runId:i,signal:a,logger:o,liveMessages:s,assistantParentFallback:c,pageLimit:l,maxMessages:u}=e;if(a.aborted)throw new t.ErrorInfo(`unable to load conversation; run ${i} was cancelled`,G.InvalidArgument,400);let d=Ht(await Ut(n,l,u),s),f=new Map,p=new Map;for(let e of d){if(It(e.name))continue;let t=N(e),n=t[_];if(n===void 0)continue;let r=t[m];r!==void 0&&p.set(r,n),f.has(n)||f.set(n,{runId:r,parentCodecMessageId:t[x]})}for(let e of d){if(e.name!==`ai-run-start`)continue;let t=N(e),n=t[m];if(n===void 0)continue;let r=p.get(n);if(r===void 0)continue;let i=f.get(r);i&&i.parentCodecMessageId===void 0&&(i.parentCodecMessageId=t[x])}let h=[],g=0;if(c!==void 0){let e=zt(f,c);g=e.length;for(let t of e){let e=f.get(t);if(e?.runId===i)continue;let n=e?.runId===void 0?Gt(r,d,t):Wt(r,d,e.runId).projection;h.push(...r.getMessages(n).map(e=>e.message))}}let{projection:v,folded:y}=Wt(r,d,i);return h.push(...r.getMessages(v).map(e=>e.message)),o?.debug(`loadConversation(); built`,{runId:i,chainLength:g,totalMessages:h.length,folded:y}),{messages:h,projection:v}},Jt=e=>{let t;return{promise:e===void 0?new Promise(()=>{}):e.aborted?Promise.resolve():new Promise(n=>{t=()=>{n()},e.addEventListener(`abort`,t,{once:!0})}),cleanup:()=>{t&&e&&e.removeEventListener(`abort`,t)}}},Yt=async(e,t,n,r,i,a)=>{a?.trace(`pipeStream();`);let o=e.getReader(),s=Jt(n),c=`complete`,l;try{for(;;){let e=await Promise.race([o.read(),s.promise.then(()=>`cancelled`)]);if(e===`cancelled`){c=`cancelled`,a?.debug(`pipeStream(); stream cancelled by AbortSignal`),r&&await r(async e=>t.publishOutput(e)),await t.cancel(`cancelled`);break}let{done:n,value:l}=e;if(n){await t.close(),a?.debug(`pipeStream(); stream completed`);break}await t.publishOutput(l,i?.(l))}}catch(e){c=`error`,l=e instanceof Error?e:Error(String(e)),a?.error(`pipeStream(); stream error`,{error:l.message});try{await t.close()}catch{}}finally{s.cleanup(),o.releaseLock()}return{reason:c,error:l}},Xt=class{constructor(e,t){this._activeRuns=new Map,this._channel=e,this._logger=t?.withContext({component:`RunManager`})}async startRun(e,t,n,r){this._logger?.trace(`DefaultRunManager.startRun();`,{runId:e,clientId:t});let i=n??new AbortController,a=t??``;this._activeRuns.set(e,{controller:i,clientId:a});let o=r?.continuation===!0,s=Ft({runId:e,runClientId:a,parent:o?void 0:r?.parent,forkOf:o?void 0:r?.forkOf,regenerates:o?void 0:r?.regenerates,invocationId:r?.invocationId,inputClientId:r?.inputClientId,inputCodecMessageId:r?.inputCodecMessageId});return await this._channel.publish({name:o?te:O,extras:{ai:{transport:s}}}),this._logger?.debug(`DefaultRunManager.startRun(); run started`,{runId:e}),i.signal}async suspendRun(e,t,n,r){this._logger?.trace(`DefaultRunManager.suspendRun();`,{runId:e}),await this._publishTerminal(ee,e,{invocationId:t,inputClientId:n,inputCodecMessageId:r}),this._logger?.debug(`DefaultRunManager.suspendRun(); run suspended`,{runId:e})}async endRun(e,t,n,r,i){this._logger?.trace(`DefaultRunManager.endRun();`,{runId:e,reason:t}),await this._publishTerminal(k,e,{reason:t,invocationId:n,inputClientId:r,inputCodecMessageId:i}),this._logger?.debug(`DefaultRunManager.endRun(); run ended`,{runId:e,reason:t})}async _publishTerminal(e,t,n){let r=Ft({runId:t,runClientId:this._activeRuns.get(t)?.clientId??``,...n});await this._channel.publish({name:e,extras:{ai:{transport:r}}}),this._activeRuns.delete(t)}getSignal(e){return this._activeRuns.get(e)?.controller.signal}getClientId(e){return this._activeRuns.get(e)?.clientId}cancel(e){this._logger?.debug(`DefaultRunManager.cancel();`,{runId:e}),this._activeRuns.get(e)?.controller.abort()}getActiveRunIds(){return[...this._activeRuns.keys()]}close(){this._logger?.trace(`DefaultRunManager.close();`,{activeRuns:this._activeRuns.size});for(let e of this._activeRuns.values())e.controller.abort();this._activeRuns.clear()}},Zt=(e,t)=>new Xt(e,t),Qt=async e=>{let{register:n,codec:r,invocationId:i,runId:a,expectedInputEventIds:o,timeoutMs:s,signal:c,logger:l}=e,u=new Set(o),d=u.size,f=[],p=[],m=new Set,h,_,v=e=>{let t=r.createDecoder(),n=N(e),i=n[`codec-message-id`]??``,{inputs:a,outputs:o}=t.decode(e),s=[...a,...o],c=r.init();for(let t of s)c=r.fold(c,t,{serial:e.serial??``,messageId:i});return r.getMessages(c).map(({message:t})=>({kind:`message`,message:t,codecMessageId:i,parentId:n[x],forkOf:n[S],headers:n,serial:e.serial}))};return new Promise((e,r)=>{let o=!1,y=new Set,b=()=>{},x,S=()=>{b(),x!==void 0&&clearTimeout(x),c.removeEventListener(`abort`,C)},C=()=>{o||(o=!0,S(),r(new t.ErrorInfo(`unable to look up input event; run ${a} was cancelled`,G.InvalidArgument,400)))};if(c.addEventListener(`abort`,C,{once:!0}),!o){if(b=n(n=>{if(o||n.serial!==void 0&&y.has(n.serial))return;n.serial!==void 0&&y.add(n.serial);let s=N(n),c=s[g];if(!c||!u.has(c)||m.has(c))return;m.add(c),h===void 0&&(h=s,_=n.clientId);let b;try{b=v(n)}catch(e){o=!0,S();let n=e instanceof t.ErrorInfo?e:void 0;r(new t.ErrorInfo(`unable to look up input event; decode failed for invocation ${i}: ${e instanceof Error?e.message:String(e)}`,G.InputEventNotFound,504,n));return}for(let e of b)f.push(e);p.push(n),!(m.size<d)&&(o=!0,S(),f.sort(R),l?.debug(`lookupInputEvents(); collected input events`,{runId:a,invocationId:i,count:f.length}),e({nodes:f,firstHeaders:h,firstClientId:_,rawMessages:p}))}),o){b();return}x=setTimeout(()=>{o||(o=!0,S(),r(new t.ErrorInfo(`unable to look up input event; received ${String(f.length)} of ${String(d)} input events for invocation ${i} within ${String(s)}ms`,G.InputEventNotFound,504)))},s)}})},$t=class{constructor(e){this._registeredRuns=new Map,this._runIdByInputCodecMessageId=new Map,this._deferredCancels=new Map,this._pendingInputEventLookups=new Map,this._inputEventBuffer=new Map,this._state=`ready`,this._codec=e.codec;let t={params:{...Nt(e.client,e.codec).params,rewind:e.rewindWindow??`2m`}};this._channel=e.client.channels.get(e.channelName,t),this._logger=e.logger?.withContext({component:`AgentSession`}),this._onError=e.onError,this._runManager=Zt(this._channel,this._logger),this._inputEventLookupTimeoutMs=e.inputEventLookupTimeoutMs??3e4,this._inputEventBufferLimit=e.inputEventBufferLimit??200,this._channelListener=e=>{this._handleChannelMessage(e)},this._hasAttachedOnce=this._channel.state===`attached`,this._onChannelStateChange=e=>{this._handleChannelStateChange(e)},this._channel.on(this._onChannelStateChange),this._logger?.debug(`DefaultAgentSession(); session created`)}connect(){return this._state===`closed`?Promise.reject(new t.ErrorInfo(`unable to connect; session is closed`,G.SessionClosed,400)):this._connectPromise?this._connectPromise:(this._logger?.trace(`DefaultAgentSession.connect();`),this._connectPromise=this._channel.subscribe(this._channelListener).then(()=>{this._logger?.debug(`DefaultAgentSession.connect(); subscribed and attached`)},e=>{let n=new t.ErrorInfo(`unable to subscribe to channel; ${e instanceof Error?e.message:String(e)}`,G.SessionSubscriptionError,500,e instanceof t.ErrorInfo?e:void 0);throw this._logger?.error(`DefaultAgentSession.connect(); subscribe failed`),this._onError?.(n),n}),this._connectPromise)}_registerInputEventListener(e,t){for(let n of e)this._pendingInputEventLookups.set(n,t);for(let n of e){let e=this._inputEventBuffer.get(n);if(e){this._inputEventBuffer.delete(n);for(let n of e)t(n)}}return()=>{for(let n of e)this._pendingInputEventLookups.get(n)===t&&this._pendingInputEventLookups.delete(n)}}createRun(e,t){return this._logger?.trace(`DefaultAgentSession.createRun();`,{inputEventId:e.inputEventId}),this._createRun(e,t??{})}async close(){if(this._state!==`closed`){this._state=`closed`,this._logger?.trace(`DefaultAgentSession.close();`),this._connectPromise&&this._channel.unsubscribe(this._channelListener),this._channel.off(this._onChannelStateChange);for(let e of this._registeredRuns.values())e.controller.abort();if(this._registeredRuns.clear(),this._runIdByInputCodecMessageId.clear(),this._deferredCancels.clear(),this._pendingInputEventLookups.clear(),this._inputEventBuffer.clear(),this._runManager.close(),this._connectPromise)try{await this._channel.detach()}catch(e){this._logger?.debug(`DefaultAgentSession.close(); channel detach failed`,{error:e})}this._logger?.debug(`DefaultAgentSession.close(); session closed`)}}async _handleCancelMessage(e){let t=N(e),n=t[m],r=t[T];if(!n&&!r){this._logger?.warn(`DefaultAgentSession._handleCancelMessage(); missing run-id and input-codec-message-id`,{serial:e.serial});return}let i=n??(r?this._runIdByInputCodecMessageId.get(r):void 0),a=i?this._registeredRuns.get(i):void 0;if(!a){r!==void 0&&this._bufferDeferredCancel(r,e);return}await this._cancelRegistration(a,e)}_bufferDeferredCancel(e,t){let n=Rt(this._deferredCancels,e,this._inputEventBufferLimit);n!==void 0&&this._logger?.warn(`DefaultAgentSession._bufferDeferredCancel(); deferred-cancel buffer full, dropping oldest`,{evictedInputCodecMessageId:n,limit:this._inputEventBufferLimit}),this._deferredCancels.set(e,t),this._logger?.debug(`DefaultAgentSession._bufferDeferredCancel(); buffered early cancel`,{inputCodecMessageId:e,serial:t.serial})}async _pullDeferredCancel(e,t){let n=this._deferredCancels.get(t);n!==void 0&&(this._deferredCancels.delete(t),this._logger?.debug(`DefaultAgentSession._pullDeferredCancel(); honouring buffered cancel`,{runId:e.runId,inputCodecMessageId:t}),await this._cancelRegistration(e,n))}async _cancelRegistration(e,n){let{runId:r}=e;this._logger?.debug(`DefaultAgentSession._cancelRegistration(); matched run`,{runId:r});let i={message:n,runId:r};try{if(e.onCancel&&!await e.onCancel(i)){this._logger?.debug(`DefaultAgentSession._cancelRegistration(); cancel rejected by onCancel`,{runId:r});return}e.controller.abort(),this._logger?.debug(`DefaultAgentSession._cancelRegistration(); run cancelled`,{runId:r})}catch(n){let i=new t.ErrorInfo(`unable to process cancel for run ${r}; onCancel handler threw: ${n instanceof Error?n.message:String(n)}`,G.CancelListenerError,500,n instanceof t.ErrorInfo?n:void 0);this._logger?.error(`DefaultAgentSession._cancelRegistration(); onCancel threw`,{runId:r}),(e.onError??this._onError)?.(i)}}_handleChannelStateChange(e){if(this._state===`closed`)return;let{current:n,resumed:r}=e;if(n===`attached`&&!this._hasAttachedOnce){this._hasAttachedOnce=!0;return}if(!(n===`failed`||n===`suspended`||n===`detached`||n===`attached`&&!r))return;this._logger?.error(`DefaultAgentSession._handleChannelStateChange(); channel continuity lost`,{current:n,resumed:r,previous:e.previous});let i=new t.ErrorInfo(`unable to deliver cancel messages; channel continuity lost (${n}${n===`attached`?`, resumed: false`:``})`,G.ChannelContinuityLost,500,e.reason);this._onError?.(i)}_handleChannelMessage(e){try{if(e.name===`ai-cancel`){this._handleCancelMessage(e).catch(e=>{let n=new t.ErrorInfo(`unable to route cancel message; ${e instanceof Error?e.message:String(e)}`,G.CancelListenerError,500,e instanceof t.ErrorInfo?e:void 0);this._logger?.error(`DefaultAgentSession._handleChannelMessage(); cancel routing error`),this._onError?.(n)});return}let n=N(e)[g];if(n!==void 0){let t=this._pendingInputEventLookups.get(n);if(t)t(e);else{let t=this._inputEventBuffer.get(n);if(t)t.push(e);else{let t=Rt(this._inputEventBuffer,n,this._inputEventBufferLimit);t!==void 0&&this._logger?.warn(`DefaultAgentSession._handleChannelMessage(); input-event buffer full, dropping oldest entry`,{evictedEventId:t,limit:this._inputEventBufferLimit}),this._inputEventBuffer.set(n,[e])}}}}catch(e){let n=new t.ErrorInfo(`unable to process channel message; ${e instanceof Error?e.message:String(e)}`,G.SessionSubscriptionError,500,e instanceof t.ErrorInfo?e:void 0);this._logger?.error(`DefaultAgentSession._handleChannelMessage(); subscription error`),this._onError?.(n)}}async _requireConnected(e){if(!this._connectPromise)throw new t.ErrorInfo(`unable to ${e}; connect() must be called before ${e}()`,G.InvalidArgument,400);return this._connectPromise}_createRun(e,n){let r=n.runId??crypto.randomUUID(),i=n.invocationId??crypto.randomUUID(),a=this._inputEventLookupTimeoutMs,{onMessage:o,onCancelled:s,onCancel:c,onError:l,signal:u}=n,d=new AbortController,f=`initialized`,p=u?AbortSignal.any([d.signal,u]):d.signal,h={runId:r,invocationId:i,controller:d,signal:p,onCancel:c,onError:l};this._registeredRuns.set(r,h);let g=this._logger,y=this._runManager,b=this._codec,w=this._channel,T=this._registeredRuns,E=this._runIdByInputCodecMessageId,D=this._deferredCancels,O=this._requireConnected.bind(this),ee=this._registerInputEventListener.bind(this),te=this._pullDeferredCancel.bind(this),k=e.inputEventId,A=[],j={get messages(){return A}},M,N,P,F,I,L,R=!1,z,B,V,H=()=>{T.delete(r),L!==void 0&&(E.delete(L),D.delete(L))},U,ne;return{get runId(){return r},get invocationId(){return i},get abortSignal(){return p},get view(){return j},get messages(){return ne===void 0?U===void 0?A.map(e=>e.message):b.getMessages(U).map(e=>e.message):[...ne]},start:async()=>{if(g?.trace(`Run.start();`,{runId:r,inputEventId:k}),await O(`start`),p.aborted)throw new t.ErrorInfo(`unable to start run; run ${r} was cancelled before start()`,G.InvalidArgument,400);if(f!==`initialized`)return;if(f=`started`,k&&a>0)try{let e=await Qt({register:e=>ee([k],e),codec:b,invocationId:i,runId:r,expectedInputEventIds:[k],timeoutMs:a,signal:p,logger:g});for(let t of e.nodes)A.push(t);e.firstHeaders!==void 0&&(z=e.firstHeaders),e.firstClientId!==void 0&&(N=e.firstClientId),V=e.rawMessages}catch(e){let n=e instanceof t.ErrorInfo?e:new t.ErrorInfo(`unable to look up input event; ${e instanceof Error?e.message:String(e)}`,G.InputEventNotFound,504);throw H(),g?.error(`Run.start(); input-event lookup failed`,{runId:r,invocationId:i}),n}let e=z??A[0]?.headers;if(e){M=e[v],P=e[x],F=e[S],I=e[C],L=e[_];let t=e[m];R=t!==void 0,t!==void 0&&t!==r&&(T.delete(r),r=t,h.runId=r,T.set(r,h))}B=A.at(-1)?.codecMessageId??P,L!==void 0&&(E.set(L,r),await te(h,L));try{await y.startRun(r,M,d,{parent:B,forkOf:F,regenerates:I,invocationId:i,inputClientId:N,inputCodecMessageId:L,continuation:R})}catch(e){let n=new t.ErrorInfo(`unable to publish run-start for run ${r}; ${e instanceof Error?e.message:String(e)}`,G.RunLifecycleError,500,e instanceof t.ErrorInfo?e:void 0);throw g?.error(`Run.start(); failed to publish run-start`,{runId:r}),n}g?.debug(`Run.start(); run started`,{runId:r,inputEventId:k})},addEvents:async e=>{if(g?.trace(`Run.addEvents();`,{runId:r,count:e.length}),await O(`addEvents`),f===`initialized`)throw new t.ErrorInfo(`unable to add events; start() must be called before addEvents() (run ${r})`,G.InvalidArgument,400);let n=y.getClientId(r);try{for(let t of e){let e=Pt({role:`assistant`,runId:r,codecMessageId:t.codecMessageId,runClientId:n,invocationId:i,inputClientId:N,inputCodecMessageId:L}),a=b.createEncoder(w,{extras:{headers:e},onMessage:o});for(let e of t.events)await a.publishOutput(e);await a.close()}}catch(e){let n=new t.ErrorInfo(`unable to publish events for run ${r}; ${e instanceof Error?e.message:String(e)}`,G.RunLifecycleError,500,e instanceof t.ErrorInfo?e:void 0);throw g?.error(`Run.addEvents(); publish failed`,{runId:r}),n}g?.debug(`Run.addEvents(); events published`,{runId:r,count:e.length})},loadProjection:async()=>{g?.trace(`Run.loadProjection();`,{runId:r}),await O(`loadProjection`);let e=await Kt({channel:w,codec:b,runId:r,signal:p,logger:g,liveMessages:V});return U=e,e},loadConversation:async e=>{g?.trace(`Run.loadConversation();`,{runId:r}),await O(`loadConversation`);let{messages:t,projection:n}=await qt({channel:w,codec:b,runId:r,signal:p,logger:g,liveMessages:V,assistantParentFallback:B,pageLimit:e?.pageLimit??200,maxMessages:e?.maxMessages??2e3});return U=n,ne=t,t},pipe:async(e,n)=>{if(g?.trace(`Run.pipe();`,{runId:r}),await O(`pipe`),f===`initialized`)throw new t.ErrorInfo(`unable to pipe stream; start() must be called before pipe() (run ${r})`,G.InvalidArgument,400);let a=y.getClientId(r),c=n?.parent??B,u=n?.forkOf??F,d=I,m=crypto.randomUUID(),h=Pt({role:`assistant`,runId:r,codecMessageId:m,runClientId:a,parent:c,forkOf:u,invocationId:i,inputClientId:N,inputCodecMessageId:L,regenerates:d}),_=await Yt(e,b.createEncoder(w,{extras:{headers:h},onMessage:o,messageId:m}),p,s,n?.resolveWriteOptions,g);if(_.error){let e=new t.ErrorInfo(`unable to pipe response for run ${r}; ${_.error.message}`,G.StreamError,500,_.error instanceof t.ErrorInfo?_.error:void 0);g?.error(`Run.pipe(); stream error`,{runId:r}),l?.(e)}return g?.debug(`Run.pipe(); stream finished`,{runId:r,reason:_.reason}),_},suspend:async()=>{if(g?.trace(`Run.suspend();`,{runId:r}),await O(`suspend`),f===`initialized`)throw new t.ErrorInfo(`unable to suspend run; start() must be called before suspend() (run ${r})`,G.InvalidArgument,400);if(f!==`ended`){f=`ended`;try{await y.suspendRun(r,i,N,L)}catch(e){let n=new t.ErrorInfo(`unable to publish run-suspend for run ${r}; ${e instanceof Error?e.message:String(e)}`,G.RunLifecycleError,500,e instanceof t.ErrorInfo?e:void 0);throw g?.error(`Run.suspend(); failed to publish run-suspend`,{runId:r}),n}finally{H()}g?.debug(`Run.suspend(); run suspended`,{runId:r})}},end:async e=>{if(g?.trace(`Run.end();`,{runId:r,reason:e}),await O(`end`),f===`initialized`)throw new t.ErrorInfo(`unable to end run; start() must be called before end() (run ${r})`,G.InvalidArgument,400);if(f!==`ended`){f=`ended`;try{await y.endRun(r,e,i,N,L)}catch(e){let n=new t.ErrorInfo(`unable to publish run-end for run ${r}; ${e instanceof Error?e.message:String(e)}`,G.RunLifecycleError,500,e instanceof t.ErrorInfo?e:void 0);throw g?.error(`Run.end(); failed to publish run-end`,{runId:r}),n}finally{H()}g?.debug(`Run.end(); run ended`,{runId:r,reason:e})}}}}},en=e=>new $t(e),tn=class e{constructor(e){this.inputEventId=e.inputEventId,this.sessionName=e.sessionName}static fromJSON(t){return new e(t)}toJSON(){return{inputEventId:this.inputEventId,sessionName:this.sessionName}}},$=e=>e.kind===`run`?e.runId:e.codecMessageId,nn=e=>e.kind===`run`?e.startSerial:e.serial,rn=(e,t,n)=>{let r=e.get(t);r||(r=new Set,e.set(t,r)),r.add(n)},an=(e,t,n)=>{let r=e.get(t);r&&(r.delete(n),r.size===0&&e.delete(t))},on=class{constructor(e,t){this._nodeIndex=new Map,this._codecMessageIdToNodeKey=new Map,this._sortedNodes=[],this._parentIndex=new Map,this._replyRunsByInput=new Map,this._seqCounter=0,this._structuralVersion=0,this._siblingCache=new Map,this._siblingCacheVersion=-1,this._codec=e,this._logger=t,this._emitter=new ht(t)}_compareNodes(e,t){let n=nn(e.node),r=nn(t.node);return n===void 0&&r===void 0?e.insertSeq-t.insertSeq:n===void 0?1:r===void 0||n<r?-1:n>r?1:e.insertSeq-t.insertSeq}_insertSortedNode(e){if(nn(e.node)===void 0){this._sortedNodes.push(e);return}let t=0,n=this._sortedNodes.length;for(;t<n;){let r=t+n>>>1,i=this._sortedNodes[r];if(!i)break;this._compareNodes(i,e)<=0?t=r+1:n=r}this._sortedNodes.splice(t,0,e)}_removeSortedNode(e){let t=this._sortedNodes.indexOf(e);t!==-1&&this._sortedNodes.splice(t,1)}_insertNode(e,t,n){this._nodeIndex.set(e,t),this._addToParentIndex(n,e),this._insertSortedNode(t),this._structuralVersion++}_promoteSerial(e){this._removeSortedNode(e),this._insertSortedNode(e),this._structuralVersion++}_foldInto(e,t,n,r){for(let i of t)try{e.node.projection=this._codec.fold(e.node.projection,i,{serial:n??``,messageId:r})}catch(t){this._logger.error(`Tree._foldInto(); fold threw`,{key:$(e.node),messageId:r,err:t})}}_addToParentIndex(e,t){rn(this._parentIndex,e,t)}_removeFromParentIndex(e,t){an(this._parentIndex,e,t)}_parentKeyOf(e){let t=e.parentCodecMessageId;return t===void 0?void 0:this._codecMessageIdToNodeKey.get(t)}_inputGroupRoot(e){let t=e,n=new Set([$(t)]);for(;t.forkOf!==void 0&&!n.has(t.forkOf);){let e=this._nodeIndex.get(t.forkOf);if(e?.node.kind!==`input`||e.node.parentCodecMessageId!==t.parentCodecMessageId)break;t=e.node,n.add($(t))}return t}_getSiblingGroup(e){this._siblingCacheVersion!==this._structuralVersion&&(this._siblingCache.clear(),this._siblingCacheVersion=this._structuralVersion);let t=this._siblingCache.get(e);if(t)return t;let n=this._nodeIndex.get(e);if(!n)return[];let r=n.node;r.kind===`input`&&(r=this._inputGroupRoot(r));let i=r.parentCodecMessageId,a=[],o=this._parentIndex.get(i);if(o)for(let e of o){let t=this._nodeIndex.get(e);t&&this._isSiblingOf(t.node,r)&&a.push(t)}a.sort((e,t)=>this._compareNodes(e,t));for(let e of a)this._siblingCache.set($(e.node),a);return this._siblingCache.set(e,a),a}_isSiblingOf(e,t){if(e.kind!==t.kind||e.parentCodecMessageId!==t.parentCodecMessageId)return!1;if(e.kind===`run`)return!0;let n=$(t);if($(e)===n)return!0;let r=e,i=new Set([$(r)]);for(;r.kind===`input`&&r.forkOf!==void 0;){if(r.forkOf===n)return!0;if(i.has(r.forkOf))break;let e=this._nodeIndex.get(r.forkOf);if(!e)break;r=e.node,i.add($(r))}return!1}getGroupRoot(e){let t=this._nodeIndex.get(e);if(!t)return e;if(t.node.kind===`input`)return $(this._inputGroupRoot(t.node));let n=this._getSiblingGroup(e)[0]?.node;return n?$(n):e}visibleNodes(e=new Map){this._logger.trace(`DefaultTree.visibleNodes();`);let t=[],n=new Set,r=new Map;for(let i of this._sortedNodes){let a=i.node,o=$(a),s=this._parentKeyOf(a);if(s!==void 0&&!n.has(s))continue;let c=this._getSiblingGroup(o);if(c.length>1){let t=this.getGroupRoot(o),n=r.get(t);if(n===void 0){let i=e.get(t);if(i!==void 0&&c.some(e=>$(e.node)===i))n=i;else{let e=c.at(-1);if(!e)break;n=$(e.node)}r.set(t,n)}if(o!==n)continue}n.add(o),t.push(a)}return t}getRunNode(e){this._logger.trace(`DefaultTree.getRunNode();`,{runId:e});let t=this._nodeIndex.get(e)?.node;return t?.kind===`run`?t:void 0}getNode(e){return this._logger.trace(`DefaultTree.getNode();`,{key:e}),this._nodeIndex.get(e)?.node}getNodeByCodecMessageId(e){this._logger.trace(`DefaultTree.getNodeByCodecMessageId();`,{codecMessageId:e});let t=this._codecMessageIdToNodeKey.get(e);return t===void 0?void 0:this._nodeIndex.get(t)?.node}getReplyRuns(e){let t=this._replyRunsByInput.get(e);if(!t)return[];let n=[];for(let e of t){let t=this._nodeIndex.get(e)?.node;t?.kind===`run`&&n.push(t)}return n}getSiblingNodes(e){return this._logger.trace(`DefaultTree.getSiblingNodes();`,{key:e}),this._getSiblingGroup(e).map(e=>e.node)}applyMessage(e,t,n){let r=t[m],i=t[_],a=r===void 0&&i!==void 0&&t.role===`user`&&e.inputs.length>0?i:void 0;if(r===void 0&&a===void 0){this._logger.warn(`Tree.applyMessage(); message has no run-id and is not a user input; skipping`);return}let o=[...e.inputs,...e.outputs],s=a??r;if(o.length===0&&s!==void 0&&!this._nodeIndex.has(s))return;let c=this._structuralVersion;a===void 0?r!==void 0&&this._applyRunMessage(r,e,t,n):this._applyInputMessage(a,t,n,o),this._structuralVersion!==c&&this._emitter.emit(`update`)}_applyInputMessage(e,t,n,r){let i=this._nodeIndex.get(e);i?i.node.kind===`input`&&n&&!i.node.serial&&(this._logger.debug(`Tree.applyMessage(); promoting input serial`,{codecMessageId:e,serial:n}),i.node.serial=n,this._promoteSerial(i)):(i=this._createInputNodeFromHeaders(e,t,n),this._insertNode(e,i,i.node.parentCodecMessageId),this._codecMessageIdToNodeKey.set(e,e),this._logger.debug(`Tree.applyMessage(); created input node`,{codecMessageId:e})),this._foldInto(i,r,n,e),this._emitter.emit(`output`,{runId:void 0,inputCodecMessageId:e,codecMessageId:e,serial:n,events:[]})}_applyRunMessage(e,t,n,r){let i=n[_],a=n[T],o=[...t.inputs,...t.outputs],s=t.outputs,c=this._nodeIndex.get(e);if(!c&&i!==void 0){let e=this._codecMessageIdToNodeKey.get(i),t=e===void 0?void 0:this._nodeIndex.get(e);t?.node.kind===`run`&&t.node.startSerial===void 0&&(c=t)}c?r&&c.node.kind===`run`&&!c.node.startSerial&&(this._logger.debug(`Tree.applyMessage(); promoting startSerial`,{runId:e,serial:r}),c.node.startSerial=r,this._promoteSerial(c)):(c=this._createRunFromHeaders(e,n,r),this._insertNode(e,c,c.node.parentCodecMessageId),this._indexReplyRun(c.node,e),this._logger.debug(`Tree.applyMessage(); created new Run`,{runId:e}));let l=$(c.node);i&&this._codecMessageIdToNodeKey.set(i,l),this._foldInto(c,o,r,i),this._emitter.emit(`output`,{runId:l,inputCodecMessageId:a,codecMessageId:i,serial:r,events:s})}_indexReplyRun(e,t){e.parentCodecMessageId!==void 0&&rn(this._replyRunsByInput,e.parentCodecMessageId,t)}applyRunLifecycle(e){this._logger.trace(`DefaultTree.applyRunLifecycle();`,{type:e.type,runId:e.runId});let t=this._structuralVersion;switch(e.type){case`start`:this._applyRunStart(e);break;case`suspend`:this._applyRunSuspend(e);break;case`resume`:this._applyRunResume(e);break;case`end`:this._applyRunEnd(e);break}this._emitter.emit(`run`,e),this._structuralVersion!==t&&this._emitter.emit(`update`)}_applyRunStart(e){let t=this._nodeIndex.get(e.runId);if(t?.node.kind===`run`){let n=t.node;if(n.status!==`active`&&(n.status=`active`),e.serial&&!n.startSerial&&(n.startSerial=e.serial,this._promoteSerial(t)),n.parentCodecMessageId===void 0&&e.parent!==void 0&&(n.parentCodecMessageId=e.parent,this._removeFromParentIndex(void 0,e.runId),this._addToParentIndex(n.parentCodecMessageId,e.runId),this._indexReplyRun(n,e.runId),this._structuralVersion++),n.forkOf===void 0&&e.forkOf!==void 0){let t=this._codecMessageIdToNodeKey.get(e.forkOf);t!==void 0&&t!==e.runId&&(n.forkOf=t,this._structuralVersion++)}n.regeneratesCodecMessageId===void 0&&e.regenerates!==void 0&&(n.regeneratesCodecMessageId=e.regenerates,this._structuralVersion++),n.invocationId===``&&e.invocationId!==``&&(n.invocationId=e.invocationId)}else if(!t){let t=this._createRunFromLifecycle(e);this._insertNode(e.runId,t,t.node.parentCodecMessageId),this._indexReplyRun(t.node,e.runId)}}_applyRunSuspend(e){let t=this._nodeIndex.get(e.runId);t?.node.kind===`run`&&(t.node.status=`suspended`,t.node.endSerial=e.serial)}_applyRunResume(e){let t=this._nodeIndex.get(e.runId);t?.node.kind===`run`&&t.node.status===`suspended`&&(t.node.status=`active`)}_applyRunEnd(e){let t=this._nodeIndex.get(e.runId);t?.node.kind===`run`&&(t.node.status=e.reason,t.node.endSerial=e.serial)}delete(e){let t=this._nodeIndex.get(e);t&&(this._logger.debug(`Tree.delete();`,{key:e}),this._removeFromParentIndex(t.node.parentCodecMessageId,e),this._removeSortedNode(t),this._nodeIndex.delete(e),t.node.kind===`run`&&t.node.parentCodecMessageId!==void 0&&an(this._replyRunsByInput,t.node.parentCodecMessageId,e),this._structuralVersion++,this._emitter.emit(`update`))}_createRunFromHeaders(e,t,n){let r=t[S];return this._buildRunNode({runId:e,parentCodecMessageId:t[x],forkOf:r?this._codecMessageIdToNodeKey.get(r):void 0,regeneratesCodecMessageId:t[C],clientId:t[`run-client-id`]??``,invocationId:t[`invocation-id`]??``,startSerial:n})}_buildRunNode(e){return{node:{kind:`run`,runId:e.runId,parentCodecMessageId:e.parentCodecMessageId,forkOf:e.forkOf,regeneratesCodecMessageId:e.regeneratesCodecMessageId,clientId:e.clientId,invocationId:e.invocationId,status:`active`,projection:this._codec.init(),startSerial:e.startSerial,endSerial:void 0},insertSeq:this._seqCounter++}}_createInputNodeFromHeaders(e,t,n){let r=t[S];return{node:{kind:`input`,codecMessageId:e,parentCodecMessageId:t[x],forkOf:r,projection:this._codec.init(),serial:n},insertSeq:this._seqCounter++}}_createRunFromLifecycle(e){let t=e.forkOf;return this._buildRunNode({runId:e.runId,parentCodecMessageId:e.parent,forkOf:t?this._codecMessageIdToNodeKey.get(t):void 0,regeneratesCodecMessageId:e.regenerates,clientId:e.clientId,invocationId:e.invocationId,startSerial:e.serial})}on(e,t){let n=t;return this._emitter.on(e,n),()=>{this._emitter.off(e,n)}}emitAblyMessage(e){this._logger.trace(`DefaultTree.emitAblyMessage();`),this._emitter.emit(`ably-message`,e)}},sn=(e,t)=>new on(e,t),cn=(e,t)=>{for(let n of t){let t=N(n),r=t[_];if(!r)continue;let i=n.action,a=i===`message.create`&&`discrete`in t,o=t.stream===`true`&&(i===`message.create`||i===`message.update`||i===`message.append`),s=t[d],c=s===`complete`||s===`cancelled`;(a||o)&&e.startedCodecMessageIds.add(r),(a||c)&&e.terminatedCodecMessageIds.add(r),e.startedCodecMessageIds.has(r)&&e.terminatedCodecMessageIds.has(r)&&e.completedCodecMessageIds.add(r)}},ln=async(e,t,n)=>{e.rawMessages.push(...t.items),e.lastAblyPage=t,cn(e,t.items);let r=e.returnedCount+n;for(;e.completedCodecMessageIds.size<r&&t.hasNext();){e.logger.debug(`loadHistory.fetchUntilLimit(); fetching next page`,{collected:e.rawMessages.length,completed:e.completedCodecMessageIds.size});let n=await t.next();if(!n)break;t=n,e.rawMessages.push(...n.items),e.lastAblyPage=n,cn(e,n.items)}},un=(e,t)=>{let n=e.completedCodecMessageIds.size,r=Math.min(t,Math.max(0,n-e.returnedCount));e.returnedCount+=r;let i=n>e.returnedCount,a=e.lastAblyPage?.hasNext()??!1,o=e.rawMessages.length-e.returnedRawCount>0?e.rawMessages.slice(e.returnedRawCount).toReversed():[];return e.returnedRawCount=e.rawMessages.length,{rawMessages:o,hasNext:()=>i||a,next:async()=>{if(i)return un(e,t);if(!a||!e.lastAblyPage)return;let n=await e.lastAblyPage.next();if(n)return await ln(e,n,t),un(e,t)}}},dn=async(e,t,n)=>{let r=t?.limit??100,i={rawMessages:[],returnedCount:0,returnedRawCount:0,lastAblyPage:void 0,startedCodecMessageIds:new Set,terminatedCodecMessageIds:new Set,completedCodecMessageIds:new Set,logger:n};n.trace(`loadHistory();`,{limit:r});let a=r*10;return await e.attach(),await ln(i,await e.history({untilAttach:!0,limit:a}),r),un(i,r)},fn=e=>Array.isArray(e)?e:[e],pn=3,mn=e=>({runId:e.runId,clientId:e.clientId,status:e.status,invocationId:e.invocationId}),hn=class{constructor(e){this._branchSelections=new Map,this._regenSelections=new Map,this._withheldRunIds=new Set,this._lastVisibleNodeKeys=[],this._lastVisibleProjections=[],this._lastVisibleMessagePairs=[],this._lastVisibleNodeKeySet=new Set,this._hasMoreHistory=!1,this._withheldBuffer=[],this._unsubs=[],this._cachedNodes=[],this._loadingOlder=!1,this._processingHistory=!1,this._closed=!1,this._tree=e.tree,this._channel=e.channel,this._codec=e.codec,this._sendDelegate=e.sendDelegate,this._onClose=e.onClose,this._logger=e.logger.withContext({component:`View`}),this._logger.trace(`DefaultView();`),this._emitter=new ht(this._logger),this._cachedNodes=this._computeFlatNodes(),this._updateVisibleSnapshot(this._cachedNodes),this._unsubs.push(this._tree.on(`update`,()=>{this._onTreeUpdate()}),this._tree.on(`ably-message`,e=>{this._onTreeAblyMessage(e)}),this._tree.on(`run`,e=>{this._onTreeRun(e)}),this._tree.on(`output`,e=>{this._onTreeOutput(e)}))}_onTreeOutput(e){this._processingHistory||(e.runId!==void 0&&this._lastVisibleNodeKeySet.has(e.runId)||e.inputCodecMessageId!==void 0&&this._lastVisibleNodeKeySet.has(e.inputCodecMessageId))&&(this._lastVisibleProjections=this._cachedNodes.map(e=>e.projection),this._lastVisibleMessagePairs=this._extractMessages(this._cachedNodes),this._emitter.emit(`update`))}getMessages(){return this._lastVisibleMessagePairs}runs(){return this._cachedNodes.filter(e=>e.kind===`run`).map(e=>mn(e))}_computeFlatNodes(){let e=this._treeVisibleNodes();return this._withheldRunIds.size===0?e:e.filter(e=>!this._withheldRunIds.has($(e)))}_recomputeAndEmit(){this._cachedNodes=this._computeFlatNodes(),this._updateVisibleSnapshot(this._cachedNodes),this._emitter.emit(`update`)}_recomputeAndEmitIfChanged(){let e=this._computeFlatNodes();this._visibleChanged(e)&&(this._cachedNodes=e,this._updateVisibleSnapshot(e),this._emitter.emit(`update`))}_runByCodecMessageId(e){let t=this._tree.getNodeByCodecMessageId(e);return t?.kind===`run`?t:void 0}_extractMessages(e){let t=[];for(let n of e)for(let e of this._codec.getMessages(n.projection))t.push(e);return t}hasOlder(){return this._withheldBuffer.length>0||this._hasMoreHistory}async loadOlder(e=100){if(!(this._closed||this._loadingOlder)){this._loadingOlder=!0,this._logger.trace(`DefaultView.loadOlder();`,{limit:e});try{if(this._withheldBuffer.length>0){let t=this._withheldBuffer.splice(-e,e);this._releaseWithheld(t);return}if(!this._hasMoreHistory&&!this._lastHistoryPage){await this._loadFirstPage(e);return}if(!this._hasMoreHistory)return;if(!this._lastHistoryPage?.hasNext()){this._hasMoreHistory=!1;return}let t=await this._lastHistoryPage.next();if(this._closed||!t){t||(this._hasMoreHistory=!1);return}await this._revealFromPage(t,e)}catch(e){throw this._logger.error(`DefaultView.loadOlder(); failed`,{error:e}),e}finally{this._loadingOlder=!1}}}runOf(e){this._logger.trace(`DefaultView.runOf();`,{codecMessageId:e});let t=this._tree.getNodeByCodecMessageId(e);if(!t)return;if(t.kind===`run`)return mn(t);let n=this._selectedReplyRun(t.codecMessageId);return n?mn(n):void 0}_selectedReplyRun(e){let t=this._tree.getReplyRuns(e);if(t.length===0)return;if(t.length===1)return t[0];let n=this._tree.getGroupRoot(t[0]?.runId??``),r=this._regenSelections.get(n),i=r&&r.kind!==`pending`?r.selectedRunId:void 0;if(i!==void 0){let e=t.find(e=>e.runId===i);if(e)return e}return t.toSorted((e,t)=>(e.startSerial??``).localeCompare(t.startSerial??``)).at(-1)}run(e){this._logger.trace(`DefaultView.run();`,{runId:e});let t=this._tree.getRunNode(e);return t?mn(t):void 0}branchSelection(e){let t=this._resolveMessageBranchPoint(e);if(t){let e=t.siblings.flatMap(e=>{let t=this._codec.getMessages(e.projection).at(0);return t?[t.message]:[]});if(e.length>0){let n=this._resolveSelectedIndex(t),r=Math.max(0,Math.min(n,e.length-1)),i=e[r];return{hasSiblings:e.length>1,siblings:e,index:r,selected:i}}}let n=this._tree.getNodeByCodecMessageId(e);if(n){let t=this._codec.getMessages(n.projection).find(t=>t.codecMessageId===e);if(t!==void 0)return{hasSiblings:!1,siblings:[t.message],index:0,selected:t.message}}return{hasSiblings:!1,siblings:[],index:0,selected:void 0}}selectSibling(e,t){this._logger.trace(`DefaultView.selectSibling();`,{codecMessageId:e,index:t});let n=this._resolveMessageBranchPoint(e);if(!n)return;let r=Math.max(0,Math.min(t,n.siblings.length-1)),i=n.siblings[r];i&&(n.kind===`fork-of`?(this._branchSelections.set(n.groupRoot,{kind:`user`,selectedKey:$(i)}),this._logger.debug(`DefaultView.selectSibling(); fork-of`,{codecMessageId:e,index:r,selectedKey:$(i)})):(this._regenSelections.set(n.groupRoot,{kind:`user`,selectedRunId:$(i)}),this._logger.debug(`DefaultView.selectSibling(); regenerate`,{codecMessageId:e,index:r,selectedRunId:$(i),groupRoot:n.groupRoot})),this._recomputeAndEmit())}_resolveSelectedIndex(e){if(e.kind===`fork-of`){let t=this._branchSelections.get(e.groupRoot);if(!t)return e.siblings.length-1;let n=e.siblings.findIndex(e=>$(e)===t.selectedKey);return n===-1?e.siblings.length-1:n}let t=this._regenSelections.get(e.groupRoot);if(!t||t.kind===`pending`)return e.siblings.length-1;let n=e.siblings.findIndex(e=>$(e)===t.selectedRunId);return n===-1?e.siblings.length-1:n}_resolveMessageBranchPoint(e){let t=this._tree.getNodeByCodecMessageId(e);if(!t)return;if(t.kind===`input`){let e=this._tree.getSiblingNodes(t.codecMessageId);return e.length>1?{kind:`fork-of`,groupRoot:this._tree.getGroupRoot(t.codecMessageId),siblings:e}:void 0}let n=this._tree.getSiblingNodes(t.runId);if(n.length>1&&this._codec.getMessages(t.projection).at(0)?.codecMessageId===e)return{kind:`regen`,groupRoot:this._tree.getGroupRoot(t.runId),siblings:n}}async send(e,n){if(this._logger.trace(`DefaultView.send();`),this._closed)throw new t.ErrorInfo(`unable to send; view is closed`,G.InvalidArgument,400);let r=fn(e),i=this._lastVisibleMessagePairs.at(-1)?.codecMessageId,a=await this._sendDelegate(r,n,i);return this._applyForkAutoSelect(a,n),a}_applyForkAutoSelect(e,t){if(!t?.forkOf)return;let n=e.optimisticCodecMessageIds.at(0);if(n===void 0)return;let r=this._tree.getGroupRoot(n);this._branchSelections.set(r,{kind:`auto`,selectedKey:n}),this._recomputeAndEmit()}_applyRegenerateAutoSelect(e,t){let n=this._runByCodecMessageId(t);if(!n)return;let r=this._tree.getGroupRoot(n.runId);this._regenSelections.set(r,{kind:`pending`,carrierCodecMessageId:e.inputCodecMessageId}),this._logger.debug(`DefaultView._applyRegenerateAutoSelect(); deferring regenerate selection`,{anchorCodecMessageId:t,groupRoot:r,carrier:e.inputCodecMessageId}),this._resolvePendingRegenSelections(),this._recomputeAndEmitIfChanged()}async regenerate(e,n){if(this._logger.trace(`DefaultView.regenerate();`,{messageId:e}),this._closed)throw new t.ErrorInfo(`unable to regenerate; view is closed`,G.InvalidArgument,400);let r=this._runByCodecMessageId(e);if(!r)throw new t.ErrorInfo(`unable to regenerate; message not found in tree: ${e}`,G.InvalidArgument,400);let i=this._findParentMsgId(r,e);if(!i)throw new t.ErrorInfo(`unable to regenerate; parent user message not found for ${e}`,G.InvalidArgument,400);let a=e;r.regeneratesCodecMessageId!==void 0&&this._codec.getMessages(r.projection).at(0)?.codecMessageId===e&&(a=r.regeneratesCodecMessageId);let o={...n,parent:i},s=this._codec.createRegenerate(a,i),c=await this._sendDelegate([s],o,i);return this._applyRegenerateAutoSelect(c,a),c}async edit(e,n,r){if(this._logger.trace(`DefaultView.edit();`,{messageId:e}),this._closed)throw new t.ErrorInfo(`unable to edit; view is closed`,G.InvalidArgument,400);let i=this._tree.getNodeByCodecMessageId(e);if(!i)throw new t.ErrorInfo(`unable to edit; message not found in tree: ${e}`,G.InvalidArgument,400);let a=this._findParentMsgId(i,e);return this.send(n,{...r,forkOf:e,parent:a})}_findParentMsgId(e,t){let n=this._lastVisibleMessagePairs,r=n.findIndex(e=>e.codecMessageId===t);if(r>0)return n[r-1]?.codecMessageId;if(r===0)return;let i=this._codec.getMessages(e.projection),a=i.findIndex(e=>e.codecMessageId===t);if(a>0)return i[a-1]?.codecMessageId;if(a===0&&e.parentCodecMessageId!==void 0){let t=this._tree.getNodeByCodecMessageId(e.parentCodecMessageId);if(t)return this._codec.getMessages(t.projection).at(-1)?.codecMessageId}}on(e,t){let n=t;return this._emitter.on(e,n),()=>{this._emitter.off(e,n)}}close(){if(!this._closed){this._logger.info(`DefaultView.close();`),this._closed=!0,this._loadingOlder=!1;for(let e of this._unsubs)e();this._unsubs.length=0,this._emitter.off(),this._branchSelections.clear(),this._regenSelections.clear(),this._withheldRunIds.clear(),this._withheldBuffer.length=0,this._onClose?.()}}async _loadFirstPage(e){let t=e*pn,n=await dn(this._channel,{limit:t},this._logger);this._closed||await this._revealFromPage(n,e)}async _revealFromPage(e,t){let n=new Set(this._treeVisibleNodes().map(e=>$(e))),{newVisible:r,lastPage:i}=await this._loadUntilVisible(e,t,n);this._closed||(this._lastHistoryPage=i,this._hasMoreHistory=i.hasNext(),this._splitReveal(r,t))}_splitReveal(e,t){let n=0,r=e.length;for(let i=e.length-1;i>=0;i--){if(e[i]?.kind===`run`){if(n===t)break;n++}r=i}let i=e.slice(r),a=e.slice(0,r);for(let e of a)this._withheldRunIds.add($(e));this._withheldBuffer.push(...a),this._releaseWithheld(i)}_processHistoryPage(e){this._processingHistory=!0;try{let t=this._codec.createDecoder();for(let n of e.rawMessages)Bt(this._tree,t,n);for(let t of e.rawMessages)this._tree.emitAblyMessage(t)}finally{this._processingHistory=!1}}async _loadUntilVisible(e,t,n){this._processHistoryPage(e);let r=e,i=()=>{let e=0;for(let t of this._treeVisibleNodes())t.kind===`run`&&!n.has($(t))&&e++;return e};for(;i()<t&&r.hasNext();){let e=await r.next();if(!e||this._closed)break;this._processHistoryPage(e),r=e}return{newVisible:this._treeVisibleNodes().filter(e=>!n.has($(e))),lastPage:r}}_releaseWithheld(e){for(let t of e)this._withheldRunIds.delete($(t));e.length>0&&this._recomputeAndEmit()}_updateVisibleSnapshot(e){let t=e??this._cachedNodes;this._lastVisibleNodeKeys=t.map(e=>$(e)),this._lastVisibleNodeKeySet=new Set(this._lastVisibleNodeKeys),this._lastVisibleProjections=t.map(e=>e.projection),this._lastVisibleMessagePairs=this._extractMessages(t)}_onTreeUpdate(){this._processingHistory||(this._pinBranchSelections(),this._resolvePendingRegenSelections(),this._recomputeAndEmitIfChanged())}_resolveSelections(){let e=new Map;for(let[t,n]of this._branchSelections)e.set(t,n.selectedKey);for(let[t,n]of this._regenSelections)n.kind!==`pending`&&e.set(t,n.selectedRunId);return e}_treeVisibleNodes(){return this._tree.visibleNodes(this._resolveSelections())}_pinBranchSelections(){for(let e of this._lastVisibleNodeKeys){if(this._tree.getNode(e)?.kind!==`input`||this._tree.getSiblingNodes(e).length<=1)continue;let t=this._tree.getGroupRoot(e);this._branchSelections.get(t)||this._branchSelections.set(t,{kind:`pinned`,selectedKey:e})}}_resolvePendingRegenSelections(){for(let[e,t]of this._regenSelections){if(t.kind===`user`)continue;let n=this._tree.getSiblingNodes(e).filter(e=>e.kind===`run`);if(n.length<=1)continue;let r=n.at(-1);r&&this._regenSelections.set(e,{kind:`auto`,selectedRunId:r.runId})}}_onTreeAblyMessage(e){let t=N(e),n=t[_],r=t[m];if(!n&&!r){this._emitter.emit(`ably-message`,e);return}r&&this._lastVisibleNodeKeySet.has(r)&&this._emitter.emit(`ably-message`,e)}_onTreeRun(e){if(this._lastVisibleNodeKeySet.has(e.runId)){this._emitter.emit(`run`,e);return}e.type===`start`&&this._isRunStartVisible(e)&&(this._lastVisibleNodeKeySet.add(e.runId),this._emitter.emit(`run`,e))}_isRunStartVisible(e){let{parent:t}=e;if(t===void 0)return!0;let n=this._tree.getNodeByCodecMessageId(t);return n?this._lastVisibleNodeKeySet.has($(n)):!0}_visibleChanged(e){if(e.length!==this._lastVisibleNodeKeys.length)return!0;for(let[t,n]of e.entries())if($(n)!==this._lastVisibleNodeKeys[t]||n.projection!==this._lastVisibleProjections[t])return!0;return!1}},gn=e=>new hn(e),_n=()=>{},vn=class{constructor(e){this._views=new Set,this._state=`ready`,this._pendingRunStarts=new Map;let t=Nt(e.client,e.codec);if(this._channel=e.client.channels.get(e.channelName,t),this._codec=e.codec,this._clientId=e.clientId,this._logger=(e.logger??vt({logLevel:gt.Silent})).withContext({component:`ClientSession`}),this._emitter=new ht(this._logger),this._hasAttachedOnce=this._channel.state===`attached`,this._tree=sn(this._codec,this._logger),this._view=gn({tree:this._tree,channel:this._channel,codec:this._codec,sendDelegate:this._internalSend.bind(this),logger:this._logger,onClose:()=>this._views.delete(this._view)}),this._decoder=this._codec.createDecoder(),this._encoder=this._codec.createEncoder(this._channel,this._clientId===void 0?void 0:{clientId:this._clientId}),this._views.add(this._view),this.tree=this._tree,this.view=this._view,e.messages){let t;for(let n of e.messages){let e=crypto.randomUUID(),r={[_]:e,[b]:`user`};t&&(r[x]=t),this._tree.applyMessage({inputs:[this._codec.createUserMessage(n)],outputs:[]},r),t=e}}this._onMessage=e=>{this._handleMessage(e)},this._onChannelStateChange=e=>{this._handleChannelStateChange(e)},this._channel.on(this._onChannelStateChange)}connect(){return this._state===`closed`?Promise.reject(new t.ErrorInfo(`unable to connect; session is closed`,G.SessionClosed,400)):this._connectPromise?this._connectPromise:(this._logger.trace(`DefaultClientSession.connect();`),this._connectPromise=this._channel.subscribe(this._onMessage).then(()=>{this._logger.debug(`DefaultClientSession.connect(); subscribed and attached`)},e=>{let n=new t.ErrorInfo(`unable to subscribe to channel; ${e instanceof Error?e.message:String(e)}`,G.SessionSubscriptionError,500,e instanceof t.ErrorInfo?e:void 0);throw this._logger.error(`DefaultClientSession.connect(); subscribe failed`),this._emitter.emit(`error`,n),n}),this._connectPromise)}async _requireConnected(e){if(!this._connectPromise)throw new t.ErrorInfo(`unable to ${e}; connect() must be called before ${e}()`,G.InvalidArgument,400);return this._connectPromise}_handleMessage(e){if(this._state!==`closed`)try{if(e.name===`ai-run-end`){let n=N(e);if((n[`run-reason`]??`complete`)===`error`){let e=n[E],r=e===void 0?NaN:Number(e),i=Number.isFinite(r)?r:G.SessionSubscriptionError,a=n[`error-message`]??`agent reported an error`,o=i>=1e4&&i<6e4?Math.floor(i/100):500,s=new t.ErrorInfo(a,i,o);this._logger.error(`ClientSession._handleMessage(); agent error received`,{runId:n[m],invocationId:n[h],code:i}),this._emitter.emit(`error`,s)}}let n=Bt(this._tree,this._decoder,e);if(n&&(n.type===`start`||n.type===`resume`)){let t=N(e)[T];if(t!==void 0){let e=this._pendingRunStarts.get(t);e&&(this._pendingRunStarts.delete(t),e.resolve(n.runId))}}this._tree.emitAblyMessage(e)}catch(e){let n=e instanceof t.ErrorInfo?e:void 0;this._emitter.emit(`error`,new t.ErrorInfo(`unable to process channel message; ${e instanceof Error?e.message:String(e)}`,G.SessionSubscriptionError,500,n))}}_handleChannelStateChange(e){if(this._state===`closed`)return;let{current:n,resumed:r}=e;if(n===`attached`&&!this._hasAttachedOnce){this._hasAttachedOnce=!0;return}if(!(n===`failed`||n===`suspended`||n===`detached`||n===`attached`&&!r))return;this._logger.error(`ClientSession._handleChannelStateChange(); channel continuity lost`,{current:n,resumed:r,previous:e.previous});let i=new t.ErrorInfo(`unable to deliver events; channel continuity lost (${n}${n===`attached`?`, resumed: false`:``})`,G.ChannelContinuityLost,500,e.reason);this._emitter.emit(`error`,i)}_cleanupFailedSend(e){for(let t of e){let e=this._tree.getNodeByCodecMessageId(t);e?.kind===`input`&&e.serial===void 0&&this._tree.delete(e.codecMessageId)}}createView(){if(this._state===`closed`)throw new t.ErrorInfo(`unable to create view; session is closed`,G.SessionClosed,400);this._logger.trace(`DefaultClientSession.createView();`);let e=gn({tree:this._tree,channel:this._channel,codec:this._codec,sendDelegate:this._internalSend.bind(this),logger:this._logger,onClose:()=>this._views.delete(e)});return this._views.add(e),e}async _internalSend(e,n,r){if(this._state===`closed`||(await this._requireConnected(`send`),this._state===`closed`))throw new t.ErrorInfo(`unable to send; session is closed`,G.SessionClosed,400);let i=this._channel.state;if(i!==`attached`&&i!==`attaching`)throw new t.ErrorInfo(`unable to send; channel is ${i}`,G.ChannelNotReady,400);this._logger.trace(`ClientSession._internalSend();`);let a=n?.runId!==void 0,o=n?.runId,s;n?.parent===void 0&&!n?.forkOf&&(s=r);let c=new Set,l=[];for(let t of e){let e=crypto.randomUUID(),r=t.codecMessageId??crypto.randomUUID();c.add(r);let i=t.kind!==`user-message`&&(t.kind===`regenerate`||t.codecMessageId!==void 0),a=t.parent??(n?.parent===void 0?s:n.parent),u=n?.forkOf,d=t.kind===`regenerate`?t.target:void 0,f=Pt({role:`user`,runId:o,codecMessageId:r,runClientId:this._clientId,...a!==void 0&&{parent:a},...u!==void 0&&{forkOf:u},...d!==void 0&&{regenerates:d},inputEventId:e});i||this._tree.applyMessage({inputs:[t],outputs:[]},f),l.push({input:t,codecMessageId:r,inputEventId:e,headers:f,isWireOnly:i}),!i&&n?.parent===void 0&&!n?.forkOf&&t.parent===void 0&&(s=r)}let u=l.at(-1);if(u===void 0)throw new t.ErrorInfo(`unable to send; inputs array is empty (include at least one input)`,G.InvalidArgument,400);let d=u.inputEventId,f=u.codecMessageId,p=new Promise((e,t)=>{this._pendingRunStarts.set(f,{resolve:e,reject:t})});return p.catch(()=>{}),await(async()=>{try{for(let e of l)await this._encoder.publishInput(e.input,{extras:{headers:e.headers},messageId:e.codecMessageId,...this._clientId!==void 0&&{clientId:this._clientId}})}catch(e){let n=e instanceof t.ErrorInfo?e:void 0,r=n?.statusCode===401||n?.statusCode===403,i=new t.ErrorInfo(r?`unable to publish events; missing publish capability on the channel`:`unable to publish events; ${e instanceof Error?e.message:String(e)}`,r?G.InsufficientCapability:G.SessionSendFailed,r?401:500,n);throw this._emitter.emit(`error`,i),this._pendingRunStarts.delete(f),a||this._cleanupFailedSend([...c]),i}})(),{inputCodecMessageId:f,runId:p,inputEventId:d,cancel:async()=>{await this._publishCancel({inputCodecMessageId:f,...o!==void 0&&{runId:o}})},optimisticCodecMessageIds:[...c],toInvocation:()=>tn.fromJSON({inputEventId:d,sessionName:this._channel.name})}}async cancel(e){return this._publishCancel({runId:e})}async _publishCancel(e){if(this._state===`closed`||(await this._requireConnected(`cancel`),this._state===`closed`))return;this._logger.debug(`ClientSession._publishCancel();`,{runId:e.runId,inputCodecMessageId:e.inputCodecMessageId});let t={[g]:crypto.randomUUID()};e.runId!==void 0&&(t[m]=e.runId),e.inputCodecMessageId!==void 0&&(t[T]=e.inputCodecMessageId),await this._channel.publish({name:D,extras:{ai:{transport:t}}})}on(e,t){if(this._state===`closed`)return _n;let n=t;return this._emitter.on(e,n),()=>{this._emitter.off(e,n)}}async close(){if(this._state!==`closed`){this._state=`closed`,this._logger.info(`ClientSession.close();`),this._connectPromise&&this._channel.unsubscribe(this._onMessage),this._channel.off(this._onChannelStateChange),this._emitter.off();for(let e of this._views)e.close();if(this._views.clear(),this._pendingRunStarts.size>0){let e=new t.ErrorInfo(`unable to await run-start; session closed`,G.SessionClosed,400);for(let t of this._pendingRunStarts.values())t.reject(e);this._pendingRunStarts.clear()}try{await this._encoder.close()}catch{}if(this._connectPromise)try{await this._channel.detach()}catch(e){this._logger.debug(`ClientSession.close(); channel detach failed`,{error:e})}}}},yn=e=>new vn(e),bn=e=>yn({...e,codec:Q}),xn=e=>en({...e,codec:Q}),Sn=async(e,t)=>{if(e.reason!==`complete`)return Promise.resolve(t).catch(()=>{}),e.reason;try{return await t===`tool-calls`?`suspend`:`complete`}catch(e){return Cn(e)?`cancelled`:`error`}},Cn=e=>typeof e!=`object`||!e?!1:e.name===`AbortError`;e.UIMessageCodec=Q,e.createAgentSession=xn,e.createChatTransport=kt,e.createClientSession=bn,e.vercelRunOutcome=Sn});
|
|
2
2
|
//# sourceMappingURL=ably-ai-transport-vercel.umd.cjs.map
|