@ag-ui/client 0.0.28 → 0.0.29
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/dist/index.d.mts +1 -0
- package/dist/index.d.ts +1 -0
- package/dist/index.js +5 -5
- package/dist/index.js.map +1 -1
- package/dist/index.mjs +5 -5
- package/dist/index.mjs.map +1 -1
- package/package.json +4 -4
package/dist/index.d.mts
CHANGED
|
@@ -196,6 +196,7 @@ declare abstract class AbstractAgent {
|
|
|
196
196
|
threadId: string;
|
|
197
197
|
messages: Message[];
|
|
198
198
|
state: State;
|
|
199
|
+
debug: boolean;
|
|
199
200
|
constructor({ agentId, description, threadId, initialMessages, initialState, debug, }?: AgentConfig);
|
|
200
201
|
protected abstract run(...args: Parameters<RunAgent>): ReturnType<RunAgent>;
|
|
201
202
|
runAgent(parameters?: RunAgentParameters): Promise<void>;
|
package/dist/index.d.ts
CHANGED
|
@@ -196,6 +196,7 @@ declare abstract class AbstractAgent {
|
|
|
196
196
|
threadId: string;
|
|
197
197
|
messages: Message[];
|
|
198
198
|
state: State;
|
|
199
|
+
debug: boolean;
|
|
199
200
|
constructor({ agentId, description, threadId, initialMessages, initialState, debug, }?: AgentConfig);
|
|
200
201
|
protected abstract run(...args: Parameters<RunAgent>): ReturnType<RunAgent>;
|
|
201
202
|
runAgent(parameters?: RunAgentParameters): Promise<void>;
|
package/dist/index.js
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
|
-
"use strict";var pe=Object.create;var w=Object.defineProperty,
|
|
1
|
+
"use strict";var pe=Object.create;var w=Object.defineProperty,Te=Object.defineProperties,de=Object.getOwnPropertyDescriptor,fe=Object.getOwnPropertyDescriptors,Se=Object.getOwnPropertyNames,W=Object.getOwnPropertySymbols,me=Object.getPrototypeOf,Q=Object.prototype.hasOwnProperty,Ae=Object.prototype.propertyIsEnumerable;var Y=(r,n,e)=>n in r?w(r,n,{enumerable:!0,configurable:!0,writable:!0,value:e}):r[n]=e,C=(r,n)=>{for(var e in n||(n={}))Q.call(n,e)&&Y(r,e,n[e]);if(W)for(var e of W(n))Ae.call(n,e)&&Y(r,e,n[e]);return r},M=(r,n)=>Te(r,fe(n));var ye=(r,n)=>{for(var e in n)w(r,e,{get:n[e],enumerable:!0})},G=(r,n,e,t)=>{if(n&&typeof n=="object"||typeof n=="function")for(let s of Se(n))!Q.call(r,s)&&s!==e&&w(r,s,{get:()=>n[s],enumerable:!(t=de(n,s))||t.enumerable});return r},h=(r,n,e)=>(G(r,n,"default"),e&&G(e,n,"default")),P=(r,n,e)=>(e=r!=null?pe(me(r)):{},G(n||!r||!r.__esModule?w(e,"default",{value:r,enumerable:!0}):e,r)),_e=r=>G(w({},"__esModule",{value:!0}),r);var L={};ye(L,{AbstractAgent:()=>O,HttpAgent:()=>V,convertToLegacyEvents:()=>J,defaultApplyEvents:()=>X,parseProtoStream:()=>k,parseSSEStream:()=>K,runHttpRequest:()=>U,transformHttpEventStream:()=>j,verifyEvents:()=>H});module.exports=_e(L);var m=require("@ag-ui/core"),Z=require("rxjs/operators");var v=r=>{if(typeof structuredClone=="function")return structuredClone(r);try{return JSON.parse(JSON.stringify(r))}catch(n){return C({},r)}};var ee=require("fast-json-patch"),te=P(require("untruncate-json"));var X=(...r)=>{let[n,e]=r,t=v(n.messages),s=v(n.state),c,g=u=>[v(u)],a=()=>[];return e.pipe((0,Z.mergeMap)(u=>{var E;switch(u.type){case m.EventType.TEXT_MESSAGE_START:{let{messageId:d,role:i}=u,T={id:d,role:i,content:""};return t.push(T),g({messages:t})}case m.EventType.TEXT_MESSAGE_CONTENT:{let{delta:d}=u,i=t[t.length-1];return i.content=i.content+d,g({messages:t})}case m.EventType.TEXT_MESSAGE_END:return a();case m.EventType.TOOL_CALL_START:{let{toolCallId:d,toolCallName:i,parentMessageId:T}=u,f;return T&&t.length>0&&t[t.length-1].id===T?f=t[t.length-1]:(f={id:T||d,role:"assistant",toolCalls:[]},t.push(f)),(E=f.toolCalls)!=null||(f.toolCalls=[]),f.toolCalls.push({id:d,type:"function",function:{name:i,arguments:""}}),g({messages:t})}case m.EventType.TOOL_CALL_ARGS:{let{delta:d}=u,i=t[t.length-1],T=i.toolCalls[i.toolCalls.length-1];if(T.function.arguments+=d,c){let f=c.find(N=>N.tool===T.function.name);if(f)try{let N=JSON.parse((0,te.default)(T.function.arguments));return f.tool_argument&&f.tool_argument in N?(s=M(C({},s),{[f.state_key]:N[f.tool_argument]}),g({messages:t,state:s})):(s=M(C({},s),{[f.state_key]:N}),g({messages:t,state:s}))}catch(N){}}return g({messages:t})}case m.EventType.TOOL_CALL_END:return a();case m.EventType.STATE_SNAPSHOT:{let{snapshot:d}=u;return s=d,g({state:s})}case m.EventType.STATE_DELTA:{let{delta:d}=u;try{return s=(0,ee.applyPatch)(s,d,!0,!1).newDocument,g({state:s})}catch(i){let T=i instanceof Error?i.message:String(i);return console.warn(`Failed to apply state patch:
|
|
2
2
|
Current state: ${JSON.stringify(s,null,2)}
|
|
3
|
-
Patch operations: ${JSON.stringify(
|
|
4
|
-
Error: ${
|
|
5
|
-
`),
|
|
6
|
-
`),S=JSON.parse(r);n.next(S)}catch(r){n.error(r)}}return n.asObservable()};var ae=require("rxjs");var oe=G(require("@ag-ui/proto")),j=a=>{let n=new ae.Subject,e=new Uint8Array(0);a.subscribe({next:s=>{if(s.type!=="headers"&&s.type==="data"&&s.data){let l=new Uint8Array(e.length+s.data.length);l.set(e,0),l.set(s.data,e.length),e=l,t()}},error:s=>n.error(s),complete:()=>{if(e.length>0)try{t()}catch(s){console.warn("Incomplete or invalid protocol buffer data at stream end")}n.complete()}});function t(){for(;e.length>=4;){let c=4+new DataView(e.buffer,e.byteOffset,4).getUint32(0,!1);if(e.length<c)break;try{let i=e.slice(4,c),r=oe.decode(i);n.next(r),e=e.slice(c)}catch(i){let r=i instanceof Error?i.message:String(i);n.error(new Error(`Failed to decode protocol buffer message: ${r}`));return}}}return n.asObservable()};var le=G(require("@ag-ui/proto")),J=a=>{let n=new k.Subject,e=new k.ReplaySubject,t=!1;return a.subscribe({next:s=>{e.next(s),s.type==="headers"&&!t?(t=!0,s.headers.get("content-type")===le.AGUI_MEDIA_TYPE?j(e).subscribe({next:c=>n.next(c),error:c=>n.error(c),complete:()=>n.complete()}):X(e).subscribe({next:c=>{try{let i=ie.EventSchemas.parse(c);n.next(i)}catch(i){n.error(i)}},error:c=>n.error(c),complete:()=>n.complete()})):t||n.error(new Error("No headers event received before data events"))},error:s=>{e.error(s),n.error(s)},complete:()=>{e.complete()}}),n.asObservable()};var ce=require("rxjs/operators"),ue=require("fast-json-patch"),v=require("@ag-ui/core");var o=require("zod"),A=o.z.enum(["TextMessageStart","TextMessageContent","TextMessageEnd","ActionExecutionStart","ActionExecutionArgs","ActionExecutionEnd","ActionExecutionResult","AgentStateMessage","MetaEvent","RunStarted","RunFinished","RunError","NodeStarted","NodeFinished"]),Ce=o.z.enum(["LangGraphInterruptEvent","PredictState","Exit"]),_e=o.z.object({type:o.z.literal(A.enum.TextMessageStart),messageId:o.z.string(),parentMessageId:o.z.string().optional()}),Le=o.z.object({type:o.z.literal(A.enum.TextMessageContent),messageId:o.z.string(),content:o.z.string()}),xe=o.z.object({type:o.z.literal(A.enum.TextMessageEnd),messageId:o.z.string()}),he=o.z.object({type:o.z.literal(A.enum.ActionExecutionStart),actionExecutionId:o.z.string(),actionName:o.z.string(),parentMessageId:o.z.string().optional()}),Me=o.z.object({type:o.z.literal(A.enum.ActionExecutionArgs),actionExecutionId:o.z.string(),args:o.z.string()}),Re=o.z.object({type:o.z.literal(A.enum.ActionExecutionEnd),actionExecutionId:o.z.string()}),Oe=o.z.object({type:o.z.literal(A.enum.ActionExecutionResult),actionName:o.z.string(),actionExecutionId:o.z.string(),result:o.z.string()}),be=o.z.object({type:o.z.literal(A.enum.AgentStateMessage),threadId:o.z.string(),agentName:o.z.string(),nodeName:o.z.string(),runId:o.z.string(),active:o.z.boolean(),role:o.z.string(),state:o.z.string(),running:o.z.boolean()}),Ne=o.z.object({type:o.z.literal(A.enum.MetaEvent),name:Ce,value:o.z.any()}),yt=o.z.discriminatedUnion("type",[_e,Le,xe,he,Me,Re,Oe,be,Ne]),vt=o.z.object({id:o.z.string(),role:o.z.string(),content:o.z.string(),parentMessageId:o.z.string().optional()}),Ct=o.z.object({id:o.z.string(),name:o.z.string(),arguments:o.z.any(),parentMessageId:o.z.string().optional()}),_t=o.z.object({id:o.z.string(),result:o.z.any(),actionExecutionId:o.z.string(),actionName:o.z.string()});var ge=G(require("untruncate-json"));var z=(a,n,e)=>t=>{let s={},l=!0,c=!0,i="",r=null,S=null,f=[],T=u=>{typeof u=="object"&&u!==null&&("messages"in u&&delete u.messages,s=u)};return t.pipe((0,ce.mergeMap)(u=>{switch(u.type){case v.EventType.TEXT_MESSAGE_START:{let E=u;return[{type:A.enum.TextMessageStart,messageId:E.messageId}]}case v.EventType.TEXT_MESSAGE_CONTENT:{let E=u;return[{type:A.enum.TextMessageContent,messageId:E.messageId,content:E.delta}]}case v.EventType.TEXT_MESSAGE_END:{let E=u;return[{type:A.enum.TextMessageEnd,messageId:E.messageId}]}case v.EventType.TOOL_CALL_START:{let E=u;return f.push({id:E.toolCallId,type:"function",function:{name:E.toolCallName,arguments:""}}),c=!0,[{type:A.enum.ActionExecutionStart,actionExecutionId:E.toolCallId,actionName:E.toolCallName,parentMessageId:E.parentMessageId}]}case v.EventType.TOOL_CALL_ARGS:{let E=u,d=f[f.length-1];d.function.arguments+=E.delta;let L=!1;if(S){let O=S.find(b=>b.tool==d.function.name);if(O)try{let b=JSON.parse((0,ge.default)(d.function.arguments));O.tool_argument&&O.tool_argument in b?(T(M(_({},s),{[O.state_key]:b[O.tool_argument]})),L=!0):O.tool_argument||(T(M(_({},s),{[O.state_key]:b})),L=!0)}catch(b){}}return[{type:A.enum.ActionExecutionArgs,actionExecutionId:E.toolCallId,args:E.delta},...L?[{type:A.enum.AgentStateMessage,threadId:a,agentName:e,nodeName:i,runId:n,running:l,role:"assistant",state:JSON.stringify(s),active:c}]:[]]}case v.EventType.TOOL_CALL_END:{let E=u;return[{type:A.enum.ActionExecutionEnd,actionExecutionId:E.toolCallId}]}case v.EventType.RAW:return[];case v.EventType.CUSTOM:{let E=u;switch(E.name){case"Exit":l=!1;break;case"PredictState":S=E.value;break}return[{type:A.enum.MetaEvent,name:E.name,value:E.value}]}case v.EventType.STATE_SNAPSHOT:return T(u.snapshot),[{type:A.enum.AgentStateMessage,threadId:a,agentName:e,nodeName:i,runId:n,running:l,role:"assistant",state:JSON.stringify(s),active:c}];case v.EventType.STATE_DELTA:{let d=(0,ue.applyPatch)(s,u.delta,!0,!1);return d?(T(d.newDocument),[{type:A.enum.AgentStateMessage,threadId:a,agentName:e,nodeName:i,runId:n,running:l,role:"assistant",state:JSON.stringify(s),active:c}]):[]}case v.EventType.MESSAGES_SNAPSHOT:return r=u.messages,[{type:A.enum.AgentStateMessage,threadId:a,agentName:e,nodeName:i,runId:n,running:l,role:"assistant",state:JSON.stringify(_(_({},s),r?{messages:r}:{})),active:!0}];case v.EventType.RUN_STARTED:return[];case v.EventType.RUN_FINISHED:return r&&(s.messages=r),[{type:A.enum.AgentStateMessage,threadId:a,agentName:e,nodeName:i,runId:n,running:l,role:"assistant",state:JSON.stringify(_(_({},s),r?{messages:Ie(r)}:{})),active:!1}];case v.EventType.RUN_ERROR:return console.error("Run error",u),[];case v.EventType.STEP_STARTED:return i=u.stepName,f=[],S=null,[{type:A.enum.AgentStateMessage,threadId:a,agentName:e,nodeName:i,runId:n,running:l,role:"assistant",state:JSON.stringify(s),active:!0}];case v.EventType.STEP_FINISHED:return f=[],S=null,[{type:A.enum.AgentStateMessage,threadId:a,agentName:e,nodeName:i,runId:n,running:l,role:"assistant",state:JSON.stringify(s),active:!1}];default:return[]}}))};function Ie(a){var e;let n=[];for(let t of a)if(t.role==="assistant"||t.role==="user"||t.role==="system"){if(t.content){let s={id:t.id,role:t.role,content:t.content};n.push(s)}if(t.role==="assistant"&&t.toolCalls&&t.toolCalls.length>0)for(let s of t.toolCalls){let l={id:s.id,name:s.function.name,arguments:JSON.parse(s.function.arguments),parentMessageId:t.id};n.push(l)}}else if(t.role==="tool"){let s="unknown";for(let c of a)if(c.role==="assistant"&&((e=c.toolCalls)!=null&&e.length)){for(let i of c.toolCalls)if(i.id===t.toolCallId){s=i.function.name;break}}let l={id:t.id,result:t.content,actionExecutionId:t.toolCallId,actionName:s};n.push(l)}return n}var H=require("uuid");var I=require("rxjs/operators"),Ee=require("rxjs/operators"),$=require("rxjs");var K=require("rxjs");var B=require("rxjs"),m=require("@ag-ui/core"),q=a=>n=>{let e,t,s,l=()=>{if(!e||s!=="text")throw new Error("No text message to close");let r={type:m.EventType.TEXT_MESSAGE_END,messageId:e.messageId};return s=void 0,e=void 0,a&&console.debug("[TRANSFORM]: TEXT_MESSAGE_END",JSON.stringify(r)),r},c=()=>{if(!t||s!=="tool")throw new Error("No tool call to close");let r={type:m.EventType.TOOL_CALL_END,toolCallId:t.toolCallId};return s=void 0,t=void 0,a&&console.debug("[TRANSFORM]: TOOL_CALL_END",JSON.stringify(r)),r},i=()=>s==="text"?[l()]:s==="tool"?[c()]:[];return n.pipe((0,B.mergeMap)(r=>{switch(r.type){case m.EventType.TEXT_MESSAGE_START:case m.EventType.TEXT_MESSAGE_CONTENT:case m.EventType.TEXT_MESSAGE_END:case m.EventType.TOOL_CALL_START:case m.EventType.TOOL_CALL_ARGS:case m.EventType.TOOL_CALL_END:case m.EventType.STATE_SNAPSHOT:case m.EventType.STATE_DELTA:case m.EventType.MESSAGES_SNAPSHOT:case m.EventType.CUSTOM:case m.EventType.RUN_STARTED:case m.EventType.RUN_FINISHED:case m.EventType.RUN_ERROR:case m.EventType.STEP_STARTED:case m.EventType.STEP_FINISHED:return[...i(),r];case m.EventType.RAW:return[r];case m.EventType.TEXT_MESSAGE_CHUNK:let f=r,T=[];if((s!=="text"||f.messageId!==void 0&&f.messageId!==(e==null?void 0:e.messageId))&&T.push(...i()),s!=="text"){if(f.messageId===void 0)throw new Error("First TEXT_MESSAGE_CHUNK must have a messageId");e={messageId:f.messageId},s="text";let d={type:m.EventType.TEXT_MESSAGE_START,messageId:f.messageId,role:"assistant"};T.push(d),a&&console.debug("[TRANSFORM]: TEXT_MESSAGE_START",JSON.stringify(d))}if(f.delta!==void 0){let d={type:m.EventType.TEXT_MESSAGE_CONTENT,messageId:e.messageId,delta:f.delta};T.push(d),a&&console.debug("[TRANSFORM]: TEXT_MESSAGE_CONTENT",JSON.stringify(d))}return T;case m.EventType.TOOL_CALL_CHUNK:let u=r,E=[];if((s!=="tool"||u.toolCallId!==void 0&&u.toolCallId!==(t==null?void 0:t.toolCallId))&&E.push(...i()),s!=="tool"){if(u.toolCallId===void 0)throw new Error("First TOOL_CALL_CHUNK must have a toolCallId");if(u.toolCallName===void 0)throw new Error("First TOOL_CALL_CHUNK must have a toolCallName");t={toolCallId:u.toolCallId,toolCallName:u.toolCallName,parentMessageId:u.parentMessageId},s="tool";let d={type:m.EventType.TOOL_CALL_START,toolCallId:u.toolCallId,toolCallName:u.toolCallName,parentMessageId:u.parentMessageId};E.push(d),a&&console.debug("[TRANSFORM]: TOOL_CALL_START",JSON.stringify(d))}if(u.delta!==void 0){let d={type:m.EventType.TOOL_CALL_ARGS,toolCallId:t.toolCallId,delta:u.delta};E.push(d),a&&console.debug("[TRANSFORM]: TOOL_CALL_ARGS",JSON.stringify(d))}return E}let S=r.type}),(0,B.finalize)(()=>i()))};var N=class{constructor({agentId:n,description:e,threadId:t,initialMessages:s,initialState:l,debug:c}={}){this.debug=!1;this.agentId=n,this.description=e!=null?e:"",this.threadId=t!=null?t:(0,H.v4)(),this.messages=C(s!=null?s:[]),this.state=C(l!=null?l:{}),this.debug=c!=null?c:!1}async runAgent(n){var s;this.agentId=(s=this.agentId)!=null?s:(0,H.v4)();let e=this.prepareRunAgentInput(n),t=(0,$.pipe)(()=>this.run(e),q(this.debug),D(this.debug),l=>this.apply(e,l),l=>this.processApplyEvents(e,l),(0,I.catchError)(l=>(this.onError(l),(0,$.throwError)(()=>l))),(0,Ee.finalize)(()=>{this.onFinalize()}));return(0,K.lastValueFrom)(t((0,K.of)(null))).then(()=>{})}abortRun(){}apply(...n){return U(...n)}processApplyEvents(n,e){return e.pipe((0,I.tap)(t=>{t.messages&&(this.messages=t.messages),t.state&&(this.state=t.state)}))}prepareRunAgentInput(n){var e,t,s;return{threadId:this.threadId,runId:(n==null?void 0:n.runId)||(0,H.v4)(),tools:C((e=n==null?void 0:n.tools)!=null?e:[]),context:C((t=n==null?void 0:n.context)!=null?t:[]),forwardedProps:C((s=n==null?void 0:n.forwardedProps)!=null?s:{}),state:C(this.state),messages:C(this.messages)}}onError(n){console.error("Agent execution failed:",n)}onFinalize(){}clone(){let n=Object.create(Object.getPrototypeOf(this));for(let e of Object.getOwnPropertyNames(this)){let t=this[e];typeof t!="function"&&(n[e]=C(t))}return n}legacy_to_be_removed_runAgentBridged(n){var t;this.agentId=(t=this.agentId)!=null?t:(0,H.v4)();let e=this.prepareRunAgentInput(n);return this.run(e).pipe(q(this.debug),D(this.debug),z(this.threadId,e.runId,this.agentId),s=>s.pipe((0,I.map)(l=>(this.debug&&console.debug("[LEGACY]:",JSON.stringify(l)),l))))}};var V=class extends N{constructor(e){var t;super(e);this.abortController=new AbortController;this.url=e.url,this.headers=C((t=e.headers)!=null?t:{})}requestInit(e){return{method:"POST",headers:M(_({},this.headers),{"Content-Type":"application/json",Accept:"text/event-stream"}),body:JSON.stringify(e),signal:this.abortController.signal}}runAgent(e){var t;return this.abortController=(t=e==null?void 0:e.abortController)!=null?t:new AbortController,super.runAgent(e)}abortRun(){this.abortController.abort(),super.abortRun()}run(e){let t=F(this.url,this.requestInit(e));return J(t)}};h(x,require("@ag-ui/core"),module.exports);0&&(module.exports={AbstractAgent,HttpAgent,convertToLegacyEvents,defaultApplyEvents,parseProtoStream,parseSSEStream,runHttpRequest,transformHttpEventStream,verifyEvents,...require("@ag-ui/core")});
|
|
3
|
+
Patch operations: ${JSON.stringify(d,null,2)}
|
|
4
|
+
Error: ${T}`),a()}}case m.EventType.MESSAGES_SNAPSHOT:{let{messages:d}=u;return t=d,g({messages:t})}case m.EventType.RAW:return a();case m.EventType.CUSTOM:{let d=u;return d.name==="PredictState"&&(c=d.value),a()}case m.EventType.RUN_STARTED:return a();case m.EventType.RUN_FINISHED:return a();case m.EventType.RUN_ERROR:return a();case m.EventType.STEP_STARTED:return a();case m.EventType.STEP_FINISHED:return c=void 0,a();case m.EventType.TEXT_MESSAGE_CHUNK:throw new Error("TEXT_MESSAGE_CHUNK must be tranformed before being applied");case m.EventType.TOOL_CALL_CHUNK:throw new Error("TOOL_CALL_CHUNK must be tranformed before being applied");case m.EventType.THINKING_START:return a();case m.EventType.THINKING_END:return a();case m.EventType.THINKING_TEXT_MESSAGE_START:return a();case m.EventType.THINKING_TEXT_MESSAGE_CONTENT:return a();case m.EventType.THINKING_TEXT_MESSAGE_END:return a()}let y=u.type;return a()}))};var l=require("@ag-ui/core"),p=require("rxjs"),ne=require("rxjs/operators"),H=r=>n=>{let e,t,s=!1,c=!1,g=!1,a=new Map,u=!1,y=!1;return n.pipe((0,ne.mergeMap)(E=>{let d=E.type;if(r&&console.debug("[VERIFY]:",JSON.stringify(E)),c)return(0,p.throwError)(()=>new l.AGUIError(`Cannot send event type '${d}': The run has already errored with 'RUN_ERROR'. No further events can be sent.`));if(s&&d!==l.EventType.RUN_ERROR)return(0,p.throwError)(()=>new l.AGUIError(`Cannot send event type '${d}': The run has already finished with 'RUN_FINISHED'. Start a new run with 'RUN_STARTED'.`));if(e!==void 0&&![l.EventType.TEXT_MESSAGE_CONTENT,l.EventType.TEXT_MESSAGE_END,l.EventType.RAW].includes(d))return(0,p.throwError)(()=>new l.AGUIError(`Cannot send event type '${d}' after 'TEXT_MESSAGE_START': Send 'TEXT_MESSAGE_END' first.`));if(t!==void 0&&![l.EventType.TOOL_CALL_ARGS,l.EventType.TOOL_CALL_END,l.EventType.RAW].includes(d))return d===l.EventType.TOOL_CALL_START?(0,p.throwError)(()=>new l.AGUIError("Cannot send 'TOOL_CALL_START' event: A tool call is already in progress. Complete it with 'TOOL_CALL_END' first.")):(0,p.throwError)(()=>new l.AGUIError(`Cannot send event type '${d}' after 'TOOL_CALL_START': Send 'TOOL_CALL_END' first.`));if(g){if(d===l.EventType.RUN_STARTED)return(0,p.throwError)(()=>new l.AGUIError("Cannot send multiple 'RUN_STARTED' events: A 'RUN_STARTED' event was already sent. Each run must have exactly one 'RUN_STARTED' event at the beginning."))}else if(g=!0,d!==l.EventType.RUN_STARTED&&d!==l.EventType.RUN_ERROR)return(0,p.throwError)(()=>new l.AGUIError("First event must be 'RUN_STARTED'"));switch(d){case l.EventType.TEXT_MESSAGE_START:return e!==void 0?(0,p.throwError)(()=>new l.AGUIError("Cannot send 'TEXT_MESSAGE_START' event: A text message is already in progress. Complete it with 'TEXT_MESSAGE_END' first.")):(e=E.messageId,(0,p.of)(E));case l.EventType.TEXT_MESSAGE_CONTENT:return e===void 0?(0,p.throwError)(()=>new l.AGUIError("Cannot send 'TEXT_MESSAGE_CONTENT' event: No active text message found. Start a text message with 'TEXT_MESSAGE_START' first.")):E.messageId!==e?(0,p.throwError)(()=>new l.AGUIError(`Cannot send 'TEXT_MESSAGE_CONTENT' event: Message ID mismatch. The ID '${E.messageId}' doesn't match the active message ID '${e}'.`)):(0,p.of)(E);case l.EventType.TEXT_MESSAGE_END:return e===void 0?(0,p.throwError)(()=>new l.AGUIError("Cannot send 'TEXT_MESSAGE_END' event: No active text message found. A 'TEXT_MESSAGE_START' event must be sent first.")):E.messageId!==e?(0,p.throwError)(()=>new l.AGUIError(`Cannot send 'TEXT_MESSAGE_END' event: Message ID mismatch. The ID '${E.messageId}' doesn't match the active message ID '${e}'.`)):(e=void 0,(0,p.of)(E));case l.EventType.TOOL_CALL_START:return t!==void 0?(0,p.throwError)(()=>new l.AGUIError("Cannot send 'TOOL_CALL_START' event: A tool call is already in progress. Complete it with 'TOOL_CALL_END' first.")):(t=E.toolCallId,(0,p.of)(E));case l.EventType.TOOL_CALL_ARGS:return t===void 0?(0,p.throwError)(()=>new l.AGUIError("Cannot send 'TOOL_CALL_ARGS' event: No active tool call found. Start a tool call with 'TOOL_CALL_START' first.")):E.toolCallId!==t?(0,p.throwError)(()=>new l.AGUIError(`Cannot send 'TOOL_CALL_ARGS' event: Tool call ID mismatch. The ID '${E.toolCallId}' doesn't match the active tool call ID '${t}'.`)):(0,p.of)(E);case l.EventType.TOOL_CALL_END:return t===void 0?(0,p.throwError)(()=>new l.AGUIError("Cannot send 'TOOL_CALL_END' event: No active tool call found. A 'TOOL_CALL_START' event must be sent first.")):E.toolCallId!==t?(0,p.throwError)(()=>new l.AGUIError(`Cannot send 'TOOL_CALL_END' event: Tool call ID mismatch. The ID '${E.toolCallId}' doesn't match the active tool call ID '${t}'.`)):(t=void 0,(0,p.of)(E));case l.EventType.STEP_STARTED:{let i=E.name;return a.has(i)?(0,p.throwError)(()=>new l.AGUIError(`Step "${i}" is already active for 'STEP_STARTED'`)):(a.set(i,!0),(0,p.of)(E))}case l.EventType.STEP_FINISHED:{let i=E.name;return a.has(i)?(a.delete(i),(0,p.of)(E)):(0,p.throwError)(()=>new l.AGUIError(`Cannot send 'STEP_FINISHED' for step "${i}" that was not started`))}case l.EventType.RUN_STARTED:return(0,p.of)(E);case l.EventType.RUN_FINISHED:{if(a.size>0){let i=Array.from(a.keys()).join(", ");return(0,p.throwError)(()=>new l.AGUIError(`Cannot send 'RUN_FINISHED' while steps are still active: ${i}`))}return s=!0,(0,p.of)(E)}case l.EventType.RUN_ERROR:return c=!0,(0,p.of)(E);case l.EventType.CUSTOM:return(0,p.of)(E);case l.EventType.THINKING_TEXT_MESSAGE_START:return u?y?(0,p.throwError)(()=>new l.AGUIError("Cannot send 'THINKING_TEXT_MESSAGE_START' event: A thinking message is already in progress. Complete it with 'THINKING_TEXT_MESSAGE_END' first.")):(y=!0,(0,p.of)(E)):(0,p.throwError)(()=>new l.AGUIError("Cannot send 'THINKING_TEXT_MESSAGE_START' event: A thinking step is not in progress. Create one with 'THINKING_START' first."));case l.EventType.THINKING_TEXT_MESSAGE_CONTENT:return y?(0,p.of)(E):(0,p.throwError)(()=>new l.AGUIError("Cannot send 'THINKING_TEXT_MESSAGE_CONTENT' event: No active thinking message found. Start a message with 'THINKING_TEXT_MESSAGE_START' first."));case l.EventType.THINKING_TEXT_MESSAGE_END:return y?(y=!1,(0,p.of)(E)):(0,p.throwError)(()=>new l.AGUIError("Cannot send 'THINKING_TEXT_MESSAGE_END' event: No active thinking message found. A 'THINKING_TEXT_MESSAGE_START' event must be sent first."));case l.EventType.THINKING_START:return u?(0,p.throwError)(()=>new l.AGUIError("Cannot send 'THINKING_START' event: A thinking step is already in progress. End it with 'THINKING_END' first.")):(u=!0,(0,p.of)(E));case l.EventType.THINKING_END:return u?(u=!1,(0,p.of)(E)):(0,p.throwError)(()=>new l.AGUIError("Cannot send 'THINKING_END' event: No active thinking step found. A 'THINKING_START' event must be sent first."));default:return(0,p.of)(E)}}))};var ie=require("@ag-ui/core"),F=require("rxjs");var R=require("rxjs"),se=require("rxjs/operators");var U=(r,n)=>(0,R.defer)(()=>(0,R.from)(fetch(r,n))).pipe((0,se.switchMap)(e=>{var c;let t={type:"headers",status:e.status,headers:e.headers},s=(c=e.body)==null?void 0:c.getReader();return s?new R.Observable(g=>(g.next(t),(async()=>{try{for(;;){let{done:a,value:u}=await s.read();if(a)break;let y={type:"data",data:u};g.next(y)}g.complete()}catch(a){g.error(a)}})(),()=>{s.cancel()})):(0,R.throwError)(()=>new Error("Failed to getReader() from response"))}));var re=require("rxjs");var K=r=>{let n=new re.Subject,e=new TextDecoder("utf-8",{fatal:!1}),t="";r.subscribe({next:c=>{if(c.type!=="headers"&&c.type==="data"&&c.data){let g=e.decode(c.data,{stream:!0});t+=g;let a=t.split(/\n\n/);t=a.pop()||"";for(let u of a)s(u)}},error:c=>n.error(c),complete:()=>{t&&(t+=e.decode(),s(t)),n.complete()}});function s(c){let g=c.split(`
|
|
5
|
+
`),a=[];for(let u of g)u.startsWith("data: ")&&a.push(u.slice(6));if(a.length>0)try{let u=a.join(`
|
|
6
|
+
`),y=JSON.parse(u);n.next(y)}catch(u){n.error(u)}}return n.asObservable()};var ae=require("rxjs");var oe=P(require("@ag-ui/proto")),k=r=>{let n=new ae.Subject,e=new Uint8Array(0);r.subscribe({next:s=>{if(s.type!=="headers"&&s.type==="data"&&s.data){let c=new Uint8Array(e.length+s.data.length);c.set(e,0),c.set(s.data,e.length),e=c,t()}},error:s=>n.error(s),complete:()=>{if(e.length>0)try{t()}catch(s){console.warn("Incomplete or invalid protocol buffer data at stream end")}n.complete()}});function t(){for(;e.length>=4;){let g=4+new DataView(e.buffer,e.byteOffset,4).getUint32(0,!1);if(e.length<g)break;try{let a=e.slice(4,g),u=oe.decode(a);n.next(u),e=e.slice(g)}catch(a){let u=a instanceof Error?a.message:String(a);n.error(new Error(`Failed to decode protocol buffer message: ${u}`));return}}}return n.asObservable()};var le=P(require("@ag-ui/proto")),j=r=>{let n=new F.Subject,e=new F.ReplaySubject,t=!1;return r.subscribe({next:s=>{e.next(s),s.type==="headers"&&!t?(t=!0,s.headers.get("content-type")===le.AGUI_MEDIA_TYPE?k(e).subscribe({next:g=>n.next(g),error:g=>n.error(g),complete:()=>n.complete()}):K(e).subscribe({next:g=>{try{let a=ie.EventSchemas.parse(g);n.next(a)}catch(a){n.error(a)}},error:g=>n.error(g),complete:()=>n.complete()})):t||n.error(new Error("No headers event received before data events"))},error:s=>{e.error(s),n.error(s)},complete:()=>{e.complete()}}),n.asObservable()};var ce=require("rxjs/operators"),ue=require("fast-json-patch"),_=require("@ag-ui/core");var o=require("zod"),A=o.z.enum(["TextMessageStart","TextMessageContent","TextMessageEnd","ActionExecutionStart","ActionExecutionArgs","ActionExecutionEnd","ActionExecutionResult","AgentStateMessage","MetaEvent","RunStarted","RunFinished","RunError","NodeStarted","NodeFinished"]),ve=o.z.enum(["LangGraphInterruptEvent","PredictState","Exit"]),Ce=o.z.object({type:o.z.literal(A.enum.TextMessageStart),messageId:o.z.string(),parentMessageId:o.z.string().optional()}),Ne=o.z.object({type:o.z.literal(A.enum.TextMessageContent),messageId:o.z.string(),content:o.z.string()}),Le=o.z.object({type:o.z.literal(A.enum.TextMessageEnd),messageId:o.z.string()}),he=o.z.object({type:o.z.literal(A.enum.ActionExecutionStart),actionExecutionId:o.z.string(),actionName:o.z.string(),parentMessageId:o.z.string().optional()}),Me=o.z.object({type:o.z.literal(A.enum.ActionExecutionArgs),actionExecutionId:o.z.string(),args:o.z.string()}),Re=o.z.object({type:o.z.literal(A.enum.ActionExecutionEnd),actionExecutionId:o.z.string()}),xe=o.z.object({type:o.z.literal(A.enum.ActionExecutionResult),actionName:o.z.string(),actionExecutionId:o.z.string(),result:o.z.string()}),Ie=o.z.object({type:o.z.literal(A.enum.AgentStateMessage),threadId:o.z.string(),agentName:o.z.string(),nodeName:o.z.string(),runId:o.z.string(),active:o.z.boolean(),role:o.z.string(),state:o.z.string(),running:o.z.boolean()}),Oe=o.z.object({type:o.z.literal(A.enum.MetaEvent),name:ve,value:o.z.any()}),yt=o.z.discriminatedUnion("type",[Ce,Ne,Le,he,Me,Re,xe,Ie,Oe]),_t=o.z.object({id:o.z.string(),role:o.z.string(),content:o.z.string(),parentMessageId:o.z.string().optional()}),vt=o.z.object({id:o.z.string(),name:o.z.string(),arguments:o.z.any(),parentMessageId:o.z.string().optional()}),Ct=o.z.object({id:o.z.string(),result:o.z.any(),actionExecutionId:o.z.string(),actionName:o.z.string()});var Ee=P(require("untruncate-json"));var J=(r,n,e)=>t=>{let s={},c=!0,g=!0,a="",u=null,y=null,E=[],d=i=>{typeof i=="object"&&i!==null&&("messages"in i&&delete i.messages,s=i)};return t.pipe((0,ce.mergeMap)(i=>{switch(i.type){case _.EventType.TEXT_MESSAGE_START:{let T=i;return[{type:A.enum.TextMessageStart,messageId:T.messageId}]}case _.EventType.TEXT_MESSAGE_CONTENT:{let T=i;return[{type:A.enum.TextMessageContent,messageId:T.messageId,content:T.delta}]}case _.EventType.TEXT_MESSAGE_END:{let T=i;return[{type:A.enum.TextMessageEnd,messageId:T.messageId}]}case _.EventType.TOOL_CALL_START:{let T=i;return E.push({id:T.toolCallId,type:"function",function:{name:T.toolCallName,arguments:""}}),g=!0,[{type:A.enum.ActionExecutionStart,actionExecutionId:T.toolCallId,actionName:T.toolCallName,parentMessageId:T.parentMessageId}]}case _.EventType.TOOL_CALL_ARGS:{let T=i,f=E[E.length-1];f.function.arguments+=T.delta;let N=!1;if(y){let x=y.find(I=>I.tool==f.function.name);if(x)try{let I=JSON.parse((0,Ee.default)(f.function.arguments));x.tool_argument&&x.tool_argument in I?(d(M(C({},s),{[x.state_key]:I[x.tool_argument]})),N=!0):x.tool_argument||(d(M(C({},s),{[x.state_key]:I})),N=!0)}catch(I){}}return[{type:A.enum.ActionExecutionArgs,actionExecutionId:T.toolCallId,args:T.delta},...N?[{type:A.enum.AgentStateMessage,threadId:r,agentName:e,nodeName:a,runId:n,running:c,role:"assistant",state:JSON.stringify(s),active:g}]:[]]}case _.EventType.TOOL_CALL_END:{let T=i;return[{type:A.enum.ActionExecutionEnd,actionExecutionId:T.toolCallId}]}case _.EventType.RAW:return[];case _.EventType.CUSTOM:{let T=i;switch(T.name){case"Exit":c=!1;break;case"PredictState":y=T.value;break}return[{type:A.enum.MetaEvent,name:T.name,value:T.value}]}case _.EventType.STATE_SNAPSHOT:return d(i.snapshot),[{type:A.enum.AgentStateMessage,threadId:r,agentName:e,nodeName:a,runId:n,running:c,role:"assistant",state:JSON.stringify(s),active:g}];case _.EventType.STATE_DELTA:{let f=(0,ue.applyPatch)(s,i.delta,!0,!1);return f?(d(f.newDocument),[{type:A.enum.AgentStateMessage,threadId:r,agentName:e,nodeName:a,runId:n,running:c,role:"assistant",state:JSON.stringify(s),active:g}]):[]}case _.EventType.MESSAGES_SNAPSHOT:return u=i.messages,[{type:A.enum.AgentStateMessage,threadId:r,agentName:e,nodeName:a,runId:n,running:c,role:"assistant",state:JSON.stringify(C(C({},s),u?{messages:u}:{})),active:!0}];case _.EventType.RUN_STARTED:return[];case _.EventType.RUN_FINISHED:return u&&(s.messages=u),[{type:A.enum.AgentStateMessage,threadId:r,agentName:e,nodeName:a,runId:n,running:c,role:"assistant",state:JSON.stringify(C(C({},s),u?{messages:be(u)}:{})),active:!1}];case _.EventType.RUN_ERROR:return console.error("Run error",i),[];case _.EventType.STEP_STARTED:return a=i.stepName,E=[],y=null,[{type:A.enum.AgentStateMessage,threadId:r,agentName:e,nodeName:a,runId:n,running:c,role:"assistant",state:JSON.stringify(s),active:!0}];case _.EventType.STEP_FINISHED:return E=[],y=null,[{type:A.enum.AgentStateMessage,threadId:r,agentName:e,nodeName:a,runId:n,running:c,role:"assistant",state:JSON.stringify(s),active:!1}];default:return[]}}))};function be(r){var e;let n=[];for(let t of r)if(t.role==="assistant"||t.role==="user"||t.role==="system"){if(t.content){let s={id:t.id,role:t.role,content:t.content};n.push(s)}if(t.role==="assistant"&&t.toolCalls&&t.toolCalls.length>0)for(let s of t.toolCalls){let c={id:s.id,name:s.function.name,arguments:JSON.parse(s.function.arguments),parentMessageId:t.id};n.push(c)}}else if(t.role==="tool"){let s="unknown";for(let g of r)if(g.role==="assistant"&&((e=g.toolCalls)!=null&&e.length)){for(let a of g.toolCalls)if(a.id===t.toolCallId){s=a.function.name;break}}let c={id:t.id,result:t.content,actionExecutionId:t.toolCallId,actionName:s};n.push(c)}return n}var D=require("uuid");var b=require("rxjs/operators"),ge=require("rxjs/operators"),B=require("rxjs");var $=require("rxjs");var z=require("rxjs"),S=require("@ag-ui/core"),q=r=>n=>{let e,t,s,c=()=>{if(!e||s!=="text")throw new Error("No text message to close");let u={type:S.EventType.TEXT_MESSAGE_END,messageId:e.messageId};return s=void 0,e=void 0,r&&console.debug("[TRANSFORM]: TEXT_MESSAGE_END",JSON.stringify(u)),u},g=()=>{if(!t||s!=="tool")throw new Error("No tool call to close");let u={type:S.EventType.TOOL_CALL_END,toolCallId:t.toolCallId};return s=void 0,t=void 0,r&&console.debug("[TRANSFORM]: TOOL_CALL_END",JSON.stringify(u)),u},a=()=>s==="text"?[c()]:s==="tool"?[g()]:[];return n.pipe((0,z.mergeMap)(u=>{switch(u.type){case S.EventType.TEXT_MESSAGE_START:case S.EventType.TEXT_MESSAGE_CONTENT:case S.EventType.TEXT_MESSAGE_END:case S.EventType.TOOL_CALL_START:case S.EventType.TOOL_CALL_ARGS:case S.EventType.TOOL_CALL_END:case S.EventType.STATE_SNAPSHOT:case S.EventType.STATE_DELTA:case S.EventType.MESSAGES_SNAPSHOT:case S.EventType.CUSTOM:case S.EventType.RUN_STARTED:case S.EventType.RUN_FINISHED:case S.EventType.RUN_ERROR:case S.EventType.STEP_STARTED:case S.EventType.STEP_FINISHED:case S.EventType.THINKING_START:case S.EventType.THINKING_END:case S.EventType.THINKING_TEXT_MESSAGE_START:case S.EventType.THINKING_TEXT_MESSAGE_CONTENT:case S.EventType.THINKING_TEXT_MESSAGE_END:return[...a(),u];case S.EventType.RAW:return[u];case S.EventType.TEXT_MESSAGE_CHUNK:let E=u,d=[];if((s!=="text"||E.messageId!==void 0&&E.messageId!==(e==null?void 0:e.messageId))&&d.push(...a()),s!=="text"){if(E.messageId===void 0)throw new Error("First TEXT_MESSAGE_CHUNK must have a messageId");e={messageId:E.messageId},s="text";let f={type:S.EventType.TEXT_MESSAGE_START,messageId:E.messageId,role:"assistant"};d.push(f),r&&console.debug("[TRANSFORM]: TEXT_MESSAGE_START",JSON.stringify(f))}if(E.delta!==void 0){let f={type:S.EventType.TEXT_MESSAGE_CONTENT,messageId:e.messageId,delta:E.delta};d.push(f),r&&console.debug("[TRANSFORM]: TEXT_MESSAGE_CONTENT",JSON.stringify(f))}return d;case S.EventType.TOOL_CALL_CHUNK:let i=u,T=[];if((s!=="tool"||i.toolCallId!==void 0&&i.toolCallId!==(t==null?void 0:t.toolCallId))&&T.push(...a()),s!=="tool"){if(i.toolCallId===void 0)throw new Error("First TOOL_CALL_CHUNK must have a toolCallId");if(i.toolCallName===void 0)throw new Error("First TOOL_CALL_CHUNK must have a toolCallName");t={toolCallId:i.toolCallId,toolCallName:i.toolCallName,parentMessageId:i.parentMessageId},s="tool";let f={type:S.EventType.TOOL_CALL_START,toolCallId:i.toolCallId,toolCallName:i.toolCallName,parentMessageId:i.parentMessageId};T.push(f),r&&console.debug("[TRANSFORM]: TOOL_CALL_START",JSON.stringify(f))}if(i.delta!==void 0){let f={type:S.EventType.TOOL_CALL_ARGS,toolCallId:t.toolCallId,delta:i.delta};T.push(f),r&&console.debug("[TRANSFORM]: TOOL_CALL_ARGS",JSON.stringify(f))}return T}let y=u.type}),(0,z.finalize)(()=>a()))};var O=class{constructor({agentId:n,description:e,threadId:t,initialMessages:s,initialState:c,debug:g}={}){this.debug=!1;this.agentId=n,this.description=e!=null?e:"",this.threadId=t!=null?t:(0,D.v4)(),this.messages=v(s!=null?s:[]),this.state=v(c!=null?c:{}),this.debug=g!=null?g:!1}async runAgent(n){var s;this.agentId=(s=this.agentId)!=null?s:(0,D.v4)();let e=this.prepareRunAgentInput(n),t=(0,B.pipe)(()=>this.run(e),q(this.debug),H(this.debug),c=>this.apply(e,c),c=>this.processApplyEvents(e,c),(0,b.catchError)(c=>(this.onError(c),(0,B.throwError)(()=>c))),(0,ge.finalize)(()=>{this.onFinalize()}));return(0,$.lastValueFrom)(t((0,$.of)(null))).then(()=>{})}abortRun(){}apply(...n){return X(...n)}processApplyEvents(n,e){return e.pipe((0,b.tap)(t=>{t.messages&&(this.messages=t.messages),t.state&&(this.state=t.state)}))}prepareRunAgentInput(n){var e,t,s;return{threadId:this.threadId,runId:(n==null?void 0:n.runId)||(0,D.v4)(),tools:v((e=n==null?void 0:n.tools)!=null?e:[]),context:v((t=n==null?void 0:n.context)!=null?t:[]),forwardedProps:v((s=n==null?void 0:n.forwardedProps)!=null?s:{}),state:v(this.state),messages:v(this.messages)}}onError(n){console.error("Agent execution failed:",n)}onFinalize(){}clone(){let n=Object.create(Object.getPrototypeOf(this));for(let e of Object.getOwnPropertyNames(this)){let t=this[e];typeof t!="function"&&(n[e]=v(t))}return n}legacy_to_be_removed_runAgentBridged(n){var t;this.agentId=(t=this.agentId)!=null?t:(0,D.v4)();let e=this.prepareRunAgentInput(n);return this.run(e).pipe(q(this.debug),H(this.debug),J(this.threadId,e.runId,this.agentId),s=>s.pipe((0,b.map)(c=>(this.debug&&console.debug("[LEGACY]:",JSON.stringify(c)),c))))}};var V=class extends O{constructor(e){var t;super(e);this.abortController=new AbortController;this.url=e.url,this.headers=v((t=e.headers)!=null?t:{})}requestInit(e){return{method:"POST",headers:M(C({},this.headers),{"Content-Type":"application/json",Accept:"text/event-stream"}),body:JSON.stringify(e),signal:this.abortController.signal}}runAgent(e){var t;return this.abortController=(t=e==null?void 0:e.abortController)!=null?t:new AbortController,super.runAgent(e)}abortRun(){this.abortController.abort(),super.abortRun()}run(e){let t=U(this.url,this.requestInit(e));return j(t)}};h(L,require("@ag-ui/core"),module.exports);0&&(module.exports={AbstractAgent,HttpAgent,convertToLegacyEvents,defaultApplyEvents,parseProtoStream,parseSSEStream,runHttpRequest,transformHttpEventStream,verifyEvents,...require("@ag-ui/core")});
|
|
7
7
|
//# sourceMappingURL=index.js.map
|