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

package/dist/message-conversion/gql-to-agui.test.js

@@ -18989,45 +18989,59 @@ function gqlToAGUI(messages, actions, coAgentStateRenders) {
   return aguiMessages;
 }
 function gqlActionExecutionMessageToAGUIMessage(message, actions, actionResults) {
-  if (actions && Object.values(actions).some((action) => action.name === message.name)) {
-    const action = Object.values(actions).find((action2) => action2.name === message.name);
-    const createRenderWrapper = (originalRender) => {
-      if (!originalRender)
-        return void 0;
-      return (props) => {
-        var _a3;
-        const actionResult = actionResults == null ? void 0 : actionResults.get(message.id);
-        let status = "inProgress";
-        if (actionResult !== void 0) {
-          status = "complete";
-        } else if (((_a3 = message.status) == null ? void 0 : _a3.code) !== "Pending" /* Pending */) {
-          status = "executing";
-        }
-        const renderProps = {
-          status: (props == null ? void 0 : props.status) || status,
-          args: message.arguments || {},
-          result: (props == null ? void 0 : props.result) || actionResult || void 0,
-          respond: (props == null ? void 0 : props.respond) || (() => {
-          }),
-          messageId: message.id,
-          ...props
-        };
-        return originalRender(renderProps);
-      };
-    };
+  const hasSpecificAction = actions && Object.values(actions).some((action2) => action2.name === message.name);
+  const hasWildcardAction = actions && Object.values(actions).some((action2) => action2.name === "*");
+  if (!actions || !hasSpecificAction && !hasWildcardAction) {
     return {
       id: message.id,
       role: "assistant",
-      content: "",
       toolCalls: [actionExecutionMessageToAGUIMessage(message)],
-      generativeUI: createRenderWrapper(action.render),
       name: message.name
     };
   }
+  const action = Object.values(actions).find((action2) => action2.name === message.name) || Object.values(actions).find((action2) => action2.name === "*");
+  const createRenderWrapper = (originalRender) => {
+    if (!originalRender)
+      return void 0;
+    return (props) => {
+      var _a3;
+      let actionResult = actionResults == null ? void 0 : actionResults.get(message.id);
+      let status = "inProgress";
+      if (actionResult !== void 0) {
+        status = "complete";
+      } else if (((_a3 = message.status) == null ? void 0 : _a3.code) !== "Pending" /* Pending */) {
+        status = "executing";
+      }
+      if (typeof (props == null ? void 0 : props.result) === "string") {
+        try {
+          props.result = JSON.parse(props.result);
+        } catch (e) {
+        }
+      }
+      if (typeof actionResult === "string") {
+        try {
+          actionResult = JSON.parse(actionResult);
+        } catch (e) {
+        }
+      }
+      const renderProps = {
+        status: (props == null ? void 0 : props.status) || status,
+        args: message.arguments || {},
+        result: (props == null ? void 0 : props.result) || actionResult || void 0,
+        respond: (props == null ? void 0 : props.respond) || (() => {
+        }),
+        messageId: message.id,
+        ...props
+      };
+      return originalRender(renderProps);
+    };
+  };
   return {
     id: message.id,
     role: "assistant",
+    content: "",
     toolCalls: [actionExecutionMessageToAGUIMessage(message)],
+    generativeUI: createRenderWrapper(action.render),
     name: message.name
   };
 }
@@ -19839,6 +19853,586 @@ describe("message-conversion", () => {
       });
     });
   });
+  describe("Wild Card Actions", () => {
+    test3("should handle action execution with specific action", () => {
+      const actions = {
+        testAction: {
+          name: "testAction",
+          render: vi.fn((props) => `Rendered: ${props.args.test}`)
+        }
+      };
+      const actionExecMsg = new ActionExecutionMessage({
+        id: "action-1",
+        name: "testAction",
+        arguments: { test: "value" },
+        parentMessageId: "parent-1"
+      });
+      const result = gqlActionExecutionMessageToAGUIMessage(actionExecMsg, actions);
+      globalExpect(result).toMatchObject({
+        id: "action-1",
+        role: "assistant",
+        content: "",
+        toolCalls: [
+          {
+            id: "action-1",
+            function: {
+              name: "testAction",
+              arguments: '{"test":"value"}'
+            },
+            type: "function"
+          }
+        ],
+        generativeUI: globalExpect.any(Function),
+        name: "testAction"
+      });
+    });
+    test3("should handle action execution with wild card action", () => {
+      const actions = {
+        "*": {
+          name: "*",
+          render: vi.fn((props) => `Wildcard rendered: ${props.args.test}`)
+        }
+      };
+      const actionExecMsg = new ActionExecutionMessage({
+        id: "action-2",
+        name: "unknownAction",
+        arguments: { test: "wildcard-value" },
+        parentMessageId: "parent-2"
+      });
+      const result = gqlActionExecutionMessageToAGUIMessage(actionExecMsg, actions);
+      globalExpect(result).toMatchObject({
+        id: "action-2",
+        role: "assistant",
+        content: "",
+        toolCalls: [
+          {
+            id: "action-2",
+            function: {
+              name: "unknownAction",
+              arguments: '{"test":"wildcard-value"}'
+            },
+            type: "function"
+          }
+        ],
+        generativeUI: globalExpect.any(Function),
+        name: "unknownAction"
+      });
+    });
+    test3("should prioritize specific action over wild card action", () => {
+      const actions = {
+        specificAction: {
+          name: "specificAction",
+          render: vi.fn((props) => "Specific action rendered")
+        },
+        "*": {
+          name: "*",
+          render: vi.fn((props) => "Wildcard action rendered")
+        }
+      };
+      const actionExecMsg = new ActionExecutionMessage({
+        id: "action-3",
+        name: "specificAction",
+        arguments: { test: "value" },
+        parentMessageId: "parent-3"
+      });
+      const result = gqlActionExecutionMessageToAGUIMessage(actionExecMsg, actions);
+      globalExpect(result).toMatchObject({
+        id: "action-3",
+        role: "assistant",
+        content: "",
+        toolCalls: [
+          {
+            id: "action-3",
+            function: {
+              name: "specificAction",
+              arguments: '{"test":"value"}'
+            },
+            type: "function"
+          }
+        ],
+        generativeUI: globalExpect.any(Function),
+        name: "specificAction"
+      });
+    });
+    test3("should handle action execution without any matching actions", () => {
+      const actions = {
+        otherAction: {
+          name: "otherAction",
+          render: vi.fn()
+        }
+      };
+      const actionExecMsg = new ActionExecutionMessage({
+        id: "action-4",
+        name: "unmatchedAction",
+        arguments: { test: "value" },
+        parentMessageId: "parent-4"
+      });
+      const result = gqlActionExecutionMessageToAGUIMessage(actionExecMsg, actions);
+      globalExpect(result).toMatchObject({
+        id: "action-4",
+        role: "assistant",
+        toolCalls: [
+          {
+            id: "action-4",
+            function: {
+              name: "unmatchedAction",
+              arguments: '{"test":"value"}'
+            },
+            type: "function"
+          }
+        ],
+        name: "unmatchedAction"
+      });
+      globalExpect(result).not.toHaveProperty("generativeUI");
+    });
+    test3("should handle action execution with no actions provided", () => {
+      const actionExecMsg = new ActionExecutionMessage({
+        id: "action-5",
+        name: "anyAction",
+        arguments: { test: "value" },
+        parentMessageId: "parent-5"
+      });
+      const result = gqlActionExecutionMessageToAGUIMessage(actionExecMsg);
+      globalExpect(result).toMatchObject({
+        id: "action-5",
+        role: "assistant",
+        toolCalls: [
+          {
+            id: "action-5",
+            function: {
+              name: "anyAction",
+              arguments: '{"test":"value"}'
+            },
+            type: "function"
+          }
+        ],
+        name: "anyAction"
+      });
+      globalExpect(result).not.toHaveProperty("generativeUI");
+    });
+    test3("should handle action execution with completed result", () => {
+      const actions = {
+        "*": {
+          name: "*",
+          render: vi.fn((props) => `Result: ${props.result}`)
+        }
+      };
+      const actionExecMsg = new ActionExecutionMessage({
+        id: "action-6",
+        name: "testAction",
+        arguments: { test: "value" },
+        parentMessageId: "parent-6"
+      });
+      const actionResults = /* @__PURE__ */ new Map([["action-6", "completed result"]]);
+      const result = gqlActionExecutionMessageToAGUIMessage(actionExecMsg, actions, actionResults);
+      globalExpect(result).toMatchObject({
+        id: "action-6",
+        role: "assistant",
+        content: "",
+        toolCalls: [
+          {
+            id: "action-6",
+            function: {
+              name: "testAction",
+              arguments: '{"test":"value"}'
+            },
+            type: "function"
+          }
+        ],
+        generativeUI: globalExpect.any(Function),
+        name: "testAction"
+      });
+    });
+    test3("should handle action execution with executing status", () => {
+      const actions = {
+        "*": {
+          name: "*",
+          render: vi.fn((props) => `Status: ${props.status}`)
+        }
+      };
+      const actionExecMsg = new ActionExecutionMessage({
+        id: "action-7",
+        name: "testAction",
+        arguments: { test: "value" },
+        parentMessageId: "parent-7",
+        status: { code: "Success" /* Success */ }
+      });
+      const result = gqlActionExecutionMessageToAGUIMessage(actionExecMsg, actions);
+      globalExpect(result).toMatchObject({
+        id: "action-7",
+        role: "assistant",
+        content: "",
+        toolCalls: [
+          {
+            id: "action-7",
+            function: {
+              name: "testAction",
+              arguments: '{"test":"value"}'
+            },
+            type: "function"
+          }
+        ],
+        generativeUI: globalExpect.any(Function),
+        name: "testAction"
+      });
+    });
+    test3("should handle action execution with pending status", () => {
+      const actions = {
+        "*": {
+          name: "*",
+          render: vi.fn((props) => `Status: ${props.status}`)
+        }
+      };
+      const actionExecMsg = new ActionExecutionMessage({
+        id: "action-8",
+        name: "testAction",
+        arguments: { test: "value" },
+        parentMessageId: "parent-8",
+        status: { code: "Pending" /* Pending */ }
+      });
+      const result = gqlActionExecutionMessageToAGUIMessage(actionExecMsg, actions);
+      globalExpect(result).toMatchObject({
+        id: "action-8",
+        role: "assistant",
+        content: "",
+        toolCalls: [
+          {
+            id: "action-8",
+            function: {
+              name: "testAction",
+              arguments: '{"test":"value"}'
+            },
+            type: "function"
+          }
+        ],
+        generativeUI: globalExpect.any(Function),
+        name: "testAction"
+      });
+    });
+    test3("should handle action execution with failed status", () => {
+      const actions = {
+        "*": {
+          name: "*",
+          render: vi.fn((props) => `Status: ${props.status}`)
+        }
+      };
+      const actionExecMsg = new ActionExecutionMessage({
+        id: "action-9",
+        name: "testAction",
+        arguments: { test: "value" },
+        parentMessageId: "parent-9",
+        status: { code: "Failed" /* Failed */ }
+      });
+      const result = gqlActionExecutionMessageToAGUIMessage(actionExecMsg, actions);
+      globalExpect(result).toMatchObject({
+        id: "action-9",
+        role: "assistant",
+        content: "",
+        toolCalls: [
+          {
+            id: "action-9",
+            function: {
+              name: "testAction",
+              arguments: '{"test":"value"}'
+            },
+            type: "function"
+          }
+        ],
+        generativeUI: globalExpect.any(Function),
+        name: "testAction"
+      });
+    });
+    test3("should handle action execution with undefined status", () => {
+      const actions = {
+        "*": {
+          name: "*",
+          render: vi.fn((props) => `Status: ${props.status}`)
+        }
+      };
+      const actionExecMsg = new ActionExecutionMessage({
+        id: "action-10",
+        name: "testAction",
+        arguments: { test: "value" },
+        parentMessageId: "parent-10"
+        // No status field
+      });
+      const result = gqlActionExecutionMessageToAGUIMessage(actionExecMsg, actions);
+      globalExpect(result).toMatchObject({
+        id: "action-10",
+        role: "assistant",
+        content: "",
+        toolCalls: [
+          {
+            id: "action-10",
+            function: {
+              name: "testAction",
+              arguments: '{"test":"value"}'
+            },
+            type: "function"
+          }
+        ],
+        generativeUI: globalExpect.any(Function),
+        name: "testAction"
+      });
+    });
+    test3("should handle action execution with empty arguments", () => {
+      const actions = {
+        "*": {
+          name: "*",
+          render: vi.fn((props) => `Args: ${JSON.stringify(props.args)}`)
+        }
+      };
+      const actionExecMsg = new ActionExecutionMessage({
+        id: "action-11",
+        name: "testAction",
+        arguments: {},
+        parentMessageId: "parent-11"
+      });
+      const result = gqlActionExecutionMessageToAGUIMessage(actionExecMsg, actions);
+      globalExpect(result).toMatchObject({
+        id: "action-11",
+        role: "assistant",
+        content: "",
+        toolCalls: [
+          {
+            id: "action-11",
+            function: {
+              name: "testAction",
+              arguments: "{}"
+            },
+            type: "function"
+          }
+        ],
+        generativeUI: globalExpect.any(Function),
+        name: "testAction"
+      });
+    });
+    test3("should handle action execution with null arguments", () => {
+      const actions = {
+        "*": {
+          name: "*",
+          render: vi.fn((props) => `Args: ${JSON.stringify(props.args)}`)
+        }
+      };
+      const actionExecMsg = new ActionExecutionMessage({
+        id: "action-12",
+        name: "testAction",
+        arguments: null,
+        parentMessageId: "parent-12"
+      });
+      const result = gqlActionExecutionMessageToAGUIMessage(actionExecMsg, actions);
+      globalExpect(result).toMatchObject({
+        id: "action-12",
+        role: "assistant",
+        content: "",
+        toolCalls: [
+          {
+            id: "action-12",
+            function: {
+              name: "testAction",
+              arguments: "null"
+            },
+            type: "function"
+          }
+        ],
+        generativeUI: globalExpect.any(Function),
+        name: "testAction"
+      });
+    });
+    test3("should handle action execution with complex nested arguments", () => {
+      const actions = {
+        "*": {
+          name: "*",
+          render: vi.fn((props) => `Complex: ${JSON.stringify(props.args)}`)
+        }
+      };
+      const complexArgs = {
+        nested: {
+          array: [1, 2, 3],
+          object: { key: "value" },
+          nullValue: null,
+          undefinedValue: void 0
+        },
+        string: "test",
+        number: 42,
+        boolean: true
+      };
+      const actionExecMsg = new ActionExecutionMessage({
+        id: "action-13",
+        name: "testAction",
+        arguments: complexArgs,
+        parentMessageId: "parent-13"
+      });
+      const result = gqlActionExecutionMessageToAGUIMessage(actionExecMsg, actions);
+      globalExpect(result).toMatchObject({
+        id: "action-13",
+        role: "assistant",
+        content: "",
+        toolCalls: [
+          {
+            id: "action-13",
+            function: {
+              name: "testAction",
+              arguments: JSON.stringify(complexArgs)
+            },
+            type: "function"
+          }
+        ],
+        generativeUI: globalExpect.any(Function),
+        name: "testAction"
+      });
+    });
+    test3("should handle multiple wild card actions (should use first one)", () => {
+      const actions = {
+        wildcard1: {
+          name: "*",
+          render: vi.fn((props) => "First wildcard")
+        },
+        wildcard2: {
+          name: "*",
+          render: vi.fn((props) => "Second wildcard")
+        }
+      };
+      const actionExecMsg = new ActionExecutionMessage({
+        id: "action-14",
+        name: "unknownAction",
+        arguments: { test: "value" },
+        parentMessageId: "parent-14"
+      });
+      const result = gqlActionExecutionMessageToAGUIMessage(actionExecMsg, actions);
+      globalExpect(result).toMatchObject({
+        id: "action-14",
+        role: "assistant",
+        content: "",
+        toolCalls: [
+          {
+            id: "action-14",
+            function: {
+              name: "unknownAction",
+              arguments: '{"test":"value"}'
+            },
+            type: "function"
+          }
+        ],
+        generativeUI: globalExpect.any(Function),
+        name: "unknownAction"
+      });
+    });
+    test3("should parse string results in generativeUI props", () => {
+      const actions = {
+        "*": {
+          name: "*",
+          render: vi.fn((props) => `Result: ${JSON.stringify(props.result)}`)
+        }
+      };
+      const actionExecMsg = new ActionExecutionMessage({
+        id: "action-string-result",
+        name: "stringResultAction",
+        arguments: { test: "value" },
+        parentMessageId: "parent-string"
+      });
+      const actionResults = /* @__PURE__ */ new Map();
+      actionResults.set("action-string-result", '{"parsed": true, "value": 42}');
+      const result = gqlActionExecutionMessageToAGUIMessage(actionExecMsg, actions, actionResults);
+      globalExpect(result.generativeUI).toBeDefined();
+      const renderResult = result.generativeUI({
+        result: '{"from": "props", "data": "test"}'
+      });
+      globalExpect(actions["*"].render).toHaveBeenCalledWith(
+        globalExpect.objectContaining({
+          result: { from: "props", data: "test" },
+          // Should be parsed from string
+          args: { test: "value" },
+          status: "complete",
+          messageId: "action-string-result"
+        })
+      );
+    });
+    test3("should handle malformed JSON strings gracefully in results", () => {
+      const actions = {
+        "*": {
+          name: "*",
+          render: vi.fn((props) => `Result: ${JSON.stringify(props.result)}`)
+        }
+      };
+      const actionExecMsg = new ActionExecutionMessage({
+        id: "action-malformed",
+        name: "malformedAction",
+        arguments: { test: "value" },
+        parentMessageId: "parent-malformed"
+      });
+      const actionResults = /* @__PURE__ */ new Map();
+      actionResults.set("action-malformed", "invalid json {");
+      const result = gqlActionExecutionMessageToAGUIMessage(actionExecMsg, actions, actionResults);
+      globalExpect(result.generativeUI).toBeDefined();
+      const renderResult = result.generativeUI({
+        result: "invalid json {"
+      });
+      globalExpect(actions["*"].render).toHaveBeenCalledWith(
+        globalExpect.objectContaining({
+          result: "invalid json {",
+          // Should remain as string due to parse error
+          args: { test: "value" },
+          status: "complete",
+          messageId: "action-malformed"
+        })
+      );
+    });
+    test3("should handle non-string results without parsing", () => {
+      const actions = {
+        "*": {
+          name: "*",
+          render: vi.fn((props) => `Result: ${JSON.stringify(props.result)}`)
+        }
+      };
+      const actionExecMsg = new ActionExecutionMessage({
+        id: "action-object-result",
+        name: "objectResultAction",
+        arguments: { test: "value" },
+        parentMessageId: "parent-object"
+      });
+      const actionResults = /* @__PURE__ */ new Map();
+      actionResults.set("action-object-result", '{"already": "parsed"}');
+      const result = gqlActionExecutionMessageToAGUIMessage(actionExecMsg, actions, actionResults);
+      globalExpect(result.generativeUI).toBeDefined();
+      const renderResult = result.generativeUI({
+        result: { from: "props", data: "object" }
+      });
+      globalExpect(actions["*"].render).toHaveBeenCalledWith(
+        globalExpect.objectContaining({
+          result: { from: "props", data: "object" },
+          // Should remain as object
+          args: { test: "value" },
+          status: "complete",
+          messageId: "action-object-result"
+        })
+      );
+    });
+    test3("should handle action execution arguments correctly with simplified conversion", () => {
+      const actionExecMsg = new ActionExecutionMessage({
+        id: "action-simplified",
+        name: "simplifiedAction",
+        arguments: { complex: { nested: "value" }, array: [1, 2, 3] },
+        parentMessageId: "parent-simplified"
+      });
+      const result = gqlActionExecutionMessageToAGUIMessage(actionExecMsg);
+      globalExpect(result).toMatchObject({
+        id: "action-simplified",
+        role: "assistant",
+        toolCalls: [
+          {
+            id: "action-simplified",
+            function: {
+              name: "simplifiedAction",
+              arguments: '{"complex":{"nested":"value"},"array":[1,2,3]}'
+            },
+            type: "function"
+          }
+        ],
+        name: "simplifiedAction"
+      });
+    });
+  });
 });
 /*! Bundled license information: