@ag-ui/client 0.0.34 → 0.0.36-alpha.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 +51 -0
- package/dist/index.d.mts +1 -1
- package/dist/index.d.ts +1 -1
- 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 +7 -6
package/README.md
ADDED
|
@@ -0,0 +1,51 @@
|
|
|
1
|
+
# @ag-ui/client
|
|
2
|
+
|
|
3
|
+
Client SDK for connecting to **Agent-User Interaction (AG-UI) Protocol** servers.
|
|
4
|
+
|
|
5
|
+
`@ag-ui/client` provides agent implementations that handle the full lifecycle of AG-UI communication: connecting to servers, processing streaming events, managing state mutations, and providing reactive subscriber hooks.
|
|
6
|
+
|
|
7
|
+
## Installation
|
|
8
|
+
|
|
9
|
+
```bash
|
|
10
|
+
npm install @ag-ui/client
|
|
11
|
+
pnpm add @ag-ui/client
|
|
12
|
+
yarn add @ag-ui/client
|
|
13
|
+
```
|
|
14
|
+
|
|
15
|
+
## Features
|
|
16
|
+
|
|
17
|
+
- 🔗 **HTTP connectivity** – `HttpAgent` for direct server connections with SSE/protobuf support
|
|
18
|
+
- 🏗️ **Custom agents** – `AbstractAgent` base class for building your own transport layer
|
|
19
|
+
- 📡 **Event streaming** – Full AG-UI event processing with validation and transformation
|
|
20
|
+
- 🔄 **State management** – Automatic message/state tracking with reactive updates
|
|
21
|
+
- 🪝 **Subscriber system** – Middleware-style hooks for logging, persistence, and custom logic
|
|
22
|
+
|
|
23
|
+
## Quick example
|
|
24
|
+
|
|
25
|
+
```ts
|
|
26
|
+
import { HttpAgent } from "@ag-ui/client";
|
|
27
|
+
|
|
28
|
+
const agent = new HttpAgent({
|
|
29
|
+
url: "https://api.example.com/agent",
|
|
30
|
+
headers: { Authorization: "Bearer token" },
|
|
31
|
+
});
|
|
32
|
+
|
|
33
|
+
const result = await agent.runAgent({
|
|
34
|
+
messages: [{ role: "user", content: "Hello!" }],
|
|
35
|
+
});
|
|
36
|
+
|
|
37
|
+
console.log(result.newMessages);
|
|
38
|
+
```
|
|
39
|
+
|
|
40
|
+
## Documentation
|
|
41
|
+
|
|
42
|
+
- Concepts & architecture: [`docs/concepts`](https://docs.ag-ui.com/concepts/architecture)
|
|
43
|
+
- Full API reference: [`docs/sdk/js/client`](https://docs.ag-ui.com/sdk/js/client/overview)
|
|
44
|
+
|
|
45
|
+
## Contributing
|
|
46
|
+
|
|
47
|
+
Bug reports and pull requests are welcome! Please read our [contributing guide](https://docs.ag-ui.com/development/contributing) first.
|
|
48
|
+
|
|
49
|
+
## License
|
|
50
|
+
|
|
51
|
+
MIT © 2025 AG-UI Protocol Contributors
|
package/dist/index.d.mts
CHANGED
|
@@ -341,4 +341,4 @@ declare class HttpAgent extends AbstractAgent {
|
|
|
341
341
|
run(input: RunAgentInput): Observable<BaseEvent>;
|
|
342
342
|
}
|
|
343
343
|
|
|
344
|
-
export { AbstractAgent, type AgentConfig, HttpAgent, convertToLegacyEvents, defaultApplyEvents, parseProtoStream, parseSSEStream, runHttpRequest, transformHttpEventStream, verifyEvents };
|
|
344
|
+
export { AbstractAgent, type AgentConfig, HttpAgent, type RunAgentResult, convertToLegacyEvents, defaultApplyEvents, parseProtoStream, parseSSEStream, runHttpRequest, transformHttpEventStream, verifyEvents };
|
package/dist/index.d.ts
CHANGED
|
@@ -341,4 +341,4 @@ declare class HttpAgent extends AbstractAgent {
|
|
|
341
341
|
run(input: RunAgentInput): Observable<BaseEvent>;
|
|
342
342
|
}
|
|
343
343
|
|
|
344
|
-
export { AbstractAgent, type AgentConfig, HttpAgent, convertToLegacyEvents, defaultApplyEvents, parseProtoStream, parseSSEStream, runHttpRequest, transformHttpEventStream, verifyEvents };
|
|
344
|
+
export { AbstractAgent, type AgentConfig, HttpAgent, type RunAgentResult, convertToLegacyEvents, defaultApplyEvents, parseProtoStream, parseSSEStream, runHttpRequest, transformHttpEventStream, verifyEvents };
|
package/dist/index.js
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
|
-
"use strict";var Me=Object.create;var
|
|
1
|
+
"use strict";var Me=Object.create;var K=Object.defineProperty,ye=Object.defineProperties,Ce=Object.getOwnPropertyDescriptor,Re=Object.getOwnPropertyDescriptors,_e=Object.getOwnPropertyNames,oe=Object.getOwnPropertySymbols,Ne=Object.getPrototypeOf,ie=Object.prototype.hasOwnProperty,xe=Object.prototype.propertyIsEnumerable;var re=(i,s,t)=>s in i?K(i,s,{enumerable:!0,configurable:!0,writable:!0,value:t}):i[s]=t,x=(i,s)=>{for(var t in s||(s={}))ie.call(s,t)&&re(i,t,s[t]);if(oe)for(var t of oe(s))xe.call(s,t)&&re(i,t,s[t]);return i},k=(i,s)=>ye(i,Re(s));var Ie=(i,s)=>{for(var t in s)K(i,t,{get:s[t],enumerable:!0})},j=(i,s,t,a)=>{if(s&&typeof s=="object"||typeof s=="function")for(let e of _e(s))!ie.call(i,e)&&e!==t&&K(i,e,{get:()=>s[e],enumerable:!(a=Ce(s,e))||a.enumerable});return i},w=(i,s,t)=>(j(i,s,"default"),t&&j(t,s,"default")),$=(i,s,t)=>(t=i!=null?Me(Ne(i)):{},j(s||!i||!i.__esModule?K(t,"default",{value:i,enumerable:!0}):t,i)),Le=i=>j(K({},"__esModule",{value:!0}),i);var O={};Ie(O,{AbstractAgent:()=>F,HttpAgent:()=>ne,convertToLegacyEvents:()=>ee,defaultApplyEvents:()=>W,parseProtoStream:()=>Y,parseSSEStream:()=>b,runHttpRequest:()=>V,transformHttpEventStream:()=>Z,verifyEvents:()=>B});module.exports=Le(O);var h=require("@ag-ui/core"),G=require("rxjs/operators"),q=require("rxjs");var R=i=>{if(typeof structuredClone=="function")return structuredClone(i);try{return JSON.parse(JSON.stringify(i))}catch(s){return x({},i)}};var le=require("fast-json-patch");async function y(i,s,t,a){let e=s,o=t,c;for(let r of i)try{let g=await a(r,R(e),R(o));if(g===void 0)continue;if(g.messages!==void 0&&(e=g.messages),g.state!==void 0&&(o=g.state),c=g.stopPropagation,c===!0)break}catch(g){process.env.NODE_ENV==="test"||process.env.JEST_WORKER_ID!==void 0||console.error("Subscriber error:",g);continue}return x(x(x({},JSON.stringify(e)!==JSON.stringify(s)?{messages:e}:{}),JSON.stringify(o)!==JSON.stringify(t)?{state:o}:{}),c!==void 0?{stopPropagation:c}:{})}var ge=$(require("untruncate-json")),W=(i,s,t,a)=>{let e=R(i.messages),o=R(i.state),c={},r=u=>{u.messages!==void 0&&(e=u.messages,c.messages=u.messages),u.state!==void 0&&(o=u.state,c.state=u.state)},g=()=>{let u=R(c);return c={},u.messages!==void 0||u.state!==void 0?(0,q.of)(u):q.EMPTY};return s.pipe((0,G.concatMap)(async u=>{var d;let C=await y(a,e,o,(l,n,m)=>{var p;return(p=l.onEvent)==null?void 0:p.call(l,{event:u,agent:t,input:i,messages:n,state:m})});if(r(C),C.stopPropagation===!0)return g();switch(u.type){case h.EventType.TEXT_MESSAGE_START:{let l=await y(a,e,o,(n,m,p)=>{var E;return(E=n.onTextMessageStartEvent)==null?void 0:E.call(n,{event:u,messages:m,state:p,agent:t,input:i})});if(r(l),l.stopPropagation!==!0){let{messageId:n,role:m="assistant"}=u,p={id:n,role:m,content:""};e.push(p),r({messages:e})}return g()}case h.EventType.TEXT_MESSAGE_CONTENT:{let{messageId:l,delta:n}=u,m=e.find(E=>E.id===l);if(!m)return console.warn(`TEXT_MESSAGE_CONTENT: No message found with ID '${l}'`),g();let p=await y(a,e,o,(E,v,N)=>{var I,D;return(D=E.onTextMessageContentEvent)==null?void 0:D.call(E,{event:u,messages:v,state:N,agent:t,input:i,textMessageBuffer:(I=m.content)!=null?I:""})});return r(p),p.stopPropagation!==!0&&(m.content=(m.content||"")+n,r({messages:e})),g()}case h.EventType.TEXT_MESSAGE_END:{let{messageId:l}=u,n=e.find(p=>p.id===l);if(!n)return console.warn(`TEXT_MESSAGE_END: No message found with ID '${l}'`),g();let m=await y(a,e,o,(p,E,v)=>{var N,I;return(I=p.onTextMessageEndEvent)==null?void 0:I.call(p,{event:u,messages:E,state:v,agent:t,input:i,textMessageBuffer:(N=n.content)!=null?N:""})});return r(m),await Promise.all(a.map(p=>{var E;(E=p.onNewMessage)==null||E.call(p,{message:n,messages:e,state:o,agent:t,input:i})})),g()}case h.EventType.TOOL_CALL_START:{let l=await y(a,e,o,(n,m,p)=>{var E;return(E=n.onToolCallStartEvent)==null?void 0:E.call(n,{event:u,messages:m,state:p,agent:t,input:i})});if(r(l),l.stopPropagation!==!0){let{toolCallId:n,toolCallName:m,parentMessageId:p}=u,E;p&&e.length>0&&e[e.length-1].id===p?E=e[e.length-1]:(E={id:p||n,role:"assistant",toolCalls:[]},e.push(E)),(d=E.toolCalls)!=null||(E.toolCalls=[]),E.toolCalls.push({id:n,type:"function",function:{name:m,arguments:""}}),r({messages:e})}return g()}case h.EventType.TOOL_CALL_ARGS:{let{toolCallId:l,delta:n}=u,m=e.find(v=>{var N;return(N=v.toolCalls)==null?void 0:N.some(I=>I.id===l)});if(!m)return console.warn(`TOOL_CALL_ARGS: No message found containing tool call with ID '${l}'`),g();let p=m.toolCalls.find(v=>v.id===l);if(!p)return console.warn(`TOOL_CALL_ARGS: No tool call found with ID '${l}'`),g();let E=await y(a,e,o,(v,N,I)=>{var ae;let D=p.function.arguments,J=p.function.name,X={};try{X=(0,ge.default)(D)}catch(Be){}return(ae=v.onToolCallArgsEvent)==null?void 0:ae.call(v,{event:u,messages:N,state:I,agent:t,input:i,toolCallBuffer:D,toolCallName:J,partialToolCallArgs:X})});return r(E),E.stopPropagation!==!0&&(p.function.arguments+=n,r({messages:e})),g()}case h.EventType.TOOL_CALL_END:{let{toolCallId:l}=u,n=e.find(E=>{var v;return(v=E.toolCalls)==null?void 0:v.some(N=>N.id===l)});if(!n)return console.warn(`TOOL_CALL_END: No message found containing tool call with ID '${l}'`),g();let m=n.toolCalls.find(E=>E.id===l);if(!m)return console.warn(`TOOL_CALL_END: No tool call found with ID '${l}'`),g();let p=await y(a,e,o,(E,v,N)=>{var X;let I=m.function.arguments,D=m.function.name,J={};try{J=JSON.parse(I)}catch(ae){}return(X=E.onToolCallEndEvent)==null?void 0:X.call(E,{event:u,messages:v,state:N,agent:t,input:i,toolCallName:D,toolCallArgs:J})});return r(p),await Promise.all(a.map(E=>{var v;(v=E.onNewToolCall)==null||v.call(E,{toolCall:m,messages:e,state:o,agent:t,input:i})})),g()}case h.EventType.TOOL_CALL_RESULT:{let l=await y(a,e,o,(n,m,p)=>{var E;return(E=n.onToolCallResultEvent)==null?void 0:E.call(n,{event:u,messages:m,state:p,agent:t,input:i})});if(r(l),l.stopPropagation!==!0){let{messageId:n,toolCallId:m,content:p,role:E}=u,v={id:n,toolCallId:m,role:E||"tool",content:p};e.push(v),await Promise.all(a.map(N=>{var I;(I=N.onNewMessage)==null||I.call(N,{message:v,messages:e,state:o,agent:t,input:i})})),r({messages:e})}return g()}case h.EventType.STATE_SNAPSHOT:{let l=await y(a,e,o,(n,m,p)=>{var E;return(E=n.onStateSnapshotEvent)==null?void 0:E.call(n,{event:u,messages:m,state:p,agent:t,input:i})});if(r(l),l.stopPropagation!==!0){let{snapshot:n}=u;o=n,r({state:o})}return g()}case h.EventType.STATE_DELTA:{let l=await y(a,e,o,(n,m,p)=>{var E;return(E=n.onStateDeltaEvent)==null?void 0:E.call(n,{event:u,messages:m,state:p,agent:t,input:i})});if(r(l),l.stopPropagation!==!0){let{delta:n}=u;try{o=(0,le.applyPatch)(o,n,!0,!1).newDocument,r({state:o})}catch(m){let p=m instanceof Error?m.message:String(m);console.warn(`Failed to apply state patch:
|
|
2
2
|
Current state: ${JSON.stringify(o,null,2)}
|
|
3
|
-
Patch operations: ${JSON.stringify(
|
|
4
|
-
Error: ${
|
|
5
|
-
`),r=[];for(let g of
|
|
6
|
-
`),u=JSON.parse(g);s.next(u)}catch(g){s.error(g)}}return s.asObservable()};var ue=require("rxjs");var pe=$(require("@ag-ui/proto")),b=i=>{let s=new ue.Subject,t=new Uint8Array(0);i.subscribe({next:e=>{if(e.type!=="headers"&&e.type==="data"&&e.data){let o=new Uint8Array(t.length+e.data.length);o.set(t,0),o.set(e.data,t.length),t=o,n()}},error:e=>s.error(e),complete:()=>{if(t.length>0)try{n()}catch(e){console.warn("Incomplete or invalid protocol buffer data at stream end")}s.complete()}});function n(){for(;t.length>=4;){let l=4+new DataView(t.buffer,t.byteOffset,4).getUint32(0,!1);if(t.length<l)break;try{let r=t.slice(4,l),g=pe.decode(r);s.next(g),t=t.slice(l)}catch(r){let g=r instanceof Error?r.message:String(r);s.error(new Error(`Failed to decode protocol buffer message: ${g}`));return}}}return s.asObservable()};var Se=$(require("@ag-ui/proto")),Z=i=>{let s=new Q.Subject,t=new Q.ReplaySubject,n=!1;return i.subscribe({next:e=>{t.next(e),e.type==="headers"&&!n?(n=!0,e.headers.get("content-type")===Se.AGUI_MEDIA_TYPE?b(t).subscribe({next:l=>s.next(l),error:l=>s.error(l),complete:()=>s.complete()}):Y(t).subscribe({next:l=>{try{let r=me.EventSchemas.parse(l);s.next(r)}catch(r){s.error(r)}},error:l=>s.error(l),complete:()=>s.complete()})):n||s.error(new Error("No headers event received before data events"))},error:e=>{t.error(e),s.error(e)},complete:()=>{t.complete()}}),s.asObservable()};var de=require("rxjs/operators"),Te=require("fast-json-patch"),N=require("@ag-ui/core");var m=require("zod"),M=m.z.enum(["TextMessageStart","TextMessageContent","TextMessageEnd","ActionExecutionStart","ActionExecutionArgs","ActionExecutionEnd","ActionExecutionResult","AgentStateMessage","MetaEvent","RunStarted","RunFinished","RunError","NodeStarted","NodeFinished"]),Oe=m.z.enum(["LangGraphInterruptEvent","PredictState","Exit"]),we=m.z.object({type:m.z.literal(M.enum.TextMessageStart),messageId:m.z.string(),parentMessageId:m.z.string().optional()}),Pe=m.z.object({type:m.z.literal(M.enum.TextMessageContent),messageId:m.z.string(),content:m.z.string()}),De=m.z.object({type:m.z.literal(M.enum.TextMessageEnd),messageId:m.z.string()}),He=m.z.object({type:m.z.literal(M.enum.ActionExecutionStart),actionExecutionId:m.z.string(),actionName:m.z.string(),parentMessageId:m.z.string().optional()}),Ge=m.z.object({type:m.z.literal(M.enum.ActionExecutionArgs),actionExecutionId:m.z.string(),args:m.z.string()}),Fe=m.z.object({type:m.z.literal(M.enum.ActionExecutionEnd),actionExecutionId:m.z.string()}),Xe=m.z.object({type:m.z.literal(M.enum.ActionExecutionResult),actionName:m.z.string(),actionExecutionId:m.z.string(),result:m.z.string()}),Ue=m.z.object({type:m.z.literal(M.enum.AgentStateMessage),threadId:m.z.string(),agentName:m.z.string(),nodeName:m.z.string(),runId:m.z.string(),active:m.z.boolean(),role:m.z.string(),state:m.z.string(),running:m.z.boolean()}),Ke=m.z.object({type:m.z.literal(M.enum.MetaEvent),name:Oe,value:m.z.any()}),jt=m.z.discriminatedUnion("type",[we,Pe,De,He,Ge,Fe,Xe,Ue,Ke]),$t=m.z.object({id:m.z.string(),role:m.z.string(),content:m.z.string(),parentMessageId:m.z.string().optional()}),Wt=m.z.object({id:m.z.string(),name:m.z.string(),arguments:m.z.any(),parentMessageId:m.z.string().optional()}),qt=m.z.object({id:m.z.string(),result:m.z.any(),actionExecutionId:m.z.string(),actionName:m.z.string()});var fe=$(require("untruncate-json"));var ee=(i,s,t)=>n=>{let e={},o=!0,l=!0,r="",g=null,u=null,T=[],C={},A=p=>{typeof p=="object"&&p!==null&&("messages"in p&&delete p.messages,e=p)};return n.pipe((0,de.mergeMap)(p=>{switch(p.type){case N.EventType.TEXT_MESSAGE_START:{let a=p;return[{type:M.enum.TextMessageStart,messageId:a.messageId}]}case N.EventType.TEXT_MESSAGE_CONTENT:{let a=p;return[{type:M.enum.TextMessageContent,messageId:a.messageId,content:a.delta}]}case N.EventType.TEXT_MESSAGE_END:{let a=p;return[{type:M.enum.TextMessageEnd,messageId:a.messageId}]}case N.EventType.TOOL_CALL_START:{let a=p;return T.push({id:a.toolCallId,type:"function",function:{name:a.toolCallName,arguments:""}}),l=!0,C[a.toolCallId]=a.toolCallName,[{type:M.enum.ActionExecutionStart,actionExecutionId:a.toolCallId,actionName:a.toolCallName,parentMessageId:a.parentMessageId}]}case N.EventType.TOOL_CALL_ARGS:{let a=p,c=T[T.length-1];c.function.arguments+=a.delta;let d=!1;if(u){let E=u.find(R=>R.tool==c.function.name);if(E)try{let R=JSON.parse((0,fe.default)(c.function.arguments));E.tool_argument&&E.tool_argument in R?(A(B(x({},e),{[E.state_key]:R[E.tool_argument]})),d=!0):E.tool_argument||(A(B(x({},e),{[E.state_key]:R})),d=!0)}catch(R){}}return[{type:M.enum.ActionExecutionArgs,actionExecutionId:a.toolCallId,args:a.delta},...d?[{type:M.enum.AgentStateMessage,threadId:i,agentName:t,nodeName:r,runId:s,running:o,role:"assistant",state:JSON.stringify(e),active:l}]:[]]}case N.EventType.TOOL_CALL_END:{let a=p;return[{type:M.enum.ActionExecutionEnd,actionExecutionId:a.toolCallId}]}case N.EventType.TOOL_CALL_RESULT:{let a=p;return[{type:M.enum.ActionExecutionResult,actionExecutionId:a.toolCallId,result:a.content,actionName:C[a.toolCallId]||"unknown"}]}case N.EventType.RAW:return[];case N.EventType.CUSTOM:{let a=p;switch(a.name){case"Exit":o=!1;break;case"PredictState":u=a.value;break}return[{type:M.enum.MetaEvent,name:a.name,value:a.value}]}case N.EventType.STATE_SNAPSHOT:return A(p.snapshot),[{type:M.enum.AgentStateMessage,threadId:i,agentName:t,nodeName:r,runId:s,running:o,role:"assistant",state:JSON.stringify(e),active:l}];case N.EventType.STATE_DELTA:{let c=(0,Te.applyPatch)(e,p.delta,!0,!1);return c?(A(c.newDocument),[{type:M.enum.AgentStateMessage,threadId:i,agentName:t,nodeName:r,runId:s,running:o,role:"assistant",state:JSON.stringify(e),active:l}]):[]}case N.EventType.MESSAGES_SNAPSHOT:return g=p.messages,[{type:M.enum.AgentStateMessage,threadId:i,agentName:t,nodeName:r,runId:s,running:o,role:"assistant",state:JSON.stringify(x(x({},e),g?{messages:g}:{})),active:!0}];case N.EventType.RUN_STARTED:return[];case N.EventType.RUN_FINISHED:return g&&(e.messages=g),[{type:M.enum.AgentStateMessage,threadId:i,agentName:t,nodeName:r,runId:s,running:o,role:"assistant",state:JSON.stringify(x(x({},e),g?{messages:ke(g)}:{})),active:!1}];case N.EventType.RUN_ERROR:return console.error("Run error",p),[];case N.EventType.STEP_STARTED:return r=p.stepName,T=[],u=null,[{type:M.enum.AgentStateMessage,threadId:i,agentName:t,nodeName:r,runId:s,running:o,role:"assistant",state:JSON.stringify(e),active:!0}];case N.EventType.STEP_FINISHED:return T=[],u=null,[{type:M.enum.AgentStateMessage,threadId:i,agentName:t,nodeName:r,runId:s,running:o,role:"assistant",state:JSON.stringify(e),active:!1}];default:return[]}}))};function ke(i){var t;let s=[];for(let n of i)if(n.role==="assistant"||n.role==="user"||n.role==="system"){if(n.content){let e={id:n.id,role:n.role,content:n.content};s.push(e)}if(n.role==="assistant"&&n.toolCalls&&n.toolCalls.length>0)for(let e of n.toolCalls){let o={id:e.id,name:e.function.name,arguments:JSON.parse(e.function.arguments),parentMessageId:n.id};s.push(o)}}else if(n.role==="tool"){let e="unknown";for(let l of i)if(l.role==="assistant"&&((t=l.toolCalls)!=null&&t.length)){for(let r of l.toolCalls)if(r.id===n.toolCallId){e=r.function.name;break}}let o={id:n.id,result:n.content,actionExecutionId:n.toolCallId,actionName:e};s.push(o)}return s}var z=require("uuid");var D=require("rxjs/operators"),Ae=require("rxjs/operators"),F=require("rxjs");var ve=require("rxjs");var te=require("rxjs"),v=require("@ag-ui/core"),ae=i=>s=>{let t,n,e,o=()=>{if(!t||e!=="text")throw new Error("No text message to close");let g={type:v.EventType.TEXT_MESSAGE_END,messageId:t.messageId};return e=void 0,t=void 0,i&&console.debug("[TRANSFORM]: TEXT_MESSAGE_END",JSON.stringify(g)),g},l=()=>{if(!n||e!=="tool")throw new Error("No tool call to close");let g={type:v.EventType.TOOL_CALL_END,toolCallId:n.toolCallId};return e=void 0,n=void 0,i&&console.debug("[TRANSFORM]: TOOL_CALL_END",JSON.stringify(g)),g},r=()=>e==="text"?[o()]:e==="tool"?[l()]:[];return s.pipe((0,te.mergeMap)(g=>{switch(g.type){case v.EventType.TEXT_MESSAGE_START:case v.EventType.TEXT_MESSAGE_CONTENT:case v.EventType.TEXT_MESSAGE_END:case v.EventType.TOOL_CALL_START:case v.EventType.TOOL_CALL_ARGS:case v.EventType.TOOL_CALL_END:case v.EventType.TOOL_CALL_RESULT:case v.EventType.STATE_SNAPSHOT:case v.EventType.STATE_DELTA:case v.EventType.MESSAGES_SNAPSHOT:case v.EventType.CUSTOM:case v.EventType.RUN_STARTED:case v.EventType.RUN_FINISHED:case v.EventType.RUN_ERROR:case v.EventType.STEP_STARTED:case v.EventType.STEP_FINISHED:case v.EventType.THINKING_START:case v.EventType.THINKING_END:case v.EventType.THINKING_TEXT_MESSAGE_START:case v.EventType.THINKING_TEXT_MESSAGE_CONTENT:case v.EventType.THINKING_TEXT_MESSAGE_END:return[...r(),g];case v.EventType.RAW:return[g];case v.EventType.TEXT_MESSAGE_CHUNK:let T=g,C=[];if((e!=="text"||T.messageId!==void 0&&T.messageId!==(t==null?void 0:t.messageId))&&C.push(...r()),e!=="text"){if(T.messageId===void 0)throw new Error("First TEXT_MESSAGE_CHUNK must have a messageId");t={messageId:T.messageId},e="text";let a={type:v.EventType.TEXT_MESSAGE_START,messageId:T.messageId,role:"assistant"};C.push(a),i&&console.debug("[TRANSFORM]: TEXT_MESSAGE_START",JSON.stringify(a))}if(T.delta!==void 0){let a={type:v.EventType.TEXT_MESSAGE_CONTENT,messageId:t.messageId,delta:T.delta};C.push(a),i&&console.debug("[TRANSFORM]: TEXT_MESSAGE_CONTENT",JSON.stringify(a))}return C;case v.EventType.TOOL_CALL_CHUNK:let A=g,p=[];if((e!=="tool"||A.toolCallId!==void 0&&A.toolCallId!==(n==null?void 0:n.toolCallId))&&p.push(...r()),e!=="tool"){if(A.toolCallId===void 0)throw new Error("First TOOL_CALL_CHUNK must have a toolCallId");if(A.toolCallName===void 0)throw new Error("First TOOL_CALL_CHUNK must have a toolCallName");n={toolCallId:A.toolCallId,toolCallName:A.toolCallName,parentMessageId:A.parentMessageId},e="tool";let a={type:v.EventType.TOOL_CALL_START,toolCallId:A.toolCallId,toolCallName:A.toolCallName,parentMessageId:A.parentMessageId};p.push(a),i&&console.debug("[TRANSFORM]: TOOL_CALL_START",JSON.stringify(a))}if(A.delta!==void 0){let a={type:v.EventType.TOOL_CALL_ARGS,toolCallId:n.toolCallId,delta:A.delta};p.push(a),i&&console.debug("[TRANSFORM]: TOOL_CALL_ARGS",JSON.stringify(a))}return p}let u=g.type}),(0,te.finalize)(()=>r()))};var G=class{constructor({agentId:s,description:t,threadId:n,initialMessages:e,initialState:o,debug:l}={}){this.debug=!1;this.subscribers=[];this.agentId=s,this.description=t!=null?t:"",this.threadId=n!=null?n:(0,z.v4)(),this.messages=_(e!=null?e:[]),this.state=_(o!=null?o:{}),this.debug=l!=null?l:!1}subscribe(s){return this.subscribers.push(s),{unsubscribe:()=>{this.subscribers=this.subscribers.filter(t=>t!==s)}}}async runAgent(s,t){var g;this.agentId=(g=this.agentId)!=null?g:(0,z.v4)();let n=this.prepareRunAgentInput(s),e,o=new Set(this.messages.map(u=>u.id)),l=[{onRunFinishedEvent:u=>{e=u.result}},...this.subscribers,t!=null?t:{}];await this.onInitialize(n,l);let r=(0,F.pipe)(()=>this.run(n),ae(this.debug),J(this.debug),u=>this.apply(n,u,l),u=>this.processApplyEvents(n,u,l),(0,D.catchError)(u=>this.onError(n,u,l)),(0,Ae.finalize)(()=>{this.onFinalize(n,l)}));return(0,ve.lastValueFrom)(r((0,F.of)(null))).then(()=>{let u=_(this.messages).filter(T=>!o.has(T.id));return{result:e,newMessages:u}})}abortRun(){}apply(s,t,n){return q(s,t,this,n)}processApplyEvents(s,t,n){return t.pipe((0,D.tap)(e=>{e.messages&&(this.messages=e.messages,n.forEach(o=>{var l;(l=o.onMessagesChanged)==null||l.call(o,{messages:this.messages,state:this.state,agent:this,input:s})})),e.state&&(this.state=e.state,n.forEach(o=>{var l;(l=o.onStateChanged)==null||l.call(o,{state:this.state,messages:this.messages,agent:this,input:s})}))}))}prepareRunAgentInput(s){var t,n,e;return{threadId:this.threadId,runId:(s==null?void 0:s.runId)||(0,z.v4)(),tools:_((t=s==null?void 0:s.tools)!=null?t:[]),context:_((n=s==null?void 0:s.context)!=null?n:[]),forwardedProps:_((e=s==null?void 0:s.forwardedProps)!=null?e:{}),state:_(this.state),messages:_(this.messages)}}async onInitialize(s,t){let n=await y(t,this.messages,this.state,(e,o,l)=>{var r;return(r=e.onRunInitialized)==null?void 0:r.call(e,{messages:o,state:l,agent:this,input:s})});(n.messages!==void 0||n.state!==void 0)&&(n.messages&&(this.messages=n.messages,s.messages=n.messages,t.forEach(e=>{var o;(o=e.onMessagesChanged)==null||o.call(e,{messages:this.messages,state:this.state,agent:this,input:s})})),n.state&&(this.state=n.state,s.state=n.state,t.forEach(e=>{var o;(o=e.onStateChanged)==null||o.call(e,{state:this.state,messages:this.messages,agent:this,input:s})})))}onError(s,t,n){return(0,F.from)(y(n,this.messages,this.state,(e,o,l)=>{var r;return(r=e.onRunFailed)==null?void 0:r.call(e,{error:t,messages:o,state:l,agent:this,input:s})})).pipe((0,D.map)(e=>{let o=e;if((o.messages!==void 0||o.state!==void 0)&&(o.messages!==void 0&&(this.messages=o.messages,n.forEach(l=>{var r;(r=l.onMessagesChanged)==null||r.call(l,{messages:this.messages,state:this.state,agent:this,input:s})})),o.state!==void 0&&(this.state=o.state,n.forEach(l=>{var r;(r=l.onStateChanged)==null||r.call(l,{state:this.state,messages:this.messages,agent:this,input:s})}))),o.stopPropagation!==!0)throw console.error("Agent execution failed:",t),t;return{}}))}async onFinalize(s,t){let n=await y(t,this.messages,this.state,(e,o,l)=>{var r;return(r=e.onRunFinalized)==null?void 0:r.call(e,{messages:o,state:l,agent:this,input:s})});(n.messages!==void 0||n.state!==void 0)&&(n.messages!==void 0&&(this.messages=n.messages,t.forEach(e=>{var o;(o=e.onMessagesChanged)==null||o.call(e,{messages:this.messages,state:this.state,agent:this,input:s})})),n.state!==void 0&&(this.state=n.state,t.forEach(e=>{var o;(o=e.onStateChanged)==null||o.call(e,{state:this.state,messages:this.messages,agent:this,input:s})})))}clone(){let s=Object.create(Object.getPrototypeOf(this));for(let t of Object.getOwnPropertyNames(this)){let n=this[t];typeof n!="function"&&(s[t]=_(n))}return s}addMessage(s){this.messages.push(s),(async()=>{var t,n,e;for(let o of this.subscribers)await((t=o.onNewMessage)==null?void 0:t.call(o,{message:s,messages:this.messages,state:this.state,agent:this}));if(s.role==="assistant"&&s.toolCalls)for(let o of s.toolCalls)for(let l of this.subscribers)await((n=l.onNewToolCall)==null?void 0:n.call(l,{toolCall:o,messages:this.messages,state:this.state,agent:this}));for(let o of this.subscribers)await((e=o.onMessagesChanged)==null?void 0:e.call(o,{messages:this.messages,state:this.state,agent:this}))})()}addMessages(s){this.messages.push(...s),(async()=>{var t,n,e;for(let o of s){for(let l of this.subscribers)await((t=l.onNewMessage)==null?void 0:t.call(l,{message:o,messages:this.messages,state:this.state,agent:this}));if(o.role==="assistant"&&o.toolCalls)for(let l of o.toolCalls)for(let r of this.subscribers)await((n=r.onNewToolCall)==null?void 0:n.call(r,{toolCall:l,messages:this.messages,state:this.state,agent:this}))}for(let o of this.subscribers)await((e=o.onMessagesChanged)==null?void 0:e.call(o,{messages:this.messages,state:this.state,agent:this}))})()}setMessages(s){this.messages=_(s),(async()=>{var t;for(let n of this.subscribers)await((t=n.onMessagesChanged)==null?void 0:t.call(n,{messages:this.messages,state:this.state,agent:this}))})()}setState(s){this.state=_(s),(async()=>{var t;for(let n of this.subscribers)await((t=n.onStateChanged)==null?void 0:t.call(n,{messages:this.messages,state:this.state,agent:this}))})()}legacy_to_be_removed_runAgentBridged(s){var n;this.agentId=(n=this.agentId)!=null?n:(0,z.v4)();let t=this.prepareRunAgentInput(s);return this.run(t).pipe(ae(this.debug),J(this.debug),ee(this.threadId,t.runId,this.agentId),e=>e.pipe((0,D.map)(o=>(this.debug&&console.debug("[LEGACY]:",JSON.stringify(o)),o))))}};var ne=class extends G{constructor(t){var n;super(t);this.abortController=new AbortController;this.url=t.url,this.headers=_((n=t.headers)!=null?n:{})}requestInit(t){return{method:"POST",headers:B(x({},this.headers),{"Content-Type":"application/json",Accept:"text/event-stream"}),body:JSON.stringify(t),signal:this.abortController.signal}}runAgent(t,n){var e;return this.abortController=(e=t==null?void 0:t.abortController)!=null?e:new AbortController,super.runAgent(t,n)}abortRun(){this.abortController.abort(),super.abortRun()}run(t){let n=V(this.url,this.requestInit(t));return Z(n)}};I(L,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(n,null,2)}
|
|
4
|
+
Error: ${p}`)}}return g()}case h.EventType.MESSAGES_SNAPSHOT:{let l=await y(a,e,o,(n,m,p)=>{var E;return(E=n.onMessagesSnapshotEvent)==null?void 0:E.call(n,{event:u,messages:m,state:p,agent:t,input:i})});if(r(l),l.stopPropagation!==!0){let{messages:n}=u;e=n,r({messages:e})}return g()}case h.EventType.RAW:{let l=await y(a,e,o,(n,m,p)=>{var E;return(E=n.onRawEvent)==null?void 0:E.call(n,{event:u,messages:m,state:p,agent:t,input:i})});return r(l),g()}case h.EventType.CUSTOM:{let l=await y(a,e,o,(n,m,p)=>{var E;return(E=n.onCustomEvent)==null?void 0:E.call(n,{event:u,messages:m,state:p,agent:t,input:i})});return r(l),g()}case h.EventType.RUN_STARTED:{let l=await y(a,e,o,(n,m,p)=>{var E;return(E=n.onRunStartedEvent)==null?void 0:E.call(n,{event:u,messages:m,state:p,agent:t,input:i})});return r(l),g()}case h.EventType.RUN_FINISHED:{let l=await y(a,e,o,(n,m,p)=>{var E;return(E=n.onRunFinishedEvent)==null?void 0:E.call(n,{event:u,messages:m,state:p,agent:t,input:i,result:u.result})});return r(l),g()}case h.EventType.RUN_ERROR:{let l=await y(a,e,o,(n,m,p)=>{var E;return(E=n.onRunErrorEvent)==null?void 0:E.call(n,{event:u,messages:m,state:p,agent:t,input:i})});return r(l),g()}case h.EventType.STEP_STARTED:{let l=await y(a,e,o,(n,m,p)=>{var E;return(E=n.onStepStartedEvent)==null?void 0:E.call(n,{event:u,messages:m,state:p,agent:t,input:i})});return r(l),g()}case h.EventType.STEP_FINISHED:{let l=await y(a,e,o,(n,m,p)=>{var E;return(E=n.onStepFinishedEvent)==null?void 0:E.call(n,{event:u,messages:m,state:p,agent:t,input:i})});return r(l),g()}case h.EventType.TEXT_MESSAGE_CHUNK:throw new Error("TEXT_MESSAGE_CHUNK must be tranformed before being applied");case h.EventType.TOOL_CALL_CHUNK:throw new Error("TOOL_CALL_CHUNK must be tranformed before being applied");case h.EventType.THINKING_START:return g();case h.EventType.THINKING_END:return g();case h.EventType.THINKING_TEXT_MESSAGE_START:return g();case h.EventType.THINKING_TEXT_MESSAGE_CONTENT:return g();case h.EventType.THINKING_TEXT_MESSAGE_END:return g()}let L=u.type;return g()}),(0,G.mergeAll)(),a.length>0?(0,G.defaultIfEmpty)({}):u=>u)};var f=require("@ag-ui/core"),T=require("rxjs"),ce=require("rxjs/operators"),B=i=>s=>{let t=new Map,a=new Map,e=!1,o=!1,c=!1,r=new Map,g=!1,u=!1,C=!1,L=()=>{t.clear(),a.clear(),r.clear(),g=!1,u=!1,e=!1,o=!1,C=!0};return s.pipe((0,ce.mergeMap)(d=>{let l=d.type;if(i&&console.debug("[VERIFY]:",JSON.stringify(d)),o)return(0,T.throwError)(()=>new f.AGUIError(`Cannot send event type '${l}': The run has already errored with 'RUN_ERROR'. No further events can be sent.`));if(e&&l!==f.EventType.RUN_ERROR&&l!==f.EventType.RUN_STARTED)return(0,T.throwError)(()=>new f.AGUIError(`Cannot send event type '${l}': The run has already finished with 'RUN_FINISHED'. Start a new run with 'RUN_STARTED'.`));if(c){if(l===f.EventType.RUN_STARTED){if(C&&!e)return(0,T.throwError)(()=>new f.AGUIError("Cannot send 'RUN_STARTED' while a run is still active. The previous run must be finished with 'RUN_FINISHED' before starting a new run."));e&&L()}}else if(c=!0,l!==f.EventType.RUN_STARTED&&l!==f.EventType.RUN_ERROR)return(0,T.throwError)(()=>new f.AGUIError("First event must be 'RUN_STARTED'"));switch(l){case f.EventType.TEXT_MESSAGE_START:{let n=d.messageId;return t.has(n)?(0,T.throwError)(()=>new f.AGUIError(`Cannot send 'TEXT_MESSAGE_START' event: A text message with ID '${n}' is already in progress. Complete it with 'TEXT_MESSAGE_END' first.`)):(t.set(n,!0),(0,T.of)(d))}case f.EventType.TEXT_MESSAGE_CONTENT:{let n=d.messageId;return t.has(n)?(0,T.of)(d):(0,T.throwError)(()=>new f.AGUIError(`Cannot send 'TEXT_MESSAGE_CONTENT' event: No active text message found with ID '${n}'. Start a text message with 'TEXT_MESSAGE_START' first.`))}case f.EventType.TEXT_MESSAGE_END:{let n=d.messageId;return t.has(n)?(t.delete(n),(0,T.of)(d)):(0,T.throwError)(()=>new f.AGUIError(`Cannot send 'TEXT_MESSAGE_END' event: No active text message found with ID '${n}'. A 'TEXT_MESSAGE_START' event must be sent first.`))}case f.EventType.TOOL_CALL_START:{let n=d.toolCallId;return a.has(n)?(0,T.throwError)(()=>new f.AGUIError(`Cannot send 'TOOL_CALL_START' event: A tool call with ID '${n}' is already in progress. Complete it with 'TOOL_CALL_END' first.`)):(a.set(n,!0),(0,T.of)(d))}case f.EventType.TOOL_CALL_ARGS:{let n=d.toolCallId;return a.has(n)?(0,T.of)(d):(0,T.throwError)(()=>new f.AGUIError(`Cannot send 'TOOL_CALL_ARGS' event: No active tool call found with ID '${n}'. Start a tool call with 'TOOL_CALL_START' first.`))}case f.EventType.TOOL_CALL_END:{let n=d.toolCallId;return a.has(n)?(a.delete(n),(0,T.of)(d)):(0,T.throwError)(()=>new f.AGUIError(`Cannot send 'TOOL_CALL_END' event: No active tool call found with ID '${n}'. A 'TOOL_CALL_START' event must be sent first.`))}case f.EventType.STEP_STARTED:{let n=d.stepName;return r.has(n)?(0,T.throwError)(()=>new f.AGUIError(`Step "${n}" is already active for 'STEP_STARTED'`)):(r.set(n,!0),(0,T.of)(d))}case f.EventType.STEP_FINISHED:{let n=d.stepName;return r.has(n)?(r.delete(n),(0,T.of)(d)):(0,T.throwError)(()=>new f.AGUIError(`Cannot send 'STEP_FINISHED' for step "${n}" that was not started`))}case f.EventType.RUN_STARTED:return C=!0,(0,T.of)(d);case f.EventType.RUN_FINISHED:{if(r.size>0){let n=Array.from(r.keys()).join(", ");return(0,T.throwError)(()=>new f.AGUIError(`Cannot send 'RUN_FINISHED' while steps are still active: ${n}`))}if(t.size>0){let n=Array.from(t.keys()).join(", ");return(0,T.throwError)(()=>new f.AGUIError(`Cannot send 'RUN_FINISHED' while text messages are still active: ${n}`))}if(a.size>0){let n=Array.from(a.keys()).join(", ");return(0,T.throwError)(()=>new f.AGUIError(`Cannot send 'RUN_FINISHED' while tool calls are still active: ${n}`))}return e=!0,(0,T.of)(d)}case f.EventType.RUN_ERROR:return o=!0,(0,T.of)(d);case f.EventType.CUSTOM:return(0,T.of)(d);case f.EventType.THINKING_TEXT_MESSAGE_START:return g?u?(0,T.throwError)(()=>new f.AGUIError("Cannot send 'THINKING_TEXT_MESSAGE_START' event: A thinking message is already in progress. Complete it with 'THINKING_TEXT_MESSAGE_END' first.")):(u=!0,(0,T.of)(d)):(0,T.throwError)(()=>new f.AGUIError("Cannot send 'THINKING_TEXT_MESSAGE_START' event: A thinking step is not in progress. Create one with 'THINKING_START' first."));case f.EventType.THINKING_TEXT_MESSAGE_CONTENT:return u?(0,T.of)(d):(0,T.throwError)(()=>new f.AGUIError("Cannot send 'THINKING_TEXT_MESSAGE_CONTENT' event: No active thinking message found. Start a message with 'THINKING_TEXT_MESSAGE_START' first."));case f.EventType.THINKING_TEXT_MESSAGE_END:return u?(u=!1,(0,T.of)(d)):(0,T.throwError)(()=>new f.AGUIError("Cannot send 'THINKING_TEXT_MESSAGE_END' event: No active thinking message found. A 'THINKING_TEXT_MESSAGE_START' event must be sent first."));case f.EventType.THINKING_START:return g?(0,T.throwError)(()=>new f.AGUIError("Cannot send 'THINKING_START' event: A thinking step is already in progress. End it with 'THINKING_END' first.")):(g=!0,(0,T.of)(d));case f.EventType.THINKING_END:return g?(g=!1,(0,T.of)(d)):(0,T.throwError)(()=>new f.AGUIError("Cannot send 'THINKING_END' event: No active thinking step found. A 'THINKING_START' event must be sent first."));default:return(0,T.of)(d)}}))};var Se=require("@ag-ui/core"),Q=require("rxjs");var P=require("rxjs"),Ee=require("rxjs/operators");var V=(i,s)=>(0,P.defer)(()=>(0,P.from)(fetch(i,s))).pipe((0,Ee.switchMap)(t=>{var o;let a={type:"headers",status:t.status,headers:t.headers},e=(o=t.body)==null?void 0:o.getReader();return e?new P.Observable(c=>(c.next(a),(async()=>{try{for(;;){let{done:r,value:g}=await e.read();if(r)break;let u={type:"data",data:g};c.next(u)}c.complete()}catch(r){c.error(r)}})(),()=>{e.cancel()})):(0,P.throwError)(()=>new Error("Failed to getReader() from response"))}));var ue=require("rxjs");var b=i=>{let s=new ue.Subject,t=new TextDecoder("utf-8",{fatal:!1}),a="";i.subscribe({next:o=>{if(o.type!=="headers"&&o.type==="data"&&o.data){let c=t.decode(o.data,{stream:!0});a+=c;let r=a.split(/\n\n/);a=r.pop()||"";for(let g of r)e(g)}},error:o=>s.error(o),complete:()=>{a&&(a+=t.decode(),e(a)),s.complete()}});function e(o){let c=o.split(`
|
|
5
|
+
`),r=[];for(let g of c)g.startsWith("data: ")&&r.push(g.slice(6));if(r.length>0)try{let g=r.join(`
|
|
6
|
+
`),u=JSON.parse(g);s.next(u)}catch(g){s.error(g)}}return s.asObservable()};var pe=require("rxjs");var me=$(require("@ag-ui/proto")),Y=i=>{let s=new pe.Subject,t=new Uint8Array(0);i.subscribe({next:e=>{if(e.type!=="headers"&&e.type==="data"&&e.data){let o=new Uint8Array(t.length+e.data.length);o.set(t,0),o.set(e.data,t.length),t=o,a()}},error:e=>s.error(e),complete:()=>{if(t.length>0)try{a()}catch(e){console.warn("Incomplete or invalid protocol buffer data at stream end")}s.complete()}});function a(){for(;t.length>=4;){let c=4+new DataView(t.buffer,t.byteOffset,4).getUint32(0,!1);if(t.length<c)break;try{let r=t.slice(4,c),g=me.decode(r);s.next(g),t=t.slice(c)}catch(r){let g=r instanceof Error?r.message:String(r);s.error(new Error(`Failed to decode protocol buffer message: ${g}`));return}}}return s.asObservable()};var de=$(require("@ag-ui/proto")),Z=i=>{let s=new Q.Subject,t=new Q.ReplaySubject,a=!1;return i.subscribe({next:e=>{t.next(e),e.type==="headers"&&!a?(a=!0,e.headers.get("content-type")===de.AGUI_MEDIA_TYPE?Y(t).subscribe({next:c=>s.next(c),error:c=>s.error(c),complete:()=>s.complete()}):b(t).subscribe({next:c=>{try{let r=Se.EventSchemas.parse(c);s.next(r)}catch(r){s.error(r)}},error:c=>s.error(c),complete:()=>s.complete()})):a||s.error(new Error("No headers event received before data events"))},error:e=>{t.error(e),s.error(e)},complete:()=>{t.complete()}}),s.asObservable()};var fe=require("rxjs/operators"),Te=require("fast-json-patch"),_=require("@ag-ui/core");var S=require("zod"),M=S.z.enum(["TextMessageStart","TextMessageContent","TextMessageEnd","ActionExecutionStart","ActionExecutionArgs","ActionExecutionEnd","ActionExecutionResult","AgentStateMessage","MetaEvent","RunStarted","RunFinished","RunError","NodeStarted","NodeFinished"]),Oe=S.z.enum(["LangGraphInterruptEvent","PredictState","Exit"]),we=S.z.object({type:S.z.literal(M.enum.TextMessageStart),messageId:S.z.string(),parentMessageId:S.z.string().optional()}),Pe=S.z.object({type:S.z.literal(M.enum.TextMessageContent),messageId:S.z.string(),content:S.z.string()}),De=S.z.object({type:S.z.literal(M.enum.TextMessageEnd),messageId:S.z.string()}),He=S.z.object({type:S.z.literal(M.enum.ActionExecutionStart),actionExecutionId:S.z.string(),actionName:S.z.string(),parentMessageId:S.z.string().optional()}),Ge=S.z.object({type:S.z.literal(M.enum.ActionExecutionArgs),actionExecutionId:S.z.string(),args:S.z.string()}),Fe=S.z.object({type:S.z.literal(M.enum.ActionExecutionEnd),actionExecutionId:S.z.string()}),Ue=S.z.object({type:S.z.literal(M.enum.ActionExecutionResult),actionName:S.z.string(),actionExecutionId:S.z.string(),result:S.z.string()}),Xe=S.z.object({type:S.z.literal(M.enum.AgentStateMessage),threadId:S.z.string(),agentName:S.z.string(),nodeName:S.z.string(),runId:S.z.string(),active:S.z.boolean(),role:S.z.string(),state:S.z.string(),running:S.z.boolean()}),Ke=S.z.object({type:S.z.literal(M.enum.MetaEvent),name:Oe,value:S.z.any()}),Vt=S.z.discriminatedUnion("type",[we,Pe,De,He,Ge,Fe,Ue,Xe,Ke]),bt=S.z.object({id:S.z.string(),role:S.z.string(),content:S.z.string(),parentMessageId:S.z.string().optional()}),Yt=S.z.object({id:S.z.string(),name:S.z.string(),arguments:S.z.any(),parentMessageId:S.z.string().optional()}),Qt=S.z.object({id:S.z.string(),result:S.z.any(),actionExecutionId:S.z.string(),actionName:S.z.string()});var Ae=$(require("untruncate-json"));var ee=(i,s,t)=>a=>{let e={},o=!0,c=!0,r="",g=null,u=null,C=[],L={},d=l=>{typeof l=="object"&&l!==null&&("messages"in l&&delete l.messages,e=l)};return a.pipe((0,fe.mergeMap)(l=>{switch(l.type){case _.EventType.TEXT_MESSAGE_START:{let n=l;return[{type:M.enum.TextMessageStart,messageId:n.messageId}]}case _.EventType.TEXT_MESSAGE_CONTENT:{let n=l;return[{type:M.enum.TextMessageContent,messageId:n.messageId,content:n.delta}]}case _.EventType.TEXT_MESSAGE_END:{let n=l;return[{type:M.enum.TextMessageEnd,messageId:n.messageId}]}case _.EventType.TOOL_CALL_START:{let n=l;return C.push({id:n.toolCallId,type:"function",function:{name:n.toolCallName,arguments:""}}),c=!0,L[n.toolCallId]=n.toolCallName,[{type:M.enum.ActionExecutionStart,actionExecutionId:n.toolCallId,actionName:n.toolCallName,parentMessageId:n.parentMessageId}]}case _.EventType.TOOL_CALL_ARGS:{let n=l,m=C.find(E=>E.id===n.toolCallId);if(!m)return console.warn(`TOOL_CALL_ARGS: No tool call found with ID '${n.toolCallId}'`),[];m.function.arguments+=n.delta;let p=!1;if(u){let E=u.find(v=>v.tool==m.function.name);if(E)try{let v=JSON.parse((0,Ae.default)(m.function.arguments));E.tool_argument&&E.tool_argument in v?(d(k(x({},e),{[E.state_key]:v[E.tool_argument]})),p=!0):E.tool_argument||(d(k(x({},e),{[E.state_key]:v})),p=!0)}catch(v){}}return[{type:M.enum.ActionExecutionArgs,actionExecutionId:n.toolCallId,args:n.delta},...p?[{type:M.enum.AgentStateMessage,threadId:i,agentName:t,nodeName:r,runId:s,running:o,role:"assistant",state:JSON.stringify(e),active:c}]:[]]}case _.EventType.TOOL_CALL_END:{let n=l;return[{type:M.enum.ActionExecutionEnd,actionExecutionId:n.toolCallId}]}case _.EventType.TOOL_CALL_RESULT:{let n=l;return[{type:M.enum.ActionExecutionResult,actionExecutionId:n.toolCallId,result:n.content,actionName:L[n.toolCallId]||"unknown"}]}case _.EventType.RAW:return[];case _.EventType.CUSTOM:{let n=l;switch(n.name){case"Exit":o=!1;break;case"PredictState":u=n.value;break}return[{type:M.enum.MetaEvent,name:n.name,value:n.value}]}case _.EventType.STATE_SNAPSHOT:return d(l.snapshot),[{type:M.enum.AgentStateMessage,threadId:i,agentName:t,nodeName:r,runId:s,running:o,role:"assistant",state:JSON.stringify(e),active:c}];case _.EventType.STATE_DELTA:{let m=(0,Te.applyPatch)(e,l.delta,!0,!1);return m?(d(m.newDocument),[{type:M.enum.AgentStateMessage,threadId:i,agentName:t,nodeName:r,runId:s,running:o,role:"assistant",state:JSON.stringify(e),active:c}]):[]}case _.EventType.MESSAGES_SNAPSHOT:return g=l.messages,[{type:M.enum.AgentStateMessage,threadId:i,agentName:t,nodeName:r,runId:s,running:o,role:"assistant",state:JSON.stringify(x(x({},e),g?{messages:g}:{})),active:!0}];case _.EventType.RUN_STARTED:return[];case _.EventType.RUN_FINISHED:return g&&(e.messages=g),[{type:M.enum.AgentStateMessage,threadId:i,agentName:t,nodeName:r,runId:s,running:o,role:"assistant",state:JSON.stringify(x(x({},e),g?{messages:ke(g)}:{})),active:!1}];case _.EventType.RUN_ERROR:return console.error("Run error",l),[];case _.EventType.STEP_STARTED:return r=l.stepName,C=[],u=null,[{type:M.enum.AgentStateMessage,threadId:i,agentName:t,nodeName:r,runId:s,running:o,role:"assistant",state:JSON.stringify(e),active:!0}];case _.EventType.STEP_FINISHED:return C=[],u=null,[{type:M.enum.AgentStateMessage,threadId:i,agentName:t,nodeName:r,runId:s,running:o,role:"assistant",state:JSON.stringify(e),active:!1}];default:return[]}}))};function ke(i){var t;let s=[];for(let a of i)if(a.role==="assistant"||a.role==="user"||a.role==="system"){if(a.content){let e={id:a.id,role:a.role,content:a.content};s.push(e)}if(a.role==="assistant"&&a.toolCalls&&a.toolCalls.length>0)for(let e of a.toolCalls){let o={id:e.id,name:e.function.name,arguments:JSON.parse(e.function.arguments),parentMessageId:a.id};s.push(o)}}else if(a.role==="tool"){let e="unknown";for(let c of i)if(c.role==="assistant"&&((t=c.toolCalls)!=null&&t.length)){for(let r of c.toolCalls)if(r.id===a.toolCallId){e=r.function.name;break}}let o={id:a.id,result:a.content,actionExecutionId:a.toolCallId,actionName:e};s.push(o)}return s}var z=require("uuid");var H=require("rxjs/operators"),ve=require("rxjs/operators"),U=require("rxjs");var he=require("rxjs");var te=require("rxjs"),A=require("@ag-ui/core"),se=i=>s=>{let t,a,e,o=()=>{if(!t||e!=="text")throw new Error("No text message to close");let g={type:A.EventType.TEXT_MESSAGE_END,messageId:t.messageId};return e=void 0,t=void 0,i&&console.debug("[TRANSFORM]: TEXT_MESSAGE_END",JSON.stringify(g)),g},c=()=>{if(!a||e!=="tool")throw new Error("No tool call to close");let g={type:A.EventType.TOOL_CALL_END,toolCallId:a.toolCallId};return e=void 0,a=void 0,i&&console.debug("[TRANSFORM]: TOOL_CALL_END",JSON.stringify(g)),g},r=()=>e==="text"?[o()]:e==="tool"?[c()]:[];return s.pipe((0,te.mergeMap)(g=>{switch(g.type){case A.EventType.TEXT_MESSAGE_START:case A.EventType.TEXT_MESSAGE_CONTENT:case A.EventType.TEXT_MESSAGE_END:case A.EventType.TOOL_CALL_START:case A.EventType.TOOL_CALL_ARGS:case A.EventType.TOOL_CALL_END:case A.EventType.TOOL_CALL_RESULT:case A.EventType.STATE_SNAPSHOT:case A.EventType.STATE_DELTA:case A.EventType.MESSAGES_SNAPSHOT:case A.EventType.CUSTOM:case A.EventType.RUN_STARTED:case A.EventType.RUN_FINISHED:case A.EventType.RUN_ERROR:case A.EventType.STEP_STARTED:case A.EventType.STEP_FINISHED:case A.EventType.THINKING_START:case A.EventType.THINKING_END:case A.EventType.THINKING_TEXT_MESSAGE_START:case A.EventType.THINKING_TEXT_MESSAGE_CONTENT:case A.EventType.THINKING_TEXT_MESSAGE_END:return[...r(),g];case A.EventType.RAW:return[g];case A.EventType.TEXT_MESSAGE_CHUNK:let C=g,L=[];if((e!=="text"||C.messageId!==void 0&&C.messageId!==(t==null?void 0:t.messageId))&&L.push(...r()),e!=="text"){if(C.messageId===void 0)throw new Error("First TEXT_MESSAGE_CHUNK must have a messageId");t={messageId:C.messageId},e="text";let n={type:A.EventType.TEXT_MESSAGE_START,messageId:C.messageId,role:C.role||"assistant"};L.push(n),i&&console.debug("[TRANSFORM]: TEXT_MESSAGE_START",JSON.stringify(n))}if(C.delta!==void 0){let n={type:A.EventType.TEXT_MESSAGE_CONTENT,messageId:t.messageId,delta:C.delta};L.push(n),i&&console.debug("[TRANSFORM]: TEXT_MESSAGE_CONTENT",JSON.stringify(n))}return L;case A.EventType.TOOL_CALL_CHUNK:let d=g,l=[];if((e!=="tool"||d.toolCallId!==void 0&&d.toolCallId!==(a==null?void 0:a.toolCallId))&&l.push(...r()),e!=="tool"){if(d.toolCallId===void 0)throw new Error("First TOOL_CALL_CHUNK must have a toolCallId");if(d.toolCallName===void 0)throw new Error("First TOOL_CALL_CHUNK must have a toolCallName");a={toolCallId:d.toolCallId,toolCallName:d.toolCallName,parentMessageId:d.parentMessageId},e="tool";let n={type:A.EventType.TOOL_CALL_START,toolCallId:d.toolCallId,toolCallName:d.toolCallName,parentMessageId:d.parentMessageId};l.push(n),i&&console.debug("[TRANSFORM]: TOOL_CALL_START",JSON.stringify(n))}if(d.delta!==void 0){let n={type:A.EventType.TOOL_CALL_ARGS,toolCallId:a.toolCallId,delta:d.delta};l.push(n),i&&console.debug("[TRANSFORM]: TOOL_CALL_ARGS",JSON.stringify(n))}return l}let u=g.type}),(0,te.finalize)(()=>r()))};var F=class{constructor({agentId:s,description:t,threadId:a,initialMessages:e,initialState:o,debug:c}={}){this.debug=!1;this.subscribers=[];this.agentId=s,this.description=t!=null?t:"",this.threadId=a!=null?a:(0,z.v4)(),this.messages=R(e!=null?e:[]),this.state=R(o!=null?o:{}),this.debug=c!=null?c:!1}subscribe(s){return this.subscribers.push(s),{unsubscribe:()=>{this.subscribers=this.subscribers.filter(t=>t!==s)}}}async runAgent(s,t){var g;this.agentId=(g=this.agentId)!=null?g:(0,z.v4)();let a=this.prepareRunAgentInput(s),e,o=new Set(this.messages.map(u=>u.id)),c=[{onRunFinishedEvent:u=>{e=u.result}},...this.subscribers,t!=null?t:{}];await this.onInitialize(a,c);let r=(0,U.pipe)(()=>this.run(a),se(this.debug),B(this.debug),u=>this.apply(a,u,c),u=>this.processApplyEvents(a,u,c),(0,H.catchError)(u=>this.onError(a,u,c)),(0,ve.finalize)(()=>{this.onFinalize(a,c)}));return(0,he.lastValueFrom)(r((0,U.of)(null))).then(()=>{let u=R(this.messages).filter(C=>!o.has(C.id));return{result:e,newMessages:u}})}abortRun(){}apply(s,t,a){return W(s,t,this,a)}processApplyEvents(s,t,a){return t.pipe((0,H.tap)(e=>{e.messages&&(this.messages=e.messages,a.forEach(o=>{var c;(c=o.onMessagesChanged)==null||c.call(o,{messages:this.messages,state:this.state,agent:this,input:s})})),e.state&&(this.state=e.state,a.forEach(o=>{var c;(c=o.onStateChanged)==null||c.call(o,{state:this.state,messages:this.messages,agent:this,input:s})}))}))}prepareRunAgentInput(s){var t,a,e;return{threadId:this.threadId,runId:(s==null?void 0:s.runId)||(0,z.v4)(),tools:R((t=s==null?void 0:s.tools)!=null?t:[]),context:R((a=s==null?void 0:s.context)!=null?a:[]),forwardedProps:R((e=s==null?void 0:s.forwardedProps)!=null?e:{}),state:R(this.state),messages:R(this.messages)}}async onInitialize(s,t){let a=await y(t,this.messages,this.state,(e,o,c)=>{var r;return(r=e.onRunInitialized)==null?void 0:r.call(e,{messages:o,state:c,agent:this,input:s})});(a.messages!==void 0||a.state!==void 0)&&(a.messages&&(this.messages=a.messages,s.messages=a.messages,t.forEach(e=>{var o;(o=e.onMessagesChanged)==null||o.call(e,{messages:this.messages,state:this.state,agent:this,input:s})})),a.state&&(this.state=a.state,s.state=a.state,t.forEach(e=>{var o;(o=e.onStateChanged)==null||o.call(e,{state:this.state,messages:this.messages,agent:this,input:s})})))}onError(s,t,a){return(0,U.from)(y(a,this.messages,this.state,(e,o,c)=>{var r;return(r=e.onRunFailed)==null?void 0:r.call(e,{error:t,messages:o,state:c,agent:this,input:s})})).pipe((0,H.map)(e=>{let o=e;if((o.messages!==void 0||o.state!==void 0)&&(o.messages!==void 0&&(this.messages=o.messages,a.forEach(c=>{var r;(r=c.onMessagesChanged)==null||r.call(c,{messages:this.messages,state:this.state,agent:this,input:s})})),o.state!==void 0&&(this.state=o.state,a.forEach(c=>{var r;(r=c.onStateChanged)==null||r.call(c,{state:this.state,messages:this.messages,agent:this,input:s})}))),o.stopPropagation!==!0)throw console.error("Agent execution failed:",t),t;return{}}))}async onFinalize(s,t){let a=await y(t,this.messages,this.state,(e,o,c)=>{var r;return(r=e.onRunFinalized)==null?void 0:r.call(e,{messages:o,state:c,agent:this,input:s})});(a.messages!==void 0||a.state!==void 0)&&(a.messages!==void 0&&(this.messages=a.messages,t.forEach(e=>{var o;(o=e.onMessagesChanged)==null||o.call(e,{messages:this.messages,state:this.state,agent:this,input:s})})),a.state!==void 0&&(this.state=a.state,t.forEach(e=>{var o;(o=e.onStateChanged)==null||o.call(e,{state:this.state,messages:this.messages,agent:this,input:s})})))}clone(){let s=Object.create(Object.getPrototypeOf(this));for(let t of Object.getOwnPropertyNames(this)){let a=this[t];typeof a!="function"&&(s[t]=R(a))}return s}addMessage(s){this.messages.push(s),(async()=>{var t,a,e;for(let o of this.subscribers)await((t=o.onNewMessage)==null?void 0:t.call(o,{message:s,messages:this.messages,state:this.state,agent:this}));if(s.role==="assistant"&&s.toolCalls)for(let o of s.toolCalls)for(let c of this.subscribers)await((a=c.onNewToolCall)==null?void 0:a.call(c,{toolCall:o,messages:this.messages,state:this.state,agent:this}));for(let o of this.subscribers)await((e=o.onMessagesChanged)==null?void 0:e.call(o,{messages:this.messages,state:this.state,agent:this}))})()}addMessages(s){this.messages.push(...s),(async()=>{var t,a,e;for(let o of s){for(let c of this.subscribers)await((t=c.onNewMessage)==null?void 0:t.call(c,{message:o,messages:this.messages,state:this.state,agent:this}));if(o.role==="assistant"&&o.toolCalls)for(let c of o.toolCalls)for(let r of this.subscribers)await((a=r.onNewToolCall)==null?void 0:a.call(r,{toolCall:c,messages:this.messages,state:this.state,agent:this}))}for(let o of this.subscribers)await((e=o.onMessagesChanged)==null?void 0:e.call(o,{messages:this.messages,state:this.state,agent:this}))})()}setMessages(s){this.messages=R(s),(async()=>{var t;for(let a of this.subscribers)await((t=a.onMessagesChanged)==null?void 0:t.call(a,{messages:this.messages,state:this.state,agent:this}))})()}setState(s){this.state=R(s),(async()=>{var t;for(let a of this.subscribers)await((t=a.onStateChanged)==null?void 0:t.call(a,{messages:this.messages,state:this.state,agent:this}))})()}legacy_to_be_removed_runAgentBridged(s){var a;this.agentId=(a=this.agentId)!=null?a:(0,z.v4)();let t=this.prepareRunAgentInput(s);return this.run(t).pipe(se(this.debug),B(this.debug),ee(this.threadId,t.runId,this.agentId),e=>e.pipe((0,H.map)(o=>(this.debug&&console.debug("[LEGACY]:",JSON.stringify(o)),o))))}};var ne=class extends F{constructor(t){var a;super(t);this.abortController=new AbortController;this.url=t.url,this.headers=R((a=t.headers)!=null?a:{})}requestInit(t){return{method:"POST",headers:k(x({},this.headers),{"Content-Type":"application/json",Accept:"text/event-stream"}),body:JSON.stringify(t),signal:this.abortController.signal}}runAgent(t,a){var e;return this.abortController=(e=t==null?void 0:t.abortController)!=null?e:new AbortController,super.runAgent(t,a)}abortRun(){this.abortController.abort(),super.abortRun()}run(t){let a=V(this.url,this.requestInit(t));return Z(a)}};w(O,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
|