@copilotkit/runtime-client-gql 1.10.0 → 1.10.1-next.0

package/dist/message-conversion/roundtrip-conversion.test.mjs

@@ -1,9 +1,9 @@
 import {
   aguiToGQL
-} from "../chunk-YNQMTL2P.mjs";
+} from "../chunk-7OSRQ2O3.mjs";
 import {
   gqlToAGUI
-} from "../chunk-MTD2RJDJ.mjs";
+} from "../chunk-IGXWE4TX.mjs";
 import {
   describe,
   globalExpect,
@@ -11,7 +11,7 @@ import {
   vi
 } from "../chunk-2R7M2FWR.mjs";
 import "../chunk-7ECCT6PK.mjs";
-import "../chunk-PAQ6AHHC.mjs";
+import "../chunk-AES3X7TL.mjs";
 import "../chunk-X2UAP3QY.mjs";
 import "../chunk-HEODM5TW.mjs";
 import "../chunk-4KTMZMM2.mjs";
@@ -215,5 +215,99 @@ describe("roundtrip message conversion", () => {
     globalExpect(aguiUserMsgs2[0].image.format).toBe("png");
     globalExpect(aguiUserMsgs2[0].image.bytes).toBe("userbase64data");
   });
+  test("wild card action roundtrip conversion", () => {
+    const mockRender = vi.fn((props) => `Wildcard rendered: ${props.args.test}`);
+    const aguiMsg = {
+      id: "assistant-wildcard-1",
+      role: "assistant",
+      content: "Running wild card action",
+      toolCalls: [
+        {
+          id: "tool-call-wildcard-1",
+          type: "function",
+          function: {
+            name: "unknownAction",
+            arguments: JSON.stringify({ test: "wildcard-value" })
+          }
+        }
+      ],
+      generativeUI: mockRender
+    };
+    const actions = {
+      "*": { name: "*" }
+    };
+    const gqlMsgs = aguiToGQL(aguiMsg, actions);
+    const aguiMsgs2 = gqlToAGUI(gqlMsgs, actions);
+    globalExpect(typeof actions["*"].render).toBe("function");
+    globalExpect(actions["*"].render).toBe(mockRender);
+    globalExpect(aguiMsgs2).toHaveLength(2);
+    globalExpect(aguiMsgs2[0].role).toBe("assistant");
+    globalExpect(aguiMsgs2[1].role).toBe("assistant");
+    if ("toolCalls" in aguiMsgs2[1]) {
+      globalExpect(aguiMsgs2[1].toolCalls[0].function.name).toBe("unknownAction");
+      globalExpect(aguiMsgs2[1].toolCalls[0].function.arguments).toBe(
+        '{"test":"wildcard-value"}'
+      );
+    }
+  });
+  test("wild card action with specific action priority roundtrip", () => {
+    const mockRender = vi.fn((props) => `Specific action rendered: ${props.args.test}`);
+    const aguiMsg = {
+      id: "assistant-priority-1",
+      role: "assistant",
+      content: "Running specific action",
+      toolCalls: [
+        {
+          id: "tool-call-priority-1",
+          type: "function",
+          function: {
+            name: "specificAction",
+            arguments: JSON.stringify({ test: "specific-value" })
+          }
+        }
+      ],
+      generativeUI: mockRender
+    };
+    const actions = {
+      specificAction: { name: "specificAction" },
+      "*": { name: "*" }
+    };
+    const gqlMsgs = aguiToGQL(aguiMsg, actions);
+    const aguiMsgs2 = gqlToAGUI(gqlMsgs, actions);
+    globalExpect(typeof actions.specificAction.render).toBe("function");
+    globalExpect(actions.specificAction.render).toBe(mockRender);
+    globalExpect(actions["*"].render).toBeUndefined();
+    globalExpect(aguiMsgs2).toHaveLength(2);
+    globalExpect(aguiMsgs2[0].role).toBe("assistant");
+    globalExpect(aguiMsgs2[1].role).toBe("assistant");
+    if ("toolCalls" in aguiMsgs2[1]) {
+      globalExpect(aguiMsgs2[1].toolCalls[0].function.name).toBe("specificAction");
+      globalExpect(aguiMsgs2[1].toolCalls[0].function.arguments).toBe(
+        '{"test":"specific-value"}'
+      );
+    }
+  });
+  test("wild card action GQL -> AGUI -> GQL roundtrip", () => {
+    const actionExecMsg = new ActionExecutionMessage({
+      id: "wildcard-action-1",
+      name: "unknownAction",
+      arguments: { test: "wildcard-gql-value" },
+      parentMessageId: "assistant-1"
+    });
+    const actions = {
+      "*": {
+        name: "*",
+        render: vi.fn((props) => `GQL wildcard rendered: ${props.args.test}`)
+      }
+    };
+    const aguiMsgs = gqlToAGUI([actionExecMsg], actions);
+    const gqlMsgs2 = aguiToGQL(aguiMsgs, actions);
+    globalExpect(gqlMsgs2).toHaveLength(2);
+    globalExpect(gqlMsgs2[0].id).toBe("wildcard-action-1");
+    globalExpect(gqlMsgs2[0].role).toBe(Role.Assistant);
+    globalExpect(gqlMsgs2[1].id).toBe("wildcard-action-1");
+    globalExpect(gqlMsgs2[1].name).toBe("unknownAction");
+    globalExpect(gqlMsgs2[1].arguments).toEqual({ test: "wildcard-gql-value" });
+  });
 });
 //# sourceMappingURL=roundtrip-conversion.test.mjs.map

package/dist/message-conversion/roundtrip-conversion.test.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/message-conversion/roundtrip-conversion.test.ts"],"sourcesContent":["import { describe, test, expect, vi } from \"vitest\";\nimport * as gql from \"../client\";\nimport agui from \"@copilotkit/shared\";\nimport { aguiToGQL } from \"./agui-to-gql\";\nimport { gqlToAGUI } from \"./gql-to-agui\";\n\n// Helper to strip functions for deep equality\nfunction stripFunctions(obj: any): any {\n  if (typeof obj === \"function\") return undefined;\n  if (Array.isArray(obj)) return obj.map(stripFunctions);\n  if (obj && typeof obj === \"object\") {\n    const out: any = {};\n    for (const k in obj) {\n      if (typeof obj[k] !== \"function\") {\n        out[k] = stripFunctions(obj[k]);\n      }\n    }\n    return out;\n  }\n  return obj;\n}\n\ndescribe(\"roundtrip message conversion\", () => {\n  test(\"text message AGUI -> GQL -> AGUI\", () => {\n    const aguiMsg: agui.Message = {\n      id: \"user-1\",\n      role: \"user\",\n      content: \"Hello!\",\n    };\n    const gqlMsgs = aguiToGQL(aguiMsg);\n    const aguiMsgs2 = gqlToAGUI(gqlMsgs);\n    expect(stripFunctions(aguiMsgs2[0])).toEqual(stripFunctions(aguiMsg));\n  });\n\n  test(\"text message GQL -> AGUI -> GQL\", () => {\n    const gqlMsg = new gql.TextMessage({\n      id: \"assistant-1\",\n      content: \"Hi!\",\n      role: gql.Role.Assistant,\n    });\n    const aguiMsgs = gqlToAGUI(gqlMsg);\n    const gqlMsgs2 = aguiToGQL(aguiMsgs);\n    // Should be equivalent in content, id, and role\n    expect(gqlMsgs2[0].id).toBe(gqlMsg.id);\n    expect((gqlMsgs2[0] as any).content).toBe(gqlMsg.content);\n    expect((gqlMsgs2[0] as any).role).toBe(gqlMsg.role);\n  });\n\n  test(\"tool message AGUI -> GQL -> AGUI\", () => {\n    const aguiMsg: agui.Message = {\n      id: \"tool-1\",\n      role: \"tool\",\n      content: \"Tool result\",\n      toolCallId: \"tool-call-1\",\n      toolName: \"testAction\",\n    };\n    const gqlMsgs = aguiToGQL(aguiMsg);\n    const aguiMsgs2 = gqlToAGUI(gqlMsgs);\n    expect(stripFunctions(aguiMsgs2[0])).toEqual(stripFunctions(aguiMsg));\n  });\n\n  test(\"tool message GQL -> AGUI -> GQL\", () => {\n    const gqlMsg = new gql.ResultMessage({\n      id: \"tool-1\",\n      result: \"Tool result\",\n      actionExecutionId: \"tool-call-1\",\n      actionName: \"testAction\",\n    });\n    const aguiMsgs = gqlToAGUI(gqlMsg);\n    const gqlMsgs2 = aguiToGQL(aguiMsgs);\n    expect(gqlMsgs2[0].id).toBe(gqlMsg.id);\n    expect((gqlMsgs2[0] as any).result).toBe(gqlMsg.result);\n    expect((gqlMsgs2[0] as any).actionExecutionId).toBe(gqlMsg.actionExecutionId);\n  });\n\n  test(\"action execution AGUI -> GQL -> AGUI\", () => {\n    const aguiMsg: agui.Message = {\n      id: \"assistant-1\",\n      role: \"assistant\",\n      content: \"Running action\",\n      toolCalls: [\n        {\n          id: \"tool-call-1\",\n          type: \"function\",\n          function: {\n            name: \"doSomething\",\n            arguments: JSON.stringify({ foo: \"bar\" }),\n          },\n        },\n      ],\n    };\n    const gqlMsgs = aguiToGQL(aguiMsg);\n    const aguiMsgs2 = gqlToAGUI(gqlMsgs);\n    // Should have an assistant message and an action execution message\n    expect(aguiMsgs2[0].role).toBe(\"assistant\");\n    expect(aguiMsgs2[1].role).toBe(\"assistant\");\n    // Only check toolCalls if present\n    if (\"toolCalls\" in aguiMsgs2[1]) {\n      expect((aguiMsgs2[1] as any).toolCalls[0].function.name).toBe(\"doSomething\");\n    }\n  });\n\n  test(\"action execution GQL -> AGUI -> GQL\", () => {\n    const actionExecMsg = new gql.ActionExecutionMessage({\n      id: \"tool-call-1\",\n      name: \"doSomething\",\n      arguments: { foo: \"bar\" },\n      parentMessageId: \"assistant-1\",\n    });\n    const aguiMsgs = gqlToAGUI([actionExecMsg]);\n    const gqlMsgs2 = aguiToGQL(aguiMsgs);\n    // The ActionExecutionMessage is at index 1, not index 0\n    expect(gqlMsgs2[1].id).toBe(\"tool-call-1\");\n    // The name should be extracted from the toolCall function name\n    expect((gqlMsgs2[1] as any).name).toBe(\"doSomething\");\n    expect((gqlMsgs2[1] as any).arguments).toEqual({ foo: \"bar\" });\n  });\n\n  test(\"agent state GQL -> AGUI -> GQL\", () => {\n    const agentStateMsg = new gql.AgentStateMessage({\n      id: \"agent-state-1\",\n      agentName: \"testAgent\",\n      state: { status: \"running\" },\n      role: gql.Role.Assistant,\n    });\n    const aguiMsgs = gqlToAGUI([agentStateMsg]);\n    const gqlMsgs2 = aguiToGQL(aguiMsgs);\n    expect(gqlMsgs2[0].id).toBe(\"agent-state-1\");\n    // The agentName should be preserved in the roundtrip\n    expect((gqlMsgs2[0] as any).agentName).toBe(\"testAgent\");\n  });\n\n  test(\"action execution with render function roundtrip\", () => {\n    const mockRender = vi.fn();\n    const aguiMsg: agui.Message = {\n      id: \"assistant-1\",\n      role: \"assistant\",\n      content: \"Running action\",\n      toolCalls: [\n        {\n          id: \"tool-call-1\",\n          type: \"function\",\n          function: {\n            name: \"doSomething\",\n            arguments: JSON.stringify({ foo: \"bar\" }),\n          },\n        },\n      ],\n      generativeUI: mockRender,\n    };\n    const actions: Record<string, any> = { doSomething: { name: \"doSomething\" } };\n    const gqlMsgs = aguiToGQL(aguiMsg, actions);\n    const aguiMsgs2 = gqlToAGUI(gqlMsgs, actions);\n    // The render function should be preserved in actions context\n    expect(typeof actions.doSomething.render).toBe(\"function\");\n    // The roundtripped message should have the same tool call\n    if (\"toolCalls\" in aguiMsgs2[1]) {\n      expect((aguiMsgs2[1] as any).toolCalls[0].function.name).toBe(\"doSomething\");\n    }\n  });\n\n  test(\"image message GQL -> AGUI -> GQL\", () => {\n    const gqlMsg = new gql.ImageMessage({\n      id: \"img-1\",\n      format: \"jpeg\",\n      bytes: \"somebase64string\",\n      role: gql.Role.User,\n    });\n    const aguiMsgs = gqlToAGUI(gqlMsg);\n    const gqlMsgs2 = aguiToGQL(aguiMsgs);\n    expect(gqlMsgs2[0].id).toBe(gqlMsg.id);\n    expect((gqlMsgs2[0] as any).format).toBe(gqlMsg.format);\n    expect((gqlMsgs2[0] as any).bytes).toBe(gqlMsg.bytes);\n    expect((gqlMsgs2[0] as any).role).toBe(gqlMsg.role);\n  });\n\n  test(\"image message AGUI -> GQL -> AGUI (assistant and user)\", () => {\n    // Assistant image message\n    const aguiAssistantImageMsg: agui.Message = {\n      id: \"img-assistant-1\",\n      role: \"assistant\",\n      image: {\n        format: \"jpeg\",\n        bytes: \"assistantbase64data\",\n      },\n      content: \"\", // required for type\n    };\n    const gqlAssistantMsgs = aguiToGQL(aguiAssistantImageMsg);\n    const aguiAssistantMsgs2 = gqlToAGUI(gqlAssistantMsgs);\n    expect(aguiAssistantMsgs2[0].id).toBe(aguiAssistantImageMsg.id);\n    expect(aguiAssistantMsgs2[0].role).toBe(\"assistant\");\n    expect((aguiAssistantMsgs2[0] as any).image.format).toBe(\"jpeg\");\n    expect((aguiAssistantMsgs2[0] as any).image.bytes).toBe(\"assistantbase64data\");\n\n    // User image message\n    const aguiUserImageMsg: agui.Message = {\n      id: \"img-user-1\",\n      role: \"user\",\n      image: {\n        format: \"png\",\n        bytes: \"userbase64data\",\n      },\n      content: \"\", // required for type\n    };\n    const gqlUserMsgs = aguiToGQL(aguiUserImageMsg);\n    const aguiUserMsgs2 = gqlToAGUI(gqlUserMsgs);\n    expect(aguiUserMsgs2[0].id).toBe(aguiUserImageMsg.id);\n    expect(aguiUserMsgs2[0].role).toBe(\"user\");\n    expect((aguiUserMsgs2[0] as any).image.format).toBe(\"png\");\n    expect((aguiUserMsgs2[0] as any).image.bytes).toBe(\"userbase64data\");\n  });\n});\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAOA,SAAS,eAAe,KAAe;AACrC,MAAI,OAAO,QAAQ;AAAY,WAAO;AACtC,MAAI,MAAM,QAAQ,GAAG;AAAG,WAAO,IAAI,IAAI,cAAc;AACrD,MAAI,OAAO,OAAO,QAAQ,UAAU;AAClC,UAAM,MAAW,CAAC;AAClB,eAAW,KAAK,KAAK;AACnB,UAAI,OAAO,IAAI,CAAC,MAAM,YAAY;AAChC,YAAI,CAAC,IAAI,eAAe,IAAI,CAAC,CAAC;AAAA,MAChC;AAAA,IACF;AACA,WAAO;AAAA,EACT;AACA,SAAO;AACT;AAEA,SAAS,gCAAgC,MAAM;AAC7C,OAAK,oCAAoC,MAAM;AAC7C,UAAM,UAAwB;AAAA,MAC5B,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,SAAS;AAAA,IACX;AACA,UAAM,UAAU,UAAU,OAAO;AACjC,UAAM,YAAY,UAAU,OAAO;AACnC,iBAAO,eAAe,UAAU,CAAC,CAAC,CAAC,EAAE,QAAQ,eAAe,OAAO,CAAC;AAAA,EACtE,CAAC;AAED,OAAK,mCAAmC,MAAM;AAC5C,UAAM,SAAS,IAAQ,YAAY;AAAA,MACjC,IAAI;AAAA,MACJ,SAAS;AAAA,MACT,MAAU,KAAK;AAAA,IACjB,CAAC;AACD,UAAM,WAAW,UAAU,MAAM;AACjC,UAAM,WAAW,UAAU,QAAQ;AAEnC,iBAAO,SAAS,CAAC,EAAE,EAAE,EAAE,KAAK,OAAO,EAAE;AACrC,iBAAQ,SAAS,CAAC,EAAU,OAAO,EAAE,KAAK,OAAO,OAAO;AACxD,iBAAQ,SAAS,CAAC,EAAU,IAAI,EAAE,KAAK,OAAO,IAAI;AAAA,EACpD,CAAC;AAED,OAAK,oCAAoC,MAAM;AAC7C,UAAM,UAAwB;AAAA,MAC5B,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,SAAS;AAAA,MACT,YAAY;AAAA,MACZ,UAAU;AAAA,IACZ;AACA,UAAM,UAAU,UAAU,OAAO;AACjC,UAAM,YAAY,UAAU,OAAO;AACnC,iBAAO,eAAe,UAAU,CAAC,CAAC,CAAC,EAAE,QAAQ,eAAe,OAAO,CAAC;AAAA,EACtE,CAAC;AAED,OAAK,mCAAmC,MAAM;AAC5C,UAAM,SAAS,IAAQ,cAAc;AAAA,MACnC,IAAI;AAAA,MACJ,QAAQ;AAAA,MACR,mBAAmB;AAAA,MACnB,YAAY;AAAA,IACd,CAAC;AACD,UAAM,WAAW,UAAU,MAAM;AACjC,UAAM,WAAW,UAAU,QAAQ;AACnC,iBAAO,SAAS,CAAC,EAAE,EAAE,EAAE,KAAK,OAAO,EAAE;AACrC,iBAAQ,SAAS,CAAC,EAAU,MAAM,EAAE,KAAK,OAAO,MAAM;AACtD,iBAAQ,SAAS,CAAC,EAAU,iBAAiB,EAAE,KAAK,OAAO,iBAAiB;AAAA,EAC9E,CAAC;AAED,OAAK,wCAAwC,MAAM;AACjD,UAAM,UAAwB;AAAA,MAC5B,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,SAAS;AAAA,MACT,WAAW;AAAA,QACT;AAAA,UACE,IAAI;AAAA,UACJ,MAAM;AAAA,UACN,UAAU;AAAA,YACR,MAAM;AAAA,YACN,WAAW,KAAK,UAAU,EAAE,KAAK,MAAM,CAAC;AAAA,UAC1C;AAAA,QACF;AAAA,MACF;AAAA,IACF;AACA,UAAM,UAAU,UAAU,OAAO;AACjC,UAAM,YAAY,UAAU,OAAO;AAEnC,iBAAO,UAAU,CAAC,EAAE,IAAI,EAAE,KAAK,WAAW;AAC1C,iBAAO,UAAU,CAAC,EAAE,IAAI,EAAE,KAAK,WAAW;AAE1C,QAAI,eAAe,UAAU,CAAC,GAAG;AAC/B,mBAAQ,UAAU,CAAC,EAAU,UAAU,CAAC,EAAE,SAAS,IAAI,EAAE,KAAK,aAAa;AAAA,IAC7E;AAAA,EACF,CAAC;AAED,OAAK,uCAAuC,MAAM;AAChD,UAAM,gBAAgB,IAAQ,uBAAuB;AAAA,MACnD,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,WAAW,EAAE,KAAK,MAAM;AAAA,MACxB,iBAAiB;AAAA,IACnB,CAAC;AACD,UAAM,WAAW,UAAU,CAAC,aAAa,CAAC;AAC1C,UAAM,WAAW,UAAU,QAAQ;AAEnC,iBAAO,SAAS,CAAC,EAAE,EAAE,EAAE,KAAK,aAAa;AAEzC,iBAAQ,SAAS,CAAC,EAAU,IAAI,EAAE,KAAK,aAAa;AACpD,iBAAQ,SAAS,CAAC,EAAU,SAAS,EAAE,QAAQ,EAAE,KAAK,MAAM,CAAC;AAAA,EAC/D,CAAC;AAED,OAAK,kCAAkC,MAAM;AAC3C,UAAM,gBAAgB,IAAQ,kBAAkB;AAAA,MAC9C,IAAI;AAAA,MACJ,WAAW;AAAA,MACX,OAAO,EAAE,QAAQ,UAAU;AAAA,MAC3B,MAAU,KAAK;AAAA,IACjB,CAAC;AACD,UAAM,WAAW,UAAU,CAAC,aAAa,CAAC;AAC1C,UAAM,WAAW,UAAU,QAAQ;AACnC,iBAAO,SAAS,CAAC,EAAE,EAAE,EAAE,KAAK,eAAe;AAE3C,iBAAQ,SAAS,CAAC,EAAU,SAAS,EAAE,KAAK,WAAW;AAAA,EACzD,CAAC;AAED,OAAK,mDAAmD,MAAM;AAC5D,UAAM,aAAa,GAAG,GAAG;AACzB,UAAM,UAAwB;AAAA,MAC5B,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,SAAS;AAAA,MACT,WAAW;AAAA,QACT;AAAA,UACE,IAAI;AAAA,UACJ,MAAM;AAAA,UACN,UAAU;AAAA,YACR,MAAM;AAAA,YACN,WAAW,KAAK,UAAU,EAAE,KAAK,MAAM,CAAC;AAAA,UAC1C;AAAA,QACF;AAAA,MACF;AAAA,MACA,cAAc;AAAA,IAChB;AACA,UAAM,UAA+B,EAAE,aAAa,EAAE,MAAM,cAAc,EAAE;AAC5E,UAAM,UAAU,UAAU,SAAS,OAAO;AAC1C,UAAM,YAAY,UAAU,SAAS,OAAO;AAE5C,iBAAO,OAAO,QAAQ,YAAY,MAAM,EAAE,KAAK,UAAU;AAEzD,QAAI,eAAe,UAAU,CAAC,GAAG;AAC/B,mBAAQ,UAAU,CAAC,EAAU,UAAU,CAAC,EAAE,SAAS,IAAI,EAAE,KAAK,aAAa;AAAA,IAC7E;AAAA,EACF,CAAC;AAED,OAAK,oCAAoC,MAAM;AAC7C,UAAM,SAAS,IAAQ,aAAa;AAAA,MAClC,IAAI;AAAA,MACJ,QAAQ;AAAA,MACR,OAAO;AAAA,MACP,MAAU,KAAK;AAAA,IACjB,CAAC;AACD,UAAM,WAAW,UAAU,MAAM;AACjC,UAAM,WAAW,UAAU,QAAQ;AACnC,iBAAO,SAAS,CAAC,EAAE,EAAE,EAAE,KAAK,OAAO,EAAE;AACrC,iBAAQ,SAAS,CAAC,EAAU,MAAM,EAAE,KAAK,OAAO,MAAM;AACtD,iBAAQ,SAAS,CAAC,EAAU,KAAK,EAAE,KAAK,OAAO,KAAK;AACpD,iBAAQ,SAAS,CAAC,EAAU,IAAI,EAAE,KAAK,OAAO,IAAI;AAAA,EACpD,CAAC;AAED,OAAK,0DAA0D,MAAM;AAEnE,UAAM,wBAAsC;AAAA,MAC1C,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,OAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO;AAAA,MACT;AAAA,MACA,SAAS;AAAA;AAAA,IACX;AACA,UAAM,mBAAmB,UAAU,qBAAqB;AACxD,UAAM,qBAAqB,UAAU,gBAAgB;AACrD,iBAAO,mBAAmB,CAAC,EAAE,EAAE,EAAE,KAAK,sBAAsB,EAAE;AAC9D,iBAAO,mBAAmB,CAAC,EAAE,IAAI,EAAE,KAAK,WAAW;AACnD,iBAAQ,mBAAmB,CAAC,EAAU,MAAM,MAAM,EAAE,KAAK,MAAM;AAC/D,iBAAQ,mBAAmB,CAAC,EAAU,MAAM,KAAK,EAAE,KAAK,qBAAqB;AAG7E,UAAM,mBAAiC;AAAA,MACrC,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,OAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO;AAAA,MACT;AAAA,MACA,SAAS;AAAA;AAAA,IACX;AACA,UAAM,cAAc,UAAU,gBAAgB;AAC9C,UAAM,gBAAgB,UAAU,WAAW;AAC3C,iBAAO,cAAc,CAAC,EAAE,EAAE,EAAE,KAAK,iBAAiB,EAAE;AACpD,iBAAO,cAAc,CAAC,EAAE,IAAI,EAAE,KAAK,MAAM;AACzC,iBAAQ,cAAc,CAAC,EAAU,MAAM,MAAM,EAAE,KAAK,KAAK;AACzD,iBAAQ,cAAc,CAAC,EAAU,MAAM,KAAK,EAAE,KAAK,gBAAgB;AAAA,EACrE,CAAC;AACH,CAAC;","names":[]}
+{"version":3,"sources":["../../src/message-conversion/roundtrip-conversion.test.ts"],"sourcesContent":["import { describe, test, expect, vi } from \"vitest\";\nimport * as gql from \"../client\";\nimport agui from \"@copilotkit/shared\";\nimport { aguiToGQL } from \"./agui-to-gql\";\nimport { gqlToAGUI } from \"./gql-to-agui\";\n\n// Helper to strip functions for deep equality\nfunction stripFunctions(obj: any): any {\n  if (typeof obj === \"function\") return undefined;\n  if (Array.isArray(obj)) return obj.map(stripFunctions);\n  if (obj && typeof obj === \"object\") {\n    const out: any = {};\n    for (const k in obj) {\n      if (typeof obj[k] !== \"function\") {\n        out[k] = stripFunctions(obj[k]);\n      }\n    }\n    return out;\n  }\n  return obj;\n}\n\ndescribe(\"roundtrip message conversion\", () => {\n  test(\"text message AGUI -> GQL -> AGUI\", () => {\n    const aguiMsg: agui.Message = {\n      id: \"user-1\",\n      role: \"user\",\n      content: \"Hello!\",\n    };\n    const gqlMsgs = aguiToGQL(aguiMsg);\n    const aguiMsgs2 = gqlToAGUI(gqlMsgs);\n    expect(stripFunctions(aguiMsgs2[0])).toEqual(stripFunctions(aguiMsg));\n  });\n\n  test(\"text message GQL -> AGUI -> GQL\", () => {\n    const gqlMsg = new gql.TextMessage({\n      id: \"assistant-1\",\n      content: \"Hi!\",\n      role: gql.Role.Assistant,\n    });\n    const aguiMsgs = gqlToAGUI(gqlMsg);\n    const gqlMsgs2 = aguiToGQL(aguiMsgs);\n    // Should be equivalent in content, id, and role\n    expect(gqlMsgs2[0].id).toBe(gqlMsg.id);\n    expect((gqlMsgs2[0] as any).content).toBe(gqlMsg.content);\n    expect((gqlMsgs2[0] as any).role).toBe(gqlMsg.role);\n  });\n\n  test(\"tool message AGUI -> GQL -> AGUI\", () => {\n    const aguiMsg: agui.Message = {\n      id: \"tool-1\",\n      role: \"tool\",\n      content: \"Tool result\",\n      toolCallId: \"tool-call-1\",\n      toolName: \"testAction\",\n    };\n    const gqlMsgs = aguiToGQL(aguiMsg);\n    const aguiMsgs2 = gqlToAGUI(gqlMsgs);\n    expect(stripFunctions(aguiMsgs2[0])).toEqual(stripFunctions(aguiMsg));\n  });\n\n  test(\"tool message GQL -> AGUI -> GQL\", () => {\n    const gqlMsg = new gql.ResultMessage({\n      id: \"tool-1\",\n      result: \"Tool result\",\n      actionExecutionId: \"tool-call-1\",\n      actionName: \"testAction\",\n    });\n    const aguiMsgs = gqlToAGUI(gqlMsg);\n    const gqlMsgs2 = aguiToGQL(aguiMsgs);\n    expect(gqlMsgs2[0].id).toBe(gqlMsg.id);\n    expect((gqlMsgs2[0] as any).result).toBe(gqlMsg.result);\n    expect((gqlMsgs2[0] as any).actionExecutionId).toBe(gqlMsg.actionExecutionId);\n  });\n\n  test(\"action execution AGUI -> GQL -> AGUI\", () => {\n    const aguiMsg: agui.Message = {\n      id: \"assistant-1\",\n      role: \"assistant\",\n      content: \"Running action\",\n      toolCalls: [\n        {\n          id: \"tool-call-1\",\n          type: \"function\",\n          function: {\n            name: \"doSomething\",\n            arguments: JSON.stringify({ foo: \"bar\" }),\n          },\n        },\n      ],\n    };\n    const gqlMsgs = aguiToGQL(aguiMsg);\n    const aguiMsgs2 = gqlToAGUI(gqlMsgs);\n    // Should have an assistant message and an action execution message\n    expect(aguiMsgs2[0].role).toBe(\"assistant\");\n    expect(aguiMsgs2[1].role).toBe(\"assistant\");\n    // Only check toolCalls if present\n    if (\"toolCalls\" in aguiMsgs2[1]) {\n      expect((aguiMsgs2[1] as any).toolCalls[0].function.name).toBe(\"doSomething\");\n    }\n  });\n\n  test(\"action execution GQL -> AGUI -> GQL\", () => {\n    const actionExecMsg = new gql.ActionExecutionMessage({\n      id: \"tool-call-1\",\n      name: \"doSomething\",\n      arguments: { foo: \"bar\" },\n      parentMessageId: \"assistant-1\",\n    });\n    const aguiMsgs = gqlToAGUI([actionExecMsg]);\n    const gqlMsgs2 = aguiToGQL(aguiMsgs);\n    // The ActionExecutionMessage is at index 1, not index 0\n    expect(gqlMsgs2[1].id).toBe(\"tool-call-1\");\n    // The name should be extracted from the toolCall function name\n    expect((gqlMsgs2[1] as any).name).toBe(\"doSomething\");\n    expect((gqlMsgs2[1] as any).arguments).toEqual({ foo: \"bar\" });\n  });\n\n  test(\"agent state GQL -> AGUI -> GQL\", () => {\n    const agentStateMsg = new gql.AgentStateMessage({\n      id: \"agent-state-1\",\n      agentName: \"testAgent\",\n      state: { status: \"running\" },\n      role: gql.Role.Assistant,\n    });\n    const aguiMsgs = gqlToAGUI([agentStateMsg]);\n    const gqlMsgs2 = aguiToGQL(aguiMsgs);\n    expect(gqlMsgs2[0].id).toBe(\"agent-state-1\");\n    // The agentName should be preserved in the roundtrip\n    expect((gqlMsgs2[0] as any).agentName).toBe(\"testAgent\");\n  });\n\n  test(\"action execution with render function roundtrip\", () => {\n    const mockRender = vi.fn();\n    const aguiMsg: agui.Message = {\n      id: \"assistant-1\",\n      role: \"assistant\",\n      content: \"Running action\",\n      toolCalls: [\n        {\n          id: \"tool-call-1\",\n          type: \"function\",\n          function: {\n            name: \"doSomething\",\n            arguments: JSON.stringify({ foo: \"bar\" }),\n          },\n        },\n      ],\n      generativeUI: mockRender,\n    };\n    const actions: Record<string, any> = { doSomething: { name: \"doSomething\" } };\n    const gqlMsgs = aguiToGQL(aguiMsg, actions);\n    const aguiMsgs2 = gqlToAGUI(gqlMsgs, actions);\n    // The render function should be preserved in actions context\n    expect(typeof actions.doSomething.render).toBe(\"function\");\n    // The roundtripped message should have the same tool call\n    if (\"toolCalls\" in aguiMsgs2[1]) {\n      expect((aguiMsgs2[1] as any).toolCalls[0].function.name).toBe(\"doSomething\");\n    }\n  });\n\n  test(\"image message GQL -> AGUI -> GQL\", () => {\n    const gqlMsg = new gql.ImageMessage({\n      id: \"img-1\",\n      format: \"jpeg\",\n      bytes: \"somebase64string\",\n      role: gql.Role.User,\n    });\n    const aguiMsgs = gqlToAGUI(gqlMsg);\n    const gqlMsgs2 = aguiToGQL(aguiMsgs);\n    expect(gqlMsgs2[0].id).toBe(gqlMsg.id);\n    expect((gqlMsgs2[0] as any).format).toBe(gqlMsg.format);\n    expect((gqlMsgs2[0] as any).bytes).toBe(gqlMsg.bytes);\n    expect((gqlMsgs2[0] as any).role).toBe(gqlMsg.role);\n  });\n\n  test(\"image message AGUI -> GQL -> AGUI (assistant and user)\", () => {\n    // Assistant image message\n    const aguiAssistantImageMsg: agui.Message = {\n      id: \"img-assistant-1\",\n      role: \"assistant\",\n      image: {\n        format: \"jpeg\",\n        bytes: \"assistantbase64data\",\n      },\n      content: \"\", // required for type\n    };\n    const gqlAssistantMsgs = aguiToGQL(aguiAssistantImageMsg);\n    const aguiAssistantMsgs2 = gqlToAGUI(gqlAssistantMsgs);\n    expect(aguiAssistantMsgs2[0].id).toBe(aguiAssistantImageMsg.id);\n    expect(aguiAssistantMsgs2[0].role).toBe(\"assistant\");\n    expect((aguiAssistantMsgs2[0] as any).image.format).toBe(\"jpeg\");\n    expect((aguiAssistantMsgs2[0] as any).image.bytes).toBe(\"assistantbase64data\");\n\n    // User image message\n    const aguiUserImageMsg: agui.Message = {\n      id: \"img-user-1\",\n      role: \"user\",\n      image: {\n        format: \"png\",\n        bytes: \"userbase64data\",\n      },\n      content: \"\", // required for type\n    };\n    const gqlUserMsgs = aguiToGQL(aguiUserImageMsg);\n    const aguiUserMsgs2 = gqlToAGUI(gqlUserMsgs);\n    expect(aguiUserMsgs2[0].id).toBe(aguiUserImageMsg.id);\n    expect(aguiUserMsgs2[0].role).toBe(\"user\");\n    expect((aguiUserMsgs2[0] as any).image.format).toBe(\"png\");\n    expect((aguiUserMsgs2[0] as any).image.bytes).toBe(\"userbase64data\");\n  });\n\n  test(\"wild card action roundtrip conversion\", () => {\n    const mockRender = vi.fn((props) => `Wildcard rendered: ${props.args.test}`);\n    const aguiMsg: agui.Message = {\n      id: \"assistant-wildcard-1\",\n      role: \"assistant\",\n      content: \"Running wild card action\",\n      toolCalls: [\n        {\n          id: \"tool-call-wildcard-1\",\n          type: \"function\",\n          function: {\n            name: \"unknownAction\",\n            arguments: JSON.stringify({ test: \"wildcard-value\" }),\n          },\n        },\n      ],\n      generativeUI: mockRender,\n    };\n\n    const actions: Record<string, any> = {\n      \"*\": { name: \"*\" },\n    };\n\n    // AGUI -> GQL -> AGUI roundtrip\n    const gqlMsgs = aguiToGQL(aguiMsg, actions);\n    const aguiMsgs2 = gqlToAGUI(gqlMsgs, actions);\n\n    // Verify the wild card action preserved the render function\n    expect(typeof actions[\"*\"].render).toBe(\"function\");\n    expect(actions[\"*\"].render).toBe(mockRender);\n\n    // Verify the roundtripped message structure\n    expect(aguiMsgs2).toHaveLength(2);\n    expect(aguiMsgs2[0].role).toBe(\"assistant\");\n    expect(aguiMsgs2[1].role).toBe(\"assistant\");\n\n    // Check that the tool call is preserved\n    if (\"toolCalls\" in aguiMsgs2[1]) {\n      expect((aguiMsgs2[1] as any).toolCalls[0].function.name).toBe(\"unknownAction\");\n      expect((aguiMsgs2[1] as any).toolCalls[0].function.arguments).toBe(\n        '{\"test\":\"wildcard-value\"}',\n      );\n    }\n  });\n\n  test(\"wild card action with specific action priority roundtrip\", () => {\n    const mockRender = vi.fn((props) => `Specific action rendered: ${props.args.test}`);\n    const aguiMsg: agui.Message = {\n      id: \"assistant-priority-1\",\n      role: \"assistant\",\n      content: \"Running specific action\",\n      toolCalls: [\n        {\n          id: \"tool-call-priority-1\",\n          type: \"function\",\n          function: {\n            name: \"specificAction\",\n            arguments: JSON.stringify({ test: \"specific-value\" }),\n          },\n        },\n      ],\n      generativeUI: mockRender,\n    };\n\n    const actions: Record<string, any> = {\n      specificAction: { name: \"specificAction\" },\n      \"*\": { name: \"*\" },\n    };\n\n    // AGUI -> GQL -> AGUI roundtrip\n    const gqlMsgs = aguiToGQL(aguiMsg, actions);\n    const aguiMsgs2 = gqlToAGUI(gqlMsgs, actions);\n\n    // Verify the specific action preserved the render function (not wild card)\n    expect(typeof actions.specificAction.render).toBe(\"function\");\n    expect(actions.specificAction.render).toBe(mockRender);\n    expect(actions[\"*\"].render).toBeUndefined();\n\n    // Verify the roundtripped message structure\n    expect(aguiMsgs2).toHaveLength(2);\n    expect(aguiMsgs2[0].role).toBe(\"assistant\");\n    expect(aguiMsgs2[1].role).toBe(\"assistant\");\n\n    // Check that the tool call is preserved\n    if (\"toolCalls\" in aguiMsgs2[1]) {\n      expect((aguiMsgs2[1] as any).toolCalls[0].function.name).toBe(\"specificAction\");\n      expect((aguiMsgs2[1] as any).toolCalls[0].function.arguments).toBe(\n        '{\"test\":\"specific-value\"}',\n      );\n    }\n  });\n\n  test(\"wild card action GQL -> AGUI -> GQL roundtrip\", () => {\n    const actionExecMsg = new gql.ActionExecutionMessage({\n      id: \"wildcard-action-1\",\n      name: \"unknownAction\",\n      arguments: { test: \"wildcard-gql-value\" },\n      parentMessageId: \"assistant-1\",\n    });\n\n    const actions: Record<string, any> = {\n      \"*\": {\n        name: \"*\",\n        render: vi.fn((props) => `GQL wildcard rendered: ${props.args.test}`),\n      },\n    };\n\n    // GQL -> AGUI -> GQL roundtrip\n    const aguiMsgs = gqlToAGUI([actionExecMsg], actions);\n    const gqlMsgs2 = aguiToGQL(aguiMsgs, actions);\n\n    // When converting ActionExecutionMessage to AGUI and back, we get:\n    // 1. A TextMessage (assistant message with toolCalls)\n    // 2. An ActionExecutionMessage (the tool call itself)\n    expect(gqlMsgs2).toHaveLength(2);\n    expect(gqlMsgs2[0].id).toBe(\"wildcard-action-1\");\n    expect((gqlMsgs2[0] as any).role).toBe(gql.Role.Assistant);\n    expect(gqlMsgs2[1].id).toBe(\"wildcard-action-1\");\n    expect((gqlMsgs2[1] as any).name).toBe(\"unknownAction\");\n    expect((gqlMsgs2[1] as any).arguments).toEqual({ test: \"wildcard-gql-value\" });\n  });\n});\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAOA,SAAS,eAAe,KAAe;AACrC,MAAI,OAAO,QAAQ;AAAY,WAAO;AACtC,MAAI,MAAM,QAAQ,GAAG;AAAG,WAAO,IAAI,IAAI,cAAc;AACrD,MAAI,OAAO,OAAO,QAAQ,UAAU;AAClC,UAAM,MAAW,CAAC;AAClB,eAAW,KAAK,KAAK;AACnB,UAAI,OAAO,IAAI,CAAC,MAAM,YAAY;AAChC,YAAI,CAAC,IAAI,eAAe,IAAI,CAAC,CAAC;AAAA,MAChC;AAAA,IACF;AACA,WAAO;AAAA,EACT;AACA,SAAO;AACT;AAEA,SAAS,gCAAgC,MAAM;AAC7C,OAAK,oCAAoC,MAAM;AAC7C,UAAM,UAAwB;AAAA,MAC5B,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,SAAS;AAAA,IACX;AACA,UAAM,UAAU,UAAU,OAAO;AACjC,UAAM,YAAY,UAAU,OAAO;AACnC,iBAAO,eAAe,UAAU,CAAC,CAAC,CAAC,EAAE,QAAQ,eAAe,OAAO,CAAC;AAAA,EACtE,CAAC;AAED,OAAK,mCAAmC,MAAM;AAC5C,UAAM,SAAS,IAAQ,YAAY;AAAA,MACjC,IAAI;AAAA,MACJ,SAAS;AAAA,MACT,MAAU,KAAK;AAAA,IACjB,CAAC;AACD,UAAM,WAAW,UAAU,MAAM;AACjC,UAAM,WAAW,UAAU,QAAQ;AAEnC,iBAAO,SAAS,CAAC,EAAE,EAAE,EAAE,KAAK,OAAO,EAAE;AACrC,iBAAQ,SAAS,CAAC,EAAU,OAAO,EAAE,KAAK,OAAO,OAAO;AACxD,iBAAQ,SAAS,CAAC,EAAU,IAAI,EAAE,KAAK,OAAO,IAAI;AAAA,EACpD,CAAC;AAED,OAAK,oCAAoC,MAAM;AAC7C,UAAM,UAAwB;AAAA,MAC5B,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,SAAS;AAAA,MACT,YAAY;AAAA,MACZ,UAAU;AAAA,IACZ;AACA,UAAM,UAAU,UAAU,OAAO;AACjC,UAAM,YAAY,UAAU,OAAO;AACnC,iBAAO,eAAe,UAAU,CAAC,CAAC,CAAC,EAAE,QAAQ,eAAe,OAAO,CAAC;AAAA,EACtE,CAAC;AAED,OAAK,mCAAmC,MAAM;AAC5C,UAAM,SAAS,IAAQ,cAAc;AAAA,MACnC,IAAI;AAAA,MACJ,QAAQ;AAAA,MACR,mBAAmB;AAAA,MACnB,YAAY;AAAA,IACd,CAAC;AACD,UAAM,WAAW,UAAU,MAAM;AACjC,UAAM,WAAW,UAAU,QAAQ;AACnC,iBAAO,SAAS,CAAC,EAAE,EAAE,EAAE,KAAK,OAAO,EAAE;AACrC,iBAAQ,SAAS,CAAC,EAAU,MAAM,EAAE,KAAK,OAAO,MAAM;AACtD,iBAAQ,SAAS,CAAC,EAAU,iBAAiB,EAAE,KAAK,OAAO,iBAAiB;AAAA,EAC9E,CAAC;AAED,OAAK,wCAAwC,MAAM;AACjD,UAAM,UAAwB;AAAA,MAC5B,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,SAAS;AAAA,MACT,WAAW;AAAA,QACT;AAAA,UACE,IAAI;AAAA,UACJ,MAAM;AAAA,UACN,UAAU;AAAA,YACR,MAAM;AAAA,YACN,WAAW,KAAK,UAAU,EAAE,KAAK,MAAM,CAAC;AAAA,UAC1C;AAAA,QACF;AAAA,MACF;AAAA,IACF;AACA,UAAM,UAAU,UAAU,OAAO;AACjC,UAAM,YAAY,UAAU,OAAO;AAEnC,iBAAO,UAAU,CAAC,EAAE,IAAI,EAAE,KAAK,WAAW;AAC1C,iBAAO,UAAU,CAAC,EAAE,IAAI,EAAE,KAAK,WAAW;AAE1C,QAAI,eAAe,UAAU,CAAC,GAAG;AAC/B,mBAAQ,UAAU,CAAC,EAAU,UAAU,CAAC,EAAE,SAAS,IAAI,EAAE,KAAK,aAAa;AAAA,IAC7E;AAAA,EACF,CAAC;AAED,OAAK,uCAAuC,MAAM;AAChD,UAAM,gBAAgB,IAAQ,uBAAuB;AAAA,MACnD,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,WAAW,EAAE,KAAK,MAAM;AAAA,MACxB,iBAAiB;AAAA,IACnB,CAAC;AACD,UAAM,WAAW,UAAU,CAAC,aAAa,CAAC;AAC1C,UAAM,WAAW,UAAU,QAAQ;AAEnC,iBAAO,SAAS,CAAC,EAAE,EAAE,EAAE,KAAK,aAAa;AAEzC,iBAAQ,SAAS,CAAC,EAAU,IAAI,EAAE,KAAK,aAAa;AACpD,iBAAQ,SAAS,CAAC,EAAU,SAAS,EAAE,QAAQ,EAAE,KAAK,MAAM,CAAC;AAAA,EAC/D,CAAC;AAED,OAAK,kCAAkC,MAAM;AAC3C,UAAM,gBAAgB,IAAQ,kBAAkB;AAAA,MAC9C,IAAI;AAAA,MACJ,WAAW;AAAA,MACX,OAAO,EAAE,QAAQ,UAAU;AAAA,MAC3B,MAAU,KAAK;AAAA,IACjB,CAAC;AACD,UAAM,WAAW,UAAU,CAAC,aAAa,CAAC;AAC1C,UAAM,WAAW,UAAU,QAAQ;AACnC,iBAAO,SAAS,CAAC,EAAE,EAAE,EAAE,KAAK,eAAe;AAE3C,iBAAQ,SAAS,CAAC,EAAU,SAAS,EAAE,KAAK,WAAW;AAAA,EACzD,CAAC;AAED,OAAK,mDAAmD,MAAM;AAC5D,UAAM,aAAa,GAAG,GAAG;AACzB,UAAM,UAAwB;AAAA,MAC5B,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,SAAS;AAAA,MACT,WAAW;AAAA,QACT;AAAA,UACE,IAAI;AAAA,UACJ,MAAM;AAAA,UACN,UAAU;AAAA,YACR,MAAM;AAAA,YACN,WAAW,KAAK,UAAU,EAAE,KAAK,MAAM,CAAC;AAAA,UAC1C;AAAA,QACF;AAAA,MACF;AAAA,MACA,cAAc;AAAA,IAChB;AACA,UAAM,UAA+B,EAAE,aAAa,EAAE,MAAM,cAAc,EAAE;AAC5E,UAAM,UAAU,UAAU,SAAS,OAAO;AAC1C,UAAM,YAAY,UAAU,SAAS,OAAO;AAE5C,iBAAO,OAAO,QAAQ,YAAY,MAAM,EAAE,KAAK,UAAU;AAEzD,QAAI,eAAe,UAAU,CAAC,GAAG;AAC/B,mBAAQ,UAAU,CAAC,EAAU,UAAU,CAAC,EAAE,SAAS,IAAI,EAAE,KAAK,aAAa;AAAA,IAC7E;AAAA,EACF,CAAC;AAED,OAAK,oCAAoC,MAAM;AAC7C,UAAM,SAAS,IAAQ,aAAa;AAAA,MAClC,IAAI;AAAA,MACJ,QAAQ;AAAA,MACR,OAAO;AAAA,MACP,MAAU,KAAK;AAAA,IACjB,CAAC;AACD,UAAM,WAAW,UAAU,MAAM;AACjC,UAAM,WAAW,UAAU,QAAQ;AACnC,iBAAO,SAAS,CAAC,EAAE,EAAE,EAAE,KAAK,OAAO,EAAE;AACrC,iBAAQ,SAAS,CAAC,EAAU,MAAM,EAAE,KAAK,OAAO,MAAM;AACtD,iBAAQ,SAAS,CAAC,EAAU,KAAK,EAAE,KAAK,OAAO,KAAK;AACpD,iBAAQ,SAAS,CAAC,EAAU,IAAI,EAAE,KAAK,OAAO,IAAI;AAAA,EACpD,CAAC;AAED,OAAK,0DAA0D,MAAM;AAEnE,UAAM,wBAAsC;AAAA,MAC1C,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,OAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO;AAAA,MACT;AAAA,MACA,SAAS;AAAA;AAAA,IACX;AACA,UAAM,mBAAmB,UAAU,qBAAqB;AACxD,UAAM,qBAAqB,UAAU,gBAAgB;AACrD,iBAAO,mBAAmB,CAAC,EAAE,EAAE,EAAE,KAAK,sBAAsB,EAAE;AAC9D,iBAAO,mBAAmB,CAAC,EAAE,IAAI,EAAE,KAAK,WAAW;AACnD,iBAAQ,mBAAmB,CAAC,EAAU,MAAM,MAAM,EAAE,KAAK,MAAM;AAC/D,iBAAQ,mBAAmB,CAAC,EAAU,MAAM,KAAK,EAAE,KAAK,qBAAqB;AAG7E,UAAM,mBAAiC;AAAA,MACrC,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,OAAO;AAAA,QACL,QAAQ;AAAA,QACR,OAAO;AAAA,MACT;AAAA,MACA,SAAS;AAAA;AAAA,IACX;AACA,UAAM,cAAc,UAAU,gBAAgB;AAC9C,UAAM,gBAAgB,UAAU,WAAW;AAC3C,iBAAO,cAAc,CAAC,EAAE,EAAE,EAAE,KAAK,iBAAiB,EAAE;AACpD,iBAAO,cAAc,CAAC,EAAE,IAAI,EAAE,KAAK,MAAM;AACzC,iBAAQ,cAAc,CAAC,EAAU,MAAM,MAAM,EAAE,KAAK,KAAK;AACzD,iBAAQ,cAAc,CAAC,EAAU,MAAM,KAAK,EAAE,KAAK,gBAAgB;AAAA,EACrE,CAAC;AAED,OAAK,yCAAyC,MAAM;AAClD,UAAM,aAAa,GAAG,GAAG,CAAC,UAAU,sBAAsB,MAAM,KAAK,MAAM;AAC3E,UAAM,UAAwB;AAAA,MAC5B,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,SAAS;AAAA,MACT,WAAW;AAAA,QACT;AAAA,UACE,IAAI;AAAA,UACJ,MAAM;AAAA,UACN,UAAU;AAAA,YACR,MAAM;AAAA,YACN,WAAW,KAAK,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAAA,UACtD;AAAA,QACF;AAAA,MACF;AAAA,MACA,cAAc;AAAA,IAChB;AAEA,UAAM,UAA+B;AAAA,MACnC,KAAK,EAAE,MAAM,IAAI;AAAA,IACnB;AAGA,UAAM,UAAU,UAAU,SAAS,OAAO;AAC1C,UAAM,YAAY,UAAU,SAAS,OAAO;AAG5C,iBAAO,OAAO,QAAQ,GAAG,EAAE,MAAM,EAAE,KAAK,UAAU;AAClD,iBAAO,QAAQ,GAAG,EAAE,MAAM,EAAE,KAAK,UAAU;AAG3C,iBAAO,SAAS,EAAE,aAAa,CAAC;AAChC,iBAAO,UAAU,CAAC,EAAE,IAAI,EAAE,KAAK,WAAW;AAC1C,iBAAO,UAAU,CAAC,EAAE,IAAI,EAAE,KAAK,WAAW;AAG1C,QAAI,eAAe,UAAU,CAAC,GAAG;AAC/B,mBAAQ,UAAU,CAAC,EAAU,UAAU,CAAC,EAAE,SAAS,IAAI,EAAE,KAAK,eAAe;AAC7E,mBAAQ,UAAU,CAAC,EAAU,UAAU,CAAC,EAAE,SAAS,SAAS,EAAE;AAAA,QAC5D;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAED,OAAK,4DAA4D,MAAM;AACrE,UAAM,aAAa,GAAG,GAAG,CAAC,UAAU,6BAA6B,MAAM,KAAK,MAAM;AAClF,UAAM,UAAwB;AAAA,MAC5B,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,SAAS;AAAA,MACT,WAAW;AAAA,QACT;AAAA,UACE,IAAI;AAAA,UACJ,MAAM;AAAA,UACN,UAAU;AAAA,YACR,MAAM;AAAA,YACN,WAAW,KAAK,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAAA,UACtD;AAAA,QACF;AAAA,MACF;AAAA,MACA,cAAc;AAAA,IAChB;AAEA,UAAM,UAA+B;AAAA,MACnC,gBAAgB,EAAE,MAAM,iBAAiB;AAAA,MACzC,KAAK,EAAE,MAAM,IAAI;AAAA,IACnB;AAGA,UAAM,UAAU,UAAU,SAAS,OAAO;AAC1C,UAAM,YAAY,UAAU,SAAS,OAAO;AAG5C,iBAAO,OAAO,QAAQ,eAAe,MAAM,EAAE,KAAK,UAAU;AAC5D,iBAAO,QAAQ,eAAe,MAAM,EAAE,KAAK,UAAU;AACrD,iBAAO,QAAQ,GAAG,EAAE,MAAM,EAAE,cAAc;AAG1C,iBAAO,SAAS,EAAE,aAAa,CAAC;AAChC,iBAAO,UAAU,CAAC,EAAE,IAAI,EAAE,KAAK,WAAW;AAC1C,iBAAO,UAAU,CAAC,EAAE,IAAI,EAAE,KAAK,WAAW;AAG1C,QAAI,eAAe,UAAU,CAAC,GAAG;AAC/B,mBAAQ,UAAU,CAAC,EAAU,UAAU,CAAC,EAAE,SAAS,IAAI,EAAE,KAAK,gBAAgB;AAC9E,mBAAQ,UAAU,CAAC,EAAU,UAAU,CAAC,EAAE,SAAS,SAAS,EAAE;AAAA,QAC5D;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAED,OAAK,iDAAiD,MAAM;AAC1D,UAAM,gBAAgB,IAAQ,uBAAuB;AAAA,MACnD,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,WAAW,EAAE,MAAM,qBAAqB;AAAA,MACxC,iBAAiB;AAAA,IACnB,CAAC;AAED,UAAM,UAA+B;AAAA,MACnC,KAAK;AAAA,QACH,MAAM;AAAA,QACN,QAAQ,GAAG,GAAG,CAAC,UAAU,0BAA0B,MAAM,KAAK,MAAM;AAAA,MACtE;AAAA,IACF;AAGA,UAAM,WAAW,UAAU,CAAC,aAAa,GAAG,OAAO;AACnD,UAAM,WAAW,UAAU,UAAU,OAAO;AAK5C,iBAAO,QAAQ,EAAE,aAAa,CAAC;AAC/B,iBAAO,SAAS,CAAC,EAAE,EAAE,EAAE,KAAK,mBAAmB;AAC/C,iBAAQ,SAAS,CAAC,EAAU,IAAI,EAAE,KAAS,KAAK,SAAS;AACzD,iBAAO,SAAS,CAAC,EAAE,EAAE,EAAE,KAAK,mBAAmB;AAC/C,iBAAQ,SAAS,CAAC,EAAU,IAAI,EAAE,KAAK,eAAe;AACtD,iBAAQ,SAAS,CAAC,EAAU,SAAS,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAAA,EAC/E,CAAC;AACH,CAAC;","names":[]}

package/package.json

@@ -9,7 +9,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "1.10.0",
+  "version": "1.10.1-next.0",
   "sideEffects": false,
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",
@@ -45,13 +45,13 @@
     "tsup": "^6.7.0",
     "typescript": "^5.4.5",
     "vitest": "^3.1.3",
-    "@copilotkit/runtime": "1.10.0"
+    "@copilotkit/runtime": "1.10.1-next.0"
   },
   "dependencies": {
     "@urql/core": "^5.0.3",
     "untruncate-json": "^0.0.1",
     "urql": "^4.1.0",
-    "@copilotkit/shared": "1.10.0"
+    "@copilotkit/shared": "1.10.1-next.0"
   },
   "keywords": [
     "copilotkit",

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

@@ -637,4 +637,430 @@ describe("agui-to-gql", () => {
       expect(result.role).toBe(gql.Role.User);
     });
   });
+
+  describe("Wild Card Actions", () => {
+    test("should preserve render function for specific action", () => {
+      const mockRender = () => "Specific Action Render";
+      const aguiMessage: agui.Message = {
+        id: "assistant-1",
+        role: "assistant",
+        content: "I'll execute a function",
+        toolCalls: [
+          {
+            id: "tool-call-1",
+            type: "function",
+            function: {
+              name: "specificAction",
+              arguments: JSON.stringify({ param: "value" }),
+            },
+          },
+        ],
+        generativeUI: mockRender,
+      };
+
+      const actions: Record<string, any> = {
+        specificAction: { name: "specificAction" },
+      };
+
+      const result = aguiToGQL(aguiMessage, actions);
+
+      expect(result).toHaveLength(2);
+      expect(result[0]).toBeInstanceOf(gql.TextMessage);
+      expect(result[1]).toBeInstanceOf(gql.ActionExecutionMessage);
+      expect(actions.specificAction.render).toBe(mockRender);
+    });
+
+    test("should preserve render function for wild card action", () => {
+      const mockRender = () => "Wild Card Action Render";
+      const aguiMessage: agui.Message = {
+        id: "assistant-2",
+        role: "assistant",
+        content: "I'll execute an unknown function",
+        toolCalls: [
+          {
+            id: "tool-call-2",
+            type: "function",
+            function: {
+              name: "unknownAction",
+              arguments: JSON.stringify({ param: "value" }),
+            },
+          },
+        ],
+        generativeUI: mockRender,
+      };
+
+      const actions: Record<string, any> = {
+        "*": { name: "*" },
+      };
+
+      const result = aguiToGQL(aguiMessage, actions);
+
+      expect(result).toHaveLength(2);
+      expect(result[0]).toBeInstanceOf(gql.TextMessage);
+      expect(result[1]).toBeInstanceOf(gql.ActionExecutionMessage);
+      expect(actions["*"].render).toBe(mockRender);
+    });
+
+    test("should prioritize specific action over wild card action", () => {
+      const mockRender = () => "Prioritized Render";
+      const aguiMessage: agui.Message = {
+        id: "assistant-3",
+        role: "assistant",
+        content: "I'll execute a function",
+        toolCalls: [
+          {
+            id: "tool-call-3",
+            type: "function",
+            function: {
+              name: "specificAction",
+              arguments: JSON.stringify({ param: "value" }),
+            },
+          },
+        ],
+        generativeUI: mockRender,
+      };
+
+      const actions: Record<string, any> = {
+        specificAction: { name: "specificAction" },
+        "*": { name: "*" },
+      };
+
+      const result = aguiToGQL(aguiMessage, actions);
+
+      expect(result).toHaveLength(2);
+      expect(result[0]).toBeInstanceOf(gql.TextMessage);
+      expect(result[1]).toBeInstanceOf(gql.ActionExecutionMessage);
+      expect(actions.specificAction.render).toBe(mockRender);
+      expect(actions["*"].render).toBeUndefined();
+    });
+
+    test("should not preserve render function when no matching action", () => {
+      const mockRender = () => "Unmatched Render";
+      const aguiMessage: agui.Message = {
+        id: "assistant-4",
+        role: "assistant",
+        content: "I'll execute an unmatched function",
+        toolCalls: [
+          {
+            id: "tool-call-4",
+            type: "function",
+            function: {
+              name: "unmatchedAction",
+              arguments: JSON.stringify({ param: "value" }),
+            },
+          },
+        ],
+        generativeUI: mockRender,
+      };
+
+      const actions: Record<string, any> = {
+        otherAction: { name: "otherAction" },
+      };
+
+      const result = aguiToGQL(aguiMessage, actions);
+
+      expect(result).toHaveLength(2);
+      expect(result[0]).toBeInstanceOf(gql.TextMessage);
+      expect(result[1]).toBeInstanceOf(gql.ActionExecutionMessage);
+      expect(actions.otherAction.render).toBeUndefined();
+    });
+
+    test("should handle multiple tool calls with wild card action", () => {
+      const mockRender = () => "Wild Card Render";
+      const aguiMessage: agui.Message = {
+        id: "assistant-5",
+        role: "assistant",
+        content: "I'll execute multiple functions",
+        toolCalls: [
+          {
+            id: "tool-call-5",
+            type: "function",
+            function: {
+              name: "firstFunction",
+              arguments: JSON.stringify({ param: "value1" }),
+            },
+          },
+          {
+            id: "tool-call-6",
+            type: "function",
+            function: {
+              name: "secondFunction",
+              arguments: JSON.stringify({ param: "value2" }),
+            },
+          },
+        ],
+        generativeUI: mockRender,
+      };
+
+      const actions: Record<string, any> = {
+        "*": { name: "*" },
+      };
+
+      const result = aguiToGQL(aguiMessage, actions);
+
+      expect(result).toHaveLength(3);
+      expect(result[0]).toBeInstanceOf(gql.TextMessage);
+      expect(result[1]).toBeInstanceOf(gql.ActionExecutionMessage);
+      expect(result[2]).toBeInstanceOf(gql.ActionExecutionMessage);
+      expect(actions["*"].render).toBe(mockRender);
+    });
+
+    test("should handle mixed specific and wild card actions", () => {
+      const mockRender = () => "Mixed Render";
+      const aguiMessage: agui.Message = {
+        id: "assistant-6",
+        role: "assistant",
+        content: "I'll execute mixed functions",
+        toolCalls: [
+          {
+            id: "tool-call-7",
+            type: "function",
+            function: {
+              name: "specificAction",
+              arguments: JSON.stringify({ param: "value1" }),
+            },
+          },
+          {
+            id: "tool-call-8",
+            type: "function",
+            function: {
+              name: "unknownAction",
+              arguments: JSON.stringify({ param: "value2" }),
+            },
+          },
+        ],
+        generativeUI: mockRender,
+      };
+
+      const actions: Record<string, any> = {
+        specificAction: { name: "specificAction" },
+        "*": { name: "*" },
+      };
+
+      const result = aguiToGQL(aguiMessage, actions);
+
+      expect(result).toHaveLength(3);
+      expect(result[0]).toBeInstanceOf(gql.TextMessage);
+      expect(result[1]).toBeInstanceOf(gql.ActionExecutionMessage);
+      expect(result[2]).toBeInstanceOf(gql.ActionExecutionMessage);
+      expect(actions.specificAction.render).toBe(mockRender);
+      // The wild card action should get the render function for the second tool call
+      expect(actions["*"].render).toBe(mockRender);
+    });
+
+    test("should handle no actions provided", () => {
+      const mockRender = () => "No Actions Render";
+      const aguiMessage: agui.Message = {
+        id: "assistant-7",
+        role: "assistant",
+        content: "I'll execute a function",
+        toolCalls: [
+          {
+            id: "tool-call-9",
+            type: "function",
+            function: {
+              name: "anyAction",
+              arguments: JSON.stringify({ param: "value" }),
+            },
+          },
+        ],
+        generativeUI: mockRender,
+      };
+
+      const result = aguiToGQL(aguiMessage);
+
+      expect(result).toHaveLength(2);
+      expect(result[0]).toBeInstanceOf(gql.TextMessage);
+      expect(result[1]).toBeInstanceOf(gql.ActionExecutionMessage);
+    });
+
+    test("should handle empty actions object", () => {
+      const mockRender = () => "Empty Actions Render";
+      const aguiMessage: agui.Message = {
+        id: "assistant-8",
+        role: "assistant",
+        content: "I'll execute a function",
+        toolCalls: [
+          {
+            id: "tool-call-10",
+            type: "function",
+            function: {
+              name: "anyAction",
+              arguments: JSON.stringify({ param: "value" }),
+            },
+          },
+        ],
+        generativeUI: mockRender,
+      };
+
+      const actions: Record<string, any> = {};
+
+      const result = aguiToGQL(aguiMessage, actions);
+
+      expect(result).toHaveLength(2);
+      expect(result[0]).toBeInstanceOf(gql.TextMessage);
+      expect(result[1]).toBeInstanceOf(gql.ActionExecutionMessage);
+    });
+
+    test("should handle actions with null render functions", () => {
+      const mockRender = () => "Null Render Test";
+      const aguiMessage: agui.Message = {
+        id: "assistant-9",
+        role: "assistant",
+        content: "I'll execute a function",
+        toolCalls: [
+          {
+            id: "tool-call-11",
+            type: "function",
+            function: {
+              name: "specificAction",
+              arguments: JSON.stringify({ param: "value" }),
+            },
+          },
+        ],
+        generativeUI: mockRender,
+      };
+
+      const actions: Record<string, any> = {
+        specificAction: { name: "specificAction", render: null },
+        "*": { name: "*", render: null },
+      };
+
+      const result = aguiToGQL(aguiMessage, actions);
+
+      expect(result).toHaveLength(2);
+      expect(result[0]).toBeInstanceOf(gql.TextMessage);
+      expect(result[1]).toBeInstanceOf(gql.ActionExecutionMessage);
+      // The render function should still be assigned even if the action had null render
+      expect(actions.specificAction.render).toBe(mockRender);
+    });
+
+    test("should handle actions with undefined render functions", () => {
+      const mockRender = () => "Undefined Render Test";
+      const aguiMessage: agui.Message = {
+        id: "assistant-10",
+        role: "assistant",
+        content: "I'll execute a function",
+        toolCalls: [
+          {
+            id: "tool-call-12",
+            type: "function",
+            function: {
+              name: "wildcardAction",
+              arguments: JSON.stringify({ param: "value" }),
+            },
+          },
+        ],
+        generativeUI: mockRender,
+      };
+
+      const actions: Record<string, any> = {
+        "*": { name: "*", render: undefined },
+      };
+
+      const result = aguiToGQL(aguiMessage, actions);
+
+      expect(result).toHaveLength(2);
+      expect(result[0]).toBeInstanceOf(gql.TextMessage);
+      expect(result[1]).toBeInstanceOf(gql.ActionExecutionMessage);
+      // The render function should still be assigned even if the action had undefined render
+      expect(actions["*"].render).toBe(mockRender);
+    });
+
+    test("should handle tool calls with malformed arguments", () => {
+      const mockRender = () => "Malformed Args Test";
+      const aguiMessage: agui.Message = {
+        id: "assistant-11",
+        role: "assistant",
+        content: "I'll execute a function",
+        toolCalls: [
+          {
+            id: "tool-call-13",
+            type: "function",
+            function: {
+              name: "wildcardAction",
+              arguments: "invalid json {", // Malformed JSON
+            },
+          },
+        ],
+        generativeUI: mockRender,
+      };
+
+      const actions: Record<string, any> = {
+        "*": { name: "*" },
+      };
+
+      const result = aguiToGQL(aguiMessage, actions);
+
+      expect(result).toHaveLength(2);
+      expect(result[0]).toBeInstanceOf(gql.TextMessage);
+      expect(result[1]).toBeInstanceOf(gql.ActionExecutionMessage);
+      expect(actions["*"].render).toBe(mockRender);
+    });
+
+    test("should handle tool calls with empty arguments string", () => {
+      const mockRender = () => "Empty Args Test";
+      const aguiMessage: agui.Message = {
+        id: "assistant-12",
+        role: "assistant",
+        content: "I'll execute a function",
+        toolCalls: [
+          {
+            id: "tool-call-14",
+            type: "function",
+            function: {
+              name: "wildcardAction",
+              arguments: "",
+            },
+          },
+        ],
+        generativeUI: mockRender,
+      };
+
+      const actions: Record<string, any> = {
+        "*": { name: "*" },
+      };
+
+      const result = aguiToGQL(aguiMessage, actions);
+
+      expect(result).toHaveLength(2);
+      expect(result[0]).toBeInstanceOf(gql.TextMessage);
+      expect(result[1]).toBeInstanceOf(gql.ActionExecutionMessage);
+      expect(actions["*"].render).toBe(mockRender);
+    });
+
+    test("should handle multiple wild card actions in same object", () => {
+      const mockRender = () => "Multiple Wildcards Test";
+      const aguiMessage: agui.Message = {
+        id: "assistant-13",
+        role: "assistant",
+        content: "I'll execute a function",
+        toolCalls: [
+          {
+            id: "tool-call-15",
+            type: "function",
+            function: {
+              name: "unknownAction",
+              arguments: JSON.stringify({ param: "value" }),
+            },
+          },
+        ],
+        generativeUI: mockRender,
+      };
+
+      const actions: Record<string, any> = {
+        wildcard1: { name: "*" },
+        wildcard2: { name: "*" },
+      };
+
+      const result = aguiToGQL(aguiMessage, actions);
+
+      expect(result).toHaveLength(2);
+      expect(result[0]).toBeInstanceOf(gql.TextMessage);
+      expect(result[1]).toBeInstanceOf(gql.ActionExecutionMessage);
+      // Should assign to the first wild card action found
+      expect(actions.wildcard1.render).toBe(mockRender);
+      expect(actions.wildcard2.render).toBeUndefined();
+    });
+  });
 });

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

@@ -86,8 +86,17 @@ export function aguiToGQL(
         // Preserve render function in actions context
         if ("generativeUI" in message && message.generativeUI && actions) {
           const actionName = toolCall.function.name;
-          if (actions[actionName]) {
-            actions[actionName].render = message.generativeUI;
+          // Check for specific action first, then wild card action
+          const specificAction = Object.values(actions).find(
+            (action: any) => action.name === actionName,
+          );
+          const wildcardAction = Object.values(actions).find((action: any) => action.name === "*");
+
+          // Assign render function to the matching action (specific takes priority)
+          if (specificAction) {
+            specificAction.render = message.generativeUI;
+          } else if (wildcardAction) {
+            wildcardAction.render = message.generativeUI;
           }
         }
         gqlMessages.push(actionExecMsg);