@ag-ui/langgraph 0.0.24 → 0.0.25

package/dist/index.d.ts

@@ -1,187 +1,203 @@
-import { TextMessageStartEvent, TextMessageContentEvent, TextMessageEndEvent, ThinkingTextMessageStartEvent, ThinkingTextMessageContentEvent, ThinkingTextMessageEndEvent, ToolCallStartEvent, ToolCallArgsEvent, ToolCallEndEvent, ToolCallResultEvent, ThinkingStartEvent, ThinkingEndEvent, StateSnapshotEvent, StateDeltaEvent, MessagesSnapshotEvent, RawEvent, CustomEvent, RunStartedEvent, RunFinishedEvent, RunErrorEvent, StepStartedEvent, StepFinishedEvent, AgentConfig, AbstractAgent, RunAgentInput as RunAgentInput$1, HttpAgent } from '@ag-ui/client';
-import * as _langchain_langgraph_sdk_dist_types_stream from '@langchain/langgraph-sdk/dist/types.stream';
-import * as _langchain_langgraph_sdk from '@langchain/langgraph-sdk';
-import { Message, AssistantGraph, Client, Config, Assistant, StreamMode, ThreadState, Thread } from '@langchain/langgraph-sdk';
-import { Subscriber, Observable } from 'rxjs';
-import { MessageType } from '@langchain/core/messages';
-import { RunAgentInput } from '@ag-ui/core';
-import { RunsStreamPayload } from '@langchain/langgraph-sdk/dist/types';
+import { AbstractAgent, AgentConfig, CustomEvent, HttpAgent, MessagesSnapshotEvent, RawEvent, ReasoningEncryptedValueEvent, ReasoningEndEvent, ReasoningMessageContentEvent, ReasoningMessageEndEvent, ReasoningMessageStartEvent, ReasoningStartEvent, RunAgentInput, RunErrorEvent, RunFinishedEvent, RunStartedEvent, StateDeltaEvent, StateSnapshotEvent, StepFinishedEvent, StepStartedEvent, TextMessageContentEvent, TextMessageEndEvent, TextMessageStartEvent, ToolCallArgsEvent, ToolCallEndEvent, ToolCallResultEvent, ToolCallStartEvent } from "@ag-ui/client";
+import * as _langchain_langgraph_sdk0 from "@langchain/langgraph-sdk";
+import { Assistant, AssistantGraph, Client, Config, Message, StreamMode, Thread, ThreadState } from "@langchain/langgraph-sdk";
+import * as _langchain_langgraph_sdk_dist_types_stream0 from "@langchain/langgraph-sdk/dist/types.stream";
+import { Observable, Subscriber } from "rxjs";
+import { MessageType } from "@langchain/core/messages";
+import { RunAgentInput as RunAgentInput$1 } from "@ag-ui/core";
+import { RunsStreamPayload } from "@langchain/langgraph-sdk/dist/types";
 
+//#region src/types.d.ts
 declare enum LangGraphEventTypes {
-    OnChainStart = "on_chain_start",
-    OnChainStream = "on_chain_stream",
-    OnChainEnd = "on_chain_end",
-    OnChatModelStart = "on_chat_model_start",
-    OnChatModelStream = "on_chat_model_stream",
-    OnChatModelEnd = "on_chat_model_end",
-    OnToolStart = "on_tool_start",
-    OnToolEnd = "on_tool_end",
-    OnCustomEvent = "on_custom_event",
-    OnInterrupt = "on_interrupt"
+  OnChainStart = "on_chain_start",
+  OnChainStream = "on_chain_stream",
+  OnChainEnd = "on_chain_end",
+  OnChatModelStart = "on_chat_model_start",
+  OnChatModelStream = "on_chat_model_stream",
+  OnChatModelEnd = "on_chat_model_end",
+  OnToolStart = "on_tool_start",
+  OnToolEnd = "on_tool_end",
+  OnCustomEvent = "on_custom_event",
+  OnInterrupt = "on_interrupt"
 }
 type LangGraphToolWithName = {
-    type: "function";
-    name?: string;
-    function: {
-        name: string;
-        description: string;
-        parameters: any;
-    };
+  type: "function";
+  name?: string;
+  function: {
+    name: string;
+    description: string;
+    parameters: any;
+  };
 };
-type State<TDefinedState = Record<string, any>> = {
-    [k in keyof TDefinedState]: TDefinedState[k] | null;
-} & Record<string, any>;
+type State<TDefinedState = Record<string, any>> = { [k in keyof TDefinedState]: TDefinedState[k] | null } & Record<string, any>;
 interface StateEnrichment {
-    messages: Message[];
+  messages: Message[];
+  tools: LangGraphToolWithName[];
+  'ag-ui': {
     tools: LangGraphToolWithName[];
-    'ag-ui': {
-        tools: LangGraphToolWithName[];
-        context: RunAgentInput['context'];
-    };
+    context: RunAgentInput$1['context'];
+  };
 }
 type SchemaKeys = {
-    input: string[] | null;
-    output: string[] | null;
-    context: string[] | null;
-    config: string[] | null;
+  input: string[] | null;
+  output: string[] | null;
+  context: string[] | null;
+  config: string[] | null;
 } | null;
 type MessageInProgress = {
-    id: string;
-    toolCallId?: string | null;
-    toolCallName?: string | null;
+  id: string;
+  toolCallId?: string | null;
+  toolCallName?: string | null;
 };
-type ThinkingInProgress = {
-    index: number;
-    type?: LangGraphReasoning['type'];
+type ReasoningInProgress = {
+  index: number;
+  type?: LangGraphReasoning['type'];
+  messageId: string;
+  signature?: string;
 };
 interface RunMetadata {
-    id: string;
-    schemaKeys?: SchemaKeys;
-    nodeName?: string;
-    prevNodeName?: string | null;
-    exitingNode?: boolean;
-    manuallyEmittedState?: State | null;
-    threadId?: string;
-    graphInfo?: AssistantGraph;
-    hasFunctionStreaming?: boolean;
-    serverRunIdKnown?: boolean;
+  id: string;
+  schemaKeys?: SchemaKeys;
+  nodeName?: string;
+  prevNodeName?: string | null;
+  exitingNode?: boolean;
+  manuallyEmittedState?: State | null;
+  threadId?: string;
+  graphInfo?: AssistantGraph;
+  hasFunctionStreaming?: boolean;
+  serverRunIdKnown?: boolean;
+  hasPredictState?: boolean;
 }
 type MessagesInProgressRecord = Record<string, MessageInProgress | null>;
 interface ToolCall {
-    id: string;
-    name: string;
-    args: Record<string, unknown>;
+  id: string;
+  name: string;
+  args: Record<string, unknown>;
 }
 type BaseLangGraphPlatformMessage = Omit<Message, "isResultMessage" | "isTextMessage" | "isImageMessage" | "isActionExecutionMessage" | "isAgentStateMessage" | "type" | "createdAt"> & {
-    content: string;
-    role: string;
-    additional_kwargs?: Record<string, unknown>;
-    type: MessageType;
+  content: string;
+  role: string;
+  additional_kwargs?: Record<string, unknown>;
+  type: MessageType;
 };
 interface LangGraphPlatformResultMessage extends BaseLangGraphPlatformMessage {
-    tool_call_id: string;
-    name: string;
+  tool_call_id: string;
+  name: string;
 }
 interface LangGraphPlatformActionExecutionMessage extends BaseLangGraphPlatformMessage {
-    tool_calls: ToolCall[];
+  tool_calls: ToolCall[];
 }
 type LangGraphPlatformMessage = LangGraphPlatformActionExecutionMessage | LangGraphPlatformResultMessage | BaseLangGraphPlatformMessage;
 declare enum CustomEventNames {
-    ManuallyEmitMessage = "manually_emit_message",
-    ManuallyEmitToolCall = "manually_emit_tool_call",
-    ManuallyEmitState = "manually_emit_state",
-    Exit = "exit"
+  ManuallyEmitMessage = "manually_emit_message",
+  ManuallyEmitToolCall = "manually_emit_tool_call",
+  ManuallyEmitState = "manually_emit_state",
+  Exit = "exit"
 }
 interface PredictStateTool {
-    tool: string;
-    state_key: string;
-    tool_argument: string;
+  tool: string;
+  state_key: string;
+  tool_argument: string;
 }
 interface LangGraphReasoning {
-    type: 'text';
-    text: string;
-    index: number;
+  type: 'text';
+  text: string;
+  index: number;
+  signature?: string;
 }
-
-type ProcessedEvents = TextMessageStartEvent | TextMessageContentEvent | TextMessageEndEvent | ThinkingTextMessageStartEvent | ThinkingTextMessageContentEvent | ThinkingTextMessageEndEvent | ToolCallStartEvent | ToolCallArgsEvent | ToolCallEndEvent | ToolCallResultEvent | ThinkingStartEvent | ThinkingEndEvent | StateSnapshotEvent | StateDeltaEvent | MessagesSnapshotEvent | RawEvent | CustomEvent | RunStartedEvent | RunFinishedEvent | RunErrorEvent | StepStartedEvent | StepFinishedEvent;
-type RunAgentExtendedInput<TStreamMode extends StreamMode | StreamMode[] = StreamMode, TSubgraphs extends boolean = false> = Omit<RunAgentInput$1, "forwardedProps"> & {
-    forwardedProps?: Omit<RunsStreamPayload<TStreamMode, TSubgraphs>, "input"> & {
-        nodeName?: string;
-        threadMetadata?: Record<string, any>;
-    };
+//#endregion
+//#region src/agent.d.ts
+type ProcessedEvents = TextMessageStartEvent | TextMessageContentEvent | TextMessageEndEvent | ReasoningStartEvent | ReasoningMessageStartEvent | ReasoningMessageContentEvent | ReasoningMessageEndEvent | ReasoningEndEvent | ReasoningEncryptedValueEvent | ToolCallStartEvent | ToolCallArgsEvent | ToolCallEndEvent | ToolCallResultEvent | StateSnapshotEvent | StateDeltaEvent | MessagesSnapshotEvent | RawEvent | CustomEvent | RunStartedEvent | RunFinishedEvent | RunErrorEvent | StepStartedEvent | StepFinishedEvent;
+type RunAgentExtendedInput<TStreamMode extends StreamMode | StreamMode[] = StreamMode, TSubgraphs extends boolean = false> = Omit<RunAgentInput, "forwardedProps"> & {
+  forwardedProps?: Omit<RunsStreamPayload<TStreamMode, TSubgraphs>, "input"> & {
+    nodeName?: string;
+    threadMetadata?: Record<string, any>;
+  };
 };
 interface RegenerateInput extends RunAgentExtendedInput {
-    messageCheckpoint: Message;
+  messageCheckpoint: Message;
 }
 interface LangGraphAgentConfig extends AgentConfig {
-    client?: Client;
-    deploymentUrl: string;
-    langsmithApiKey?: string;
-    propertyHeaders?: Record<string, string>;
-    assistantConfig?: Config;
-    agentName?: string;
-    graphId: string;
+  client?: Client;
+  deploymentUrl: string;
+  langsmithApiKey?: string;
+  propertyHeaders?: Record<string, string>;
+  assistantConfig?: Config;
+  agentName?: string;
+  graphId: string;
 }
 declare class LangGraphAgent extends AbstractAgent {
-    client: Client;
-    assistantConfig?: Config;
-    agentName?: string;
-    graphId: string;
-    assistant?: Assistant;
-    messagesInProcess: MessagesInProgressRecord;
-    thinkingProcess: null | ThinkingInProgress;
-    activeRun?: RunMetadata;
-    private cancelRequested;
-    private cancelSent;
-    subscriber: Subscriber<ProcessedEvents>;
-    constantSchemaKeys: string[];
-    config: LangGraphAgentConfig;
-    constructor(config: LangGraphAgentConfig);
-    clone(): any;
-    dispatchEvent(event: ProcessedEvents): boolean;
-    run(input: RunAgentInput$1): Observable<ProcessedEvents>;
-    runAgentStream(input: RunAgentExtendedInput, subscriber: Subscriber<ProcessedEvents>): Promise<void>;
-    prepareRegenerateStream(input: RegenerateInput, streamMode: StreamMode | StreamMode[]): Promise<void | {
-        streamResponse: _langchain_langgraph_sdk_dist_types_stream.TypedAsyncGenerator<"messages" | "values" | "updates" | "events" | "debug" | "tasks" | "checkpoints" | "custom" | "messages-tuple" | StreamMode[], false, _langchain_langgraph_sdk.DefaultValues, _langchain_langgraph_sdk.DefaultValues, unknown>;
-        state: ThreadState<State>;
-        streamMode: StreamMode | StreamMode[];
-    }>;
-    prepareStream(input: RunAgentExtendedInput, streamMode: StreamMode | StreamMode[]): Promise<void | {
-        streamResponse: _langchain_langgraph_sdk_dist_types_stream.TypedAsyncGenerator<"messages" | "values" | "updates" | "events" | "debug" | "tasks" | "checkpoints" | "custom" | "messages-tuple" | StreamMode[], false, _langchain_langgraph_sdk.DefaultValues, _langchain_langgraph_sdk.DefaultValues, unknown>;
-        state: ThreadState<State>;
-        streamMode: StreamMode | StreamMode[];
-    } | {
-        streamResponse: _langchain_langgraph_sdk_dist_types_stream.TypedAsyncGenerator<"messages" | "values" | "updates" | "events" | "debug" | "tasks" | "checkpoints" | "custom" | "messages-tuple" | StreamMode[], false, _langchain_langgraph_sdk.DefaultValues, _langchain_langgraph_sdk.DefaultValues, unknown>;
-        state: ThreadState<State>;
-    }>;
-    handleStreamEvents(stream: Awaited<ReturnType<typeof this.prepareStream> | ReturnType<typeof this.prepareRegenerateStream>>, threadId: string, subscriber: Subscriber<ProcessedEvents>, input: RunAgentExtendedInput, streamModes: StreamMode | StreamMode[]): Promise<void>;
-    handleSingleEvent(event: any): void;
-    abortRun(): void;
-    handleThinkingEvent(reasoningData: LangGraphReasoning): void;
-    getStateSnapshot(threadState: ThreadState<State>): State;
-    getOrCreateThread(threadId: string, threadMetadata?: Record<string, any>): Promise<Thread>;
-    getThread(threadId: string): Promise<Thread<_langchain_langgraph_sdk.DefaultValues>>;
-    createThread(payload?: Parameters<typeof this$1.client.threads.create>[0]): Promise<Thread<_langchain_langgraph_sdk.DefaultValues>>;
-    mergeConfigs({ configs, assistant, schemaKeys, }: {
-        configs: Config[];
-        assistant: Assistant;
-        schemaKeys: SchemaKeys;
-    }): Promise<Config>;
-    getMessageInProgress(runId: string): MessageInProgress | null;
-    setMessageInProgress(runId: string, data: MessageInProgress): void;
-    getAssistant(): Promise<Assistant>;
-    getSchemaKeys(): Promise<SchemaKeys>;
-    langGraphDefaultMergeState(state: State, messages: Message[], input: RunAgentExtendedInput): State<StateEnrichment>;
-    handleNodeChange(nodeName: string | undefined): void;
-    startStep(nodeName: string): void;
-    endStep(): void;
-    getCheckpointByMessage(messageId: string, threadId: string, checkpoint?: null | {
-        checkpoint_id?: null | string;
-        checkpoint_ns: string;
-    }): Promise<ThreadState>;
-}
-
-declare class LangGraphHttpAgent extends HttpAgent {
+  client: Client;
+  assistantConfig?: Config;
+  agentName?: string;
+  graphId: string;
+  assistant?: Assistant;
+  messagesInProcess: MessagesInProgressRecord;
+  reasoningProcess: null | ReasoningInProgress;
+  activeRun?: RunMetadata;
+  private cancelRequested;
+  private cancelSent;
+  private eventsStreamActive;
+  subscriber: Subscriber<ProcessedEvents>;
+  constantSchemaKeys: string[];
+  config: LangGraphAgentConfig;
+  constructor(config: LangGraphAgentConfig);
+  clone(): any;
+  dispatchEvent(event: ProcessedEvents): boolean;
+  run(input: RunAgentInput): Observable<ProcessedEvents>;
+  runAgentStream(input: RunAgentExtendedInput, subscriber: Subscriber<ProcessedEvents>): Promise<void>;
+  prepareRegenerateStream(input: RegenerateInput, streamMode: StreamMode | StreamMode[]): Promise<void | {
+    streamResponse: _langchain_langgraph_sdk_dist_types_stream0.TypedAsyncGenerator<"messages" | "values" | "custom" | "updates" | "events" | "debug" | "tasks" | "checkpoints" | "messages-tuple" | StreamMode[], false, _langchain_langgraph_sdk0.DefaultValues, _langchain_langgraph_sdk0.DefaultValues, unknown>;
+    state: ThreadState<State>;
+    streamMode: StreamMode | StreamMode[];
+  }>;
+  prepareStream(input: RunAgentExtendedInput, streamMode: StreamMode | StreamMode[]): Promise<void | {
+    streamResponse: _langchain_langgraph_sdk_dist_types_stream0.TypedAsyncGenerator<"messages" | "values" | "custom" | "updates" | "events" | "debug" | "tasks" | "checkpoints" | "messages-tuple" | StreamMode[], false, _langchain_langgraph_sdk0.DefaultValues, _langchain_langgraph_sdk0.DefaultValues, unknown>;
+    state: ThreadState<State>;
+    streamMode: StreamMode | StreamMode[];
+  } | {
+    streamResponse: _langchain_langgraph_sdk_dist_types_stream0.TypedAsyncGenerator<"messages" | "values" | "custom" | "updates" | "events" | "debug" | "tasks" | "checkpoints" | "messages-tuple" | StreamMode[], false, _langchain_langgraph_sdk0.DefaultValues, _langchain_langgraph_sdk0.DefaultValues, unknown>;
+    state: ThreadState<State>;
+  }>;
+  handleStreamEvents(stream: Awaited<ReturnType<typeof this.prepareStream> | ReturnType<typeof this.prepareRegenerateStream>>, threadId: string, subscriber: Subscriber<ProcessedEvents>, input: RunAgentExtendedInput, streamModes: StreamMode | StreamMode[]): Promise<void>;
+  handleSingleEvent(event: any): void;
+  /**
+   * Process [AIMessageChunk, metadata] tuples from messages-tuple stream mode
+   * and convert them into AG-UI text message and tool call events.
+   * Uses the same messagesInProcess tracking as events-mode streaming.
+   */
+  private handleMessagesTupleEvent;
+  abortRun(): void;
+  handleReasoningEvent(reasoningData: LangGraphReasoning): void;
+  getStateSnapshot(threadState: ThreadState<State>): State;
+  getOrCreateThread(threadId: string, threadMetadata?: Record<string, any>): Promise<Thread>;
+  getThread(threadId: string): Promise<Thread<_langchain_langgraph_sdk0.DefaultValues>>;
+  createThread(payload?: Parameters<typeof this.client.threads.create>[0]): Promise<Thread<_langchain_langgraph_sdk0.DefaultValues>>;
+  mergeConfigs({
+    configs,
+    assistant,
+    schemaKeys
+  }: {
+    configs: Config[];
+    assistant: Assistant;
+    schemaKeys: SchemaKeys;
+  }): Promise<Config>;
+  getMessageInProgress(runId: string): MessageInProgress | null;
+  setMessageInProgress(runId: string, data: MessageInProgress): void;
+  getAssistant(): Promise<Assistant>;
+  getSchemaKeys(): Promise<SchemaKeys>;
+  langGraphDefaultMergeState(state: State, messages: Message[], input: RunAgentExtendedInput): State<StateEnrichment>;
+  handleNodeChange(nodeName: string | undefined): void;
+  startStep(nodeName: string): void;
+  endStep(): void;
+  getCheckpointByMessage(messageId: string, threadId: string, checkpoint?: null | {
+    checkpoint_id?: null | string;
+    checkpoint_ns: string;
+  }): Promise<ThreadState>;
 }
-
-export { CustomEventNames, LangGraphAgent, type LangGraphAgentConfig, LangGraphEventTypes, LangGraphHttpAgent, type LangGraphPlatformMessage, type LangGraphReasoning, type LangGraphToolWithName, type MessageInProgress, type MessagesInProgressRecord, type PredictStateTool, type ProcessedEvents, type RunMetadata, type SchemaKeys, type State, type StateEnrichment, type ThinkingInProgress, type ToolCall };
+//#endregion
+//#region src/index.d.ts
+declare class LangGraphHttpAgent extends HttpAgent {}
+//#endregion
+export { CustomEventNames, LangGraphAgent, LangGraphAgentConfig, LangGraphEventTypes, LangGraphHttpAgent, LangGraphPlatformMessage, LangGraphReasoning, LangGraphToolWithName, MessageInProgress, MessagesInProgressRecord, PredictStateTool, ProcessedEvents, ReasoningInProgress, RunMetadata, SchemaKeys, State, StateEnrichment, ToolCall };
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","names":[],"sources":["../src/types.ts","../src/agent.ts","../src/index.ts"],"mappings":";;;;;;;;;;aAIY,mBAAA;EACV,YAAA;EACA,aAAA;EACA,UAAA;EACA,gBAAA;EACA,iBAAA;EACA,cAAA;EACA,WAAA;EACA,SAAA;EACA,aAAA;EACA,WAAA;AAAA;AAAA,KAGU,qBAAA;EACV,IAAA;EACA,IAAA;EACA,QAAA;IACE,IAAA;IACA,WAAA;IACA,UAAA;EAAA;AAAA;AAAA,KAIQ,KAAA,iBAAsB,MAAA,+BACpB,aAAA,GAAgB,aAAA,CAAc,CAAA,aACxC,MAAA;AAAA,UACa,eAAA;EACf,QAAA,EAAU,OAAA;EACV,KAAA,EAAO,qBAAA;EACP,OAAA;IACE,KAAA,EAAO,qBAAA;IACP,OAAA,EAAS,eAAA;EAAA;AAAA;AAAA,KAID,UAAA;EACV,KAAA;EACA,MAAA;EACA,OAAA;EACA,MAAA;AAAA;AAAA,KAGU,iBAAA;EACV,EAAA;EACA,UAAA;EACA,YAAA;AAAA;AAAA,KAGU,mBAAA;EACV,KAAA;EACA,IAAA,GAAO,kBAAA;EACP,SAAA;EACA,SAAA;AAAA;AAAA,UAGe,WAAA;EACf,EAAA;EACA,UAAA,GAAa,UAAA;EACb,QAAA;EACA,YAAA;EACA,WAAA;EACA,oBAAA,GAAuB,KAAA;EACvB,QAAA;EACA,SAAA,GAAY,cAAA;EACZ,oBAAA;EAEA,gBAAA;EAEA,eAAA;AAAA;AAAA,KAGU,wBAAA,GAA2B,MAAA,SAAe,iBAAA;AAAA,UAGrC,QAAA;EACf,EAAA;EACA,IAAA;EACA,IAAA,EAAM,MAAA;AAAA;AAAA,KAGH,4BAAA,GAA+B,IAAA,CAClC,OAAA;EASA,OAAA;EACA,IAAA;EACA,iBAAA,GAAoB,MAAA;EACpB,IAAA,EAAM,WAAA;AAAA;AAAA,UAGE,8BAAA,SAAuC,4BAAA;EAC/C,YAAA;EACA,IAAA;AAAA;AAAA,UAGQ,uCAAA,SAAgD,4BAAA;EACxD,UAAA,EAAY,QAAA;AAAA;AAAA,KAGF,wBAAA,GACR,uCAAA,GACA,8BAAA,GACA,4BAAA;AAAA,aAEQ,gBAAA;EACV,mBAAA;EACA,oBAAA;EACA,iBAAA;EACA,IAAA;AAAA;AAAA,UAGe,gBAAA;EACf,IAAA;EACA,SAAA;EACA,aAAA;AAAA;AAAA,UAGe,kBAAA;EACf,IAAA;EACA,IAAA;EACA,KAAA;EACA,SAAA;AAAA;;;KC3DU,eAAA,GACR,qBAAA,GACA,uBAAA,GACA,mBAAA,GACA,mBAAA,GACA,0BAAA,GACA,4BAAA,GACA,wBAAA,GACA,iBAAA,GACA,4BAAA,GACA,kBAAA,GACA,iBAAA,GACA,gBAAA,GACA,mBAAA,GACA,kBAAA,GACA,eAAA,GACA,qBAAA,GACA,QAAA,GACA,WAAA,GACA,eAAA,GACA,gBAAA,GACA,aAAA,GACA,gBAAA,GACA,iBAAA;AAAA,KAEC,qBAAA,qBACiB,UAAA,GAAa,UAAA,KAAe,UAAA,wCAE9C,IAAA,CAAK,aAAA;EACP,cAAA,GAAiB,IAAA,CAAK,iBAAA,CAAkB,WAAA,EAAa,UAAA;IACnD,QAAA;IACA,cAAA,GAAiB,MAAA;EAAA;AAAA;AAAA,UAIX,eAAA,SAAwB,qBAAA;EAChC,iBAAA,EAAmB,OAAA;AAAA;AAAA,UAGJ,oBAAA,SAA6B,WAAA;EAC5C,MAAA,GAAS,MAAA;EACT,aAAA;EACA,eAAA;EACA,eAAA,GAAkB,MAAA;EAClB,eAAA,GAAkB,MAAA;EAClB,SAAA;EACA,OAAA;AAAA;AAAA,cAGW,cAAA,SAAuB,aAAA;EAClC,MAAA,EAAQ,MAAA;EACR,eAAA,GAAkB,MAAA;EAClB,SAAA;EACA,OAAA;EACA,SAAA,GAAY,SAAA;EACZ,iBAAA,EAAmB,wBAAA;EACnB,gBAAA,SAAyB,mBAAA;EACzB,SAAA,GAAY,WAAA;EAAA,QAEJ,eAAA;EAAA,QACA,UAAA;EAAA,QAEA,kBAAA;EAER,UAAA,EAAY,UAAA,CAAW,eAAA;EACvB,kBAAA;EACA,MAAA,EAAQ,oBAAA;cAEI,MAAA,EAAQ,oBAAA;EAiBb,KAAA,CAAA;EAoBP,aAAA,CAAc,KAAA,EAAO,eAAA;EAKrB,GAAA,CAAI,KAAA,EAAO,aAAA,GAAa,UAAA,CAAA,eAAA;EAOlB,cAAA,CAAe,KAAA,EAAO,qBAAA,EAAuB,UAAA,EAAY,UAAA,CAAW,eAAA,IAAgB,OAAA;EA2BpF,uBAAA,CAAwB,KAAA,EAAO,eAAA,EAAiB,UAAA,EAAY,UAAA,GAAa,UAAA,KAAY,OAAA;;WAkCxD,WAAA,CAAY,KAAA;;;EAKzC,aAAA,CAAc,KAAA,EAAO,qBAAA,EAAuB,UAAA,EAAY,UAAA,GAAa,UAAA,KAAY,OAAA;;WALpD,WAAA,CAAY,KAAA;;;;WA+JrB,WAAA,CAAY,KAAA;EAAA;EAIhC,kBAAA,CACJ,MAAA,EAAQ,OAAA,CACN,UAAA,aAAuB,aAAA,IAAiB,UAAA,aAAuB,uBAAA,IAEjE,QAAA,UACA,UAAA,EAAY,UAAA,CAAW,eAAA,GACvB,KAAA,EAAO,qBAAA,EACP,WAAA,EAAa,UAAA,GAAa,UAAA,KAAY,OAAA;EAsOxC,iBAAA,CAAkB,KAAA;;;;;;UA4VV,wBAAA;EA8FD,QAAA,CAAA;EAiBP,oBAAA,CAAqB,aAAA,EAAe,kBAAA;EAyDpC,gBAAA,CAAiB,WAAA,EAAa,WAAA,CAAY,KAAA,IAAM,KAAA;EAW1C,iBAAA,CAAkB,QAAA,UAAkB,cAAA,GAAiB,MAAA,gBAAsB,OAAA,CAAQ,MAAA;EAkBnF,SAAA,CAAU,QAAA,WAAgB,OAAA,CAAA,MAAA,CAlBwD,yBAAA,CAkBxD,aAAA;EAI1B,YAAA,CAAa,OAAA,GAAU,UAAA,aAAuB,MAAA,CAAO,OAAA,CAAQ,MAAA,OAAU,OAAA,CAAA,MAAA,CAAtC,yBAAA,CAAsC,aAAA;EAIvE,YAAA,CAAA;IACJ,OAAA;IACA,SAAA;IACA;EAAA;IAEA,OAAA,EAAS,MAAA;IACT,SAAA,EAAW,SAAA;IACX,UAAA,EAAY,UAAA;EAAA,IACb,OAAA,CAAA,MAAA;EA0CD,oBAAA,CAAqB,KAAA,WAAa,iBAAA;EAIlC,oBAAA,CAAqB,KAAA,UAAe,IAAA,EAAM,iBAAA;EAUpC,YAAA,CAAA,GAAgB,OAAA,CAAQ,SAAA;EA4BxB,aAAA,CAAA,GAAiB,OAAA,CAAQ,UAAA;EAgC/B,0BAAA,CAA2B,KAAA,EAAO,KAAA,EAAO,QAAA,EAAU,OAAA,IAAoB,KAAA,EAAO,qBAAA,GAAwB,KAAA,CAAM,eAAA;EA+C5G,gBAAA,CAAiB,QAAA;EAiBjB,SAAA,CAAU,QAAA;EAOV,OAAA,CAAA;EAOM,sBAAA,CACJ,SAAA,UACA,QAAA,UACA,UAAA;IACE,aAAA;IACA,aAAA;EAAA,IAED,OAAA,CAAQ,WAAA;AAAA;;;cCj4CA,kBAAA,SAA2B,SAAA"}

package/dist/index.js

@@ -1,7 +1,6 @@
-"use strict";var Y=Object.defineProperty,Et=Object.defineProperties,vt=Object.getOwnPropertyDescriptor,_t=Object.getOwnPropertyDescriptors,Tt=Object.getOwnPropertyNames,V=Object.getOwnPropertySymbols;var tt=Object.prototype.hasOwnProperty,ht=Object.prototype.propertyIsEnumerable;var ot=(a,n)=>(n=Symbol[a])?n:Symbol.for("Symbol."+a);var lt=(a,n,t)=>n in a?Y(a,n,{enumerable:!0,configurable:!0,writable:!0,value:t}):a[n]=t,f=(a,n)=>{for(var t in n||(n={}))tt.call(n,t)&&lt(a,t,n[t]);if(V)for(var t of V(n))ht.call(n,t)&&lt(a,t,n[t]);return a},T=(a,n)=>Et(a,_t(n));var et=(a,n)=>{var t={};for(var e in a)tt.call(a,e)&&n.indexOf(e)<0&&(t[e]=a[e]);if(a!=null&&V)for(var e of V(a))n.indexOf(e)<0&&ht.call(a,e)&&(t[e]=a[e]);return t};var It=(a,n)=>{for(var t in n)Y(a,t,{get:n[t],enumerable:!0})},Rt=(a,n,t,e)=>{if(n&&typeof n=="object"||typeof n=="function")for(let s of Tt(n))!tt.call(a,s)&&s!==t&&Y(a,s,{get:()=>n[s],enumerable:!(e=vt(n,s))||e.enumerable});return a};var Mt=a=>Rt(Y({},"__esModule",{value:!0}),a);var ct=(a,n,t)=>(n=a[ot("asyncIterator")])?n.call(a):(a=a[ot("iterator")](),n={},t=(e,s)=>(s=a[e])&&(n[e]=i=>new Promise((r,u,g)=>(i=s.call(a,i),g=i.done,Promise.resolve(i.value).then(l=>r({value:l,done:g}),u)))),t("next"),t("return"),n);var Nt={};It(Nt,{CustomEventNames:()=>ut,LangGraphAgent:()=>at,LangGraphEventTypes:()=>dt,LangGraphHttpAgent:()=>it});module.exports=Mt(Nt);var yt=require("@ag-ui/client");var ft=require("rxjs"),St=require("@langchain/langgraph-sdk"),B=require("@ag-ui/client");var dt=(c=>(c.OnChainStart="on_chain_start",c.OnChainStream="on_chain_stream",c.OnChainEnd="on_chain_end",c.OnChatModelStart="on_chat_model_start",c.OnChatModelStream="on_chat_model_stream",c.OnChatModelEnd="on_chat_model_end",c.OnToolStart="on_tool_start",c.OnToolEnd="on_tool_end",c.OnCustomEvent="on_custom_event",c.OnInterrupt="on_interrupt",c))(dt||{}),ut=(s=>(s.ManuallyEmitMessage="manually_emit_message",s.ManuallyEmitToolCall="manually_emit_tool_call",s.ManuallyEmitState="manually_emit_state",s.Exit="exit",s))(ut||{});var o=require("@ag-ui/client");var nt=["messages","tools"];function Q(a,n){return Object.fromEntries(Object.entries(a).filter(([t])=>n.includes(t)))}function gt({mode:a,state:n,schemaKeys:t}){let e=a==="start"?n:null;return e&&(t!=null&&t.input)&&(e=Q(e,[...nt,...t.input])),e}function Ct(a){var t;let n=[];for(let e of a)if(e.type==="text"&&e.text)n.push({type:"text",text:e.text});else if(e.type==="image_url"){let s=typeof e.image_url=="string"?e.image_url:(t=e.image_url)==null?void 0:t.url;if(!s)continue;if(s.startsWith("data:")){let[i,r]=s.split(",",2),u=i.includes(":")?i.split(":")[1].split(";")[0]:"image/png";n.push({type:"binary",mimeType:u,data:r||""})}else n.push({type:"binary",mimeType:"image/png",url:s})}return n}function xt(a){let n=[];for(let t of a)if(t.type==="text")n.push({type:"text",text:t.text});else if(t.type==="binary"){let e;if(t.url)e=t.url;else if(t.data)e=`data:${t.mimeType};base64,${t.data}`;else if(t.id)e=t.id;else continue;n.push({type:"image_url",image_url:{url:e}})}return n}function pt(a){return a.map(n=>{var t;switch(n.type){case"human":let e;return Array.isArray(n.content)?e=Ct(n.content):e=z(K(n.content)),{id:n.id,role:"user",content:e};case"ai":let s=K(n.content);return{id:n.id,role:"assistant",content:s?z(s):"",toolCalls:(t=n.tool_calls)==null?void 0:t.map(i=>({id:i.id,type:"function",function:{name:i.name,arguments:JSON.stringify(i.args)}}))};case"system":return{id:n.id,role:"system",content:z(K(n.content))};case"tool":return{id:n.id,role:"tool",content:z(K(n.content)),toolCallId:n.tool_call_id};default:throw new Error("message type returned from LangGraph is not supported.")}})}function st(a){return a.map((n,t)=>{var e,s;switch(n.role){case"user":let i;return typeof n.content=="string"?i=n.content:Array.isArray(n.content)?i=xt(n.content):i=String(n.content),{id:n.id,role:n.role,content:i,type:"human"};case"assistant":return{id:n.id,type:"ai",role:n.role,content:(e=n.content)!=null?e:"",tool_calls:((s=n.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:n.id,role:n.role,content:n.content,type:"system"};case"tool":return{content:n.content,role:n.role,type:n.role,tool_call_id:n.toolCallId,id:n.id};default:throw console.error(`Message role ${n.role} is not implemented`),new Error("message role is not supported.")}})}function z(a){return typeof a=="string"?a:JSON.stringify(a)}function mt(a){var t,e,s,i,r;let n=(t=a.chunk)==null?void 0:t.content;if(n&&Array.isArray(n)&&n.length&&n[0])return n[0].thinking?{text:n[0].thinking,type:"text",index:n[0].index}:null;if((i=(s=(e=a.chunk.additional_kwargs)==null?void 0:e.reasoning)==null?void 0:s.summary)!=null&&i[0]){let u=(r=a.chunk.additional_kwargs)==null?void 0:r.reasoning.summary[0];return!u||!u.text?null:{type:"text",text:u.text,index:u.index}}return null}function K(a){var n;if(!a)return null;if(typeof a=="string")return a;if(Array.isArray(a)&&a.length){let t=(n=a.find(e=>e.type==="text"))==null?void 0:n.text;return t!=null?t:null}return null}var at=class extends o.AbstractAgent{constructor(t){var e,s;super(t);this.cancelRequested=!1;this.cancelSent=!1;this.constantSchemaKeys=nt;this.config=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 St.Client({apiUrl:t.deploymentUrl,apiKey:t.langsmithApiKey,defaultHeaders:f({},(e=t.propertyHeaders)!=null?e:{})})}clone(){return Object.assign(super.clone(),{config:this.config,messagesInProcess:structuredClone(this.messagesInProcess),agentName:this.agentName,graphId:this.graphId,assistantConfig:this.assistantConfig,thinkingProcess:this.thinkingProcess?structuredClone(this.thinkingProcess):null,constantSchemaKeys:[...this.constantSchemaKeys],client:this.client,assistant:this.assistant,activeRun:this.activeRun?structuredClone(this.activeRun):void 0,cancelRequested:this.cancelRequested,cancelSent:this.cancelSent})}dispatchEvent(t){return this.subscriber.next(t),!0}run(t){return new ft.Observable(e=>(this.runAgentStream(t,e),()=>{}))}async runAgentStream(t,e){var u,g,l;this.activeRun={id:t.runId,threadId:t.threadId,hasFunctionStreaming:!1},this.cancelRequested=!1,this.cancelSent=!1,this.subscriber=e,this.assistant||(this.assistant=await this.getAssistant());let s=(u=t.threadId)!=null?u:(0,B.randomUUID)(),i=(l=(g=t.forwardedProps)==null?void 0:g.streamMode)!=null?l:["events","values","updates"],r=await this.prepareStream(T(f({},t),{threadId:s}),i);if(!r)return e.error("No stream to regenerate");await this.handleStreamEvents(r,s,e,t,Array.isArray(i)?i:[i])}async prepareRegenerateStream(t,e){var l,c,d;let{threadId:s,messageCheckpoint:i}=t,r=await this.getCheckpointByMessage(i.id,s);if(this.assistant||(this.assistant=await this.getAssistant()),!r)return this.subscriber.error("No checkpoint found for message");let u=await this.client.threads.updateState(s,{values:this.langGraphDefaultMergeState(r.values,[],t),checkpointId:r.checkpoint.checkpoint_id,asNode:(c=(l=r.next)==null?void 0:l[0])!=null?c:"__start__"}),g=T(f({},(d=t.forwardedProps)!=null?d:{}),{input:this.langGraphDefaultMergeState(r.values,[i],t),checkpointId:u.checkpoint.checkpoint_id,streamMode:e});return{streamResponse:this.client.runs.stream(s,this.assistant.assistant_id,g),state:r,streamMode:e}}async prepareStream(t,e){var L,P,b,S,G,O,$,X,U,E;let{threadId:s,state:i,messages:r,tools:u,context:g,forwardedProps:l}=t;this.activeRun.manuallyEmittedState=null;let c=l==null?void 0:l.nodeName,d=s!=null?s:(0,B.randomUUID)();this.assistant||(this.assistant=await this.getAssistant());let y=await this.getOrCreateThread(d,l==null?void 0:l.threadMetadata);this.activeRun.threadId=y.thread_id;let _=(L=await this.client.threads.getState(y.thread_id))!=null?L:{values:{}},I=(P=_.values.messages)!=null?P:[],x=st(r),M=this.langGraphDefaultMergeState(T(f({},i),{messages:I}),x,t),N=T(f({},_),{values:T(f({},M),{messages:[...I,...(b=M.messages)!=null?b:[]]})}),A=N.values;if(this.activeRun.schemaKeys=await this.getSchemaKeys(),((S=_.values.messages)!=null?S:[]).length>r.filter(v=>v.role!=="system").length){let v=null;for(let w=r.length-1;w>=0;w--)if(r[w].role==="user"){v=st([r[w]])[0];break}return v?this.prepareRegenerateStream(T(f({},t),{messageCheckpoint:v}),e):this.subscriber.error("No user message found in messages to regenerate")}this.activeRun.graphInfo=await this.client.assistants.getGraph(this.assistant.assistant_id);let p=!((G=l==null?void 0:l.command)!=null&&G.resume)&&d&&this.activeRun.nodeName!="__end__"&&this.activeRun.nodeName?"continue":"start";if(p==="continue"){let v=this.activeRun.graphInfo.edges.find(w=>w.target===this.activeRun.nodeName);await this.client.threads.updateState(d,{values:i,asNode:v==null?void 0:v.source})}let R=gt({mode:p,state:A,schemaKeys:this.activeRun.schemaKeys}),m,J=[this.assistantConfig,l==null?void 0:l.config].filter(Boolean);J.length&&(m=await this.mergeConfigs({configs:J,assistant:this.assistant,schemaKeys:this.activeRun.schemaKeys}));let h=l,{command:k}=h,W=et(h,["command"]);if(k!=null&&k.resume&&typeof k.resume=="string")try{k.resume=JSON.parse(k.resume)}catch(v){}let j=T(f({},W),{command:k,streamMode:e,input:R,config:m,context:f(f({},g),(O=m==null?void 0:m.configurable)!=null?O:{})}),C=(U=(X=($=_.tasks)==null?void 0:$[0])==null?void 0:X.interrupts)!=null?U:[];return C!=null&&C.length&&!((E=l==null?void 0:l.command)!=null&&E.resume)?(this.dispatchEvent({type:o.EventType.RUN_STARTED,threadId:d,runId:t.runId}),this.handleNodeChange(c),C.forEach(v=>{this.dispatchEvent({type:o.EventType.CUSTOM,name:"on_interrupt",value:typeof v.value=="string"?v.value:JSON.stringify(v.value),rawEvent:v})}),this.dispatchEvent({type:o.EventType.RUN_FINISHED,threadId:d,runId:t.runId}),this.subscriber.complete()):{streamResponse:this.client.runs.stream(d,this.assistant.assistant_id,j),state:N}}async handleStreamEvents(t,e,s,i,r){var I,x,M,N,A,p,R,m,C,L,P,b,S,G;let{forwardedProps:u}=i,g=u==null?void 0:u.nodeName;this.subscriber=s;let l=!1;if(!t)return;let{streamResponse:c,state:d}=t;this.activeRun.prevNodeName=null;let y={},_=d;try{this.dispatchEvent({type:o.EventType.RUN_STARTED,threadId:e,runId:this.activeRun.id}),this.handleNodeChange(g);try{for(var J=ct(c),k,W,j;k=!(W=await J.next()).done;k=!1){let E=W.value;if(this.cancelRequested&&!this.cancelSent&&((I=this.activeRun)!=null&&I.threadId)&&((x=this.activeRun)!=null&&x.id)){try{await this.client.runs.cancel(this.activeRun.threadId,this.activeRun.id)}catch(q){}finally{this.cancelSent=!0}try{await((M=c==null?void 0:c.return)==null?void 0:M.call(c))}catch(q){}break}let v=(N=i.forwardedProps)==null?void 0:N.streamSubgraphs,w=v&&(E.event.startsWith("events")||E.event.startsWith("values"));if(!r.includes(E.event)&&!w&&E.event!=="error")continue;let F=E;if(E.event==="error"){this.dispatchEvent({type:o.EventType.RUN_ERROR,message:E.data.message,rawEvent:E});break}if(E.event==="updates")continue;if(E.event==="values"){y=F.data;continue}else if(v&&F.event.startsWith("values|")){y=f(f({},y),F.data);continue}let H=F.data,Z=(A=H.metadata)!=null?A:{},D=Z.langgraph_node,rt=H.event;if(Z.run_id&&(this.activeRun.id=Z.run_id,this.activeRun.serverRunIdKnown=!0,this.cancelRequested&&!this.cancelSent&&((p=this.activeRun)!=null&&p.threadId)))try{await this.client.runs.cancel(this.activeRun.threadId,this.activeRun.id)}catch(q){}finally{this.cancelSent=!0}if(D&&D!==this.activeRun.nodeName&&this.handleNodeChange(D),l=l||rt==="on_custom_event"&&H.name==="exit",rt==="on_chain_end"&&this.activeRun.nodeName===D&&(this.activeRun.exitingNode=!0),this.activeRun.exitingNode&&(this.activeRun.manuallyEmittedState=null),(R=this.activeRun.graphInfo)!=null&&R.nodes.some(q=>q.id===D)&&this.handleNodeChange(D),_.values=(m=this.activeRun.manuallyEmittedState)!=null?m:y,!this.activeRun.nodeName)continue;(JSON.stringify(_)!==JSON.stringify(d)||this.activeRun.prevNodeName!=this.activeRun.nodeName||this.activeRun.exitingNode)&&!this.getMessageInProgress(this.activeRun.id)&&(d=_,this.activeRun.prevNodeName=this.activeRun.nodeName,this.dispatchEvent({type:o.EventType.STATE_SNAPSHOT,snapshot:this.getStateSnapshot(d),rawEvent:F})),this.dispatchEvent({type:o.EventType.RAW,event:H}),this.handleSingleEvent(H)}}catch(W){j=[W]}finally{try{k&&(W=J.return)&&await W.call(J)}finally{if(j)throw j[0]}}d=await this.client.threads.getState(e);let h=d.tasks,O=(L=(C=h==null?void 0:h[0])==null?void 0:C.interrupts)!=null?L:[],$=d.next.length===0,X=(b=(P=d.metadata)==null?void 0:P.writes)!=null?b:{},U=this.activeRun.nodeName;return O!=null&&O.length||(U=$?"__end__":(S=d.next[0])!=null?S:Object.keys(X)[0]),O.forEach(E=>{this.dispatchEvent({type:o.EventType.CUSTOM,name:"on_interrupt",value:typeof E.value=="string"?E.value:JSON.stringify(E.value),rawEvent:E})}),this.handleNodeChange(U),this.handleNodeChange(void 0),this.dispatchEvent({type:o.EventType.STATE_SNAPSHOT,snapshot:this.getStateSnapshot(d)}),this.dispatchEvent({type:o.EventType.MESSAGES_SNAPSHOT,messages:pt((G=d.values.messages)!=null?G:[])}),this.dispatchEvent({type:o.EventType.RUN_FINISHED,threadId:e,runId:this.activeRun.id}),this.cancelRequested=!1,this.cancelSent=!1,this.activeRun=void 0,s.complete()}catch(h){return s.error(h)}}handleSingleEvent(t){var e,s,i,r,u,g,l,c,d,y,_,I,x,M;switch(t.event){case"on_chat_model_stream":let N=(e=t.metadata["emit-messages"])!=null?e:!0,A=(s=t.metadata["emit-tool-calls"])!=null?s:!0;if(t.data.chunk.response_metadata.finish_reason)return;let p=this.getMessageInProgress(this.activeRun.id),R=!!(p!=null&&p.id),m=(i=t.data.chunk.tool_call_chunks)==null?void 0:i[0],J=(r=t.metadata.predict_state)==null?void 0:r.some(h=>h.tool===(m==null?void 0:m.name)),k=!R&&(m==null?void 0:m.name),W=R&&(p==null?void 0:p.toolCallId)&&(m==null?void 0:m.args),j=R&&(p==null?void 0:p.toolCallId)&&!m;(j||W||k)&&(this.activeRun.hasFunctionStreaming=!0);let C=mt(t.data),L=K(t.data.chunk.content),P=!!(!m&&L),b=R&&!(p!=null&&p.toolCallId)&&!P;if(C){this.handleThinkingEvent(C);break}if(!C&&this.thinkingProcess&&(this.dispatchEvent({type:o.EventType.THINKING_TEXT_MESSAGE_END}),this.dispatchEvent({type:o.EventType.THINKING_END}),this.thinkingProcess=null),J&&this.dispatchEvent({type:o.EventType.CUSTOM,name:"PredictState",value:t.metadata.predict_state}),j){this.dispatchEvent({type:o.EventType.TOOL_CALL_END,toolCallId:p==null?void 0:p.toolCallId,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if(b){this.dispatchEvent({type:o.EventType.TEXT_MESSAGE_END,messageId:p.id,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if(k&&A){this.dispatchEvent({type:o.EventType.TOOL_CALL_START,toolCallId:m.id,toolCallName:m.name,parentMessageId:t.data.chunk.id,rawEvent:t})&&this.setMessageInProgress(this.activeRun.id,{id:t.data.chunk.id,toolCallId:m.id,toolCallName:m.name});break}if(W&&A){this.dispatchEvent({type:o.EventType.TOOL_CALL_ARGS,toolCallId:p==null?void 0:p.toolCallId,delta:m.args,rawEvent:t});break}if(P&&N){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}),p=this.getMessageInProgress(this.activeRun.id)),this.dispatchEvent({type:o.EventType.TEXT_MESSAGE_CONTENT,messageId:p.id,delta:L,rawEvent:t});break}break;case"on_chat_model_end":if((u=this.getMessageInProgress(this.activeRun.id))!=null&&u.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((g=this.getMessageInProgress(this.activeRun.id))!=null&&g.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({values:this.activeRun.manuallyEmittedState}),rawEvent:t})),this.dispatchEvent({type:o.EventType.CUSTOM,name:t.name,value:t.data,rawEvent:t});break;case"on_tool_end":let S=(l=t.data)==null?void 0:l.output;if(S&&!S.tool_call_id&&((d=(c=S.update)==null?void 0:c.messages)!=null&&d.find(h=>h.type==="tool"))&&(S=(_=(y=S.update)==null?void 0:y.messages)==null?void 0:_.find(h=>h.type==="tool")),S&&((x=(I=S.update)==null?void 0:I.messages)!=null&&x.length)){(M=S.update)==null||M.messages.filter(h=>h.type==="tool").forEach(h=>{var O;this.activeRun.hasFunctionStreaming||(this.dispatchEvent({type:o.EventType.TOOL_CALL_START,toolCallId:h.tool_call_id,toolCallName:(O=h.name)!=null?O:"",parentMessageId:h.id,rawEvent:t}),this.dispatchEvent({type:o.EventType.TOOL_CALL_ARGS,toolCallId:h.tool_call_id,delta:JSON.stringify(t.data.input),rawEvent:t})),this.dispatchEvent({type:o.EventType.TOOL_CALL_RESULT,toolCallId:h.tool_call_id,content:typeof(h==null?void 0:h.content)=="string"?h==null?void 0:h.content:JSON.stringify(h==null?void 0:h.content),messageId:(0,B.randomUUID)(),rawEvent:t,role:"tool"})});break}this.activeRun.hasFunctionStreaming||(this.dispatchEvent({type:o.EventType.TOOL_CALL_START,toolCallId:S.tool_call_id,toolCallName:S.name,parentMessageId:S.id,rawEvent:t}),this.dispatchEvent({type:o.EventType.TOOL_CALL_ARGS,toolCallId:S.tool_call_id,delta:JSON.stringify(t.data.input),rawEvent:t}),this.dispatchEvent({type:o.EventType.TOOL_CALL_END,toolCallId:S.tool_call_id,rawEvent:t}));let G=Array.isArray(S.content)?S.content.map(h=>typeof h=="string"?h:h.type==="text"?h.text:JSON.stringify(h)).join(""):S.content;this.dispatchEvent({type:o.EventType.TOOL_CALL_RESULT,toolCallId:S.tool_call_id,content:G,messageId:(0,B.randomUUID)(),role:"tool",rawEvent:t});break}}abortRun(){var s,i;this.cancelRequested=!0;let t=(s=this.activeRun)==null?void 0:s.threadId,e=(i=this.activeRun)==null?void 0:i.id;t&&e&&!this.cancelSent&&this.client.runs.cancel(t,e).then(()=>{this.cancelSent=!0}).catch(()=>{}),super.abortRun()}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:o.EventType.THINKING_TEXT_MESSAGE_END}),this.dispatchEvent({type:o.EventType.THINKING_END}),this.thinkingProcess=null),this.thinkingProcess||(this.dispatchEvent({type:o.EventType.THINKING_START}),this.thinkingProcess={index:e}),this.thinkingProcess.type!==t.type&&(this.dispatchEvent({type:o.EventType.THINKING_TEXT_MESSAGE_START}),this.thinkingProcess.type=t.type),this.thinkingProcess.type&&this.dispatchEvent({type:o.EventType.THINKING_TEXT_MESSAGE_CONTENT,delta:t.text})}getStateSnapshot(t){let e=t.values,s=this.activeRun.schemaKeys;return s!=null&&s.output&&(e=Q(e,[...this.constantSchemaKeys,...s.output])),e}async getOrCreateThread(t,e){let s;try{try{s=await this.getThread(t)}catch(i){s=await this.createThread({threadId:t,metadata:e})}}catch(i){throw new Error(`Failed to create thread: ${i.message}`)}return s}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((i,r)=>{var y;let u=i.configurable;r.configurable&&(u=s!=null&&s.config?Q(r==null?void 0:r.configurable,[...this.constantSchemaKeys,...(y=s==null?void 0:s.config)!=null?y:[]]):r==null?void 0:r.configurable);let g=T(f(f({},i),r),{configurable:u}),l=i.recursion_limit==null&&r.recursion_limit===25,c=JSON.stringify(g)!==JSON.stringify(i),d=l&&JSON.stringify(T(f({},g),{recursion_limit:null}))===JSON.stringify(T(f({},i),{recursion_limit:null}));return c&&!d?f(f({},i),g):i},e.config)}getMessageInProgress(t){return this.messagesInProcess[t]}setMessageInProgress(t,e){this.messagesInProcess=T(f({},this.messagesInProcess),{[t]:f(f({},this.messagesInProcess[t]),e)})}async getAssistant(){try{let t=await this.client.assistants.search(),e=t.find(s=>s.graph_id===this.graphId);if(!e){let s=`
-      No agent found with graph ID ${this.graphId} found..
+let e=require(`@ag-ui/client`),t=require(`rxjs`),n=require(`@langchain/langgraph-sdk`),r=function(e){return e.OnChainStart=`on_chain_start`,e.OnChainStream=`on_chain_stream`,e.OnChainEnd=`on_chain_end`,e.OnChatModelStart=`on_chat_model_start`,e.OnChatModelStream=`on_chat_model_stream`,e.OnChatModelEnd=`on_chat_model_end`,e.OnToolStart=`on_tool_start`,e.OnToolEnd=`on_tool_end`,e.OnCustomEvent=`on_custom_event`,e.OnInterrupt=`on_interrupt`,e}({}),i=function(e){return e.ManuallyEmitMessage=`manually_emit_message`,e.ManuallyEmitToolCall=`manually_emit_tool_call`,e.ManuallyEmitState=`manually_emit_state`,e.Exit=`exit`,e}({});const a=[`messages`,`tools`];function o(e,t){return Object.fromEntries(Object.entries(e).filter(([e])=>t.includes(e)))}function s({mode:e,state:t,schemaKeys:n}){let r=e===`start`?t:null;return r&&n?.input&&(r=o(r,[...a,...n.input])),r}function c(e){let t=[];for(let n of e)if(n.type===`text`&&n.text)t.push({type:`text`,text:n.text});else if(n.type===`image_url`){let e=typeof n.image_url==`string`?n.image_url:n.image_url?.url;if(!e)continue;if(e.startsWith(`data:`)){let[n,r]=e.split(`,`,2),i=n.includes(`:`)?n.split(`:`)[1].split(`;`)[0]:`image/png`;t.push({type:`binary`,mimeType:i,data:r||``})}else t.push({type:`binary`,mimeType:`image/png`,url:e})}return t}function l(e){let t=[];for(let n of e)if(n.type===`text`)t.push({type:`text`,text:n.text});else if(n.type===`binary`){let e;if(n.url)e=n.url;else if(n.data)e=`data:${n.mimeType};base64,${n.data}`;else if(n.id)e=n.id;else continue;t.push({type:`image_url`,image_url:{url:e}})}return t}function u(e){return e.map(e=>{switch(e.type){case`human`:let t;return t=Array.isArray(e.content)?c(e.content):f(h(e.content)),{id:e.id,role:`user`,content:t};case`ai`:let n=h(e.content);return{id:e.id,role:`assistant`,content:n?f(n):``,toolCalls:e.tool_calls?.map(e=>({id:e.id,type:`function`,function:{name:e.name,arguments:JSON.stringify(e.args)}}))};case`system`:return{id:e.id,role:`system`,content:f(h(e.content))};case`tool`:return{id:e.id,role:`tool`,content:f(h(e.content)),toolCallId:e.tool_call_id};default:throw Error(`message type returned from LangGraph is not supported.`)}})}function d(e){return e.map((e,t)=>{switch(e.role){case`user`:let t;return t=typeof e.content==`string`?e.content:Array.isArray(e.content)?l(e.content):String(e.content),{id:e.id,role:e.role,content:t,type:`human`};case`assistant`:return{id:e.id,type:`ai`,role:e.role,content:e.content??``,tool_calls:(e.toolCalls??[]).map(e=>({id:e.id,name:e.function.name,args:JSON.parse(e.function.arguments),type:`tool_call`}))};case`system`:return{id:e.id,role:e.role,content:e.content,type:`system`};case`tool`:return{content:e.content,role:e.role,type:e.role,tool_call_id:e.toolCallId,id:e.id};default:throw console.error(`Message role ${e.role} is not implemented`),Error(`message role is not supported.`)}})}function f(e){return typeof e==`string`?e:JSON.stringify(e)}function p(e){let t=e.chunk?.content;if(t&&Array.isArray(t)&&t.length&&t[0]){let e=t[0];if(e.type===`thinking`&&e.thinking){let t={text:e.thinking,type:`text`,index:e.index??0};return e.signature&&(t.signature=e.signature),t}if(e.type===`reasoning`&&e.reasoning)return{text:e.reasoning,type:`text`,index:e.index??0};if(e.type===`reasoning`&&e.summary?.[0]?.text)return{type:`text`,text:e.summary[0].text,index:e.summary[0].index??0}}if(e.chunk?.additional_kwargs?.reasoning?.summary?.[0]){let t=e.chunk.additional_kwargs.reasoning.summary[0];return!t||!t.text?null:{type:`text`,text:t.text,index:t.index??0}}return null}function m(e){let t=e.chunk?.content;return!t||!Array.isArray(t)||!t.length||!t[0]?null:t[0].type===`redacted_thinking`&&t[0].data?t[0].data:null}function h(e){return e?typeof e==`string`?e:Array.isArray(e)&&e.length?e.find(e=>e.type===`text`)?.text??null:null:null}var g=class extends e.AbstractAgent{constructor(e){super(e),this.cancelRequested=!1,this.cancelSent=!1,this.eventsStreamActive=!1,this.constantSchemaKeys=a,this.config=e,this.messagesInProcess={},this.agentName=e.agentName,this.graphId=e.graphId,this.assistantConfig=e.assistantConfig,this.reasoningProcess=null,this.client=e?.client??new n.Client({apiUrl:e.deploymentUrl,apiKey:e.langsmithApiKey,defaultHeaders:{...e.propertyHeaders??{}}})}clone(){return Object.assign(super.clone(),{config:this.config,messagesInProcess:structuredClone(this.messagesInProcess),agentName:this.agentName,graphId:this.graphId,assistantConfig:this.assistantConfig,reasoningProcess:this.reasoningProcess?structuredClone(this.reasoningProcess):null,constantSchemaKeys:[...this.constantSchemaKeys],client:this.client,assistant:this.assistant,activeRun:this.activeRun?structuredClone(this.activeRun):void 0,cancelRequested:this.cancelRequested,cancelSent:this.cancelSent})}dispatchEvent(e){return this.subscriber.next(e),!0}run(e){return new t.Observable(t=>(this.runAgentStream(e,t),()=>{}))}async runAgentStream(t,n){this.activeRun={id:t.runId,threadId:t.threadId,hasFunctionStreaming:!1,hasPredictState:!1},this.cancelRequested=!1,this.cancelSent=!1,this.eventsStreamActive=!1,this.subscriber=n,this.assistant||=await this.getAssistant();let r=t.threadId??(0,e.randomUUID)(),i=t.forwardedProps?.streamMode??[`events`,`values`,`updates`,`messages-tuple`],a=await this.prepareStream({...t,threadId:r},i);if(!a)return n.error(`No stream to regenerate`);await this.handleStreamEvents(a,r,n,t,Array.isArray(i)?i:[i])}async prepareRegenerateStream(e,t){let{threadId:n,messageCheckpoint:r}=e,i=await this.getCheckpointByMessage(r.id,n);if(this.assistant||=await this.getAssistant(),!i)return this.subscriber.error(`No checkpoint found for message`);let a=await this.client.threads.updateState(n,{values:this.langGraphDefaultMergeState(i.values,[],e),checkpointId:i.checkpoint.checkpoint_id,asNode:i.next?.[0]??`__start__`}),o={...e.forwardedProps??{},input:this.langGraphDefaultMergeState(i.values,[r],e),checkpointId:a.checkpoint.checkpoint_id,streamMode:t};return{streamResponse:this.client.runs.stream(n,this.assistant.assistant_id,o),state:i,streamMode:t}}async prepareStream(t,n){let{threadId:i,state:a,messages:o,tools:c,context:l,forwardedProps:u}=t;this.activeRun.manuallyEmittedState=null;let f=u?.nodeName,p=i??(0,e.randomUUID)();this.assistant||=await this.getAssistant();let m=await this.getOrCreateThread(p,u?.threadMetadata);this.activeRun.threadId=m.thread_id;let h=await this.client.threads.getState(m.thread_id)??{values:{}},g=h.values.messages??[],_=d(o),v=this.langGraphDefaultMergeState({...a,messages:g},_,t),y={...h,values:{...v,messages:[...g,...v.messages??[]]}},b=y.values;if(this.activeRun.schemaKeys=await this.getSchemaKeys(),(h.values.messages??[]).length>o.filter(e=>e.role!==`system`).length){let e=null;for(let t=o.length-1;t>=0;t--)if(o[t].role===`user`){e=d([o[t]])[0];break}return e?this.prepareRegenerateStream({...t,messageCheckpoint:e},n):this.subscriber.error(`No user message found in messages to regenerate`)}this.activeRun.graphInfo=await this.client.assistants.getGraph(this.assistant.assistant_id);let x=!u?.command?.resume&&p&&this.activeRun.nodeName!=`__end__`&&this.activeRun.nodeName?`continue`:`start`;if(x===`continue`){let e=this.activeRun.graphInfo.edges.find(e=>e.target===this.activeRun.nodeName);await this.client.threads.updateState(p,{values:a,asNode:e?.source})}let S=s({mode:x,state:b,schemaKeys:this.activeRun.schemaKeys}),C,w=[this.assistantConfig,u?.config].filter(Boolean);w.length&&(C=await this.mergeConfigs({configs:w,assistant:this.assistant,schemaKeys:this.activeRun.schemaKeys}));let{command:T,...E}=u;if(T?.resume&&typeof T.resume==`string`)try{T.resume=JSON.parse(T.resume)}catch{}let D={...E,command:T,streamMode:n,input:S,config:C,context:{...l,...C?.configurable??{}}},O=h.tasks?.[0]?.interrupts??[];return O?.length&&!u?.command?.resume?(this.dispatchEvent({type:e.EventType.RUN_STARTED,threadId:p,runId:t.runId}),this.handleNodeChange(f),O.forEach(t=>{this.dispatchEvent({type:e.EventType.CUSTOM,name:r.OnInterrupt,value:typeof t.value==`string`?t.value:JSON.stringify(t.value),rawEvent:t})}),this.dispatchEvent({type:e.EventType.RUN_FINISHED,threadId:p,runId:t.runId}),this.subscriber.complete()):{streamResponse:this.client.runs.stream(p,this.assistant.assistant_id,D),state:y}}async handleStreamEvents(t,n,a,o,s){let{forwardedProps:c}=o,l=c?.nodeName;this.subscriber=a;let d=!1;if(!t)return;let{streamResponse:f,state:p}=t;this.activeRun.prevNodeName=null;let m={},h=p;try{this.dispatchEvent({type:e.EventType.RUN_STARTED,threadId:n,runId:this.activeRun.id}),this.handleNodeChange(l);for await(let t of f){if(this.cancelRequested&&!this.cancelSent&&this.activeRun?.threadId&&this.activeRun?.id){try{await this.client.runs.cancel(this.activeRun.threadId,this.activeRun.id)}catch{}finally{this.cancelSent=!0}try{await f?.return?.()}catch{}break}let n=o.forwardedProps?.streamSubgraphs,a=n&&(t.event.startsWith(`events`)||t.event.startsWith(`values`)),c=t.event===`messages`&&(Array.isArray(s)?s:[s]).includes(`messages-tuple`);if(!s.includes(t.event)&&!a&&!c&&t.event!==`error`)continue;let l=t;if(t.event===`error`){this.dispatchEvent({type:e.EventType.RUN_ERROR,message:t.data.message,rawEvent:t});break}if(t.event===`updates`)continue;if(t.event===`values`){m=l.data;continue}else if(n&&l.event.startsWith(`values|`)){m={...m,...l.data};continue}let u=l.data,g=u.metadata??{},_=g.langgraph_node,v=u.event;if(g.run_id&&(this.activeRun.id=g.run_id,this.activeRun.serverRunIdKnown=!0,this.cancelRequested&&!this.cancelSent&&this.activeRun?.threadId))try{await this.client.runs.cancel(this.activeRun.threadId,this.activeRun.id)}catch{}finally{this.cancelSent=!0}if(_&&_!==this.activeRun.nodeName&&this.handleNodeChange(_),d||=v===r.OnCustomEvent&&u.name===i.Exit,v===r.OnChainEnd&&this.activeRun.nodeName===_&&(this.activeRun.exitingNode=!0),this.activeRun.exitingNode&&(this.activeRun.manuallyEmittedState=null),this.activeRun.graphInfo?.nodes.some(e=>e.id===_)&&this.handleNodeChange(_),h.values=this.activeRun.manuallyEmittedState??m,!this.activeRun.nodeName)continue;let y=JSON.stringify(h)!==JSON.stringify(p),b=this.activeRun.exitingNode&&this.activeRun.hasPredictState;(y||this.activeRun.prevNodeName!=this.activeRun.nodeName||this.activeRun.exitingNode)&&!this.getMessageInProgress(this.activeRun.id)&&!b?(p=h,this.activeRun.prevNodeName=this.activeRun.nodeName,this.dispatchEvent({type:e.EventType.STATE_SNAPSHOT,snapshot:this.getStateSnapshot(p),rawEvent:l})):b&&console.debug(`[ag-ui/langgraph] Suppressing STATE_SNAPSHOT on node exit (node=${this.activeRun.nodeName}, hasPredictState=${this.activeRun.hasPredictState})`),this.dispatchEvent({type:e.EventType.RAW,event:u}),this.handleSingleEvent(u)}p=await this.client.threads.getState(n);let t=p.tasks?.[0]?.interrupts??[],c=p.next.length===0,g=p.metadata?.writes??{},_=this.activeRun.nodeName;return t?.length||(_=c?`__end__`:p.next[0]??Object.keys(g)[0]),t.forEach(t=>{this.dispatchEvent({type:e.EventType.CUSTOM,name:r.OnInterrupt,value:typeof t.value==`string`?t.value:JSON.stringify(t.value),rawEvent:t})}),this.handleNodeChange(_),this.handleNodeChange(void 0),this.dispatchEvent({type:e.EventType.STATE_SNAPSHOT,snapshot:this.getStateSnapshot(p)}),this.dispatchEvent({type:e.EventType.MESSAGES_SNAPSHOT,messages:u(p.values.messages??[])}),this.dispatchEvent({type:e.EventType.RUN_FINISHED,threadId:n,runId:this.activeRun.id}),this.cancelRequested=!1,this.cancelSent=!1,this.activeRun=void 0,a.complete()}catch(e){return a.error(e)}}handleSingleEvent(t){if(Array.isArray(t)){this.eventsStreamActive||this.handleMessagesTupleEvent(t);return}switch(t.event===r.OnChatModelStream&&(this.eventsStreamActive=!0),t.event){case r.OnChatModelStream:let n=t.metadata?.[`emit-messages`]??!0,a=t.metadata?.[`emit-tool-calls`]??!0;if(t.data.chunk.response_metadata.finish_reason)return;let o=this.getMessageInProgress(this.activeRun.id),s=!!o?.id,c=t.data.chunk.tool_call_chunks?.[0],l=t.metadata?.predict_state?.some(e=>e.tool===c?.name),u=!s&&c?.name,d=s&&o?.toolCallId&&c?.args,f=s&&o?.toolCallId&&!c;(f||d||u)&&(this.activeRun.hasFunctionStreaming=!0);let g=p(t.data),_=m(t.data),v=h(t.data.chunk.content),y=!!(!c&&v),b=s&&!o?.toolCallId&&!y;if(g){this.handleReasoningEvent(g);break}if(_&&this.reasoningProcess){this.dispatchEvent({type:e.EventType.REASONING_ENCRYPTED_VALUE,subtype:`message`,entityId:this.reasoningProcess.messageId,encryptedValue:_});break}if(!g&&this.reasoningProcess&&(this.reasoningProcess.signature&&this.dispatchEvent({type:e.EventType.REASONING_ENCRYPTED_VALUE,subtype:`message`,entityId:this.reasoningProcess.messageId,encryptedValue:this.reasoningProcess.signature}),this.dispatchEvent({type:e.EventType.REASONING_MESSAGE_END,messageId:this.reasoningProcess.messageId}),this.dispatchEvent({type:e.EventType.REASONING_END,messageId:this.reasoningProcess.messageId}),this.reasoningProcess=null),l&&(this.activeRun.hasPredictState=!0,this.dispatchEvent({type:e.EventType.CUSTOM,name:`PredictState`,value:t.metadata?.predict_state})),f){this.dispatchEvent({type:e.EventType.TOOL_CALL_END,toolCallId:o?.toolCallId,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if(b){this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_END,messageId:o.id,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if(u&&a){this.dispatchEvent({type:e.EventType.TOOL_CALL_START,toolCallId:c.id,toolCallName:c.name,parentMessageId:t.data.chunk.id,rawEvent:t})&&this.setMessageInProgress(this.activeRun.id,{id:t.data.chunk.id,toolCallId:c.id,toolCallName:c.name});break}if(d&&a){this.dispatchEvent({type:e.EventType.TOOL_CALL_ARGS,toolCallId:o?.toolCallId,delta:c.args,rawEvent:t});break}if(y&&n){o||=(this.dispatchEvent({type:e.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}),this.getMessageInProgress(this.activeRun.id)),this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_CONTENT,messageId:o.id,delta:v,rawEvent:t});break}break;case r.OnChatModelEnd:if(this.getMessageInProgress(this.activeRun.id)?.toolCallId){this.dispatchEvent({type:e.EventType.TOOL_CALL_END,toolCallId:this.getMessageInProgress(this.activeRun.id).toolCallId,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}if(this.getMessageInProgress(this.activeRun.id)?.id){this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_END,messageId:this.getMessageInProgress(this.activeRun.id).id,rawEvent:t})&&(this.messagesInProcess[this.activeRun.id]=null);break}break;case r.OnCustomEvent:if(t.name===i.ManuallyEmitMessage){this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_START,role:`assistant`,messageId:t.data.message_id,rawEvent:t}),this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_CONTENT,messageId:t.data.message_id,delta:t.data.message,rawEvent:t}),this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_END,messageId:t.data.message_id,rawEvent:t});break}if(t.name===i.ManuallyEmitToolCall){this.dispatchEvent({type:e.EventType.TOOL_CALL_START,toolCallId:t.data.id,toolCallName:t.data.name,parentMessageId:t.data.id,rawEvent:t}),this.dispatchEvent({type:e.EventType.TOOL_CALL_ARGS,toolCallId:t.data.id,delta:t.data.args,rawEvent:t}),this.dispatchEvent({type:e.EventType.TOOL_CALL_END,toolCallId:t.data.id,rawEvent:t});break}t.name===i.ManuallyEmitState&&(this.activeRun.manuallyEmittedState=t.data,this.dispatchEvent({type:e.EventType.STATE_SNAPSHOT,snapshot:this.getStateSnapshot({values:this.activeRun.manuallyEmittedState}),rawEvent:t})),this.dispatchEvent({type:e.EventType.CUSTOM,name:t.name,value:t.data,rawEvent:t});break;case r.OnToolEnd:this.activeRun.hasPredictState=!1;let x=t.data?.output;if(x&&!x.tool_call_id&&x.update?.messages?.find(e=>e.type===`tool`)&&(x=x.update?.messages?.find(e=>e.type===`tool`)),x&&x.update?.messages?.length){x.update?.messages.filter(e=>e.type===`tool`).forEach(n=>{this.activeRun.hasFunctionStreaming||(this.dispatchEvent({type:e.EventType.TOOL_CALL_START,toolCallId:n.tool_call_id,toolCallName:n.name??``,parentMessageId:n.id,rawEvent:t}),this.dispatchEvent({type:e.EventType.TOOL_CALL_ARGS,toolCallId:n.tool_call_id,delta:JSON.stringify(t.data.input),rawEvent:t})),this.dispatchEvent({type:e.EventType.TOOL_CALL_RESULT,toolCallId:n.tool_call_id,content:typeof n?.content==`string`?n?.content:JSON.stringify(n?.content),messageId:(0,e.randomUUID)(),rawEvent:t,role:`tool`})});break}this.activeRun.hasFunctionStreaming||(this.dispatchEvent({type:e.EventType.TOOL_CALL_START,toolCallId:x.tool_call_id,toolCallName:x.name,parentMessageId:x.id,rawEvent:t}),this.dispatchEvent({type:e.EventType.TOOL_CALL_ARGS,toolCallId:x.tool_call_id,delta:JSON.stringify(t.data.input),rawEvent:t}),this.dispatchEvent({type:e.EventType.TOOL_CALL_END,toolCallId:x.tool_call_id,rawEvent:t}));let S=Array.isArray(x.content)?x.content.map(e=>typeof e==`string`?e:e.type===`text`?e.text:JSON.stringify(e)).join(``):x.content;this.dispatchEvent({type:e.EventType.TOOL_CALL_RESULT,toolCallId:x.tool_call_id,content:S,messageId:(0,e.randomUUID)(),role:`tool`,rawEvent:t});break}}handleMessagesTupleEvent(t){let n=t[0];if(n.type&&n.type!==`AIMessageChunk`)return;let r=typeof n.content==`string`?n.content:Array.isArray(n.content)?n.content.find(e=>e.type===`text`)?.text:null,i=n.tool_call_chunks,a=n.response_metadata?.finish_reason===`stop`,o=this.getMessageInProgress(this.activeRun.id);if(i?.length>0){let t=i[0];t.name?(o?.id&&!o?.toolCallId&&(this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_END,messageId:o.id}),this.messagesInProcess[this.activeRun.id]=null),this.dispatchEvent({type:e.EventType.TOOL_CALL_START,toolCallId:t.id||n.id,toolCallName:t.name,parentMessageId:n.id}),this.setMessageInProgress(this.activeRun.id,{id:n.id,toolCallId:t.id||n.id,toolCallName:t.name}),this.activeRun.hasFunctionStreaming=!0):t.args&&o?.toolCallId&&this.dispatchEvent({type:e.EventType.TOOL_CALL_ARGS,toolCallId:o.toolCallId,delta:t.args});return}if(a){o?.toolCallId?this.dispatchEvent({type:e.EventType.TOOL_CALL_END,toolCallId:o.toolCallId}):o?.id&&this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_END,messageId:o.id}),this.messagesInProcess[this.activeRun.id]=null;return}!r&&!i?.length||r&&(o||(this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_START,role:`assistant`,messageId:n.id}),this.setMessageInProgress(this.activeRun.id,{id:n.id,toolCallId:null,toolCallName:null})),this.dispatchEvent({type:e.EventType.TEXT_MESSAGE_CONTENT,messageId:(this.getMessageInProgress(this.activeRun.id)??{id:n.id}).id,delta:r}))}abortRun(){this.cancelRequested=!0;let e=this.activeRun?.threadId,t=this.activeRun?.id;e&&t&&!this.cancelSent&&this.client.runs.cancel(e,t).then(()=>{this.cancelSent=!0}).catch(()=>{}),super.abortRun()}handleReasoningEvent(t){if(!t||!t.type||!t.text)return;let n=t.index;if(this.reasoningProcess?.index&&this.reasoningProcess.index!==n&&(this.reasoningProcess.type&&this.dispatchEvent({type:e.EventType.REASONING_MESSAGE_END,messageId:this.reasoningProcess.messageId}),this.dispatchEvent({type:e.EventType.REASONING_END,messageId:this.reasoningProcess.messageId}),this.reasoningProcess=null),!this.reasoningProcess){let t=(0,e.randomUUID)();this.dispatchEvent({type:e.EventType.REASONING_START,messageId:t}),this.reasoningProcess={index:n,messageId:t}}this.reasoningProcess.type!==t.type&&(this.dispatchEvent({type:e.EventType.REASONING_MESSAGE_START,messageId:this.reasoningProcess.messageId,role:`reasoning`}),this.reasoningProcess.type=t.type),t.signature&&(this.reasoningProcess.signature=t.signature),this.reasoningProcess.type&&this.dispatchEvent({type:e.EventType.REASONING_MESSAGE_CONTENT,messageId:this.reasoningProcess.messageId,delta:t.text})}getStateSnapshot(e){let t=e.values,n=this.activeRun.schemaKeys;return n?.output&&(t=o(t,[...this.constantSchemaKeys,...n.output])),t}async getOrCreateThread(e,t){let n;try{try{n=await this.getThread(e)}catch{n=await this.createThread({threadId:e,metadata:t})}}catch(e){throw Error(`Failed to create thread: ${e.message}`)}return n}async getThread(e){return this.client.threads.get(e)}async createThread(e){return this.client.threads.create(e)}async mergeConfigs({configs:e,assistant:t,schemaKeys:n}){return e.reduce((e,t)=>{let r=e.configurable;t.configurable&&(r=n?.config?o(t?.configurable,[...this.constantSchemaKeys,...n?.config??[]]):t?.configurable);let i={...e,...t,configurable:r},a=e.recursion_limit==null&&t.recursion_limit===25,s=JSON.stringify(i)!==JSON.stringify(e),c=a&&JSON.stringify({...i,recursion_limit:null})===JSON.stringify({...e,recursion_limit:null});return s&&!c?{...e,...i}:e},t.config)}getMessageInProgress(e){return this.messagesInProcess[e]}setMessageInProgress(e,t){this.messagesInProcess={...this.messagesInProcess,[e]:{...this.messagesInProcess[e],...t}}}async getAssistant(){try{let e=await this.client.assistants.search(),t=e.find(e=>e.graph_id===this.graphId);if(!t){let t=`
+      No agent found with graph ID ${this.graphId} found..\n
 
-
-      These are the available agents: [${t.map(i=>`${i.graph_id} (ID: ${i.assistant_id})`).join(", ")}]
-      `;throw console.error(s),new Error(s)}return e}catch(t){let e=new Error(`Failed to retrieve assistant: ${t.message}`);throw this.dispatchEvent({type:o.EventType.RUN_ERROR,message:e.message}),this.subscriber.error(),e}}async getSchemaKeys(){var t,e,s,i;try{let r=await this.client.assistants.getSchemas(this.assistant.assistant_id),u=null,g=[];if("context_schema"in r&&((t=r.context_schema)!=null&&t.properties)&&(g=Object.keys(r.context_schema.properties)),(e=r.config_schema)!=null&&e.properties&&(u=Object.keys(r.config_schema.properties)),!((s=r.input_schema)!=null&&s.properties)||!((i=r.output_schema)!=null&&i.properties))return{config:[],input:null,output:null,context:g};let l=Object.keys(r.input_schema.properties),c=Object.keys(r.output_schema.properties);return{input:l&&l.length?[...l,...this.constantSchemaKeys]:null,output:c&&c.length?[...c,...this.constantSchemaKeys]:null,context:g,config:u}}catch(r){return{config:[],input:this.constantSchemaKeys,output:this.constantSchemaKeys,context:[]}}}langGraphDefaultMergeState(t,e,s){var l,c;e.length>0&&"role"in e[0]&&e[0].role==="system"&&(e=e.slice(1));let i=t.messages||[],r=new Set(i.map(d=>d.id)),u=e.filter(d=>!r.has(d.id)),g=[...(l=t.tools)!=null?l:[],...(c=s.tools)!=null?c:[]].reduce((d,y)=>{let _=y;return y.type||(_={type:"function",name:y.name,function:{name:y.name,description:y.description,parameters:y.parameters}}),d.find(I=>I.name===_.name||I.function.name===_.function.name)?d:[...d,_]},[]);return T(f({},t),{messages:u,tools:g,"ag-ui":{tools:g,context:s.context}})}handleNodeChange(t){var e,s;t==="__end__"&&(t=void 0),t!==((e=this.activeRun)==null?void 0:e.nodeName)&&((s=this.activeRun)!=null&&s.nodeName&&this.endStep(),t&&this.startStep(t)),this.activeRun.nodeName=t}startStep(t){this.dispatchEvent({type:o.EventType.STEP_STARTED,stepName:t})}endStep(){this.dispatchEvent({type:o.EventType.STEP_FINISHED,stepName:this.activeRun.nodeName})}async getCheckpointByMessage(t,e,s){var M,A;let i=s!=null&&s.checkpoint_id?{checkpoint:{checkpoint_id:s.checkpoint_id}}:void 0,u=[...await this.client.threads.getHistory(e,i)].reverse(),g=u.find(p=>{var R;return(R=p.values.messages)==null?void 0:R.some(m=>m.id===t)});if(!g)throw new Error("Message not found");let l=(M=g.values.messages)!=null?M:[],c=l.findIndex(p=>p.id===t);if(l.slice(c+1).length)return this.getCheckpointByMessage(t,e,g.parent_checkpoint);let y=u.indexOf(g),N=g.values,{messages:_}=N,I=et(N,["messages"]),x=(A=u[y-1])!=null?A:T(f({},g),{values:{}});return T(f({},x),{values:f(f({},x.values),I)})}};var it=class extends yt.HttpAgent{};0&&(module.exports={CustomEventNames,LangGraphAgent,LangGraphEventTypes,LangGraphHttpAgent});
+      These are the available agents: [${e.map(e=>`${e.graph_id} (ID: ${e.assistant_id})`).join(`, `)}]
+      `;throw console.error(t),Error(t)}return t}catch(t){let n=Error(`Failed to retrieve assistant: ${t.message}`);throw this.dispatchEvent({type:e.EventType.RUN_ERROR,message:n.message}),this.subscriber.error(),n}}async getSchemaKeys(){try{let e=await this.client.assistants.getSchemas(this.assistant.assistant_id),t=null,n=[];if(`context_schema`in e&&e.context_schema?.properties&&(n=Object.keys(e.context_schema.properties)),e.config_schema?.properties&&(t=Object.keys(e.config_schema.properties)),!e.input_schema?.properties||!e.output_schema?.properties)return{config:[],input:null,output:null,context:n};let r=Object.keys(e.input_schema.properties),i=Object.keys(e.output_schema.properties);return{input:r&&r.length?[...r,...this.constantSchemaKeys]:null,output:i&&i.length?[...i,...this.constantSchemaKeys]:null,context:n,config:t}}catch{return{config:[],input:this.constantSchemaKeys,output:this.constantSchemaKeys,context:[]}}}langGraphDefaultMergeState(e,t,n){t.length>0&&`role`in t[0]&&t[0].role===`system`&&(t=t.slice(1));let r=e.messages||[],i=new Set(r.map(e=>e.id)),a=t.filter(e=>!i.has(e.id)),o=[...e.tools??[],...n.tools??[]].reduce((e,t)=>{let n=t;return t.type||(n={type:`function`,name:t.name,function:{name:t.name,description:t.description,parameters:t.parameters}}),e.find(e=>e.name===n.name||e.function.name===n.function.name)?e:[...e,n]},[]);return{...e,messages:a,tools:o,"ag-ui":{tools:o,context:n.context},copilotkit:{...e.copilotkit,actions:o}}}handleNodeChange(e){e===`__end__`&&(e=void 0),e!==this.activeRun?.nodeName&&(this.activeRun?.nodeName&&this.endStep(),e&&this.startStep(e)),this.activeRun.nodeName=e}startStep(t){this.dispatchEvent({type:e.EventType.STEP_STARTED,stepName:t})}endStep(){this.dispatchEvent({type:e.EventType.STEP_FINISHED,stepName:this.activeRun.nodeName})}async getCheckpointByMessage(e,t,n){let r=n?.checkpoint_id?{checkpoint:{checkpoint_id:n.checkpoint_id}}:void 0,i=[...await this.client.threads.getHistory(t,r)].reverse(),a=i.find(t=>t.values.messages?.some(t=>t.id===e));if(!a)throw Error(`Message not found`);let o=a.values.messages??[],s=o.findIndex(t=>t.id===e);if(o.slice(s+1).length)return this.getCheckpointByMessage(e,t,a.parent_checkpoint);let c=i.indexOf(a),{messages:l,...u}=a.values,d=i[c-1]??{...a,values:{}};return{...d,values:{...d.values,...u}}}},_=class extends e.HttpAgent{};exports.CustomEventNames=i,exports.LangGraphAgent=g,exports.LangGraphEventTypes=r,exports.LangGraphHttpAgent=_;
 //# sourceMappingURL=index.js.map