@ag-ui/langgraph 0.0.36 → 0.0.38

package/dist/index.d.mts

@@ -4,7 +4,7 @@ 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_core_tools0 from "@langchain/core/tools";
 import { MessageType } from "@langchain/core/messages";
-import { A2UI_OPERATIONS_KEY, BASIC_CATALOG_ID } from "@ag-ui/a2ui-toolkit";
+import { A2UIAttemptRecord, A2UIRecoveryConfig, A2UIValidationCatalog, A2UI_OPERATIONS_KEY, BASIC_CATALOG_ID } from "@ag-ui/a2ui-toolkit";
 import * as _langchain_langgraph_sdk_dist_types_stream0 from "@langchain/langgraph-sdk/dist/types.stream";
 import { RunAgentInput as RunAgentInput$1 } from "@ag-ui/core";
 import { RunsStreamPayload } from "@langchain/langgraph-sdk/dist/types";
@@ -39,6 +39,7 @@ interface StateEnrichment {
   'ag-ui': {
     tools: LangGraphToolWithName[];
     context: RunAgentInput$1['context'];
+    inject_a2ui_tool?: boolean | string;
   };
 }
 type SchemaKeys = {
@@ -115,6 +116,7 @@ type RunAgentExtendedInput<TStreamMode extends StreamMode | StreamMode[] = Strea
   forwardedProps?: Omit<RunsStreamPayload<TStreamMode, TSubgraphs>, "input"> & {
     nodeName?: string;
     threadMetadata?: Record<string, any>;
+    injectA2UITool?: boolean | string;
   };
 };
 interface RegenerateInput extends RunAgentExtendedInput {
@@ -255,6 +257,18 @@ interface A2UISubagentToolOptions {
   toolName?: string;
   /** Description shown to the main agent's planner. */
   toolDescription?: string;
+  /**
+   * Inline catalog (component name → JSON Schema with `required`) enabling
+   * catalog-aware recovery (unknown-component / missing-required-prop). Pass the
+   * SAME catalog the host gives `@ag-ui/a2ui-middleware` so the retry decision
+   * (here) and the paint gate (middleware) agree. Omit for structural-only
+   * recovery (malformed JSON, missing root, broken refs/bindings).
+   */
+  catalog?: A2UIValidationCatalog;
+  /** Recovery loop config: attempt cap, retry-UI threshold, debug exposure. */
+  recovery?: A2UIRecoveryConfig;
+  /** Per-attempt hook for emitting recovery status / dev logs (non-disruptive). */
+  onA2UIAttempt?: (record: A2UIAttemptRecord) => void;
 }
 /** Tool arguments exposed to the main agent's planner. */
 interface GenerateA2UIArgs {

package/dist/index.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.mts","names":[],"sources":["../src/types.ts","../src/agent.ts","../src/a2ui-tool.ts","../src/index.ts"],"mappings":";;;;;;;;;;;;aAIY,mBAAA;EACV,YAAA;EACA,aAAA;EACA,UAAA;EACA,gBAAA;EACA,iBAAA;EACA,cAAA;EACA,WAAA;EACA,SAAA;EACA,WAAA;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;EAKA,iBAAA;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;;;KC/DU,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;ED1G5C;;;;;;;;;;ECqHA,MAAA,GAAS,MAAA;EACT,aAAA;EACA,eAAA;EACA,eAAA,GAAkB,MAAA;EAClB,eAAA,GAAkB,MAAA;EAClB,SAAA;EACA,OAAA;ED7GA;;;;;;;;AASF;;;ECgHE,aAAA,SAAsB,MAAA;AAAA;AAAA,cAKX,cAAA,SAAuB,aAAA;EAClC,MAAA,EAAQ,MAAA;EACR,eAAA,GAAkB,MAAA;EAClB,SAAA;EACA,OAAA;EDzHgB;;EC4HT,OAAA,EAAS,MAAA;EAChB,SAAA,GAAY,SAAA;EACZ,iBAAA,EAAmB,wBAAA;EACnB,uBAAA,EAAyB,GAAA;EACzB,gBAAA,SAAyB,mBAAA;EACzB,SAAA,GAAY,WAAA;EAAA,QAEJ,SAAA;EAAA,QACA,eAAA;EAAA,QAEA,eAAA;EAAA,QACA,UAAA;EAAA,QAMA,kBAAA;EAER,UAAA,EAAY,UAAA,CAAW,eAAA;EACvB,kBAAA;EACA,MAAA,EAAQ,oBAAA;cAEI,MAAA,EAAQ,oBAAA;EA4Cb,KAAA,CAAA;EAgDP,aAAA,CAAc,KAAA,EAAO,eAAA;EAKrB,GAAA,CAAI,KAAA,EAAO,aAAA,GAAa,UAAA,CAAA,eAAA;EAYlB,cAAA,CACJ,KAAA,EAAO,qBAAA,EACP,UAAA,EAAY,UAAA,CAAW,eAAA,IAAgB,OAAA;EA2CnC,uBAAA,CACJ,KAAA,EAAO,eAAA,EACP,UAAA,EAAY,UAAA,GAAa,UAAA,KAAY,OAAA;;WAyDJ,WAAA,CAAY,KAAA;;;EAKzC,aAAA,CACJ,KAAA,EAAO,qBAAA,EACP,UAAA,EAAY,UAAA,GAAa,UAAA,KAAY,OAAA;;WAPJ,WAAA,CAAY,KAAA;;;;WA8RrB,WAAA,CAAY,KAAA;EAAA;EAIhC,kBAAA,CACJ,MAAA,EAAQ,OAAA,CACJ,UAAA,aAAuB,aAAA,IACvB,UAAA,aAAuB,uBAAA,IAE3B,QAAA,UACA,UAAA,EAAY,UAAA,CAAW,eAAA,GACvB,KAAA,EAAO,qBAAA,EACP,WAAA,EAAa,UAAA,GAAa,UAAA,KAAY,OAAA;EAAA,QAqU1B,4BAAA;EAed,iBAAA,CAAkB,KAAA;EDt9BZ;;AAGR;;;;;;;;EAHQ,QCg2CE,wBAAA;EAgGD,QAAA,CAAA;EAiBP,oBAAA,CAAqB,aAAA,EAAe,kBAAA;EA4DpC,gBAAA,CAAiB,WAAA,EAAa,WAAA,CAAY,KAAA,IAAM,KAAA;EAc1C,iBAAA,CACJ,QAAA,UACA,cAAA,GAAiB,MAAA,gBAChB,OAAA,CAAQ,MAAA;EAkBL,SAAA,CAAU,QAAA,WAAgB,OAAA,CAAA,MAAA,CAlBtB,yBAAA,CAkBsB,aAAA;EAI1B,YAAA,CACJ,OAAA,GAAU,UAAA,aAAuB,MAAA,CAAO,OAAA,CAAQ,MAAA,OAAU,OAAA,CAAA,MAAA,CAAtC,yBAAA,CAAsC,aAAA;EAKtD,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;EA4CD,oBAAA,CAAqB,KAAA,WAAa,iBAAA;EAIlC,oBAAA,CAAqB,KAAA,UAAe,IAAA,EAAM,iBAAA;EAUpC,YAAA,CAAA,GAAgB,OAAA,CAAQ,SAAA;EAiCxB,aAAA,CAAA,GAAiB,OAAA,CAAQ,UAAA;EAoD/B,0BAAA,CACE,KAAA,EAAO,KAAA,EACP,QAAA,EAAU,OAAA,IACV,KAAA,EAAO,qBAAA,GACN,KAAA,CAAM,eAAA;EAkET,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;;;;;;;;;;;KC7zDD,iBAAA;AAAA,UAMK,uBAAA;EFhCgB;EEkC/B,gBAAA;EFhCA;EEkCA,gBAAA;EFhCE;;EEmCF,gBAAA;EFjCY;EEmCZ,QAAA;EF/BU;EEiCV,eAAA;AAAA;;UAIQ,gBAAA;EFpCoB;;;;EEyC5B,MAAA;EF1CgB;;;;EE+ChB,iBAAA;EF9C0C;EEgD1C,OAAA;AAAA;;AF9CF;;;;;;;;iBE0DgB,YAAA,CACd,KAAA,EAAO,iBAAA,EACP,OAAA,GAAS,uBAAA,GAA4B,sBAAA,CAAA,qBAAA,MAAA,gBAAA;;;cCjF1B,kBAAA,SAA2B,SAAA"}
+{"version":3,"file":"index.d.mts","names":[],"sources":["../src/types.ts","../src/agent.ts","../src/a2ui-tool.ts","../src/index.ts"],"mappings":";;;;;;;;;;;;aAIY,mBAAA;EACV,YAAA;EACA,aAAA;EACA,UAAA;EACA,gBAAA;EACA,iBAAA;EACA,cAAA;EACA,WAAA;EACA,SAAA;EACA,WAAA;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;IAGT,gBAAA;EAAA;AAAA;AAAA,KAIQ,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;EAKA,iBAAA;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;;;KClEU,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;IAGjB,cAAA;EAAA;AAAA;AAAA,UAIM,eAAA,SAAwB,qBAAA;EAChC,iBAAA,EAAmB,OAAA;AAAA;AAAA,UAGJ,oBAAA,SAA6B,WAAA;ED5G5C;;;;;;;;;;ECuHA,MAAA,GAAS,MAAA;EACT,aAAA;EACA,eAAA;EACA,eAAA,GAAkB,MAAA;EAClB,eAAA,GAAkB,MAAA;EAClB,SAAA;EACA,OAAA;ED/GA;;;;;;;AAQF;;;;ECmHE,aAAA,SAAsB,MAAA;AAAA;AAAA,cAKX,cAAA,SAAuB,aAAA;EAClC,MAAA,EAAQ,MAAA;EACR,eAAA,GAAkB,MAAA;EAClB,SAAA;EACA,OAAA;ED5HgC;;EC+HzB,OAAA,EAAS,MAAA;EAChB,SAAA,GAAY,SAAA;EACZ,iBAAA,EAAmB,wBAAA;EACnB,uBAAA,EAAyB,GAAA;EACzB,gBAAA,SAAyB,mBAAA;EACzB,SAAA,GAAY,WAAA;EAAA,QAEJ,SAAA;EAAA,QACA,eAAA;EAAA,QAEA,eAAA;EAAA,QACA,UAAA;EAAA,QAMA,kBAAA;EAER,UAAA,EAAY,UAAA,CAAW,eAAA;EACvB,kBAAA;EACA,MAAA,EAAQ,oBAAA;cAEI,MAAA,EAAQ,oBAAA;EA4Cb,KAAA,CAAA;EAgDP,aAAA,CAAc,KAAA,EAAO,eAAA;EAKrB,GAAA,CAAI,KAAA,EAAO,aAAA,GAAa,UAAA,CAAA,eAAA;EAYlB,cAAA,CACJ,KAAA,EAAO,qBAAA,EACP,UAAA,EAAY,UAAA,CAAW,eAAA,IAAgB,OAAA;EA2CnC,uBAAA,CACJ,KAAA,EAAO,eAAA,EACP,UAAA,EAAY,UAAA,GAAa,UAAA,KAAY,OAAA;;WAyDJ,WAAA,CAAY,KAAA;;;EAKzC,aAAA,CACJ,KAAA,EAAO,qBAAA,EACP,UAAA,EAAY,UAAA,GAAa,UAAA,KAAY,OAAA;;WAPJ,WAAA,CAAY,KAAA;;;;WA8RrB,WAAA,CAAY,KAAA;EAAA;EAIhC,kBAAA,CACJ,MAAA,EAAQ,OAAA,CACJ,UAAA,aAAuB,aAAA,IACvB,UAAA,aAAuB,uBAAA,IAE3B,QAAA,UACA,UAAA,EAAY,UAAA,CAAW,eAAA,GACvB,KAAA,EAAO,qBAAA,EACP,WAAA,EAAa,UAAA,GAAa,UAAA,KAAY,OAAA;EAAA,QAqU1B,4BAAA;EAed,iBAAA,CAAkB,KAAA;EDt9BZ;;AAGR;;;;;;;;EAHQ,QCg2CE,wBAAA;EAgGD,QAAA,CAAA;EAiBP,oBAAA,CAAqB,aAAA,EAAe,kBAAA;EA4DpC,gBAAA,CAAiB,WAAA,EAAa,WAAA,CAAY,KAAA,IAAM,KAAA;EAc1C,iBAAA,CACJ,QAAA,UACA,cAAA,GAAiB,MAAA,gBAChB,OAAA,CAAQ,MAAA;EAkBL,SAAA,CAAU,QAAA,WAAgB,OAAA,CAAA,MAAA,CAlBtB,yBAAA,CAkBsB,aAAA;EAI1B,YAAA,CACJ,OAAA,GAAU,UAAA,aAAuB,MAAA,CAAO,OAAA,CAAQ,MAAA,OAAU,OAAA,CAAA,MAAA,CAAtC,yBAAA,CAAsC,aAAA;EAKtD,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;EA4CD,oBAAA,CAAqB,KAAA,WAAa,iBAAA;EAIlC,oBAAA,CAAqB,KAAA,UAAe,IAAA,EAAM,iBAAA;EAUpC,YAAA,CAAA,GAAgB,OAAA,CAAQ,SAAA;EAiCxB,aAAA,CAAA,GAAiB,OAAA,CAAQ,UAAA;EAoD/B,0BAAA,CACE,KAAA,EAAO,KAAA,EACP,QAAA,EAAU,OAAA,IACV,KAAA,EAAO,qBAAA,GACN,KAAA,CAAM,eAAA;EA4ET,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;;;;;;;;;;;KCt0DD,iBAAA;AAAA,UAMK,uBAAA;EFpCgB;EEsC/B,gBAAA;EFpCA;EEsCA,gBAAA;EFpCE;;EEuCF,gBAAA;EFrCY;EEuCZ,QAAA;EFnCU;EEqCV,eAAA;EFrCe;;;;;;;EE6Cf,OAAA,GAAU,qBAAA;EF7CM;EE+ChB,QAAA,GAAW,kBAAA;EF9CV;EEgDD,aAAA,IAAiB,MAAA,EAAQ,iBAAA;AAAA;;UAIjB,gBAAA;EFnDA;;AACV;;EEuDE,MAAA;EFtDU;;;;EE2DV,iBAAA;EFvDwB;EEyDxB,OAAA;AAAA;;;;;;;;;;iBAYc,YAAA,CACd,KAAA,EAAO,iBAAA,EACP,OAAA,GAAS,uBAAA,GAA4B,sBAAA,CAAA,qBAAA,MAAA,gBAAA;;;cCjG1B,kBAAA,SAA2B,SAAA"}

package/dist/index.d.ts

@@ -6,8 +6,8 @@ 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";
-import * as _langchain_core_tools1 from "@langchain/core/tools";
-import { A2UI_OPERATIONS_KEY, BASIC_CATALOG_ID } from "@ag-ui/a2ui-toolkit";
+import * as _langchain_core_tools0 from "@langchain/core/tools";
+import { A2UIAttemptRecord, A2UIRecoveryConfig, A2UIValidationCatalog, A2UI_OPERATIONS_KEY, BASIC_CATALOG_ID } from "@ag-ui/a2ui-toolkit";
 
 //#region src/types.d.ts
 declare enum LangGraphEventTypes {
@@ -39,6 +39,7 @@ interface StateEnrichment {
   'ag-ui': {
     tools: LangGraphToolWithName[];
     context: RunAgentInput$1['context'];
+    inject_a2ui_tool?: boolean | string;
   };
 }
 type SchemaKeys = {
@@ -115,6 +116,7 @@ type RunAgentExtendedInput<TStreamMode extends StreamMode | StreamMode[] = Strea
   forwardedProps?: Omit<RunsStreamPayload<TStreamMode, TSubgraphs>, "input"> & {
     nodeName?: string;
     threadMetadata?: Record<string, any>;
+    injectA2UITool?: boolean | string;
   };
 };
 interface RegenerateInput extends RunAgentExtendedInput {
@@ -255,6 +257,18 @@ interface A2UISubagentToolOptions {
   toolName?: string;
   /** Description shown to the main agent's planner. */
   toolDescription?: string;
+  /**
+   * Inline catalog (component name → JSON Schema with `required`) enabling
+   * catalog-aware recovery (unknown-component / missing-required-prop). Pass the
+   * SAME catalog the host gives `@ag-ui/a2ui-middleware` so the retry decision
+   * (here) and the paint gate (middleware) agree. Omit for structural-only
+   * recovery (malformed JSON, missing root, broken refs/bindings).
+   */
+  catalog?: A2UIValidationCatalog;
+  /** Recovery loop config: attempt cap, retry-UI threshold, debug exposure. */
+  recovery?: A2UIRecoveryConfig;
+  /** Per-attempt hook for emitting recovery status / dev logs (non-disruptive). */
+  onA2UIAttempt?: (record: A2UIAttemptRecord) => void;
 }
 /** Tool arguments exposed to the main agent's planner. */
 interface GenerateA2UIArgs {
@@ -280,7 +294,7 @@ interface GenerateA2UIArgs {
  *   Using the same provider/model as the main agent is fine.
  * @param options Optional behavior overrides.
  */
-declare function getA2UITools(model: A2UISubagentModel, options?: A2UISubagentToolOptions): _langchain_core_tools1.DynamicStructuredTool<any, GenerateA2UIArgs, unknown, string, unknown, string>;
+declare function getA2UITools(model: A2UISubagentModel, options?: A2UISubagentToolOptions): _langchain_core_tools0.DynamicStructuredTool<any, GenerateA2UIArgs, unknown, string, unknown, string>;
 //#endregion
 //#region src/index.d.ts
 declare class LangGraphHttpAgent extends HttpAgent {}

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","names":[],"sources":["../src/types.ts","../src/agent.ts","../src/a2ui-tool.ts","../src/index.ts"],"mappings":";;;;;;;;;;;;aAIY,mBAAA;EACV,YAAA;EACA,aAAA;EACA,UAAA;EACA,gBAAA;EACA,iBAAA;EACA,cAAA;EACA,WAAA;EACA,SAAA;EACA,WAAA;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;EAKA,iBAAA;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;;;KC/DU,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;ED1G5C;;;;;;;;;;ECqHA,MAAA,GAAS,MAAA;EACT,aAAA;EACA,eAAA;EACA,eAAA,GAAkB,MAAA;EAClB,eAAA,GAAkB,MAAA;EAClB,SAAA;EACA,OAAA;ED7GA;;;;;;;;AASF;;;ECgHE,aAAA,SAAsB,MAAA;AAAA;AAAA,cAKX,cAAA,SAAuB,aAAA;EAClC,MAAA,EAAQ,MAAA;EACR,eAAA,GAAkB,MAAA;EAClB,SAAA;EACA,OAAA;EDzHgB;;EC4HT,OAAA,EAAS,MAAA;EAChB,SAAA,GAAY,SAAA;EACZ,iBAAA,EAAmB,wBAAA;EACnB,uBAAA,EAAyB,GAAA;EACzB,gBAAA,SAAyB,mBAAA;EACzB,SAAA,GAAY,WAAA;EAAA,QAEJ,SAAA;EAAA,QACA,eAAA;EAAA,QAEA,eAAA;EAAA,QACA,UAAA;EAAA,QAMA,kBAAA;EAER,UAAA,EAAY,UAAA,CAAW,eAAA;EACvB,kBAAA;EACA,MAAA,EAAQ,oBAAA;cAEI,MAAA,EAAQ,oBAAA;EA4Cb,KAAA,CAAA;EAgDP,aAAA,CAAc,KAAA,EAAO,eAAA;EAKrB,GAAA,CAAI,KAAA,EAAO,aAAA,GAAa,UAAA,CAAA,eAAA;EAYlB,cAAA,CACJ,KAAA,EAAO,qBAAA,EACP,UAAA,EAAY,UAAA,CAAW,eAAA,IAAgB,OAAA;EA2CnC,uBAAA,CACJ,KAAA,EAAO,eAAA,EACP,UAAA,EAAY,UAAA,GAAa,UAAA,KAAY,OAAA;;WAyDJ,WAAA,CAAY,KAAA;;;EAKzC,aAAA,CACJ,KAAA,EAAO,qBAAA,EACP,UAAA,EAAY,UAAA,GAAa,UAAA,KAAY,OAAA;;WAPJ,WAAA,CAAY,KAAA;;;;WA8RrB,WAAA,CAAY,KAAA;EAAA;EAIhC,kBAAA,CACJ,MAAA,EAAQ,OAAA,CACJ,UAAA,aAAuB,aAAA,IACvB,UAAA,aAAuB,uBAAA,IAE3B,QAAA,UACA,UAAA,EAAY,UAAA,CAAW,eAAA,GACvB,KAAA,EAAO,qBAAA,EACP,WAAA,EAAa,UAAA,GAAa,UAAA,KAAY,OAAA;EAAA,QAqU1B,4BAAA;EAed,iBAAA,CAAkB,KAAA;EDt9BZ;;AAGR;;;;;;;;EAHQ,QCg2CE,wBAAA;EAgGD,QAAA,CAAA;EAiBP,oBAAA,CAAqB,aAAA,EAAe,kBAAA;EA4DpC,gBAAA,CAAiB,WAAA,EAAa,WAAA,CAAY,KAAA,IAAM,KAAA;EAc1C,iBAAA,CACJ,QAAA,UACA,cAAA,GAAiB,MAAA,gBAChB,OAAA,CAAQ,MAAA;EAkBL,SAAA,CAAU,QAAA,WAAgB,OAAA,CAAA,MAAA,CAlBtB,yBAAA,CAkBsB,aAAA;EAI1B,YAAA,CACJ,OAAA,GAAU,UAAA,aAAuB,MAAA,CAAO,OAAA,CAAQ,MAAA,OAAU,OAAA,CAAA,MAAA,CAAtC,yBAAA,CAAsC,aAAA;EAKtD,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;EA4CD,oBAAA,CAAqB,KAAA,WAAa,iBAAA;EAIlC,oBAAA,CAAqB,KAAA,UAAe,IAAA,EAAM,iBAAA;EAUpC,YAAA,CAAA,GAAgB,OAAA,CAAQ,SAAA;EAiCxB,aAAA,CAAA,GAAiB,OAAA,CAAQ,UAAA;EAoD/B,0BAAA,CACE,KAAA,EAAO,KAAA,EACP,QAAA,EAAU,OAAA,IACV,KAAA,EAAO,qBAAA,GACN,KAAA,CAAM,eAAA;EAkET,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;;;;;;;;;;;KC7zDD,iBAAA;AAAA,UAMK,uBAAA;EFhCgB;EEkC/B,gBAAA;EFhCA;EEkCA,gBAAA;EFhCE;;EEmCF,gBAAA;EFjCY;EEmCZ,QAAA;EF/BU;EEiCV,eAAA;AAAA;;UAIQ,gBAAA;EFpCoB;;;;EEyC5B,MAAA;EF1CgB;;;;EE+ChB,iBAAA;EF9C0C;EEgD1C,OAAA;AAAA;;AF9CF;;;;;;;;iBE0DgB,YAAA,CACd,KAAA,EAAO,iBAAA,EACP,OAAA,GAAS,uBAAA,GAA4B,sBAAA,CAAA,qBAAA,MAAA,gBAAA;;;cCjF1B,kBAAA,SAA2B,SAAA"}
+{"version":3,"file":"index.d.ts","names":[],"sources":["../src/types.ts","../src/agent.ts","../src/a2ui-tool.ts","../src/index.ts"],"mappings":";;;;;;;;;;;;aAIY,mBAAA;EACV,YAAA;EACA,aAAA;EACA,UAAA;EACA,gBAAA;EACA,iBAAA;EACA,cAAA;EACA,WAAA;EACA,SAAA;EACA,WAAA;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;IAGT,gBAAA;EAAA;AAAA;AAAA,KAIQ,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;EAKA,iBAAA;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;;;KClEU,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;IAGjB,cAAA;EAAA;AAAA;AAAA,UAIM,eAAA,SAAwB,qBAAA;EAChC,iBAAA,EAAmB,OAAA;AAAA;AAAA,UAGJ,oBAAA,SAA6B,WAAA;ED5G5C;;;;;;;;;;ECuHA,MAAA,GAAS,MAAA;EACT,aAAA;EACA,eAAA;EACA,eAAA,GAAkB,MAAA;EAClB,eAAA,GAAkB,MAAA;EAClB,SAAA;EACA,OAAA;ED/GA;;;;;;;AAQF;;;;ECmHE,aAAA,SAAsB,MAAA;AAAA;AAAA,cAKX,cAAA,SAAuB,aAAA;EAClC,MAAA,EAAQ,MAAA;EACR,eAAA,GAAkB,MAAA;EAClB,SAAA;EACA,OAAA;ED5HgC;;EC+HzB,OAAA,EAAS,MAAA;EAChB,SAAA,GAAY,SAAA;EACZ,iBAAA,EAAmB,wBAAA;EACnB,uBAAA,EAAyB,GAAA;EACzB,gBAAA,SAAyB,mBAAA;EACzB,SAAA,GAAY,WAAA;EAAA,QAEJ,SAAA;EAAA,QACA,eAAA;EAAA,QAEA,eAAA;EAAA,QACA,UAAA;EAAA,QAMA,kBAAA;EAER,UAAA,EAAY,UAAA,CAAW,eAAA;EACvB,kBAAA;EACA,MAAA,EAAQ,oBAAA;cAEI,MAAA,EAAQ,oBAAA;EA4Cb,KAAA,CAAA;EAgDP,aAAA,CAAc,KAAA,EAAO,eAAA;EAKrB,GAAA,CAAI,KAAA,EAAO,aAAA,GAAa,UAAA,CAAA,eAAA;EAYlB,cAAA,CACJ,KAAA,EAAO,qBAAA,EACP,UAAA,EAAY,UAAA,CAAW,eAAA,IAAgB,OAAA;EA2CnC,uBAAA,CACJ,KAAA,EAAO,eAAA,EACP,UAAA,EAAY,UAAA,GAAa,UAAA,KAAY,OAAA;;WAyDJ,WAAA,CAAY,KAAA;;;EAKzC,aAAA,CACJ,KAAA,EAAO,qBAAA,EACP,UAAA,EAAY,UAAA,GAAa,UAAA,KAAY,OAAA;;WAPJ,WAAA,CAAY,KAAA;;;;WA8RrB,WAAA,CAAY,KAAA;EAAA;EAIhC,kBAAA,CACJ,MAAA,EAAQ,OAAA,CACJ,UAAA,aAAuB,aAAA,IACvB,UAAA,aAAuB,uBAAA,IAE3B,QAAA,UACA,UAAA,EAAY,UAAA,CAAW,eAAA,GACvB,KAAA,EAAO,qBAAA,EACP,WAAA,EAAa,UAAA,GAAa,UAAA,KAAY,OAAA;EAAA,QAqU1B,4BAAA;EAed,iBAAA,CAAkB,KAAA;EDt9BZ;;AAGR;;;;;;;;EAHQ,QCg2CE,wBAAA;EAgGD,QAAA,CAAA;EAiBP,oBAAA,CAAqB,aAAA,EAAe,kBAAA;EA4DpC,gBAAA,CAAiB,WAAA,EAAa,WAAA,CAAY,KAAA,IAAM,KAAA;EAc1C,iBAAA,CACJ,QAAA,UACA,cAAA,GAAiB,MAAA,gBAChB,OAAA,CAAQ,MAAA;EAkBL,SAAA,CAAU,QAAA,WAAgB,OAAA,CAAA,MAAA,CAlBtB,yBAAA,CAkBsB,aAAA;EAI1B,YAAA,CACJ,OAAA,GAAU,UAAA,aAAuB,MAAA,CAAO,OAAA,CAAQ,MAAA,OAAU,OAAA,CAAA,MAAA,CAAtC,yBAAA,CAAsC,aAAA;EAKtD,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;EA4CD,oBAAA,CAAqB,KAAA,WAAa,iBAAA;EAIlC,oBAAA,CAAqB,KAAA,UAAe,IAAA,EAAM,iBAAA;EAUpC,YAAA,CAAA,GAAgB,OAAA,CAAQ,SAAA;EAiCxB,aAAA,CAAA,GAAiB,OAAA,CAAQ,UAAA;EAoD/B,0BAAA,CACE,KAAA,EAAO,KAAA,EACP,QAAA,EAAU,OAAA,IACV,KAAA,EAAO,qBAAA,GACN,KAAA,CAAM,eAAA;EA4ET,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;;;;;;;;;;;KCt0DD,iBAAA;AAAA,UAMK,uBAAA;EFpCgB;EEsC/B,gBAAA;EFpCA;EEsCA,gBAAA;EFpCE;;EEuCF,gBAAA;EFrCY;EEuCZ,QAAA;EFnCU;EEqCV,eAAA;EFrCe;;;;;;;EE6Cf,OAAA,GAAU,qBAAA;EF7CM;EE+ChB,QAAA,GAAW,kBAAA;EF9CV;EEgDD,aAAA,IAAiB,MAAA,EAAQ,iBAAA;AAAA;;UAIjB,gBAAA;EFnDA;;AACV;;EEuDE,MAAA;EFtDU;;;;EE2DV,iBAAA;EFvDwB;EEyDxB,OAAA;AAAA;;;;;;;;;;iBAYc,YAAA,CACd,KAAA,EAAO,iBAAA,EACP,OAAA,GAAS,uBAAA,GAA4B,sBAAA,CAAA,qBAAA,MAAA,gBAAA;;;cCjG1B,kBAAA,SAA2B,SAAA"}

package/dist/index.js

@@ -2,5 +2,5 @@ let e=require(`@ag-ui/client`),t=require(`rxjs`),n=require(`@langchain/langgraph
       No agent found with graph ID ${this.graphId} found..\n
 
       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=[...n.tools??[],...e.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}}}};function C(e,t={}){let{compositionGuide:n,defaultSurfaceId:o,defaultCatalogId:s,toolName:c,toolDescription:l}=t,u=o||a.DEFAULT_SURFACE_ID,d=s||a.BASIC_CATALOG_ID;return(0,r.tool)(async(t,r)=>{let o=r.state,s=(o.messages??[]).slice(0,-1),c=(0,a.prepareA2UIRequest)({intent:t.intent,targetSurfaceId:t.target_surface_id,changes:t.changes,messages:s,state:o,compositionGuide:n});if(c.error)return(0,a.wrapErrorEnvelope)(c.error);if(!e.bindTools)return(0,a.wrapErrorEnvelope)(`Provided model does not support bindTools`);let l=(await e.bindTools([a.RENDER_A2UI_TOOL_DEF],{tool_choice:{type:`function`,function:{name:`render_a2ui`}}}).invoke([new i.SystemMessage(c.prompt),...s])).tool_calls??[];return l.length===0?(0,a.wrapErrorEnvelope)(`LLM did not call render_a2ui`):(0,a.buildA2UIEnvelope)({args:l[0].args??{},isUpdate:c.isUpdate,targetSurfaceId:t.target_surface_id,prior:c.prior,defaultSurfaceId:u,defaultCatalogId:d})},{name:c||a.GENERATE_A2UI_TOOL_NAME,description:l||a.GENERATE_A2UI_TOOL_DESCRIPTION,schema:{type:`object`,properties:{intent:{type:`string`,enum:[`create`,`update`],description:a.GENERATE_A2UI_ARG_DESCRIPTIONS.intent},target_surface_id:{type:`string`,description:a.GENERATE_A2UI_ARG_DESCRIPTIONS.target_surface_id},changes:{type:`string`,description:a.GENERATE_A2UI_ARG_DESCRIPTIONS.changes}}}})}var w=class extends e.HttpAgent{};exports.A2UI_OPERATIONS_KEY=a.A2UI_OPERATIONS_KEY,exports.BASIC_CATALOG_ID=a.BASIC_CATALOG_ID,exports.CustomEventNames=s,exports.LangGraphAgent=S,exports.LangGraphEventTypes=o,exports.LangGraphHttpAgent=w,exports.getA2UITools=C;
+      `;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=[...n.tools??[],...e.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]},[]),s=n.forwardedProps?.injectA2UITool,c={tools:o,context:n.context};return s!==void 0&&(c.inject_a2ui_tool=s),{...e,messages:a,tools:o,"ag-ui":c,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}}}};function C(e,t={}){let{compositionGuide:n,defaultSurfaceId:o,defaultCatalogId:s,toolName:c,toolDescription:l,catalog:u,recovery:d,onA2UIAttempt:f}=t,p=o||a.DEFAULT_SURFACE_ID,m=s||a.BASIC_CATALOG_ID;return(0,r.tool)(async(t,r)=>{let o=r.state??{},s=(o.messages??[]).slice(0,-1),c=(0,a.prepareA2UIRequest)({intent:t.intent,targetSurfaceId:t.target_surface_id,changes:t.changes,messages:s,state:o,compositionGuide:n});if(c.error)return(0,a.wrapErrorEnvelope)(c.error);if(!e.bindTools)return(0,a.wrapErrorEnvelope)(`Provided model does not support bindTools`);let l=e.bindTools([a.RENDER_A2UI_TOOL_DEF],{tool_choice:{type:`function`,function:{name:`render_a2ui`}}}),{envelope:h}=await(0,a.runA2UIGenerationWithRecovery)({basePrompt:c.prompt,catalog:u,config:d,onAttempt:f,invokeSubagent:async e=>{let t=(await l.invoke([new i.SystemMessage(e),...s])).tool_calls??[];return t.length?t[0].args??{}:null},buildEnvelope:e=>(0,a.buildA2UIEnvelope)({args:e,isUpdate:c.isUpdate,targetSurfaceId:t.target_surface_id,prior:c.prior,defaultSurfaceId:p,defaultCatalogId:m})});return h},{name:c||a.GENERATE_A2UI_TOOL_NAME,description:l||a.GENERATE_A2UI_TOOL_DESCRIPTION,schema:{type:`object`,properties:{intent:{type:`string`,enum:[`create`,`update`],description:a.GENERATE_A2UI_ARG_DESCRIPTIONS.intent},target_surface_id:{type:`string`,description:a.GENERATE_A2UI_ARG_DESCRIPTIONS.target_surface_id},changes:{type:`string`,description:a.GENERATE_A2UI_ARG_DESCRIPTIONS.changes}}}})}var w=class extends e.HttpAgent{};exports.A2UI_OPERATIONS_KEY=a.A2UI_OPERATIONS_KEY,exports.BASIC_CATALOG_ID=a.BASIC_CATALOG_ID,exports.CustomEventNames=s,exports.LangGraphAgent=S,exports.LangGraphEventTypes=o,exports.LangGraphHttpAgent=w,exports.getA2UITools=C;
 //# sourceMappingURL=index.js.map