@copilotkit/runtime 1.50.0-beta.1 → 1.50.0-beta.3

package/dist/langgraph.mjs

@@ -0,0 +1,206 @@
+import { map } from 'rxjs';
+import { LangGraphAgent as LangGraphAgent$1 } from '@ag-ui/langgraph';
+export { LangGraphHttpAgent } from '@ag-ui/langgraph';
+import { EventType } from '@ag-ui/client';
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+
+// src/lib/runtime/agent-integrations/langgraph/consts.ts
+var CustomEventNames;
+(function(CustomEventNames3) {
+  CustomEventNames3["CopilotKitManuallyEmitMessage"] = "copilotkit_manually_emit_message";
+  CustomEventNames3["CopilotKitManuallyEmitToolCall"] = "copilotkit_manually_emit_tool_call";
+  CustomEventNames3["CopilotKitManuallyEmitIntermediateState"] = "copilotkit_manually_emit_intermediate_state";
+  CustomEventNames3["CopilotKitExit"] = "copilotkit_exit";
+})(CustomEventNames || (CustomEventNames = {}));
+
+// src/agents/langgraph/events.ts
+var LangGraphEventTypes;
+(function(LangGraphEventTypes2) {
+  LangGraphEventTypes2["OnChainStart"] = "on_chain_start";
+  LangGraphEventTypes2["OnChainStream"] = "on_chain_stream";
+  LangGraphEventTypes2["OnChainEnd"] = "on_chain_end";
+  LangGraphEventTypes2["OnChatModelStart"] = "on_chat_model_start";
+  LangGraphEventTypes2["OnChatModelStream"] = "on_chat_model_stream";
+  LangGraphEventTypes2["OnChatModelEnd"] = "on_chat_model_end";
+  LangGraphEventTypes2["OnToolStart"] = "on_tool_start";
+  LangGraphEventTypes2["OnToolEnd"] = "on_tool_end";
+  LangGraphEventTypes2["OnCopilotKitStateSync"] = "on_copilotkit_state_sync";
+  LangGraphEventTypes2["OnCopilotKitEmitMessage"] = "on_copilotkit_emit_message";
+  LangGraphEventTypes2["OnCopilotKitEmitToolCall"] = "on_copilotkit_emit_tool_call";
+  LangGraphEventTypes2["OnCustomEvent"] = "on_custom_event";
+  LangGraphEventTypes2["OnInterrupt"] = "on_interrupt";
+  LangGraphEventTypes2["OnCopilotKitInterrupt"] = "on_copilotkit_interrupt";
+  LangGraphEventTypes2["OnCopilotKitError"] = "on_copilotkit_error";
+})(LangGraphEventTypes || (LangGraphEventTypes = {}));
+var MetaEventNames;
+(function(MetaEventNames2) {
+  MetaEventNames2["LangGraphInterruptEvent"] = "LangGraphInterruptEvent";
+  MetaEventNames2["CopilotKitLangGraphInterruptEvent"] = "CopilotKitLangGraphInterruptEvent";
+})(MetaEventNames || (MetaEventNames = {}));
+var CustomEventNames2;
+(function(CustomEventNames3) {
+  CustomEventNames3["CopilotKitManuallyEmitMessage"] = "copilotkit_manually_emit_message";
+  CustomEventNames3["CopilotKitManuallyEmitToolCall"] = "copilotkit_manually_emit_tool_call";
+  CustomEventNames3["CopilotKitManuallyEmitIntermediateState"] = "copilotkit_manually_emit_intermediate_state";
+  CustomEventNames3["CopilotKitExit"] = "copilotkit_exit";
+})(CustomEventNames2 || (CustomEventNames2 = {}));
+var LangGraphAgent = class extends LangGraphAgent$1 {
+  constructor(config) {
+    super(config);
+  }
+  // @ts-ignore
+  clone() {
+    return new LangGraphAgent(this.config);
+  }
+  dispatchEvent(event) {
+    if (event.type === EventType.CUSTOM) {
+      const customEvent = event;
+      if (customEvent.name === CustomEventNames.CopilotKitManuallyEmitMessage) {
+        this.subscriber.next({
+          type: EventType.TEXT_MESSAGE_START,
+          role: "assistant",
+          messageId: customEvent.value.message_id,
+          rawEvent: event
+        });
+        this.subscriber.next({
+          type: EventType.TEXT_MESSAGE_CONTENT,
+          messageId: customEvent.value.message_id,
+          delta: customEvent.value.message,
+          rawEvent: event
+        });
+        this.subscriber.next({
+          type: EventType.TEXT_MESSAGE_END,
+          messageId: customEvent.value.message_id,
+          rawEvent: event
+        });
+        return true;
+      }
+      if (customEvent.name === CustomEventNames.CopilotKitManuallyEmitToolCall) {
+        this.subscriber.next({
+          type: EventType.TOOL_CALL_START,
+          toolCallId: customEvent.value.id,
+          toolCallName: customEvent.value.name,
+          parentMessageId: customEvent.value.id,
+          rawEvent: event
+        });
+        this.subscriber.next({
+          type: EventType.TOOL_CALL_ARGS,
+          toolCallId: customEvent.value.id,
+          delta: customEvent.value.args,
+          rawEvent: event
+        });
+        this.subscriber.next({
+          type: EventType.TOOL_CALL_END,
+          toolCallId: customEvent.value.id,
+          rawEvent: event
+        });
+        return true;
+      }
+      if (customEvent.name === CustomEventNames.CopilotKitManuallyEmitIntermediateState) {
+        this.activeRun.manuallyEmittedState = customEvent.value;
+        this.dispatchEvent({
+          type: EventType.STATE_SNAPSHOT,
+          snapshot: this.getStateSnapshot({
+            values: this.activeRun.manuallyEmittedState
+          }),
+          rawEvent: event
+        });
+        return true;
+      }
+      if (customEvent.name === CustomEventNames.CopilotKitExit) {
+        this.subscriber.next({
+          type: EventType.CUSTOM,
+          name: "Exit",
+          value: true
+        });
+        return true;
+      }
+    }
+    const rawEvent = event.rawEvent;
+    if (!rawEvent) {
+      this.subscriber.next(event);
+      return true;
+    }
+    const isMessageEvent = event.type === EventType.TEXT_MESSAGE_START || event.type === EventType.TEXT_MESSAGE_CONTENT || event.type === EventType.TEXT_MESSAGE_END;
+    const isToolEvent = event.type === EventType.TOOL_CALL_START || event.type === EventType.TOOL_CALL_ARGS || event.type === EventType.TOOL_CALL_END;
+    if ("copilotkit:emit-tool-calls" in (rawEvent.metadata || {})) {
+      if (rawEvent.metadata["copilotkit:emit-tool-calls"] === false && isToolEvent) {
+        return false;
+      }
+    }
+    if ("copilotkit:emit-messages" in (rawEvent.metadata || {})) {
+      if (rawEvent.metadata["copilotkit:emit-messages"] === false && isMessageEvent) {
+        return false;
+      }
+    }
+    this.subscriber.next(event);
+    return true;
+  }
+  // @ts-ignore
+  run(input) {
+    return super.run(input).pipe(map((processedEvent) => {
+      var _a, _b, _c, _d, _e;
+      if (processedEvent.type === EventType.RAW) {
+        const event = processedEvent.event ?? processedEvent.rawEvent;
+        const eventType = event.event;
+        const toolCallData = (_c = (_b = (_a = event.data) == null ? void 0 : _a.chunk) == null ? void 0 : _b.tool_call_chunks) == null ? void 0 : _c[0];
+        const toolCallUsedToPredictState = (_e = (_d = event.metadata) == null ? void 0 : _d["copilotkit:emit-intermediate-state"]) == null ? void 0 : _e.some((predictStateTool) => predictStateTool.tool === (toolCallData == null ? void 0 : toolCallData.name));
+        if (eventType === LangGraphEventTypes.OnChatModelStream && toolCallUsedToPredictState) {
+          return {
+            type: EventType.CUSTOM,
+            name: "PredictState",
+            value: event.metadata["copilotkit:emit-intermediate-state"]
+          };
+        }
+      }
+      return processedEvent;
+    }));
+  }
+  langGraphDefaultMergeState(state, messages, input) {
+    const aguiMergedState = super.langGraphDefaultMergeState(state, messages, input);
+    const { tools: returnedTools, "ag-ui": agui } = aguiMergedState;
+    const rawCombinedTools = [
+      ...returnedTools ?? [],
+      ...(agui == null ? void 0 : agui.tools) ?? []
+    ];
+    const combinedTools = Array.from(new Map(rawCombinedTools.map((t) => [
+      (t == null ? void 0 : t.id) ?? (t == null ? void 0 : t.name) ?? (t == null ? void 0 : t.key) ?? JSON.stringify(t),
+      t
+    ])).values());
+    return {
+      ...aguiMergedState,
+      copilotkit: {
+        actions: combinedTools,
+        context: (agui == null ? void 0 : agui.context) ?? []
+      }
+    };
+  }
+  async getSchemaKeys() {
+    const CONSTANT_KEYS = [
+      "copilotkit"
+    ];
+    const schemaKeys = await super.getSchemaKeys();
+    return {
+      config: schemaKeys.config,
+      input: schemaKeys.input ? [
+        ...schemaKeys.input,
+        ...CONSTANT_KEYS
+      ] : null,
+      output: schemaKeys.output ? [
+        ...schemaKeys.output,
+        ...CONSTANT_KEYS
+      ] : null,
+      context: schemaKeys.context ? [
+        ...schemaKeys.context,
+        ...CONSTANT_KEYS
+      ] : null
+    };
+  }
+};
+__name(LangGraphAgent, "LangGraphAgent");
+
+export { CustomEventNames, LangGraphAgent };
+//# sourceMappingURL=out.js.map
+//# sourceMappingURL=langgraph.mjs.map

package/dist/langgraph.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/lib/runtime/agent-integrations/langgraph/consts.ts","../src/lib/runtime/agent-integrations/langgraph/agent.ts","../src/agents/langgraph/events.ts"],"names":["CustomEventNames","map","LangGraphEventTypes","MetaEventNames","LangGraphAgent","AGUILangGraphAgent","LangGraphHttpAgent","EventType","constructor","config","clone","dispatchEvent","event","type","CUSTOM","customEvent","name","CopilotKitManuallyEmitMessage","subscriber","next","TEXT_MESSAGE_START","role","messageId","value","message_id","rawEvent","TEXT_MESSAGE_CONTENT","delta","message","TEXT_MESSAGE_END","CopilotKitManuallyEmitToolCall","TOOL_CALL_START","toolCallId","id","toolCallName","parentMessageId","TOOL_CALL_ARGS","args","TOOL_CALL_END","CopilotKitManuallyEmitIntermediateState","activeRun","manuallyEmittedState","STATE_SNAPSHOT","snapshot","getStateSnapshot","values","CopilotKitExit","isMessageEvent","isToolEvent","metadata","run","input","pipe","processedEvent","RAW","eventType","toolCallData","data","chunk","tool_call_chunks","toolCallUsedToPredictState","some","predictStateTool","tool","OnChatModelStream","langGraphDefaultMergeState","state","messages","aguiMergedState","tools","returnedTools","agui","rawCombinedTools","combinedTools","Array","from","Map","t","key","JSON","stringify","copilotkit","actions","context","getSchemaKeys","CONSTANT_KEYS","schemaKeys","output"],"mappings":";;;;AAIC,IAAA;UAkBWA,mBAAAA;;;;;GAAAA,qBAAAA,mBAAAA,CAAAA,EAAAA;;;ACtBZ,SAASC,WAAW;;;;UCERC,sBAAAA;;;;;;;;;;;;;;;;GAAAA,wBAAAA,sBAAAA,CAAAA,EAAAA;;UAkBAC,iBAAAA;;;GAAAA,mBAAAA,iBAAAA,CAAAA,EAAAA;;UAKAH,mBAAAA;;;;;GAAAA,sBAAAA,oBAAAA,CAAAA,EAAAA;;;ADtBZ,SACEI,kBAAkBC,oBAClBC,0BAMK;AAWP,SAAwBC,iBAA8B;AAM/C,IAAMH,iBAAN,cAA6BC,mBAAAA;EAClCG,YAAYC,QAA8B;AACxC,UAAMA,MAAAA;EACR;;EAGOC,QAAQ;AACb,WAAO,IAAIN,eAAe,KAAKK,MAAM;EACvC;EAEAE,cAAcC,OAAwB;AACpC,QAAIA,MAAMC,SAASN,UAAUO,QAAQ;AAEnC,YAAMC,cAAcH;AAEpB,UAAIG,YAAYC,SAAShB,iBAAiBiB,+BAA+B;AACvE,aAAKC,WAAWC,KAAK;UACnBN,MAAMN,UAAUa;UAChBC,MAAM;UACNC,WAAWP,YAAYQ,MAAMC;UAC7BC,UAAUb;QACZ,CAAA;AACA,aAAKM,WAAWC,KAAK;UACnBN,MAAMN,UAAUmB;UAChBJ,WAAWP,YAAYQ,MAAMC;UAC7BG,OAAOZ,YAAYQ,MAAMK;UACzBH,UAAUb;QACZ,CAAA;AACA,aAAKM,WAAWC,KAAK;UACnBN,MAAMN,UAAUsB;UAChBP,WAAWP,YAAYQ,MAAMC;UAC7BC,UAAUb;QACZ,CAAA;AACA,eAAO;MACT;AAEA,UAAIG,YAAYC,SAAShB,iBAAiB8B,gCAAgC;AACxE,aAAKZ,WAAWC,KAAK;UACnBN,MAAMN,UAAUwB;UAChBC,YAAYjB,YAAYQ,MAAMU;UAC9BC,cAAcnB,YAAYQ,MAAMP;UAChCmB,iBAAiBpB,YAAYQ,MAAMU;UACnCR,UAAUb;QACZ,CAAA;AACA,aAAKM,WAAWC,KAAK;UACnBN,MAAMN,UAAU6B;UAChBJ,YAAYjB,YAAYQ,MAAMU;UAC9BN,OAAOZ,YAAYQ,MAAMc;UACzBZ,UAAUb;QACZ,CAAA;AACA,aAAKM,WAAWC,KAAK;UACnBN,MAAMN,UAAU+B;UAChBN,YAAYjB,YAAYQ,MAAMU;UAC9BR,UAAUb;QACZ,CAAA;AACA,eAAO;MACT;AAEA,UAAIG,YAAYC,SAAShB,iBAAiBuC,yCAAyC;AACjF,aAAKC,UAAUC,uBAAuB1B,YAAYQ;AAClD,aAAKZ,cAAc;UACjBE,MAAMN,UAAUmC;UAChBC,UAAU,KAAKC,iBAAiB;YAC9BC,QAAQ,KAAKL,UAAUC;UACzB,CAAA;UACAhB,UAAUb;QACZ,CAAA;AACA,eAAO;MACT;AAEA,UAAIG,YAAYC,SAAShB,iBAAiB8C,gBAAgB;AACxD,aAAK5B,WAAWC,KAAK;UACnBN,MAAMN,UAAUO;UAChBE,MAAM;UACNO,OAAO;QACT,CAAA;AACA,eAAO;MACT;IACF;AAGA,UAAME,WAAYb,MAA6Ca;AAC/D,QAAI,CAACA,UAAU;AACb,WAAKP,WAAWC,KAAKP,KAAAA;AACrB,aAAO;IACT;AAEA,UAAMmC,iBACJnC,MAAMC,SAASN,UAAUa,sBACzBR,MAAMC,SAASN,UAAUmB,wBACzBd,MAAMC,SAASN,UAAUsB;AAC3B,UAAMmB,cACJpC,MAAMC,SAASN,UAAUwB,mBACzBnB,MAAMC,SAASN,UAAU6B,kBACzBxB,MAAMC,SAASN,UAAU+B;AAC3B,QAAI,iCAAiCb,SAASwB,YAAY,CAAC,IAAI;AAC7D,UAAIxB,SAASwB,SAAS,4BAAA,MAAkC,SAASD,aAAa;AAC5E,eAAO;MACT;IACF;AACA,QAAI,+BAA+BvB,SAASwB,YAAY,CAAC,IAAI;AAC3D,UAAIxB,SAASwB,SAAS,0BAAA,MAAgC,SAASF,gBAAgB;AAC7E,eAAO;MACT;IACF;AAEA,SAAK7B,WAAWC,KAAKP,KAAAA;AACrB,WAAO;EACT;;EAGAsC,IAAIC,OAAsB;AACxB,WAAO,MAAMD,IAAIC,KAAAA,EAAOC,KACtBnD,IAAI,CAACoD,mBAAAA;AA7IX;AA+IQ,UAAIA,eAAexC,SAASN,UAAU+C,KAAK;AAEzC,cAAM1C,QAASyC,eAA4BzC,SAAUyC,eAA4B5B;AAEjF,cAAM8B,YAAY3C,MAAMA;AACxB,cAAM4C,gBAAe5C,uBAAM6C,SAAN7C,mBAAY8C,UAAZ9C,mBAAmB+C,qBAAnB/C,mBAAsC;AAC3D,cAAMgD,8BAA6BhD,iBAAMqC,aAANrC,mBACjC,0CADiCA,mBAEhCiD,KACD,CAACC,qBAAuCA,iBAAiBC,UAASP,6CAAcxC;AAGlF,YAAIuC,cAAcrD,oBAAoB8D,qBAAqBJ,4BAA4B;AACrF,iBAAO;YACL/C,MAAMN,UAAUO;YAChBE,MAAM;YACNO,OAAOX,MAAMqC,SAAS,oCAAA;UACxB;QACF;MACF;AAEA,aAAOI;IACT,CAAA,CAAA;EAEJ;EAEAY,2BACEC,OACAC,UACAhB,OACoD;AACpD,UAAMiB,kBAAkB,MAAMH,2BAA2BC,OAAOC,UAAUhB,KAAAA;AAC1E,UAAM,EAAEkB,OAAOC,eAAe,SAASC,KAAI,IAAKH;AAEhD,UAAMI,mBAAmB;SAClBF,iBAA2B,CAAA;UAC3BC,6BAAMF,UAAmB,CAAA;;AAEhC,UAAMI,gBAAgBC,MAAMC,KAC1B,IAAIC,IACFJ,iBAAiBvE,IAAI,CAAC4E,MAAW;OAACA,uBAAG5C,QAAM4C,uBAAG7D,UAAQ6D,uBAAGC,QAAOC,KAAKC,UAAUH,CAAAA;MAAIA;KAAE,CAAA,EACrFhC,OAAM,CAAA;AAGV,WAAO;MACL,GAAGuB;MACHa,YAAY;QACVC,SAAST;QACTU,UAASZ,6BAAMY,YAAW,CAAA;MAC5B;IACF;EACF;EAEA,MAAMC,gBAAqC;AACzC,UAAMC,gBAAgB;MAAC;;AACvB,UAAMC,aAAa,MAAM,MAAMF,cAAAA;AAC/B,WAAO;MACL3E,QAAQ6E,WAAW7E;MACnB0C,OAAOmC,WAAWnC,QAAQ;WAAImC,WAAWnC;WAAUkC;UAAiB;MACpEE,QAAQD,WAAWC,SAAS;WAAID,WAAWC;WAAWF;UAAiB;MACvEF,SAASG,WAAWH,UAAU;WAAIG,WAAWH;WAAYE;UAAiB;IAC5E;EACF;AACF;AAlLajF","sourcesContent":["/**\n * Constants for LangGraph integration.\n * This file is separate from langgraph.agent.ts to avoid pulling in @ag-ui/langgraph\n * when only these constants are needed.\n */\n\nimport {\n  TextMessageStartEvent,\n  TextMessageContentEvent,\n  TextMessageEndEvent,\n  ToolCallStartEvent,\n  ToolCallArgsEvent,\n  ToolCallEndEvent,\n} from \"@ag-ui/client\";\n\nexport type TextMessageEvents =\n  | TextMessageStartEvent\n  | TextMessageContentEvent\n  | TextMessageEndEvent;\n\nexport type ToolCallEvents = ToolCallStartEvent | ToolCallArgsEvent | ToolCallEndEvent;\n\nexport enum CustomEventNames {\n  CopilotKitManuallyEmitMessage = \"copilotkit_manually_emit_message\",\n  CopilotKitManuallyEmitToolCall = \"copilotkit_manually_emit_tool_call\",\n  CopilotKitManuallyEmitIntermediateState = \"copilotkit_manually_emit_intermediate_state\",\n  CopilotKitExit = \"copilotkit_exit\",\n}\n\nexport interface PredictStateTool {\n  tool: string;\n  state_key: string;\n  tool_argument: string;\n}\n","import { map } from \"rxjs\";\nimport { LangGraphEventTypes } from \"../../../../agents/langgraph/events\";\nimport { RawEvent } from \"@ag-ui/core\";\nimport {\n  LangGraphAgent as AGUILangGraphAgent,\n  LangGraphHttpAgent,\n  type LangGraphAgentConfig,\n  ProcessedEvents,\n  SchemaKeys,\n  type State,\n  StateEnrichment,\n} from \"@ag-ui/langgraph\";\nimport { Message as LangGraphMessage } from \"@langchain/langgraph-sdk/dist/types.messages\";\nimport { ThreadState } from \"@langchain/langgraph-sdk\";\n\ninterface CopilotKitStateEnrichment {\n  copilotkit: {\n    actions: StateEnrichment[\"ag-ui\"][\"tools\"];\n    context: StateEnrichment[\"ag-ui\"][\"context\"];\n  };\n}\n\nimport { RunAgentInput, EventType, CustomEvent } from \"@ag-ui/client\";\n\n// Import and re-export from separate file to maintain API compatibility\nimport { CustomEventNames, TextMessageEvents, ToolCallEvents, PredictStateTool } from \"./consts\";\nexport { CustomEventNames };\n\nexport class LangGraphAgent extends AGUILangGraphAgent {\n  constructor(config: LangGraphAgentConfig) {\n    super(config);\n  }\n\n  // @ts-ignore\n  public clone() {\n    return new LangGraphAgent(this.config);\n  }\n\n  dispatchEvent(event: ProcessedEvents) {\n    if (event.type === EventType.CUSTOM) {\n      // const event = processedEvent as unknown as CustomEvent;\n      const customEvent = event as unknown as CustomEvent;\n\n      if (customEvent.name === CustomEventNames.CopilotKitManuallyEmitMessage) {\n        this.subscriber.next({\n          type: EventType.TEXT_MESSAGE_START,\n          role: \"assistant\",\n          messageId: customEvent.value.message_id,\n          rawEvent: event,\n        });\n        this.subscriber.next({\n          type: EventType.TEXT_MESSAGE_CONTENT,\n          messageId: customEvent.value.message_id,\n          delta: customEvent.value.message,\n          rawEvent: event,\n        });\n        this.subscriber.next({\n          type: EventType.TEXT_MESSAGE_END,\n          messageId: customEvent.value.message_id,\n          rawEvent: event,\n        });\n        return true;\n      }\n\n      if (customEvent.name === CustomEventNames.CopilotKitManuallyEmitToolCall) {\n        this.subscriber.next({\n          type: EventType.TOOL_CALL_START,\n          toolCallId: customEvent.value.id,\n          toolCallName: customEvent.value.name,\n          parentMessageId: customEvent.value.id,\n          rawEvent: event,\n        });\n        this.subscriber.next({\n          type: EventType.TOOL_CALL_ARGS,\n          toolCallId: customEvent.value.id,\n          delta: customEvent.value.args,\n          rawEvent: event,\n        });\n        this.subscriber.next({\n          type: EventType.TOOL_CALL_END,\n          toolCallId: customEvent.value.id,\n          rawEvent: event,\n        });\n        return true;\n      }\n\n      if (customEvent.name === CustomEventNames.CopilotKitManuallyEmitIntermediateState) {\n        this.activeRun.manuallyEmittedState = customEvent.value;\n        this.dispatchEvent({\n          type: EventType.STATE_SNAPSHOT,\n          snapshot: this.getStateSnapshot({\n            values: this.activeRun.manuallyEmittedState,\n          } as ThreadState<State>),\n          rawEvent: event,\n        });\n        return true;\n      }\n\n      if (customEvent.name === CustomEventNames.CopilotKitExit) {\n        this.subscriber.next({\n          type: EventType.CUSTOM,\n          name: \"Exit\",\n          value: true,\n        });\n        return true;\n      }\n    }\n\n    // Intercept all text message and tool call events and check if should disable\n    const rawEvent = (event as ToolCallEvents | TextMessageEvents).rawEvent;\n    if (!rawEvent) {\n      this.subscriber.next(event);\n      return true;\n    }\n\n    const isMessageEvent =\n      event.type === EventType.TEXT_MESSAGE_START ||\n      event.type === EventType.TEXT_MESSAGE_CONTENT ||\n      event.type === EventType.TEXT_MESSAGE_END;\n    const isToolEvent =\n      event.type === EventType.TOOL_CALL_START ||\n      event.type === EventType.TOOL_CALL_ARGS ||\n      event.type === EventType.TOOL_CALL_END;\n    if (\"copilotkit:emit-tool-calls\" in (rawEvent.metadata || {})) {\n      if (rawEvent.metadata[\"copilotkit:emit-tool-calls\"] === false && isToolEvent) {\n        return false;\n      }\n    }\n    if (\"copilotkit:emit-messages\" in (rawEvent.metadata || {})) {\n      if (rawEvent.metadata[\"copilotkit:emit-messages\"] === false && isMessageEvent) {\n        return false;\n      }\n    }\n\n    this.subscriber.next(event);\n    return true;\n  }\n\n  // @ts-ignore\n  run(input: RunAgentInput) {\n    return super.run(input).pipe(\n      map((processedEvent) => {\n        // Turn raw event into emit state snapshot from tool call event\n        if (processedEvent.type === EventType.RAW) {\n          // Get the LangGraph event from the AGUI event.\n          const event = (processedEvent as RawEvent).event ?? (processedEvent as RawEvent).rawEvent;\n\n          const eventType = event.event;\n          const toolCallData = event.data?.chunk?.tool_call_chunks?.[0];\n          const toolCallUsedToPredictState = event.metadata?.[\n            \"copilotkit:emit-intermediate-state\"\n          ]?.some(\n            (predictStateTool: PredictStateTool) => predictStateTool.tool === toolCallData?.name,\n          );\n\n          if (eventType === LangGraphEventTypes.OnChatModelStream && toolCallUsedToPredictState) {\n            return {\n              type: EventType.CUSTOM,\n              name: \"PredictState\",\n              value: event.metadata[\"copilotkit:emit-intermediate-state\"],\n            };\n          }\n        }\n\n        return processedEvent;\n      }),\n    );\n  }\n\n  langGraphDefaultMergeState(\n    state: State,\n    messages: LangGraphMessage[],\n    input: RunAgentInput,\n  ): State<StateEnrichment & CopilotKitStateEnrichment> {\n    const aguiMergedState = super.langGraphDefaultMergeState(state, messages, input);\n    const { tools: returnedTools, \"ag-ui\": agui } = aguiMergedState;\n    // tolerate undefined and de-duplicate by stable key (id | name | key)\n    const rawCombinedTools = [\n      ...((returnedTools as any[]) ?? []),\n      ...((agui?.tools as any[]) ?? []),\n    ];\n    const combinedTools = Array.from(\n      new Map(\n        rawCombinedTools.map((t: any) => [t?.id ?? t?.name ?? t?.key ?? JSON.stringify(t), t]),\n      ).values(),\n    );\n\n    return {\n      ...aguiMergedState,\n      copilotkit: {\n        actions: combinedTools,\n        context: agui?.context ?? [],\n      },\n    };\n  }\n\n  async getSchemaKeys(): Promise<SchemaKeys> {\n    const CONSTANT_KEYS = [\"copilotkit\"];\n    const schemaKeys = await super.getSchemaKeys();\n    return {\n      config: schemaKeys.config,\n      input: schemaKeys.input ? [...schemaKeys.input, ...CONSTANT_KEYS] : null,\n      output: schemaKeys.output ? [...schemaKeys.output, ...CONSTANT_KEYS] : null,\n      context: schemaKeys.context ? [...schemaKeys.context, ...CONSTANT_KEYS] : null,\n    };\n  }\n}\n\nexport { LangGraphHttpAgent };\n","import { ActionExecutionMessage, ResultMessage, TextMessage } from \"../../graphql/types/converted\";\n\nexport enum LangGraphEventTypes {\n  OnChainStart = \"on_chain_start\",\n  OnChainStream = \"on_chain_stream\",\n  OnChainEnd = \"on_chain_end\",\n  OnChatModelStart = \"on_chat_model_start\",\n  OnChatModelStream = \"on_chat_model_stream\",\n  OnChatModelEnd = \"on_chat_model_end\",\n  OnToolStart = \"on_tool_start\",\n  OnToolEnd = \"on_tool_end\",\n  OnCopilotKitStateSync = \"on_copilotkit_state_sync\",\n  OnCopilotKitEmitMessage = \"on_copilotkit_emit_message\",\n  OnCopilotKitEmitToolCall = \"on_copilotkit_emit_tool_call\",\n  OnCustomEvent = \"on_custom_event\",\n  OnInterrupt = \"on_interrupt\",\n  OnCopilotKitInterrupt = \"on_copilotkit_interrupt\",\n  OnCopilotKitError = \"on_copilotkit_error\",\n}\n\nexport enum MetaEventNames {\n  LangGraphInterruptEvent = \"LangGraphInterruptEvent\",\n  CopilotKitLangGraphInterruptEvent = \"CopilotKitLangGraphInterruptEvent\",\n}\n\nexport enum CustomEventNames {\n  CopilotKitManuallyEmitMessage = \"copilotkit_manually_emit_message\",\n  CopilotKitManuallyEmitToolCall = \"copilotkit_manually_emit_tool_call\",\n  CopilotKitManuallyEmitIntermediateState = \"copilotkit_manually_emit_intermediate_state\",\n  CopilotKitExit = \"copilotkit_exit\",\n}\n\ntype LangGraphOnCopilotKitStateSyncEvent = {\n  event: LangGraphEventTypes.OnCopilotKitStateSync;\n  thread_id: string;\n  agent_name: string;\n  node_name: string;\n  run_id: string;\n  active: boolean;\n  role: string;\n  state: any;\n  running: boolean;\n};\n\ntype LangGraphOnChainStartEvent = {\n  event: LangGraphEventTypes.OnChainStart;\n  run_id: string;\n  name: string;\n  tags: string[];\n  metadata: { thread_id: string };\n  data: {\n    input: any;\n  };\n  parent_ids: string[];\n};\n\ntype LangGraphOnChainEndEvent = {\n  event: LangGraphEventTypes.OnChainEnd;\n  name: string;\n  run_id: string;\n  tags: string[];\n  metadata: {\n    thread_id: string;\n    langgraph_step: number;\n    langgraph_node: string;\n    langgraph_triggers: string[];\n    langgraph_task_idx: number;\n    thread_ts: string;\n  };\n  data: {\n    input: any;\n    output: any;\n  };\n  parent_ids: string[];\n};\n\ntype LangGraphOnChatModelStartEvent = {\n  event: LangGraphEventTypes.OnChatModelStart;\n  name: string;\n  run_id: string;\n  tags: string[];\n  metadata: {\n    thread_id: string;\n    langgraph_step: number;\n    langgraph_node: string;\n    langgraph_triggers: string[];\n    langgraph_task_idx: number;\n    thread_ts: string;\n    ls_provider: string;\n    ls_model_name: string;\n    ls_model_type: string;\n    ls_temperature: number;\n  };\n  data: {\n    input: {\n      messages: {\n        lc: number;\n        type: string;\n        id: string[];\n        kwargs: {\n          content: string;\n          type: string;\n          id: string;\n        };\n      }[][];\n    };\n  };\n  parent_ids: string[];\n};\n\ntype LangGraphOnChatModelStreamEvent = {\n  event: LangGraphEventTypes.OnChatModelStream;\n  name: string;\n  run_id: string;\n  tags: string[];\n  metadata: {\n    thread_id: string;\n    langgraph_step: number;\n    langgraph_node: string;\n    langgraph_triggers: string[];\n    langgraph_task_idx: number;\n    thread_ts: string;\n    ls_provider: string;\n    ls_model_name: string;\n    ls_model_type: string;\n    ls_temperature: number;\n  };\n  data: {\n    chunk: {\n      lc: number;\n      type: string;\n      id: string;\n      kwargs: {\n        content: string | { text: string; type: string; index: number }[];\n        additional_kwargs: {\n          tool_calls: {\n            index: number;\n            id: string;\n            function: { arguments: string; name: string };\n            type: string;\n          }[];\n        };\n        type: string;\n        id: string;\n        tool_calls: { name: string; args: {}; id: string; type: string }[];\n        tool_call_chunks: {\n          name: string;\n          args: string;\n          id: string;\n          index: number;\n          type: string;\n        }[];\n        invalid_tool_calls: any[];\n      };\n    };\n  };\n  parent_ids: string[];\n};\n\ntype LangGraphOnChatModelEndEvent = {\n  event: LangGraphEventTypes.OnChatModelEnd;\n  name: string;\n  run_id: string;\n  tags: string[];\n  metadata: {\n    thread_id: string;\n    langgraph_step: number;\n    langgraph_node: string;\n    langgraph_triggers: string[];\n    langgraph_task_idx: number;\n    thread_ts: string;\n    ls_provider: string;\n    ls_model_name: string;\n    ls_model_type: string;\n    ls_temperature: number;\n  };\n  data: {\n    input: any;\n    output: {\n      generations: {\n        text: string;\n        generation_info: {\n          finish_reason: string;\n          model_name: string;\n          system_fingerprint: string;\n        };\n        type: string;\n        message: {\n          lc: number;\n          type: string;\n          id: string[];\n          kwargs: {\n            content: string;\n            additional_kwargs: {\n              tool_calls: {\n                index: number;\n                id: string;\n                function: { arguments: string; name: string };\n                type: string;\n              }[];\n            };\n            response_metadata: {\n              finish_reason: string;\n              model_name: string;\n              system_fingerprint: string;\n            };\n            type: string;\n            id: string;\n            tool_calls: { name: string; args: { query: string }; id: string; type: string }[];\n            invalid_tool_calls: any[];\n          };\n        };\n      }[][];\n      llm_output: any;\n      run: any;\n    };\n  };\n  parent_ids: string[];\n};\n\ntype LangGraphOnChainStreamEvent = {\n  event: LangGraphEventTypes.OnChainStream;\n  name: string;\n  run_id: string;\n  tags: string[];\n  metadata: {\n    thread_id: string;\n    langgraph_step?: number;\n    langgraph_node?: string;\n    langgraph_triggers?: string[];\n    langgraph_task_idx?: number;\n    thread_ts?: string;\n  };\n  data: {\n    chunk: {\n      messages: {\n        lc: number;\n        type: string;\n        id: string[];\n        kwargs: {\n          content: string;\n          additional_kwargs?: {\n            tool_calls?: {\n              index: number;\n              id: string;\n              function: { arguments: string; name: string };\n              type: string;\n            }[];\n          };\n          response_metadata?: {\n            finish_reason: string;\n            model_name: string;\n            system_fingerprint: string;\n          };\n          type: string;\n          id: string;\n          tool_calls?: { name: string; args: { query: string }; id: string; type: string }[];\n          invalid_tool_calls?: any[];\n        };\n      }[];\n    };\n  };\n  parent_ids: string[];\n};\n\ntype LangGraphOnToolStartEvent = {\n  event: LangGraphEventTypes.OnToolStart;\n  name: string;\n  run_id: string;\n  tags: string[];\n  metadata: {\n    thread_id: string;\n    langgraph_step: number;\n    langgraph_node: string;\n    langgraph_triggers: string[];\n    langgraph_task_idx: number;\n    thread_ts: string;\n  };\n  data: {\n    input: {\n      query: string;\n    };\n  };\n  parent_ids: string[];\n};\n\ntype LangGraphOnToolEndEvent = {\n  event: LangGraphEventTypes.OnToolEnd;\n  name: string;\n  run_id: string;\n  tags: string[];\n  metadata: {\n    thread_id: string;\n    langgraph_step: number;\n    langgraph_node: string;\n    langgraph_triggers: string[];\n    langgraph_task_idx: number;\n    thread_ts: string;\n  };\n  data: {\n    input: {\n      query: string;\n    };\n    output: {\n      lc: number;\n      type: string;\n      id: string[];\n      kwargs: {\n        content: string[];\n        type: string;\n        name: string;\n        tool_call_id: string;\n        status: string;\n      };\n    };\n  };\n  parent_ids: string[];\n};\n\ntype LangGraphOnCustomEvent = {\n  event: LangGraphEventTypes.OnCustomEvent;\n  run_id: string;\n  name: string;\n  tags: string[];\n  metadata: {\n    thread_id: string;\n    langgraph_step: number;\n    langgraph_node: string;\n    langgraph_triggers: string[];\n    langgraph_path: [string, string];\n    langgraph_checkpoint_ns: string;\n    checkpoint_ns: string;\n  };\n  data: any;\n  parent_ids: string[];\n};\n\ninterface LangGraphInterruptEvent {\n  event: LangGraphEventTypes.OnInterrupt;\n  value: string;\n}\n\ninterface CopilotKitLangGraphInterruptEvent {\n  event: LangGraphEventTypes.OnCopilotKitInterrupt;\n  data: { value: string; messages: (TextMessage | ActionExecutionMessage | ResultMessage)[] };\n}\n\ninterface CopilotKitLangGraphErrorEvent {\n  event: LangGraphEventTypes.OnCopilotKitError;\n  data: {\n    error: {\n      message: string;\n      type: string;\n      agent_name: string;\n      status_code?: number;\n      response_data?: any;\n    };\n    thread_id: string;\n    agent_name: string;\n    node_name: string;\n  };\n}\n\nexport type LangGraphEvent =\n  | LangGraphOnChainStartEvent\n  | LangGraphOnChainStreamEvent\n  | LangGraphOnChainEndEvent\n  | LangGraphOnChatModelStartEvent\n  | LangGraphOnChatModelStreamEvent\n  | LangGraphOnChatModelEndEvent\n  | LangGraphOnToolStartEvent\n  | LangGraphOnToolEndEvent\n  | LangGraphOnCopilotKitStateSyncEvent\n  | LangGraphOnCustomEvent\n  | LangGraphInterruptEvent\n  | CopilotKitLangGraphInterruptEvent\n  | CopilotKitLangGraphErrorEvent;\n"]}

package/dist/v2/index.d.ts

@@ -1 +1,2 @@
 export * from '@copilotkitnext/runtime';
+export * from '@copilotkitnext/agent';

package/dist/v2/index.js

@@ -2,6 +2,7 @@
 
 require('reflect-metadata');
 var runtime = require('@copilotkitnext/runtime');
+var agent = require('@copilotkitnext/agent');
 
 
 
@@ -11,5 +12,11 @@ Object.keys(runtime).forEach(function (k) {
 		get: function () { return runtime[k]; }
 	});
 });
+Object.keys(agent).forEach(function (k) {
+	if (k !== 'default' && !Object.prototype.hasOwnProperty.call(exports, k)) Object.defineProperty(exports, k, {
+		enumerable: true,
+		get: function () { return agent[k]; }
+	});
+});
 //# sourceMappingURL=out.js.map
 //# sourceMappingURL=index.js.map

package/dist/v2/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/v2/index.ts"],"names":[],"mappings":";AAAA,OAAO;AACP,cAAc","sourcesContent":["import \"reflect-metadata\";\nexport * from \"@copilotkitnext/runtime\";\n"]}
+{"version":3,"sources":["../../src/v2/index.ts"],"names":[],"mappings":";AAAA,OAAO;AACP,cAAc;AACd,cAAc","sourcesContent":["import \"reflect-metadata\";\nexport * from \"@copilotkitnext/runtime\";\nexport * from \"@copilotkitnext/agent\";\n"]}

package/dist/v2/index.mjs

@@ -1,4 +1,5 @@
 import 'reflect-metadata';
 export * from '@copilotkitnext/runtime';
+export * from '@copilotkitnext/agent';
 //# sourceMappingURL=out.js.map
 //# sourceMappingURL=index.mjs.map

package/dist/v2/index.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/v2/index.ts"],"names":[],"mappings":";AAAA,OAAO;AACP,cAAc","sourcesContent":["import \"reflect-metadata\";\nexport * from \"@copilotkitnext/runtime\";\n"]}
+{"version":3,"sources":["../../src/v2/index.ts"],"names":[],"mappings":";AAAA,OAAO;AACP,cAAc;AACd,cAAc","sourcesContent":["import \"reflect-metadata\";\nexport * from \"@copilotkitnext/runtime\";\nexport * from \"@copilotkitnext/agent\";\n"]}

package/package.json

@@ -9,7 +9,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "1.50.0-beta.1",
+  "version": "1.50.0-beta.3",
   "sideEffects": false,
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",
@@ -23,6 +23,11 @@
       "import": "./dist/v2/index.mjs",
       "require": "./dist/v2/index.js",
       "types": "./dist/v2/index.d.ts"
+    },
+    "./langgraph": {
+      "import": "./dist/langgraph.mjs",
+      "require": "./dist/langgraph.js",
+      "types": "./dist/langgraph.d.ts"
     }
   },
   "types": "./dist/index.d.ts",
@@ -41,29 +46,21 @@
     "tsup": "^6.7.0",
     "typescript": "^5.2.3",
     "vitest": "^3.2.4",
-    "zod-to-json-schema": "^3.23.5",
-    "eslint-config-custom": "1.4.6",
-    "tsconfig": "1.4.6"
+    "tsconfig": "1.4.6",
+    "eslint-config-custom": "1.4.6"
   },
   "dependencies": {
-    "@anthropic-ai/sdk": "^0.57.0",
-    "@copilotkitnext/agent": "0.0.22",
-    "@copilotkitnext/runtime": "0.0.22",
+    "@copilotkitnext/agent": "0.0.23",
+    "@copilotkitnext/runtime": "0.0.23",
     "@graphql-yoga/plugin-defer-stream": "^3.3.1",
-    "@langchain/aws": "^0.1.9",
-    "@langchain/community": "^0.3.29",
+    "@hono/node-server": "^1.13.5",
     "@langchain/core": "^0.3.38",
-    "@langchain/google-gauth": "^0.1.0",
-    "@langchain/langgraph-sdk": "^0.0.70",
-    "@langchain/openai": "^0.4.2",
     "@scarf/scarf": "^1.3.0",
     "class-transformer": "^0.5.1",
     "class-validator": "^0.14.1",
     "graphql": "^16.8.1",
     "graphql-scalars": "^1.23.0",
     "graphql-yoga": "^5.3.1",
-    "groq-sdk": "^0.5.0",
-    "@hono/node-server": "^1.13.5",
     "hono": "^4.10.3",
     "langchain": "^0.3.3",
     "openai": "^4.85.1",
@@ -74,14 +71,55 @@
     "rxjs": "7.8.1",
     "type-graphql": "2.0.0-rc.1",
     "zod": "^3.23.3",
-    "@copilotkit/shared": "1.50.0-beta.1"
+    "@copilotkit/shared": "1.50.0-beta.3"
   },
   "peerDependencies": {
-    "@ag-ui/client": ">=0.0.39",
+    "@anthropic-ai/sdk": "^0.57.0",
+    "@ag-ui/client": "^0.0.41",
     "@ag-ui/core": ">=0.0.39",
     "@ag-ui/encoder": ">=0.0.39",
     "@ag-ui/langgraph": ">=0.0.18",
-    "@ag-ui/proto": ">=0.0.39"
+    "@ag-ui/proto": ">=0.0.39",
+    "@langchain/aws": "^0.1.9",
+    "@langchain/community": "^0.3.58",
+    "@langchain/google-gauth": "^0.1.0",
+    "@langchain/langgraph-sdk": "^0.0.70",
+    "@langchain/openai": "^0.4.2",
+    "groq-sdk": "^0.5.0",
+    "langchain": "^0.3.3",
+    "openai": "^4.85.1"
+  },
+  "peerDependenciesMeta": {
+    "@anthropic-ai/sdk": {
+      "optional": true
+    },
+    "@ag-ui/langgraph": {
+      "optional": true
+    },
+    "@langchain/aws": {
+      "optional": true
+    },
+    "@langchain/community": {
+      "optional": true
+    },
+    "@langchain/google-gauth": {
+      "optional": true
+    },
+    "@langchain/langgraph-sdk": {
+      "optional": true
+    },
+    "@langchain/openai": {
+      "optional": true
+    },
+    "groq-sdk": {
+      "optional": true
+    },
+    "langchain": {
+      "optional": true
+    },
+    "openai": {
+      "optional": true
+    }
   },
   "keywords": [
     "copilotkit",

package/src/graphql/message-conversion/agui-to-gql.test.ts

@@ -1,4 +1,4 @@
-import { describe, test, expect, vi } from "vitest";
+import { describe, test, expect } from "@jest/globals";
 import * as gql from "../types/converted/index";
 import agui from "@copilotkit/shared";
 import {
@@ -587,7 +587,7 @@ describe("agui-to-gql", () => {
     });
 
     test("should verify render function receives correct props including name", () => {
-      const mockRender = vi.fn(
+      const mockRender = jest.fn(
         (props) => `Rendered: ${props.name} with args: ${JSON.stringify(props.args)}`,
       );
       const aguiMessage: agui.Message = {

package/src/graphql/message-conversion/gql-to-agui.test.ts

@@ -1,4 +1,4 @@
-import { describe, test, expect, vi } from "vitest";
+import { describe, test, expect } from "@jest/globals";
 import * as gql from "../types/converted/index";
 import { MessageStatusCode } from "../types/message-status.type";
 import {
@@ -370,7 +370,7 @@ describe("message-conversion", () => {
         status: { code: MessageStatusCode.Pending },
       });
 
-      const mockRender = vi.fn();
+      const mockRender = jest.fn();
       const mockRenderAndWaitForResponse = (props: any) => "Test Render With Response";
 
       const actions = {
@@ -414,7 +414,7 @@ describe("message-conversion", () => {
         status: { code: MessageStatusCode.Pending },
       });
 
-      const mockRender = vi.fn();
+      const mockRender = jest.fn();
       const actions = {
         testAction: {
           name: "testAction",
@@ -445,7 +445,7 @@ describe("message-conversion", () => {
         status: { code: MessageStatusCode.Success },
       });
 
-      const mockRender = vi.fn();
+      const mockRender = jest.fn();
       const actions = {
         testAction: {
           name: "testAction",
@@ -483,7 +483,7 @@ describe("message-conversion", () => {
         actionName: "testAction",
       });
 
-      const mockRender = vi.fn();
+      const mockRender = jest.fn();
       const actions = {
         testAction: {
           name: "testAction",
@@ -517,7 +517,7 @@ describe("message-conversion", () => {
         status: { code: MessageStatusCode.Pending },
       });
 
-      const mockRender = vi.fn();
+      const mockRender = jest.fn();
       const actions = {
         testAction: {
           name: "testAction",
@@ -627,7 +627,7 @@ describe("message-conversion", () => {
         role: gql.Role.assistant,
       });
 
-      const mockRender = vi.fn();
+      const mockRender = jest.fn();
       const coAgentStateRenders = {
         testAgent: {
           name: "testAgent",
@@ -740,7 +740,7 @@ describe("message-conversion", () => {
         role: gql.Role.assistant,
       });
 
-      const mockRender = vi.fn();
+      const mockRender = jest.fn();
       const coAgentStateRenders = {
         testAgent: {
           name: "testAgent",
@@ -853,7 +853,7 @@ describe("message-conversion", () => {
       const actions = {
         testAction: {
           name: "testAction",
-          render: vi.fn((props) => `Rendered: ${props.args.test}`),
+          render: jest.fn((props) => `Rendered: ${props.args.test}`),
         },
       };
 
@@ -889,7 +889,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Wildcard rendered: ${props.args.test}`),
+          render: jest.fn((props) => `Wildcard rendered: ${props.args.test}`),
         },
       };
 
@@ -922,7 +922,7 @@ describe("message-conversion", () => {
     });
 
     test("should pass tool name to wildcard action render function", () => {
-      const mockRender = vi.fn(
+      const mockRender = jest.fn(
         (props) => `Wildcard rendered: ${props.name} with args: ${JSON.stringify(props.args)}`,
       );
       const actions = {
@@ -954,7 +954,9 @@ describe("message-conversion", () => {
     });
 
     test("should pass tool name to regular action render function", () => {
-      const mockRender = vi.fn((props) => `Regular action rendered: ${JSON.stringify(props.args)}`);
+      const mockRender = jest.fn(
+        (props) => `Regular action rendered: ${JSON.stringify(props.args)}`,
+      );
       const actions = {
         testAction: {
           name: "testAction",
@@ -992,11 +994,11 @@ describe("message-conversion", () => {
       const actions = {
         specificAction: {
           name: "specificAction",
-          render: vi.fn((props) => "Specific action rendered"),
+          render: jest.fn((props) => "Specific action rendered"),
         },
         "*": {
           name: "*",
-          render: vi.fn((props) => "Wildcard action rendered"),
+          render: jest.fn((props) => "Wildcard action rendered"),
         },
       };
 
@@ -1032,7 +1034,7 @@ describe("message-conversion", () => {
       const actions = {
         otherAction: {
           name: "otherAction",
-          render: vi.fn(),
+          render: jest.fn(),
         },
       };
 
@@ -1095,7 +1097,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Result: ${props.result}`),
+          render: jest.fn((props) => `Result: ${props.result}`),
         },
       };
 
@@ -1133,7 +1135,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Status: ${props.status}`),
+          render: jest.fn((props) => `Status: ${props.status}`),
         },
       };
 
@@ -1170,7 +1172,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Status: ${props.status}`),
+          render: jest.fn((props) => `Status: ${props.status}`),
         },
       };
 
@@ -1207,7 +1209,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Status: ${props.status}`),
+          render: jest.fn((props) => `Status: ${props.status}`),
         },
       };
 
@@ -1244,7 +1246,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Status: ${props.status}`),
+          render: jest.fn((props) => `Status: ${props.status}`),
         },
       };
 
@@ -1281,7 +1283,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Args: ${JSON.stringify(props.args)}`),
+          render: jest.fn((props) => `Args: ${JSON.stringify(props.args)}`),
         },
       };
 
@@ -1317,7 +1319,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Args: ${JSON.stringify(props.args)}`),
+          render: jest.fn((props) => `Args: ${JSON.stringify(props.args)}`),
         },
       };
 
@@ -1353,7 +1355,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Complex: ${JSON.stringify(props.args)}`),
+          render: jest.fn((props) => `Complex: ${JSON.stringify(props.args)}`),
         },
       };
 
@@ -1401,11 +1403,11 @@ describe("message-conversion", () => {
       const actions = {
         wildcard1: {
           name: "*",
-          render: vi.fn((props) => "First wildcard"),
+          render: jest.fn((props) => "First wildcard"),
         },
         wildcard2: {
           name: "*",
-          render: vi.fn((props) => "Second wildcard"),
+          render: jest.fn((props) => "Second wildcard"),
         },
       };
 
@@ -1441,7 +1443,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Result: ${JSON.stringify(props.result)}`),
+          render: jest.fn((props) => `Result: ${JSON.stringify(props.result)}`),
         },
       };
 
@@ -1478,7 +1480,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Result: ${JSON.stringify(props.result)}`),
+          render: jest.fn((props) => `Result: ${JSON.stringify(props.result)}`),
         },
       };
 
@@ -1515,7 +1517,7 @@ describe("message-conversion", () => {
       const actions = {
         "*": {
           name: "*",
-          render: vi.fn((props) => `Result: ${JSON.stringify(props.result)}`),
+          render: jest.fn((props) => `Result: ${JSON.stringify(props.result)}`),
         },
       };