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

package/dist/graphql/definitions/queries.mjs

@@ -4,6 +4,7 @@ import {
 } from "../../chunk-X2UAP3QY.mjs";
 import "../../chunk-4KTMZMM2.mjs";
 import "../../chunk-WM3ARNBD.mjs";
+import "../../chunk-DELDZXUX.mjs";
 export {
   getAvailableAgentsQuery,
   loadAgentStateQuery

package/dist/index.d.ts

@@ -3,6 +3,9 @@ export { convertGqlOutputToMessages, convertMessagesToGqlInput, filterAdjacentAg
 export { ActionExecutionMessage, AgentStateMessage, ImageMessage, LangGraphInterruptEvent, Message, MetaEvent, ResultMessage, Role, TextMessage, langGraphInterruptEvent } from './client/types.js';
 export { GraphQLError } from 'graphql';
 export { ActionExecutionMessageInput, ActionExecutionMessageOutput, ActionInput, ActionInputAvailability, Agent, AgentSessionInput, AgentStateInput, AgentStateMessageInput, AgentStateMessageOutput, AgentsResponse, AvailableAgentsDocument, AvailableAgentsQuery, AvailableAgentsQueryVariables, BaseMessageOutput, BaseMetaEvent, BaseResponseStatus, CloudInput, CopilotKitLangGraphInterruptEvent, CopilotKitLangGraphInterruptEventData, CopilotRequestType, CopilotResponse, Exact, ExtensionsInput, ExtensionsResponse, FailedMessageStatus, FailedResponseStatus, FailedResponseStatusReason, ForwardedParametersInput, FrontendInput, GenerateCopilotResponseDocument, GenerateCopilotResponseInput, GenerateCopilotResponseMetadataInput, GenerateCopilotResponseMutation, GenerateCopilotResponseMutationVariables, GuardrailsInput, GuardrailsRuleInput, ImageMessageInput, ImageMessageOutput, Incremental, InputMaybe, LoadAgentStateDocument, LoadAgentStateInput, LoadAgentStateQuery, LoadAgentStateQueryVariables, LoadAgentStateResponse, MakeEmpty, MakeMaybe, MakeOptional, Maybe, MessageInput, MessageRole, MessageStatus, MessageStatusCode, MetaEventInput, MetaEventName, Mutation, MutationGenerateCopilotResponseArgs, OpenAiApiAssistantApiInput, OpenAiApiAssistantApiResponse, PendingMessageStatus, PendingResponseStatus, Query, QueryLoadAgentStateArgs, ResponseStatus, ResponseStatusCode, ResultMessageInput, ResultMessageOutput, Scalars, SuccessMessageStatus, SuccessResponseStatus, TextMessageInput, TextMessageOutput } from './graphql/@generated/graphql.js';
+export { aguiMessageWithImageToGQLMessage, aguiMessageWithRenderToGQL, aguiTextMessageToGQLMessage, aguiToGQL, aguiToolCallToGQLActionExecution, aguiToolMessageToGQLResultMessage } from './message-conversion/agui-to-gql.js';
+export { gqlImageMessageToAGUIMessage, gqlResultMessageToAGUIMessage, gqlTextMessageToAGUIMessage, gqlToAGUI } from './message-conversion/gql-to-agui.js';
 import 'urql';
 import '@urql/core';
 import '@graphql-typed-document-node/core';
+import '@copilotkit/shared';

package/dist/index.js

@@ -48,10 +48,20 @@ __export(src_exports, {
   ResultMessage: () => ResultMessage,
   Role: () => Role,
   TextMessage: () => TextMessage,
+  aguiMessageWithImageToGQLMessage: () => aguiMessageWithImageToGQLMessage,
+  aguiMessageWithRenderToGQL: () => aguiMessageWithRenderToGQL,
+  aguiTextMessageToGQLMessage: () => aguiTextMessageToGQLMessage,
+  aguiToGQL: () => aguiToGQL,
+  aguiToolCallToGQLActionExecution: () => aguiToolCallToGQLActionExecution,
+  aguiToolMessageToGQLResultMessage: () => aguiToolMessageToGQLResultMessage,
   convertGqlOutputToMessages: () => convertGqlOutputToMessages,
   convertMessagesToGqlInput: () => convertMessagesToGqlInput,
   filterAdjacentAgentStateMessages: () => filterAdjacentAgentStateMessages,
   filterAgentStateMessages: () => filterAgentStateMessages,
+  gqlImageMessageToAGUIMessage: () => gqlImageMessageToAGUIMessage,
+  gqlResultMessageToAGUIMessage: () => gqlResultMessageToAGUIMessage,
+  gqlTextMessageToAGUIMessage: () => gqlTextMessageToAGUIMessage,
+  gqlToAGUI: () => gqlToAGUI,
   langGraphInterruptEvent: () => langGraphInterruptEvent,
   loadMessagesFromJsonRepresentation: () => loadMessagesFromJsonRepresentation
 });
@@ -61,7 +71,7 @@ module.exports = __toCommonJS(src_exports);
 var import_core = require("@urql/core");
 
 // package.json
-var version = "1.9.3-next.4";
+var version = "1.10.0-next.0";
 
 // src/graphql/@generated/graphql.ts
 var ActionInputAvailability = /* @__PURE__ */ ((ActionInputAvailability2) => {
@@ -724,6 +734,355 @@ function getPartialArguments(args) {
     return {};
   }
 }
+
+// src/message-conversion/agui-to-gql.ts
+function extractAgentName(message) {
+  if (message.role !== "assistant") {
+    throw new Error(`Cannot extract agent name from message with role ${message.role}`);
+  }
+  return message.agentName || "unknown";
+}
+function isAgentStateMessage(message) {
+  return message.role === "assistant" && "agentName" in message && "state" in message;
+}
+function hasImageProperty(message) {
+  const canContainImage = message.role === "assistant" || message.role === "user";
+  if (!canContainImage || message.image === void 0) {
+    return false;
+  }
+  const isMalformed = message.image.format === void 0 || message.image.bytes === void 0;
+  if (isMalformed) {
+    return false;
+  }
+  return true;
+}
+function aguiToGQL(messages, actions, coAgentStateRenders) {
+  const gqlMessages = [];
+  messages = Array.isArray(messages) ? messages : [messages];
+  const toolCallNames = {};
+  for (const message of messages) {
+    if (isAgentStateMessage(message)) {
+      const agentName = extractAgentName(message);
+      const state = "state" in message && message.state ? message.state : {};
+      gqlMessages.push(
+        new AgentStateMessage({
+          id: message.id,
+          agentName,
+          state,
+          role: Role.Assistant
+        })
+      );
+      if ("generativeUI" in message && message.generativeUI && coAgentStateRenders) {
+        coAgentStateRenders[agentName] = {
+          name: agentName,
+          render: message.generativeUI
+        };
+      }
+      continue;
+    }
+    if (hasImageProperty(message)) {
+      gqlMessages.push(aguiMessageWithImageToGQLMessage(message));
+      continue;
+    }
+    if (message.role === "assistant" && message.toolCalls) {
+      gqlMessages.push(aguiTextMessageToGQLMessage(message));
+      for (const toolCall of message.toolCalls) {
+        toolCallNames[toolCall.id] = toolCall.function.name;
+        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;
+          }
+        }
+        gqlMessages.push(actionExecMsg);
+      }
+      continue;
+    }
+    if (message.role === "developer" || message.role === "system" || message.role === "assistant" || message.role === "user") {
+      gqlMessages.push(aguiTextMessageToGQLMessage(message));
+      continue;
+    }
+    if (message.role === "tool") {
+      gqlMessages.push(aguiToolMessageToGQLResultMessage(message, toolCallNames));
+      continue;
+    }
+    throw new Error(
+      `Unknown message role: "${message.role}" in message with id: ${message.id}`
+    );
+  }
+  return gqlMessages;
+}
+function aguiTextMessageToGQLMessage(message) {
+  if (message.role !== "developer" && message.role !== "system" && message.role !== "assistant" && message.role !== "user") {
+    throw new Error(`Cannot convert message with role ${message.role} to TextMessage`);
+  }
+  let roleValue;
+  if (message.role === "developer") {
+    roleValue = Role.Developer;
+  } else if (message.role === "system") {
+    roleValue = Role.System;
+  } else if (message.role === "assistant") {
+    roleValue = Role.Assistant;
+  } else {
+    roleValue = Role.User;
+  }
+  return new TextMessage({
+    id: message.id,
+    content: message.content || "",
+    role: roleValue
+  });
+}
+function aguiToolCallToGQLActionExecution(toolCall, parentMessageId) {
+  if (toolCall.type !== "function") {
+    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);
+    argumentsObj = {};
+  }
+  return new ActionExecutionMessage({
+    id: toolCall.id,
+    name: toolCall.function.name,
+    arguments: argumentsObj,
+    parentMessageId
+  });
+}
+function aguiToolMessageToGQLResultMessage(message, toolCallNames) {
+  if (message.role !== "tool") {
+    throw new Error(`Cannot convert message with role ${message.role} to ResultMessage`);
+  }
+  if (!message.toolCallId) {
+    throw new Error("Tool message must have a toolCallId");
+  }
+  const actionName = toolCallNames[message.toolCallId] || "unknown";
+  return new ResultMessage({
+    id: message.id,
+    result: message.content || "",
+    actionExecutionId: message.toolCallId,
+    actionName: message.toolName || actionName
+  });
+}
+function aguiMessageWithRenderToGQL(message, actions, coAgentStateRenders) {
+  if (message.role === "assistant" && "generativeUI" in message && message.generativeUI && !message.toolCalls) {
+    const gqlMessages = [];
+    gqlMessages.push(
+      new AgentStateMessage({
+        id: message.id,
+        agentName: "unknown",
+        state: {},
+        role: Role.Assistant
+      })
+    );
+    if (coAgentStateRenders) {
+      coAgentStateRenders.unknown = {
+        name: "unknown",
+        render: message.generativeUI
+      };
+    }
+    return gqlMessages;
+  }
+  return aguiToGQL([message], actions, coAgentStateRenders);
+}
+function aguiMessageWithImageToGQLMessage(message) {
+  if (!hasImageProperty(message)) {
+    throw new Error(`Cannot convert message to ImageMessage: missing format or bytes`);
+  }
+  let roleValue;
+  if (message.role === "assistant") {
+    roleValue = Role.Assistant;
+  } else {
+    roleValue = Role.User;
+  }
+  if (message.role !== "assistant" && message.role !== "user") {
+    throw new Error(`Cannot convert message with role ${message.role} to ImageMessage`);
+  }
+  return new ImageMessage({
+    id: message.id,
+    format: message.image.format,
+    bytes: message.image.bytes,
+    role: roleValue
+  });
+}
+
+// src/message-conversion/gql-to-agui.ts
+var VALID_IMAGE_FORMATS = ["jpeg", "png", "webp", "gif"];
+function validateImageFormat(format) {
+  return VALID_IMAGE_FORMATS.includes(format);
+}
+function gqlToAGUI(messages, actions, coAgentStateRenders) {
+  let aguiMessages = [];
+  messages = Array.isArray(messages) ? messages : [messages];
+  const actionResults = /* @__PURE__ */ new Map();
+  for (const message of messages) {
+    if (message.isResultMessage()) {
+      actionResults.set(message.actionExecutionId, message.result);
+    }
+  }
+  for (const message of messages) {
+    if (message.isTextMessage()) {
+      aguiMessages.push(gqlTextMessageToAGUIMessage(message));
+    } else if (message.isResultMessage()) {
+      aguiMessages.push(gqlResultMessageToAGUIMessage(message));
+    } else if (message.isActionExecutionMessage()) {
+      aguiMessages.push(gqlActionExecutionMessageToAGUIMessage(message, actions, actionResults));
+    } else if (message.isAgentStateMessage()) {
+      aguiMessages.push(gqlAgentStateMessageToAGUIMessage(message, coAgentStateRenders));
+    } else if (message.isImageMessage()) {
+      aguiMessages.push(gqlImageMessageToAGUIMessage(message));
+    } else {
+      throw new Error("Unknown message type");
+    }
+  }
+  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 _a;
+        const actionResult = actionResults == null ? void 0 : actionResults.get(message.id);
+        let status = "inProgress";
+        if (actionResult !== void 0) {
+          status = "complete";
+        } else if (((_a = message.status) == null ? void 0 : _a.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);
+      };
+    };
+    return {
+      id: message.id,
+      role: "assistant",
+      content: "",
+      toolCalls: [actionExecutionMessageToAGUIMessage(message)],
+      generativeUI: createRenderWrapper(action.render),
+      name: message.name
+    };
+  }
+  return {
+    id: message.id,
+    role: "assistant",
+    toolCalls: [actionExecutionMessageToAGUIMessage(message)],
+    name: message.name
+  };
+}
+function gqlAgentStateMessageToAGUIMessage(message, coAgentStateRenders) {
+  if (coAgentStateRenders && Object.values(coAgentStateRenders).some((render) => render.name === message.agentName)) {
+    const render = Object.values(coAgentStateRenders).find(
+      (render2) => render2.name === message.agentName
+    );
+    const createRenderWrapper = (originalRender) => {
+      if (!originalRender)
+        return void 0;
+      return (props) => {
+        const state = message.state;
+        const renderProps = {
+          state
+        };
+        return originalRender(renderProps);
+      };
+    };
+    return {
+      id: message.id,
+      role: "assistant",
+      generativeUI: createRenderWrapper(render.render),
+      agentName: message.agentName,
+      state: message.state
+    };
+  }
+  return {
+    id: message.id,
+    role: "assistant",
+    agentName: message.agentName,
+    state: message.state
+  };
+}
+function actionExecutionMessageToAGUIMessage(actionExecutionMessage) {
+  return {
+    id: actionExecutionMessage.id,
+    function: {
+      name: actionExecutionMessage.name,
+      arguments: JSON.stringify(actionExecutionMessage.arguments)
+    },
+    type: "function"
+  };
+}
+function gqlTextMessageToAGUIMessage(message) {
+  switch (message.role) {
+    case Role.Developer:
+      return {
+        id: message.id,
+        role: "developer",
+        content: message.content
+      };
+    case Role.System:
+      return {
+        id: message.id,
+        role: "system",
+        content: message.content
+      };
+    case Role.Assistant:
+      return {
+        id: message.id,
+        role: "assistant",
+        content: message.content
+      };
+    case Role.User:
+      return {
+        id: message.id,
+        role: "user",
+        content: message.content
+      };
+    default:
+      throw new Error("Unknown message role");
+  }
+}
+function gqlResultMessageToAGUIMessage(message) {
+  return {
+    id: message.id,
+    role: "tool",
+    content: message.result,
+    toolCallId: message.actionExecutionId,
+    toolName: message.actionName
+  };
+}
+function gqlImageMessageToAGUIMessage(message) {
+  if (!validateImageFormat(message.format)) {
+    throw new Error(
+      `Invalid image format: ${message.format}. Supported formats are: ${VALID_IMAGE_FORMATS.join(", ")}`
+    );
+  }
+  if (!message.bytes || typeof message.bytes !== "string" || message.bytes.trim() === "") {
+    throw new Error("Image bytes must be a non-empty string");
+  }
+  const role = message.role === Role.Assistant ? "assistant" : "user";
+  const imageMessage = {
+    id: message.id,
+    role,
+    content: "",
+    image: {
+      format: message.format,
+      bytes: message.bytes
+    }
+  };
+  return imageMessage;
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   ActionExecutionMessage,
@@ -744,10 +1103,20 @@ function getPartialArguments(args) {
   ResultMessage,
   Role,
   TextMessage,
+  aguiMessageWithImageToGQLMessage,
+  aguiMessageWithRenderToGQL,
+  aguiTextMessageToGQLMessage,
+  aguiToGQL,
+  aguiToolCallToGQLActionExecution,
+  aguiToolMessageToGQLResultMessage,
   convertGqlOutputToMessages,
   convertMessagesToGqlInput,
   filterAdjacentAgentStateMessages,
   filterAgentStateMessages,
+  gqlImageMessageToAGUIMessage,
+  gqlResultMessageToAGUIMessage,
+  gqlTextMessageToAGUIMessage,
+  gqlToAGUI,
   langGraphInterruptEvent,
   loadMessagesFromJsonRepresentation
 });