@ag-ui/langgraph 0.0.3 → 0.0.4-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/dist/index.d.mts +19 -5
- package/dist/index.d.ts +19 -5
- package/dist/index.js +4 -4
- package/dist/index.js.map +1 -1
- package/dist/index.mjs +3 -3
- package/dist/index.mjs.map +1 -1
- package/package.json +1 -1
package/dist/index.d.mts
CHANGED
|
@@ -1,7 +1,8 @@
|
|
|
1
|
+
import * as _langchain_langgraph_sdk from '@langchain/langgraph-sdk';
|
|
2
|
+
import { Message, Client, Config, Assistant, StreamMode, Thread } from '@langchain/langgraph-sdk';
|
|
1
3
|
import { Subscriber, Observable } from 'rxjs';
|
|
2
|
-
import { Message, Client, Config, Assistant, StreamMode, ThreadState } from '@langchain/langgraph-sdk';
|
|
3
4
|
import { MessageType } from '@langchain/core/messages';
|
|
4
|
-
import { TextMessageStartEvent, TextMessageContentEvent, TextMessageEndEvent, ToolCallStartEvent, ToolCallArgsEvent, ToolCallEndEvent, StateSnapshotEvent, StateDeltaEvent, MessagesSnapshotEvent, RawEvent, CustomEvent, RunStartedEvent, RunFinishedEvent, RunErrorEvent, StepStartedEvent, StepFinishedEvent, AgentConfig, AbstractAgent, RunAgentInput } from '@ag-ui/client';
|
|
5
|
+
import { TextMessageStartEvent, TextMessageContentEvent, TextMessageEndEvent, ThinkingTextMessageStartEvent, ThinkingTextMessageContentEvent, ThinkingTextMessageEndEvent, ToolCallStartEvent, ToolCallArgsEvent, ToolCallEndEvent, ThinkingStartEvent, ThinkingEndEvent, StateSnapshotEvent, StateDeltaEvent, MessagesSnapshotEvent, RawEvent, CustomEvent, RunStartedEvent, RunFinishedEvent, RunErrorEvent, StepStartedEvent, StepFinishedEvent, AgentConfig, AbstractAgent, RunAgentInput } from '@ag-ui/client';
|
|
5
6
|
import { RunsStreamPayload } from '@langchain/langgraph-sdk/dist/types';
|
|
6
7
|
|
|
7
8
|
declare enum LangGraphEventTypes {
|
|
@@ -27,6 +28,10 @@ type MessageInProgress = {
|
|
|
27
28
|
toolCallId?: string | null;
|
|
28
29
|
toolCallName?: string | null;
|
|
29
30
|
};
|
|
31
|
+
type ThinkingInProgress = {
|
|
32
|
+
index: number;
|
|
33
|
+
type?: LangGraphReasoning['type'];
|
|
34
|
+
};
|
|
30
35
|
interface RunMetadata {
|
|
31
36
|
id: string;
|
|
32
37
|
schemaKeys?: SchemaKeys;
|
|
@@ -67,8 +72,13 @@ interface PredictStateTool {
|
|
|
67
72
|
state_key: string;
|
|
68
73
|
tool_argument: string;
|
|
69
74
|
}
|
|
75
|
+
interface LangGraphReasoning {
|
|
76
|
+
type: 'text';
|
|
77
|
+
text: string;
|
|
78
|
+
index: number;
|
|
79
|
+
}
|
|
70
80
|
|
|
71
|
-
type ProcessedEvents = TextMessageStartEvent | TextMessageContentEvent | TextMessageEndEvent | ToolCallStartEvent | ToolCallArgsEvent | ToolCallEndEvent | StateSnapshotEvent | StateDeltaEvent | MessagesSnapshotEvent | RawEvent | CustomEvent | RunStartedEvent | RunFinishedEvent | RunErrorEvent | StepStartedEvent | StepFinishedEvent;
|
|
81
|
+
type ProcessedEvents = TextMessageStartEvent | TextMessageContentEvent | TextMessageEndEvent | ThinkingTextMessageStartEvent | ThinkingTextMessageContentEvent | ThinkingTextMessageEndEvent | ToolCallStartEvent | ToolCallArgsEvent | ToolCallEndEvent | ThinkingStartEvent | ThinkingEndEvent | StateSnapshotEvent | StateDeltaEvent | MessagesSnapshotEvent | RawEvent | CustomEvent | RunStartedEvent | RunFinishedEvent | RunErrorEvent | StepStartedEvent | StepFinishedEvent;
|
|
72
82
|
type RunAgentExtendedInput<TStreamMode extends StreamMode | StreamMode[] = StreamMode, TSubgraphs extends boolean = false> = Omit<RunAgentInput, "forwardedProps"> & {
|
|
73
83
|
forwardedProps?: Omit<RunsStreamPayload<TStreamMode, TSubgraphs>, "input"> & {
|
|
74
84
|
nodeName?: string;
|
|
@@ -90,6 +100,7 @@ declare class LangGraphAgent extends AbstractAgent {
|
|
|
90
100
|
graphId: string;
|
|
91
101
|
assistant?: Assistant;
|
|
92
102
|
messagesInProcess: MessagesInProgressRecord;
|
|
103
|
+
thinkingProcess: null | ThinkingInProgress;
|
|
93
104
|
activeRun?: RunMetadata;
|
|
94
105
|
subscriber: Subscriber<ProcessedEvents>;
|
|
95
106
|
constructor(config: LangGraphAgentConfig);
|
|
@@ -97,8 +108,11 @@ declare class LangGraphAgent extends AbstractAgent {
|
|
|
97
108
|
run(input: RunAgentInput): Observable<ProcessedEvents>;
|
|
98
109
|
handleStreamEvents(input: RunAgentExtendedInput, subscriber: Subscriber<ProcessedEvents>): Promise<void>;
|
|
99
110
|
handleSingleEvent(event: any, state: State): void;
|
|
111
|
+
handleThinkingEvent(reasoningData: LangGraphReasoning): void;
|
|
100
112
|
getStateSnapshot(state: State): State;
|
|
101
|
-
|
|
113
|
+
getOrCreateThread(threadId: string): Promise<Thread>;
|
|
114
|
+
getThread(threadId: string): Promise<Thread<_langchain_langgraph_sdk.DefaultValues>>;
|
|
115
|
+
createThread(payload?: Parameters<typeof this$1.client.threads.create>[0]): Promise<Thread<_langchain_langgraph_sdk.DefaultValues>>;
|
|
102
116
|
mergeConfigs({ configs, assistant, schemaKeys, }: {
|
|
103
117
|
configs: Config[];
|
|
104
118
|
assistant: Assistant;
|
|
@@ -111,4 +125,4 @@ declare class LangGraphAgent extends AbstractAgent {
|
|
|
111
125
|
langGraphDefaultMergeState(state: State, messages: Message[], tools: any): State;
|
|
112
126
|
}
|
|
113
127
|
|
|
114
|
-
export { CustomEventNames, LangGraphAgent, type LangGraphAgentConfig, LangGraphEventTypes, type LangGraphPlatformMessage, type MessageInProgress, type MessagesInProgressRecord, type PredictStateTool, type ProcessedEvents, type RunMetadata, type SchemaKeys, type State, type ToolCall };
|
|
128
|
+
export { CustomEventNames, LangGraphAgent, type LangGraphAgentConfig, LangGraphEventTypes, type LangGraphPlatformMessage, type LangGraphReasoning, type MessageInProgress, type MessagesInProgressRecord, type PredictStateTool, type ProcessedEvents, type RunMetadata, type SchemaKeys, type State, type ThinkingInProgress, type ToolCall };
|
package/dist/index.d.ts
CHANGED
|
@@ -1,7 +1,8 @@
|
|
|
1
|
+
import * as _langchain_langgraph_sdk from '@langchain/langgraph-sdk';
|
|
2
|
+
import { Message, Client, Config, Assistant, StreamMode, Thread } from '@langchain/langgraph-sdk';
|
|
1
3
|
import { Subscriber, Observable } from 'rxjs';
|
|
2
|
-
import { Message, Client, Config, Assistant, StreamMode, ThreadState } from '@langchain/langgraph-sdk';
|
|
3
4
|
import { MessageType } from '@langchain/core/messages';
|
|
4
|
-
import { TextMessageStartEvent, TextMessageContentEvent, TextMessageEndEvent, ToolCallStartEvent, ToolCallArgsEvent, ToolCallEndEvent, StateSnapshotEvent, StateDeltaEvent, MessagesSnapshotEvent, RawEvent, CustomEvent, RunStartedEvent, RunFinishedEvent, RunErrorEvent, StepStartedEvent, StepFinishedEvent, AgentConfig, AbstractAgent, RunAgentInput } from '@ag-ui/client';
|
|
5
|
+
import { TextMessageStartEvent, TextMessageContentEvent, TextMessageEndEvent, ThinkingTextMessageStartEvent, ThinkingTextMessageContentEvent, ThinkingTextMessageEndEvent, ToolCallStartEvent, ToolCallArgsEvent, ToolCallEndEvent, ThinkingStartEvent, ThinkingEndEvent, StateSnapshotEvent, StateDeltaEvent, MessagesSnapshotEvent, RawEvent, CustomEvent, RunStartedEvent, RunFinishedEvent, RunErrorEvent, StepStartedEvent, StepFinishedEvent, AgentConfig, AbstractAgent, RunAgentInput } from '@ag-ui/client';
|
|
5
6
|
import { RunsStreamPayload } from '@langchain/langgraph-sdk/dist/types';
|
|
6
7
|
|
|
7
8
|
declare enum LangGraphEventTypes {
|
|
@@ -27,6 +28,10 @@ type MessageInProgress = {
|
|
|
27
28
|
toolCallId?: string | null;
|
|
28
29
|
toolCallName?: string | null;
|
|
29
30
|
};
|
|
31
|
+
type ThinkingInProgress = {
|
|
32
|
+
index: number;
|
|
33
|
+
type?: LangGraphReasoning['type'];
|
|
34
|
+
};
|
|
30
35
|
interface RunMetadata {
|
|
31
36
|
id: string;
|
|
32
37
|
schemaKeys?: SchemaKeys;
|
|
@@ -67,8 +72,13 @@ interface PredictStateTool {
|
|
|
67
72
|
state_key: string;
|
|
68
73
|
tool_argument: string;
|
|
69
74
|
}
|
|
75
|
+
interface LangGraphReasoning {
|
|
76
|
+
type: 'text';
|
|
77
|
+
text: string;
|
|
78
|
+
index: number;
|
|
79
|
+
}
|
|
70
80
|
|
|
71
|
-
type ProcessedEvents = TextMessageStartEvent | TextMessageContentEvent | TextMessageEndEvent | ToolCallStartEvent | ToolCallArgsEvent | ToolCallEndEvent | StateSnapshotEvent | StateDeltaEvent | MessagesSnapshotEvent | RawEvent | CustomEvent | RunStartedEvent | RunFinishedEvent | RunErrorEvent | StepStartedEvent | StepFinishedEvent;
|
|
81
|
+
type ProcessedEvents = TextMessageStartEvent | TextMessageContentEvent | TextMessageEndEvent | ThinkingTextMessageStartEvent | ThinkingTextMessageContentEvent | ThinkingTextMessageEndEvent | ToolCallStartEvent | ToolCallArgsEvent | ToolCallEndEvent | ThinkingStartEvent | ThinkingEndEvent | StateSnapshotEvent | StateDeltaEvent | MessagesSnapshotEvent | RawEvent | CustomEvent | RunStartedEvent | RunFinishedEvent | RunErrorEvent | StepStartedEvent | StepFinishedEvent;
|
|
72
82
|
type RunAgentExtendedInput<TStreamMode extends StreamMode | StreamMode[] = StreamMode, TSubgraphs extends boolean = false> = Omit<RunAgentInput, "forwardedProps"> & {
|
|
73
83
|
forwardedProps?: Omit<RunsStreamPayload<TStreamMode, TSubgraphs>, "input"> & {
|
|
74
84
|
nodeName?: string;
|
|
@@ -90,6 +100,7 @@ declare class LangGraphAgent extends AbstractAgent {
|
|
|
90
100
|
graphId: string;
|
|
91
101
|
assistant?: Assistant;
|
|
92
102
|
messagesInProcess: MessagesInProgressRecord;
|
|
103
|
+
thinkingProcess: null | ThinkingInProgress;
|
|
93
104
|
activeRun?: RunMetadata;
|
|
94
105
|
subscriber: Subscriber<ProcessedEvents>;
|
|
95
106
|
constructor(config: LangGraphAgentConfig);
|
|
@@ -97,8 +108,11 @@ declare class LangGraphAgent extends AbstractAgent {
|
|
|
97
108
|
run(input: RunAgentInput): Observable<ProcessedEvents>;
|
|
98
109
|
handleStreamEvents(input: RunAgentExtendedInput, subscriber: Subscriber<ProcessedEvents>): Promise<void>;
|
|
99
110
|
handleSingleEvent(event: any, state: State): void;
|
|
111
|
+
handleThinkingEvent(reasoningData: LangGraphReasoning): void;
|
|
100
112
|
getStateSnapshot(state: State): State;
|
|
101
|
-
|
|
113
|
+
getOrCreateThread(threadId: string): Promise<Thread>;
|
|
114
|
+
getThread(threadId: string): Promise<Thread<_langchain_langgraph_sdk.DefaultValues>>;
|
|
115
|
+
createThread(payload?: Parameters<typeof this$1.client.threads.create>[0]): Promise<Thread<_langchain_langgraph_sdk.DefaultValues>>;
|
|
102
116
|
mergeConfigs({ configs, assistant, schemaKeys, }: {
|
|
103
117
|
configs: Config[];
|
|
104
118
|
assistant: Assistant;
|
|
@@ -111,4 +125,4 @@ declare class LangGraphAgent extends AbstractAgent {
|
|
|
111
125
|
langGraphDefaultMergeState(state: State, messages: Message[], tools: any): State;
|
|
112
126
|
}
|
|
113
127
|
|
|
114
|
-
export { CustomEventNames, LangGraphAgent, type LangGraphAgentConfig, LangGraphEventTypes, type LangGraphPlatformMessage, type MessageInProgress, type MessagesInProgressRecord, type PredictStateTool, type ProcessedEvents, type RunMetadata, type SchemaKeys, type State, type ToolCall };
|
|
128
|
+
export { CustomEventNames, LangGraphAgent, type LangGraphAgentConfig, LangGraphEventTypes, type LangGraphPlatformMessage, type LangGraphReasoning, type MessageInProgress, type MessagesInProgressRecord, type PredictStateTool, type ProcessedEvents, type RunMetadata, type SchemaKeys, type State, type ThinkingInProgress, type ToolCall };
|
package/dist/index.js
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
|
-
"use strict";var N=Object.defineProperty,gt=Object.defineProperties,ut=Object.getOwnPropertyDescriptor,pt=Object.getOwnPropertyDescriptors,mt=Object.getOwnPropertyNames,q=Object.getOwnPropertySymbols;var Z=Object.prototype.hasOwnProperty,Et=Object.prototype.propertyIsEnumerable;var z=(n,t)=>(t=Symbol[n])?t:Symbol.for("Symbol."+n);var Q=(n,t,e)=>t in n?N(n,t,{enumerable:!0,configurable:!0,writable:!0,value:e}):n[t]=e,u=(n,t)=>{for(var e in t||(t={}))Z.call(t,e)&&Q(n,e,t[e]);if(q)for(var e of q(t))Et.call(t,e)&&Q(n,e,t[e]);return n},_=(n,t)=>gt(n,pt(t));var St=(n,t)=>{for(var e in t)N(n,e,{get:t[e],enumerable:!0})},ft=(n,t,e,s)=>{if(t&&typeof t=="object"||typeof t=="function")for(let a of mt(t))!Z.call(n,a)&&a!==e&&N(n,a,{get:()=>t[a],enumerable:!(s=ut(t,a))||s.enumerable});return n};var vt=n=>ft(N({},"__esModule",{value:!0}),n);var tt=(n,t,e)=>(t=n[z("asyncIterator")])?t.call(n):(n=n[z("iterator")](),t={},e=(s,a)=>(a=n[s])&&(t[s]=r=>new Promise((g,c,l)=>(r=a.call(n,r),l=r.done,Promise.resolve(r.value).then(p=>g({value:p,done:l}),c)))),e("next"),e("return"),t);var _t={};St(_t,{CustomEventNames:()=>at,LangGraphAgent:()=>k,LangGraphEventTypes:()=>et});module.exports=vt(_t);var ot=require("rxjs"),rt=require("@langchain/langgraph-sdk"),lt=require("crypto"),dt=require("@langchain/core/messages");var et=(i=>(i.OnChainStart="on_chain_start",i.OnChainStream="on_chain_stream",i.OnChainEnd="on_chain_end",i.OnChatModelStart="on_chat_model_start",i.OnChatModelStream="on_chat_model_stream",i.OnChatModelEnd="on_chat_model_end",i.OnToolStart="on_tool_start",i.OnToolEnd="on_tool_end",i.OnCustomEvent="on_custom_event",i.OnInterrupt="on_interrupt",i))(et||{}),at=(a=>(a.ManuallyEmitMessage="manually_emit_message",a.ManuallyEmitToolCall="manually_emit_tool_call",a.ManuallyEmitState="manually_emit_state",a.Exit="exit",a))(at||{});var o=require("@ag-ui/client");var O=["tools"];function P(n,t){return Object.fromEntries(Object.entries(n).filter(([e])=>t.includes(e)))}function st({mode:n,state:t,schemaKeys:e}){let s=n==="start"?t:null;return s&&(e!=null&&e.input)&&(s=P(s,[...O,...e.input])),s}function nt(n){return n.map(t=>{var e;switch(t.type){case"human":return{id:t.id,role:"user",content:A(t.content)};case"ai":return{id:t.id,role:"assistant",content:A(t.content),toolCalls:(e=t.tool_calls)==null?void 0:e.map(s=>({id:s.id,type:"function",function:{name:s.name,arguments:JSON.stringify(s.args)}}))};case"system":return{id:t.id,role:"system",content:A(t.content)};case"tool":return{id:t.id,role:"tool",content:A(t.content),toolCallId:t.tool_call_id};default:throw new Error("message type returned from LangGraph is not supported.")}})}function it(n){return n.map((t,e)=>{var s,a;switch(t.role){case"user":return{id:t.id,role:t.role,content:t.content,type:"human"};case"assistant":return{id:t.id,type:"ai",role:t.role,content:(s=t.content)!=null?s:"",tool_calls:((a=t.toolCalls)!=null?a:[]).map(r=>({id:r.id,name:r.function.name,args:JSON.parse(r.function.arguments),type:"tool_call"}))};case"system":return{id:t.id,role:t.role,content:t.content,type:"system"};case"tool":return{content:t.content,role:t.role,type:t.role,tool_call_id:t.toolCallId,id:t.id};default:throw console.error(`Message role ${t.role} is not implemented`),new Error("message role is not supported.")}})}function A(n){return typeof n=="string"?n:JSON.stringify(n)}var k=class extends o.AbstractAgent{constructor(t){var e,s;super(t),this.messagesInProcess={},this.agentName=t.agentName,this.graphId=t.graphId,this.assistantConfig=t.assistantConfig,this.client=(s=t==null?void 0:t.client)!=null?s:new rt.Client({apiUrl:t.deploymentUrl,apiKey:t.langsmithApiKey,defaultHeaders:u({},(e=t.propertyHeaders)!=null?e:{})})}dispatchEvent(t){return this.subscriber.next(t),!0}run(t){return this.activeRun={id:t.runId,threadId:t.threadId},new ot.Observable(e=>(this.handleStreamEvents(t,e),()=>{}))}async handleStreamEvents(t,e){var D,U,H,J,K,F,j,B,X,$,Y,V;let{threadId:s,state:a,messages:r,tools:g,context:c,forwardedProps:l}=t;this.subscriber=e;let p=!1;this.activeRun.manuallyEmittedState=null,this.activeRun.nodeName=(D=t.forwardedProps)==null?void 0:D.nodeName;let i=s!=null?s:(0,lt.randomUUID)();this.assistant||(this.assistant=await this.getAssistant());let d=await this.getOrCreateThreadAndReturnState(i),f=d.values,h=it(r);a.messages=f.messages,a=this.langGraphDefaultMergeState(a,h,g);let R=i&&this.activeRun.nodeName!="__end__"&&this.activeRun.nodeName?"continue":"start";R==="continue"&&!((U=l==null?void 0:l.command)!=null&&U.resume)&&await this.client.threads.updateState(i,{values:a,asNode:this.activeRun.nodeName}),this.activeRun.schemaKeys=await this.getSchemaKeys();let w=st({mode:R,state:a,schemaKeys:this.activeRun.schemaKeys}),x=await this.client.assistants.getGraph(this.assistant.assistant_id),I,T=[this.assistantConfig,l==null?void 0:l.config].filter(Boolean);T.length&&(I=await this.mergeConfigs({configs:T,assistant:this.assistant,schemaKeys:this.activeRun.schemaKeys}));let C=_(u({},l),{streamMode:(H=l==null?void 0:l.streamMode)!=null?H:["events","values","updates"],input:w,config:I}),y=(F=(K=(J=d.tasks)==null?void 0:J[0])==null?void 0:K.interrupts)!=null?F:[];if(y!=null&&y.length&&!((j=l==null?void 0:l.command)!=null&&j.resume))return this.dispatchEvent({type:o.EventType.RUN_STARTED,threadId:i,runId:t.runId}),y.forEach(S=>{this.dispatchEvent({type:o.EventType.CUSTOM,name:"on_interrupt",value:typeof S.value=="string"?S.value:JSON.stringify(S.value),rawEvent:S})}),this.dispatchEvent({type:o.EventType.RUN_FINISHED,threadId:i,runId:t.runId}),e.complete();let E=this.client.runs.stream(i,this.assistant.assistant_id,C);this.activeRun.prevNodeName=null;let G={},L=a;try{this.dispatchEvent({type:o.EventType.RUN_STARTED,threadId:i,runId:this.activeRun.id});try{for(var ee=tt(E),ae,se,ne;ae=!(se=await ee.next()).done;ae=!1){let m=se.value;if(!C.streamMode.includes(m.event))continue;let b=m;if(m.event==="error"){this.dispatchEvent({type:o.EventType.RUN_ERROR,message:m.data.message,rawEvent:m});break}if(m.event==="updates")continue;if(m.event==="values"){G=b.data;continue}let M=b.data,v=M.metadata.langgraph_node,W=M.event,ht=M.metadata;if(this.activeRun.id=ht.run_id,v&&v!==this.activeRun.nodeName&&(this.activeRun.nodeName&&this.dispatchEvent({type:o.EventType.STEP_FINISHED,stepName:this.activeRun.nodeName}),v&&(this.dispatchEvent({type:o.EventType.STEP_STARTED,stepName:v}),this.activeRun.nodeName=v)),p=p||W==="on_custom_event"&&M.name==="exit",this.activeRun.exitingNode=this.activeRun.nodeName===v&&W==="on_chain_end",this.activeRun.exitingNode&&(this.activeRun.manuallyEmittedState=null),x.nodes.some(ct=>ct.id===v)&&(this.activeRun.nodeName=v),L=(B=this.activeRun.manuallyEmittedState)!=null?B:G,!this.activeRun.nodeName)continue;(JSON.stringify(L)!==JSON.stringify(a)||this.activeRun.prevNodeName!=this.activeRun.nodeName||this.activeRun.exitingNode)&&!this.getMessageInProgress(this.activeRun.id)&&(a=L,this.activeRun.prevNodeName=this.activeRun.nodeName,this.dispatchEvent({type:o.EventType.STATE_SNAPSHOT,snapshot:this.getStateSnapshot(a),rawEvent:b})),this.dispatchEvent({type:o.EventType.RAW,event:M}),this.handleSingleEvent(M,a)}}catch(se){ne=[se]}finally{try{ae&&(se=ee.return)&&await se.call(ee)}finally{if(ne)throw ne[0]}}a=await this.client.threads.getState(i);let S=(Y=($=(X=a.tasks)==null?void 0:X[0])==null?void 0:$.interrupts)!=null?Y:[];return this.activeRun.nodeName=S?this.activeRun.nodeName:Object.keys(a.metadata.writes)[0],S.forEach(m=>{this.dispatchEvent({type:o.EventType.CUSTOM,name:"on_interrupt",value:typeof m.value=="string"?m.value:JSON.stringify(m.value),rawEvent:m})}),this.dispatchEvent({type:o.EventType.STATE_SNAPSHOT,snapshot:this.getStateSnapshot(a.values)}),this.dispatchEvent({type:o.EventType.MESSAGES_SNAPSHOT,messages:nt((V=a.values.messages)!=null?V:[])}),this.activeRun.nodeName&&this.dispatchEvent({type:o.EventType.STEP_FINISHED,stepName:this.activeRun.nodeName}),this.dispatchEvent({type:o.EventType.RUN_FINISHED,threadId:i,runId:this.activeRun.id}),e.complete()}catch(S){return e.error(S)}}handleSingleEvent(t,e){var s,a,r,g,c,l;switch(t.event){case"on_chat_model_stream":let p=(s=t.metadata["emit-messages"])!=null?s:!0,i=(a=t.metadata["emit-tool-calls"])!=null?a:!0;if(t.data.chunk.response_metadata.finish_reason)return;let d=this.getMessageInProgress(this.activeRun.id),f=!!(d!=null&&d.id),h=(r=t.data.chunk.tool_call_chunks)==null?void 0:r[0],R=(g=t.metadata.predict_state)==null?void 0:g.some(E=>E.tool===(h==null?void 0:h.name)),w=!f&&(h==null?void 0:h.name),x=f&&(d==null?void 0:d.toolCallId)&&h.args,I=f&&(d==null?void 0:d.toolCallId)&&!h,T=!f&&!h,C=f&&!h,y=f&&!(d!=null&&d.toolCallId)&&!C;if(R&&this.dispatchEvent({type:o.EventType.CUSTOM,name:"PredictState",value:t.metadata.predict_state}),I){this.dispatchEvent({type:o.EventType.TOOL_CALL_END,toolCallId:d.toolCallId,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if(y){this.dispatchEvent({type:o.EventType.TEXT_MESSAGE_END,messageId:d.id,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if(w&&i){this.dispatchEvent({type:o.EventType.TOOL_CALL_START,toolCallId:h.id,toolCallName:h.name,parentMessageId:t.data.chunk.id,rawEvent:t})&&this.setMessageInProgress(this.activeRun.id,{id:t.data.chunk.id,toolCallId:h.id,toolCallName:h.name});break}if(x&&i){this.dispatchEvent({type:o.EventType.TOOL_CALL_ARGS,toolCallId:d.toolCallId,delta:h.args,rawEvent:t});break}if(T&&p){this.dispatchEvent({type:o.EventType.TEXT_MESSAGE_START,role:"assistant",messageId:t.data.chunk.id,rawEvent:t})&&this.setMessageInProgress(this.activeRun.id,{id:t.data.chunk.id,toolCallId:null,toolCallName:null});break}if(C&&p){this.dispatchEvent({type:o.EventType.TEXT_MESSAGE_CONTENT,messageId:d.id,delta:t.data.chunk.content,rawEvent:t});break}break;case"on_chat_model_end":if((c=this.getMessageInProgress(this.activeRun.id))!=null&&c.toolCallId){this.dispatchEvent({type:o.EventType.TOOL_CALL_END,toolCallId:this.getMessageInProgress(this.activeRun.id).toolCallId,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if((l=this.getMessageInProgress(this.activeRun.id))!=null&&l.id){this.dispatchEvent({type:o.EventType.TEXT_MESSAGE_END,messageId:this.getMessageInProgress(this.activeRun.id).id,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}break;case"on_custom_event":if(t.name==="manually_emit_message"){this.dispatchEvent({type:o.EventType.TEXT_MESSAGE_START,role:"assistant",messageId:t.data.message_id,rawEvent:t}),this.dispatchEvent({type:o.EventType.TEXT_MESSAGE_CONTENT,messageId:t.data.message_id,delta:t.data.message,rawEvent:t}),this.dispatchEvent({type:o.EventType.TEXT_MESSAGE_END,messageId:t.data.message_id,rawEvent:t});break}if(t.name==="manually_emit_tool_call"){this.dispatchEvent({type:o.EventType.TOOL_CALL_START,toolCallId:t.data.id,toolCallName:t.data.name,parentMessageId:t.data.id,rawEvent:t}),this.dispatchEvent({type:o.EventType.TOOL_CALL_ARGS,toolCallId:t.data.id,delta:t.data.args,rawEvent:t}),this.dispatchEvent({type:o.EventType.TOOL_CALL_END,toolCallId:t.data.id,rawEvent:t});break}t.name==="manually_emit_state"&&(this.activeRun.manuallyEmittedState=t.data,this.dispatchEvent({type:o.EventType.STATE_SNAPSHOT,snapshot:this.getStateSnapshot(e),rawEvent:t})),this.dispatchEvent({type:o.EventType.CUSTOM,name:t.name,value:t.data,rawEvent:t});break}}getStateSnapshot(t){let e=this.activeRun.schemaKeys;return e!=null&&e.output&&(t=P(t,[...O,...e.output])),t}async getOrCreateThreadAndReturnState(t){let e={values:{}};try{await this.client.threads.get(t),e=await this.client.threads.getState(t)}catch(s){await this.client.threads.create({threadId:t})}return e}async mergeConfigs({configs:t,assistant:e,schemaKeys:s}){return t.reduce((a,r)=>{var d;let g=a.configurable;r.configurable&&(g=s!=null&&s.config?P(r==null?void 0:r.configurable,[...O,...(d=s==null?void 0:s.config)!=null?d:[]]):r==null?void 0:r.configurable);let c=_(u(u({},a),r),{configurable:g}),l=a.recursion_limit==null&&r.recursion_limit===25,p=JSON.stringify(c)!==JSON.stringify(a),i=l&&JSON.stringify(_(u({},c),{recursion_limit:null}))===JSON.stringify(_(u({},a),{recursion_limit:null}));return p&&!i?u(u({},a),c):a},e.config)}getMessageInProgress(t){return this.messagesInProcess[t]}setMessageInProgress(t,e){this.messagesInProcess=_(u({},this.messagesInProcess),{[t]:u(u({},this.messagesInProcess[t]),e)})}async getAssistant(){let t=await this.client.assistants.search(),e=t.find(s=>s.assistant_id===this.agentId||s.graph_id===this.graphId);if(!e)throw console.error(`
|
|
2
|
-
No agent found with graph ID ${this.graphId}
|
|
1
|
+
"use strict";var x=Object.defineProperty,Et=Object.defineProperties,St=Object.getOwnPropertyDescriptor,ft=Object.getOwnPropertyDescriptors,vt=Object.getOwnPropertyNames,Z=Object.getOwnPropertySymbols;var st=Object.prototype.hasOwnProperty,yt=Object.prototype.propertyIsEnumerable;var tt=(a,t)=>(t=Symbol[a])?t:Symbol.for("Symbol."+a);var et=(a,t,e)=>t in a?x(a,t,{enumerable:!0,configurable:!0,writable:!0,value:e}):a[t]=e,u=(a,t)=>{for(var e in t||(t={}))st.call(t,e)&&et(a,e,t[e]);if(Z)for(var e of Z(t))yt.call(t,e)&&et(a,e,t[e]);return a},y=(a,t)=>Et(a,ft(t));var Tt=(a,t)=>{for(var e in t)x(a,e,{get:t[e],enumerable:!0})},_t=(a,t,e,n)=>{if(t&&typeof t=="object"||typeof t=="function")for(let s of vt(t))!st.call(a,s)&&s!==e&&x(a,s,{get:()=>t[s],enumerable:!(n=St(t,s))||n.enumerable});return a};var Mt=a=>_t(x({},"__esModule",{value:!0}),a);var nt=(a,t,e)=>(t=a[tt("asyncIterator")])?t.call(a):(a=a[tt("iterator")](),t={},e=(n,s)=>(s=a[n])&&(t[n]=r=>new Promise((c,h,d)=>(r=s.call(a,r),d=r.done,Promise.resolve(r.value).then(m=>c({value:m,done:d}),h)))),e("next"),e("return"),t);var It={};Tt(It,{CustomEventNames:()=>it,LangGraphAgent:()=>D,LangGraphEventTypes:()=>at});module.exports=Mt(It);var ht=require("rxjs"),gt=require("@langchain/langgraph-sdk"),ct=require("crypto"),ut=require("@langchain/core/messages");var at=(o=>(o.OnChainStart="on_chain_start",o.OnChainStream="on_chain_stream",o.OnChainEnd="on_chain_end",o.OnChatModelStart="on_chat_model_start",o.OnChatModelStream="on_chat_model_stream",o.OnChatModelEnd="on_chat_model_end",o.OnToolStart="on_tool_start",o.OnToolEnd="on_tool_end",o.OnCustomEvent="on_custom_event",o.OnInterrupt="on_interrupt",o))(at||{}),it=(s=>(s.ManuallyEmitMessage="manually_emit_message",s.ManuallyEmitToolCall="manually_emit_tool_call",s.ManuallyEmitState="manually_emit_state",s.Exit="exit",s))(it||{});var i=require("@ag-ui/client");var P=["tools"];function k(a,t){return Object.fromEntries(Object.entries(a).filter(([e])=>t.includes(e)))}function rt({mode:a,state:t,schemaKeys:e}){let n=a==="start"?t:null;return n&&(e!=null&&e.input)&&(n=k(n,[...P,...e.input])),n}function ot(a){return a.map(t=>{var e;switch(t.type){case"human":return{id:t.id,role:"user",content:A(I(t.content))};case"ai":let n=I(t.content);return{id:t.id,role:"assistant",content:n?A(n):"",toolCalls:(e=t.tool_calls)==null?void 0:e.map(s=>({id:s.id,type:"function",function:{name:s.name,arguments:JSON.stringify(s.args)}}))};case"system":return{id:t.id,role:"system",content:A(I(t.content))};case"tool":return{id:t.id,role:"tool",content:A(I(t.content)),toolCallId:t.tool_call_id};default:throw new Error("message type returned from LangGraph is not supported.")}})}function lt(a){return a.map((t,e)=>{var n,s;switch(t.role){case"user":return{id:t.id,role:t.role,content:t.content,type:"human"};case"assistant":return{id:t.id,type:"ai",role:t.role,content:(n=t.content)!=null?n:"",tool_calls:((s=t.toolCalls)!=null?s:[]).map(r=>({id:r.id,name:r.function.name,args:JSON.parse(r.function.arguments),type:"tool_call"}))};case"system":return{id:t.id,role:t.role,content:t.content,type:"system"};case"tool":return{content:t.content,role:t.role,type:t.role,tool_call_id:t.toolCallId,id:t.id};default:throw console.error(`Message role ${t.role} is not implemented`),new Error("message role is not supported.")}})}function A(a){return typeof a=="string"?a:JSON.stringify(a)}function dt(a){var e,n,s,r,c;let t=(e=a.chunk)==null?void 0:e.content;if(t&&Array.isArray(t)&&t.length&&t[0])return t[0].thinking?{text:t[0].thinking,type:"text",index:t[0].index}:null;if((r=(s=(n=a.chunk.additional_kwargs)==null?void 0:n.reasoning)==null?void 0:s.summary)!=null&&r[0]){let h=(c=a.chunk.additional_kwargs)==null?void 0:c.reasoning.summary[0];return!h||!h.text?null:{type:"text",text:h.text,index:h.index}}return null}function I(a){var t;if(!a)return null;if(typeof a=="string")return a;if(Array.isArray(a)&&a.length){let e=(t=a.find(n=>n.type==="text"))==null?void 0:t.text;return e!=null?e:null}return null}var D=class extends i.AbstractAgent{constructor(t){var e,n;super(t),this.messagesInProcess={},this.agentName=t.agentName,this.graphId=t.graphId,this.assistantConfig=t.assistantConfig,this.thinkingProcess=null,this.client=(n=t==null?void 0:t.client)!=null?n:new gt.Client({apiUrl:t.deploymentUrl,apiKey:t.langsmithApiKey,defaultHeaders:u({},(e=t.propertyHeaders)!=null?e:{})})}dispatchEvent(t){return this.subscriber.next(t),!0}run(t){return this.activeRun={id:t.runId,threadId:t.threadId},new ht.Observable(e=>(this.handleStreamEvents(t,e),()=>{}))}async handleStreamEvents(t,e){var K,U,F,J,X,B,j,$,Y,V,W,q,z;let{threadId:n,state:s,messages:r,tools:c,context:h,forwardedProps:d}=t;this.subscriber=e;let m=!1;this.activeRun.manuallyEmittedState=null,this.activeRun.nodeName=(K=t.forwardedProps)==null?void 0:K.nodeName;let o=n!=null?n:(0,ct.randomUUID)();this.assistant||(this.assistant=await this.getAssistant());let l=await this.getOrCreateThread(o);this.activeRun.threadId=l.thread_id;let v=(U=await this.client.threads.getState(l.thread_id))!=null?U:{values:{}},g=v.values,O=lt(r);s.messages=g.messages,s=this.langGraphDefaultMergeState(s,O,c);let N=o&&this.activeRun.nodeName!="__end__"&&this.activeRun.nodeName?"continue":"start";N==="continue"&&!((F=d==null?void 0:d.command)!=null&&F.resume)&&await this.client.threads.updateState(o,{values:s,asNode:this.activeRun.nodeName}),this.activeRun.schemaKeys=await this.getSchemaKeys();let w=rt({mode:N,state:s,schemaKeys:this.activeRun.schemaKeys}),L=await this.client.assistants.getGraph(this.assistant.assistant_id),T,R=[this.assistantConfig,d==null?void 0:d.config].filter(Boolean);R.length&&(T=await this.mergeConfigs({configs:R,assistant:this.assistant,schemaKeys:this.activeRun.schemaKeys}));let C=y(u({},d),{streamMode:(J=d==null?void 0:d.streamMode)!=null?J:["events","values","updates"],input:w,config:T}),_=(j=(B=(X=v.tasks)==null?void 0:X[0])==null?void 0:B.interrupts)!=null?j:[];if(_!=null&&_.length&&!(($=d==null?void 0:d.command)!=null&&$.resume))return this.dispatchEvent({type:i.EventType.RUN_STARTED,threadId:o,runId:t.runId}),_.forEach(E=>{this.dispatchEvent({type:i.EventType.CUSTOM,name:"on_interrupt",value:typeof E.value=="string"?E.value:JSON.stringify(E.value),rawEvent:E})}),this.dispatchEvent({type:i.EventType.RUN_FINISHED,threadId:o,runId:t.runId}),e.complete();let S=this.client.runs.stream(o,this.assistant.assistant_id,C);this.activeRun.prevNodeName=null;let H={},G=s;try{this.dispatchEvent({type:i.EventType.RUN_STARTED,threadId:o,runId:this.activeRun.id});try{for(var ge=nt(S),ce,ue,pe;ce=!(ue=await ge.next()).done;ce=!1){let p=ue.value;if(!C.streamMode.includes(p.event))continue;let b=p;if(p.event==="error"){this.dispatchEvent({type:i.EventType.RUN_ERROR,message:p.data.message,rawEvent:p});break}if(p.event==="updates")continue;if(p.event==="values"){H=b.data;continue}let M=b.data,f=M.metadata.langgraph_node,Q=M.event,pt=M.metadata;if(this.activeRun.id=pt.run_id,f&&f!==this.activeRun.nodeName&&(this.activeRun.nodeName&&this.dispatchEvent({type:i.EventType.STEP_FINISHED,stepName:this.activeRun.nodeName}),f&&(this.dispatchEvent({type:i.EventType.STEP_STARTED,stepName:f}),this.activeRun.nodeName=f)),m=m||Q==="on_custom_event"&&M.name==="exit",this.activeRun.exitingNode=this.activeRun.nodeName===f&&Q==="on_chain_end",this.activeRun.exitingNode&&(this.activeRun.manuallyEmittedState=null),L.nodes.some(mt=>mt.id===f)&&(this.activeRun.nodeName=f),G=(Y=this.activeRun.manuallyEmittedState)!=null?Y:H,!this.activeRun.nodeName)continue;(JSON.stringify(G)!==JSON.stringify(s)||this.activeRun.prevNodeName!=this.activeRun.nodeName||this.activeRun.exitingNode)&&!this.getMessageInProgress(this.activeRun.id)&&(s=G,this.activeRun.prevNodeName=this.activeRun.nodeName,this.dispatchEvent({type:i.EventType.STATE_SNAPSHOT,snapshot:this.getStateSnapshot(s),rawEvent:b})),this.dispatchEvent({type:i.EventType.RAW,event:M}),this.handleSingleEvent(M,s)}}catch(ue){pe=[ue]}finally{try{ce&&(ue=ge.return)&&await ue.call(ge)}finally{if(pe)throw pe[0]}}s=await this.client.threads.getState(o);let E=(q=(W=(V=s.tasks)==null?void 0:V[0])==null?void 0:W.interrupts)!=null?q:[];return this.activeRun.nodeName=E?this.activeRun.nodeName:Object.keys(s.metadata.writes)[0],E.forEach(p=>{this.dispatchEvent({type:i.EventType.CUSTOM,name:"on_interrupt",value:typeof p.value=="string"?p.value:JSON.stringify(p.value),rawEvent:p})}),this.dispatchEvent({type:i.EventType.STATE_SNAPSHOT,snapshot:this.getStateSnapshot(s.values)}),this.dispatchEvent({type:i.EventType.MESSAGES_SNAPSHOT,messages:ot((z=s.values.messages)!=null?z:[])}),this.activeRun.nodeName&&this.dispatchEvent({type:i.EventType.STEP_FINISHED,stepName:this.activeRun.nodeName}),this.dispatchEvent({type:i.EventType.RUN_FINISHED,threadId:o,runId:this.activeRun.id}),e.complete()}catch(E){return e.error(E)}}handleSingleEvent(t,e){var n,s,r,c,h,d;switch(t.event){case"on_chat_model_stream":let m=(n=t.metadata["emit-messages"])!=null?n:!0,o=(s=t.metadata["emit-tool-calls"])!=null?s:!0;if(t.data.chunk.response_metadata.finish_reason)return;let l=this.getMessageInProgress(this.activeRun.id),v=!!(l!=null&&l.id),g=(r=t.data.chunk.tool_call_chunks)==null?void 0:r[0],O=(c=t.metadata.predict_state)==null?void 0:c.some(S=>S.tool===(g==null?void 0:g.name)),N=!v&&(g==null?void 0:g.name),w=v&&(l==null?void 0:l.toolCallId)&&(g==null?void 0:g.args),L=v&&(l==null?void 0:l.toolCallId)&&!g,T=dt(t.data),R=I(t.data.chunk.content),C=!!(!g&&R),_=v&&!(l!=null&&l.toolCallId)&&!C;if(T){this.handleThinkingEvent(T);break}if(!T&&this.thinkingProcess&&(this.dispatchEvent({type:i.EventType.THINKING_TEXT_MESSAGE_END}),this.dispatchEvent({type:i.EventType.THINKING_END}),this.thinkingProcess=null),O&&this.dispatchEvent({type:i.EventType.CUSTOM,name:"PredictState",value:t.metadata.predict_state}),L){this.dispatchEvent({type:i.EventType.TOOL_CALL_END,toolCallId:l==null?void 0:l.toolCallId,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if(_){this.dispatchEvent({type:i.EventType.TEXT_MESSAGE_END,messageId:l.id,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if(N&&o){this.dispatchEvent({type:i.EventType.TOOL_CALL_START,toolCallId:g.id,toolCallName:g.name,parentMessageId:t.data.chunk.id,rawEvent:t})&&this.setMessageInProgress(this.activeRun.id,{id:t.data.chunk.id,toolCallId:g.id,toolCallName:g.name});break}if(w&&o){this.dispatchEvent({type:i.EventType.TOOL_CALL_ARGS,toolCallId:l==null?void 0:l.toolCallId,delta:g.args,rawEvent:t});break}if(C&&m){l||(this.dispatchEvent({type:i.EventType.TEXT_MESSAGE_START,role:"assistant",messageId:t.data.chunk.id,rawEvent:t}),this.setMessageInProgress(this.activeRun.id,{id:t.data.chunk.id,toolCallId:null,toolCallName:null}),l=this.getMessageInProgress(this.activeRun.id)),this.dispatchEvent({type:i.EventType.TEXT_MESSAGE_CONTENT,messageId:l.id,delta:R,rawEvent:t});break}break;case"on_chat_model_end":if((h=this.getMessageInProgress(this.activeRun.id))!=null&&h.toolCallId){this.dispatchEvent({type:i.EventType.TOOL_CALL_END,toolCallId:this.getMessageInProgress(this.activeRun.id).toolCallId,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if((d=this.getMessageInProgress(this.activeRun.id))!=null&&d.id){this.dispatchEvent({type:i.EventType.TEXT_MESSAGE_END,messageId:this.getMessageInProgress(this.activeRun.id).id,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}break;case"on_custom_event":if(t.name==="manually_emit_message"){this.dispatchEvent({type:i.EventType.TEXT_MESSAGE_START,role:"assistant",messageId:t.data.message_id,rawEvent:t}),this.dispatchEvent({type:i.EventType.TEXT_MESSAGE_CONTENT,messageId:t.data.message_id,delta:t.data.message,rawEvent:t}),this.dispatchEvent({type:i.EventType.TEXT_MESSAGE_END,messageId:t.data.message_id,rawEvent:t});break}if(t.name==="manually_emit_tool_call"){this.dispatchEvent({type:i.EventType.TOOL_CALL_START,toolCallId:t.data.id,toolCallName:t.data.name,parentMessageId:t.data.id,rawEvent:t}),this.dispatchEvent({type:i.EventType.TOOL_CALL_ARGS,toolCallId:t.data.id,delta:t.data.args,rawEvent:t}),this.dispatchEvent({type:i.EventType.TOOL_CALL_END,toolCallId:t.data.id,rawEvent:t});break}t.name==="manually_emit_state"&&(this.activeRun.manuallyEmittedState=t.data,this.dispatchEvent({type:i.EventType.STATE_SNAPSHOT,snapshot:this.getStateSnapshot(e),rawEvent:t})),this.dispatchEvent({type:i.EventType.CUSTOM,name:t.name,value:t.data,rawEvent:t});break}}handleThinkingEvent(t){var n;if(!t||!t.type||!t.text)return;let e=t.index;(n=this.thinkingProcess)!=null&&n.index&&this.thinkingProcess.index!==e&&(this.thinkingProcess.type&&this.dispatchEvent({type:i.EventType.THINKING_TEXT_MESSAGE_END}),this.dispatchEvent({type:i.EventType.THINKING_END}),this.thinkingProcess=null),this.thinkingProcess||(this.dispatchEvent({type:i.EventType.THINKING_START}),this.thinkingProcess={index:e}),this.thinkingProcess.type!==t.type&&(this.dispatchEvent({type:i.EventType.THINKING_TEXT_MESSAGE_START}),this.thinkingProcess.type=t.type),this.thinkingProcess.type&&this.dispatchEvent({type:i.EventType.THINKING_TEXT_MESSAGE_CONTENT,delta:t.text})}getStateSnapshot(t){let e=this.activeRun.schemaKeys;return e!=null&&e.output&&(t=k(t,[...P,...e.output])),t}async getOrCreateThread(t){let e;try{try{e=await this.getThread(t)}catch(n){e=await this.createThread({threadId:t})}}catch(n){throw new Error(`Failed to create thread: ${n.message}`)}return e}async getThread(t){return this.client.threads.get(t)}async createThread(t){return this.client.threads.create(t)}async mergeConfigs({configs:t,assistant:e,schemaKeys:n}){return t.reduce((s,r)=>{var l;let c=s.configurable;r.configurable&&(c=n!=null&&n.config?k(r==null?void 0:r.configurable,[...P,...(l=n==null?void 0:n.config)!=null?l:[]]):r==null?void 0:r.configurable);let h=y(u(u({},s),r),{configurable:c}),d=s.recursion_limit==null&&r.recursion_limit===25,m=JSON.stringify(h)!==JSON.stringify(s),o=d&&JSON.stringify(y(u({},h),{recursion_limit:null}))===JSON.stringify(y(u({},s),{recursion_limit:null}));return m&&!o?u(u({},s),h):s},e.config)}getMessageInProgress(t){return this.messagesInProcess[t]}setMessageInProgress(t,e){this.messagesInProcess=y(u({},this.messagesInProcess),{[t]:u(u({},this.messagesInProcess[t]),e)})}async getAssistant(){let t=await this.client.assistants.search(),e=t.find(n=>n.graph_id===this.graphId);if(!e)throw console.error(`
|
|
2
|
+
No agent found with graph ID ${this.graphId} found..
|
|
3
3
|
|
|
4
4
|
|
|
5
|
-
These are the available agents: [${t.map(
|
|
6
|
-
`),new Error("No agent id found");return e}async getSchemaKeys(){var e,s
|
|
5
|
+
These are the available agents: [${t.map(n=>`${n.graph_id} (ID: ${n.assistant_id})`).join(", ")}]
|
|
6
|
+
`),new Error("No agent id found");return e}async getSchemaKeys(){var e,n,s;let t=["messages"];try{let r=await this.client.assistants.getSchemas(this.assistant.assistant_id),c=null;if((e=r.config_schema)!=null&&e.properties&&(c=Object.keys(r.config_schema.properties)),!((n=r.input_schema)!=null&&n.properties)||!((s=r.output_schema)!=null&&s.properties))return{config:[],input:t,output:t};let h=Object.keys(r.input_schema.properties),d=Object.keys(r.output_schema.properties);return{input:h&&h.length?[...h,...t]:null,output:d&&d.length?[...d,...t]:null,config:c}}catch(r){return{config:[],input:t,output:t}}}langGraphDefaultMergeState(t,e,n){var m;e.length>0&&"role"in e[0]&&e[0].role==="system"&&(e=e.slice(1));let s=t.messages||[],r=new Set(s.map(o=>o.id)),c=new Set(e.map(o=>o.id)),h=[];e.length<s.length&&(h=s.filter(o=>!c.has(o.id)).map(o=>new ut.RemoveMessage({id:o.id})));let d=e.filter(o=>!r.has(o.id));return y(u({},t),{messages:[...h,...d],tools:[...(m=t.tools)!=null?m:[],...n]})}};0&&(module.exports={CustomEventNames,LangGraphAgent,LangGraphEventTypes});
|
|
7
7
|
//# sourceMappingURL=index.js.map
|
package/dist/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["../src/index.ts","../src/types.ts","../src/utils.ts"],"sourcesContent":["import { Observable, Subscriber } from \"rxjs\";\nimport {\n Client as LangGraphClient,\n EventsStreamEvent,\n StreamMode,\n Config as LangGraphConfig,\n ThreadState,\n Assistant,\n Message as LangGraphMessage,\n Config,\n Interrupt,\n} from \"@langchain/langgraph-sdk\";\nimport { randomUUID } from \"node:crypto\";\nimport { RemoveMessage } from \"@langchain/core/messages\";\nimport {\n LangGraphPlatformMessage,\n CustomEventNames,\n LangGraphEventTypes,\n State,\n MessagesInProgressRecord,\n SchemaKeys,\n MessageInProgress,\n RunMetadata,\n PredictStateTool,\n} from \"./types\";\nimport {\n AbstractAgent,\n AgentConfig,\n CustomEvent,\n EventType,\n MessagesSnapshotEvent,\n RawEvent,\n RunAgentInput,\n RunErrorEvent,\n RunFinishedEvent,\n RunStartedEvent,\n StateDeltaEvent,\n StateSnapshotEvent,\n StepFinishedEvent,\n StepStartedEvent,\n TextMessageContentEvent,\n TextMessageEndEvent,\n TextMessageStartEvent,\n ToolCallArgsEvent,\n ToolCallEndEvent,\n ToolCallStartEvent,\n} from \"@ag-ui/client\";\nimport { RunsStreamPayload } from \"@langchain/langgraph-sdk/dist/types\";\nimport {\n aguiMessagesToLangChain,\n DEFAULT_SCHEMA_KEYS,\n filterObjectBySchemaKeys,\n getStreamPayloadInput,\n langchainMessagesToAgui,\n} from \"@/utils\";\n\nexport type ProcessedEvents =\n | TextMessageStartEvent\n | TextMessageContentEvent\n | TextMessageEndEvent\n | ToolCallStartEvent\n | ToolCallArgsEvent\n | ToolCallEndEvent\n | StateSnapshotEvent\n | StateDeltaEvent\n | MessagesSnapshotEvent\n | RawEvent\n | CustomEvent\n | RunStartedEvent\n | RunFinishedEvent\n | RunErrorEvent\n | StepStartedEvent\n | StepFinishedEvent;\n\ntype RunAgentExtendedInput<\n TStreamMode extends StreamMode | StreamMode[] = StreamMode,\n TSubgraphs extends boolean = false,\n> = Omit<RunAgentInput, \"forwardedProps\"> & {\n forwardedProps?: Omit<RunsStreamPayload<TStreamMode, TSubgraphs>, \"input\"> & {\n nodeName?: string;\n };\n};\n\nexport interface LangGraphAgentConfig extends AgentConfig {\n client?: LangGraphClient;\n deploymentUrl: string;\n langsmithApiKey?: string;\n propertyHeaders?: Record<string, string>;\n assistantConfig?: LangGraphConfig;\n agentName?: string;\n graphId: string;\n}\n\nexport class LangGraphAgent extends AbstractAgent {\n client: LangGraphClient;\n assistantConfig?: LangGraphConfig;\n agentName?: string;\n graphId: string;\n assistant?: Assistant;\n messagesInProcess: MessagesInProgressRecord;\n activeRun?: RunMetadata;\n // @ts-expect-error no need to initialize subscriber right now\n subscriber: Subscriber<ProcessedEvents>;\n\n constructor(config: LangGraphAgentConfig) {\n super(config);\n this.messagesInProcess = {};\n this.agentName = config.agentName;\n this.graphId = config.graphId;\n this.assistantConfig = config.assistantConfig;\n this.client =\n config?.client ??\n new LangGraphClient({\n apiUrl: config.deploymentUrl,\n apiKey: config.langsmithApiKey,\n defaultHeaders: { ...(config.propertyHeaders ?? {}) },\n });\n }\n\n dispatchEvent(event: ProcessedEvents) {\n this.subscriber.next(event);\n return true;\n }\n\n run(input: RunAgentInput) {\n this.activeRun = {\n id: input.runId,\n threadId: input.threadId,\n };\n return new Observable<ProcessedEvents>((subscriber) => {\n this.handleStreamEvents(input, subscriber);\n return () => {};\n });\n }\n\n async handleStreamEvents(input: RunAgentExtendedInput, subscriber: Subscriber<ProcessedEvents>) {\n let { threadId: inputThreadId, state, messages, tools, context, forwardedProps } = input;\n this.subscriber = subscriber;\n let shouldExit = false;\n // If a manual emittance happens, it is the ultimate source of truth of state, unless a node has exited.\n // Therefore, this value should either hold null, or the only edition of state that should be used.\n this.activeRun!.manuallyEmittedState = null;\n\n this.activeRun!.nodeName = input.forwardedProps?.nodeName;\n\n const threadId = inputThreadId ?? randomUUID();\n\n if (!this.assistant) {\n this.assistant = await this.getAssistant();\n }\n\n let agentState = await this.getOrCreateThreadAndReturnState(threadId);\n\n const agentStateValues = agentState.values as State;\n const aguiToLangChainMessage = aguiMessagesToLangChain(messages);\n\n state.messages = agentStateValues.messages;\n state = this.langGraphDefaultMergeState(state, aguiToLangChainMessage, tools);\n\n const mode =\n threadId && this.activeRun!.nodeName != \"__end__\" && this.activeRun!.nodeName\n ? \"continue\"\n : \"start\";\n\n if (mode === \"continue\" && !forwardedProps?.command?.resume) {\n await this.client.threads.updateState(threadId, {\n values: state,\n asNode: this.activeRun!.nodeName,\n });\n }\n\n this.activeRun!.schemaKeys = await this.getSchemaKeys();\n\n const payloadInput = getStreamPayloadInput({\n mode,\n state,\n schemaKeys: this.activeRun!.schemaKeys,\n });\n\n const graphInfo = await this.client.assistants.getGraph(this.assistant.assistant_id);\n\n let payloadConfig: LangGraphConfig | undefined;\n const configsToMerge = [this.assistantConfig, forwardedProps?.config].filter(\n Boolean,\n ) as LangGraphConfig[];\n if (configsToMerge.length) {\n payloadConfig = await this.mergeConfigs({\n configs: configsToMerge,\n assistant: this.assistant,\n schemaKeys: this.activeRun!.schemaKeys,\n });\n }\n const payload = {\n ...forwardedProps,\n streamMode:\n forwardedProps?.streamMode ?? ([\"events\", \"values\", \"updates\"] satisfies StreamMode[]),\n input: payloadInput,\n config: payloadConfig,\n };\n\n // If there are still outstanding unresolved interrupts, we must force resolution of them before moving forward\n const interrupts = (agentState.tasks?.[0]?.interrupts ?? []) as Interrupt[];\n if (interrupts?.length && !forwardedProps?.command?.resume) {\n this.dispatchEvent({\n type: EventType.RUN_STARTED,\n threadId,\n runId: input.runId,\n });\n\n interrupts.forEach((interrupt) => {\n this.dispatchEvent({\n type: EventType.CUSTOM,\n name: LangGraphEventTypes.OnInterrupt,\n value:\n typeof interrupt.value === \"string\" ? interrupt.value : JSON.stringify(interrupt.value),\n rawEvent: interrupt,\n });\n });\n\n this.dispatchEvent({\n type: EventType.RUN_FINISHED,\n threadId,\n runId: input.runId,\n });\n return subscriber.complete();\n }\n const streamResponse = this.client.runs.stream(threadId, this.assistant.assistant_id, payload);\n\n this.activeRun!.prevNodeName = null;\n let latestStateValues = {};\n let updatedState = state;\n\n try {\n this.dispatchEvent({\n type: EventType.RUN_STARTED,\n threadId,\n runId: this.activeRun!.id,\n });\n\n for await (let streamResponseChunk of streamResponse) {\n // @ts-ignore\n if (!payload.streamMode.includes(streamResponseChunk.event as StreamMode)) {\n continue;\n }\n // Force event type, as data is not properly defined on the LG side.\n type EventsChunkData = {\n __interrupt__?: any;\n metadata: Record<string, any>;\n event: string;\n data: any;\n [key: string]: unknown;\n };\n const chunk = streamResponseChunk as EventsStreamEvent & { data: EventsChunkData };\n\n if (streamResponseChunk.event === \"error\") {\n this.dispatchEvent({\n type: EventType.RUN_ERROR,\n message: streamResponseChunk.data.message,\n rawEvent: streamResponseChunk,\n });\n break;\n }\n\n if (streamResponseChunk.event === \"updates\") continue;\n\n if (streamResponseChunk.event === \"values\") {\n latestStateValues = chunk.data;\n continue;\n }\n\n const chunkData = chunk.data;\n const currentNodeName = chunkData.metadata.langgraph_node;\n const eventType = chunkData.event;\n const metadata = chunkData.metadata;\n this.activeRun!.id = metadata.run_id;\n\n if (currentNodeName && currentNodeName !== this.activeRun!.nodeName) {\n if (this.activeRun!.nodeName) {\n this.dispatchEvent({\n type: EventType.STEP_FINISHED,\n stepName: this.activeRun!.nodeName,\n });\n }\n\n if (currentNodeName) {\n this.dispatchEvent({\n type: EventType.STEP_STARTED,\n stepName: currentNodeName,\n });\n this.activeRun!.nodeName = currentNodeName;\n }\n }\n\n shouldExit =\n shouldExit ||\n (eventType === LangGraphEventTypes.OnCustomEvent &&\n chunkData.name === CustomEventNames.Exit);\n\n this.activeRun!.exitingNode =\n this.activeRun!.nodeName === currentNodeName &&\n eventType === LangGraphEventTypes.OnChainEnd;\n if (this.activeRun!.exitingNode) {\n this.activeRun!.manuallyEmittedState = null;\n }\n\n // we only want to update the node name under certain conditions\n // since we don't need any internal node names to be sent to the frontend\n if (graphInfo[\"nodes\"].some((node) => node.id === currentNodeName)) {\n this.activeRun!.nodeName = currentNodeName;\n }\n\n updatedState = this.activeRun!.manuallyEmittedState ?? latestStateValues;\n\n if (!this.activeRun!.nodeName) {\n continue;\n }\n\n const hasStateDiff = JSON.stringify(updatedState) !== JSON.stringify(state);\n // We should not update snapshot while a message is in progress.\n if (\n (hasStateDiff ||\n this.activeRun!.prevNodeName != this.activeRun!.nodeName ||\n this.activeRun!.exitingNode) &&\n !Boolean(this.getMessageInProgress(this.activeRun!.id))\n ) {\n state = updatedState;\n this.activeRun!.prevNodeName = this.activeRun!.nodeName;\n\n this.dispatchEvent({\n type: EventType.STATE_SNAPSHOT,\n snapshot: this.getStateSnapshot(state),\n rawEvent: chunk,\n });\n }\n\n this.dispatchEvent({\n type: EventType.RAW,\n event: chunkData,\n });\n\n this.handleSingleEvent(chunkData, state);\n }\n\n state = await this.client.threads.getState(threadId);\n const interrupts = (state.tasks?.[0]?.interrupts ?? []) as Interrupt[];\n this.activeRun!.nodeName = interrupts\n ? this.activeRun!.nodeName\n : Object.keys(state.metadata.writes)[0];\n\n interrupts.forEach((interrupt) => {\n this.dispatchEvent({\n type: EventType.CUSTOM,\n name: LangGraphEventTypes.OnInterrupt,\n value:\n typeof interrupt.value === \"string\" ? interrupt.value : JSON.stringify(interrupt.value),\n rawEvent: interrupt,\n });\n });\n\n this.dispatchEvent({\n type: EventType.STATE_SNAPSHOT,\n snapshot: this.getStateSnapshot(state.values),\n });\n this.dispatchEvent({\n type: EventType.MESSAGES_SNAPSHOT,\n messages: langchainMessagesToAgui(state.values.messages ?? []),\n });\n if (this.activeRun!.nodeName) {\n this.dispatchEvent({\n type: EventType.STEP_FINISHED,\n stepName: this.activeRun!.nodeName!,\n });\n }\n this.dispatchEvent({\n type: EventType.RUN_FINISHED,\n threadId,\n runId: this.activeRun!.id,\n });\n return subscriber.complete();\n } catch (e) {\n return subscriber.error(e);\n }\n }\n\n handleSingleEvent(event: any, state: State): void {\n switch (event.event) {\n case LangGraphEventTypes.OnChatModelStream:\n let shouldEmitMessages = event.metadata[\"emit-messages\"] ?? true;\n let shouldEmitToolCalls = event.metadata[\"emit-tool-calls\"] ?? true;\n\n if (event.data.chunk.response_metadata.finish_reason) return;\n const currentStream = this.getMessageInProgress(this.activeRun!.id);\n const hasCurrentStream = Boolean(currentStream?.id);\n const toolCallData = event.data.chunk.tool_call_chunks?.[0];\n const toolCallUsedToPredictState = event.metadata[\"predict_state\"]?.some(\n (predictStateTool: PredictStateTool) => predictStateTool.tool === toolCallData?.name,\n );\n\n const isToolCallStartEvent = !hasCurrentStream && toolCallData?.name;\n const isToolCallArgsEvent =\n hasCurrentStream && currentStream?.toolCallId && toolCallData.args;\n const isToolCallEndEvent = hasCurrentStream && currentStream?.toolCallId && !toolCallData;\n\n const isMessageStartEvent = !hasCurrentStream && !toolCallData;\n const isMessageContentEvent = hasCurrentStream && !toolCallData;\n const isMessageEndEvent =\n hasCurrentStream && !currentStream?.toolCallId && !isMessageContentEvent;\n\n if (toolCallUsedToPredictState) {\n this.dispatchEvent({\n type: EventType.CUSTOM,\n name: \"PredictState\",\n value: event.metadata[\"predict_state\"],\n });\n }\n\n if (isToolCallEndEvent) {\n const resolved = this.dispatchEvent({\n type: EventType.TOOL_CALL_END,\n toolCallId: currentStream.toolCallId!,\n rawEvent: event,\n });\n if (resolved) {\n this.messagesInProcess[this.activeRun!.id] = null;\n }\n break;\n }\n\n if (isMessageEndEvent) {\n const resolved = this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_END,\n messageId: currentStream!.id,\n rawEvent: event,\n });\n if (resolved) {\n this.messagesInProcess[this.activeRun!.id] = null;\n }\n break;\n }\n\n if (isToolCallStartEvent && shouldEmitToolCalls) {\n const resolved = this.dispatchEvent({\n type: EventType.TOOL_CALL_START,\n toolCallId: toolCallData.id,\n toolCallName: toolCallData.name,\n parentMessageId: event.data.chunk.id,\n rawEvent: event,\n });\n if (resolved) {\n this.setMessageInProgress(this.activeRun!.id, {\n id: event.data.chunk.id,\n toolCallId: toolCallData.id,\n toolCallName: toolCallData.name,\n });\n }\n break;\n }\n\n // Tool call args: emit ActionExecutionArgs\n if (isToolCallArgsEvent && shouldEmitToolCalls) {\n this.dispatchEvent({\n type: EventType.TOOL_CALL_ARGS,\n toolCallId: currentStream.toolCallId!,\n delta: toolCallData.args,\n rawEvent: event,\n });\n break;\n }\n\n // Message started: emit TextMessageStart\n if (isMessageStartEvent && shouldEmitMessages) {\n const resolved = this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_START,\n role: \"assistant\",\n messageId: event.data.chunk.id,\n rawEvent: event,\n });\n if (resolved) {\n this.setMessageInProgress(this.activeRun!.id, {\n id: event.data.chunk.id,\n toolCallId: null,\n toolCallName: null,\n });\n }\n break;\n }\n // Message content: emit TextMessageContent\n if (isMessageContentEvent && shouldEmitMessages) {\n this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_CONTENT,\n messageId: currentStream!.id,\n delta: event.data.chunk.content,\n rawEvent: event,\n });\n break;\n }\n\n break;\n case LangGraphEventTypes.OnChatModelEnd:\n if (this.getMessageInProgress(this.activeRun!.id)?.toolCallId) {\n const resolved = this.dispatchEvent({\n type: EventType.TOOL_CALL_END,\n toolCallId: this.getMessageInProgress(this.activeRun!.id)!.toolCallId!,\n rawEvent: event,\n });\n if (resolved) {\n this.messagesInProcess[this.activeRun!.id] = null;\n }\n break;\n }\n if (this.getMessageInProgress(this.activeRun!.id)?.id) {\n const resolved = this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_END,\n messageId: this.getMessageInProgress(this.activeRun!.id)!.id,\n rawEvent: event,\n });\n if (resolved) {\n this.messagesInProcess[this.activeRun!.id] = null;\n }\n break;\n }\n break;\n case LangGraphEventTypes.OnCustomEvent:\n if (event.name === CustomEventNames.ManuallyEmitMessage) {\n this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_START,\n role: \"assistant\",\n messageId: event.data.message_id,\n rawEvent: event,\n });\n this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_CONTENT,\n messageId: event.data.message_id,\n delta: event.data.message,\n rawEvent: event,\n });\n this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_END,\n messageId: event.data.message_id,\n rawEvent: event,\n });\n break;\n }\n\n if (event.name === CustomEventNames.ManuallyEmitToolCall) {\n this.dispatchEvent({\n type: EventType.TOOL_CALL_START,\n toolCallId: event.data.id,\n toolCallName: event.data.name,\n parentMessageId: event.data.id,\n rawEvent: event,\n });\n this.dispatchEvent({\n type: EventType.TOOL_CALL_ARGS,\n toolCallId: event.data.id,\n delta: event.data.args,\n rawEvent: event,\n });\n this.dispatchEvent({\n type: EventType.TOOL_CALL_END,\n toolCallId: event.data.id,\n rawEvent: event,\n });\n break;\n }\n\n if (event.name === CustomEventNames.ManuallyEmitState) {\n this.activeRun!.manuallyEmittedState = event.data;\n this.dispatchEvent({\n type: EventType.STATE_SNAPSHOT,\n snapshot: this.getStateSnapshot(state),\n rawEvent: event,\n });\n }\n\n this.dispatchEvent({\n type: EventType.CUSTOM,\n name: event.name,\n value: event.data,\n rawEvent: event,\n });\n break;\n }\n }\n\n getStateSnapshot(state: State) {\n const schemaKeys = this.activeRun!.schemaKeys!;\n // Do not emit state keys that are not part of the output schema\n if (schemaKeys?.output) {\n state = filterObjectBySchemaKeys(state, [...DEFAULT_SCHEMA_KEYS, ...schemaKeys.output]);\n }\n // return state\n return state;\n }\n\n async getOrCreateThreadAndReturnState(threadId: string): Promise<ThreadState<{}>> {\n let agentState = { values: {} } as ThreadState;\n try {\n await this.client.threads.get(threadId);\n agentState = await this.client.threads.getState(threadId);\n } catch (error) {\n await this.client.threads.create({ threadId });\n }\n\n return agentState;\n }\n\n async mergeConfigs({\n configs,\n assistant,\n schemaKeys,\n }: {\n configs: Config[];\n assistant: Assistant;\n schemaKeys: SchemaKeys;\n }) {\n return configs.reduce((acc, cfg) => {\n let filteredConfigurable = acc.configurable;\n\n if (cfg.configurable) {\n filteredConfigurable = schemaKeys?.config\n ? filterObjectBySchemaKeys(cfg?.configurable, [\n ...DEFAULT_SCHEMA_KEYS,\n ...(schemaKeys?.config ?? []),\n ])\n : cfg?.configurable;\n }\n\n const newConfig = {\n ...acc,\n ...cfg,\n configurable: filteredConfigurable,\n };\n\n // LG does not return recursion limit if it's the default, therefore we check: if no recursion limit is currently set, and the user asked for 25, there is no change.\n const isRecursionLimitSetToDefault =\n acc.recursion_limit == null && cfg.recursion_limit === 25;\n // Deep compare configs to avoid unnecessary update calls\n const configsAreDifferent = JSON.stringify(newConfig) !== JSON.stringify(acc);\n\n // Check if the only difference is the recursion_limit being set to default\n const isOnlyRecursionLimitDifferent =\n isRecursionLimitSetToDefault &&\n JSON.stringify({ ...newConfig, recursion_limit: null }) ===\n JSON.stringify({ ...acc, recursion_limit: null });\n\n if (configsAreDifferent && !isOnlyRecursionLimitDifferent) {\n return {\n ...acc,\n ...newConfig,\n };\n }\n\n return acc;\n }, assistant.config);\n }\n\n getMessageInProgress(runId: string) {\n return this.messagesInProcess[runId];\n }\n\n setMessageInProgress(runId: string, data: MessageInProgress) {\n this.messagesInProcess = {\n ...this.messagesInProcess,\n [runId]: {\n ...(this.messagesInProcess[runId] as MessageInProgress),\n ...data,\n },\n };\n }\n\n async getAssistant(): Promise<Assistant> {\n const assistants = await this.client.assistants.search();\n const retrievedAssistant = assistants.find(\n (searchResult) =>\n searchResult.assistant_id === this.agentId || searchResult.graph_id === this.graphId,\n );\n if (!retrievedAssistant) {\n console.error(`\n No agent found with graph ID ${this.graphId} or agent ID ${this.agentId} found..\\n\n \n These are the available agents: [${assistants.map((a) => `${a.graph_id} (ID: ${a.assistant_id})`).join(\", \")}]\n `);\n throw new Error(\"No agent id found\");\n }\n\n return retrievedAssistant;\n }\n\n async getSchemaKeys(): Promise<SchemaKeys> {\n const CONSTANT_KEYS = [\"messages\"];\n\n try {\n const graphSchema = await this.client.assistants.getSchemas(this.assistant!.assistant_id);\n let configSchema = null;\n if (graphSchema.config_schema?.properties) {\n configSchema = Object.keys(graphSchema.config_schema.properties);\n }\n if (!graphSchema.input_schema?.properties || !graphSchema.output_schema?.properties) {\n return { config: [], input: CONSTANT_KEYS, output: CONSTANT_KEYS };\n }\n const inputSchema = Object.keys(graphSchema.input_schema.properties);\n const outputSchema = Object.keys(graphSchema.output_schema.properties);\n\n return {\n input: inputSchema && inputSchema.length ? [...inputSchema, ...CONSTANT_KEYS] : null,\n output: outputSchema && outputSchema.length ? [...outputSchema, ...CONSTANT_KEYS] : null,\n config: configSchema,\n };\n } catch (e) {\n return { config: [], input: CONSTANT_KEYS, output: CONSTANT_KEYS };\n }\n }\n\n langGraphDefaultMergeState(state: State, messages: LangGraphMessage[], tools: any): State {\n if (messages.length > 0 && \"role\" in messages[0] && messages[0].role === \"system\") {\n // remove system message\n messages = messages.slice(1);\n }\n\n // merge with existing messages\n const existingMessages: LangGraphPlatformMessage[] = state.messages || [];\n const existingMessageIds = new Set(existingMessages.map((message) => message.id));\n const messageIds = new Set(messages.map((message) => message.id));\n\n let removedMessages: RemoveMessage[] = [];\n if (messages.length < existingMessages.length) {\n // Messages were removed\n removedMessages = existingMessages\n .filter((m) => !messageIds.has(m.id))\n .map((m) => new RemoveMessage({ id: m.id! }));\n }\n\n const newMessages = messages.filter((message) => !existingMessageIds.has(message.id));\n\n return {\n ...state,\n messages: [...removedMessages, ...newMessages],\n tools: [...(state.tools ?? []), ...tools],\n };\n }\n}\n\nexport * from \"./types\";\n","import { Message } from \"@langchain/langgraph-sdk\";\nimport { MessageType } from \"@langchain/core/messages\";\n\nexport enum LangGraphEventTypes {\n OnChainStart = \"on_chain_start\",\n OnChainStream = \"on_chain_stream\",\n OnChainEnd = \"on_chain_end\",\n OnChatModelStart = \"on_chat_model_start\",\n OnChatModelStream = \"on_chat_model_stream\",\n OnChatModelEnd = \"on_chat_model_end\",\n OnToolStart = \"on_tool_start\",\n OnToolEnd = \"on_tool_end\",\n OnCustomEvent = \"on_custom_event\",\n OnInterrupt = \"on_interrupt\",\n}\n\nexport type State = Record<string, any>;\n\nexport type SchemaKeys = {\n input: string[] | null;\n output: string[] | null;\n config: string[] | null;\n} | null;\n\nexport type MessageInProgress = {\n id: string;\n toolCallId?: string | null;\n toolCallName?: string | null;\n};\n\nexport interface RunMetadata {\n id: string;\n schemaKeys?: SchemaKeys;\n nodeName?: string;\n prevNodeName?: string | null;\n exitingNode?: boolean;\n manuallyEmittedState?: State | null;\n threadId?: string;\n}\n\nexport type MessagesInProgressRecord = Record<string, MessageInProgress | null>;\n\n// The following types are our own definition to the messages accepted by LangGraph Platform, enhanced with some of our extra data.\nexport interface ToolCall {\n id: string;\n name: string;\n args: Record<string, unknown>;\n}\n\ntype BaseLangGraphPlatformMessage = Omit<\n Message,\n | \"isResultMessage\"\n | \"isTextMessage\"\n | \"isImageMessage\"\n | \"isActionExecutionMessage\"\n | \"isAgentStateMessage\"\n | \"type\"\n | \"createdAt\"\n> & {\n content: string;\n role: string;\n additional_kwargs?: Record<string, unknown>;\n type: MessageType;\n};\n\ninterface LangGraphPlatformResultMessage extends BaseLangGraphPlatformMessage {\n tool_call_id: string;\n name: string;\n}\n\ninterface LangGraphPlatformActionExecutionMessage extends BaseLangGraphPlatformMessage {\n tool_calls: ToolCall[];\n}\n\nexport type LangGraphPlatformMessage =\n | LangGraphPlatformActionExecutionMessage\n | LangGraphPlatformResultMessage\n | BaseLangGraphPlatformMessage;\n\nexport enum CustomEventNames {\n ManuallyEmitMessage = \"manually_emit_message\",\n ManuallyEmitToolCall = \"manually_emit_tool_call\",\n ManuallyEmitState = \"manually_emit_state\",\n Exit = \"exit\",\n}\n\nexport interface PredictStateTool {\n tool: string;\n state_key: string;\n tool_argument: string;\n}\n","import { Message as LangGraphMessage } from \"@langchain/langgraph-sdk\";\nimport { State, SchemaKeys } from \"./types\";\nimport { Message, ToolCall } from \"@ag-ui/client\";\n\nexport const DEFAULT_SCHEMA_KEYS = [\"tools\"];\n\nexport function filterObjectBySchemaKeys(obj: Record<string, any>, schemaKeys: string[]) {\n return Object.fromEntries(Object.entries(obj).filter(([key]) => schemaKeys.includes(key)));\n}\n\nexport function getStreamPayloadInput({\n mode,\n state,\n schemaKeys,\n}: {\n mode: \"start\" | \"continue\";\n state: State;\n schemaKeys: SchemaKeys;\n}) {\n let input = mode === \"start\" ? state : null;\n // Do not input keys that are not part of the input schema\n if (input && schemaKeys?.input) {\n input = filterObjectBySchemaKeys(input, [...DEFAULT_SCHEMA_KEYS, ...schemaKeys.input]);\n }\n\n return input;\n}\n\nexport function langchainMessagesToAgui(messages: LangGraphMessage[]): Message[] {\n return messages.map((message) => {\n switch (message.type) {\n case \"human\":\n return {\n id: message.id!,\n role: \"user\",\n content: stringifyIfNeeded(message.content),\n };\n case \"ai\":\n return {\n id: message.id!,\n role: \"assistant\",\n content: stringifyIfNeeded(message.content),\n toolCalls: message.tool_calls?.map((tc) => ({\n id: tc.id!,\n type: \"function\",\n function: {\n name: tc.name,\n arguments: JSON.stringify(tc.args),\n },\n })),\n };\n case \"system\":\n return {\n id: message.id!,\n role: \"system\",\n content: stringifyIfNeeded(message.content),\n };\n case \"tool\":\n return {\n id: message.id!,\n role: \"tool\",\n content: stringifyIfNeeded(message.content),\n toolCallId: message.tool_call_id,\n };\n default:\n throw new Error(\"message type returned from LangGraph is not supported.\");\n }\n });\n}\n\nexport function aguiMessagesToLangChain(messages: Message[]): LangGraphMessage[] {\n return messages.map((message, index) => {\n switch (message.role) {\n case \"user\":\n return {\n id: message.id,\n role: message.role,\n content: message.content,\n type: \"human\",\n };\n case \"assistant\":\n return {\n id: message.id,\n type: \"ai\",\n role: message.role,\n content: message.content ?? \"\",\n tool_calls: (message.toolCalls ?? []).map((tc: ToolCall) => ({\n id: tc.id,\n name: tc.function.name,\n args: JSON.parse(tc.function.arguments),\n type: \"tool_call\",\n })),\n };\n case \"system\":\n return {\n id: message.id,\n role: message.role,\n content: message.content,\n type: \"system\",\n };\n case \"tool\":\n return {\n content: message.content,\n role: message.role,\n type: message.role,\n tool_call_id: message.toolCallId,\n id: message.id,\n };\n default:\n console.error(`Message role ${message.role} is not implemented`);\n throw new Error(\"message role is not supported.\");\n }\n });\n}\n\nfunction stringifyIfNeeded(item: any) {\n if (typeof item === \"string\") return item;\n return JSON.stringify(item);\n}\n"],"mappings":"mjCAAA,IAAAA,GAAA,GAAAC,GAAAD,GAAA,sBAAAE,GAAA,mBAAAC,EAAA,wBAAAC,KAAA,eAAAC,GAAAL,IAAA,IAAAM,GAAuC,gBACvCC,GAUO,oCACPC,GAA2B,kBAC3BC,GAA8B,oCCVvB,IAAKC,QACVA,EAAA,aAAe,iBACfA,EAAA,cAAgB,kBAChBA,EAAA,WAAa,eACbA,EAAA,iBAAmB,sBACnBA,EAAA,kBAAoB,uBACpBA,EAAA,eAAiB,oBACjBA,EAAA,YAAc,gBACdA,EAAA,UAAY,cACZA,EAAA,cAAgB,kBAChBA,EAAA,YAAc,eAVJA,QAAA,IA4EAC,QACVA,EAAA,oBAAsB,wBACtBA,EAAA,qBAAuB,0BACvBA,EAAA,kBAAoB,sBACpBA,EAAA,KAAO,OAJGA,QAAA,IDtDZ,IAAAC,EAqBO,yBE1CA,IAAMC,EAAsB,CAAC,OAAO,EAEpC,SAASC,EAAyBC,EAA0BC,EAAsB,CACvF,OAAO,OAAO,YAAY,OAAO,QAAQD,CAAG,EAAE,OAAO,CAAC,CAACE,CAAG,IAAMD,EAAW,SAASC,CAAG,CAAC,CAAC,CAC3F,CAEO,SAASC,GAAsB,CACpC,KAAAC,EACA,MAAAC,EACA,WAAAJ,CACF,EAIG,CACD,IAAIK,EAAQF,IAAS,QAAUC,EAAQ,KAEvC,OAAIC,IAASL,GAAA,MAAAA,EAAY,SACvBK,EAAQP,EAAyBO,EAAO,CAAC,GAAGR,EAAqB,GAAGG,EAAW,KAAK,CAAC,GAGhFK,CACT,CAEO,SAASC,GAAwBC,EAAyC,CAC/E,OAAOA,EAAS,IAAKC,GAAY,CA7BnC,IAAAC,EA8BI,OAAQD,EAAQ,KAAM,CACpB,IAAK,QACH,MAAO,CACL,GAAIA,EAAQ,GACZ,KAAM,OACN,QAASE,EAAkBF,EAAQ,OAAO,CAC5C,EACF,IAAK,KACH,MAAO,CACL,GAAIA,EAAQ,GACZ,KAAM,YACN,QAASE,EAAkBF,EAAQ,OAAO,EAC1C,WAAWC,EAAAD,EAAQ,aAAR,YAAAC,EAAoB,IAAKE,IAAQ,CAC1C,GAAIA,EAAG,GACP,KAAM,WACN,SAAU,CACR,KAAMA,EAAG,KACT,UAAW,KAAK,UAAUA,EAAG,IAAI,CACnC,CACF,GACF,EACF,IAAK,SACH,MAAO,CACL,GAAIH,EAAQ,GACZ,KAAM,SACN,QAASE,EAAkBF,EAAQ,OAAO,CAC5C,EACF,IAAK,OACH,MAAO,CACL,GAAIA,EAAQ,GACZ,KAAM,OACN,QAASE,EAAkBF,EAAQ,OAAO,EAC1C,WAAYA,EAAQ,YACtB,EACF,QACE,MAAM,IAAI,MAAM,wDAAwD,CAC5E,CACF,CAAC,CACH,CAEO,SAASI,GAAwBL,EAAyC,CAC/E,OAAOA,EAAS,IAAI,CAACC,EAASK,IAAU,CAvE1C,IAAAJ,EAAAK,EAwEI,OAAQN,EAAQ,KAAM,CACpB,IAAK,OACH,MAAO,CACL,GAAIA,EAAQ,GACZ,KAAMA,EAAQ,KACd,QAASA,EAAQ,QACjB,KAAM,OACR,EACF,IAAK,YACH,MAAO,CACL,GAAIA,EAAQ,GACZ,KAAM,KACN,KAAMA,EAAQ,KACd,SAASC,EAAAD,EAAQ,UAAR,KAAAC,EAAmB,GAC5B,aAAaK,EAAAN,EAAQ,YAAR,KAAAM,EAAqB,CAAC,GAAG,IAAKH,IAAkB,CAC3D,GAAIA,EAAG,GACP,KAAMA,EAAG,SAAS,KAClB,KAAM,KAAK,MAAMA,EAAG,SAAS,SAAS,EACtC,KAAM,WACR,EAAE,CACJ,EACF,IAAK,SACH,MAAO,CACL,GAAIH,EAAQ,GACZ,KAAMA,EAAQ,KACd,QAASA,EAAQ,QACjB,KAAM,QACR,EACF,IAAK,OACH,MAAO,CACL,QAASA,EAAQ,QACjB,KAAMA,EAAQ,KACd,KAAMA,EAAQ,KACd,aAAcA,EAAQ,WACtB,GAAIA,EAAQ,EACd,EACF,QACE,cAAQ,MAAM,gBAAgBA,EAAQ,IAAI,qBAAqB,EACzD,IAAI,MAAM,gCAAgC,CACpD,CACF,CAAC,CACH,CAEA,SAASE,EAAkBK,EAAW,CACpC,OAAI,OAAOA,GAAS,SAAiBA,EAC9B,KAAK,UAAUA,CAAI,CAC5B,CFzBO,IAAMC,EAAN,cAA6B,eAAc,CAWhD,YAAYC,EAA8B,CAxG5C,IAAAC,EAAAC,EAyGI,MAAMF,CAAM,EACZ,KAAK,kBAAoB,CAAC,EAC1B,KAAK,UAAYA,EAAO,UACxB,KAAK,QAAUA,EAAO,QACtB,KAAK,gBAAkBA,EAAO,gBAC9B,KAAK,QACHE,EAAAF,GAAA,YAAAA,EAAQ,SAAR,KAAAE,EACA,IAAI,GAAAC,OAAgB,CAClB,OAAQH,EAAO,cACf,OAAQA,EAAO,gBACf,eAAgBI,EAAA,IAAMH,EAAAD,EAAO,kBAAP,KAAAC,EAA0B,CAAC,EACnD,CAAC,CACL,CAEA,cAAcI,EAAwB,CACpC,YAAK,WAAW,KAAKA,CAAK,EACnB,EACT,CAEA,IAAIC,EAAsB,CACxB,YAAK,UAAY,CACf,GAAIA,EAAM,MACV,SAAUA,EAAM,QAClB,EACO,IAAI,cAA6BC,IACtC,KAAK,mBAAmBD,EAAOC,CAAU,EAClC,IAAM,CAAC,EACf,CACH,CAEA,MAAM,mBAAmBD,EAA8BC,EAAyC,CAvIlG,IAAAN,EAAAC,EAAAM,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAwII,GAAI,CAAE,SAAUC,EAAe,MAAAC,EAAO,SAAAC,EAAU,MAAAC,EAAO,QAAAC,EAAS,eAAAC,CAAe,EAAIjB,EACnF,KAAK,WAAaC,EAClB,IAAIiB,EAAa,GAGjB,KAAK,UAAW,qBAAuB,KAEvC,KAAK,UAAW,UAAWvB,EAAAK,EAAM,iBAAN,YAAAL,EAAsB,SAEjD,IAAMwB,EAAWP,GAAA,KAAAA,KAAiB,eAAW,EAExC,KAAK,YACR,KAAK,UAAY,MAAM,KAAK,aAAa,GAG3C,IAAIQ,EAAa,MAAM,KAAK,gCAAgCD,CAAQ,EAE9DE,EAAmBD,EAAW,OAC9BE,EAAyBC,GAAwBT,CAAQ,EAE/DD,EAAM,SAAWQ,EAAiB,SAClCR,EAAQ,KAAK,2BAA2BA,EAAOS,EAAwBP,CAAK,EAE5E,IAAMS,EACJL,GAAY,KAAK,UAAW,UAAY,WAAa,KAAK,UAAW,SACjE,WACA,QAEFK,IAAS,YAAc,GAAC5B,EAAAqB,GAAA,YAAAA,EAAgB,UAAhB,MAAArB,EAAyB,SACnD,MAAM,KAAK,OAAO,QAAQ,YAAYuB,EAAU,CAC9C,OAAQN,EACR,OAAQ,KAAK,UAAW,QAC1B,CAAC,EAGH,KAAK,UAAW,WAAa,MAAM,KAAK,cAAc,EAEtD,IAAMY,EAAeC,GAAsB,CACzC,KAAAF,EACA,MAAAX,EACA,WAAY,KAAK,UAAW,UAC9B,CAAC,EAEKc,EAAY,MAAM,KAAK,OAAO,WAAW,SAAS,KAAK,UAAU,YAAY,EAE/EC,EACEC,EAAiB,CAAC,KAAK,gBAAiBZ,GAAA,YAAAA,EAAgB,MAAM,EAAE,OACpE,OACF,EACIY,EAAe,SACjBD,EAAgB,MAAM,KAAK,aAAa,CACtC,QAASC,EACT,UAAW,KAAK,UAChB,WAAY,KAAK,UAAW,UAC9B,CAAC,GAEH,IAAMC,EAAUC,EAAAjC,EAAA,GACXmB,GADW,CAEd,YACEf,EAAAe,GAAA,YAAAA,EAAgB,aAAhB,KAAAf,EAA+B,CAAC,SAAU,SAAU,SAAS,EAC/D,MAAOuB,EACP,OAAQG,CACV,GAGMI,GAAc3B,GAAAD,GAAAD,EAAAiB,EAAW,QAAX,YAAAjB,EAAmB,KAAnB,YAAAC,EAAuB,aAAvB,KAAAC,EAAqC,CAAC,EAC1D,GAAI2B,GAAA,MAAAA,EAAY,QAAU,GAAC1B,EAAAW,GAAA,YAAAA,EAAgB,UAAhB,MAAAX,EAAyB,QAClD,YAAK,cAAc,CACjB,KAAM,YAAU,YAChB,SAAAa,EACA,MAAOnB,EAAM,KACf,CAAC,EAEDgC,EAAW,QAASC,GAAc,CAChC,KAAK,cAAc,CACjB,KAAM,YAAU,OAChB,oBACA,MACE,OAAOA,EAAU,OAAU,SAAWA,EAAU,MAAQ,KAAK,UAAUA,EAAU,KAAK,EACxF,SAAUA,CACZ,CAAC,CACH,CAAC,EAED,KAAK,cAAc,CACjB,KAAM,YAAU,aAChB,SAAAd,EACA,MAAOnB,EAAM,KACf,CAAC,EACMC,EAAW,SAAS,EAE7B,IAAMiC,EAAiB,KAAK,OAAO,KAAK,OAAOf,EAAU,KAAK,UAAU,aAAcW,CAAO,EAE7F,KAAK,UAAW,aAAe,KAC/B,IAAIK,EAAoB,CAAC,EACrBC,EAAevB,EAEnB,GAAI,CACF,KAAK,cAAc,CACjB,KAAM,YAAU,YAChB,SAAAM,EACA,MAAO,KAAK,UAAW,EACzB,CAAC,EAED,YAAAkB,GAAAC,GAAsCJ,GAAtCK,GAAAC,GAAAC,GAAAF,GAAA,EAAAC,GAAA,MAAAH,GAAA,aAAAE,GAAA,GAAsD,CAA3C,IAAIG,EAAfF,GAAA,MAEE,GAAI,CAACV,EAAQ,WAAW,SAASY,EAAoB,KAAmB,EACtE,SAUF,IAAMC,EAAQD,EAEd,GAAIA,EAAoB,QAAU,QAAS,CACzC,KAAK,cAAc,CACjB,KAAM,YAAU,UAChB,QAASA,EAAoB,KAAK,QAClC,SAAUA,CACZ,CAAC,EACD,KACF,CAEA,GAAIA,EAAoB,QAAU,UAAW,SAE7C,GAAIA,EAAoB,QAAU,SAAU,CAC1CP,EAAoBQ,EAAM,KAC1B,QACF,CAEA,IAAMC,EAAYD,EAAM,KAClBE,EAAkBD,EAAU,SAAS,eACrCE,EAAYF,EAAU,MACtBG,GAAWH,EAAU,SAwC3B,GAvCA,KAAK,UAAW,GAAKG,GAAS,OAE1BF,GAAmBA,IAAoB,KAAK,UAAW,WACrD,KAAK,UAAW,UAClB,KAAK,cAAc,CACjB,KAAM,YAAU,cAChB,SAAU,KAAK,UAAW,QAC5B,CAAC,EAGCA,IACF,KAAK,cAAc,CACjB,KAAM,YAAU,aAChB,SAAUA,CACZ,CAAC,EACD,KAAK,UAAW,SAAWA,IAI/B3B,EACEA,GACC4B,IAAc,mBACbF,EAAU,OAAS,OAEvB,KAAK,UAAW,YACd,KAAK,UAAW,WAAaC,GAC7BC,IAAc,eACZ,KAAK,UAAW,cAClB,KAAK,UAAW,qBAAuB,MAKrCnB,EAAU,MAAS,KAAMqB,IAASA,GAAK,KAAOH,CAAe,IAC/D,KAAK,UAAW,SAAWA,GAG7BT,GAAe7B,EAAA,KAAK,UAAW,uBAAhB,KAAAA,EAAwC4B,EAEnD,CAAC,KAAK,UAAW,SACnB,UAGmB,KAAK,UAAUC,CAAY,IAAM,KAAK,UAAUvB,CAAK,GAItE,KAAK,UAAW,cAAgB,KAAK,UAAW,UAChD,KAAK,UAAW,cAClB,CAAS,KAAK,qBAAqB,KAAK,UAAW,EAAE,IAErDA,EAAQuB,EACR,KAAK,UAAW,aAAe,KAAK,UAAW,SAE/C,KAAK,cAAc,CACjB,KAAM,YAAU,eAChB,SAAU,KAAK,iBAAiBvB,CAAK,EACrC,SAAU8B,CACZ,CAAC,GAGH,KAAK,cAAc,CACjB,KAAM,YAAU,IAChB,MAAOC,CACT,CAAC,EAED,KAAK,kBAAkBA,EAAW/B,CAAK,CACzC,QAtGA2B,GA/ON,CA+OMC,GAAA,CAAAD,WAAA,KAAAD,KAAAC,GAAAH,GAAA,eAAAG,GAAA,KAAAH,WAAA,IAAAI,GAAA,MAAAA,GAAA,IAwGA5B,EAAQ,MAAM,KAAK,OAAO,QAAQ,SAASM,CAAQ,EACnD,IAAMa,GAActB,GAAAD,GAAAD,EAAAK,EAAM,QAAN,YAAAL,EAAc,KAAd,YAAAC,EAAkB,aAAlB,KAAAC,EAAgC,CAAC,EACrD,YAAK,UAAW,SAAWsB,EACvB,KAAK,UAAW,SAChB,OAAO,KAAKnB,EAAM,SAAS,MAAM,EAAE,CAAC,EAExCmB,EAAW,QAASC,GAAc,CAChC,KAAK,cAAc,CACjB,KAAM,YAAU,OAChB,oBACA,MACE,OAAOA,EAAU,OAAU,SAAWA,EAAU,MAAQ,KAAK,UAAUA,EAAU,KAAK,EACxF,SAAUA,CACZ,CAAC,CACH,CAAC,EAED,KAAK,cAAc,CACjB,KAAM,YAAU,eAChB,SAAU,KAAK,iBAAiBpB,EAAM,MAAM,CAC9C,CAAC,EACD,KAAK,cAAc,CACjB,KAAM,YAAU,kBAChB,SAAUoC,IAAwBtC,EAAAE,EAAM,OAAO,WAAb,KAAAF,EAAyB,CAAC,CAAC,CAC/D,CAAC,EACG,KAAK,UAAW,UAClB,KAAK,cAAc,CACjB,KAAM,YAAU,cAChB,SAAU,KAAK,UAAW,QAC5B,CAAC,EAEH,KAAK,cAAc,CACjB,KAAM,YAAU,aAChB,SAAAQ,EACA,MAAO,KAAK,UAAW,EACzB,CAAC,EACMlB,EAAW,SAAS,CAC7B,OAASiD,EAAG,CACV,OAAOjD,EAAW,MAAMiD,CAAC,CAC3B,CACF,CAEA,kBAAkBnD,EAAYc,EAAoB,CAhYpD,IAAAlB,EAAAC,EAAAM,EAAAC,EAAAC,EAAAC,EAiYI,OAAQN,EAAM,MAAO,CACnB,2BACE,IAAIoD,GAAqBxD,EAAAI,EAAM,SAAS,eAAe,IAA9B,KAAAJ,EAAmC,GACxDyD,GAAsBxD,EAAAG,EAAM,SAAS,iBAAiB,IAAhC,KAAAH,EAAqC,GAE/D,GAAIG,EAAM,KAAK,MAAM,kBAAkB,cAAe,OACtD,IAAMsD,EAAgB,KAAK,qBAAqB,KAAK,UAAW,EAAE,EAC5DC,EAAmB,GAAQD,GAAA,MAAAA,EAAe,IAC1CE,GAAerD,EAAAH,EAAM,KAAK,MAAM,mBAAjB,YAAAG,EAAoC,GACnDsD,GAA6BrD,EAAAJ,EAAM,SAAS,gBAAf,YAAAI,EAAiC,KACjEsD,GAAuCA,EAAiB,QAASF,GAAA,YAAAA,EAAc,OAG5EG,EAAuB,CAACJ,IAAoBC,GAAA,YAAAA,EAAc,MAC1DI,EACJL,IAAoBD,GAAA,YAAAA,EAAe,aAAcE,EAAa,KAC1DK,EAAqBN,IAAoBD,GAAA,YAAAA,EAAe,aAAc,CAACE,EAEvEM,EAAsB,CAACP,GAAoB,CAACC,EAC5CO,EAAwBR,GAAoB,CAACC,EAC7CQ,EACJT,GAAoB,EAACD,GAAA,MAAAA,EAAe,aAAc,CAACS,EAUrD,GARIN,GACF,KAAK,cAAc,CACjB,KAAM,YAAU,OAChB,KAAM,eACN,MAAOzD,EAAM,SAAS,aACxB,CAAC,EAGC6D,EAAoB,CACL,KAAK,cAAc,CAClC,KAAM,YAAU,cAChB,WAAYP,EAAc,WAC1B,SAAUtD,CACZ,CAAC,IAEC,KAAK,kBAAkB,KAAK,UAAW,EAAE,EAAI,MAE/C,KACF,CAEA,GAAIgE,EAAmB,CACJ,KAAK,cAAc,CAClC,KAAM,YAAU,iBAChB,UAAWV,EAAe,GAC1B,SAAUtD,CACZ,CAAC,IAEC,KAAK,kBAAkB,KAAK,UAAW,EAAE,EAAI,MAE/C,KACF,CAEA,GAAI2D,GAAwBN,EAAqB,CAC9B,KAAK,cAAc,CAClC,KAAM,YAAU,gBAChB,WAAYG,EAAa,GACzB,aAAcA,EAAa,KAC3B,gBAAiBxD,EAAM,KAAK,MAAM,GAClC,SAAUA,CACZ,CAAC,GAEC,KAAK,qBAAqB,KAAK,UAAW,GAAI,CAC5C,GAAIA,EAAM,KAAK,MAAM,GACrB,WAAYwD,EAAa,GACzB,aAAcA,EAAa,IAC7B,CAAC,EAEH,KACF,CAGA,GAAII,GAAuBP,EAAqB,CAC9C,KAAK,cAAc,CACjB,KAAM,YAAU,eAChB,WAAYC,EAAc,WAC1B,MAAOE,EAAa,KACpB,SAAUxD,CACZ,CAAC,EACD,KACF,CAGA,GAAI8D,GAAuBV,EAAoB,CAC5B,KAAK,cAAc,CAClC,KAAM,YAAU,mBAChB,KAAM,YACN,UAAWpD,EAAM,KAAK,MAAM,GAC5B,SAAUA,CACZ,CAAC,GAEC,KAAK,qBAAqB,KAAK,UAAW,GAAI,CAC5C,GAAIA,EAAM,KAAK,MAAM,GACrB,WAAY,KACZ,aAAc,IAChB,CAAC,EAEH,KACF,CAEA,GAAI+D,GAAyBX,EAAoB,CAC/C,KAAK,cAAc,CACjB,KAAM,YAAU,qBAChB,UAAWE,EAAe,GAC1B,MAAOtD,EAAM,KAAK,MAAM,QACxB,SAAUA,CACZ,CAAC,EACD,KACF,CAEA,MACF,wBACE,IAAIK,EAAA,KAAK,qBAAqB,KAAK,UAAW,EAAE,IAA5C,MAAAA,EAA+C,WAAY,CAC5C,KAAK,cAAc,CAClC,KAAM,YAAU,cAChB,WAAY,KAAK,qBAAqB,KAAK,UAAW,EAAE,EAAG,WAC3D,SAAUL,CACZ,CAAC,IAEC,KAAK,kBAAkB,KAAK,UAAW,EAAE,EAAI,MAE/C,KACF,CACA,IAAIM,EAAA,KAAK,qBAAqB,KAAK,UAAW,EAAE,IAA5C,MAAAA,EAA+C,GAAI,CACpC,KAAK,cAAc,CAClC,KAAM,YAAU,iBAChB,UAAW,KAAK,qBAAqB,KAAK,UAAW,EAAE,EAAG,GAC1D,SAAUN,CACZ,CAAC,IAEC,KAAK,kBAAkB,KAAK,UAAW,EAAE,EAAI,MAE/C,KACF,CACA,MACF,sBACE,GAAIA,EAAM,OAAS,wBAAsC,CACvD,KAAK,cAAc,CACjB,KAAM,YAAU,mBAChB,KAAM,YACN,UAAWA,EAAM,KAAK,WACtB,SAAUA,CACZ,CAAC,EACD,KAAK,cAAc,CACjB,KAAM,YAAU,qBAChB,UAAWA,EAAM,KAAK,WACtB,MAAOA,EAAM,KAAK,QAClB,SAAUA,CACZ,CAAC,EACD,KAAK,cAAc,CACjB,KAAM,YAAU,iBAChB,UAAWA,EAAM,KAAK,WACtB,SAAUA,CACZ,CAAC,EACD,KACF,CAEA,GAAIA,EAAM,OAAS,0BAAuC,CACxD,KAAK,cAAc,CACjB,KAAM,YAAU,gBAChB,WAAYA,EAAM,KAAK,GACvB,aAAcA,EAAM,KAAK,KACzB,gBAAiBA,EAAM,KAAK,GAC5B,SAAUA,CACZ,CAAC,EACD,KAAK,cAAc,CACjB,KAAM,YAAU,eAChB,WAAYA,EAAM,KAAK,GACvB,MAAOA,EAAM,KAAK,KAClB,SAAUA,CACZ,CAAC,EACD,KAAK,cAAc,CACjB,KAAM,YAAU,cAChB,WAAYA,EAAM,KAAK,GACvB,SAAUA,CACZ,CAAC,EACD,KACF,CAEIA,EAAM,OAAS,wBACjB,KAAK,UAAW,qBAAuBA,EAAM,KAC7C,KAAK,cAAc,CACjB,KAAM,YAAU,eAChB,SAAU,KAAK,iBAAiBc,CAAK,EACrC,SAAUd,CACZ,CAAC,GAGH,KAAK,cAAc,CACjB,KAAM,YAAU,OAChB,KAAMA,EAAM,KACZ,MAAOA,EAAM,KACb,SAAUA,CACZ,CAAC,EACD,KACJ,CACF,CAEA,iBAAiBc,EAAc,CAC7B,IAAMmD,EAAa,KAAK,UAAW,WAEnC,OAAIA,GAAA,MAAAA,EAAY,SACdnD,EAAQoD,EAAyBpD,EAAO,CAAC,GAAGqD,EAAqB,GAAGF,EAAW,MAAM,CAAC,GAGjFnD,CACT,CAEA,MAAM,gCAAgCM,EAA4C,CAChF,IAAIC,EAAa,CAAE,OAAQ,CAAC,CAAE,EAC9B,GAAI,CACF,MAAM,KAAK,OAAO,QAAQ,IAAID,CAAQ,EACtCC,EAAa,MAAM,KAAK,OAAO,QAAQ,SAASD,CAAQ,CAC1D,OAASsB,EAAO,CACd,MAAM,KAAK,OAAO,QAAQ,OAAO,CAAE,SAAAtB,CAAS,CAAC,CAC/C,CAEA,OAAOC,CACT,CAEA,MAAM,aAAa,CACjB,QAAA+C,EACA,UAAAC,EACA,WAAAJ,CACF,EAIG,CACD,OAAOG,EAAQ,OAAO,CAACE,EAAKC,IAAQ,CAxmBxC,IAAA3E,EAymBM,IAAI4E,EAAuBF,EAAI,aAE3BC,EAAI,eACNC,EAAuBP,GAAA,MAAAA,EAAY,OAC/BC,EAAyBK,GAAA,YAAAA,EAAK,aAAc,CAC1C,GAAGJ,EACH,IAAIvE,EAAAqE,GAAA,YAAAA,EAAY,SAAZ,KAAArE,EAAsB,CAAC,CAC7B,CAAC,EACD2E,GAAA,YAAAA,EAAK,cAGX,IAAME,EAAYzC,EAAAjC,IAAA,GACbuE,GACAC,GAFa,CAGhB,aAAcC,CAChB,GAGME,EACJJ,EAAI,iBAAmB,MAAQC,EAAI,kBAAoB,GAEnDI,EAAsB,KAAK,UAAUF,CAAS,IAAM,KAAK,UAAUH,CAAG,EAGtEM,EACJF,GACA,KAAK,UAAU1C,EAAAjC,EAAA,GAAK0E,GAAL,CAAgB,gBAAiB,IAAK,EAAC,IACpD,KAAK,UAAUzC,EAAAjC,EAAA,GAAKuE,GAAL,CAAU,gBAAiB,IAAK,EAAC,EAEpD,OAAIK,GAAuB,CAACC,EACnB7E,IAAA,GACFuE,GACAG,GAIAH,CACT,EAAGD,EAAU,MAAM,CACrB,CAEA,qBAAqBQ,EAAe,CAClC,OAAO,KAAK,kBAAkBA,CAAK,CACrC,CAEA,qBAAqBA,EAAeC,EAAyB,CAC3D,KAAK,kBAAoB9C,EAAAjC,EAAA,GACpB,KAAK,mBADe,CAEvB,CAAC8E,CAAK,EAAG9E,IAAA,GACH,KAAK,kBAAkB8E,CAAK,GAC7BC,EAEP,EACF,CAEA,MAAM,cAAmC,CACvC,IAAMC,EAAa,MAAM,KAAK,OAAO,WAAW,OAAO,EACjDC,EAAqBD,EAAW,KACnCE,GACCA,EAAa,eAAiB,KAAK,SAAWA,EAAa,WAAa,KAAK,OACjF,EACA,GAAI,CAACD,EACH,cAAQ,MAAM;AAAA,qCACiB,KAAK,OAAO,gBAAgB,KAAK,OAAO;AAAA;AAAA;AAAA,yCAEpCD,EAAW,IAAKG,GAAM,GAAGA,EAAE,QAAQ,SAASA,EAAE,YAAY,GAAG,EAAE,KAAK,IAAI,CAAC;AAAA,OAC3G,EACK,IAAI,MAAM,mBAAmB,EAGrC,OAAOF,CACT,CAEA,MAAM,eAAqC,CAjrB7C,IAAApF,EAAAC,EAAAM,EAkrBI,IAAMgF,EAAgB,CAAC,UAAU,EAEjC,GAAI,CACF,IAAMC,EAAc,MAAM,KAAK,OAAO,WAAW,WAAW,KAAK,UAAW,YAAY,EACpFC,EAAe,KAInB,IAHIzF,EAAAwF,EAAY,gBAAZ,MAAAxF,EAA2B,aAC7ByF,EAAe,OAAO,KAAKD,EAAY,cAAc,UAAU,GAE7D,GAACvF,EAAAuF,EAAY,eAAZ,MAAAvF,EAA0B,aAAc,GAACM,EAAAiF,EAAY,gBAAZ,MAAAjF,EAA2B,YACvE,MAAO,CAAE,OAAQ,CAAC,EAAG,MAAOgF,EAAe,OAAQA,CAAc,EAEnE,IAAMG,EAAc,OAAO,KAAKF,EAAY,aAAa,UAAU,EAC7DG,EAAe,OAAO,KAAKH,EAAY,cAAc,UAAU,EAErE,MAAO,CACL,MAAOE,GAAeA,EAAY,OAAS,CAAC,GAAGA,EAAa,GAAGH,CAAa,EAAI,KAChF,OAAQI,GAAgBA,EAAa,OAAS,CAAC,GAAGA,EAAc,GAAGJ,CAAa,EAAI,KACpF,OAAQE,CACV,CACF,OAASlC,EAAG,CACV,MAAO,CAAE,OAAQ,CAAC,EAAG,MAAOgC,EAAe,OAAQA,CAAc,CACnE,CACF,CAEA,2BAA2BrE,EAAcC,EAA8BC,EAAmB,CA1sB5F,IAAApB,EA2sBQmB,EAAS,OAAS,GAAK,SAAUA,EAAS,CAAC,GAAKA,EAAS,CAAC,EAAE,OAAS,WAEvEA,EAAWA,EAAS,MAAM,CAAC,GAI7B,IAAMyE,EAA+C1E,EAAM,UAAY,CAAC,EAClE2E,EAAqB,IAAI,IAAID,EAAiB,IAAKE,GAAYA,EAAQ,EAAE,CAAC,EAC1EC,EAAa,IAAI,IAAI5E,EAAS,IAAK2E,GAAYA,EAAQ,EAAE,CAAC,EAE5DE,EAAmC,CAAC,EACpC7E,EAAS,OAASyE,EAAiB,SAErCI,EAAkBJ,EACf,OAAQK,GAAM,CAACF,EAAW,IAAIE,EAAE,EAAE,CAAC,EACnC,IAAKA,GAAM,IAAI,iBAAc,CAAE,GAAIA,EAAE,EAAI,CAAC,CAAC,GAGhD,IAAMC,EAAc/E,EAAS,OAAQ2E,GAAY,CAACD,EAAmB,IAAIC,EAAQ,EAAE,CAAC,EAEpF,OAAO1D,EAAAjC,EAAA,GACFe,GADE,CAEL,SAAU,CAAC,GAAG8E,EAAiB,GAAGE,CAAW,EAC7C,MAAO,CAAC,IAAIlG,EAAAkB,EAAM,QAAN,KAAAlB,EAAe,CAAC,EAAI,GAAGoB,CAAK,CAC1C,EACF,CACF","names":["index_exports","__export","CustomEventNames","LangGraphAgent","LangGraphEventTypes","__toCommonJS","import_rxjs","import_langgraph_sdk","import_node_crypto","import_messages","LangGraphEventTypes","CustomEventNames","import_client","DEFAULT_SCHEMA_KEYS","filterObjectBySchemaKeys","obj","schemaKeys","key","getStreamPayloadInput","mode","state","input","langchainMessagesToAgui","messages","message","_a","stringifyIfNeeded","tc","aguiMessagesToLangChain","index","_b","item","LangGraphAgent","config","_a","_b","LangGraphClient","__spreadValues","event","input","subscriber","_c","_d","_e","_f","_g","_h","_i","_j","_k","_l","inputThreadId","state","messages","tools","context","forwardedProps","shouldExit","threadId","agentState","agentStateValues","aguiToLangChainMessage","aguiMessagesToLangChain","mode","payloadInput","getStreamPayloadInput","graphInfo","payloadConfig","configsToMerge","payload","__spreadProps","interrupts","interrupt","streamResponse","latestStateValues","updatedState","iter","__forAwait","more","temp","error","streamResponseChunk","chunk","chunkData","currentNodeName","eventType","metadata","node","langchainMessagesToAgui","e","shouldEmitMessages","shouldEmitToolCalls","currentStream","hasCurrentStream","toolCallData","toolCallUsedToPredictState","predictStateTool","isToolCallStartEvent","isToolCallArgsEvent","isToolCallEndEvent","isMessageStartEvent","isMessageContentEvent","isMessageEndEvent","schemaKeys","filterObjectBySchemaKeys","DEFAULT_SCHEMA_KEYS","configs","assistant","acc","cfg","filteredConfigurable","newConfig","isRecursionLimitSetToDefault","configsAreDifferent","isOnlyRecursionLimitDifferent","runId","data","assistants","retrievedAssistant","searchResult","a","CONSTANT_KEYS","graphSchema","configSchema","inputSchema","outputSchema","existingMessages","existingMessageIds","message","messageIds","removedMessages","m","newMessages"]}
|
|
1
|
+
{"version":3,"sources":["../src/index.ts","../src/types.ts","../src/utils.ts"],"sourcesContent":["import { Observable, Subscriber } from \"rxjs\";\nimport {\n Client as LangGraphClient,\n EventsStreamEvent,\n StreamMode,\n Config as LangGraphConfig,\n ThreadState,\n Assistant,\n Message as LangGraphMessage,\n Config,\n Interrupt,\n Thread\n} from \"@langchain/langgraph-sdk\";\nimport { randomUUID } from \"node:crypto\";\nimport { RemoveMessage } from \"@langchain/core/messages\";\nimport {\n LangGraphPlatformMessage,\n CustomEventNames,\n LangGraphEventTypes,\n State,\n MessagesInProgressRecord,\n ThinkingInProgress,\n SchemaKeys,\n MessageInProgress,\n RunMetadata,\n PredictStateTool,\n LangGraphReasoning\n} from \"./types\";\nimport {\n AbstractAgent,\n AgentConfig,\n CustomEvent,\n EventType,\n MessagesSnapshotEvent,\n RawEvent,\n RunAgentInput,\n RunErrorEvent,\n RunFinishedEvent,\n RunStartedEvent,\n StateDeltaEvent,\n StateSnapshotEvent,\n StepFinishedEvent,\n StepStartedEvent,\n TextMessageContentEvent,\n TextMessageEndEvent,\n TextMessageStartEvent,\n ToolCallArgsEvent,\n ToolCallEndEvent,\n ToolCallStartEvent,\n ThinkingTextMessageStartEvent,\n ThinkingTextMessageContentEvent,\n ThinkingTextMessageEndEvent,\n ThinkingStartEvent,\n ThinkingEndEvent,\n} from \"@ag-ui/client\";\nimport { RunsStreamPayload } from \"@langchain/langgraph-sdk/dist/types\";\nimport {\n aguiMessagesToLangChain,\n DEFAULT_SCHEMA_KEYS,\n filterObjectBySchemaKeys,\n getStreamPayloadInput,\n langchainMessagesToAgui,\n resolveMessageContent,\n resolveReasoningContent\n} from \"@/utils\";\n\nexport type ProcessedEvents =\n | TextMessageStartEvent\n | TextMessageContentEvent\n | TextMessageEndEvent\n | ThinkingTextMessageStartEvent\n | ThinkingTextMessageContentEvent\n | ThinkingTextMessageEndEvent\n | ToolCallStartEvent\n | ToolCallArgsEvent\n | ToolCallEndEvent\n | ThinkingStartEvent\n | ThinkingEndEvent\n | StateSnapshotEvent\n | StateDeltaEvent\n | MessagesSnapshotEvent\n | RawEvent\n | CustomEvent\n | RunStartedEvent\n | RunFinishedEvent\n | RunErrorEvent\n | StepStartedEvent\n | StepFinishedEvent;\n\ntype RunAgentExtendedInput<\n TStreamMode extends StreamMode | StreamMode[] = StreamMode,\n TSubgraphs extends boolean = false,\n> = Omit<RunAgentInput, \"forwardedProps\"> & {\n forwardedProps?: Omit<RunsStreamPayload<TStreamMode, TSubgraphs>, \"input\"> & {\n nodeName?: string;\n };\n};\n\nexport interface LangGraphAgentConfig extends AgentConfig {\n client?: LangGraphClient;\n deploymentUrl: string;\n langsmithApiKey?: string;\n propertyHeaders?: Record<string, string>;\n assistantConfig?: LangGraphConfig;\n agentName?: string;\n graphId: string;\n}\n\nexport class LangGraphAgent extends AbstractAgent {\n client: LangGraphClient;\n assistantConfig?: LangGraphConfig;\n agentName?: string;\n graphId: string;\n assistant?: Assistant;\n messagesInProcess: MessagesInProgressRecord;\n thinkingProcess: null | ThinkingInProgress;\n activeRun?: RunMetadata;\n // @ts-expect-error no need to initialize subscriber right now\n subscriber: Subscriber<ProcessedEvents>;\n\n constructor(config: LangGraphAgentConfig) {\n super(config);\n this.messagesInProcess = {};\n this.agentName = config.agentName;\n this.graphId = config.graphId;\n this.assistantConfig = config.assistantConfig;\n this.thinkingProcess = null\n this.client =\n config?.client ??\n new LangGraphClient({\n apiUrl: config.deploymentUrl,\n apiKey: config.langsmithApiKey,\n defaultHeaders: { ...(config.propertyHeaders ?? {}) },\n });\n }\n\n dispatchEvent(event: ProcessedEvents) {\n this.subscriber.next(event);\n return true;\n }\n\n run(input: RunAgentInput) {\n this.activeRun = {\n id: input.runId,\n threadId: input.threadId,\n };\n return new Observable<ProcessedEvents>((subscriber) => {\n this.handleStreamEvents(input, subscriber);\n return () => {};\n });\n }\n\n async handleStreamEvents(input: RunAgentExtendedInput, subscriber: Subscriber<ProcessedEvents>) {\n let { threadId: inputThreadId, state, messages, tools, context, forwardedProps } = input;\n this.subscriber = subscriber;\n let shouldExit = false;\n // If a manual emittance happens, it is the ultimate source of truth of state, unless a node has exited.\n // Therefore, this value should either hold null, or the only edition of state that should be used.\n this.activeRun!.manuallyEmittedState = null;\n\n this.activeRun!.nodeName = input.forwardedProps?.nodeName;\n\n const threadId = inputThreadId ?? randomUUID();\n\n if (!this.assistant) {\n this.assistant = await this.getAssistant();\n }\n\n const thread = await this.getOrCreateThread(threadId);\n this.activeRun!.threadId = thread.thread_id;\n const agentState = await this.client.threads.getState(thread.thread_id) ?? { values: {} } as ThreadState\n\n const agentStateValues = agentState.values as State;\n const aguiToLangChainMessage = aguiMessagesToLangChain(messages);\n\n state.messages = agentStateValues.messages;\n state = this.langGraphDefaultMergeState(state, aguiToLangChainMessage, tools);\n\n const mode =\n threadId && this.activeRun!.nodeName != \"__end__\" && this.activeRun!.nodeName\n ? \"continue\"\n : \"start\";\n\n if (mode === \"continue\" && !forwardedProps?.command?.resume) {\n await this.client.threads.updateState(threadId, {\n values: state,\n asNode: this.activeRun!.nodeName,\n });\n }\n\n this.activeRun!.schemaKeys = await this.getSchemaKeys();\n\n const payloadInput = getStreamPayloadInput({\n mode,\n state,\n schemaKeys: this.activeRun!.schemaKeys,\n });\n\n const graphInfo = await this.client.assistants.getGraph(this.assistant.assistant_id);\n\n let payloadConfig: LangGraphConfig | undefined;\n const configsToMerge = [this.assistantConfig, forwardedProps?.config].filter(\n Boolean,\n ) as LangGraphConfig[];\n if (configsToMerge.length) {\n payloadConfig = await this.mergeConfigs({\n configs: configsToMerge,\n assistant: this.assistant,\n schemaKeys: this.activeRun!.schemaKeys,\n });\n }\n const payload = {\n ...forwardedProps,\n streamMode:\n forwardedProps?.streamMode ?? ([\"events\", \"values\", \"updates\"] satisfies StreamMode[]),\n input: payloadInput,\n config: payloadConfig,\n };\n\n // If there are still outstanding unresolved interrupts, we must force resolution of them before moving forward\n const interrupts = (agentState.tasks?.[0]?.interrupts ?? []) as Interrupt[];\n if (interrupts?.length && !forwardedProps?.command?.resume) {\n this.dispatchEvent({\n type: EventType.RUN_STARTED,\n threadId,\n runId: input.runId,\n });\n\n interrupts.forEach((interrupt) => {\n this.dispatchEvent({\n type: EventType.CUSTOM,\n name: LangGraphEventTypes.OnInterrupt,\n value:\n typeof interrupt.value === \"string\" ? interrupt.value : JSON.stringify(interrupt.value),\n rawEvent: interrupt,\n });\n });\n\n this.dispatchEvent({\n type: EventType.RUN_FINISHED,\n threadId,\n runId: input.runId,\n });\n return subscriber.complete();\n }\n const streamResponse = this.client.runs.stream(threadId, this.assistant.assistant_id, payload);\n\n this.activeRun!.prevNodeName = null;\n let latestStateValues = {};\n let updatedState = state;\n\n try {\n this.dispatchEvent({\n type: EventType.RUN_STARTED,\n threadId,\n runId: this.activeRun!.id,\n });\n\n for await (let streamResponseChunk of streamResponse) {\n // @ts-ignore\n if (!payload.streamMode.includes(streamResponseChunk.event as StreamMode)) {\n continue;\n }\n // Force event type, as data is not properly defined on the LG side.\n type EventsChunkData = {\n __interrupt__?: any;\n metadata: Record<string, any>;\n event: string;\n data: any;\n [key: string]: unknown;\n };\n const chunk = streamResponseChunk as EventsStreamEvent & { data: EventsChunkData };\n\n if (streamResponseChunk.event === \"error\") {\n this.dispatchEvent({\n type: EventType.RUN_ERROR,\n message: streamResponseChunk.data.message,\n rawEvent: streamResponseChunk,\n });\n break;\n }\n\n if (streamResponseChunk.event === \"updates\") continue;\n\n if (streamResponseChunk.event === \"values\") {\n latestStateValues = chunk.data;\n continue;\n }\n\n const chunkData = chunk.data;\n const currentNodeName = chunkData.metadata.langgraph_node;\n const eventType = chunkData.event;\n const metadata = chunkData.metadata;\n this.activeRun!.id = metadata.run_id;\n\n if (currentNodeName && currentNodeName !== this.activeRun!.nodeName) {\n if (this.activeRun!.nodeName) {\n this.dispatchEvent({\n type: EventType.STEP_FINISHED,\n stepName: this.activeRun!.nodeName,\n });\n }\n\n if (currentNodeName) {\n this.dispatchEvent({\n type: EventType.STEP_STARTED,\n stepName: currentNodeName,\n });\n this.activeRun!.nodeName = currentNodeName;\n }\n }\n\n shouldExit =\n shouldExit ||\n (eventType === LangGraphEventTypes.OnCustomEvent &&\n chunkData.name === CustomEventNames.Exit);\n\n this.activeRun!.exitingNode =\n this.activeRun!.nodeName === currentNodeName &&\n eventType === LangGraphEventTypes.OnChainEnd;\n if (this.activeRun!.exitingNode) {\n this.activeRun!.manuallyEmittedState = null;\n }\n\n // we only want to update the node name under certain conditions\n // since we don't need any internal node names to be sent to the frontend\n if (graphInfo[\"nodes\"].some((node) => node.id === currentNodeName)) {\n this.activeRun!.nodeName = currentNodeName;\n }\n\n updatedState = this.activeRun!.manuallyEmittedState ?? latestStateValues;\n\n if (!this.activeRun!.nodeName) {\n continue;\n }\n\n const hasStateDiff = JSON.stringify(updatedState) !== JSON.stringify(state);\n // We should not update snapshot while a message is in progress.\n if (\n (hasStateDiff ||\n this.activeRun!.prevNodeName != this.activeRun!.nodeName ||\n this.activeRun!.exitingNode) &&\n !Boolean(this.getMessageInProgress(this.activeRun!.id))\n ) {\n state = updatedState;\n this.activeRun!.prevNodeName = this.activeRun!.nodeName;\n\n this.dispatchEvent({\n type: EventType.STATE_SNAPSHOT,\n snapshot: this.getStateSnapshot(state),\n rawEvent: chunk,\n });\n }\n\n this.dispatchEvent({\n type: EventType.RAW,\n event: chunkData,\n });\n\n this.handleSingleEvent(chunkData, state);\n }\n\n state = await this.client.threads.getState(threadId);\n const interrupts = (state.tasks?.[0]?.interrupts ?? []) as Interrupt[];\n this.activeRun!.nodeName = interrupts\n ? this.activeRun!.nodeName\n : Object.keys(state.metadata.writes)[0];\n\n interrupts.forEach((interrupt) => {\n this.dispatchEvent({\n type: EventType.CUSTOM,\n name: LangGraphEventTypes.OnInterrupt,\n value:\n typeof interrupt.value === \"string\" ? interrupt.value : JSON.stringify(interrupt.value),\n rawEvent: interrupt,\n });\n });\n\n this.dispatchEvent({\n type: EventType.STATE_SNAPSHOT,\n snapshot: this.getStateSnapshot(state.values),\n });\n this.dispatchEvent({\n type: EventType.MESSAGES_SNAPSHOT,\n messages: langchainMessagesToAgui(state.values.messages ?? []),\n });\n if (this.activeRun!.nodeName) {\n this.dispatchEvent({\n type: EventType.STEP_FINISHED,\n stepName: this.activeRun!.nodeName!,\n });\n }\n this.dispatchEvent({\n type: EventType.RUN_FINISHED,\n threadId,\n runId: this.activeRun!.id,\n });\n return subscriber.complete();\n } catch (e) {\n return subscriber.error(e);\n }\n }\n\n handleSingleEvent(event: any, state: State): void {\n switch (event.event) {\n case LangGraphEventTypes.OnChatModelStream:\n let shouldEmitMessages = event.metadata[\"emit-messages\"] ?? true;\n let shouldEmitToolCalls = event.metadata[\"emit-tool-calls\"] ?? true;\n\n if (event.data.chunk.response_metadata.finish_reason) return;\n let currentStream = this.getMessageInProgress(this.activeRun!.id);\n const hasCurrentStream = Boolean(currentStream?.id);\n const toolCallData = event.data.chunk.tool_call_chunks?.[0];\n const toolCallUsedToPredictState = event.metadata[\"predict_state\"]?.some(\n (predictStateTool: PredictStateTool) => predictStateTool.tool === toolCallData?.name,\n );\n\n const isToolCallStartEvent = !hasCurrentStream && toolCallData?.name;\n const isToolCallArgsEvent =\n hasCurrentStream && currentStream?.toolCallId && toolCallData?.args;\n const isToolCallEndEvent = hasCurrentStream && currentStream?.toolCallId && !toolCallData;\n\n const reasoningData = resolveReasoningContent(event.data);\n const messageContent = resolveMessageContent(event.data.chunk.content);\n const isMessageContentEvent = Boolean(!toolCallData && messageContent);\n\n const isMessageEndEvent =\n hasCurrentStream && !currentStream?.toolCallId && !isMessageContentEvent;\n\n if (reasoningData) {\n this.handleThinkingEvent(reasoningData)\n break;\n }\n\n if (!reasoningData && this.thinkingProcess) {\n this.dispatchEvent({\n type: EventType.THINKING_TEXT_MESSAGE_END,\n })\n this.dispatchEvent({\n type: EventType.THINKING_END,\n })\n this.thinkingProcess = null;\n }\n\n if (toolCallUsedToPredictState) {\n this.dispatchEvent({\n type: EventType.CUSTOM,\n name: \"PredictState\",\n value: event.metadata[\"predict_state\"],\n });\n }\n\n if (isToolCallEndEvent) {\n const resolved = this.dispatchEvent({\n type: EventType.TOOL_CALL_END,\n toolCallId: currentStream?.toolCallId!,\n rawEvent: event,\n });\n if (resolved) {\n this.messagesInProcess[this.activeRun!.id] = null;\n }\n break;\n }\n\n if (isMessageEndEvent) {\n const resolved = this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_END,\n messageId: currentStream!.id,\n rawEvent: event,\n });\n if (resolved) {\n this.messagesInProcess[this.activeRun!.id] = null;\n }\n break;\n }\n\n if (isToolCallStartEvent && shouldEmitToolCalls) {\n const resolved = this.dispatchEvent({\n type: EventType.TOOL_CALL_START,\n toolCallId: toolCallData.id,\n toolCallName: toolCallData.name,\n parentMessageId: event.data.chunk.id,\n rawEvent: event,\n });\n if (resolved) {\n this.setMessageInProgress(this.activeRun!.id, {\n id: event.data.chunk.id,\n toolCallId: toolCallData.id,\n toolCallName: toolCallData.name,\n });\n }\n break;\n }\n\n // Tool call args: emit ActionExecutionArgs\n if (isToolCallArgsEvent && shouldEmitToolCalls) {\n this.dispatchEvent({\n type: EventType.TOOL_CALL_ARGS,\n toolCallId: currentStream?.toolCallId!,\n delta: toolCallData.args,\n rawEvent: event,\n });\n break;\n }\n\n // Message content: emit TextMessageContent\n if (isMessageContentEvent && shouldEmitMessages) {\n // No existing message yet, also init the message\n if (!currentStream) {\n this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_START,\n role: \"assistant\",\n messageId: event.data.chunk.id,\n rawEvent: event,\n });\n this.setMessageInProgress(this.activeRun!.id, {\n id: event.data.chunk.id,\n toolCallId: null,\n toolCallName: null,\n });\n currentStream = this.getMessageInProgress(this.activeRun!.id);\n }\n\n this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_CONTENT,\n messageId: currentStream!.id,\n delta: messageContent!,\n rawEvent: event,\n });\n break;\n }\n\n break;\n case LangGraphEventTypes.OnChatModelEnd:\n if (this.getMessageInProgress(this.activeRun!.id)?.toolCallId) {\n const resolved = this.dispatchEvent({\n type: EventType.TOOL_CALL_END,\n toolCallId: this.getMessageInProgress(this.activeRun!.id)!.toolCallId!,\n rawEvent: event,\n });\n if (resolved) {\n this.messagesInProcess[this.activeRun!.id] = null;\n }\n break;\n }\n if (this.getMessageInProgress(this.activeRun!.id)?.id) {\n const resolved = this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_END,\n messageId: this.getMessageInProgress(this.activeRun!.id)!.id,\n rawEvent: event,\n });\n if (resolved) {\n this.messagesInProcess[this.activeRun!.id] = null;\n }\n break;\n }\n break;\n case LangGraphEventTypes.OnCustomEvent:\n if (event.name === CustomEventNames.ManuallyEmitMessage) {\n this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_START,\n role: \"assistant\",\n messageId: event.data.message_id,\n rawEvent: event,\n });\n this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_CONTENT,\n messageId: event.data.message_id,\n delta: event.data.message,\n rawEvent: event,\n });\n this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_END,\n messageId: event.data.message_id,\n rawEvent: event,\n });\n break;\n }\n\n if (event.name === CustomEventNames.ManuallyEmitToolCall) {\n this.dispatchEvent({\n type: EventType.TOOL_CALL_START,\n toolCallId: event.data.id,\n toolCallName: event.data.name,\n parentMessageId: event.data.id,\n rawEvent: event,\n });\n this.dispatchEvent({\n type: EventType.TOOL_CALL_ARGS,\n toolCallId: event.data.id,\n delta: event.data.args,\n rawEvent: event,\n });\n this.dispatchEvent({\n type: EventType.TOOL_CALL_END,\n toolCallId: event.data.id,\n rawEvent: event,\n });\n break;\n }\n\n if (event.name === CustomEventNames.ManuallyEmitState) {\n this.activeRun!.manuallyEmittedState = event.data;\n this.dispatchEvent({\n type: EventType.STATE_SNAPSHOT,\n snapshot: this.getStateSnapshot(state),\n rawEvent: event,\n });\n }\n\n this.dispatchEvent({\n type: EventType.CUSTOM,\n name: event.name,\n value: event.data,\n rawEvent: event,\n });\n break;\n }\n }\n\n handleThinkingEvent(reasoningData: LangGraphReasoning) {\n if (!reasoningData || !reasoningData.type || !reasoningData.text) {\n return;\n }\n\n const thinkingStepIndex = reasoningData.index;\n\n if (this.thinkingProcess?.index && this.thinkingProcess.index !== thinkingStepIndex) {\n if (this.thinkingProcess.type) {\n this.dispatchEvent({\n type: EventType.THINKING_TEXT_MESSAGE_END,\n })\n }\n this.dispatchEvent({\n type: EventType.THINKING_END,\n })\n this.thinkingProcess = null;\n }\n\n if (!this.thinkingProcess) {\n // No thinking step yet. Start a new one\n this.dispatchEvent({\n type: EventType.THINKING_START,\n })\n this.thinkingProcess = {\n index: thinkingStepIndex,\n };\n }\n\n\n if (this.thinkingProcess.type !== reasoningData.type) {\n this.dispatchEvent({\n type: EventType.THINKING_TEXT_MESSAGE_START,\n })\n this.thinkingProcess.type = reasoningData.type\n }\n\n if (this.thinkingProcess.type) {\n this.dispatchEvent({\n type: EventType.THINKING_TEXT_MESSAGE_CONTENT,\n delta: reasoningData.text\n })\n }\n }\n\n getStateSnapshot(state: State) {\n const schemaKeys = this.activeRun!.schemaKeys!;\n // Do not emit state keys that are not part of the output schema\n if (schemaKeys?.output) {\n state = filterObjectBySchemaKeys(state, [...DEFAULT_SCHEMA_KEYS, ...schemaKeys.output]);\n }\n // return state\n return state;\n }\n\n async getOrCreateThread(threadId: string): Promise<Thread> {\n let thread: Thread;\n try {\n try {\n thread = await this.getThread(threadId);\n } catch (error) {\n thread = await this.createThread({ threadId });\n }\n } catch (error: unknown) {\n throw new Error(`Failed to create thread: ${(error as Error).message}`);\n }\n\n return thread;\n }\n\n async getThread(threadId: string) {\n return this.client.threads.get(threadId);\n }\n\n async createThread(payload?: Parameters<typeof this.client.threads.create>[0]) {\n return this.client.threads.create(payload);\n }\n\n async mergeConfigs({\n configs,\n assistant,\n schemaKeys,\n }: {\n configs: Config[];\n assistant: Assistant;\n schemaKeys: SchemaKeys;\n }) {\n return configs.reduce((acc, cfg) => {\n let filteredConfigurable = acc.configurable;\n\n if (cfg.configurable) {\n filteredConfigurable = schemaKeys?.config\n ? filterObjectBySchemaKeys(cfg?.configurable, [\n ...DEFAULT_SCHEMA_KEYS,\n ...(schemaKeys?.config ?? []),\n ])\n : cfg?.configurable;\n }\n\n const newConfig = {\n ...acc,\n ...cfg,\n configurable: filteredConfigurable,\n };\n\n // LG does not return recursion limit if it's the default, therefore we check: if no recursion limit is currently set, and the user asked for 25, there is no change.\n const isRecursionLimitSetToDefault =\n acc.recursion_limit == null && cfg.recursion_limit === 25;\n // Deep compare configs to avoid unnecessary update calls\n const configsAreDifferent = JSON.stringify(newConfig) !== JSON.stringify(acc);\n\n // Check if the only difference is the recursion_limit being set to default\n const isOnlyRecursionLimitDifferent =\n isRecursionLimitSetToDefault &&\n JSON.stringify({ ...newConfig, recursion_limit: null }) ===\n JSON.stringify({ ...acc, recursion_limit: null });\n\n if (configsAreDifferent && !isOnlyRecursionLimitDifferent) {\n return {\n ...acc,\n ...newConfig,\n };\n }\n\n return acc;\n }, assistant.config);\n }\n\n getMessageInProgress(runId: string) {\n return this.messagesInProcess[runId];\n }\n\n setMessageInProgress(runId: string, data: MessageInProgress) {\n this.messagesInProcess = {\n ...this.messagesInProcess,\n [runId]: {\n ...(this.messagesInProcess[runId] as MessageInProgress),\n ...data,\n },\n };\n }\n\n async getAssistant(): Promise<Assistant> {\n const assistants = await this.client.assistants.search();\n const retrievedAssistant = assistants.find(\n (searchResult) =>\n searchResult.graph_id === this.graphId,\n );\n if (!retrievedAssistant) {\n console.error(`\n No agent found with graph ID ${this.graphId} found..\\n\n \n These are the available agents: [${assistants.map((a) => `${a.graph_id} (ID: ${a.assistant_id})`).join(\", \")}]\n `);\n throw new Error(\"No agent id found\");\n }\n\n return retrievedAssistant;\n }\n\n async getSchemaKeys(): Promise<SchemaKeys> {\n const CONSTANT_KEYS = [\"messages\"];\n\n try {\n const graphSchema = await this.client.assistants.getSchemas(this.assistant!.assistant_id);\n let configSchema = null;\n if (graphSchema.config_schema?.properties) {\n configSchema = Object.keys(graphSchema.config_schema.properties);\n }\n if (!graphSchema.input_schema?.properties || !graphSchema.output_schema?.properties) {\n return { config: [], input: CONSTANT_KEYS, output: CONSTANT_KEYS };\n }\n const inputSchema = Object.keys(graphSchema.input_schema.properties);\n const outputSchema = Object.keys(graphSchema.output_schema.properties);\n\n return {\n input: inputSchema && inputSchema.length ? [...inputSchema, ...CONSTANT_KEYS] : null,\n output: outputSchema && outputSchema.length ? [...outputSchema, ...CONSTANT_KEYS] : null,\n config: configSchema,\n };\n } catch (e) {\n return { config: [], input: CONSTANT_KEYS, output: CONSTANT_KEYS };\n }\n }\n\n langGraphDefaultMergeState(state: State, messages: LangGraphMessage[], tools: any): State {\n if (messages.length > 0 && \"role\" in messages[0] && messages[0].role === \"system\") {\n // remove system message\n messages = messages.slice(1);\n }\n\n // merge with existing messages\n const existingMessages: LangGraphPlatformMessage[] = state.messages || [];\n const existingMessageIds = new Set(existingMessages.map((message) => message.id));\n const messageIds = new Set(messages.map((message) => message.id));\n\n let removedMessages: RemoveMessage[] = [];\n if (messages.length < existingMessages.length) {\n // Messages were removed\n removedMessages = existingMessages\n .filter((m) => !messageIds.has(m.id))\n .map((m) => new RemoveMessage({ id: m.id! }));\n }\n\n const newMessages = messages.filter((message) => !existingMessageIds.has(message.id));\n\n return {\n ...state,\n messages: [...removedMessages, ...newMessages],\n tools: [...(state.tools ?? []), ...tools],\n };\n }\n}\n\nexport * from \"./types\";\n","import { Message } from \"@langchain/langgraph-sdk\";\nimport { MessageType } from \"@langchain/core/messages\";\n\nexport enum LangGraphEventTypes {\n OnChainStart = \"on_chain_start\",\n OnChainStream = \"on_chain_stream\",\n OnChainEnd = \"on_chain_end\",\n OnChatModelStart = \"on_chat_model_start\",\n OnChatModelStream = \"on_chat_model_stream\",\n OnChatModelEnd = \"on_chat_model_end\",\n OnToolStart = \"on_tool_start\",\n OnToolEnd = \"on_tool_end\",\n OnCustomEvent = \"on_custom_event\",\n OnInterrupt = \"on_interrupt\",\n}\n\nexport type State = Record<string, any>;\n\nexport type SchemaKeys = {\n input: string[] | null;\n output: string[] | null;\n config: string[] | null;\n} | null;\n\nexport type MessageInProgress = {\n id: string;\n toolCallId?: string | null;\n toolCallName?: string | null;\n};\n\nexport type ThinkingInProgress = {\n index: number;\n type?: LangGraphReasoning['type'];\n}\n\nexport interface RunMetadata {\n id: string;\n schemaKeys?: SchemaKeys;\n nodeName?: string;\n prevNodeName?: string | null;\n exitingNode?: boolean;\n manuallyEmittedState?: State | null;\n threadId?: string;\n}\n\nexport type MessagesInProgressRecord = Record<string, MessageInProgress | null>;\n\n// The following types are our own definition to the messages accepted by LangGraph Platform, enhanced with some of our extra data.\nexport interface ToolCall {\n id: string;\n name: string;\n args: Record<string, unknown>;\n}\n\ntype BaseLangGraphPlatformMessage = Omit<\n Message,\n | \"isResultMessage\"\n | \"isTextMessage\"\n | \"isImageMessage\"\n | \"isActionExecutionMessage\"\n | \"isAgentStateMessage\"\n | \"type\"\n | \"createdAt\"\n> & {\n content: string;\n role: string;\n additional_kwargs?: Record<string, unknown>;\n type: MessageType;\n};\n\ninterface LangGraphPlatformResultMessage extends BaseLangGraphPlatformMessage {\n tool_call_id: string;\n name: string;\n}\n\ninterface LangGraphPlatformActionExecutionMessage extends BaseLangGraphPlatformMessage {\n tool_calls: ToolCall[];\n}\n\nexport type LangGraphPlatformMessage =\n | LangGraphPlatformActionExecutionMessage\n | LangGraphPlatformResultMessage\n | BaseLangGraphPlatformMessage;\n\nexport enum CustomEventNames {\n ManuallyEmitMessage = \"manually_emit_message\",\n ManuallyEmitToolCall = \"manually_emit_tool_call\",\n ManuallyEmitState = \"manually_emit_state\",\n Exit = \"exit\",\n}\n\nexport interface PredictStateTool {\n tool: string;\n state_key: string;\n tool_argument: string;\n}\n\nexport interface LangGraphReasoning {\n type: 'text';\n text: string;\n index: number\n}\n","import { Message as LangGraphMessage } from \"@langchain/langgraph-sdk\";\nimport { State, SchemaKeys, LangGraphReasoning } from \"./types\";\nimport { Message, ToolCall } from \"@ag-ui/client\";\n\nexport const DEFAULT_SCHEMA_KEYS = [\"tools\"];\n\nexport function filterObjectBySchemaKeys(obj: Record<string, any>, schemaKeys: string[]) {\n return Object.fromEntries(Object.entries(obj).filter(([key]) => schemaKeys.includes(key)));\n}\n\nexport function getStreamPayloadInput({\n mode,\n state,\n schemaKeys,\n}: {\n mode: \"start\" | \"continue\";\n state: State;\n schemaKeys: SchemaKeys;\n}) {\n let input = mode === \"start\" ? state : null;\n // Do not input keys that are not part of the input schema\n if (input && schemaKeys?.input) {\n input = filterObjectBySchemaKeys(input, [...DEFAULT_SCHEMA_KEYS, ...schemaKeys.input]);\n }\n\n return input;\n}\n\nexport function langchainMessagesToAgui(messages: LangGraphMessage[]): Message[] {\n return messages.map((message) => {\n switch (message.type) {\n case \"human\":\n return {\n id: message.id!,\n role: \"user\",\n content: stringifyIfNeeded(resolveMessageContent(message.content)),\n };\n case \"ai\":\n const content = resolveMessageContent(message.content)\n return {\n id: message.id!,\n role: \"assistant\",\n content: content ? stringifyIfNeeded(content) : '',\n toolCalls: message.tool_calls?.map((tc) => ({\n id: tc.id!,\n type: \"function\",\n function: {\n name: tc.name,\n arguments: JSON.stringify(tc.args),\n },\n })),\n };\n case \"system\":\n return {\n id: message.id!,\n role: \"system\",\n content: stringifyIfNeeded(resolveMessageContent(message.content)),\n };\n case \"tool\":\n return {\n id: message.id!,\n role: \"tool\",\n content: stringifyIfNeeded(resolveMessageContent(message.content)),\n toolCallId: message.tool_call_id,\n };\n default:\n throw new Error(\"message type returned from LangGraph is not supported.\");\n }\n });\n}\n\nexport function aguiMessagesToLangChain(messages: Message[]): LangGraphMessage[] {\n return messages.map((message, index) => {\n switch (message.role) {\n case \"user\":\n return {\n id: message.id,\n role: message.role,\n content: message.content,\n type: \"human\",\n };\n case \"assistant\":\n return {\n id: message.id,\n type: \"ai\",\n role: message.role,\n content: message.content ?? \"\",\n tool_calls: (message.toolCalls ?? []).map((tc: ToolCall) => ({\n id: tc.id,\n name: tc.function.name,\n args: JSON.parse(tc.function.arguments),\n type: \"tool_call\",\n })),\n };\n case \"system\":\n return {\n id: message.id,\n role: message.role,\n content: message.content,\n type: \"system\",\n };\n case \"tool\":\n return {\n content: message.content,\n role: message.role,\n type: message.role,\n tool_call_id: message.toolCallId,\n id: message.id,\n };\n default:\n console.error(`Message role ${message.role} is not implemented`);\n throw new Error(\"message role is not supported.\");\n }\n });\n}\n\nfunction stringifyIfNeeded(item: any) {\n if (typeof item === \"string\") return item;\n return JSON.stringify(item);\n}\n\nexport function resolveReasoningContent(eventData: any): LangGraphReasoning | null {\n const content = eventData.chunk?.content\n\n // Anthropic reasoning response\n if (content && Array.isArray(content) && content.length && content[0]) {\n if (!content[0].thinking) return null\n return {\n text: content[0].thinking,\n type: 'text',\n index: content[0].index,\n }\n }\n\n /// OpenAI reasoning response\n if (eventData.chunk.additional_kwargs?.reasoning?.summary?.[0]) {\n const data = eventData.chunk.additional_kwargs?.reasoning.summary[0]\n if (!data || !data.text) return null\n return {\n type: 'text',\n text: data.text,\n index: data.index,\n }\n }\n\n return null\n}\n\nexport function resolveMessageContent(content?: LangGraphMessage['content']): string | null {\n if (!content) return null;\n\n if (typeof content === 'string') {\n return content;\n }\n\n if (Array.isArray(content) && content.length) {\n const contentText = content.find(c => c.type === 'text')?.text\n return contentText ?? null;\n }\n\n return null\n}\n"],"mappings":"4jCAAA,IAAAA,GAAA,GAAAC,GAAAD,GAAA,sBAAAE,GAAA,mBAAAC,EAAA,wBAAAC,KAAA,eAAAC,GAAAL,IAAA,IAAAM,GAAuC,gBACvCC,GAWO,oCACPC,GAA2B,kBAC3BC,GAA8B,oCCXvB,IAAKC,QACVA,EAAA,aAAe,iBACfA,EAAA,cAAgB,kBAChBA,EAAA,WAAa,eACbA,EAAA,iBAAmB,sBACnBA,EAAA,kBAAoB,uBACpBA,EAAA,eAAiB,oBACjBA,EAAA,YAAc,gBACdA,EAAA,UAAY,cACZA,EAAA,cAAgB,kBAChBA,EAAA,YAAc,eAVJA,QAAA,IAiFAC,QACVA,EAAA,oBAAsB,wBACtBA,EAAA,qBAAuB,0BACvBA,EAAA,kBAAoB,sBACpBA,EAAA,KAAO,OAJGA,QAAA,IDxDZ,IAAAC,EA0BO,yBElDA,IAAMC,EAAsB,CAAC,OAAO,EAEpC,SAASC,EAAyBC,EAA0BC,EAAsB,CACvF,OAAO,OAAO,YAAY,OAAO,QAAQD,CAAG,EAAE,OAAO,CAAC,CAACE,CAAG,IAAMD,EAAW,SAASC,CAAG,CAAC,CAAC,CAC3F,CAEO,SAASC,GAAsB,CACpC,KAAAC,EACA,MAAAC,EACA,WAAAJ,CACF,EAIG,CACD,IAAIK,EAAQF,IAAS,QAAUC,EAAQ,KAEvC,OAAIC,IAASL,GAAA,MAAAA,EAAY,SACvBK,EAAQP,EAAyBO,EAAO,CAAC,GAAGR,EAAqB,GAAGG,EAAW,KAAK,CAAC,GAGhFK,CACT,CAEO,SAASC,GAAwBC,EAAyC,CAC/E,OAAOA,EAAS,IAAKC,GAAY,CA7BnC,IAAAC,EA8BI,OAAQD,EAAQ,KAAM,CACpB,IAAK,QACH,MAAO,CACL,GAAIA,EAAQ,GACZ,KAAM,OACN,QAASE,EAAkBC,EAAsBH,EAAQ,OAAO,CAAC,CACnE,EACF,IAAK,KACH,IAAMI,EAAUD,EAAsBH,EAAQ,OAAO,EACrD,MAAO,CACL,GAAIA,EAAQ,GACZ,KAAM,YACN,QAASI,EAAUF,EAAkBE,CAAO,EAAI,GAChD,WAAWH,EAAAD,EAAQ,aAAR,YAAAC,EAAoB,IAAKI,IAAQ,CAC1C,GAAIA,EAAG,GACP,KAAM,WACN,SAAU,CACR,KAAMA,EAAG,KACT,UAAW,KAAK,UAAUA,EAAG,IAAI,CACnC,CACF,GACF,EACF,IAAK,SACH,MAAO,CACL,GAAIL,EAAQ,GACZ,KAAM,SACN,QAASE,EAAkBC,EAAsBH,EAAQ,OAAO,CAAC,CACnE,EACF,IAAK,OACH,MAAO,CACL,GAAIA,EAAQ,GACZ,KAAM,OACN,QAASE,EAAkBC,EAAsBH,EAAQ,OAAO,CAAC,EACjE,WAAYA,EAAQ,YACtB,EACF,QACE,MAAM,IAAI,MAAM,wDAAwD,CAC5E,CACF,CAAC,CACH,CAEO,SAASM,GAAwBP,EAAyC,CAC/E,OAAOA,EAAS,IAAI,CAACC,EAASO,IAAU,CAxE1C,IAAAN,EAAAO,EAyEI,OAAQR,EAAQ,KAAM,CACpB,IAAK,OACH,MAAO,CACL,GAAIA,EAAQ,GACZ,KAAMA,EAAQ,KACd,QAASA,EAAQ,QACjB,KAAM,OACR,EACF,IAAK,YACH,MAAO,CACL,GAAIA,EAAQ,GACZ,KAAM,KACN,KAAMA,EAAQ,KACd,SAASC,EAAAD,EAAQ,UAAR,KAAAC,EAAmB,GAC5B,aAAaO,EAAAR,EAAQ,YAAR,KAAAQ,EAAqB,CAAC,GAAG,IAAKH,IAAkB,CAC3D,GAAIA,EAAG,GACP,KAAMA,EAAG,SAAS,KAClB,KAAM,KAAK,MAAMA,EAAG,SAAS,SAAS,EACtC,KAAM,WACR,EAAE,CACJ,EACF,IAAK,SACH,MAAO,CACL,GAAIL,EAAQ,GACZ,KAAMA,EAAQ,KACd,QAASA,EAAQ,QACjB,KAAM,QACR,EACF,IAAK,OACH,MAAO,CACL,QAASA,EAAQ,QACjB,KAAMA,EAAQ,KACd,KAAMA,EAAQ,KACd,aAAcA,EAAQ,WACtB,GAAIA,EAAQ,EACd,EACF,QACE,cAAQ,MAAM,gBAAgBA,EAAQ,IAAI,qBAAqB,EACzD,IAAI,MAAM,gCAAgC,CACpD,CACF,CAAC,CACH,CAEA,SAASE,EAAkBO,EAAW,CACpC,OAAI,OAAOA,GAAS,SAAiBA,EAC9B,KAAK,UAAUA,CAAI,CAC5B,CAEO,SAASC,GAAwBC,EAA2C,CAzHnF,IAAAV,EAAAO,EAAAI,EAAAC,EAAAC,EA0HE,IAAMV,GAAUH,EAAAU,EAAU,QAAV,YAAAV,EAAiB,QAGjC,GAAIG,GAAW,MAAM,QAAQA,CAAO,GAAKA,EAAQ,QAAUA,EAAQ,CAAC,EAClE,OAAKA,EAAQ,CAAC,EAAE,SACT,CACL,KAAMA,EAAQ,CAAC,EAAE,SACjB,KAAM,OACN,MAAOA,EAAQ,CAAC,EAAE,KACpB,EALiC,KASnC,IAAIS,GAAAD,GAAAJ,EAAAG,EAAU,MAAM,oBAAhB,YAAAH,EAAmC,YAAnC,YAAAI,EAA8C,UAA9C,MAAAC,EAAwD,GAAI,CAC9D,IAAME,GAAOD,EAAAH,EAAU,MAAM,oBAAhB,YAAAG,EAAmC,UAAU,QAAQ,GAClE,MAAI,CAACC,GAAQ,CAACA,EAAK,KAAa,KACzB,CACL,KAAM,OACN,KAAMA,EAAK,KACX,MAAOA,EAAK,KACd,CACF,CAEA,OAAO,IACT,CAEO,SAASZ,EAAsBC,EAAsD,CApJ5F,IAAAH,EAqJE,GAAI,CAACG,EAAS,OAAO,KAErB,GAAI,OAAOA,GAAY,SACrB,OAAOA,EAGT,GAAI,MAAM,QAAQA,CAAO,GAAKA,EAAQ,OAAQ,CAC5C,IAAMY,GAAcf,EAAAG,EAAQ,KAAKa,GAAKA,EAAE,OAAS,MAAM,IAAnC,YAAAhB,EAAsC,KAC1D,OAAOe,GAAA,KAAAA,EAAe,IACxB,CAEA,OAAO,IACT,CFrDO,IAAME,EAAN,cAA6B,eAAc,CAYhD,YAAYC,EAA8B,CAxH5C,IAAAC,EAAAC,EAyHI,MAAMF,CAAM,EACZ,KAAK,kBAAoB,CAAC,EAC1B,KAAK,UAAYA,EAAO,UACxB,KAAK,QAAUA,EAAO,QACtB,KAAK,gBAAkBA,EAAO,gBAC9B,KAAK,gBAAkB,KACvB,KAAK,QACHE,EAAAF,GAAA,YAAAA,EAAQ,SAAR,KAAAE,EACA,IAAI,GAAAC,OAAgB,CAClB,OAAQH,EAAO,cACf,OAAQA,EAAO,gBACf,eAAgBI,EAAA,IAAMH,EAAAD,EAAO,kBAAP,KAAAC,EAA0B,CAAC,EACnD,CAAC,CACL,CAEA,cAAcI,EAAwB,CACpC,YAAK,WAAW,KAAKA,CAAK,EACnB,EACT,CAEA,IAAIC,EAAsB,CACxB,YAAK,UAAY,CACf,GAAIA,EAAM,MACV,SAAUA,EAAM,QAClB,EACO,IAAI,cAA6BC,IACtC,KAAK,mBAAmBD,EAAOC,CAAU,EAClC,IAAM,CAAC,EACf,CACH,CAEA,MAAM,mBAAmBD,EAA8BC,EAAyC,CAxJlG,IAAAN,EAAAC,EAAAM,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAyJI,GAAI,CAAE,SAAUC,EAAe,MAAAC,EAAO,SAAAC,EAAU,MAAAC,EAAO,QAAAC,EAAS,eAAAC,CAAe,EAAIlB,EACnF,KAAK,WAAaC,EAClB,IAAIkB,EAAa,GAGjB,KAAK,UAAW,qBAAuB,KAEvC,KAAK,UAAW,UAAWxB,EAAAK,EAAM,iBAAN,YAAAL,EAAsB,SAEjD,IAAMyB,EAAWP,GAAA,KAAAA,KAAiB,eAAW,EAExC,KAAK,YACR,KAAK,UAAY,MAAM,KAAK,aAAa,GAG3C,IAAMQ,EAAS,MAAM,KAAK,kBAAkBD,CAAQ,EACpD,KAAK,UAAW,SAAWC,EAAO,UAClC,IAAMC,GAAa1B,EAAA,MAAM,KAAK,OAAO,QAAQ,SAASyB,EAAO,SAAS,IAAnD,KAAAzB,EAAwD,CAAE,OAAQ,CAAC,CAAE,EAElF2B,EAAmBD,EAAW,OAC9BE,EAAyBC,GAAwBV,CAAQ,EAE/DD,EAAM,SAAWS,EAAiB,SAClCT,EAAQ,KAAK,2BAA2BA,EAAOU,EAAwBR,CAAK,EAE5E,IAAMU,EACJN,GAAY,KAAK,UAAW,UAAY,WAAa,KAAK,UAAW,SACjE,WACA,QAEFM,IAAS,YAAc,GAACxB,EAAAgB,GAAA,YAAAA,EAAgB,UAAhB,MAAAhB,EAAyB,SACnD,MAAM,KAAK,OAAO,QAAQ,YAAYkB,EAAU,CAC9C,OAAQN,EACR,OAAQ,KAAK,UAAW,QAC1B,CAAC,EAGH,KAAK,UAAW,WAAa,MAAM,KAAK,cAAc,EAEtD,IAAMa,EAAeC,GAAsB,CACzC,KAAAF,EACA,MAAAZ,EACA,WAAY,KAAK,UAAW,UAC9B,CAAC,EAEKe,EAAY,MAAM,KAAK,OAAO,WAAW,SAAS,KAAK,UAAU,YAAY,EAE/EC,EACEC,EAAiB,CAAC,KAAK,gBAAiBb,GAAA,YAAAA,EAAgB,MAAM,EAAE,OACpE,OACF,EACIa,EAAe,SACjBD,EAAgB,MAAM,KAAK,aAAa,CACtC,QAASC,EACT,UAAW,KAAK,UAChB,WAAY,KAAK,UAAW,UAC9B,CAAC,GAEH,IAAMC,EAAUC,EAAAnC,EAAA,GACXoB,GADW,CAEd,YACEf,EAAAe,GAAA,YAAAA,EAAgB,aAAhB,KAAAf,EAA+B,CAAC,SAAU,SAAU,SAAS,EAC/D,MAAOwB,EACP,OAAQG,CACV,GAGMI,GAAc5B,GAAAD,GAAAD,EAAAkB,EAAW,QAAX,YAAAlB,EAAmB,KAAnB,YAAAC,EAAuB,aAAvB,KAAAC,EAAqC,CAAC,EAC1D,GAAI4B,GAAA,MAAAA,EAAY,QAAU,GAAC3B,EAAAW,GAAA,YAAAA,EAAgB,UAAhB,MAAAX,EAAyB,QAClD,YAAK,cAAc,CACjB,KAAM,YAAU,YAChB,SAAAa,EACA,MAAOpB,EAAM,KACf,CAAC,EAEDkC,EAAW,QAASC,GAAc,CAChC,KAAK,cAAc,CACjB,KAAM,YAAU,OAChB,oBACA,MACE,OAAOA,EAAU,OAAU,SAAWA,EAAU,MAAQ,KAAK,UAAUA,EAAU,KAAK,EACxF,SAAUA,CACZ,CAAC,CACH,CAAC,EAED,KAAK,cAAc,CACjB,KAAM,YAAU,aAChB,SAAAf,EACA,MAAOpB,EAAM,KACf,CAAC,EACMC,EAAW,SAAS,EAE7B,IAAMmC,EAAiB,KAAK,OAAO,KAAK,OAAOhB,EAAU,KAAK,UAAU,aAAcY,CAAO,EAE7F,KAAK,UAAW,aAAe,KAC/B,IAAIK,EAAoB,CAAC,EACrBC,EAAexB,EAEnB,GAAI,CACF,KAAK,cAAc,CACjB,KAAM,YAAU,YAChB,SAAAM,EACA,MAAO,KAAK,UAAW,EACzB,CAAC,EAED,YAAAmB,GAAAC,GAAsCJ,GAAtCK,GAAAC,GAAAC,GAAAF,GAAA,EAAAC,GAAA,MAAAH,GAAA,aAAAE,GAAA,GAAsD,CAA3C,IAAIG,EAAfF,GAAA,MAEE,GAAI,CAACV,EAAQ,WAAW,SAASY,EAAoB,KAAmB,EACtE,SAUF,IAAMC,EAAQD,EAEd,GAAIA,EAAoB,QAAU,QAAS,CACzC,KAAK,cAAc,CACjB,KAAM,YAAU,UAChB,QAASA,EAAoB,KAAK,QAClC,SAAUA,CACZ,CAAC,EACD,KACF,CAEA,GAAIA,EAAoB,QAAU,UAAW,SAE7C,GAAIA,EAAoB,QAAU,SAAU,CAC1CP,EAAoBQ,EAAM,KAC1B,QACF,CAEA,IAAMC,EAAYD,EAAM,KAClBE,EAAkBD,EAAU,SAAS,eACrCE,EAAYF,EAAU,MACtBG,GAAWH,EAAU,SAwC3B,GAvCA,KAAK,UAAW,GAAKG,GAAS,OAE1BF,GAAmBA,IAAoB,KAAK,UAAW,WACrD,KAAK,UAAW,UAClB,KAAK,cAAc,CACjB,KAAM,YAAU,cAChB,SAAU,KAAK,UAAW,QAC5B,CAAC,EAGCA,IACF,KAAK,cAAc,CACjB,KAAM,YAAU,aAChB,SAAUA,CACZ,CAAC,EACD,KAAK,UAAW,SAAWA,IAI/B5B,EACEA,GACC6B,IAAc,mBACbF,EAAU,OAAS,OAEvB,KAAK,UAAW,YACd,KAAK,UAAW,WAAaC,GAC7BC,IAAc,eACZ,KAAK,UAAW,cAClB,KAAK,UAAW,qBAAuB,MAKrCnB,EAAU,MAAS,KAAMqB,IAASA,GAAK,KAAOH,CAAe,IAC/D,KAAK,UAAW,SAAWA,GAG7BT,GAAe9B,EAAA,KAAK,UAAW,uBAAhB,KAAAA,EAAwC6B,EAEnD,CAAC,KAAK,UAAW,SACnB,UAGmB,KAAK,UAAUC,CAAY,IAAM,KAAK,UAAUxB,CAAK,GAItE,KAAK,UAAW,cAAgB,KAAK,UAAW,UAChD,KAAK,UAAW,cAClB,CAAS,KAAK,qBAAqB,KAAK,UAAW,EAAE,IAErDA,EAAQwB,EACR,KAAK,UAAW,aAAe,KAAK,UAAW,SAE/C,KAAK,cAAc,CACjB,KAAM,YAAU,eAChB,SAAU,KAAK,iBAAiBxB,CAAK,EACrC,SAAU+B,CACZ,CAAC,GAGH,KAAK,cAAc,CACjB,KAAM,YAAU,IAChB,MAAOC,CACT,CAAC,EAED,KAAK,kBAAkBA,EAAWhC,CAAK,CACzC,QAtGA4B,GAlQN,CAkQMC,GAAA,CAAAD,WAAA,KAAAD,KAAAC,GAAAH,GAAA,eAAAG,GAAA,KAAAH,WAAA,IAAAI,GAAA,MAAAA,GAAA,IAwGA7B,EAAQ,MAAM,KAAK,OAAO,QAAQ,SAASM,CAAQ,EACnD,IAAMc,GAAcvB,GAAAD,GAAAD,EAAAK,EAAM,QAAN,YAAAL,EAAc,KAAd,YAAAC,EAAkB,aAAlB,KAAAC,EAAgC,CAAC,EACrD,YAAK,UAAW,SAAWuB,EACvB,KAAK,UAAW,SAChB,OAAO,KAAKpB,EAAM,SAAS,MAAM,EAAE,CAAC,EAExCoB,EAAW,QAASC,GAAc,CAChC,KAAK,cAAc,CACjB,KAAM,YAAU,OAChB,oBACA,MACE,OAAOA,EAAU,OAAU,SAAWA,EAAU,MAAQ,KAAK,UAAUA,EAAU,KAAK,EACxF,SAAUA,CACZ,CAAC,CACH,CAAC,EAED,KAAK,cAAc,CACjB,KAAM,YAAU,eAChB,SAAU,KAAK,iBAAiBrB,EAAM,MAAM,CAC9C,CAAC,EACD,KAAK,cAAc,CACjB,KAAM,YAAU,kBAChB,SAAUqC,IAAwBvC,EAAAE,EAAM,OAAO,WAAb,KAAAF,EAAyB,CAAC,CAAC,CAC/D,CAAC,EACG,KAAK,UAAW,UAClB,KAAK,cAAc,CACjB,KAAM,YAAU,cAChB,SAAU,KAAK,UAAW,QAC5B,CAAC,EAEH,KAAK,cAAc,CACjB,KAAM,YAAU,aAChB,SAAAQ,EACA,MAAO,KAAK,UAAW,EACzB,CAAC,EACMnB,EAAW,SAAS,CAC7B,OAASmD,EAAG,CACV,OAAOnD,EAAW,MAAMmD,CAAC,CAC3B,CACF,CAEA,kBAAkBrD,EAAYe,EAAoB,CAnZpD,IAAAnB,EAAAC,EAAAM,EAAAC,EAAAC,EAAAC,EAoZI,OAAQN,EAAM,MAAO,CACnB,2BACE,IAAIsD,GAAqB1D,EAAAI,EAAM,SAAS,eAAe,IAA9B,KAAAJ,EAAmC,GACxD2D,GAAsB1D,EAAAG,EAAM,SAAS,iBAAiB,IAAhC,KAAAH,EAAqC,GAE/D,GAAIG,EAAM,KAAK,MAAM,kBAAkB,cAAe,OACtD,IAAIwD,EAAgB,KAAK,qBAAqB,KAAK,UAAW,EAAE,EAC1DC,EAAmB,GAAQD,GAAA,MAAAA,EAAe,IAC1CE,GAAevD,EAAAH,EAAM,KAAK,MAAM,mBAAjB,YAAAG,EAAoC,GACnDwD,GAA6BvD,EAAAJ,EAAM,SAAS,gBAAf,YAAAI,EAAiC,KACjEwD,GAAuCA,EAAiB,QAASF,GAAA,YAAAA,EAAc,OAG5EG,EAAuB,CAACJ,IAAoBC,GAAA,YAAAA,EAAc,MAC1DI,EACJL,IAAoBD,GAAA,YAAAA,EAAe,cAAcE,GAAA,YAAAA,EAAc,MAC3DK,EAAqBN,IAAoBD,GAAA,YAAAA,EAAe,aAAc,CAACE,EAEvEM,EAAgBC,GAAwBjE,EAAM,IAAI,EAClDkE,EAAiBC,EAAsBnE,EAAM,KAAK,MAAM,OAAO,EAC/DoE,EAAwB,GAAQ,CAACV,GAAgBQ,GAEjDG,EACJZ,GAAoB,EAACD,GAAA,MAAAA,EAAe,aAAc,CAACY,EAErD,GAAIJ,EAAe,CACjB,KAAK,oBAAoBA,CAAa,EACtC,KACF,CAoBA,GAlBI,CAACA,GAAiB,KAAK,kBACzB,KAAK,cAAc,CACjB,KAAM,YAAU,yBAClB,CAAC,EACD,KAAK,cAAc,CACjB,KAAM,YAAU,YAClB,CAAC,EACD,KAAK,gBAAkB,MAGrBL,GACF,KAAK,cAAc,CACjB,KAAM,YAAU,OAChB,KAAM,eACN,MAAO3D,EAAM,SAAS,aACxB,CAAC,EAGC+D,EAAoB,CACL,KAAK,cAAc,CAClC,KAAM,YAAU,cAChB,WAAYP,GAAA,YAAAA,EAAe,WAC3B,SAAUxD,CACZ,CAAC,IAEC,KAAK,kBAAkB,KAAK,UAAW,EAAE,EAAI,MAE/C,KACF,CAEA,GAAIqE,EAAmB,CACJ,KAAK,cAAc,CAClC,KAAM,YAAU,iBAChB,UAAWb,EAAe,GAC1B,SAAUxD,CACZ,CAAC,IAEC,KAAK,kBAAkB,KAAK,UAAW,EAAE,EAAI,MAE/C,KACF,CAEA,GAAI6D,GAAwBN,EAAqB,CAC9B,KAAK,cAAc,CAClC,KAAM,YAAU,gBAChB,WAAYG,EAAa,GACzB,aAAcA,EAAa,KAC3B,gBAAiB1D,EAAM,KAAK,MAAM,GAClC,SAAUA,CACZ,CAAC,GAEC,KAAK,qBAAqB,KAAK,UAAW,GAAI,CAC5C,GAAIA,EAAM,KAAK,MAAM,GACrB,WAAY0D,EAAa,GACzB,aAAcA,EAAa,IAC7B,CAAC,EAEH,KACF,CAGA,GAAII,GAAuBP,EAAqB,CAC9C,KAAK,cAAc,CACjB,KAAM,YAAU,eAChB,WAAYC,GAAA,YAAAA,EAAe,WAC3B,MAAOE,EAAa,KACpB,SAAU1D,CACZ,CAAC,EACD,KACF,CAGA,GAAIoE,GAAyBd,EAAoB,CAE1CE,IACH,KAAK,cAAc,CACjB,KAAM,YAAU,mBAChB,KAAM,YACN,UAAWxD,EAAM,KAAK,MAAM,GAC5B,SAAUA,CACZ,CAAC,EACD,KAAK,qBAAqB,KAAK,UAAW,GAAI,CAC5C,GAAIA,EAAM,KAAK,MAAM,GACrB,WAAY,KACZ,aAAc,IAChB,CAAC,EACDwD,EAAgB,KAAK,qBAAqB,KAAK,UAAW,EAAE,GAG9D,KAAK,cAAc,CACjB,KAAM,YAAU,qBAChB,UAAWA,EAAe,GAC1B,MAAOU,EACP,SAAUlE,CACZ,CAAC,EACD,KACF,CAEA,MACF,wBACE,IAAIK,EAAA,KAAK,qBAAqB,KAAK,UAAW,EAAE,IAA5C,MAAAA,EAA+C,WAAY,CAC5C,KAAK,cAAc,CAClC,KAAM,YAAU,cAChB,WAAY,KAAK,qBAAqB,KAAK,UAAW,EAAE,EAAG,WAC3D,SAAUL,CACZ,CAAC,IAEC,KAAK,kBAAkB,KAAK,UAAW,EAAE,EAAI,MAE/C,KACF,CACA,IAAIM,EAAA,KAAK,qBAAqB,KAAK,UAAW,EAAE,IAA5C,MAAAA,EAA+C,GAAI,CACpC,KAAK,cAAc,CAClC,KAAM,YAAU,iBAChB,UAAW,KAAK,qBAAqB,KAAK,UAAW,EAAE,EAAG,GAC1D,SAAUN,CACZ,CAAC,IAEC,KAAK,kBAAkB,KAAK,UAAW,EAAE,EAAI,MAE/C,KACF,CACA,MACF,sBACE,GAAIA,EAAM,OAAS,wBAAsC,CACvD,KAAK,cAAc,CACjB,KAAM,YAAU,mBAChB,KAAM,YACN,UAAWA,EAAM,KAAK,WACtB,SAAUA,CACZ,CAAC,EACD,KAAK,cAAc,CACjB,KAAM,YAAU,qBAChB,UAAWA,EAAM,KAAK,WACtB,MAAOA,EAAM,KAAK,QAClB,SAAUA,CACZ,CAAC,EACD,KAAK,cAAc,CACjB,KAAM,YAAU,iBAChB,UAAWA,EAAM,KAAK,WACtB,SAAUA,CACZ,CAAC,EACD,KACF,CAEA,GAAIA,EAAM,OAAS,0BAAuC,CACxD,KAAK,cAAc,CACjB,KAAM,YAAU,gBAChB,WAAYA,EAAM,KAAK,GACvB,aAAcA,EAAM,KAAK,KACzB,gBAAiBA,EAAM,KAAK,GAC5B,SAAUA,CACZ,CAAC,EACD,KAAK,cAAc,CACjB,KAAM,YAAU,eAChB,WAAYA,EAAM,KAAK,GACvB,MAAOA,EAAM,KAAK,KAClB,SAAUA,CACZ,CAAC,EACD,KAAK,cAAc,CACjB,KAAM,YAAU,cAChB,WAAYA,EAAM,KAAK,GACvB,SAAUA,CACZ,CAAC,EACD,KACF,CAEIA,EAAM,OAAS,wBACjB,KAAK,UAAW,qBAAuBA,EAAM,KAC7C,KAAK,cAAc,CACjB,KAAM,YAAU,eAChB,SAAU,KAAK,iBAAiBe,CAAK,EACrC,SAAUf,CACZ,CAAC,GAGH,KAAK,cAAc,CACjB,KAAM,YAAU,OAChB,KAAMA,EAAM,KACZ,MAAOA,EAAM,KACb,SAAUA,CACZ,CAAC,EACD,KACJ,CACF,CAEA,oBAAoBgE,EAAmC,CA5mBzD,IAAApE,EA6mBI,GAAI,CAACoE,GAAiB,CAACA,EAAc,MAAQ,CAACA,EAAc,KAC1D,OAGF,IAAMM,EAAoBN,EAAc,OAEpCpE,EAAA,KAAK,kBAAL,MAAAA,EAAsB,OAAS,KAAK,gBAAgB,QAAU0E,IAC5D,KAAK,gBAAgB,MACvB,KAAK,cAAc,CACjB,KAAM,YAAU,yBAClB,CAAC,EAEH,KAAK,cAAc,CACjB,KAAM,YAAU,YAClB,CAAC,EACD,KAAK,gBAAkB,MAGpB,KAAK,kBAER,KAAK,cAAc,CACjB,KAAM,YAAU,cAClB,CAAC,EACD,KAAK,gBAAkB,CACrB,MAAOA,CACT,GAIE,KAAK,gBAAgB,OAASN,EAAc,OAC9C,KAAK,cAAc,CACjB,KAAM,YAAU,2BAClB,CAAC,EACD,KAAK,gBAAgB,KAAOA,EAAc,MAGxC,KAAK,gBAAgB,MACvB,KAAK,cAAc,CACjB,KAAM,YAAU,8BAChB,MAAOA,EAAc,IACvB,CAAC,CAEL,CAEA,iBAAiBjD,EAAc,CAC7B,IAAMwD,EAAa,KAAK,UAAW,WAEnC,OAAIA,GAAA,MAAAA,EAAY,SACdxD,EAAQyD,EAAyBzD,EAAO,CAAC,GAAG0D,EAAqB,GAAGF,EAAW,MAAM,CAAC,GAGjFxD,CACT,CAEA,MAAM,kBAAkBM,EAAmC,CACzD,IAAIC,EACJ,GAAI,CACF,GAAI,CACFA,EAAS,MAAM,KAAK,UAAUD,CAAQ,CACxC,OAASuB,EAAO,CACdtB,EAAS,MAAM,KAAK,aAAa,CAAE,SAAAD,CAAS,CAAC,CAC/C,CACF,OAASuB,EAAgB,CACvB,MAAM,IAAI,MAAM,4BAA6BA,EAAgB,OAAO,EAAE,CACxE,CAEA,OAAOtB,CACT,CAEA,MAAM,UAAUD,EAAkB,CAChC,OAAO,KAAK,OAAO,QAAQ,IAAIA,CAAQ,CACzC,CAEA,MAAM,aAAaY,EAA4D,CAC7E,OAAO,KAAK,OAAO,QAAQ,OAAOA,CAAO,CAC3C,CAEA,MAAM,aAAa,CACjB,QAAAyC,EACA,UAAAC,EACA,WAAAJ,CACF,EAIG,CACD,OAAOG,EAAQ,OAAO,CAACE,EAAKC,IAAQ,CAnsBxC,IAAAjF,EAosBM,IAAIkF,EAAuBF,EAAI,aAE3BC,EAAI,eACNC,EAAuBP,GAAA,MAAAA,EAAY,OAC/BC,EAAyBK,GAAA,YAAAA,EAAK,aAAc,CAC1C,GAAGJ,EACH,IAAI7E,EAAA2E,GAAA,YAAAA,EAAY,SAAZ,KAAA3E,EAAsB,CAAC,CAC7B,CAAC,EACDiF,GAAA,YAAAA,EAAK,cAGX,IAAME,EAAY7C,EAAAnC,IAAA,GACb6E,GACAC,GAFa,CAGhB,aAAcC,CAChB,GAGME,EACJJ,EAAI,iBAAmB,MAAQC,EAAI,kBAAoB,GAEnDI,EAAsB,KAAK,UAAUF,CAAS,IAAM,KAAK,UAAUH,CAAG,EAGtEM,EACJF,GACA,KAAK,UAAU9C,EAAAnC,EAAA,GAAKgF,GAAL,CAAgB,gBAAiB,IAAK,EAAC,IACpD,KAAK,UAAU7C,EAAAnC,EAAA,GAAK6E,GAAL,CAAU,gBAAiB,IAAK,EAAC,EAEpD,OAAIK,GAAuB,CAACC,EACnBnF,IAAA,GACF6E,GACAG,GAIAH,CACT,EAAGD,EAAU,MAAM,CACrB,CAEA,qBAAqBQ,EAAe,CAClC,OAAO,KAAK,kBAAkBA,CAAK,CACrC,CAEA,qBAAqBA,EAAeC,EAAyB,CAC3D,KAAK,kBAAoBlD,EAAAnC,EAAA,GACpB,KAAK,mBADe,CAEvB,CAACoF,CAAK,EAAGpF,IAAA,GACH,KAAK,kBAAkBoF,CAAK,GAC7BC,EAEP,EACF,CAEA,MAAM,cAAmC,CACvC,IAAMC,EAAa,MAAM,KAAK,OAAO,WAAW,OAAO,EACjDC,EAAqBD,EAAW,KACnCE,GACCA,EAAa,WAAa,KAAK,OACnC,EACA,GAAI,CAACD,EACH,cAAQ,MAAM;AAAA,qCACiB,KAAK,OAAO;AAAA;AAAA;AAAA,yCAERD,EAAW,IAAKG,GAAM,GAAGA,EAAE,QAAQ,SAASA,EAAE,YAAY,GAAG,EAAE,KAAK,IAAI,CAAC;AAAA,OAC3G,EACK,IAAI,MAAM,mBAAmB,EAGrC,OAAOF,CACT,CAEA,MAAM,eAAqC,CA5wB7C,IAAA1F,EAAAC,EAAAM,EA6wBI,IAAMsF,EAAgB,CAAC,UAAU,EAEjC,GAAI,CACF,IAAMC,EAAc,MAAM,KAAK,OAAO,WAAW,WAAW,KAAK,UAAW,YAAY,EACpFC,EAAe,KAInB,IAHI/F,EAAA8F,EAAY,gBAAZ,MAAA9F,EAA2B,aAC7B+F,EAAe,OAAO,KAAKD,EAAY,cAAc,UAAU,GAE7D,GAAC7F,EAAA6F,EAAY,eAAZ,MAAA7F,EAA0B,aAAc,GAACM,EAAAuF,EAAY,gBAAZ,MAAAvF,EAA2B,YACvE,MAAO,CAAE,OAAQ,CAAC,EAAG,MAAOsF,EAAe,OAAQA,CAAc,EAEnE,IAAMG,EAAc,OAAO,KAAKF,EAAY,aAAa,UAAU,EAC7DG,EAAe,OAAO,KAAKH,EAAY,cAAc,UAAU,EAErE,MAAO,CACL,MAAOE,GAAeA,EAAY,OAAS,CAAC,GAAGA,EAAa,GAAGH,CAAa,EAAI,KAChF,OAAQI,GAAgBA,EAAa,OAAS,CAAC,GAAGA,EAAc,GAAGJ,CAAa,EAAI,KACpF,OAAQE,CACV,CACF,OAAStC,EAAG,CACV,MAAO,CAAE,OAAQ,CAAC,EAAG,MAAOoC,EAAe,OAAQA,CAAc,CACnE,CACF,CAEA,2BAA2B1E,EAAcC,EAA8BC,EAAmB,CAryB5F,IAAArB,EAsyBQoB,EAAS,OAAS,GAAK,SAAUA,EAAS,CAAC,GAAKA,EAAS,CAAC,EAAE,OAAS,WAEvEA,EAAWA,EAAS,MAAM,CAAC,GAI7B,IAAM8E,EAA+C/E,EAAM,UAAY,CAAC,EAClEgF,EAAqB,IAAI,IAAID,EAAiB,IAAKE,GAAYA,EAAQ,EAAE,CAAC,EAC1EC,EAAa,IAAI,IAAIjF,EAAS,IAAKgF,GAAYA,EAAQ,EAAE,CAAC,EAE5DE,EAAmC,CAAC,EACpClF,EAAS,OAAS8E,EAAiB,SAErCI,EAAkBJ,EACf,OAAQK,GAAM,CAACF,EAAW,IAAIE,EAAE,EAAE,CAAC,EACnC,IAAKA,GAAM,IAAI,iBAAc,CAAE,GAAIA,EAAE,EAAI,CAAC,CAAC,GAGhD,IAAMC,EAAcpF,EAAS,OAAQgF,GAAY,CAACD,EAAmB,IAAIC,EAAQ,EAAE,CAAC,EAEpF,OAAO9D,EAAAnC,EAAA,GACFgB,GADE,CAEL,SAAU,CAAC,GAAGmF,EAAiB,GAAGE,CAAW,EAC7C,MAAO,CAAC,IAAIxG,EAAAmB,EAAM,QAAN,KAAAnB,EAAe,CAAC,EAAI,GAAGqB,CAAK,CAC1C,EACF,CACF","names":["index_exports","__export","CustomEventNames","LangGraphAgent","LangGraphEventTypes","__toCommonJS","import_rxjs","import_langgraph_sdk","import_node_crypto","import_messages","LangGraphEventTypes","CustomEventNames","import_client","DEFAULT_SCHEMA_KEYS","filterObjectBySchemaKeys","obj","schemaKeys","key","getStreamPayloadInput","mode","state","input","langchainMessagesToAgui","messages","message","_a","stringifyIfNeeded","resolveMessageContent","content","tc","aguiMessagesToLangChain","index","_b","item","resolveReasoningContent","eventData","_c","_d","_e","data","contentText","c","LangGraphAgent","config","_a","_b","LangGraphClient","__spreadValues","event","input","subscriber","_c","_d","_e","_f","_g","_h","_i","_j","_k","_l","_m","inputThreadId","state","messages","tools","context","forwardedProps","shouldExit","threadId","thread","agentState","agentStateValues","aguiToLangChainMessage","aguiMessagesToLangChain","mode","payloadInput","getStreamPayloadInput","graphInfo","payloadConfig","configsToMerge","payload","__spreadProps","interrupts","interrupt","streamResponse","latestStateValues","updatedState","iter","__forAwait","more","temp","error","streamResponseChunk","chunk","chunkData","currentNodeName","eventType","metadata","node","langchainMessagesToAgui","e","shouldEmitMessages","shouldEmitToolCalls","currentStream","hasCurrentStream","toolCallData","toolCallUsedToPredictState","predictStateTool","isToolCallStartEvent","isToolCallArgsEvent","isToolCallEndEvent","reasoningData","resolveReasoningContent","messageContent","resolveMessageContent","isMessageContentEvent","isMessageEndEvent","thinkingStepIndex","schemaKeys","filterObjectBySchemaKeys","DEFAULT_SCHEMA_KEYS","configs","assistant","acc","cfg","filteredConfigurable","newConfig","isRecursionLimitSetToDefault","configsAreDifferent","isOnlyRecursionLimitDifferent","runId","data","assistants","retrievedAssistant","searchResult","a","CONSTANT_KEYS","graphSchema","configSchema","inputSchema","outputSchema","existingMessages","existingMessageIds","message","messageIds","removedMessages","m","newMessages"]}
|
package/dist/index.mjs
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
|
-
var nt=Object.defineProperty,it=Object.defineProperties;var ot=Object.getOwnPropertyDescriptors;var V=Object.getOwnPropertySymbols;var rt=Object.prototype.hasOwnProperty,lt=Object.prototype.propertyIsEnumerable;var W=(o,t)=>(t=Symbol[o])?t:Symbol.for("Symbol."+o);var q=(o,t,e)=>t in o?nt(o,t,{enumerable:!0,configurable:!0,writable:!0,value:e}):o[t]=e,u=(o,t)=>{for(var e in t||(t={}))rt.call(t,e)&&q(o,e,t[e]);if(V)for(var e of V(t))lt.call(t,e)&&q(o,e,t[e]);return o},_=(o,t)=>it(o,ot(t));var z=(o,t,e)=>(t=o[W("asyncIterator")])?t.call(o):(o=o[W("iterator")](),t={},e=(s,a)=>(a=o[s])&&(t[s]=i=>new Promise((g,c,l)=>(i=a.call(o,i),l=i.done,Promise.resolve(i.value).then(p=>g({value:p,done:l}),c)))),e("next"),e("return"),t);import{Observable as ct}from"rxjs";import{Client as gt}from"@langchain/langgraph-sdk";import{randomUUID as ut}from"crypto";import{RemoveMessage as pt}from"@langchain/core/messages";var dt=(n=>(n.OnChainStart="on_chain_start",n.OnChainStream="on_chain_stream",n.OnChainEnd="on_chain_end",n.OnChatModelStart="on_chat_model_start",n.OnChatModelStream="on_chat_model_stream",n.OnChatModelEnd="on_chat_model_end",n.OnToolStart="on_tool_start",n.OnToolEnd="on_tool_end",n.OnCustomEvent="on_custom_event",n.OnInterrupt="on_interrupt",n))(dt||{}),ht=(a=>(a.ManuallyEmitMessage="manually_emit_message",a.ManuallyEmitToolCall="manually_emit_tool_call",a.ManuallyEmitState="manually_emit_state",a.Exit="exit",a))(ht||{});import{AbstractAgent as mt,EventType as r}from"@ag-ui/client";var A=["tools"];function O(o,t){return Object.fromEntries(Object.entries(o).filter(([e])=>t.includes(e)))}function Q({mode:o,state:t,schemaKeys:e}){let s=o==="start"?t:null;return s&&(e!=null&&e.input)&&(s=O(s,[...A,...e.input])),s}function Z(o){return o.map(t=>{var e;switch(t.type){case"human":return{id:t.id,role:"user",content:N(t.content)};case"ai":return{id:t.id,role:"assistant",content:N(t.content),toolCalls:(e=t.tool_calls)==null?void 0:e.map(s=>({id:s.id,type:"function",function:{name:s.name,arguments:JSON.stringify(s.args)}}))};case"system":return{id:t.id,role:"system",content:N(t.content)};case"tool":return{id:t.id,role:"tool",content:N(t.content),toolCallId:t.tool_call_id};default:throw new Error("message type returned from LangGraph is not supported.")}})}function tt(o){return o.map((t,e)=>{var s,a;switch(t.role){case"user":return{id:t.id,role:t.role,content:t.content,type:"human"};case"assistant":return{id:t.id,type:"ai",role:t.role,content:(s=t.content)!=null?s:"",tool_calls:((a=t.toolCalls)!=null?a:[]).map(i=>({id:i.id,name:i.function.name,args:JSON.parse(i.function.arguments),type:"tool_call"}))};case"system":return{id:t.id,role:t.role,content:t.content,type:"system"};case"tool":return{content:t.content,role:t.role,type:t.role,tool_call_id:t.toolCallId,id:t.id};default:throw console.error(`Message role ${t.role} is not implemented`),new Error("message role is not supported.")}})}function N(o){return typeof o=="string"?o:JSON.stringify(o)}var et=class extends mt{constructor(t){var e,s;super(t),this.messagesInProcess={},this.agentName=t.agentName,this.graphId=t.graphId,this.assistantConfig=t.assistantConfig,this.client=(s=t==null?void 0:t.client)!=null?s:new gt({apiUrl:t.deploymentUrl,apiKey:t.langsmithApiKey,defaultHeaders:u({},(e=t.propertyHeaders)!=null?e:{})})}dispatchEvent(t){return this.subscriber.next(t),!0}run(t){return this.activeRun={id:t.runId,threadId:t.threadId},new ct(e=>(this.handleStreamEvents(t,e),()=>{}))}async handleStreamEvents(t,e){var k,G,D,U,H,J,K,F,j,B,X,$;let{threadId:s,state:a,messages:i,tools:g,context:c,forwardedProps:l}=t;this.subscriber=e;let p=!1;this.activeRun.manuallyEmittedState=null,this.activeRun.nodeName=(k=t.forwardedProps)==null?void 0:k.nodeName;let n=s!=null?s:ut();this.assistant||(this.assistant=await this.getAssistant());let d=await this.getOrCreateThreadAndReturnState(n),f=d.values,h=tt(i);a.messages=f.messages,a=this.langGraphDefaultMergeState(a,h,g);let R=n&&this.activeRun.nodeName!="__end__"&&this.activeRun.nodeName?"continue":"start";R==="continue"&&!((G=l==null?void 0:l.command)!=null&&G.resume)&&await this.client.threads.updateState(n,{values:a,asNode:this.activeRun.nodeName}),this.activeRun.schemaKeys=await this.getSchemaKeys();let P=Q({mode:R,state:a,schemaKeys:this.activeRun.schemaKeys}),w=await this.client.assistants.getGraph(this.assistant.assistant_id),I,T=[this.assistantConfig,l==null?void 0:l.config].filter(Boolean);T.length&&(I=await this.mergeConfigs({configs:T,assistant:this.assistant,schemaKeys:this.activeRun.schemaKeys}));let C=_(u({},l),{streamMode:(D=l==null?void 0:l.streamMode)!=null?D:["events","values","updates"],input:P,config:I}),y=(J=(H=(U=d.tasks)==null?void 0:U[0])==null?void 0:H.interrupts)!=null?J:[];if(y!=null&&y.length&&!((K=l==null?void 0:l.command)!=null&&K.resume))return this.dispatchEvent({type:r.RUN_STARTED,threadId:n,runId:t.runId}),y.forEach(S=>{this.dispatchEvent({type:r.CUSTOM,name:"on_interrupt",value:typeof S.value=="string"?S.value:JSON.stringify(S.value),rawEvent:S})}),this.dispatchEvent({type:r.RUN_FINISHED,threadId:n,runId:t.runId}),e.complete();let E=this.client.runs.stream(n,this.assistant.assistant_id,C);this.activeRun.prevNodeName=null;let b={},x=a;try{this.dispatchEvent({type:r.RUN_STARTED,threadId:n,runId:this.activeRun.id});try{for(var ae=z(E),se,ne,ie;se=!(ne=await ae.next()).done;se=!1){let m=ne.value;if(!C.streamMode.includes(m.event))continue;let L=m;if(m.event==="error"){this.dispatchEvent({type:r.RUN_ERROR,message:m.data.message,rawEvent:m});break}if(m.event==="updates")continue;if(m.event==="values"){b=L.data;continue}let M=L.data,v=M.metadata.langgraph_node,Y=M.event,at=M.metadata;if(this.activeRun.id=at.run_id,v&&v!==this.activeRun.nodeName&&(this.activeRun.nodeName&&this.dispatchEvent({type:r.STEP_FINISHED,stepName:this.activeRun.nodeName}),v&&(this.dispatchEvent({type:r.STEP_STARTED,stepName:v}),this.activeRun.nodeName=v)),p=p||Y==="on_custom_event"&&M.name==="exit",this.activeRun.exitingNode=this.activeRun.nodeName===v&&Y==="on_chain_end",this.activeRun.exitingNode&&(this.activeRun.manuallyEmittedState=null),w.nodes.some(st=>st.id===v)&&(this.activeRun.nodeName=v),x=(F=this.activeRun.manuallyEmittedState)!=null?F:b,!this.activeRun.nodeName)continue;(JSON.stringify(x)!==JSON.stringify(a)||this.activeRun.prevNodeName!=this.activeRun.nodeName||this.activeRun.exitingNode)&&!this.getMessageInProgress(this.activeRun.id)&&(a=x,this.activeRun.prevNodeName=this.activeRun.nodeName,this.dispatchEvent({type:r.STATE_SNAPSHOT,snapshot:this.getStateSnapshot(a),rawEvent:L})),this.dispatchEvent({type:r.RAW,event:M}),this.handleSingleEvent(M,a)}}catch(ne){ie=[ne]}finally{try{se&&(ne=ae.return)&&await ne.call(ae)}finally{if(ie)throw ie[0]}}a=await this.client.threads.getState(n);let S=(X=(B=(j=a.tasks)==null?void 0:j[0])==null?void 0:B.interrupts)!=null?X:[];return this.activeRun.nodeName=S?this.activeRun.nodeName:Object.keys(a.metadata.writes)[0],S.forEach(m=>{this.dispatchEvent({type:r.CUSTOM,name:"on_interrupt",value:typeof m.value=="string"?m.value:JSON.stringify(m.value),rawEvent:m})}),this.dispatchEvent({type:r.STATE_SNAPSHOT,snapshot:this.getStateSnapshot(a.values)}),this.dispatchEvent({type:r.MESSAGES_SNAPSHOT,messages:Z(($=a.values.messages)!=null?$:[])}),this.activeRun.nodeName&&this.dispatchEvent({type:r.STEP_FINISHED,stepName:this.activeRun.nodeName}),this.dispatchEvent({type:r.RUN_FINISHED,threadId:n,runId:this.activeRun.id}),e.complete()}catch(S){return e.error(S)}}handleSingleEvent(t,e){var s,a,i,g,c,l;switch(t.event){case"on_chat_model_stream":let p=(s=t.metadata["emit-messages"])!=null?s:!0,n=(a=t.metadata["emit-tool-calls"])!=null?a:!0;if(t.data.chunk.response_metadata.finish_reason)return;let d=this.getMessageInProgress(this.activeRun.id),f=!!(d!=null&&d.id),h=(i=t.data.chunk.tool_call_chunks)==null?void 0:i[0],R=(g=t.metadata.predict_state)==null?void 0:g.some(E=>E.tool===(h==null?void 0:h.name)),P=!f&&(h==null?void 0:h.name),w=f&&(d==null?void 0:d.toolCallId)&&h.args,I=f&&(d==null?void 0:d.toolCallId)&&!h,T=!f&&!h,C=f&&!h,y=f&&!(d!=null&&d.toolCallId)&&!C;if(R&&this.dispatchEvent({type:r.CUSTOM,name:"PredictState",value:t.metadata.predict_state}),I){this.dispatchEvent({type:r.TOOL_CALL_END,toolCallId:d.toolCallId,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if(y){this.dispatchEvent({type:r.TEXT_MESSAGE_END,messageId:d.id,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if(P&&n){this.dispatchEvent({type:r.TOOL_CALL_START,toolCallId:h.id,toolCallName:h.name,parentMessageId:t.data.chunk.id,rawEvent:t})&&this.setMessageInProgress(this.activeRun.id,{id:t.data.chunk.id,toolCallId:h.id,toolCallName:h.name});break}if(w&&n){this.dispatchEvent({type:r.TOOL_CALL_ARGS,toolCallId:d.toolCallId,delta:h.args,rawEvent:t});break}if(T&&p){this.dispatchEvent({type:r.TEXT_MESSAGE_START,role:"assistant",messageId:t.data.chunk.id,rawEvent:t})&&this.setMessageInProgress(this.activeRun.id,{id:t.data.chunk.id,toolCallId:null,toolCallName:null});break}if(C&&p){this.dispatchEvent({type:r.TEXT_MESSAGE_CONTENT,messageId:d.id,delta:t.data.chunk.content,rawEvent:t});break}break;case"on_chat_model_end":if((c=this.getMessageInProgress(this.activeRun.id))!=null&&c.toolCallId){this.dispatchEvent({type:r.TOOL_CALL_END,toolCallId:this.getMessageInProgress(this.activeRun.id).toolCallId,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if((l=this.getMessageInProgress(this.activeRun.id))!=null&&l.id){this.dispatchEvent({type:r.TEXT_MESSAGE_END,messageId:this.getMessageInProgress(this.activeRun.id).id,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}break;case"on_custom_event":if(t.name==="manually_emit_message"){this.dispatchEvent({type:r.TEXT_MESSAGE_START,role:"assistant",messageId:t.data.message_id,rawEvent:t}),this.dispatchEvent({type:r.TEXT_MESSAGE_CONTENT,messageId:t.data.message_id,delta:t.data.message,rawEvent:t}),this.dispatchEvent({type:r.TEXT_MESSAGE_END,messageId:t.data.message_id,rawEvent:t});break}if(t.name==="manually_emit_tool_call"){this.dispatchEvent({type:r.TOOL_CALL_START,toolCallId:t.data.id,toolCallName:t.data.name,parentMessageId:t.data.id,rawEvent:t}),this.dispatchEvent({type:r.TOOL_CALL_ARGS,toolCallId:t.data.id,delta:t.data.args,rawEvent:t}),this.dispatchEvent({type:r.TOOL_CALL_END,toolCallId:t.data.id,rawEvent:t});break}t.name==="manually_emit_state"&&(this.activeRun.manuallyEmittedState=t.data,this.dispatchEvent({type:r.STATE_SNAPSHOT,snapshot:this.getStateSnapshot(e),rawEvent:t})),this.dispatchEvent({type:r.CUSTOM,name:t.name,value:t.data,rawEvent:t});break}}getStateSnapshot(t){let e=this.activeRun.schemaKeys;return e!=null&&e.output&&(t=O(t,[...A,...e.output])),t}async getOrCreateThreadAndReturnState(t){let e={values:{}};try{await this.client.threads.get(t),e=await this.client.threads.getState(t)}catch(s){await this.client.threads.create({threadId:t})}return e}async mergeConfigs({configs:t,assistant:e,schemaKeys:s}){return t.reduce((a,i)=>{var d;let g=a.configurable;i.configurable&&(g=s!=null&&s.config?O(i==null?void 0:i.configurable,[...A,...(d=s==null?void 0:s.config)!=null?d:[]]):i==null?void 0:i.configurable);let c=_(u(u({},a),i),{configurable:g}),l=a.recursion_limit==null&&i.recursion_limit===25,p=JSON.stringify(c)!==JSON.stringify(a),n=l&&JSON.stringify(_(u({},c),{recursion_limit:null}))===JSON.stringify(_(u({},a),{recursion_limit:null}));return p&&!n?u(u({},a),c):a},e.config)}getMessageInProgress(t){return this.messagesInProcess[t]}setMessageInProgress(t,e){this.messagesInProcess=_(u({},this.messagesInProcess),{[t]:u(u({},this.messagesInProcess[t]),e)})}async getAssistant(){let t=await this.client.assistants.search(),e=t.find(s=>s.assistant_id===this.agentId||s.graph_id===this.graphId);if(!e)throw console.error(`
|
|
2
|
-
No agent found with graph ID ${this.graphId}
|
|
1
|
+
var lt=Object.defineProperty,dt=Object.defineProperties;var ht=Object.getOwnPropertyDescriptors;var z=Object.getOwnPropertySymbols;var gt=Object.prototype.hasOwnProperty,ct=Object.prototype.propertyIsEnumerable;var Q=(a,t)=>(t=Symbol[a])?t:Symbol.for("Symbol."+a);var Z=(a,t,e)=>t in a?lt(a,t,{enumerable:!0,configurable:!0,writable:!0,value:e}):a[t]=e,u=(a,t)=>{for(var e in t||(t={}))gt.call(t,e)&&Z(a,e,t[e]);if(z)for(var e of z(t))ct.call(t,e)&&Z(a,e,t[e]);return a},y=(a,t)=>dt(a,ht(t));var tt=(a,t,e)=>(t=a[Q("asyncIterator")])?t.call(a):(a=a[Q("iterator")](),t={},e=(s,n)=>(n=a[s])&&(t[s]=r=>new Promise((c,h,d)=>(r=n.call(a,r),d=r.done,Promise.resolve(r.value).then(m=>c({value:m,done:d}),h)))),e("next"),e("return"),t);import{Observable as mt}from"rxjs";import{Client as Et}from"@langchain/langgraph-sdk";import{randomUUID as St}from"crypto";import{RemoveMessage as ft}from"@langchain/core/messages";var ut=(o=>(o.OnChainStart="on_chain_start",o.OnChainStream="on_chain_stream",o.OnChainEnd="on_chain_end",o.OnChatModelStart="on_chat_model_start",o.OnChatModelStream="on_chat_model_stream",o.OnChatModelEnd="on_chat_model_end",o.OnToolStart="on_tool_start",o.OnToolEnd="on_tool_end",o.OnCustomEvent="on_custom_event",o.OnInterrupt="on_interrupt",o))(ut||{}),pt=(n=>(n.ManuallyEmitMessage="manually_emit_message",n.ManuallyEmitToolCall="manually_emit_tool_call",n.ManuallyEmitState="manually_emit_state",n.Exit="exit",n))(pt||{});import{AbstractAgent as vt,EventType as i}from"@ag-ui/client";var A=["tools"];function P(a,t){return Object.fromEntries(Object.entries(a).filter(([e])=>t.includes(e)))}function et({mode:a,state:t,schemaKeys:e}){let s=a==="start"?t:null;return s&&(e!=null&&e.input)&&(s=P(s,[...A,...e.input])),s}function st(a){return a.map(t=>{var e;switch(t.type){case"human":return{id:t.id,role:"user",content:x(I(t.content))};case"ai":let s=I(t.content);return{id:t.id,role:"assistant",content:s?x(s):"",toolCalls:(e=t.tool_calls)==null?void 0:e.map(n=>({id:n.id,type:"function",function:{name:n.name,arguments:JSON.stringify(n.args)}}))};case"system":return{id:t.id,role:"system",content:x(I(t.content))};case"tool":return{id:t.id,role:"tool",content:x(I(t.content)),toolCallId:t.tool_call_id};default:throw new Error("message type returned from LangGraph is not supported.")}})}function nt(a){return a.map((t,e)=>{var s,n;switch(t.role){case"user":return{id:t.id,role:t.role,content:t.content,type:"human"};case"assistant":return{id:t.id,type:"ai",role:t.role,content:(s=t.content)!=null?s:"",tool_calls:((n=t.toolCalls)!=null?n:[]).map(r=>({id:r.id,name:r.function.name,args:JSON.parse(r.function.arguments),type:"tool_call"}))};case"system":return{id:t.id,role:t.role,content:t.content,type:"system"};case"tool":return{content:t.content,role:t.role,type:t.role,tool_call_id:t.toolCallId,id:t.id};default:throw console.error(`Message role ${t.role} is not implemented`),new Error("message role is not supported.")}})}function x(a){return typeof a=="string"?a:JSON.stringify(a)}function at(a){var e,s,n,r,c;let t=(e=a.chunk)==null?void 0:e.content;if(t&&Array.isArray(t)&&t.length&&t[0])return t[0].thinking?{text:t[0].thinking,type:"text",index:t[0].index}:null;if((r=(n=(s=a.chunk.additional_kwargs)==null?void 0:s.reasoning)==null?void 0:n.summary)!=null&&r[0]){let h=(c=a.chunk.additional_kwargs)==null?void 0:c.reasoning.summary[0];return!h||!h.text?null:{type:"text",text:h.text,index:h.index}}return null}function I(a){var t;if(!a)return null;if(typeof a=="string")return a;if(Array.isArray(a)&&a.length){let e=(t=a.find(s=>s.type==="text"))==null?void 0:t.text;return e!=null?e:null}return null}var it=class extends vt{constructor(t){var e,s;super(t),this.messagesInProcess={},this.agentName=t.agentName,this.graphId=t.graphId,this.assistantConfig=t.assistantConfig,this.thinkingProcess=null,this.client=(s=t==null?void 0:t.client)!=null?s:new Et({apiUrl:t.deploymentUrl,apiKey:t.langsmithApiKey,defaultHeaders:u({},(e=t.propertyHeaders)!=null?e:{})})}dispatchEvent(t){return this.subscriber.next(t),!0}run(t){return this.activeRun={id:t.runId,threadId:t.threadId},new mt(e=>(this.handleStreamEvents(t,e),()=>{}))}async handleStreamEvents(t,e){var D,H,K,U,F,J,X,B,j,$,Y,V,W;let{threadId:s,state:n,messages:r,tools:c,context:h,forwardedProps:d}=t;this.subscriber=e;let m=!1;this.activeRun.manuallyEmittedState=null,this.activeRun.nodeName=(D=t.forwardedProps)==null?void 0:D.nodeName;let o=s!=null?s:St();this.assistant||(this.assistant=await this.getAssistant());let l=await this.getOrCreateThread(o);this.activeRun.threadId=l.thread_id;let v=(H=await this.client.threads.getState(l.thread_id))!=null?H:{values:{}},g=v.values,k=nt(r);n.messages=g.messages,n=this.langGraphDefaultMergeState(n,k,c);let N=o&&this.activeRun.nodeName!="__end__"&&this.activeRun.nodeName?"continue":"start";N==="continue"&&!((K=d==null?void 0:d.command)!=null&&K.resume)&&await this.client.threads.updateState(o,{values:n,asNode:this.activeRun.nodeName}),this.activeRun.schemaKeys=await this.getSchemaKeys();let O=et({mode:N,state:n,schemaKeys:this.activeRun.schemaKeys}),w=await this.client.assistants.getGraph(this.assistant.assistant_id),T,R=[this.assistantConfig,d==null?void 0:d.config].filter(Boolean);R.length&&(T=await this.mergeConfigs({configs:R,assistant:this.assistant,schemaKeys:this.activeRun.schemaKeys}));let C=y(u({},d),{streamMode:(U=d==null?void 0:d.streamMode)!=null?U:["events","values","updates"],input:O,config:T}),_=(X=(J=(F=v.tasks)==null?void 0:F[0])==null?void 0:J.interrupts)!=null?X:[];if(_!=null&&_.length&&!((B=d==null?void 0:d.command)!=null&&B.resume))return this.dispatchEvent({type:i.RUN_STARTED,threadId:o,runId:t.runId}),_.forEach(E=>{this.dispatchEvent({type:i.CUSTOM,name:"on_interrupt",value:typeof E.value=="string"?E.value:JSON.stringify(E.value),rawEvent:E})}),this.dispatchEvent({type:i.RUN_FINISHED,threadId:o,runId:t.runId}),e.complete();let S=this.client.runs.stream(o,this.assistant.assistant_id,C);this.activeRun.prevNodeName=null;let b={},L=n;try{this.dispatchEvent({type:i.RUN_STARTED,threadId:o,runId:this.activeRun.id});try{for(var ce=tt(S),ue,pe,me;ue=!(pe=await ce.next()).done;ue=!1){let p=pe.value;if(!C.streamMode.includes(p.event))continue;let G=p;if(p.event==="error"){this.dispatchEvent({type:i.RUN_ERROR,message:p.data.message,rawEvent:p});break}if(p.event==="updates")continue;if(p.event==="values"){b=G.data;continue}let M=G.data,f=M.metadata.langgraph_node,q=M.event,rt=M.metadata;if(this.activeRun.id=rt.run_id,f&&f!==this.activeRun.nodeName&&(this.activeRun.nodeName&&this.dispatchEvent({type:i.STEP_FINISHED,stepName:this.activeRun.nodeName}),f&&(this.dispatchEvent({type:i.STEP_STARTED,stepName:f}),this.activeRun.nodeName=f)),m=m||q==="on_custom_event"&&M.name==="exit",this.activeRun.exitingNode=this.activeRun.nodeName===f&&q==="on_chain_end",this.activeRun.exitingNode&&(this.activeRun.manuallyEmittedState=null),w.nodes.some(ot=>ot.id===f)&&(this.activeRun.nodeName=f),L=(j=this.activeRun.manuallyEmittedState)!=null?j:b,!this.activeRun.nodeName)continue;(JSON.stringify(L)!==JSON.stringify(n)||this.activeRun.prevNodeName!=this.activeRun.nodeName||this.activeRun.exitingNode)&&!this.getMessageInProgress(this.activeRun.id)&&(n=L,this.activeRun.prevNodeName=this.activeRun.nodeName,this.dispatchEvent({type:i.STATE_SNAPSHOT,snapshot:this.getStateSnapshot(n),rawEvent:G})),this.dispatchEvent({type:i.RAW,event:M}),this.handleSingleEvent(M,n)}}catch(pe){me=[pe]}finally{try{ue&&(pe=ce.return)&&await pe.call(ce)}finally{if(me)throw me[0]}}n=await this.client.threads.getState(o);let E=(V=(Y=($=n.tasks)==null?void 0:$[0])==null?void 0:Y.interrupts)!=null?V:[];return this.activeRun.nodeName=E?this.activeRun.nodeName:Object.keys(n.metadata.writes)[0],E.forEach(p=>{this.dispatchEvent({type:i.CUSTOM,name:"on_interrupt",value:typeof p.value=="string"?p.value:JSON.stringify(p.value),rawEvent:p})}),this.dispatchEvent({type:i.STATE_SNAPSHOT,snapshot:this.getStateSnapshot(n.values)}),this.dispatchEvent({type:i.MESSAGES_SNAPSHOT,messages:st((W=n.values.messages)!=null?W:[])}),this.activeRun.nodeName&&this.dispatchEvent({type:i.STEP_FINISHED,stepName:this.activeRun.nodeName}),this.dispatchEvent({type:i.RUN_FINISHED,threadId:o,runId:this.activeRun.id}),e.complete()}catch(E){return e.error(E)}}handleSingleEvent(t,e){var s,n,r,c,h,d;switch(t.event){case"on_chat_model_stream":let m=(s=t.metadata["emit-messages"])!=null?s:!0,o=(n=t.metadata["emit-tool-calls"])!=null?n:!0;if(t.data.chunk.response_metadata.finish_reason)return;let l=this.getMessageInProgress(this.activeRun.id),v=!!(l!=null&&l.id),g=(r=t.data.chunk.tool_call_chunks)==null?void 0:r[0],k=(c=t.metadata.predict_state)==null?void 0:c.some(S=>S.tool===(g==null?void 0:g.name)),N=!v&&(g==null?void 0:g.name),O=v&&(l==null?void 0:l.toolCallId)&&(g==null?void 0:g.args),w=v&&(l==null?void 0:l.toolCallId)&&!g,T=at(t.data),R=I(t.data.chunk.content),C=!!(!g&&R),_=v&&!(l!=null&&l.toolCallId)&&!C;if(T){this.handleThinkingEvent(T);break}if(!T&&this.thinkingProcess&&(this.dispatchEvent({type:i.THINKING_TEXT_MESSAGE_END}),this.dispatchEvent({type:i.THINKING_END}),this.thinkingProcess=null),k&&this.dispatchEvent({type:i.CUSTOM,name:"PredictState",value:t.metadata.predict_state}),w){this.dispatchEvent({type:i.TOOL_CALL_END,toolCallId:l==null?void 0:l.toolCallId,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if(_){this.dispatchEvent({type:i.TEXT_MESSAGE_END,messageId:l.id,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if(N&&o){this.dispatchEvent({type:i.TOOL_CALL_START,toolCallId:g.id,toolCallName:g.name,parentMessageId:t.data.chunk.id,rawEvent:t})&&this.setMessageInProgress(this.activeRun.id,{id:t.data.chunk.id,toolCallId:g.id,toolCallName:g.name});break}if(O&&o){this.dispatchEvent({type:i.TOOL_CALL_ARGS,toolCallId:l==null?void 0:l.toolCallId,delta:g.args,rawEvent:t});break}if(C&&m){l||(this.dispatchEvent({type:i.TEXT_MESSAGE_START,role:"assistant",messageId:t.data.chunk.id,rawEvent:t}),this.setMessageInProgress(this.activeRun.id,{id:t.data.chunk.id,toolCallId:null,toolCallName:null}),l=this.getMessageInProgress(this.activeRun.id)),this.dispatchEvent({type:i.TEXT_MESSAGE_CONTENT,messageId:l.id,delta:R,rawEvent:t});break}break;case"on_chat_model_end":if((h=this.getMessageInProgress(this.activeRun.id))!=null&&h.toolCallId){this.dispatchEvent({type:i.TOOL_CALL_END,toolCallId:this.getMessageInProgress(this.activeRun.id).toolCallId,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if((d=this.getMessageInProgress(this.activeRun.id))!=null&&d.id){this.dispatchEvent({type:i.TEXT_MESSAGE_END,messageId:this.getMessageInProgress(this.activeRun.id).id,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}break;case"on_custom_event":if(t.name==="manually_emit_message"){this.dispatchEvent({type:i.TEXT_MESSAGE_START,role:"assistant",messageId:t.data.message_id,rawEvent:t}),this.dispatchEvent({type:i.TEXT_MESSAGE_CONTENT,messageId:t.data.message_id,delta:t.data.message,rawEvent:t}),this.dispatchEvent({type:i.TEXT_MESSAGE_END,messageId:t.data.message_id,rawEvent:t});break}if(t.name==="manually_emit_tool_call"){this.dispatchEvent({type:i.TOOL_CALL_START,toolCallId:t.data.id,toolCallName:t.data.name,parentMessageId:t.data.id,rawEvent:t}),this.dispatchEvent({type:i.TOOL_CALL_ARGS,toolCallId:t.data.id,delta:t.data.args,rawEvent:t}),this.dispatchEvent({type:i.TOOL_CALL_END,toolCallId:t.data.id,rawEvent:t});break}t.name==="manually_emit_state"&&(this.activeRun.manuallyEmittedState=t.data,this.dispatchEvent({type:i.STATE_SNAPSHOT,snapshot:this.getStateSnapshot(e),rawEvent:t})),this.dispatchEvent({type:i.CUSTOM,name:t.name,value:t.data,rawEvent:t});break}}handleThinkingEvent(t){var s;if(!t||!t.type||!t.text)return;let e=t.index;(s=this.thinkingProcess)!=null&&s.index&&this.thinkingProcess.index!==e&&(this.thinkingProcess.type&&this.dispatchEvent({type:i.THINKING_TEXT_MESSAGE_END}),this.dispatchEvent({type:i.THINKING_END}),this.thinkingProcess=null),this.thinkingProcess||(this.dispatchEvent({type:i.THINKING_START}),this.thinkingProcess={index:e}),this.thinkingProcess.type!==t.type&&(this.dispatchEvent({type:i.THINKING_TEXT_MESSAGE_START}),this.thinkingProcess.type=t.type),this.thinkingProcess.type&&this.dispatchEvent({type:i.THINKING_TEXT_MESSAGE_CONTENT,delta:t.text})}getStateSnapshot(t){let e=this.activeRun.schemaKeys;return e!=null&&e.output&&(t=P(t,[...A,...e.output])),t}async getOrCreateThread(t){let e;try{try{e=await this.getThread(t)}catch(s){e=await this.createThread({threadId:t})}}catch(s){throw new Error(`Failed to create thread: ${s.message}`)}return e}async getThread(t){return this.client.threads.get(t)}async createThread(t){return this.client.threads.create(t)}async mergeConfigs({configs:t,assistant:e,schemaKeys:s}){return t.reduce((n,r)=>{var l;let c=n.configurable;r.configurable&&(c=s!=null&&s.config?P(r==null?void 0:r.configurable,[...A,...(l=s==null?void 0:s.config)!=null?l:[]]):r==null?void 0:r.configurable);let h=y(u(u({},n),r),{configurable:c}),d=n.recursion_limit==null&&r.recursion_limit===25,m=JSON.stringify(h)!==JSON.stringify(n),o=d&&JSON.stringify(y(u({},h),{recursion_limit:null}))===JSON.stringify(y(u({},n),{recursion_limit:null}));return m&&!o?u(u({},n),h):n},e.config)}getMessageInProgress(t){return this.messagesInProcess[t]}setMessageInProgress(t,e){this.messagesInProcess=y(u({},this.messagesInProcess),{[t]:u(u({},this.messagesInProcess[t]),e)})}async getAssistant(){let t=await this.client.assistants.search(),e=t.find(s=>s.graph_id===this.graphId);if(!e)throw console.error(`
|
|
2
|
+
No agent found with graph ID ${this.graphId} found..
|
|
3
3
|
|
|
4
4
|
|
|
5
5
|
These are the available agents: [${t.map(s=>`${s.graph_id} (ID: ${s.assistant_id})`).join(", ")}]
|
|
6
|
-
`),new Error("No agent id found");return e}async getSchemaKeys(){var e,s,
|
|
6
|
+
`),new Error("No agent id found");return e}async getSchemaKeys(){var e,s,n;let t=["messages"];try{let r=await this.client.assistants.getSchemas(this.assistant.assistant_id),c=null;if((e=r.config_schema)!=null&&e.properties&&(c=Object.keys(r.config_schema.properties)),!((s=r.input_schema)!=null&&s.properties)||!((n=r.output_schema)!=null&&n.properties))return{config:[],input:t,output:t};let h=Object.keys(r.input_schema.properties),d=Object.keys(r.output_schema.properties);return{input:h&&h.length?[...h,...t]:null,output:d&&d.length?[...d,...t]:null,config:c}}catch(r){return{config:[],input:t,output:t}}}langGraphDefaultMergeState(t,e,s){var m;e.length>0&&"role"in e[0]&&e[0].role==="system"&&(e=e.slice(1));let n=t.messages||[],r=new Set(n.map(o=>o.id)),c=new Set(e.map(o=>o.id)),h=[];e.length<n.length&&(h=n.filter(o=>!c.has(o.id)).map(o=>new ft({id:o.id})));let d=e.filter(o=>!r.has(o.id));return y(u({},t),{messages:[...h,...d],tools:[...(m=t.tools)!=null?m:[],...s]})}};export{pt as CustomEventNames,it as LangGraphAgent,ut as LangGraphEventTypes};
|
|
7
7
|
//# sourceMappingURL=index.mjs.map
|
package/dist/index.mjs.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["../src/index.ts","../src/types.ts","../src/utils.ts"],"sourcesContent":["import { Observable, Subscriber } from \"rxjs\";\nimport {\n Client as LangGraphClient,\n EventsStreamEvent,\n StreamMode,\n Config as LangGraphConfig,\n ThreadState,\n Assistant,\n Message as LangGraphMessage,\n Config,\n Interrupt,\n} from \"@langchain/langgraph-sdk\";\nimport { randomUUID } from \"node:crypto\";\nimport { RemoveMessage } from \"@langchain/core/messages\";\nimport {\n LangGraphPlatformMessage,\n CustomEventNames,\n LangGraphEventTypes,\n State,\n MessagesInProgressRecord,\n SchemaKeys,\n MessageInProgress,\n RunMetadata,\n PredictStateTool,\n} from \"./types\";\nimport {\n AbstractAgent,\n AgentConfig,\n CustomEvent,\n EventType,\n MessagesSnapshotEvent,\n RawEvent,\n RunAgentInput,\n RunErrorEvent,\n RunFinishedEvent,\n RunStartedEvent,\n StateDeltaEvent,\n StateSnapshotEvent,\n StepFinishedEvent,\n StepStartedEvent,\n TextMessageContentEvent,\n TextMessageEndEvent,\n TextMessageStartEvent,\n ToolCallArgsEvent,\n ToolCallEndEvent,\n ToolCallStartEvent,\n} from \"@ag-ui/client\";\nimport { RunsStreamPayload } from \"@langchain/langgraph-sdk/dist/types\";\nimport {\n aguiMessagesToLangChain,\n DEFAULT_SCHEMA_KEYS,\n filterObjectBySchemaKeys,\n getStreamPayloadInput,\n langchainMessagesToAgui,\n} from \"@/utils\";\n\nexport type ProcessedEvents =\n | TextMessageStartEvent\n | TextMessageContentEvent\n | TextMessageEndEvent\n | ToolCallStartEvent\n | ToolCallArgsEvent\n | ToolCallEndEvent\n | StateSnapshotEvent\n | StateDeltaEvent\n | MessagesSnapshotEvent\n | RawEvent\n | CustomEvent\n | RunStartedEvent\n | RunFinishedEvent\n | RunErrorEvent\n | StepStartedEvent\n | StepFinishedEvent;\n\ntype RunAgentExtendedInput<\n TStreamMode extends StreamMode | StreamMode[] = StreamMode,\n TSubgraphs extends boolean = false,\n> = Omit<RunAgentInput, \"forwardedProps\"> & {\n forwardedProps?: Omit<RunsStreamPayload<TStreamMode, TSubgraphs>, \"input\"> & {\n nodeName?: string;\n };\n};\n\nexport interface LangGraphAgentConfig extends AgentConfig {\n client?: LangGraphClient;\n deploymentUrl: string;\n langsmithApiKey?: string;\n propertyHeaders?: Record<string, string>;\n assistantConfig?: LangGraphConfig;\n agentName?: string;\n graphId: string;\n}\n\nexport class LangGraphAgent extends AbstractAgent {\n client: LangGraphClient;\n assistantConfig?: LangGraphConfig;\n agentName?: string;\n graphId: string;\n assistant?: Assistant;\n messagesInProcess: MessagesInProgressRecord;\n activeRun?: RunMetadata;\n // @ts-expect-error no need to initialize subscriber right now\n subscriber: Subscriber<ProcessedEvents>;\n\n constructor(config: LangGraphAgentConfig) {\n super(config);\n this.messagesInProcess = {};\n this.agentName = config.agentName;\n this.graphId = config.graphId;\n this.assistantConfig = config.assistantConfig;\n this.client =\n config?.client ??\n new LangGraphClient({\n apiUrl: config.deploymentUrl,\n apiKey: config.langsmithApiKey,\n defaultHeaders: { ...(config.propertyHeaders ?? {}) },\n });\n }\n\n dispatchEvent(event: ProcessedEvents) {\n this.subscriber.next(event);\n return true;\n }\n\n run(input: RunAgentInput) {\n this.activeRun = {\n id: input.runId,\n threadId: input.threadId,\n };\n return new Observable<ProcessedEvents>((subscriber) => {\n this.handleStreamEvents(input, subscriber);\n return () => {};\n });\n }\n\n async handleStreamEvents(input: RunAgentExtendedInput, subscriber: Subscriber<ProcessedEvents>) {\n let { threadId: inputThreadId, state, messages, tools, context, forwardedProps } = input;\n this.subscriber = subscriber;\n let shouldExit = false;\n // If a manual emittance happens, it is the ultimate source of truth of state, unless a node has exited.\n // Therefore, this value should either hold null, or the only edition of state that should be used.\n this.activeRun!.manuallyEmittedState = null;\n\n this.activeRun!.nodeName = input.forwardedProps?.nodeName;\n\n const threadId = inputThreadId ?? randomUUID();\n\n if (!this.assistant) {\n this.assistant = await this.getAssistant();\n }\n\n let agentState = await this.getOrCreateThreadAndReturnState(threadId);\n\n const agentStateValues = agentState.values as State;\n const aguiToLangChainMessage = aguiMessagesToLangChain(messages);\n\n state.messages = agentStateValues.messages;\n state = this.langGraphDefaultMergeState(state, aguiToLangChainMessage, tools);\n\n const mode =\n threadId && this.activeRun!.nodeName != \"__end__\" && this.activeRun!.nodeName\n ? \"continue\"\n : \"start\";\n\n if (mode === \"continue\" && !forwardedProps?.command?.resume) {\n await this.client.threads.updateState(threadId, {\n values: state,\n asNode: this.activeRun!.nodeName,\n });\n }\n\n this.activeRun!.schemaKeys = await this.getSchemaKeys();\n\n const payloadInput = getStreamPayloadInput({\n mode,\n state,\n schemaKeys: this.activeRun!.schemaKeys,\n });\n\n const graphInfo = await this.client.assistants.getGraph(this.assistant.assistant_id);\n\n let payloadConfig: LangGraphConfig | undefined;\n const configsToMerge = [this.assistantConfig, forwardedProps?.config].filter(\n Boolean,\n ) as LangGraphConfig[];\n if (configsToMerge.length) {\n payloadConfig = await this.mergeConfigs({\n configs: configsToMerge,\n assistant: this.assistant,\n schemaKeys: this.activeRun!.schemaKeys,\n });\n }\n const payload = {\n ...forwardedProps,\n streamMode:\n forwardedProps?.streamMode ?? ([\"events\", \"values\", \"updates\"] satisfies StreamMode[]),\n input: payloadInput,\n config: payloadConfig,\n };\n\n // If there are still outstanding unresolved interrupts, we must force resolution of them before moving forward\n const interrupts = (agentState.tasks?.[0]?.interrupts ?? []) as Interrupt[];\n if (interrupts?.length && !forwardedProps?.command?.resume) {\n this.dispatchEvent({\n type: EventType.RUN_STARTED,\n threadId,\n runId: input.runId,\n });\n\n interrupts.forEach((interrupt) => {\n this.dispatchEvent({\n type: EventType.CUSTOM,\n name: LangGraphEventTypes.OnInterrupt,\n value:\n typeof interrupt.value === \"string\" ? interrupt.value : JSON.stringify(interrupt.value),\n rawEvent: interrupt,\n });\n });\n\n this.dispatchEvent({\n type: EventType.RUN_FINISHED,\n threadId,\n runId: input.runId,\n });\n return subscriber.complete();\n }\n const streamResponse = this.client.runs.stream(threadId, this.assistant.assistant_id, payload);\n\n this.activeRun!.prevNodeName = null;\n let latestStateValues = {};\n let updatedState = state;\n\n try {\n this.dispatchEvent({\n type: EventType.RUN_STARTED,\n threadId,\n runId: this.activeRun!.id,\n });\n\n for await (let streamResponseChunk of streamResponse) {\n // @ts-ignore\n if (!payload.streamMode.includes(streamResponseChunk.event as StreamMode)) {\n continue;\n }\n // Force event type, as data is not properly defined on the LG side.\n type EventsChunkData = {\n __interrupt__?: any;\n metadata: Record<string, any>;\n event: string;\n data: any;\n [key: string]: unknown;\n };\n const chunk = streamResponseChunk as EventsStreamEvent & { data: EventsChunkData };\n\n if (streamResponseChunk.event === \"error\") {\n this.dispatchEvent({\n type: EventType.RUN_ERROR,\n message: streamResponseChunk.data.message,\n rawEvent: streamResponseChunk,\n });\n break;\n }\n\n if (streamResponseChunk.event === \"updates\") continue;\n\n if (streamResponseChunk.event === \"values\") {\n latestStateValues = chunk.data;\n continue;\n }\n\n const chunkData = chunk.data;\n const currentNodeName = chunkData.metadata.langgraph_node;\n const eventType = chunkData.event;\n const metadata = chunkData.metadata;\n this.activeRun!.id = metadata.run_id;\n\n if (currentNodeName && currentNodeName !== this.activeRun!.nodeName) {\n if (this.activeRun!.nodeName) {\n this.dispatchEvent({\n type: EventType.STEP_FINISHED,\n stepName: this.activeRun!.nodeName,\n });\n }\n\n if (currentNodeName) {\n this.dispatchEvent({\n type: EventType.STEP_STARTED,\n stepName: currentNodeName,\n });\n this.activeRun!.nodeName = currentNodeName;\n }\n }\n\n shouldExit =\n shouldExit ||\n (eventType === LangGraphEventTypes.OnCustomEvent &&\n chunkData.name === CustomEventNames.Exit);\n\n this.activeRun!.exitingNode =\n this.activeRun!.nodeName === currentNodeName &&\n eventType === LangGraphEventTypes.OnChainEnd;\n if (this.activeRun!.exitingNode) {\n this.activeRun!.manuallyEmittedState = null;\n }\n\n // we only want to update the node name under certain conditions\n // since we don't need any internal node names to be sent to the frontend\n if (graphInfo[\"nodes\"].some((node) => node.id === currentNodeName)) {\n this.activeRun!.nodeName = currentNodeName;\n }\n\n updatedState = this.activeRun!.manuallyEmittedState ?? latestStateValues;\n\n if (!this.activeRun!.nodeName) {\n continue;\n }\n\n const hasStateDiff = JSON.stringify(updatedState) !== JSON.stringify(state);\n // We should not update snapshot while a message is in progress.\n if (\n (hasStateDiff ||\n this.activeRun!.prevNodeName != this.activeRun!.nodeName ||\n this.activeRun!.exitingNode) &&\n !Boolean(this.getMessageInProgress(this.activeRun!.id))\n ) {\n state = updatedState;\n this.activeRun!.prevNodeName = this.activeRun!.nodeName;\n\n this.dispatchEvent({\n type: EventType.STATE_SNAPSHOT,\n snapshot: this.getStateSnapshot(state),\n rawEvent: chunk,\n });\n }\n\n this.dispatchEvent({\n type: EventType.RAW,\n event: chunkData,\n });\n\n this.handleSingleEvent(chunkData, state);\n }\n\n state = await this.client.threads.getState(threadId);\n const interrupts = (state.tasks?.[0]?.interrupts ?? []) as Interrupt[];\n this.activeRun!.nodeName = interrupts\n ? this.activeRun!.nodeName\n : Object.keys(state.metadata.writes)[0];\n\n interrupts.forEach((interrupt) => {\n this.dispatchEvent({\n type: EventType.CUSTOM,\n name: LangGraphEventTypes.OnInterrupt,\n value:\n typeof interrupt.value === \"string\" ? interrupt.value : JSON.stringify(interrupt.value),\n rawEvent: interrupt,\n });\n });\n\n this.dispatchEvent({\n type: EventType.STATE_SNAPSHOT,\n snapshot: this.getStateSnapshot(state.values),\n });\n this.dispatchEvent({\n type: EventType.MESSAGES_SNAPSHOT,\n messages: langchainMessagesToAgui(state.values.messages ?? []),\n });\n if (this.activeRun!.nodeName) {\n this.dispatchEvent({\n type: EventType.STEP_FINISHED,\n stepName: this.activeRun!.nodeName!,\n });\n }\n this.dispatchEvent({\n type: EventType.RUN_FINISHED,\n threadId,\n runId: this.activeRun!.id,\n });\n return subscriber.complete();\n } catch (e) {\n return subscriber.error(e);\n }\n }\n\n handleSingleEvent(event: any, state: State): void {\n switch (event.event) {\n case LangGraphEventTypes.OnChatModelStream:\n let shouldEmitMessages = event.metadata[\"emit-messages\"] ?? true;\n let shouldEmitToolCalls = event.metadata[\"emit-tool-calls\"] ?? true;\n\n if (event.data.chunk.response_metadata.finish_reason) return;\n const currentStream = this.getMessageInProgress(this.activeRun!.id);\n const hasCurrentStream = Boolean(currentStream?.id);\n const toolCallData = event.data.chunk.tool_call_chunks?.[0];\n const toolCallUsedToPredictState = event.metadata[\"predict_state\"]?.some(\n (predictStateTool: PredictStateTool) => predictStateTool.tool === toolCallData?.name,\n );\n\n const isToolCallStartEvent = !hasCurrentStream && toolCallData?.name;\n const isToolCallArgsEvent =\n hasCurrentStream && currentStream?.toolCallId && toolCallData.args;\n const isToolCallEndEvent = hasCurrentStream && currentStream?.toolCallId && !toolCallData;\n\n const isMessageStartEvent = !hasCurrentStream && !toolCallData;\n const isMessageContentEvent = hasCurrentStream && !toolCallData;\n const isMessageEndEvent =\n hasCurrentStream && !currentStream?.toolCallId && !isMessageContentEvent;\n\n if (toolCallUsedToPredictState) {\n this.dispatchEvent({\n type: EventType.CUSTOM,\n name: \"PredictState\",\n value: event.metadata[\"predict_state\"],\n });\n }\n\n if (isToolCallEndEvent) {\n const resolved = this.dispatchEvent({\n type: EventType.TOOL_CALL_END,\n toolCallId: currentStream.toolCallId!,\n rawEvent: event,\n });\n if (resolved) {\n this.messagesInProcess[this.activeRun!.id] = null;\n }\n break;\n }\n\n if (isMessageEndEvent) {\n const resolved = this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_END,\n messageId: currentStream!.id,\n rawEvent: event,\n });\n if (resolved) {\n this.messagesInProcess[this.activeRun!.id] = null;\n }\n break;\n }\n\n if (isToolCallStartEvent && shouldEmitToolCalls) {\n const resolved = this.dispatchEvent({\n type: EventType.TOOL_CALL_START,\n toolCallId: toolCallData.id,\n toolCallName: toolCallData.name,\n parentMessageId: event.data.chunk.id,\n rawEvent: event,\n });\n if (resolved) {\n this.setMessageInProgress(this.activeRun!.id, {\n id: event.data.chunk.id,\n toolCallId: toolCallData.id,\n toolCallName: toolCallData.name,\n });\n }\n break;\n }\n\n // Tool call args: emit ActionExecutionArgs\n if (isToolCallArgsEvent && shouldEmitToolCalls) {\n this.dispatchEvent({\n type: EventType.TOOL_CALL_ARGS,\n toolCallId: currentStream.toolCallId!,\n delta: toolCallData.args,\n rawEvent: event,\n });\n break;\n }\n\n // Message started: emit TextMessageStart\n if (isMessageStartEvent && shouldEmitMessages) {\n const resolved = this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_START,\n role: \"assistant\",\n messageId: event.data.chunk.id,\n rawEvent: event,\n });\n if (resolved) {\n this.setMessageInProgress(this.activeRun!.id, {\n id: event.data.chunk.id,\n toolCallId: null,\n toolCallName: null,\n });\n }\n break;\n }\n // Message content: emit TextMessageContent\n if (isMessageContentEvent && shouldEmitMessages) {\n this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_CONTENT,\n messageId: currentStream!.id,\n delta: event.data.chunk.content,\n rawEvent: event,\n });\n break;\n }\n\n break;\n case LangGraphEventTypes.OnChatModelEnd:\n if (this.getMessageInProgress(this.activeRun!.id)?.toolCallId) {\n const resolved = this.dispatchEvent({\n type: EventType.TOOL_CALL_END,\n toolCallId: this.getMessageInProgress(this.activeRun!.id)!.toolCallId!,\n rawEvent: event,\n });\n if (resolved) {\n this.messagesInProcess[this.activeRun!.id] = null;\n }\n break;\n }\n if (this.getMessageInProgress(this.activeRun!.id)?.id) {\n const resolved = this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_END,\n messageId: this.getMessageInProgress(this.activeRun!.id)!.id,\n rawEvent: event,\n });\n if (resolved) {\n this.messagesInProcess[this.activeRun!.id] = null;\n }\n break;\n }\n break;\n case LangGraphEventTypes.OnCustomEvent:\n if (event.name === CustomEventNames.ManuallyEmitMessage) {\n this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_START,\n role: \"assistant\",\n messageId: event.data.message_id,\n rawEvent: event,\n });\n this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_CONTENT,\n messageId: event.data.message_id,\n delta: event.data.message,\n rawEvent: event,\n });\n this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_END,\n messageId: event.data.message_id,\n rawEvent: event,\n });\n break;\n }\n\n if (event.name === CustomEventNames.ManuallyEmitToolCall) {\n this.dispatchEvent({\n type: EventType.TOOL_CALL_START,\n toolCallId: event.data.id,\n toolCallName: event.data.name,\n parentMessageId: event.data.id,\n rawEvent: event,\n });\n this.dispatchEvent({\n type: EventType.TOOL_CALL_ARGS,\n toolCallId: event.data.id,\n delta: event.data.args,\n rawEvent: event,\n });\n this.dispatchEvent({\n type: EventType.TOOL_CALL_END,\n toolCallId: event.data.id,\n rawEvent: event,\n });\n break;\n }\n\n if (event.name === CustomEventNames.ManuallyEmitState) {\n this.activeRun!.manuallyEmittedState = event.data;\n this.dispatchEvent({\n type: EventType.STATE_SNAPSHOT,\n snapshot: this.getStateSnapshot(state),\n rawEvent: event,\n });\n }\n\n this.dispatchEvent({\n type: EventType.CUSTOM,\n name: event.name,\n value: event.data,\n rawEvent: event,\n });\n break;\n }\n }\n\n getStateSnapshot(state: State) {\n const schemaKeys = this.activeRun!.schemaKeys!;\n // Do not emit state keys that are not part of the output schema\n if (schemaKeys?.output) {\n state = filterObjectBySchemaKeys(state, [...DEFAULT_SCHEMA_KEYS, ...schemaKeys.output]);\n }\n // return state\n return state;\n }\n\n async getOrCreateThreadAndReturnState(threadId: string): Promise<ThreadState<{}>> {\n let agentState = { values: {} } as ThreadState;\n try {\n await this.client.threads.get(threadId);\n agentState = await this.client.threads.getState(threadId);\n } catch (error) {\n await this.client.threads.create({ threadId });\n }\n\n return agentState;\n }\n\n async mergeConfigs({\n configs,\n assistant,\n schemaKeys,\n }: {\n configs: Config[];\n assistant: Assistant;\n schemaKeys: SchemaKeys;\n }) {\n return configs.reduce((acc, cfg) => {\n let filteredConfigurable = acc.configurable;\n\n if (cfg.configurable) {\n filteredConfigurable = schemaKeys?.config\n ? filterObjectBySchemaKeys(cfg?.configurable, [\n ...DEFAULT_SCHEMA_KEYS,\n ...(schemaKeys?.config ?? []),\n ])\n : cfg?.configurable;\n }\n\n const newConfig = {\n ...acc,\n ...cfg,\n configurable: filteredConfigurable,\n };\n\n // LG does not return recursion limit if it's the default, therefore we check: if no recursion limit is currently set, and the user asked for 25, there is no change.\n const isRecursionLimitSetToDefault =\n acc.recursion_limit == null && cfg.recursion_limit === 25;\n // Deep compare configs to avoid unnecessary update calls\n const configsAreDifferent = JSON.stringify(newConfig) !== JSON.stringify(acc);\n\n // Check if the only difference is the recursion_limit being set to default\n const isOnlyRecursionLimitDifferent =\n isRecursionLimitSetToDefault &&\n JSON.stringify({ ...newConfig, recursion_limit: null }) ===\n JSON.stringify({ ...acc, recursion_limit: null });\n\n if (configsAreDifferent && !isOnlyRecursionLimitDifferent) {\n return {\n ...acc,\n ...newConfig,\n };\n }\n\n return acc;\n }, assistant.config);\n }\n\n getMessageInProgress(runId: string) {\n return this.messagesInProcess[runId];\n }\n\n setMessageInProgress(runId: string, data: MessageInProgress) {\n this.messagesInProcess = {\n ...this.messagesInProcess,\n [runId]: {\n ...(this.messagesInProcess[runId] as MessageInProgress),\n ...data,\n },\n };\n }\n\n async getAssistant(): Promise<Assistant> {\n const assistants = await this.client.assistants.search();\n const retrievedAssistant = assistants.find(\n (searchResult) =>\n searchResult.assistant_id === this.agentId || searchResult.graph_id === this.graphId,\n );\n if (!retrievedAssistant) {\n console.error(`\n No agent found with graph ID ${this.graphId} or agent ID ${this.agentId} found..\\n\n \n These are the available agents: [${assistants.map((a) => `${a.graph_id} (ID: ${a.assistant_id})`).join(\", \")}]\n `);\n throw new Error(\"No agent id found\");\n }\n\n return retrievedAssistant;\n }\n\n async getSchemaKeys(): Promise<SchemaKeys> {\n const CONSTANT_KEYS = [\"messages\"];\n\n try {\n const graphSchema = await this.client.assistants.getSchemas(this.assistant!.assistant_id);\n let configSchema = null;\n if (graphSchema.config_schema?.properties) {\n configSchema = Object.keys(graphSchema.config_schema.properties);\n }\n if (!graphSchema.input_schema?.properties || !graphSchema.output_schema?.properties) {\n return { config: [], input: CONSTANT_KEYS, output: CONSTANT_KEYS };\n }\n const inputSchema = Object.keys(graphSchema.input_schema.properties);\n const outputSchema = Object.keys(graphSchema.output_schema.properties);\n\n return {\n input: inputSchema && inputSchema.length ? [...inputSchema, ...CONSTANT_KEYS] : null,\n output: outputSchema && outputSchema.length ? [...outputSchema, ...CONSTANT_KEYS] : null,\n config: configSchema,\n };\n } catch (e) {\n return { config: [], input: CONSTANT_KEYS, output: CONSTANT_KEYS };\n }\n }\n\n langGraphDefaultMergeState(state: State, messages: LangGraphMessage[], tools: any): State {\n if (messages.length > 0 && \"role\" in messages[0] && messages[0].role === \"system\") {\n // remove system message\n messages = messages.slice(1);\n }\n\n // merge with existing messages\n const existingMessages: LangGraphPlatformMessage[] = state.messages || [];\n const existingMessageIds = new Set(existingMessages.map((message) => message.id));\n const messageIds = new Set(messages.map((message) => message.id));\n\n let removedMessages: RemoveMessage[] = [];\n if (messages.length < existingMessages.length) {\n // Messages were removed\n removedMessages = existingMessages\n .filter((m) => !messageIds.has(m.id))\n .map((m) => new RemoveMessage({ id: m.id! }));\n }\n\n const newMessages = messages.filter((message) => !existingMessageIds.has(message.id));\n\n return {\n ...state,\n messages: [...removedMessages, ...newMessages],\n tools: [...(state.tools ?? []), ...tools],\n };\n }\n}\n\nexport * from \"./types\";\n","import { Message } from \"@langchain/langgraph-sdk\";\nimport { MessageType } from \"@langchain/core/messages\";\n\nexport enum LangGraphEventTypes {\n OnChainStart = \"on_chain_start\",\n OnChainStream = \"on_chain_stream\",\n OnChainEnd = \"on_chain_end\",\n OnChatModelStart = \"on_chat_model_start\",\n OnChatModelStream = \"on_chat_model_stream\",\n OnChatModelEnd = \"on_chat_model_end\",\n OnToolStart = \"on_tool_start\",\n OnToolEnd = \"on_tool_end\",\n OnCustomEvent = \"on_custom_event\",\n OnInterrupt = \"on_interrupt\",\n}\n\nexport type State = Record<string, any>;\n\nexport type SchemaKeys = {\n input: string[] | null;\n output: string[] | null;\n config: string[] | null;\n} | null;\n\nexport type MessageInProgress = {\n id: string;\n toolCallId?: string | null;\n toolCallName?: string | null;\n};\n\nexport interface RunMetadata {\n id: string;\n schemaKeys?: SchemaKeys;\n nodeName?: string;\n prevNodeName?: string | null;\n exitingNode?: boolean;\n manuallyEmittedState?: State | null;\n threadId?: string;\n}\n\nexport type MessagesInProgressRecord = Record<string, MessageInProgress | null>;\n\n// The following types are our own definition to the messages accepted by LangGraph Platform, enhanced with some of our extra data.\nexport interface ToolCall {\n id: string;\n name: string;\n args: Record<string, unknown>;\n}\n\ntype BaseLangGraphPlatformMessage = Omit<\n Message,\n | \"isResultMessage\"\n | \"isTextMessage\"\n | \"isImageMessage\"\n | \"isActionExecutionMessage\"\n | \"isAgentStateMessage\"\n | \"type\"\n | \"createdAt\"\n> & {\n content: string;\n role: string;\n additional_kwargs?: Record<string, unknown>;\n type: MessageType;\n};\n\ninterface LangGraphPlatformResultMessage extends BaseLangGraphPlatformMessage {\n tool_call_id: string;\n name: string;\n}\n\ninterface LangGraphPlatformActionExecutionMessage extends BaseLangGraphPlatformMessage {\n tool_calls: ToolCall[];\n}\n\nexport type LangGraphPlatformMessage =\n | LangGraphPlatformActionExecutionMessage\n | LangGraphPlatformResultMessage\n | BaseLangGraphPlatformMessage;\n\nexport enum CustomEventNames {\n ManuallyEmitMessage = \"manually_emit_message\",\n ManuallyEmitToolCall = \"manually_emit_tool_call\",\n ManuallyEmitState = \"manually_emit_state\",\n Exit = \"exit\",\n}\n\nexport interface PredictStateTool {\n tool: string;\n state_key: string;\n tool_argument: string;\n}\n","import { Message as LangGraphMessage } from \"@langchain/langgraph-sdk\";\nimport { State, SchemaKeys } from \"./types\";\nimport { Message, ToolCall } from \"@ag-ui/client\";\n\nexport const DEFAULT_SCHEMA_KEYS = [\"tools\"];\n\nexport function filterObjectBySchemaKeys(obj: Record<string, any>, schemaKeys: string[]) {\n return Object.fromEntries(Object.entries(obj).filter(([key]) => schemaKeys.includes(key)));\n}\n\nexport function getStreamPayloadInput({\n mode,\n state,\n schemaKeys,\n}: {\n mode: \"start\" | \"continue\";\n state: State;\n schemaKeys: SchemaKeys;\n}) {\n let input = mode === \"start\" ? state : null;\n // Do not input keys that are not part of the input schema\n if (input && schemaKeys?.input) {\n input = filterObjectBySchemaKeys(input, [...DEFAULT_SCHEMA_KEYS, ...schemaKeys.input]);\n }\n\n return input;\n}\n\nexport function langchainMessagesToAgui(messages: LangGraphMessage[]): Message[] {\n return messages.map((message) => {\n switch (message.type) {\n case \"human\":\n return {\n id: message.id!,\n role: \"user\",\n content: stringifyIfNeeded(message.content),\n };\n case \"ai\":\n return {\n id: message.id!,\n role: \"assistant\",\n content: stringifyIfNeeded(message.content),\n toolCalls: message.tool_calls?.map((tc) => ({\n id: tc.id!,\n type: \"function\",\n function: {\n name: tc.name,\n arguments: JSON.stringify(tc.args),\n },\n })),\n };\n case \"system\":\n return {\n id: message.id!,\n role: \"system\",\n content: stringifyIfNeeded(message.content),\n };\n case \"tool\":\n return {\n id: message.id!,\n role: \"tool\",\n content: stringifyIfNeeded(message.content),\n toolCallId: message.tool_call_id,\n };\n default:\n throw new Error(\"message type returned from LangGraph is not supported.\");\n }\n });\n}\n\nexport function aguiMessagesToLangChain(messages: Message[]): LangGraphMessage[] {\n return messages.map((message, index) => {\n switch (message.role) {\n case \"user\":\n return {\n id: message.id,\n role: message.role,\n content: message.content,\n type: \"human\",\n };\n case \"assistant\":\n return {\n id: message.id,\n type: \"ai\",\n role: message.role,\n content: message.content ?? \"\",\n tool_calls: (message.toolCalls ?? []).map((tc: ToolCall) => ({\n id: tc.id,\n name: tc.function.name,\n args: JSON.parse(tc.function.arguments),\n type: \"tool_call\",\n })),\n };\n case \"system\":\n return {\n id: message.id,\n role: message.role,\n content: message.content,\n type: \"system\",\n };\n case \"tool\":\n return {\n content: message.content,\n role: message.role,\n type: message.role,\n tool_call_id: message.toolCallId,\n id: message.id,\n };\n default:\n console.error(`Message role ${message.role} is not implemented`);\n throw new Error(\"message role is not supported.\");\n }\n });\n}\n\nfunction stringifyIfNeeded(item: any) {\n if (typeof item === \"string\") return item;\n return JSON.stringify(item);\n}\n"],"mappings":"utBAAA,OAAS,cAAAA,OAA8B,OACvC,OACE,UAAUC,OASL,2BACP,OAAS,cAAAC,OAAkB,SAC3B,OAAS,iBAAAC,OAAqB,2BCVvB,IAAKC,QACVA,EAAA,aAAe,iBACfA,EAAA,cAAgB,kBAChBA,EAAA,WAAa,eACbA,EAAA,iBAAmB,sBACnBA,EAAA,kBAAoB,uBACpBA,EAAA,eAAiB,oBACjBA,EAAA,YAAc,gBACdA,EAAA,UAAY,cACZA,EAAA,cAAgB,kBAChBA,EAAA,YAAc,eAVJA,QAAA,IA4EAC,QACVA,EAAA,oBAAsB,wBACtBA,EAAA,qBAAuB,0BACvBA,EAAA,kBAAoB,sBACpBA,EAAA,KAAO,OAJGA,QAAA,IDtDZ,OACE,iBAAAC,GAGA,aAAAC,MAiBK,gBE1CA,IAAMC,EAAsB,CAAC,OAAO,EAEpC,SAASC,EAAyBC,EAA0BC,EAAsB,CACvF,OAAO,OAAO,YAAY,OAAO,QAAQD,CAAG,EAAE,OAAO,CAAC,CAACE,CAAG,IAAMD,EAAW,SAASC,CAAG,CAAC,CAAC,CAC3F,CAEO,SAASC,EAAsB,CACpC,KAAAC,EACA,MAAAC,EACA,WAAAJ,CACF,EAIG,CACD,IAAIK,EAAQF,IAAS,QAAUC,EAAQ,KAEvC,OAAIC,IAASL,GAAA,MAAAA,EAAY,SACvBK,EAAQP,EAAyBO,EAAO,CAAC,GAAGR,EAAqB,GAAGG,EAAW,KAAK,CAAC,GAGhFK,CACT,CAEO,SAASC,EAAwBC,EAAyC,CAC/E,OAAOA,EAAS,IAAKC,GAAY,CA7BnC,IAAAC,EA8BI,OAAQD,EAAQ,KAAM,CACpB,IAAK,QACH,MAAO,CACL,GAAIA,EAAQ,GACZ,KAAM,OACN,QAASE,EAAkBF,EAAQ,OAAO,CAC5C,EACF,IAAK,KACH,MAAO,CACL,GAAIA,EAAQ,GACZ,KAAM,YACN,QAASE,EAAkBF,EAAQ,OAAO,EAC1C,WAAWC,EAAAD,EAAQ,aAAR,YAAAC,EAAoB,IAAKE,IAAQ,CAC1C,GAAIA,EAAG,GACP,KAAM,WACN,SAAU,CACR,KAAMA,EAAG,KACT,UAAW,KAAK,UAAUA,EAAG,IAAI,CACnC,CACF,GACF,EACF,IAAK,SACH,MAAO,CACL,GAAIH,EAAQ,GACZ,KAAM,SACN,QAASE,EAAkBF,EAAQ,OAAO,CAC5C,EACF,IAAK,OACH,MAAO,CACL,GAAIA,EAAQ,GACZ,KAAM,OACN,QAASE,EAAkBF,EAAQ,OAAO,EAC1C,WAAYA,EAAQ,YACtB,EACF,QACE,MAAM,IAAI,MAAM,wDAAwD,CAC5E,CACF,CAAC,CACH,CAEO,SAASI,GAAwBL,EAAyC,CAC/E,OAAOA,EAAS,IAAI,CAACC,EAASK,IAAU,CAvE1C,IAAAJ,EAAAK,EAwEI,OAAQN,EAAQ,KAAM,CACpB,IAAK,OACH,MAAO,CACL,GAAIA,EAAQ,GACZ,KAAMA,EAAQ,KACd,QAASA,EAAQ,QACjB,KAAM,OACR,EACF,IAAK,YACH,MAAO,CACL,GAAIA,EAAQ,GACZ,KAAM,KACN,KAAMA,EAAQ,KACd,SAASC,EAAAD,EAAQ,UAAR,KAAAC,EAAmB,GAC5B,aAAaK,EAAAN,EAAQ,YAAR,KAAAM,EAAqB,CAAC,GAAG,IAAKH,IAAkB,CAC3D,GAAIA,EAAG,GACP,KAAMA,EAAG,SAAS,KAClB,KAAM,KAAK,MAAMA,EAAG,SAAS,SAAS,EACtC,KAAM,WACR,EAAE,CACJ,EACF,IAAK,SACH,MAAO,CACL,GAAIH,EAAQ,GACZ,KAAMA,EAAQ,KACd,QAASA,EAAQ,QACjB,KAAM,QACR,EACF,IAAK,OACH,MAAO,CACL,QAASA,EAAQ,QACjB,KAAMA,EAAQ,KACd,KAAMA,EAAQ,KACd,aAAcA,EAAQ,WACtB,GAAIA,EAAQ,EACd,EACF,QACE,cAAQ,MAAM,gBAAgBA,EAAQ,IAAI,qBAAqB,EACzD,IAAI,MAAM,gCAAgC,CACpD,CACF,CAAC,CACH,CAEA,SAASE,EAAkBK,EAAW,CACpC,OAAI,OAAOA,GAAS,SAAiBA,EAC9B,KAAK,UAAUA,CAAI,CAC5B,CFzBO,IAAMC,GAAN,cAA6BC,EAAc,CAWhD,YAAYC,EAA8B,CAxG5C,IAAAC,EAAAC,EAyGI,MAAMF,CAAM,EACZ,KAAK,kBAAoB,CAAC,EAC1B,KAAK,UAAYA,EAAO,UACxB,KAAK,QAAUA,EAAO,QACtB,KAAK,gBAAkBA,EAAO,gBAC9B,KAAK,QACHE,EAAAF,GAAA,YAAAA,EAAQ,SAAR,KAAAE,EACA,IAAIC,GAAgB,CAClB,OAAQH,EAAO,cACf,OAAQA,EAAO,gBACf,eAAgBI,EAAA,IAAMH,EAAAD,EAAO,kBAAP,KAAAC,EAA0B,CAAC,EACnD,CAAC,CACL,CAEA,cAAcI,EAAwB,CACpC,YAAK,WAAW,KAAKA,CAAK,EACnB,EACT,CAEA,IAAIC,EAAsB,CACxB,YAAK,UAAY,CACf,GAAIA,EAAM,MACV,SAAUA,EAAM,QAClB,EACO,IAAIC,GAA6BC,IACtC,KAAK,mBAAmBF,EAAOE,CAAU,EAClC,IAAM,CAAC,EACf,CACH,CAEA,MAAM,mBAAmBF,EAA8BE,EAAyC,CAvIlG,IAAAP,EAAAC,EAAAO,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAwII,GAAI,CAAE,SAAUC,EAAe,MAAAC,EAAO,SAAAC,EAAU,MAAAC,EAAO,QAAAC,EAAS,eAAAC,CAAe,EAAIlB,EACnF,KAAK,WAAaE,EAClB,IAAIiB,EAAa,GAGjB,KAAK,UAAW,qBAAuB,KAEvC,KAAK,UAAW,UAAWxB,EAAAK,EAAM,iBAAN,YAAAL,EAAsB,SAEjD,IAAMyB,EAAWP,GAAA,KAAAA,EAAiBQ,GAAW,EAExC,KAAK,YACR,KAAK,UAAY,MAAM,KAAK,aAAa,GAG3C,IAAIC,EAAa,MAAM,KAAK,gCAAgCF,CAAQ,EAE9DG,EAAmBD,EAAW,OAC9BE,EAAyBC,GAAwBV,CAAQ,EAE/DD,EAAM,SAAWS,EAAiB,SAClCT,EAAQ,KAAK,2BAA2BA,EAAOU,EAAwBR,CAAK,EAE5E,IAAMU,EACJN,GAAY,KAAK,UAAW,UAAY,WAAa,KAAK,UAAW,SACjE,WACA,QAEFM,IAAS,YAAc,GAAC9B,EAAAsB,GAAA,YAAAA,EAAgB,UAAhB,MAAAtB,EAAyB,SACnD,MAAM,KAAK,OAAO,QAAQ,YAAYwB,EAAU,CAC9C,OAAQN,EACR,OAAQ,KAAK,UAAW,QAC1B,CAAC,EAGH,KAAK,UAAW,WAAa,MAAM,KAAK,cAAc,EAEtD,IAAMa,EAAeC,EAAsB,CACzC,KAAAF,EACA,MAAAZ,EACA,WAAY,KAAK,UAAW,UAC9B,CAAC,EAEKe,EAAY,MAAM,KAAK,OAAO,WAAW,SAAS,KAAK,UAAU,YAAY,EAE/EC,EACEC,EAAiB,CAAC,KAAK,gBAAiBb,GAAA,YAAAA,EAAgB,MAAM,EAAE,OACpE,OACF,EACIa,EAAe,SACjBD,EAAgB,MAAM,KAAK,aAAa,CACtC,QAASC,EACT,UAAW,KAAK,UAChB,WAAY,KAAK,UAAW,UAC9B,CAAC,GAEH,IAAMC,EAAUC,EAAAnC,EAAA,GACXoB,GADW,CAEd,YACEf,EAAAe,GAAA,YAAAA,EAAgB,aAAhB,KAAAf,EAA+B,CAAC,SAAU,SAAU,SAAS,EAC/D,MAAOwB,EACP,OAAQG,CACV,GAGMI,GAAc5B,GAAAD,GAAAD,EAAAkB,EAAW,QAAX,YAAAlB,EAAmB,KAAnB,YAAAC,EAAuB,aAAvB,KAAAC,EAAqC,CAAC,EAC1D,GAAI4B,GAAA,MAAAA,EAAY,QAAU,GAAC3B,EAAAW,GAAA,YAAAA,EAAgB,UAAhB,MAAAX,EAAyB,QAClD,YAAK,cAAc,CACjB,KAAM4B,EAAU,YAChB,SAAAf,EACA,MAAOpB,EAAM,KACf,CAAC,EAEDkC,EAAW,QAASE,GAAc,CAChC,KAAK,cAAc,CACjB,KAAMD,EAAU,OAChB,oBACA,MACE,OAAOC,EAAU,OAAU,SAAWA,EAAU,MAAQ,KAAK,UAAUA,EAAU,KAAK,EACxF,SAAUA,CACZ,CAAC,CACH,CAAC,EAED,KAAK,cAAc,CACjB,KAAMD,EAAU,aAChB,SAAAf,EACA,MAAOpB,EAAM,KACf,CAAC,EACME,EAAW,SAAS,EAE7B,IAAMmC,EAAiB,KAAK,OAAO,KAAK,OAAOjB,EAAU,KAAK,UAAU,aAAcY,CAAO,EAE7F,KAAK,UAAW,aAAe,KAC/B,IAAIM,EAAoB,CAAC,EACrBC,EAAezB,EAEnB,GAAI,CACF,KAAK,cAAc,CACjB,KAAMqB,EAAU,YAChB,SAAAf,EACA,MAAO,KAAK,UAAW,EACzB,CAAC,EAED,YAAAoB,GAAAC,EAAsCJ,GAAtCK,GAAAC,GAAAC,GAAAF,GAAA,EAAAC,GAAA,MAAAH,GAAA,aAAAE,GAAA,GAAsD,CAA3C,IAAIG,EAAfF,GAAA,MAEE,GAAI,CAACX,EAAQ,WAAW,SAASa,EAAoB,KAAmB,EACtE,SAUF,IAAMC,EAAQD,EAEd,GAAIA,EAAoB,QAAU,QAAS,CACzC,KAAK,cAAc,CACjB,KAAMV,EAAU,UAChB,QAASU,EAAoB,KAAK,QAClC,SAAUA,CACZ,CAAC,EACD,KACF,CAEA,GAAIA,EAAoB,QAAU,UAAW,SAE7C,GAAIA,EAAoB,QAAU,SAAU,CAC1CP,EAAoBQ,EAAM,KAC1B,QACF,CAEA,IAAMC,EAAYD,EAAM,KAClBE,EAAkBD,EAAU,SAAS,eACrCE,EAAYF,EAAU,MACtBG,GAAWH,EAAU,SAwC3B,GAvCA,KAAK,UAAW,GAAKG,GAAS,OAE1BF,GAAmBA,IAAoB,KAAK,UAAW,WACrD,KAAK,UAAW,UAClB,KAAK,cAAc,CACjB,KAAMb,EAAU,cAChB,SAAU,KAAK,UAAW,QAC5B,CAAC,EAGCa,IACF,KAAK,cAAc,CACjB,KAAMb,EAAU,aAChB,SAAUa,CACZ,CAAC,EACD,KAAK,UAAW,SAAWA,IAI/B7B,EACEA,GACC8B,IAAc,mBACbF,EAAU,OAAS,OAEvB,KAAK,UAAW,YACd,KAAK,UAAW,WAAaC,GAC7BC,IAAc,eACZ,KAAK,UAAW,cAClB,KAAK,UAAW,qBAAuB,MAKrCpB,EAAU,MAAS,KAAMsB,IAASA,GAAK,KAAOH,CAAe,IAC/D,KAAK,UAAW,SAAWA,GAG7BT,GAAe/B,EAAA,KAAK,UAAW,uBAAhB,KAAAA,EAAwC8B,EAEnD,CAAC,KAAK,UAAW,SACnB,UAGmB,KAAK,UAAUC,CAAY,IAAM,KAAK,UAAUzB,CAAK,GAItE,KAAK,UAAW,cAAgB,KAAK,UAAW,UAChD,KAAK,UAAW,cAClB,CAAS,KAAK,qBAAqB,KAAK,UAAW,EAAE,IAErDA,EAAQyB,EACR,KAAK,UAAW,aAAe,KAAK,UAAW,SAE/C,KAAK,cAAc,CACjB,KAAMJ,EAAU,eAChB,SAAU,KAAK,iBAAiBrB,CAAK,EACrC,SAAUgC,CACZ,CAAC,GAGH,KAAK,cAAc,CACjB,KAAMX,EAAU,IAChB,MAAOY,CACT,CAAC,EAED,KAAK,kBAAkBA,EAAWjC,CAAK,CACzC,QAtGA6B,GA/ON,CA+OMC,GAAA,CAAAD,WAAA,KAAAD,KAAAC,GAAAH,GAAA,eAAAG,GAAA,KAAAH,WAAA,IAAAI,GAAA,MAAAA,GAAA,IAwGA9B,EAAQ,MAAM,KAAK,OAAO,QAAQ,SAASM,CAAQ,EACnD,IAAMc,GAAcvB,GAAAD,GAAAD,EAAAK,EAAM,QAAN,YAAAL,EAAc,KAAd,YAAAC,EAAkB,aAAlB,KAAAC,EAAgC,CAAC,EACrD,YAAK,UAAW,SAAWuB,EACvB,KAAK,UAAW,SAChB,OAAO,KAAKpB,EAAM,SAAS,MAAM,EAAE,CAAC,EAExCoB,EAAW,QAASE,GAAc,CAChC,KAAK,cAAc,CACjB,KAAMD,EAAU,OAChB,oBACA,MACE,OAAOC,EAAU,OAAU,SAAWA,EAAU,MAAQ,KAAK,UAAUA,EAAU,KAAK,EACxF,SAAUA,CACZ,CAAC,CACH,CAAC,EAED,KAAK,cAAc,CACjB,KAAMD,EAAU,eAChB,SAAU,KAAK,iBAAiBrB,EAAM,MAAM,CAC9C,CAAC,EACD,KAAK,cAAc,CACjB,KAAMqB,EAAU,kBAChB,SAAUiB,GAAwBxC,EAAAE,EAAM,OAAO,WAAb,KAAAF,EAAyB,CAAC,CAAC,CAC/D,CAAC,EACG,KAAK,UAAW,UAClB,KAAK,cAAc,CACjB,KAAMuB,EAAU,cAChB,SAAU,KAAK,UAAW,QAC5B,CAAC,EAEH,KAAK,cAAc,CACjB,KAAMA,EAAU,aAChB,SAAAf,EACA,MAAO,KAAK,UAAW,EACzB,CAAC,EACMlB,EAAW,SAAS,CAC7B,OAASmD,EAAG,CACV,OAAOnD,EAAW,MAAMmD,CAAC,CAC3B,CACF,CAEA,kBAAkBtD,EAAYe,EAAoB,CAhYpD,IAAAnB,EAAAC,EAAAO,EAAAC,EAAAC,EAAAC,EAiYI,OAAQP,EAAM,MAAO,CACnB,2BACE,IAAIuD,GAAqB3D,EAAAI,EAAM,SAAS,eAAe,IAA9B,KAAAJ,EAAmC,GACxD4D,GAAsB3D,EAAAG,EAAM,SAAS,iBAAiB,IAAhC,KAAAH,EAAqC,GAE/D,GAAIG,EAAM,KAAK,MAAM,kBAAkB,cAAe,OACtD,IAAMyD,EAAgB,KAAK,qBAAqB,KAAK,UAAW,EAAE,EAC5DC,EAAmB,GAAQD,GAAA,MAAAA,EAAe,IAC1CE,GAAevD,EAAAJ,EAAM,KAAK,MAAM,mBAAjB,YAAAI,EAAoC,GACnDwD,GAA6BvD,EAAAL,EAAM,SAAS,gBAAf,YAAAK,EAAiC,KACjEwD,GAAuCA,EAAiB,QAASF,GAAA,YAAAA,EAAc,OAG5EG,EAAuB,CAACJ,IAAoBC,GAAA,YAAAA,EAAc,MAC1DI,EACJL,IAAoBD,GAAA,YAAAA,EAAe,aAAcE,EAAa,KAC1DK,EAAqBN,IAAoBD,GAAA,YAAAA,EAAe,aAAc,CAACE,EAEvEM,EAAsB,CAACP,GAAoB,CAACC,EAC5CO,EAAwBR,GAAoB,CAACC,EAC7CQ,EACJT,GAAoB,EAACD,GAAA,MAAAA,EAAe,aAAc,CAACS,EAUrD,GARIN,GACF,KAAK,cAAc,CACjB,KAAMxB,EAAU,OAChB,KAAM,eACN,MAAOpC,EAAM,SAAS,aACxB,CAAC,EAGCgE,EAAoB,CACL,KAAK,cAAc,CAClC,KAAM5B,EAAU,cAChB,WAAYqB,EAAc,WAC1B,SAAUzD,CACZ,CAAC,IAEC,KAAK,kBAAkB,KAAK,UAAW,EAAE,EAAI,MAE/C,KACF,CAEA,GAAImE,EAAmB,CACJ,KAAK,cAAc,CAClC,KAAM/B,EAAU,iBAChB,UAAWqB,EAAe,GAC1B,SAAUzD,CACZ,CAAC,IAEC,KAAK,kBAAkB,KAAK,UAAW,EAAE,EAAI,MAE/C,KACF,CAEA,GAAI8D,GAAwBN,EAAqB,CAC9B,KAAK,cAAc,CAClC,KAAMpB,EAAU,gBAChB,WAAYuB,EAAa,GACzB,aAAcA,EAAa,KAC3B,gBAAiB3D,EAAM,KAAK,MAAM,GAClC,SAAUA,CACZ,CAAC,GAEC,KAAK,qBAAqB,KAAK,UAAW,GAAI,CAC5C,GAAIA,EAAM,KAAK,MAAM,GACrB,WAAY2D,EAAa,GACzB,aAAcA,EAAa,IAC7B,CAAC,EAEH,KACF,CAGA,GAAII,GAAuBP,EAAqB,CAC9C,KAAK,cAAc,CACjB,KAAMpB,EAAU,eAChB,WAAYqB,EAAc,WAC1B,MAAOE,EAAa,KACpB,SAAU3D,CACZ,CAAC,EACD,KACF,CAGA,GAAIiE,GAAuBV,EAAoB,CAC5B,KAAK,cAAc,CAClC,KAAMnB,EAAU,mBAChB,KAAM,YACN,UAAWpC,EAAM,KAAK,MAAM,GAC5B,SAAUA,CACZ,CAAC,GAEC,KAAK,qBAAqB,KAAK,UAAW,GAAI,CAC5C,GAAIA,EAAM,KAAK,MAAM,GACrB,WAAY,KACZ,aAAc,IAChB,CAAC,EAEH,KACF,CAEA,GAAIkE,GAAyBX,EAAoB,CAC/C,KAAK,cAAc,CACjB,KAAMnB,EAAU,qBAChB,UAAWqB,EAAe,GAC1B,MAAOzD,EAAM,KAAK,MAAM,QACxB,SAAUA,CACZ,CAAC,EACD,KACF,CAEA,MACF,wBACE,IAAIM,EAAA,KAAK,qBAAqB,KAAK,UAAW,EAAE,IAA5C,MAAAA,EAA+C,WAAY,CAC5C,KAAK,cAAc,CAClC,KAAM8B,EAAU,cAChB,WAAY,KAAK,qBAAqB,KAAK,UAAW,EAAE,EAAG,WAC3D,SAAUpC,CACZ,CAAC,IAEC,KAAK,kBAAkB,KAAK,UAAW,EAAE,EAAI,MAE/C,KACF,CACA,IAAIO,EAAA,KAAK,qBAAqB,KAAK,UAAW,EAAE,IAA5C,MAAAA,EAA+C,GAAI,CACpC,KAAK,cAAc,CAClC,KAAM6B,EAAU,iBAChB,UAAW,KAAK,qBAAqB,KAAK,UAAW,EAAE,EAAG,GAC1D,SAAUpC,CACZ,CAAC,IAEC,KAAK,kBAAkB,KAAK,UAAW,EAAE,EAAI,MAE/C,KACF,CACA,MACF,sBACE,GAAIA,EAAM,OAAS,wBAAsC,CACvD,KAAK,cAAc,CACjB,KAAMoC,EAAU,mBAChB,KAAM,YACN,UAAWpC,EAAM,KAAK,WACtB,SAAUA,CACZ,CAAC,EACD,KAAK,cAAc,CACjB,KAAMoC,EAAU,qBAChB,UAAWpC,EAAM,KAAK,WACtB,MAAOA,EAAM,KAAK,QAClB,SAAUA,CACZ,CAAC,EACD,KAAK,cAAc,CACjB,KAAMoC,EAAU,iBAChB,UAAWpC,EAAM,KAAK,WACtB,SAAUA,CACZ,CAAC,EACD,KACF,CAEA,GAAIA,EAAM,OAAS,0BAAuC,CACxD,KAAK,cAAc,CACjB,KAAMoC,EAAU,gBAChB,WAAYpC,EAAM,KAAK,GACvB,aAAcA,EAAM,KAAK,KACzB,gBAAiBA,EAAM,KAAK,GAC5B,SAAUA,CACZ,CAAC,EACD,KAAK,cAAc,CACjB,KAAMoC,EAAU,eAChB,WAAYpC,EAAM,KAAK,GACvB,MAAOA,EAAM,KAAK,KAClB,SAAUA,CACZ,CAAC,EACD,KAAK,cAAc,CACjB,KAAMoC,EAAU,cAChB,WAAYpC,EAAM,KAAK,GACvB,SAAUA,CACZ,CAAC,EACD,KACF,CAEIA,EAAM,OAAS,wBACjB,KAAK,UAAW,qBAAuBA,EAAM,KAC7C,KAAK,cAAc,CACjB,KAAMoC,EAAU,eAChB,SAAU,KAAK,iBAAiBrB,CAAK,EACrC,SAAUf,CACZ,CAAC,GAGH,KAAK,cAAc,CACjB,KAAMoC,EAAU,OAChB,KAAMpC,EAAM,KACZ,MAAOA,EAAM,KACb,SAAUA,CACZ,CAAC,EACD,KACJ,CACF,CAEA,iBAAiBe,EAAc,CAC7B,IAAMqD,EAAa,KAAK,UAAW,WAEnC,OAAIA,GAAA,MAAAA,EAAY,SACdrD,EAAQsD,EAAyBtD,EAAO,CAAC,GAAGuD,EAAqB,GAAGF,EAAW,MAAM,CAAC,GAGjFrD,CACT,CAEA,MAAM,gCAAgCM,EAA4C,CAChF,IAAIE,EAAa,CAAE,OAAQ,CAAC,CAAE,EAC9B,GAAI,CACF,MAAM,KAAK,OAAO,QAAQ,IAAIF,CAAQ,EACtCE,EAAa,MAAM,KAAK,OAAO,QAAQ,SAASF,CAAQ,CAC1D,OAASwB,EAAO,CACd,MAAM,KAAK,OAAO,QAAQ,OAAO,CAAE,SAAAxB,CAAS,CAAC,CAC/C,CAEA,OAAOE,CACT,CAEA,MAAM,aAAa,CACjB,QAAAgD,EACA,UAAAC,EACA,WAAAJ,CACF,EAIG,CACD,OAAOG,EAAQ,OAAO,CAACE,EAAKC,IAAQ,CAxmBxC,IAAA9E,EAymBM,IAAI+E,EAAuBF,EAAI,aAE3BC,EAAI,eACNC,EAAuBP,GAAA,MAAAA,EAAY,OAC/BC,EAAyBK,GAAA,YAAAA,EAAK,aAAc,CAC1C,GAAGJ,EACH,IAAI1E,EAAAwE,GAAA,YAAAA,EAAY,SAAZ,KAAAxE,EAAsB,CAAC,CAC7B,CAAC,EACD8E,GAAA,YAAAA,EAAK,cAGX,IAAME,EAAY1C,EAAAnC,IAAA,GACb0E,GACAC,GAFa,CAGhB,aAAcC,CAChB,GAGME,EACJJ,EAAI,iBAAmB,MAAQC,EAAI,kBAAoB,GAEnDI,EAAsB,KAAK,UAAUF,CAAS,IAAM,KAAK,UAAUH,CAAG,EAGtEM,EACJF,GACA,KAAK,UAAU3C,EAAAnC,EAAA,GAAK6E,GAAL,CAAgB,gBAAiB,IAAK,EAAC,IACpD,KAAK,UAAU1C,EAAAnC,EAAA,GAAK0E,GAAL,CAAU,gBAAiB,IAAK,EAAC,EAEpD,OAAIK,GAAuB,CAACC,EACnBhF,IAAA,GACF0E,GACAG,GAIAH,CACT,EAAGD,EAAU,MAAM,CACrB,CAEA,qBAAqBQ,EAAe,CAClC,OAAO,KAAK,kBAAkBA,CAAK,CACrC,CAEA,qBAAqBA,EAAeC,EAAyB,CAC3D,KAAK,kBAAoB/C,EAAAnC,EAAA,GACpB,KAAK,mBADe,CAEvB,CAACiF,CAAK,EAAGjF,IAAA,GACH,KAAK,kBAAkBiF,CAAK,GAC7BC,EAEP,EACF,CAEA,MAAM,cAAmC,CACvC,IAAMC,EAAa,MAAM,KAAK,OAAO,WAAW,OAAO,EACjDC,EAAqBD,EAAW,KACnCE,GACCA,EAAa,eAAiB,KAAK,SAAWA,EAAa,WAAa,KAAK,OACjF,EACA,GAAI,CAACD,EACH,cAAQ,MAAM;AAAA,qCACiB,KAAK,OAAO,gBAAgB,KAAK,OAAO;AAAA;AAAA;AAAA,yCAEpCD,EAAW,IAAKG,GAAM,GAAGA,EAAE,QAAQ,SAASA,EAAE,YAAY,GAAG,EAAE,KAAK,IAAI,CAAC;AAAA,OAC3G,EACK,IAAI,MAAM,mBAAmB,EAGrC,OAAOF,CACT,CAEA,MAAM,eAAqC,CAjrB7C,IAAAvF,EAAAC,EAAAO,EAkrBI,IAAMkF,EAAgB,CAAC,UAAU,EAEjC,GAAI,CACF,IAAMC,EAAc,MAAM,KAAK,OAAO,WAAW,WAAW,KAAK,UAAW,YAAY,EACpFC,EAAe,KAInB,IAHI5F,EAAA2F,EAAY,gBAAZ,MAAA3F,EAA2B,aAC7B4F,EAAe,OAAO,KAAKD,EAAY,cAAc,UAAU,GAE7D,GAAC1F,EAAA0F,EAAY,eAAZ,MAAA1F,EAA0B,aAAc,GAACO,EAAAmF,EAAY,gBAAZ,MAAAnF,EAA2B,YACvE,MAAO,CAAE,OAAQ,CAAC,EAAG,MAAOkF,EAAe,OAAQA,CAAc,EAEnE,IAAMG,EAAc,OAAO,KAAKF,EAAY,aAAa,UAAU,EAC7DG,EAAe,OAAO,KAAKH,EAAY,cAAc,UAAU,EAErE,MAAO,CACL,MAAOE,GAAeA,EAAY,OAAS,CAAC,GAAGA,EAAa,GAAGH,CAAa,EAAI,KAChF,OAAQI,GAAgBA,EAAa,OAAS,CAAC,GAAGA,EAAc,GAAGJ,CAAa,EAAI,KACpF,OAAQE,CACV,CACF,OAASlC,EAAG,CACV,MAAO,CAAE,OAAQ,CAAC,EAAG,MAAOgC,EAAe,OAAQA,CAAc,CACnE,CACF,CAEA,2BAA2BvE,EAAcC,EAA8BC,EAAmB,CA1sB5F,IAAArB,EA2sBQoB,EAAS,OAAS,GAAK,SAAUA,EAAS,CAAC,GAAKA,EAAS,CAAC,EAAE,OAAS,WAEvEA,EAAWA,EAAS,MAAM,CAAC,GAI7B,IAAM2E,EAA+C5E,EAAM,UAAY,CAAC,EAClE6E,EAAqB,IAAI,IAAID,EAAiB,IAAKE,GAAYA,EAAQ,EAAE,CAAC,EAC1EC,EAAa,IAAI,IAAI9E,EAAS,IAAK6E,GAAYA,EAAQ,EAAE,CAAC,EAE5DE,EAAmC,CAAC,EACpC/E,EAAS,OAAS2E,EAAiB,SAErCI,EAAkBJ,EACf,OAAQK,GAAM,CAACF,EAAW,IAAIE,EAAE,EAAE,CAAC,EACnC,IAAKA,GAAM,IAAIC,GAAc,CAAE,GAAID,EAAE,EAAI,CAAC,CAAC,GAGhD,IAAME,EAAclF,EAAS,OAAQ6E,GAAY,CAACD,EAAmB,IAAIC,EAAQ,EAAE,CAAC,EAEpF,OAAO3D,EAAAnC,EAAA,GACFgB,GADE,CAEL,SAAU,CAAC,GAAGgF,EAAiB,GAAGG,CAAW,EAC7C,MAAO,CAAC,IAAItG,EAAAmB,EAAM,QAAN,KAAAnB,EAAe,CAAC,EAAI,GAAGqB,CAAK,CAC1C,EACF,CACF","names":["Observable","LangGraphClient","randomUUID","RemoveMessage","LangGraphEventTypes","CustomEventNames","AbstractAgent","EventType","DEFAULT_SCHEMA_KEYS","filterObjectBySchemaKeys","obj","schemaKeys","key","getStreamPayloadInput","mode","state","input","langchainMessagesToAgui","messages","message","_a","stringifyIfNeeded","tc","aguiMessagesToLangChain","index","_b","item","LangGraphAgent","AbstractAgent","config","_a","_b","LangGraphClient","__spreadValues","event","input","Observable","subscriber","_c","_d","_e","_f","_g","_h","_i","_j","_k","_l","inputThreadId","state","messages","tools","context","forwardedProps","shouldExit","threadId","randomUUID","agentState","agentStateValues","aguiToLangChainMessage","aguiMessagesToLangChain","mode","payloadInput","getStreamPayloadInput","graphInfo","payloadConfig","configsToMerge","payload","__spreadProps","interrupts","EventType","interrupt","streamResponse","latestStateValues","updatedState","iter","__forAwait","more","temp","error","streamResponseChunk","chunk","chunkData","currentNodeName","eventType","metadata","node","langchainMessagesToAgui","e","shouldEmitMessages","shouldEmitToolCalls","currentStream","hasCurrentStream","toolCallData","toolCallUsedToPredictState","predictStateTool","isToolCallStartEvent","isToolCallArgsEvent","isToolCallEndEvent","isMessageStartEvent","isMessageContentEvent","isMessageEndEvent","schemaKeys","filterObjectBySchemaKeys","DEFAULT_SCHEMA_KEYS","configs","assistant","acc","cfg","filteredConfigurable","newConfig","isRecursionLimitSetToDefault","configsAreDifferent","isOnlyRecursionLimitDifferent","runId","data","assistants","retrievedAssistant","searchResult","a","CONSTANT_KEYS","graphSchema","configSchema","inputSchema","outputSchema","existingMessages","existingMessageIds","message","messageIds","removedMessages","m","RemoveMessage","newMessages"]}
|
|
1
|
+
{"version":3,"sources":["../src/index.ts","../src/types.ts","../src/utils.ts"],"sourcesContent":["import { Observable, Subscriber } from \"rxjs\";\nimport {\n Client as LangGraphClient,\n EventsStreamEvent,\n StreamMode,\n Config as LangGraphConfig,\n ThreadState,\n Assistant,\n Message as LangGraphMessage,\n Config,\n Interrupt,\n Thread\n} from \"@langchain/langgraph-sdk\";\nimport { randomUUID } from \"node:crypto\";\nimport { RemoveMessage } from \"@langchain/core/messages\";\nimport {\n LangGraphPlatformMessage,\n CustomEventNames,\n LangGraphEventTypes,\n State,\n MessagesInProgressRecord,\n ThinkingInProgress,\n SchemaKeys,\n MessageInProgress,\n RunMetadata,\n PredictStateTool,\n LangGraphReasoning\n} from \"./types\";\nimport {\n AbstractAgent,\n AgentConfig,\n CustomEvent,\n EventType,\n MessagesSnapshotEvent,\n RawEvent,\n RunAgentInput,\n RunErrorEvent,\n RunFinishedEvent,\n RunStartedEvent,\n StateDeltaEvent,\n StateSnapshotEvent,\n StepFinishedEvent,\n StepStartedEvent,\n TextMessageContentEvent,\n TextMessageEndEvent,\n TextMessageStartEvent,\n ToolCallArgsEvent,\n ToolCallEndEvent,\n ToolCallStartEvent,\n ThinkingTextMessageStartEvent,\n ThinkingTextMessageContentEvent,\n ThinkingTextMessageEndEvent,\n ThinkingStartEvent,\n ThinkingEndEvent,\n} from \"@ag-ui/client\";\nimport { RunsStreamPayload } from \"@langchain/langgraph-sdk/dist/types\";\nimport {\n aguiMessagesToLangChain,\n DEFAULT_SCHEMA_KEYS,\n filterObjectBySchemaKeys,\n getStreamPayloadInput,\n langchainMessagesToAgui,\n resolveMessageContent,\n resolveReasoningContent\n} from \"@/utils\";\n\nexport type ProcessedEvents =\n | TextMessageStartEvent\n | TextMessageContentEvent\n | TextMessageEndEvent\n | ThinkingTextMessageStartEvent\n | ThinkingTextMessageContentEvent\n | ThinkingTextMessageEndEvent\n | ToolCallStartEvent\n | ToolCallArgsEvent\n | ToolCallEndEvent\n | ThinkingStartEvent\n | ThinkingEndEvent\n | StateSnapshotEvent\n | StateDeltaEvent\n | MessagesSnapshotEvent\n | RawEvent\n | CustomEvent\n | RunStartedEvent\n | RunFinishedEvent\n | RunErrorEvent\n | StepStartedEvent\n | StepFinishedEvent;\n\ntype RunAgentExtendedInput<\n TStreamMode extends StreamMode | StreamMode[] = StreamMode,\n TSubgraphs extends boolean = false,\n> = Omit<RunAgentInput, \"forwardedProps\"> & {\n forwardedProps?: Omit<RunsStreamPayload<TStreamMode, TSubgraphs>, \"input\"> & {\n nodeName?: string;\n };\n};\n\nexport interface LangGraphAgentConfig extends AgentConfig {\n client?: LangGraphClient;\n deploymentUrl: string;\n langsmithApiKey?: string;\n propertyHeaders?: Record<string, string>;\n assistantConfig?: LangGraphConfig;\n agentName?: string;\n graphId: string;\n}\n\nexport class LangGraphAgent extends AbstractAgent {\n client: LangGraphClient;\n assistantConfig?: LangGraphConfig;\n agentName?: string;\n graphId: string;\n assistant?: Assistant;\n messagesInProcess: MessagesInProgressRecord;\n thinkingProcess: null | ThinkingInProgress;\n activeRun?: RunMetadata;\n // @ts-expect-error no need to initialize subscriber right now\n subscriber: Subscriber<ProcessedEvents>;\n\n constructor(config: LangGraphAgentConfig) {\n super(config);\n this.messagesInProcess = {};\n this.agentName = config.agentName;\n this.graphId = config.graphId;\n this.assistantConfig = config.assistantConfig;\n this.thinkingProcess = null\n this.client =\n config?.client ??\n new LangGraphClient({\n apiUrl: config.deploymentUrl,\n apiKey: config.langsmithApiKey,\n defaultHeaders: { ...(config.propertyHeaders ?? {}) },\n });\n }\n\n dispatchEvent(event: ProcessedEvents) {\n this.subscriber.next(event);\n return true;\n }\n\n run(input: RunAgentInput) {\n this.activeRun = {\n id: input.runId,\n threadId: input.threadId,\n };\n return new Observable<ProcessedEvents>((subscriber) => {\n this.handleStreamEvents(input, subscriber);\n return () => {};\n });\n }\n\n async handleStreamEvents(input: RunAgentExtendedInput, subscriber: Subscriber<ProcessedEvents>) {\n let { threadId: inputThreadId, state, messages, tools, context, forwardedProps } = input;\n this.subscriber = subscriber;\n let shouldExit = false;\n // If a manual emittance happens, it is the ultimate source of truth of state, unless a node has exited.\n // Therefore, this value should either hold null, or the only edition of state that should be used.\n this.activeRun!.manuallyEmittedState = null;\n\n this.activeRun!.nodeName = input.forwardedProps?.nodeName;\n\n const threadId = inputThreadId ?? randomUUID();\n\n if (!this.assistant) {\n this.assistant = await this.getAssistant();\n }\n\n const thread = await this.getOrCreateThread(threadId);\n this.activeRun!.threadId = thread.thread_id;\n const agentState = await this.client.threads.getState(thread.thread_id) ?? { values: {} } as ThreadState\n\n const agentStateValues = agentState.values as State;\n const aguiToLangChainMessage = aguiMessagesToLangChain(messages);\n\n state.messages = agentStateValues.messages;\n state = this.langGraphDefaultMergeState(state, aguiToLangChainMessage, tools);\n\n const mode =\n threadId && this.activeRun!.nodeName != \"__end__\" && this.activeRun!.nodeName\n ? \"continue\"\n : \"start\";\n\n if (mode === \"continue\" && !forwardedProps?.command?.resume) {\n await this.client.threads.updateState(threadId, {\n values: state,\n asNode: this.activeRun!.nodeName,\n });\n }\n\n this.activeRun!.schemaKeys = await this.getSchemaKeys();\n\n const payloadInput = getStreamPayloadInput({\n mode,\n state,\n schemaKeys: this.activeRun!.schemaKeys,\n });\n\n const graphInfo = await this.client.assistants.getGraph(this.assistant.assistant_id);\n\n let payloadConfig: LangGraphConfig | undefined;\n const configsToMerge = [this.assistantConfig, forwardedProps?.config].filter(\n Boolean,\n ) as LangGraphConfig[];\n if (configsToMerge.length) {\n payloadConfig = await this.mergeConfigs({\n configs: configsToMerge,\n assistant: this.assistant,\n schemaKeys: this.activeRun!.schemaKeys,\n });\n }\n const payload = {\n ...forwardedProps,\n streamMode:\n forwardedProps?.streamMode ?? ([\"events\", \"values\", \"updates\"] satisfies StreamMode[]),\n input: payloadInput,\n config: payloadConfig,\n };\n\n // If there are still outstanding unresolved interrupts, we must force resolution of them before moving forward\n const interrupts = (agentState.tasks?.[0]?.interrupts ?? []) as Interrupt[];\n if (interrupts?.length && !forwardedProps?.command?.resume) {\n this.dispatchEvent({\n type: EventType.RUN_STARTED,\n threadId,\n runId: input.runId,\n });\n\n interrupts.forEach((interrupt) => {\n this.dispatchEvent({\n type: EventType.CUSTOM,\n name: LangGraphEventTypes.OnInterrupt,\n value:\n typeof interrupt.value === \"string\" ? interrupt.value : JSON.stringify(interrupt.value),\n rawEvent: interrupt,\n });\n });\n\n this.dispatchEvent({\n type: EventType.RUN_FINISHED,\n threadId,\n runId: input.runId,\n });\n return subscriber.complete();\n }\n const streamResponse = this.client.runs.stream(threadId, this.assistant.assistant_id, payload);\n\n this.activeRun!.prevNodeName = null;\n let latestStateValues = {};\n let updatedState = state;\n\n try {\n this.dispatchEvent({\n type: EventType.RUN_STARTED,\n threadId,\n runId: this.activeRun!.id,\n });\n\n for await (let streamResponseChunk of streamResponse) {\n // @ts-ignore\n if (!payload.streamMode.includes(streamResponseChunk.event as StreamMode)) {\n continue;\n }\n // Force event type, as data is not properly defined on the LG side.\n type EventsChunkData = {\n __interrupt__?: any;\n metadata: Record<string, any>;\n event: string;\n data: any;\n [key: string]: unknown;\n };\n const chunk = streamResponseChunk as EventsStreamEvent & { data: EventsChunkData };\n\n if (streamResponseChunk.event === \"error\") {\n this.dispatchEvent({\n type: EventType.RUN_ERROR,\n message: streamResponseChunk.data.message,\n rawEvent: streamResponseChunk,\n });\n break;\n }\n\n if (streamResponseChunk.event === \"updates\") continue;\n\n if (streamResponseChunk.event === \"values\") {\n latestStateValues = chunk.data;\n continue;\n }\n\n const chunkData = chunk.data;\n const currentNodeName = chunkData.metadata.langgraph_node;\n const eventType = chunkData.event;\n const metadata = chunkData.metadata;\n this.activeRun!.id = metadata.run_id;\n\n if (currentNodeName && currentNodeName !== this.activeRun!.nodeName) {\n if (this.activeRun!.nodeName) {\n this.dispatchEvent({\n type: EventType.STEP_FINISHED,\n stepName: this.activeRun!.nodeName,\n });\n }\n\n if (currentNodeName) {\n this.dispatchEvent({\n type: EventType.STEP_STARTED,\n stepName: currentNodeName,\n });\n this.activeRun!.nodeName = currentNodeName;\n }\n }\n\n shouldExit =\n shouldExit ||\n (eventType === LangGraphEventTypes.OnCustomEvent &&\n chunkData.name === CustomEventNames.Exit);\n\n this.activeRun!.exitingNode =\n this.activeRun!.nodeName === currentNodeName &&\n eventType === LangGraphEventTypes.OnChainEnd;\n if (this.activeRun!.exitingNode) {\n this.activeRun!.manuallyEmittedState = null;\n }\n\n // we only want to update the node name under certain conditions\n // since we don't need any internal node names to be sent to the frontend\n if (graphInfo[\"nodes\"].some((node) => node.id === currentNodeName)) {\n this.activeRun!.nodeName = currentNodeName;\n }\n\n updatedState = this.activeRun!.manuallyEmittedState ?? latestStateValues;\n\n if (!this.activeRun!.nodeName) {\n continue;\n }\n\n const hasStateDiff = JSON.stringify(updatedState) !== JSON.stringify(state);\n // We should not update snapshot while a message is in progress.\n if (\n (hasStateDiff ||\n this.activeRun!.prevNodeName != this.activeRun!.nodeName ||\n this.activeRun!.exitingNode) &&\n !Boolean(this.getMessageInProgress(this.activeRun!.id))\n ) {\n state = updatedState;\n this.activeRun!.prevNodeName = this.activeRun!.nodeName;\n\n this.dispatchEvent({\n type: EventType.STATE_SNAPSHOT,\n snapshot: this.getStateSnapshot(state),\n rawEvent: chunk,\n });\n }\n\n this.dispatchEvent({\n type: EventType.RAW,\n event: chunkData,\n });\n\n this.handleSingleEvent(chunkData, state);\n }\n\n state = await this.client.threads.getState(threadId);\n const interrupts = (state.tasks?.[0]?.interrupts ?? []) as Interrupt[];\n this.activeRun!.nodeName = interrupts\n ? this.activeRun!.nodeName\n : Object.keys(state.metadata.writes)[0];\n\n interrupts.forEach((interrupt) => {\n this.dispatchEvent({\n type: EventType.CUSTOM,\n name: LangGraphEventTypes.OnInterrupt,\n value:\n typeof interrupt.value === \"string\" ? interrupt.value : JSON.stringify(interrupt.value),\n rawEvent: interrupt,\n });\n });\n\n this.dispatchEvent({\n type: EventType.STATE_SNAPSHOT,\n snapshot: this.getStateSnapshot(state.values),\n });\n this.dispatchEvent({\n type: EventType.MESSAGES_SNAPSHOT,\n messages: langchainMessagesToAgui(state.values.messages ?? []),\n });\n if (this.activeRun!.nodeName) {\n this.dispatchEvent({\n type: EventType.STEP_FINISHED,\n stepName: this.activeRun!.nodeName!,\n });\n }\n this.dispatchEvent({\n type: EventType.RUN_FINISHED,\n threadId,\n runId: this.activeRun!.id,\n });\n return subscriber.complete();\n } catch (e) {\n return subscriber.error(e);\n }\n }\n\n handleSingleEvent(event: any, state: State): void {\n switch (event.event) {\n case LangGraphEventTypes.OnChatModelStream:\n let shouldEmitMessages = event.metadata[\"emit-messages\"] ?? true;\n let shouldEmitToolCalls = event.metadata[\"emit-tool-calls\"] ?? true;\n\n if (event.data.chunk.response_metadata.finish_reason) return;\n let currentStream = this.getMessageInProgress(this.activeRun!.id);\n const hasCurrentStream = Boolean(currentStream?.id);\n const toolCallData = event.data.chunk.tool_call_chunks?.[0];\n const toolCallUsedToPredictState = event.metadata[\"predict_state\"]?.some(\n (predictStateTool: PredictStateTool) => predictStateTool.tool === toolCallData?.name,\n );\n\n const isToolCallStartEvent = !hasCurrentStream && toolCallData?.name;\n const isToolCallArgsEvent =\n hasCurrentStream && currentStream?.toolCallId && toolCallData?.args;\n const isToolCallEndEvent = hasCurrentStream && currentStream?.toolCallId && !toolCallData;\n\n const reasoningData = resolveReasoningContent(event.data);\n const messageContent = resolveMessageContent(event.data.chunk.content);\n const isMessageContentEvent = Boolean(!toolCallData && messageContent);\n\n const isMessageEndEvent =\n hasCurrentStream && !currentStream?.toolCallId && !isMessageContentEvent;\n\n if (reasoningData) {\n this.handleThinkingEvent(reasoningData)\n break;\n }\n\n if (!reasoningData && this.thinkingProcess) {\n this.dispatchEvent({\n type: EventType.THINKING_TEXT_MESSAGE_END,\n })\n this.dispatchEvent({\n type: EventType.THINKING_END,\n })\n this.thinkingProcess = null;\n }\n\n if (toolCallUsedToPredictState) {\n this.dispatchEvent({\n type: EventType.CUSTOM,\n name: \"PredictState\",\n value: event.metadata[\"predict_state\"],\n });\n }\n\n if (isToolCallEndEvent) {\n const resolved = this.dispatchEvent({\n type: EventType.TOOL_CALL_END,\n toolCallId: currentStream?.toolCallId!,\n rawEvent: event,\n });\n if (resolved) {\n this.messagesInProcess[this.activeRun!.id] = null;\n }\n break;\n }\n\n if (isMessageEndEvent) {\n const resolved = this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_END,\n messageId: currentStream!.id,\n rawEvent: event,\n });\n if (resolved) {\n this.messagesInProcess[this.activeRun!.id] = null;\n }\n break;\n }\n\n if (isToolCallStartEvent && shouldEmitToolCalls) {\n const resolved = this.dispatchEvent({\n type: EventType.TOOL_CALL_START,\n toolCallId: toolCallData.id,\n toolCallName: toolCallData.name,\n parentMessageId: event.data.chunk.id,\n rawEvent: event,\n });\n if (resolved) {\n this.setMessageInProgress(this.activeRun!.id, {\n id: event.data.chunk.id,\n toolCallId: toolCallData.id,\n toolCallName: toolCallData.name,\n });\n }\n break;\n }\n\n // Tool call args: emit ActionExecutionArgs\n if (isToolCallArgsEvent && shouldEmitToolCalls) {\n this.dispatchEvent({\n type: EventType.TOOL_CALL_ARGS,\n toolCallId: currentStream?.toolCallId!,\n delta: toolCallData.args,\n rawEvent: event,\n });\n break;\n }\n\n // Message content: emit TextMessageContent\n if (isMessageContentEvent && shouldEmitMessages) {\n // No existing message yet, also init the message\n if (!currentStream) {\n this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_START,\n role: \"assistant\",\n messageId: event.data.chunk.id,\n rawEvent: event,\n });\n this.setMessageInProgress(this.activeRun!.id, {\n id: event.data.chunk.id,\n toolCallId: null,\n toolCallName: null,\n });\n currentStream = this.getMessageInProgress(this.activeRun!.id);\n }\n\n this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_CONTENT,\n messageId: currentStream!.id,\n delta: messageContent!,\n rawEvent: event,\n });\n break;\n }\n\n break;\n case LangGraphEventTypes.OnChatModelEnd:\n if (this.getMessageInProgress(this.activeRun!.id)?.toolCallId) {\n const resolved = this.dispatchEvent({\n type: EventType.TOOL_CALL_END,\n toolCallId: this.getMessageInProgress(this.activeRun!.id)!.toolCallId!,\n rawEvent: event,\n });\n if (resolved) {\n this.messagesInProcess[this.activeRun!.id] = null;\n }\n break;\n }\n if (this.getMessageInProgress(this.activeRun!.id)?.id) {\n const resolved = this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_END,\n messageId: this.getMessageInProgress(this.activeRun!.id)!.id,\n rawEvent: event,\n });\n if (resolved) {\n this.messagesInProcess[this.activeRun!.id] = null;\n }\n break;\n }\n break;\n case LangGraphEventTypes.OnCustomEvent:\n if (event.name === CustomEventNames.ManuallyEmitMessage) {\n this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_START,\n role: \"assistant\",\n messageId: event.data.message_id,\n rawEvent: event,\n });\n this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_CONTENT,\n messageId: event.data.message_id,\n delta: event.data.message,\n rawEvent: event,\n });\n this.dispatchEvent({\n type: EventType.TEXT_MESSAGE_END,\n messageId: event.data.message_id,\n rawEvent: event,\n });\n break;\n }\n\n if (event.name === CustomEventNames.ManuallyEmitToolCall) {\n this.dispatchEvent({\n type: EventType.TOOL_CALL_START,\n toolCallId: event.data.id,\n toolCallName: event.data.name,\n parentMessageId: event.data.id,\n rawEvent: event,\n });\n this.dispatchEvent({\n type: EventType.TOOL_CALL_ARGS,\n toolCallId: event.data.id,\n delta: event.data.args,\n rawEvent: event,\n });\n this.dispatchEvent({\n type: EventType.TOOL_CALL_END,\n toolCallId: event.data.id,\n rawEvent: event,\n });\n break;\n }\n\n if (event.name === CustomEventNames.ManuallyEmitState) {\n this.activeRun!.manuallyEmittedState = event.data;\n this.dispatchEvent({\n type: EventType.STATE_SNAPSHOT,\n snapshot: this.getStateSnapshot(state),\n rawEvent: event,\n });\n }\n\n this.dispatchEvent({\n type: EventType.CUSTOM,\n name: event.name,\n value: event.data,\n rawEvent: event,\n });\n break;\n }\n }\n\n handleThinkingEvent(reasoningData: LangGraphReasoning) {\n if (!reasoningData || !reasoningData.type || !reasoningData.text) {\n return;\n }\n\n const thinkingStepIndex = reasoningData.index;\n\n if (this.thinkingProcess?.index && this.thinkingProcess.index !== thinkingStepIndex) {\n if (this.thinkingProcess.type) {\n this.dispatchEvent({\n type: EventType.THINKING_TEXT_MESSAGE_END,\n })\n }\n this.dispatchEvent({\n type: EventType.THINKING_END,\n })\n this.thinkingProcess = null;\n }\n\n if (!this.thinkingProcess) {\n // No thinking step yet. Start a new one\n this.dispatchEvent({\n type: EventType.THINKING_START,\n })\n this.thinkingProcess = {\n index: thinkingStepIndex,\n };\n }\n\n\n if (this.thinkingProcess.type !== reasoningData.type) {\n this.dispatchEvent({\n type: EventType.THINKING_TEXT_MESSAGE_START,\n })\n this.thinkingProcess.type = reasoningData.type\n }\n\n if (this.thinkingProcess.type) {\n this.dispatchEvent({\n type: EventType.THINKING_TEXT_MESSAGE_CONTENT,\n delta: reasoningData.text\n })\n }\n }\n\n getStateSnapshot(state: State) {\n const schemaKeys = this.activeRun!.schemaKeys!;\n // Do not emit state keys that are not part of the output schema\n if (schemaKeys?.output) {\n state = filterObjectBySchemaKeys(state, [...DEFAULT_SCHEMA_KEYS, ...schemaKeys.output]);\n }\n // return state\n return state;\n }\n\n async getOrCreateThread(threadId: string): Promise<Thread> {\n let thread: Thread;\n try {\n try {\n thread = await this.getThread(threadId);\n } catch (error) {\n thread = await this.createThread({ threadId });\n }\n } catch (error: unknown) {\n throw new Error(`Failed to create thread: ${(error as Error).message}`);\n }\n\n return thread;\n }\n\n async getThread(threadId: string) {\n return this.client.threads.get(threadId);\n }\n\n async createThread(payload?: Parameters<typeof this.client.threads.create>[0]) {\n return this.client.threads.create(payload);\n }\n\n async mergeConfigs({\n configs,\n assistant,\n schemaKeys,\n }: {\n configs: Config[];\n assistant: Assistant;\n schemaKeys: SchemaKeys;\n }) {\n return configs.reduce((acc, cfg) => {\n let filteredConfigurable = acc.configurable;\n\n if (cfg.configurable) {\n filteredConfigurable = schemaKeys?.config\n ? filterObjectBySchemaKeys(cfg?.configurable, [\n ...DEFAULT_SCHEMA_KEYS,\n ...(schemaKeys?.config ?? []),\n ])\n : cfg?.configurable;\n }\n\n const newConfig = {\n ...acc,\n ...cfg,\n configurable: filteredConfigurable,\n };\n\n // LG does not return recursion limit if it's the default, therefore we check: if no recursion limit is currently set, and the user asked for 25, there is no change.\n const isRecursionLimitSetToDefault =\n acc.recursion_limit == null && cfg.recursion_limit === 25;\n // Deep compare configs to avoid unnecessary update calls\n const configsAreDifferent = JSON.stringify(newConfig) !== JSON.stringify(acc);\n\n // Check if the only difference is the recursion_limit being set to default\n const isOnlyRecursionLimitDifferent =\n isRecursionLimitSetToDefault &&\n JSON.stringify({ ...newConfig, recursion_limit: null }) ===\n JSON.stringify({ ...acc, recursion_limit: null });\n\n if (configsAreDifferent && !isOnlyRecursionLimitDifferent) {\n return {\n ...acc,\n ...newConfig,\n };\n }\n\n return acc;\n }, assistant.config);\n }\n\n getMessageInProgress(runId: string) {\n return this.messagesInProcess[runId];\n }\n\n setMessageInProgress(runId: string, data: MessageInProgress) {\n this.messagesInProcess = {\n ...this.messagesInProcess,\n [runId]: {\n ...(this.messagesInProcess[runId] as MessageInProgress),\n ...data,\n },\n };\n }\n\n async getAssistant(): Promise<Assistant> {\n const assistants = await this.client.assistants.search();\n const retrievedAssistant = assistants.find(\n (searchResult) =>\n searchResult.graph_id === this.graphId,\n );\n if (!retrievedAssistant) {\n console.error(`\n No agent found with graph ID ${this.graphId} found..\\n\n \n These are the available agents: [${assistants.map((a) => `${a.graph_id} (ID: ${a.assistant_id})`).join(\", \")}]\n `);\n throw new Error(\"No agent id found\");\n }\n\n return retrievedAssistant;\n }\n\n async getSchemaKeys(): Promise<SchemaKeys> {\n const CONSTANT_KEYS = [\"messages\"];\n\n try {\n const graphSchema = await this.client.assistants.getSchemas(this.assistant!.assistant_id);\n let configSchema = null;\n if (graphSchema.config_schema?.properties) {\n configSchema = Object.keys(graphSchema.config_schema.properties);\n }\n if (!graphSchema.input_schema?.properties || !graphSchema.output_schema?.properties) {\n return { config: [], input: CONSTANT_KEYS, output: CONSTANT_KEYS };\n }\n const inputSchema = Object.keys(graphSchema.input_schema.properties);\n const outputSchema = Object.keys(graphSchema.output_schema.properties);\n\n return {\n input: inputSchema && inputSchema.length ? [...inputSchema, ...CONSTANT_KEYS] : null,\n output: outputSchema && outputSchema.length ? [...outputSchema, ...CONSTANT_KEYS] : null,\n config: configSchema,\n };\n } catch (e) {\n return { config: [], input: CONSTANT_KEYS, output: CONSTANT_KEYS };\n }\n }\n\n langGraphDefaultMergeState(state: State, messages: LangGraphMessage[], tools: any): State {\n if (messages.length > 0 && \"role\" in messages[0] && messages[0].role === \"system\") {\n // remove system message\n messages = messages.slice(1);\n }\n\n // merge with existing messages\n const existingMessages: LangGraphPlatformMessage[] = state.messages || [];\n const existingMessageIds = new Set(existingMessages.map((message) => message.id));\n const messageIds = new Set(messages.map((message) => message.id));\n\n let removedMessages: RemoveMessage[] = [];\n if (messages.length < existingMessages.length) {\n // Messages were removed\n removedMessages = existingMessages\n .filter((m) => !messageIds.has(m.id))\n .map((m) => new RemoveMessage({ id: m.id! }));\n }\n\n const newMessages = messages.filter((message) => !existingMessageIds.has(message.id));\n\n return {\n ...state,\n messages: [...removedMessages, ...newMessages],\n tools: [...(state.tools ?? []), ...tools],\n };\n }\n}\n\nexport * from \"./types\";\n","import { Message } from \"@langchain/langgraph-sdk\";\nimport { MessageType } from \"@langchain/core/messages\";\n\nexport enum LangGraphEventTypes {\n OnChainStart = \"on_chain_start\",\n OnChainStream = \"on_chain_stream\",\n OnChainEnd = \"on_chain_end\",\n OnChatModelStart = \"on_chat_model_start\",\n OnChatModelStream = \"on_chat_model_stream\",\n OnChatModelEnd = \"on_chat_model_end\",\n OnToolStart = \"on_tool_start\",\n OnToolEnd = \"on_tool_end\",\n OnCustomEvent = \"on_custom_event\",\n OnInterrupt = \"on_interrupt\",\n}\n\nexport type State = Record<string, any>;\n\nexport type SchemaKeys = {\n input: string[] | null;\n output: string[] | null;\n config: string[] | null;\n} | null;\n\nexport type MessageInProgress = {\n id: string;\n toolCallId?: string | null;\n toolCallName?: string | null;\n};\n\nexport type ThinkingInProgress = {\n index: number;\n type?: LangGraphReasoning['type'];\n}\n\nexport interface RunMetadata {\n id: string;\n schemaKeys?: SchemaKeys;\n nodeName?: string;\n prevNodeName?: string | null;\n exitingNode?: boolean;\n manuallyEmittedState?: State | null;\n threadId?: string;\n}\n\nexport type MessagesInProgressRecord = Record<string, MessageInProgress | null>;\n\n// The following types are our own definition to the messages accepted by LangGraph Platform, enhanced with some of our extra data.\nexport interface ToolCall {\n id: string;\n name: string;\n args: Record<string, unknown>;\n}\n\ntype BaseLangGraphPlatformMessage = Omit<\n Message,\n | \"isResultMessage\"\n | \"isTextMessage\"\n | \"isImageMessage\"\n | \"isActionExecutionMessage\"\n | \"isAgentStateMessage\"\n | \"type\"\n | \"createdAt\"\n> & {\n content: string;\n role: string;\n additional_kwargs?: Record<string, unknown>;\n type: MessageType;\n};\n\ninterface LangGraphPlatformResultMessage extends BaseLangGraphPlatformMessage {\n tool_call_id: string;\n name: string;\n}\n\ninterface LangGraphPlatformActionExecutionMessage extends BaseLangGraphPlatformMessage {\n tool_calls: ToolCall[];\n}\n\nexport type LangGraphPlatformMessage =\n | LangGraphPlatformActionExecutionMessage\n | LangGraphPlatformResultMessage\n | BaseLangGraphPlatformMessage;\n\nexport enum CustomEventNames {\n ManuallyEmitMessage = \"manually_emit_message\",\n ManuallyEmitToolCall = \"manually_emit_tool_call\",\n ManuallyEmitState = \"manually_emit_state\",\n Exit = \"exit\",\n}\n\nexport interface PredictStateTool {\n tool: string;\n state_key: string;\n tool_argument: string;\n}\n\nexport interface LangGraphReasoning {\n type: 'text';\n text: string;\n index: number\n}\n","import { Message as LangGraphMessage } from \"@langchain/langgraph-sdk\";\nimport { State, SchemaKeys, LangGraphReasoning } from \"./types\";\nimport { Message, ToolCall } from \"@ag-ui/client\";\n\nexport const DEFAULT_SCHEMA_KEYS = [\"tools\"];\n\nexport function filterObjectBySchemaKeys(obj: Record<string, any>, schemaKeys: string[]) {\n return Object.fromEntries(Object.entries(obj).filter(([key]) => schemaKeys.includes(key)));\n}\n\nexport function getStreamPayloadInput({\n mode,\n state,\n schemaKeys,\n}: {\n mode: \"start\" | \"continue\";\n state: State;\n schemaKeys: SchemaKeys;\n}) {\n let input = mode === \"start\" ? state : null;\n // Do not input keys that are not part of the input schema\n if (input && schemaKeys?.input) {\n input = filterObjectBySchemaKeys(input, [...DEFAULT_SCHEMA_KEYS, ...schemaKeys.input]);\n }\n\n return input;\n}\n\nexport function langchainMessagesToAgui(messages: LangGraphMessage[]): Message[] {\n return messages.map((message) => {\n switch (message.type) {\n case \"human\":\n return {\n id: message.id!,\n role: \"user\",\n content: stringifyIfNeeded(resolveMessageContent(message.content)),\n };\n case \"ai\":\n const content = resolveMessageContent(message.content)\n return {\n id: message.id!,\n role: \"assistant\",\n content: content ? stringifyIfNeeded(content) : '',\n toolCalls: message.tool_calls?.map((tc) => ({\n id: tc.id!,\n type: \"function\",\n function: {\n name: tc.name,\n arguments: JSON.stringify(tc.args),\n },\n })),\n };\n case \"system\":\n return {\n id: message.id!,\n role: \"system\",\n content: stringifyIfNeeded(resolveMessageContent(message.content)),\n };\n case \"tool\":\n return {\n id: message.id!,\n role: \"tool\",\n content: stringifyIfNeeded(resolveMessageContent(message.content)),\n toolCallId: message.tool_call_id,\n };\n default:\n throw new Error(\"message type returned from LangGraph is not supported.\");\n }\n });\n}\n\nexport function aguiMessagesToLangChain(messages: Message[]): LangGraphMessage[] {\n return messages.map((message, index) => {\n switch (message.role) {\n case \"user\":\n return {\n id: message.id,\n role: message.role,\n content: message.content,\n type: \"human\",\n };\n case \"assistant\":\n return {\n id: message.id,\n type: \"ai\",\n role: message.role,\n content: message.content ?? \"\",\n tool_calls: (message.toolCalls ?? []).map((tc: ToolCall) => ({\n id: tc.id,\n name: tc.function.name,\n args: JSON.parse(tc.function.arguments),\n type: \"tool_call\",\n })),\n };\n case \"system\":\n return {\n id: message.id,\n role: message.role,\n content: message.content,\n type: \"system\",\n };\n case \"tool\":\n return {\n content: message.content,\n role: message.role,\n type: message.role,\n tool_call_id: message.toolCallId,\n id: message.id,\n };\n default:\n console.error(`Message role ${message.role} is not implemented`);\n throw new Error(\"message role is not supported.\");\n }\n });\n}\n\nfunction stringifyIfNeeded(item: any) {\n if (typeof item === \"string\") return item;\n return JSON.stringify(item);\n}\n\nexport function resolveReasoningContent(eventData: any): LangGraphReasoning | null {\n const content = eventData.chunk?.content\n\n // Anthropic reasoning response\n if (content && Array.isArray(content) && content.length && content[0]) {\n if (!content[0].thinking) return null\n return {\n text: content[0].thinking,\n type: 'text',\n index: content[0].index,\n }\n }\n\n /// OpenAI reasoning response\n if (eventData.chunk.additional_kwargs?.reasoning?.summary?.[0]) {\n const data = eventData.chunk.additional_kwargs?.reasoning.summary[0]\n if (!data || !data.text) return null\n return {\n type: 'text',\n text: data.text,\n index: data.index,\n }\n }\n\n return null\n}\n\nexport function resolveMessageContent(content?: LangGraphMessage['content']): string | null {\n if (!content) return null;\n\n if (typeof content === 'string') {\n return content;\n }\n\n if (Array.isArray(content) && content.length) {\n const contentText = content.find(c => c.type === 'text')?.text\n return contentText ?? null;\n }\n\n return null\n}\n"],"mappings":"wtBAAA,OAAS,cAAAA,OAA8B,OACvC,OACE,UAAUC,OAUL,2BACP,OAAS,cAAAC,OAAkB,SAC3B,OAAS,iBAAAC,OAAqB,2BCXvB,IAAKC,QACVA,EAAA,aAAe,iBACfA,EAAA,cAAgB,kBAChBA,EAAA,WAAa,eACbA,EAAA,iBAAmB,sBACnBA,EAAA,kBAAoB,uBACpBA,EAAA,eAAiB,oBACjBA,EAAA,YAAc,gBACdA,EAAA,UAAY,cACZA,EAAA,cAAgB,kBAChBA,EAAA,YAAc,eAVJA,QAAA,IAiFAC,QACVA,EAAA,oBAAsB,wBACtBA,EAAA,qBAAuB,0BACvBA,EAAA,kBAAoB,sBACpBA,EAAA,KAAO,OAJGA,QAAA,IDxDZ,OACE,iBAAAC,GAGA,aAAAC,MAsBK,gBElDA,IAAMC,EAAsB,CAAC,OAAO,EAEpC,SAASC,EAAyBC,EAA0BC,EAAsB,CACvF,OAAO,OAAO,YAAY,OAAO,QAAQD,CAAG,EAAE,OAAO,CAAC,CAACE,CAAG,IAAMD,EAAW,SAASC,CAAG,CAAC,CAAC,CAC3F,CAEO,SAASC,GAAsB,CACpC,KAAAC,EACA,MAAAC,EACA,WAAAJ,CACF,EAIG,CACD,IAAIK,EAAQF,IAAS,QAAUC,EAAQ,KAEvC,OAAIC,IAASL,GAAA,MAAAA,EAAY,SACvBK,EAAQP,EAAyBO,EAAO,CAAC,GAAGR,EAAqB,GAAGG,EAAW,KAAK,CAAC,GAGhFK,CACT,CAEO,SAASC,GAAwBC,EAAyC,CAC/E,OAAOA,EAAS,IAAKC,GAAY,CA7BnC,IAAAC,EA8BI,OAAQD,EAAQ,KAAM,CACpB,IAAK,QACH,MAAO,CACL,GAAIA,EAAQ,GACZ,KAAM,OACN,QAASE,EAAkBC,EAAsBH,EAAQ,OAAO,CAAC,CACnE,EACF,IAAK,KACH,IAAMI,EAAUD,EAAsBH,EAAQ,OAAO,EACrD,MAAO,CACL,GAAIA,EAAQ,GACZ,KAAM,YACN,QAASI,EAAUF,EAAkBE,CAAO,EAAI,GAChD,WAAWH,EAAAD,EAAQ,aAAR,YAAAC,EAAoB,IAAKI,IAAQ,CAC1C,GAAIA,EAAG,GACP,KAAM,WACN,SAAU,CACR,KAAMA,EAAG,KACT,UAAW,KAAK,UAAUA,EAAG,IAAI,CACnC,CACF,GACF,EACF,IAAK,SACH,MAAO,CACL,GAAIL,EAAQ,GACZ,KAAM,SACN,QAASE,EAAkBC,EAAsBH,EAAQ,OAAO,CAAC,CACnE,EACF,IAAK,OACH,MAAO,CACL,GAAIA,EAAQ,GACZ,KAAM,OACN,QAASE,EAAkBC,EAAsBH,EAAQ,OAAO,CAAC,EACjE,WAAYA,EAAQ,YACtB,EACF,QACE,MAAM,IAAI,MAAM,wDAAwD,CAC5E,CACF,CAAC,CACH,CAEO,SAASM,GAAwBP,EAAyC,CAC/E,OAAOA,EAAS,IAAI,CAACC,EAASO,IAAU,CAxE1C,IAAAN,EAAAO,EAyEI,OAAQR,EAAQ,KAAM,CACpB,IAAK,OACH,MAAO,CACL,GAAIA,EAAQ,GACZ,KAAMA,EAAQ,KACd,QAASA,EAAQ,QACjB,KAAM,OACR,EACF,IAAK,YACH,MAAO,CACL,GAAIA,EAAQ,GACZ,KAAM,KACN,KAAMA,EAAQ,KACd,SAASC,EAAAD,EAAQ,UAAR,KAAAC,EAAmB,GAC5B,aAAaO,EAAAR,EAAQ,YAAR,KAAAQ,EAAqB,CAAC,GAAG,IAAKH,IAAkB,CAC3D,GAAIA,EAAG,GACP,KAAMA,EAAG,SAAS,KAClB,KAAM,KAAK,MAAMA,EAAG,SAAS,SAAS,EACtC,KAAM,WACR,EAAE,CACJ,EACF,IAAK,SACH,MAAO,CACL,GAAIL,EAAQ,GACZ,KAAMA,EAAQ,KACd,QAASA,EAAQ,QACjB,KAAM,QACR,EACF,IAAK,OACH,MAAO,CACL,QAASA,EAAQ,QACjB,KAAMA,EAAQ,KACd,KAAMA,EAAQ,KACd,aAAcA,EAAQ,WACtB,GAAIA,EAAQ,EACd,EACF,QACE,cAAQ,MAAM,gBAAgBA,EAAQ,IAAI,qBAAqB,EACzD,IAAI,MAAM,gCAAgC,CACpD,CACF,CAAC,CACH,CAEA,SAASE,EAAkBO,EAAW,CACpC,OAAI,OAAOA,GAAS,SAAiBA,EAC9B,KAAK,UAAUA,CAAI,CAC5B,CAEO,SAASC,GAAwBC,EAA2C,CAzHnF,IAAAV,EAAAO,EAAAI,EAAAC,EAAAC,EA0HE,IAAMV,GAAUH,EAAAU,EAAU,QAAV,YAAAV,EAAiB,QAGjC,GAAIG,GAAW,MAAM,QAAQA,CAAO,GAAKA,EAAQ,QAAUA,EAAQ,CAAC,EAClE,OAAKA,EAAQ,CAAC,EAAE,SACT,CACL,KAAMA,EAAQ,CAAC,EAAE,SACjB,KAAM,OACN,MAAOA,EAAQ,CAAC,EAAE,KACpB,EALiC,KASnC,IAAIS,GAAAD,GAAAJ,EAAAG,EAAU,MAAM,oBAAhB,YAAAH,EAAmC,YAAnC,YAAAI,EAA8C,UAA9C,MAAAC,EAAwD,GAAI,CAC9D,IAAME,GAAOD,EAAAH,EAAU,MAAM,oBAAhB,YAAAG,EAAmC,UAAU,QAAQ,GAClE,MAAI,CAACC,GAAQ,CAACA,EAAK,KAAa,KACzB,CACL,KAAM,OACN,KAAMA,EAAK,KACX,MAAOA,EAAK,KACd,CACF,CAEA,OAAO,IACT,CAEO,SAASZ,EAAsBC,EAAsD,CApJ5F,IAAAH,EAqJE,GAAI,CAACG,EAAS,OAAO,KAErB,GAAI,OAAOA,GAAY,SACrB,OAAOA,EAGT,GAAI,MAAM,QAAQA,CAAO,GAAKA,EAAQ,OAAQ,CAC5C,IAAMY,GAAcf,EAAAG,EAAQ,KAAKa,GAAKA,EAAE,OAAS,MAAM,IAAnC,YAAAhB,EAAsC,KAC1D,OAAOe,GAAA,KAAAA,EAAe,IACxB,CAEA,OAAO,IACT,CFrDO,IAAME,GAAN,cAA6BC,EAAc,CAYhD,YAAYC,EAA8B,CAxH5C,IAAAC,EAAAC,EAyHI,MAAMF,CAAM,EACZ,KAAK,kBAAoB,CAAC,EAC1B,KAAK,UAAYA,EAAO,UACxB,KAAK,QAAUA,EAAO,QACtB,KAAK,gBAAkBA,EAAO,gBAC9B,KAAK,gBAAkB,KACvB,KAAK,QACHE,EAAAF,GAAA,YAAAA,EAAQ,SAAR,KAAAE,EACA,IAAIC,GAAgB,CAClB,OAAQH,EAAO,cACf,OAAQA,EAAO,gBACf,eAAgBI,EAAA,IAAMH,EAAAD,EAAO,kBAAP,KAAAC,EAA0B,CAAC,EACnD,CAAC,CACL,CAEA,cAAcI,EAAwB,CACpC,YAAK,WAAW,KAAKA,CAAK,EACnB,EACT,CAEA,IAAIC,EAAsB,CACxB,YAAK,UAAY,CACf,GAAIA,EAAM,MACV,SAAUA,EAAM,QAClB,EACO,IAAIC,GAA6BC,IACtC,KAAK,mBAAmBF,EAAOE,CAAU,EAClC,IAAM,CAAC,EACf,CACH,CAEA,MAAM,mBAAmBF,EAA8BE,EAAyC,CAxJlG,IAAAP,EAAAC,EAAAO,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAAAC,EAyJI,GAAI,CAAE,SAAUC,EAAe,MAAAC,EAAO,SAAAC,EAAU,MAAAC,EAAO,QAAAC,EAAS,eAAAC,CAAe,EAAInB,EACnF,KAAK,WAAaE,EAClB,IAAIkB,EAAa,GAGjB,KAAK,UAAW,qBAAuB,KAEvC,KAAK,UAAW,UAAWzB,EAAAK,EAAM,iBAAN,YAAAL,EAAsB,SAEjD,IAAM0B,EAAWP,GAAA,KAAAA,EAAiBQ,GAAW,EAExC,KAAK,YACR,KAAK,UAAY,MAAM,KAAK,aAAa,GAG3C,IAAMC,EAAS,MAAM,KAAK,kBAAkBF,CAAQ,EACpD,KAAK,UAAW,SAAWE,EAAO,UAClC,IAAMC,GAAa5B,EAAA,MAAM,KAAK,OAAO,QAAQ,SAAS2B,EAAO,SAAS,IAAnD,KAAA3B,EAAwD,CAAE,OAAQ,CAAC,CAAE,EAElF6B,EAAmBD,EAAW,OAC9BE,EAAyBC,GAAwBX,CAAQ,EAE/DD,EAAM,SAAWU,EAAiB,SAClCV,EAAQ,KAAK,2BAA2BA,EAAOW,EAAwBT,CAAK,EAE5E,IAAMW,EACJP,GAAY,KAAK,UAAW,UAAY,WAAa,KAAK,UAAW,SACjE,WACA,QAEFO,IAAS,YAAc,GAACzB,EAAAgB,GAAA,YAAAA,EAAgB,UAAhB,MAAAhB,EAAyB,SACnD,MAAM,KAAK,OAAO,QAAQ,YAAYkB,EAAU,CAC9C,OAAQN,EACR,OAAQ,KAAK,UAAW,QAC1B,CAAC,EAGH,KAAK,UAAW,WAAa,MAAM,KAAK,cAAc,EAEtD,IAAMc,EAAeC,GAAsB,CACzC,KAAAF,EACA,MAAAb,EACA,WAAY,KAAK,UAAW,UAC9B,CAAC,EAEKgB,EAAY,MAAM,KAAK,OAAO,WAAW,SAAS,KAAK,UAAU,YAAY,EAE/EC,EACEC,EAAiB,CAAC,KAAK,gBAAiBd,GAAA,YAAAA,EAAgB,MAAM,EAAE,OACpE,OACF,EACIc,EAAe,SACjBD,EAAgB,MAAM,KAAK,aAAa,CACtC,QAASC,EACT,UAAW,KAAK,UAChB,WAAY,KAAK,UAAW,UAC9B,CAAC,GAEH,IAAMC,EAAUC,EAAArC,EAAA,GACXqB,GADW,CAEd,YACEf,EAAAe,GAAA,YAAAA,EAAgB,aAAhB,KAAAf,EAA+B,CAAC,SAAU,SAAU,SAAS,EAC/D,MAAOyB,EACP,OAAQG,CACV,GAGMI,GAAc7B,GAAAD,GAAAD,EAAAmB,EAAW,QAAX,YAAAnB,EAAmB,KAAnB,YAAAC,EAAuB,aAAvB,KAAAC,EAAqC,CAAC,EAC1D,GAAI6B,GAAA,MAAAA,EAAY,QAAU,GAAC5B,EAAAW,GAAA,YAAAA,EAAgB,UAAhB,MAAAX,EAAyB,QAClD,YAAK,cAAc,CACjB,KAAM6B,EAAU,YAChB,SAAAhB,EACA,MAAOrB,EAAM,KACf,CAAC,EAEDoC,EAAW,QAASE,GAAc,CAChC,KAAK,cAAc,CACjB,KAAMD,EAAU,OAChB,oBACA,MACE,OAAOC,EAAU,OAAU,SAAWA,EAAU,MAAQ,KAAK,UAAUA,EAAU,KAAK,EACxF,SAAUA,CACZ,CAAC,CACH,CAAC,EAED,KAAK,cAAc,CACjB,KAAMD,EAAU,aAChB,SAAAhB,EACA,MAAOrB,EAAM,KACf,CAAC,EACME,EAAW,SAAS,EAE7B,IAAMqC,EAAiB,KAAK,OAAO,KAAK,OAAOlB,EAAU,KAAK,UAAU,aAAca,CAAO,EAE7F,KAAK,UAAW,aAAe,KAC/B,IAAIM,EAAoB,CAAC,EACrBC,EAAe1B,EAEnB,GAAI,CACF,KAAK,cAAc,CACjB,KAAMsB,EAAU,YAChB,SAAAhB,EACA,MAAO,KAAK,UAAW,EACzB,CAAC,EAED,YAAAqB,GAAAC,GAAsCJ,GAAtCK,GAAAC,GAAAC,GAAAF,GAAA,EAAAC,GAAA,MAAAH,GAAA,aAAAE,GAAA,GAAsD,CAA3C,IAAIG,EAAfF,GAAA,MAEE,GAAI,CAACX,EAAQ,WAAW,SAASa,EAAoB,KAAmB,EACtE,SAUF,IAAMC,EAAQD,EAEd,GAAIA,EAAoB,QAAU,QAAS,CACzC,KAAK,cAAc,CACjB,KAAMV,EAAU,UAChB,QAASU,EAAoB,KAAK,QAClC,SAAUA,CACZ,CAAC,EACD,KACF,CAEA,GAAIA,EAAoB,QAAU,UAAW,SAE7C,GAAIA,EAAoB,QAAU,SAAU,CAC1CP,EAAoBQ,EAAM,KAC1B,QACF,CAEA,IAAMC,EAAYD,EAAM,KAClBE,EAAkBD,EAAU,SAAS,eACrCE,EAAYF,EAAU,MACtBG,GAAWH,EAAU,SAwC3B,GAvCA,KAAK,UAAW,GAAKG,GAAS,OAE1BF,GAAmBA,IAAoB,KAAK,UAAW,WACrD,KAAK,UAAW,UAClB,KAAK,cAAc,CACjB,KAAMb,EAAU,cAChB,SAAU,KAAK,UAAW,QAC5B,CAAC,EAGCa,IACF,KAAK,cAAc,CACjB,KAAMb,EAAU,aAChB,SAAUa,CACZ,CAAC,EACD,KAAK,UAAW,SAAWA,IAI/B9B,EACEA,GACC+B,IAAc,mBACbF,EAAU,OAAS,OAEvB,KAAK,UAAW,YACd,KAAK,UAAW,WAAaC,GAC7BC,IAAc,eACZ,KAAK,UAAW,cAClB,KAAK,UAAW,qBAAuB,MAKrCpB,EAAU,MAAS,KAAMsB,IAASA,GAAK,KAAOH,CAAe,IAC/D,KAAK,UAAW,SAAWA,GAG7BT,GAAehC,EAAA,KAAK,UAAW,uBAAhB,KAAAA,EAAwC+B,EAEnD,CAAC,KAAK,UAAW,SACnB,UAGmB,KAAK,UAAUC,CAAY,IAAM,KAAK,UAAU1B,CAAK,GAItE,KAAK,UAAW,cAAgB,KAAK,UAAW,UAChD,KAAK,UAAW,cAClB,CAAS,KAAK,qBAAqB,KAAK,UAAW,EAAE,IAErDA,EAAQ0B,EACR,KAAK,UAAW,aAAe,KAAK,UAAW,SAE/C,KAAK,cAAc,CACjB,KAAMJ,EAAU,eAChB,SAAU,KAAK,iBAAiBtB,CAAK,EACrC,SAAUiC,CACZ,CAAC,GAGH,KAAK,cAAc,CACjB,KAAMX,EAAU,IAChB,MAAOY,CACT,CAAC,EAED,KAAK,kBAAkBA,EAAWlC,CAAK,CACzC,QAtGA8B,GAlQN,CAkQMC,GAAA,CAAAD,WAAA,KAAAD,KAAAC,GAAAH,GAAA,eAAAG,GAAA,KAAAH,WAAA,IAAAI,GAAA,MAAAA,GAAA,IAwGA/B,EAAQ,MAAM,KAAK,OAAO,QAAQ,SAASM,CAAQ,EACnD,IAAMe,GAAcxB,GAAAD,GAAAD,EAAAK,EAAM,QAAN,YAAAL,EAAc,KAAd,YAAAC,EAAkB,aAAlB,KAAAC,EAAgC,CAAC,EACrD,YAAK,UAAW,SAAWwB,EACvB,KAAK,UAAW,SAChB,OAAO,KAAKrB,EAAM,SAAS,MAAM,EAAE,CAAC,EAExCqB,EAAW,QAASE,GAAc,CAChC,KAAK,cAAc,CACjB,KAAMD,EAAU,OAChB,oBACA,MACE,OAAOC,EAAU,OAAU,SAAWA,EAAU,MAAQ,KAAK,UAAUA,EAAU,KAAK,EACxF,SAAUA,CACZ,CAAC,CACH,CAAC,EAED,KAAK,cAAc,CACjB,KAAMD,EAAU,eAChB,SAAU,KAAK,iBAAiBtB,EAAM,MAAM,CAC9C,CAAC,EACD,KAAK,cAAc,CACjB,KAAMsB,EAAU,kBAChB,SAAUiB,IAAwBzC,EAAAE,EAAM,OAAO,WAAb,KAAAF,EAAyB,CAAC,CAAC,CAC/D,CAAC,EACG,KAAK,UAAW,UAClB,KAAK,cAAc,CACjB,KAAMwB,EAAU,cAChB,SAAU,KAAK,UAAW,QAC5B,CAAC,EAEH,KAAK,cAAc,CACjB,KAAMA,EAAU,aAChB,SAAAhB,EACA,MAAO,KAAK,UAAW,EACzB,CAAC,EACMnB,EAAW,SAAS,CAC7B,OAASqD,EAAG,CACV,OAAOrD,EAAW,MAAMqD,CAAC,CAC3B,CACF,CAEA,kBAAkBxD,EAAYgB,EAAoB,CAnZpD,IAAApB,EAAAC,EAAAO,EAAAC,EAAAC,EAAAC,EAoZI,OAAQP,EAAM,MAAO,CACnB,2BACE,IAAIyD,GAAqB7D,EAAAI,EAAM,SAAS,eAAe,IAA9B,KAAAJ,EAAmC,GACxD8D,GAAsB7D,EAAAG,EAAM,SAAS,iBAAiB,IAAhC,KAAAH,EAAqC,GAE/D,GAAIG,EAAM,KAAK,MAAM,kBAAkB,cAAe,OACtD,IAAI2D,EAAgB,KAAK,qBAAqB,KAAK,UAAW,EAAE,EAC1DC,EAAmB,GAAQD,GAAA,MAAAA,EAAe,IAC1CE,GAAezD,EAAAJ,EAAM,KAAK,MAAM,mBAAjB,YAAAI,EAAoC,GACnD0D,GAA6BzD,EAAAL,EAAM,SAAS,gBAAf,YAAAK,EAAiC,KACjE0D,GAAuCA,EAAiB,QAASF,GAAA,YAAAA,EAAc,OAG5EG,EAAuB,CAACJ,IAAoBC,GAAA,YAAAA,EAAc,MAC1DI,EACJL,IAAoBD,GAAA,YAAAA,EAAe,cAAcE,GAAA,YAAAA,EAAc,MAC3DK,EAAqBN,IAAoBD,GAAA,YAAAA,EAAe,aAAc,CAACE,EAEvEM,EAAgBC,GAAwBpE,EAAM,IAAI,EAClDqE,EAAiBC,EAAsBtE,EAAM,KAAK,MAAM,OAAO,EAC/DuE,EAAwB,GAAQ,CAACV,GAAgBQ,GAEjDG,EACJZ,GAAoB,EAACD,GAAA,MAAAA,EAAe,aAAc,CAACY,EAErD,GAAIJ,EAAe,CACjB,KAAK,oBAAoBA,CAAa,EACtC,KACF,CAoBA,GAlBI,CAACA,GAAiB,KAAK,kBACzB,KAAK,cAAc,CACjB,KAAM7B,EAAU,yBAClB,CAAC,EACD,KAAK,cAAc,CACjB,KAAMA,EAAU,YAClB,CAAC,EACD,KAAK,gBAAkB,MAGrBwB,GACF,KAAK,cAAc,CACjB,KAAMxB,EAAU,OAChB,KAAM,eACN,MAAOtC,EAAM,SAAS,aACxB,CAAC,EAGCkE,EAAoB,CACL,KAAK,cAAc,CAClC,KAAM5B,EAAU,cAChB,WAAYqB,GAAA,YAAAA,EAAe,WAC3B,SAAU3D,CACZ,CAAC,IAEC,KAAK,kBAAkB,KAAK,UAAW,EAAE,EAAI,MAE/C,KACF,CAEA,GAAIwE,EAAmB,CACJ,KAAK,cAAc,CAClC,KAAMlC,EAAU,iBAChB,UAAWqB,EAAe,GAC1B,SAAU3D,CACZ,CAAC,IAEC,KAAK,kBAAkB,KAAK,UAAW,EAAE,EAAI,MAE/C,KACF,CAEA,GAAIgE,GAAwBN,EAAqB,CAC9B,KAAK,cAAc,CAClC,KAAMpB,EAAU,gBAChB,WAAYuB,EAAa,GACzB,aAAcA,EAAa,KAC3B,gBAAiB7D,EAAM,KAAK,MAAM,GAClC,SAAUA,CACZ,CAAC,GAEC,KAAK,qBAAqB,KAAK,UAAW,GAAI,CAC5C,GAAIA,EAAM,KAAK,MAAM,GACrB,WAAY6D,EAAa,GACzB,aAAcA,EAAa,IAC7B,CAAC,EAEH,KACF,CAGA,GAAII,GAAuBP,EAAqB,CAC9C,KAAK,cAAc,CACjB,KAAMpB,EAAU,eAChB,WAAYqB,GAAA,YAAAA,EAAe,WAC3B,MAAOE,EAAa,KACpB,SAAU7D,CACZ,CAAC,EACD,KACF,CAGA,GAAIuE,GAAyBd,EAAoB,CAE1CE,IACH,KAAK,cAAc,CACjB,KAAMrB,EAAU,mBAChB,KAAM,YACN,UAAWtC,EAAM,KAAK,MAAM,GAC5B,SAAUA,CACZ,CAAC,EACD,KAAK,qBAAqB,KAAK,UAAW,GAAI,CAC5C,GAAIA,EAAM,KAAK,MAAM,GACrB,WAAY,KACZ,aAAc,IAChB,CAAC,EACD2D,EAAgB,KAAK,qBAAqB,KAAK,UAAW,EAAE,GAG9D,KAAK,cAAc,CACjB,KAAMrB,EAAU,qBAChB,UAAWqB,EAAe,GAC1B,MAAOU,EACP,SAAUrE,CACZ,CAAC,EACD,KACF,CAEA,MACF,wBACE,IAAIM,EAAA,KAAK,qBAAqB,KAAK,UAAW,EAAE,IAA5C,MAAAA,EAA+C,WAAY,CAC5C,KAAK,cAAc,CAClC,KAAMgC,EAAU,cAChB,WAAY,KAAK,qBAAqB,KAAK,UAAW,EAAE,EAAG,WAC3D,SAAUtC,CACZ,CAAC,IAEC,KAAK,kBAAkB,KAAK,UAAW,EAAE,EAAI,MAE/C,KACF,CACA,IAAIO,EAAA,KAAK,qBAAqB,KAAK,UAAW,EAAE,IAA5C,MAAAA,EAA+C,GAAI,CACpC,KAAK,cAAc,CAClC,KAAM+B,EAAU,iBAChB,UAAW,KAAK,qBAAqB,KAAK,UAAW,EAAE,EAAG,GAC1D,SAAUtC,CACZ,CAAC,IAEC,KAAK,kBAAkB,KAAK,UAAW,EAAE,EAAI,MAE/C,KACF,CACA,MACF,sBACE,GAAIA,EAAM,OAAS,wBAAsC,CACvD,KAAK,cAAc,CACjB,KAAMsC,EAAU,mBAChB,KAAM,YACN,UAAWtC,EAAM,KAAK,WACtB,SAAUA,CACZ,CAAC,EACD,KAAK,cAAc,CACjB,KAAMsC,EAAU,qBAChB,UAAWtC,EAAM,KAAK,WACtB,MAAOA,EAAM,KAAK,QAClB,SAAUA,CACZ,CAAC,EACD,KAAK,cAAc,CACjB,KAAMsC,EAAU,iBAChB,UAAWtC,EAAM,KAAK,WACtB,SAAUA,CACZ,CAAC,EACD,KACF,CAEA,GAAIA,EAAM,OAAS,0BAAuC,CACxD,KAAK,cAAc,CACjB,KAAMsC,EAAU,gBAChB,WAAYtC,EAAM,KAAK,GACvB,aAAcA,EAAM,KAAK,KACzB,gBAAiBA,EAAM,KAAK,GAC5B,SAAUA,CACZ,CAAC,EACD,KAAK,cAAc,CACjB,KAAMsC,EAAU,eAChB,WAAYtC,EAAM,KAAK,GACvB,MAAOA,EAAM,KAAK,KAClB,SAAUA,CACZ,CAAC,EACD,KAAK,cAAc,CACjB,KAAMsC,EAAU,cAChB,WAAYtC,EAAM,KAAK,GACvB,SAAUA,CACZ,CAAC,EACD,KACF,CAEIA,EAAM,OAAS,wBACjB,KAAK,UAAW,qBAAuBA,EAAM,KAC7C,KAAK,cAAc,CACjB,KAAMsC,EAAU,eAChB,SAAU,KAAK,iBAAiBtB,CAAK,EACrC,SAAUhB,CACZ,CAAC,GAGH,KAAK,cAAc,CACjB,KAAMsC,EAAU,OAChB,KAAMtC,EAAM,KACZ,MAAOA,EAAM,KACb,SAAUA,CACZ,CAAC,EACD,KACJ,CACF,CAEA,oBAAoBmE,EAAmC,CA5mBzD,IAAAvE,EA6mBI,GAAI,CAACuE,GAAiB,CAACA,EAAc,MAAQ,CAACA,EAAc,KAC1D,OAGF,IAAMM,EAAoBN,EAAc,OAEpCvE,EAAA,KAAK,kBAAL,MAAAA,EAAsB,OAAS,KAAK,gBAAgB,QAAU6E,IAC5D,KAAK,gBAAgB,MACvB,KAAK,cAAc,CACjB,KAAMnC,EAAU,yBAClB,CAAC,EAEH,KAAK,cAAc,CACjB,KAAMA,EAAU,YAClB,CAAC,EACD,KAAK,gBAAkB,MAGpB,KAAK,kBAER,KAAK,cAAc,CACjB,KAAMA,EAAU,cAClB,CAAC,EACD,KAAK,gBAAkB,CACrB,MAAOmC,CACT,GAIE,KAAK,gBAAgB,OAASN,EAAc,OAC9C,KAAK,cAAc,CACjB,KAAM7B,EAAU,2BAClB,CAAC,EACD,KAAK,gBAAgB,KAAO6B,EAAc,MAGxC,KAAK,gBAAgB,MACvB,KAAK,cAAc,CACjB,KAAM7B,EAAU,8BAChB,MAAO6B,EAAc,IACvB,CAAC,CAEL,CAEA,iBAAiBnD,EAAc,CAC7B,IAAM0D,EAAa,KAAK,UAAW,WAEnC,OAAIA,GAAA,MAAAA,EAAY,SACd1D,EAAQ2D,EAAyB3D,EAAO,CAAC,GAAG4D,EAAqB,GAAGF,EAAW,MAAM,CAAC,GAGjF1D,CACT,CAEA,MAAM,kBAAkBM,EAAmC,CACzD,IAAIE,EACJ,GAAI,CACF,GAAI,CACFA,EAAS,MAAM,KAAK,UAAUF,CAAQ,CACxC,OAASyB,EAAO,CACdvB,EAAS,MAAM,KAAK,aAAa,CAAE,SAAAF,CAAS,CAAC,CAC/C,CACF,OAASyB,EAAgB,CACvB,MAAM,IAAI,MAAM,4BAA6BA,EAAgB,OAAO,EAAE,CACxE,CAEA,OAAOvB,CACT,CAEA,MAAM,UAAUF,EAAkB,CAChC,OAAO,KAAK,OAAO,QAAQ,IAAIA,CAAQ,CACzC,CAEA,MAAM,aAAaa,EAA4D,CAC7E,OAAO,KAAK,OAAO,QAAQ,OAAOA,CAAO,CAC3C,CAEA,MAAM,aAAa,CACjB,QAAA0C,EACA,UAAAC,EACA,WAAAJ,CACF,EAIG,CACD,OAAOG,EAAQ,OAAO,CAACE,EAAKC,IAAQ,CAnsBxC,IAAApF,EAosBM,IAAIqF,EAAuBF,EAAI,aAE3BC,EAAI,eACNC,EAAuBP,GAAA,MAAAA,EAAY,OAC/BC,EAAyBK,GAAA,YAAAA,EAAK,aAAc,CAC1C,GAAGJ,EACH,IAAIhF,EAAA8E,GAAA,YAAAA,EAAY,SAAZ,KAAA9E,EAAsB,CAAC,CAC7B,CAAC,EACDoF,GAAA,YAAAA,EAAK,cAGX,IAAME,EAAY9C,EAAArC,IAAA,GACbgF,GACAC,GAFa,CAGhB,aAAcC,CAChB,GAGME,EACJJ,EAAI,iBAAmB,MAAQC,EAAI,kBAAoB,GAEnDI,EAAsB,KAAK,UAAUF,CAAS,IAAM,KAAK,UAAUH,CAAG,EAGtEM,EACJF,GACA,KAAK,UAAU/C,EAAArC,EAAA,GAAKmF,GAAL,CAAgB,gBAAiB,IAAK,EAAC,IACpD,KAAK,UAAU9C,EAAArC,EAAA,GAAKgF,GAAL,CAAU,gBAAiB,IAAK,EAAC,EAEpD,OAAIK,GAAuB,CAACC,EACnBtF,IAAA,GACFgF,GACAG,GAIAH,CACT,EAAGD,EAAU,MAAM,CACrB,CAEA,qBAAqBQ,EAAe,CAClC,OAAO,KAAK,kBAAkBA,CAAK,CACrC,CAEA,qBAAqBA,EAAeC,EAAyB,CAC3D,KAAK,kBAAoBnD,EAAArC,EAAA,GACpB,KAAK,mBADe,CAEvB,CAACuF,CAAK,EAAGvF,IAAA,GACH,KAAK,kBAAkBuF,CAAK,GAC7BC,EAEP,EACF,CAEA,MAAM,cAAmC,CACvC,IAAMC,EAAa,MAAM,KAAK,OAAO,WAAW,OAAO,EACjDC,EAAqBD,EAAW,KACnCE,GACCA,EAAa,WAAa,KAAK,OACnC,EACA,GAAI,CAACD,EACH,cAAQ,MAAM;AAAA,qCACiB,KAAK,OAAO;AAAA;AAAA;AAAA,yCAERD,EAAW,IAAKG,GAAM,GAAGA,EAAE,QAAQ,SAASA,EAAE,YAAY,GAAG,EAAE,KAAK,IAAI,CAAC;AAAA,OAC3G,EACK,IAAI,MAAM,mBAAmB,EAGrC,OAAOF,CACT,CAEA,MAAM,eAAqC,CA5wB7C,IAAA7F,EAAAC,EAAAO,EA6wBI,IAAMwF,EAAgB,CAAC,UAAU,EAEjC,GAAI,CACF,IAAMC,EAAc,MAAM,KAAK,OAAO,WAAW,WAAW,KAAK,UAAW,YAAY,EACpFC,EAAe,KAInB,IAHIlG,EAAAiG,EAAY,gBAAZ,MAAAjG,EAA2B,aAC7BkG,EAAe,OAAO,KAAKD,EAAY,cAAc,UAAU,GAE7D,GAAChG,EAAAgG,EAAY,eAAZ,MAAAhG,EAA0B,aAAc,GAACO,EAAAyF,EAAY,gBAAZ,MAAAzF,EAA2B,YACvE,MAAO,CAAE,OAAQ,CAAC,EAAG,MAAOwF,EAAe,OAAQA,CAAc,EAEnE,IAAMG,EAAc,OAAO,KAAKF,EAAY,aAAa,UAAU,EAC7DG,EAAe,OAAO,KAAKH,EAAY,cAAc,UAAU,EAErE,MAAO,CACL,MAAOE,GAAeA,EAAY,OAAS,CAAC,GAAGA,EAAa,GAAGH,CAAa,EAAI,KAChF,OAAQI,GAAgBA,EAAa,OAAS,CAAC,GAAGA,EAAc,GAAGJ,CAAa,EAAI,KACpF,OAAQE,CACV,CACF,OAAStC,EAAG,CACV,MAAO,CAAE,OAAQ,CAAC,EAAG,MAAOoC,EAAe,OAAQA,CAAc,CACnE,CACF,CAEA,2BAA2B5E,EAAcC,EAA8BC,EAAmB,CAryB5F,IAAAtB,EAsyBQqB,EAAS,OAAS,GAAK,SAAUA,EAAS,CAAC,GAAKA,EAAS,CAAC,EAAE,OAAS,WAEvEA,EAAWA,EAAS,MAAM,CAAC,GAI7B,IAAMgF,EAA+CjF,EAAM,UAAY,CAAC,EAClEkF,EAAqB,IAAI,IAAID,EAAiB,IAAKE,GAAYA,EAAQ,EAAE,CAAC,EAC1EC,EAAa,IAAI,IAAInF,EAAS,IAAKkF,GAAYA,EAAQ,EAAE,CAAC,EAE5DE,EAAmC,CAAC,EACpCpF,EAAS,OAASgF,EAAiB,SAErCI,EAAkBJ,EACf,OAAQK,GAAM,CAACF,EAAW,IAAIE,EAAE,EAAE,CAAC,EACnC,IAAKA,GAAM,IAAIC,GAAc,CAAE,GAAID,EAAE,EAAI,CAAC,CAAC,GAGhD,IAAME,EAAcvF,EAAS,OAAQkF,GAAY,CAACD,EAAmB,IAAIC,EAAQ,EAAE,CAAC,EAEpF,OAAO/D,EAAArC,EAAA,GACFiB,GADE,CAEL,SAAU,CAAC,GAAGqF,EAAiB,GAAGG,CAAW,EAC7C,MAAO,CAAC,IAAI5G,EAAAoB,EAAM,QAAN,KAAApB,EAAe,CAAC,EAAI,GAAGsB,CAAK,CAC1C,EACF,CACF","names":["Observable","LangGraphClient","randomUUID","RemoveMessage","LangGraphEventTypes","CustomEventNames","AbstractAgent","EventType","DEFAULT_SCHEMA_KEYS","filterObjectBySchemaKeys","obj","schemaKeys","key","getStreamPayloadInput","mode","state","input","langchainMessagesToAgui","messages","message","_a","stringifyIfNeeded","resolveMessageContent","content","tc","aguiMessagesToLangChain","index","_b","item","resolveReasoningContent","eventData","_c","_d","_e","data","contentText","c","LangGraphAgent","AbstractAgent","config","_a","_b","LangGraphClient","__spreadValues","event","input","Observable","subscriber","_c","_d","_e","_f","_g","_h","_i","_j","_k","_l","_m","inputThreadId","state","messages","tools","context","forwardedProps","shouldExit","threadId","randomUUID","thread","agentState","agentStateValues","aguiToLangChainMessage","aguiMessagesToLangChain","mode","payloadInput","getStreamPayloadInput","graphInfo","payloadConfig","configsToMerge","payload","__spreadProps","interrupts","EventType","interrupt","streamResponse","latestStateValues","updatedState","iter","__forAwait","more","temp","error","streamResponseChunk","chunk","chunkData","currentNodeName","eventType","metadata","node","langchainMessagesToAgui","e","shouldEmitMessages","shouldEmitToolCalls","currentStream","hasCurrentStream","toolCallData","toolCallUsedToPredictState","predictStateTool","isToolCallStartEvent","isToolCallArgsEvent","isToolCallEndEvent","reasoningData","resolveReasoningContent","messageContent","resolveMessageContent","isMessageContentEvent","isMessageEndEvent","thinkingStepIndex","schemaKeys","filterObjectBySchemaKeys","DEFAULT_SCHEMA_KEYS","configs","assistant","acc","cfg","filteredConfigurable","newConfig","isRecursionLimitSetToDefault","configsAreDifferent","isOnlyRecursionLimitDifferent","runId","data","assistants","retrievedAssistant","searchResult","a","CONSTANT_KEYS","graphSchema","configSchema","inputSchema","outputSchema","existingMessages","existingMessageIds","message","messageIds","removedMessages","m","RemoveMessage","newMessages"]}
|