@copilotkit/runtime-client-gql/dist/message-conversion/roundtrip-conversion.test.mjs.map

<img src="https://github.com/user-attachments/assets/0a6b64d9-e193-4940-a3f6-60334ac34084" alt="banner" style="border-radius: 12px; border: 2px solid #d6d4fa;" />

{"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":[]}