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

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

@@ -19013,8 +19013,14 @@ function aguiToGQL(messages, actions, coAgentStateRenders) {
         const actionExecMsg = aguiToolCallToGQLActionExecution(toolCall, message.id);
         if ("generativeUI" in message && message.generativeUI && actions) {
           const actionName = toolCall.function.name;
-          if (actions[actionName]) {
-            actions[actionName].render = message.generativeUI;
+          const specificAction = Object.values(actions).find(
+            (action) => action.name === actionName
+          );
+          const wildcardAction = Object.values(actions).find((action) => action.name === "*");
+          if (specificAction) {
+            specificAction.render = message.generativeUI;
+          } else if (wildcardAction) {
+            wildcardAction.render = message.generativeUI;
           }
         }
         gqlMessages.push(actionExecMsg);
@@ -19060,10 +19066,20 @@ function aguiToolCallToGQLActionExecution(toolCall, parentMessageId) {
     throw new Error(`Unsupported tool call type: ${toolCall.type}`);
   }
   let argumentsObj;
-  try {
-    argumentsObj = JSON.parse(toolCall.function.arguments);
-  } catch (error) {
-    console.warn(`Failed to parse tool call arguments for ${toolCall.function.name}:`, error);
+  if (typeof toolCall.function.arguments === "string") {
+    try {
+      argumentsObj = JSON.parse(toolCall.function.arguments);
+    } catch (error) {
+      console.warn(`Failed to parse tool call arguments for ${toolCall.function.name}:`, error);
+      argumentsObj = {};
+    }
+  } else if (typeof toolCall.function.arguments === "object" && toolCall.function.arguments !== null) {
+    argumentsObj = toolCall.function.arguments;
+  } else {
+    console.warn(
+      `Invalid tool call arguments type for ${toolCall.function.name}:`,
+      typeof toolCall.function.arguments
+    );
     argumentsObj = {};
   }
   return new ActionExecutionMessage({
@@ -19081,9 +19097,23 @@ function aguiToolMessageToGQLResultMessage(message, toolCallNames) {
     throw new Error("Tool message must have a toolCallId");
   }
   const actionName = toolCallNames[message.toolCallId] || "unknown";
+  let resultContent;
+  const messageContent = message.content || "";
+  if (typeof messageContent === "string") {
+    resultContent = messageContent;
+  } else if (typeof messageContent === "object" && messageContent !== null) {
+    try {
+      resultContent = JSON.stringify(messageContent);
+    } catch (error) {
+      console.warn(`Failed to stringify tool result for ${actionName}:`, error);
+      resultContent = String(messageContent);
+    }
+  } else {
+    resultContent = String(messageContent);
+  }
   return new ResultMessage({
     id: message.id,
-    result: message.content || "",
+    result: resultContent,
     actionExecutionId: message.toolCallId,
     actionName: message.toolName || actionName
   });
@@ -19141,45 +19171,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
   };
 }
@@ -19473,6 +19517,177 @@ describe("roundtrip message conversion", () => {
     globalExpect(aguiUserMsgs2[0].image.format).toBe("png");
     globalExpect(aguiUserMsgs2[0].image.bytes).toBe("userbase64data");
   });
+  test3("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"}'
+      );
+    }
+  });
+  test3("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"}'
+      );
+    }
+  });
+  test3("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" });
+  });
+  test3("roundtrip conversion with result parsing edge cases", () => {
+    const toolResultMsg = {
+      id: "tool-result-json",
+      role: "tool",
+      content: '{"status": "success", "data": {"value": 42}}',
+      toolCallId: "tool-call-json",
+      toolName: "jsonAction"
+    };
+    const gqlMsgs = aguiToGQL(toolResultMsg);
+    const aguiMsgs = gqlToAGUI(gqlMsgs);
+    globalExpect(gqlMsgs).toHaveLength(1);
+    globalExpect(gqlMsgs[0]).toBeInstanceOf(ResultMessage);
+    globalExpect(gqlMsgs[0].result).toBe('{"status": "success", "data": {"value": 42}}');
+    globalExpect(aguiMsgs).toHaveLength(1);
+    globalExpect(aguiMsgs[0].role).toBe("tool");
+    globalExpect(aguiMsgs[0].content).toBe('{"status": "success", "data": {"value": 42}}');
+  });
+  test3("roundtrip conversion with object content in tool results", () => {
+    const toolResultMsg = {
+      id: "tool-result-object",
+      role: "tool",
+      content: { status: "success", data: { value: 42 } },
+      toolCallId: "tool-call-object",
+      toolName: "objectAction"
+    };
+    const gqlMsgs = aguiToGQL(toolResultMsg);
+    const aguiMsgs = gqlToAGUI(gqlMsgs);
+    globalExpect(gqlMsgs).toHaveLength(1);
+    globalExpect(gqlMsgs[0]).toBeInstanceOf(ResultMessage);
+    globalExpect(gqlMsgs[0].result).toBe('{"status":"success","data":{"value":42}}');
+    globalExpect(aguiMsgs).toHaveLength(1);
+    globalExpect(aguiMsgs[0].role).toBe("tool");
+    globalExpect(aguiMsgs[0].content).toBe('{"status":"success","data":{"value":42}}');
+  });
+  test3("roundtrip conversion with action execution and result parsing", () => {
+    const mockRender = vi.fn((props) => `Rendered: ${JSON.stringify(props.result)}`);
+    const actionExecMsg = new ActionExecutionMessage({
+      id: "action-with-result",
+      name: "testAction",
+      arguments: { input: "test-value" },
+      parentMessageId: "parent-result"
+    });
+    const resultMsg = new ResultMessage({
+      id: "result-with-json",
+      result: '{"output": "processed", "count": 5}',
+      actionExecutionId: "action-with-result",
+      actionName: "testAction"
+    });
+    const actions = {
+      testAction: {
+        name: "testAction",
+        render: mockRender
+      }
+    };
+    const aguiMsgs = gqlToAGUI([actionExecMsg, resultMsg], actions);
+    globalExpect(aguiMsgs).toHaveLength(2);
+    globalExpect(aguiMsgs[0].role).toBe("assistant");
+    globalExpect("generativeUI" in aguiMsgs[0]).toBe(true);
+    globalExpect(aguiMsgs[1].role).toBe("tool");
+    globalExpect(aguiMsgs[1].content).toBe('{"output": "processed", "count": 5}');
+    if ("generativeUI" in aguiMsgs[0] && aguiMsgs[0].generativeUI) {
+      aguiMsgs[0].generativeUI({ result: '{"parsed": true}' });
+      globalExpect(mockRender).toHaveBeenCalledWith(
+        globalExpect.objectContaining({
+          result: { parsed: true }
+          // Should be parsed from string
+        })
+      );
+    }
+    const gqlMsgs2 = aguiToGQL(aguiMsgs, actions);
+    globalExpect(gqlMsgs2).toHaveLength(3);
+    globalExpect(gqlMsgs2[0]).toBeInstanceOf(TextMessage);
+    globalExpect(gqlMsgs2[1]).toBeInstanceOf(ActionExecutionMessage);
+    globalExpect(gqlMsgs2[2]).toBeInstanceOf(ResultMessage);
+    globalExpect(gqlMsgs2[1].arguments).toEqual({ input: "test-value" });
+    globalExpect(gqlMsgs2[2].result).toBe('{"output": "processed", "count": 5}');
+  });
 });
 /*! Bundled license information: