@copilotkit/runtime 1.3.16-mme-improve-langsmith.0 → 1.4.0-pre-1-4-0.0

package/dist/index.js

@@ -44,7 +44,7 @@ var require_package = __commonJS({
       publishConfig: {
         access: "public"
       },
-      version: "1.3.16-mme-improve-langsmith.0",
+      version: "1.4.0-pre-1-4-0.0",
       sideEffects: false,
       main: "./dist/index.js",
       module: "./dist/index.mjs",
@@ -87,6 +87,7 @@ var require_package = __commonJS({
         "@langchain/community": "^0.0.53",
         "@langchain/core": "^0.3.13",
         "@langchain/google-gauth": "^0.1.0",
+        "@langchain/langgraph-sdk": "^0.0.16",
         "@langchain/openai": "^0.0.28",
         "class-transformer": "^0.5.1",
         express: "^4.19.2",
@@ -96,6 +97,7 @@ var require_package = __commonJS({
         "groq-sdk": "^0.5.0",
         langchain: "^0.3.3",
         openai: "^4.50.0",
+        "partial-json": "^0.1.7",
         pino: "^9.2.0",
         "pino-pretty": "^11.2.1",
         "reflect-metadata": "^0.2.2",
@@ -136,6 +138,7 @@ __export(src_exports, {
   UnknownErrorResponse: () => UnknownErrorResponse,
   buildSchema: () => buildSchema,
   config: () => config,
+  copilotKitEndpoint: () => copilotKitEndpoint,
   copilotRuntimeNestEndpoint: () => copilotRuntimeNestEndpoint,
   copilotRuntimeNextJSAppRouterEndpoint: () => copilotRuntimeNextJSAppRouterEndpoint,
   copilotRuntimeNextJSPagesRouterEndpoint: () => copilotRuntimeNextJSPagesRouterEndpoint,
@@ -143,7 +146,8 @@ __export(src_exports, {
   copilotRuntimeNodeHttpEndpoint: () => copilotRuntimeNodeHttpEndpoint,
   createContext: () => createContext,
   flattenToolCallsNoDuplicates: () => flattenToolCallsNoDuplicates,
-  getCommonConfig: () => getCommonConfig
+  getCommonConfig: () => getCommonConfig,
+  langGraphCloudEndpoint: () => langGraphCloudEndpoint
 });
 module.exports = __toCommonJS(src_exports);
 var import_reflect_metadata = require("reflect-metadata");
@@ -683,8 +687,7 @@ var LangChainAdapter = class {
     this.options = options;
   }
   async process(request) {
-    const { eventSource, model, actions, messages, runId } = request;
-    const threadId = request.threadId ?? (0, import_shared3.randomId)();
+    const { eventSource, model, actions, messages, threadId, runId } = request;
     const result = await this.options.chainFn({
       messages: messages.map(convertMessageToLangChainMessage),
       tools: actions.map(convertActionInputToLangChainTool),
@@ -699,7 +702,7 @@ var LangChainAdapter = class {
       });
     });
     return {
-      threadId
+      threadId: threadId || (0, import_shared3.randomId)()
     };
   }
 };
@@ -710,16 +713,12 @@ var import_google_gauth = require("@langchain/google-gauth");
 var GoogleGenerativeAIAdapter = class extends LangChainAdapter {
   constructor(options) {
     super({
-      chainFn: async ({ messages, tools, threadId }) => {
+      chainFn: async ({ messages, tools }) => {
         const model = new import_google_gauth.ChatGoogle({
           modelName: (options == null ? void 0 : options.model) ?? "gemini-1.5-pro",
           apiVersion: "v1beta"
         }).bindTools(tools);
-        return model.stream(messages, {
-          metadata: {
-            conversation_id: threadId
-          }
-        });
+        return model.stream(messages);
       }
     });
   }
@@ -1355,58 +1354,19 @@ var LangGraphEventTypes;
   LangGraphEventTypes2["OnToolStart"] = "on_tool_start";
   LangGraphEventTypes2["OnToolEnd"] = "on_tool_end";
   LangGraphEventTypes2["OnCopilotKitStateSync"] = "on_copilotkit_state_sync";
-  LangGraphEventTypes2["OnCopilotKitEmitMessage"] = "on_copilotkit_emit_message";
-  LangGraphEventTypes2["OnCopilotKitEmitToolCall"] = "on_copilotkit_emit_tool_call";
+  LangGraphEventTypes2["OnCustomEvent"] = "on_custom_event";
 })(LangGraphEventTypes || (LangGraphEventTypes = {}));
+var CustomEventNames;
+(function(CustomEventNames2) {
+  CustomEventNames2["CopilotKitManuallyEmitMessage"] = "copilotkit_manually_emit_message";
+  CustomEventNames2["CopilotKitManuallyEmitToolCall"] = "copilotkit_manually_emit_tool_call";
+  CustomEventNames2["CopilotKitManuallyEmitIntermediateState"] = "copilotkit_manually_emit_intermediate_state";
+})(CustomEventNames || (CustomEventNames = {}));
 
 // src/agents/langgraph/event-source.ts
 var import_shared8 = require("@copilotkit/shared");
 var RemoteLangGraphEventSource = class {
   eventStream$ = new import_rxjs.ReplaySubject();
-  async streamResponse(response) {
-    const reader = response.body.getReader();
-    const decoder = new TextDecoder();
-    let buffer = [];
-    const eventStream$ = this.eventStream$;
-    function flushBuffer() {
-      const currentBuffer = buffer.join("");
-      if (currentBuffer.trim().length === 0) {
-        return;
-      }
-      const parts = currentBuffer.split("\n");
-      if (parts.length === 0) {
-        return;
-      }
-      const lastPartIsComplete = currentBuffer.endsWith("\n");
-      buffer = [];
-      if (!lastPartIsComplete) {
-        buffer.push(parts.pop());
-      }
-      parts.map((part) => part.trim()).filter((part) => part != "").forEach((part) => {
-        eventStream$.next(JSON.parse(part));
-      });
-    }
-    __name(flushBuffer, "flushBuffer");
-    try {
-      while (true) {
-        const { done, value } = await reader.read();
-        if (!done) {
-          buffer.push(decoder.decode(value, {
-            stream: true
-          }));
-        }
-        flushBuffer();
-        if (done) {
-          break;
-        }
-      }
-    } catch (error) {
-      console.error("Error in stream", error);
-      eventStream$.error(error);
-      return;
-    }
-    eventStream$.complete();
-  }
   shouldEmitToolCall(shouldEmitToolCalls, toolCallName) {
     if (typeof shouldEmitToolCalls === "boolean") {
       return shouldEmitToolCalls;
@@ -1429,20 +1389,25 @@ var RemoteLangGraphEventSource = class {
         } else {
           acc.content = null;
         }
-        if ((_h = (_g = (_f = event.data) == null ? void 0 : _f.chunk) == null ? void 0 : _g.kwargs) == null ? void 0 : _h.tool_call_chunks) {
+        const toolCallChunks = (
+          // @ts-expect-error -- LangGraph Cloud implementation stores data outside of kwargs
+          ((_h = (_g = (_f = event.data) == null ? void 0 : _f.chunk) == null ? void 0 : _g.kwargs) == null ? void 0 : _h.tool_call_chunks) ?? ((_j = (_i = event.data) == null ? void 0 : _i.chunk) == null ? void 0 : _j.tool_call_chunks)
+        );
+        const toolCallMessageId = ((_m = (_l = (_k = event.data) == null ? void 0 : _k.chunk) == null ? void 0 : _l.kwargs) == null ? void 0 : _m.id) ?? ((_o = (_n = event.data) == null ? void 0 : _n.chunk) == null ? void 0 : _o.id);
+        if (toolCallChunks && toolCallChunks.length > 0) {
           acc.prevToolCallMessageId = acc.toolCallMessageId;
-          acc.toolCallMessageId = (_i = event.data.chunk.kwargs) == null ? void 0 : _i.id;
-          if ((_j = event.data.chunk.kwargs.tool_call_chunks[0]) == null ? void 0 : _j.name) {
-            acc.toolCallName = event.data.chunk.kwargs.tool_call_chunks[0].name;
+          acc.toolCallMessageId = toolCallMessageId;
+          if ((_p = toolCallChunks[0]) == null ? void 0 : _p.name) {
+            acc.toolCallName = toolCallChunks[0].name;
           }
-          if ((_k = event.data.chunk.kwargs.tool_call_chunks[0]) == null ? void 0 : _k.id) {
-            acc.toolCallId = event.data.chunk.kwargs.tool_call_chunks[0].id;
+          if ((_q = toolCallChunks[0]) == null ? void 0 : _q.id) {
+            acc.toolCallId = toolCallChunks[0].id;
           }
           acc.prevMessageId = acc.messageId;
-          acc.messageId = (_n = (_m = (_l = event.data) == null ? void 0 : _l.chunk) == null ? void 0 : _m.kwargs) == null ? void 0 : _n.id;
+          acc.messageId = toolCallMessageId;
         } else if (acc.content && acc.content != "") {
           acc.prevMessageId = acc.messageId;
-          acc.messageId = (_q = (_p = (_o = event.data) == null ? void 0 : _o.chunk) == null ? void 0 : _p.kwargs) == null ? void 0 : _q.id;
+          acc.messageId = toolCallMessageId;
         } else {
           acc.prevToolCallMessageId = acc.toolCallMessageId;
           acc.prevMessageId = acc.messageId;
@@ -1467,7 +1432,7 @@ var RemoteLangGraphEventSource = class {
       prevMessageId: null,
       content: null
     }), (0, import_rxjs.mergeMap)((eventWithState) => {
-      var _a, _b, _c, _d, _e;
+      var _a, _b, _c, _d, _e, _f, _g, _h, _i;
       const events = [];
       let shouldEmitMessages = true;
       let shouldEmitToolCalls = false;
@@ -1490,32 +1455,33 @@ var RemoteLangGraphEventSource = class {
         });
       }
       switch (eventWithState.event.event) {
-        case LangGraphEventTypes.OnCopilotKitEmitMessage:
-          events.push({
-            type: RuntimeEventTypes.TextMessageStart,
-            messageId: eventWithState.event.message_id
-          });
-          events.push({
-            type: RuntimeEventTypes.TextMessageContent,
-            content: eventWithState.event.message
-          });
-          events.push({
-            type: RuntimeEventTypes.TextMessageEnd
-          });
-          break;
-        case LangGraphEventTypes.OnCopilotKitEmitToolCall:
-          events.push({
-            type: RuntimeEventTypes.ActionExecutionStart,
-            actionExecutionId: eventWithState.event.id,
-            actionName: eventWithState.event.name
-          });
-          events.push({
-            type: RuntimeEventTypes.ActionExecutionArgs,
-            args: JSON.stringify(eventWithState.event.args)
-          });
-          events.push({
-            type: RuntimeEventTypes.ActionExecutionEnd
-          });
+        case LangGraphEventTypes.OnCustomEvent:
+          if (eventWithState.event.name === CustomEventNames.CopilotKitManuallyEmitMessage) {
+            events.push({
+              type: RuntimeEventTypes.TextMessageStart,
+              messageId: eventWithState.event.data.message_id
+            });
+            events.push({
+              type: RuntimeEventTypes.TextMessageContent,
+              content: eventWithState.event.data.message
+            });
+            events.push({
+              type: RuntimeEventTypes.TextMessageEnd
+            });
+          } else if (eventWithState.event.name === CustomEventNames.CopilotKitManuallyEmitToolCall) {
+            events.push({
+              type: RuntimeEventTypes.ActionExecutionStart,
+              actionExecutionId: eventWithState.event.data.id,
+              actionName: eventWithState.event.data.name
+            });
+            events.push({
+              type: RuntimeEventTypes.ActionExecutionArgs,
+              args: JSON.stringify(eventWithState.event.data.args)
+            });
+            events.push({
+              type: RuntimeEventTypes.ActionExecutionEnd
+            });
+          }
           break;
         case LangGraphEventTypes.OnCopilotKitStateSync:
           events.push({
@@ -1550,7 +1516,8 @@ var RemoteLangGraphEventSource = class {
               });
             }
           }
-          const args = (_e = (_d = (_c = (_b = (_a = eventWithState.event.data) == null ? void 0 : _a.chunk) == null ? void 0 : _b.kwargs) == null ? void 0 : _c.tool_call_chunks) == null ? void 0 : _d[0]) == null ? void 0 : _e.args;
+          const args = ((_e = (_d = (_c = (_b = (_a = eventWithState.event.data) == null ? void 0 : _a.chunk) == null ? void 0 : _b.kwargs) == null ? void 0 : _c.tool_call_chunks) == null ? void 0 : _d[0]) == null ? void 0 : _e.args) ?? // @ts-expect-error -- sdf
+          ((_i = (_h = (_g = (_f = eventWithState.event.data) == null ? void 0 : _f.chunk) == null ? void 0 : _g.tool_call_chunks) == null ? void 0 : _h[0]) == null ? void 0 : _i.args);
           const content = eventWithState.content;
           if (args) {
             if (this.shouldEmitToolCall(shouldEmitToolCalls, eventWithState.toolCallName)) {
@@ -1601,70 +1568,543 @@ var RemoteLangGraphEventSource = class {
 };
 __name(RemoteLangGraphEventSource, "RemoteLangGraphEventSource");
 
-// src/lib/runtime/remote-actions.ts
-function isLangGraphAgentAction(action) {
-  if (!action) {
-    return false;
-  }
-  return typeof action.langGraphAgentHandler === "function";
+// src/lib/runtime/remote-lg-cloud-action.ts
+var import_langgraph_sdk = require("@langchain/langgraph-sdk");
+var import_node_crypto = require("crypto");
+var import_partial_json = require("partial-json");
+
+// src/graphql/types/enums.ts
+var import_type_graphql = require("type-graphql");
+var MessageRole;
+(function(MessageRole2) {
+  MessageRole2["user"] = "user";
+  MessageRole2["assistant"] = "assistant";
+  MessageRole2["system"] = "system";
+})(MessageRole || (MessageRole = {}));
+var ActionExecutionScope;
+(function(ActionExecutionScope2) {
+  ActionExecutionScope2["server"] = "server";
+  ActionExecutionScope2["client"] = "client";
+  ActionExecutionScope2["passThrough"] = "passThrough";
+})(ActionExecutionScope || (ActionExecutionScope = {}));
+var CopilotRequestType;
+(function(CopilotRequestType2) {
+  CopilotRequestType2["Chat"] = "Chat";
+  CopilotRequestType2["Task"] = "Task";
+  CopilotRequestType2["TextareaCompletion"] = "TextareaCompletion";
+  CopilotRequestType2["TextareaPopover"] = "TextareaPopover";
+  CopilotRequestType2["Suggestion"] = "Suggestion";
+})(CopilotRequestType || (CopilotRequestType = {}));
+(0, import_type_graphql.registerEnumType)(MessageRole, {
+  name: "MessageRole",
+  description: "The role of the message"
+});
+(0, import_type_graphql.registerEnumType)(ActionExecutionScope, {
+  name: "ActionExecutionScope",
+  description: "The scope of the action"
+});
+(0, import_type_graphql.registerEnumType)(CopilotRequestType, {
+  name: "CopilotRequestType",
+  description: "The type of Copilot request"
+});
+
+// src/lib/runtime/remote-lg-cloud-action.ts
+async function execute(args) {
+  return new ReadableStream({
+    async start(controller) {
+      try {
+        await streamEvents(controller, args);
+        controller.close();
+      } catch (err) {
+      }
+    }
+  });
 }
-__name(isLangGraphAgentAction, "isLangGraphAgentAction");
-function createHeaders(onBeforeRequest, graphqlContext) {
-  const headers = {
-    "Content-Type": "application/json"
+__name(execute, "execute");
+async function streamEvents(controller, args) {
+  const { deploymentUrl, langsmithApiKey, threadId: agrsInitialThreadId, agent, nodeName: initialNodeName, state: initialState, messages, actions, logger: logger2 } = args;
+  let nodeName = initialNodeName;
+  let state = initialState;
+  const { name, assistantId: initialAssistantId } = agent;
+  const client = new import_langgraph_sdk.Client({
+    apiUrl: deploymentUrl,
+    apiKey: langsmithApiKey
+  });
+  let initialThreadId = agrsInitialThreadId;
+  const wasInitiatedWithExistingThread = !!initialThreadId;
+  if (initialThreadId && initialThreadId.startsWith("ck-")) {
+    initialThreadId = initialThreadId.substring(3);
+  }
+  const assistants = await client.assistants.search();
+  const retrievedAssistant = assistants.find((a) => a.name === name);
+  const threadId = initialThreadId ?? (0, import_node_crypto.randomUUID)();
+  if (initialThreadId === threadId) {
+    await client.threads.get(threadId);
+  } else {
+    await client.threads.create({
+      threadId
+    });
+  }
+  let agentState = {
+    values: {}
   };
-  if (onBeforeRequest) {
-    const { headers: additionalHeaders } = onBeforeRequest({
-      ctx: graphqlContext
+  if (wasInitiatedWithExistingThread) {
+    agentState = await client.threads.getState(threadId);
+  }
+  const agentStateValues = agentState.values;
+  state.messages = agentStateValues.messages;
+  const mode = wasInitiatedWithExistingThread && nodeName != "__end__" ? "continue" : "start";
+  state = langGraphDefaultMergeState(state, formatMessages(messages), actions, name);
+  if (mode === "continue") {
+    await client.threads.updateState(threadId, {
+      values: state,
+      asNode: nodeName
     });
-    if (additionalHeaders) {
-      Object.assign(headers, additionalHeaders);
+  }
+  const assistantId = initialAssistantId ?? (retrievedAssistant == null ? void 0 : retrievedAssistant.assistant_id);
+  if (!assistantId) {
+    console.error(`
+      No agent found for the agent name specified in CopilotKit provider
+      Please check your available agents or provide an agent ID in the LangGraph Cloud endpoint definition.
+
+      
+      These are the available agents: [${assistants.map((a) => `${a.name} (ID: ${a.assistant_id})`).join(", ")}]
+      `);
+    throw new Error("No agent id found");
+  }
+  const graphInfo = await client.assistants.getGraph(assistantId);
+  const streamInput = mode === "start" ? state : null;
+  let streamingStateExtractor = new StreamingStateExtractor([]);
+  let prevNodeName = null;
+  let emitIntermediateStateUntilEnd = null;
+  let shouldExit = null;
+  let externalRunId = null;
+  const streamResponse2 = client.runs.stream(threadId, assistantId, {
+    input: streamInput,
+    streamMode: [
+      "events",
+      "values"
+    ]
+  });
+  const emit = /* @__PURE__ */ __name((message) => controller.enqueue(new TextEncoder().encode(message)), "emit");
+  let latestStateValues = {};
+  for await (const chunk of streamResponse2) {
+    if (![
+      "events",
+      "values",
+      "error"
+    ].includes(chunk.event))
+      continue;
+    if (chunk.event === "error") {
+      logger2.error(chunk, `Error event thrown: ${chunk.data.message}`);
+      throw new Error(`Error event thrown: ${chunk.data.message}`);
+    }
+    if (chunk.event === "values") {
+      latestStateValues = chunk.data;
+      continue;
+    }
+    const event = chunk.data;
+    const currentNodeName = event.name;
+    const eventType = event.event;
+    const runId = event.metadata.run_id;
+    externalRunId = runId;
+    const metadata = event.metadata;
+    shouldExit = shouldExit != null ? shouldExit : metadata["copilotkit:exit"];
+    const emitIntermediateState = metadata["copilotkit:emit-intermediate-state"];
+    const forceEmitIntermediateState = metadata["copilotkit:force-emit-intermediate-state"];
+    const manuallyEmitMessage = metadata["copilotkit:manually-emit-messages"];
+    const manuallyEmitToolCall = metadata["copilotkit:manually-emit-tool-calls"];
+    if (graphInfo["nodes"].some((node) => node.id === currentNodeName)) {
+      nodeName = currentNodeName;
+    }
+    if (!nodeName) {
+      continue;
+    }
+    if (forceEmitIntermediateState) {
+      if (eventType === "on_chain_end") {
+        state = event.data.output;
+        emit(getStateSyncEvent({
+          threadId,
+          runId,
+          agentName: agent.name,
+          nodeName,
+          state: event.data.output,
+          running: true,
+          active: true
+        }));
+      }
+      continue;
+    }
+    if (manuallyEmitMessage) {
+      if (eventType === "on_chain_end") {
+        state = event.data.output;
+        emit(JSON.stringify({
+          event: "on_copilotkit_emit_message",
+          message: event.data.output,
+          messageId: (0, import_node_crypto.randomUUID)(),
+          role: MessageRole.assistant
+        }) + "\n");
+      }
+      continue;
+    }
+    if (manuallyEmitToolCall) {
+      if (eventType === "on_chain_end") {
+        state = event.data.output;
+        emit(JSON.stringify({
+          event: "on_copilotkit_emit_tool_call",
+          name: event.data.output.name,
+          args: event.data.output.args,
+          id: event.data.output.id
+        }) + "\n");
+      }
+      continue;
+    }
+    if (emitIntermediateState && emitIntermediateStateUntilEnd == null) {
+      emitIntermediateStateUntilEnd = nodeName;
     }
+    if (emitIntermediateState && eventType === "on_chat_model_start") {
+      streamingStateExtractor = new StreamingStateExtractor(emitIntermediateState);
+    }
+    let updatedState = latestStateValues;
+    if (emitIntermediateState && eventType === "on_chat_model_stream") {
+      streamingStateExtractor.bufferToolCalls(event);
+    }
+    if (emitIntermediateStateUntilEnd !== null) {
+      updatedState = {
+        ...updatedState,
+        ...streamingStateExtractor.extractState()
+      };
+    }
+    if (!emitIntermediateState && currentNodeName === emitIntermediateStateUntilEnd && eventType === "on_chain_end") {
+      emitIntermediateStateUntilEnd = null;
+    }
+    const exitingNode = nodeName === currentNodeName && eventType === "on_chain_end";
+    if (JSON.stringify(updatedState) !== JSON.stringify(state) || prevNodeName != nodeName || exitingNode) {
+      state = updatedState;
+      prevNodeName = nodeName;
+      emit(getStateSyncEvent({
+        threadId,
+        runId,
+        agentName: agent.name,
+        nodeName,
+        state,
+        running: true,
+        active: !exitingNode
+      }));
+    }
+    emit(JSON.stringify(event) + "\n");
   }
-  return headers;
+  state = await client.threads.getState(threadId);
+  const isEndNode = state.next.length === 0;
+  nodeName = Object.keys(state.metadata.writes)[0];
+  emit(getStateSyncEvent({
+    threadId,
+    runId: externalRunId,
+    agentName: agent.name,
+    nodeName: isEndNode ? "__end__" : nodeName,
+    state: state.values,
+    running: !shouldExit,
+    active: false
+  }));
+  return Promise.resolve();
 }
-__name(createHeaders, "createHeaders");
-async function fetchRemoteInfo({ url, onBeforeRequest, graphqlContext, logger: logger2, frontendUrl }) {
-  logger2.debug({
-    url
-  }, "Fetching actions from url");
-  const headers = createHeaders(onBeforeRequest, graphqlContext);
-  try {
-    const response = await fetch(`${url}/info`, {
-      method: "POST",
-      headers,
-      body: JSON.stringify({
-        properties: graphqlContext.properties,
-        frontendUrl
-      })
-    });
-    if (!response.ok) {
-      logger2.error({
-        url,
-        status: response.status,
-        body: await response.text()
-      }, "Failed to fetch actions from url");
+__name(streamEvents, "streamEvents");
+function getStateSyncEvent({ threadId, runId, agentName, nodeName, state, running, active }) {
+  const stateWithoutMessages = Object.keys(state).reduce((acc, key) => {
+    if (key !== "messages") {
+      acc[key] = state[key];
+    }
+    return acc;
+  }, {});
+  return JSON.stringify({
+    event: "on_copilotkit_state_sync",
+    thread_id: threadId,
+    run_id: runId,
+    agent_name: agentName,
+    node_name: nodeName,
+    active,
+    state: stateWithoutMessages,
+    running,
+    role: "assistant"
+  }) + "\n";
+}
+__name(getStateSyncEvent, "getStateSyncEvent");
+var StreamingStateExtractor = /* @__PURE__ */ __name(class StreamingStateExtractor2 {
+  emitIntermediateState;
+  toolCallBuffer;
+  currentToolCall;
+  previouslyParsableState;
+  constructor(emitIntermediateState) {
+    this.emitIntermediateState = emitIntermediateState;
+    this.toolCallBuffer = {};
+    this.currentToolCall = null;
+    this.previouslyParsableState = {};
+  }
+  bufferToolCalls(event) {
+    if (event.data.chunk.tool_call_chunks.length > 0) {
+      const chunk = event.data.chunk.tool_call_chunks[0];
+      if (chunk.name !== null) {
+        this.currentToolCall = chunk.name;
+        this.toolCallBuffer[this.currentToolCall] = chunk.args;
+      } else if (this.currentToolCall !== null) {
+        this.toolCallBuffer[this.currentToolCall] += chunk.args;
+      }
+    }
+  }
+  getEmitStateConfig(currentToolName) {
+    for (const config2 of this.emitIntermediateState) {
+      const stateKey = config2["state_key"];
+      const tool = config2["tool"];
+      const toolArgument = config2["tool_argument"];
+      if (currentToolName === tool) {
+        return [
+          toolArgument,
+          stateKey
+        ];
+      }
+    }
+    return [
+      null,
+      null
+    ];
+  }
+  extractState() {
+    const state = {};
+    for (const [key, value] of Object.entries(this.toolCallBuffer)) {
+      const [argumentName, stateKey] = this.getEmitStateConfig(key);
+      if (stateKey === null) {
+        continue;
+      }
+      let parsedValue;
+      try {
+        parsedValue = (0, import_partial_json.parse)(value);
+      } catch (error) {
+        if (key in this.previouslyParsableState) {
+          parsedValue = this.previouslyParsableState[key];
+        } else {
+          continue;
+        }
+      }
+      this.previouslyParsableState[key] = parsedValue;
+      if (!argumentName) {
+        state[stateKey] = parsedValue;
+      } else {
+        state[stateKey] = parsedValue[argumentName];
+      }
+    }
+    return state;
+  }
+}, "StreamingStateExtractor");
+function langGraphDefaultMergeState(state, messages, actions, agentName) {
+  if (messages.length > 0 && "role" in messages[0] && messages[0].role === "system") {
+    messages = messages.slice(1);
+  }
+  const mergedMessages = state.messages || [];
+  const existingMessageIds = new Set(mergedMessages.map((message) => message.id));
+  const existingToolCallResults = /* @__PURE__ */ new Set();
+  for (const message of mergedMessages) {
+    if ("tool_call_id" in message) {
+      existingToolCallResults.add(message.tool_call_id);
+    }
+  }
+  for (const message of messages) {
+    if ("tool_calls" in message && message.tool_calls.length > 0 && message.tool_calls[0].name === agentName) {
+      continue;
+    }
+    if ("name" in message && message.name === agentName) {
+      continue;
+    }
+    if (!existingMessageIds.has(message.id)) {
+      if ("tool_call_id" in message && existingToolCallResults.has(message.tool_call_id)) {
+        console.warn("Warning: Duplicate tool call result, skipping:", message.tool_call_id);
+        continue;
+      }
+      mergedMessages.push(message);
+    } else {
+      for (let i = 0; i < mergedMessages.length; i++) {
+        if (mergedMessages[i].id === message.id) {
+          if ("tool_calls" in message) {
+            if (("tool_calls" in mergedMessages[i] || "additional_kwargs" in mergedMessages[i]) && mergedMessages[i].content) {
+              message.tool_calls = mergedMessages[i]["tool_calls"];
+              message.additional_kwargs = mergedMessages[i].additional_kwargs;
+            }
+          }
+          mergedMessages[i] = message;
+        }
+      }
+    }
+  }
+  for (let i = 0; i < mergedMessages.length - 1; i++) {
+    const currentMessage = mergedMessages[i];
+    const nextMessage = mergedMessages[i + 1];
+    if ("tool_calls" in currentMessage && currentMessage.tool_calls.length > 0 && "tool_call_id" in nextMessage) {
+      nextMessage.tool_call_id = currentMessage.tool_calls[0].id;
+    }
+  }
+  const correctedMessages = [];
+  for (let i = 0; i < mergedMessages.length; i++) {
+    const currentMessage = mergedMessages[i];
+    const nextMessage = mergedMessages[i + 1] || null;
+    const prevMessage = mergedMessages[i - 1] || null;
+    if ("tool_calls" in currentMessage && currentMessage.tool_calls.length > 0) {
+      if (!nextMessage) {
+        console.warn("No next message to auto-correct tool call, skipping:", currentMessage.tool_calls[0].id);
+        continue;
+      }
+      if (!("tool_call_id" in nextMessage) || nextMessage.tool_call_id !== currentMessage.tool_calls[0].id) {
+        const toolMessage = mergedMessages.find((m) => "tool_call_id" in m && m.tool_call_id === currentMessage.tool_calls[0].id);
+        if (toolMessage) {
+          console.warn("Auto-corrected tool call alignment issue:", currentMessage.tool_calls[0].id);
+          correctedMessages.push(currentMessage, toolMessage);
+          continue;
+        } else {
+          console.warn("No corresponding tool call result found for tool call, skipping:", currentMessage.tool_calls[0].id);
+          continue;
+        }
+      }
+      correctedMessages.push(currentMessage);
+      continue;
+    }
+    if ("tool_call_id" in currentMessage) {
+      if (!prevMessage || !("tool_calls" in prevMessage)) {
+        console.warn("No previous tool call, skipping tool call result:", currentMessage.id);
+        continue;
+      }
+      if (prevMessage.tool_calls && prevMessage.tool_calls[0].id !== currentMessage.tool_call_id) {
+        console.warn("Tool call id is incorrect, skipping tool call result:", currentMessage.id);
+        continue;
+      }
+      correctedMessages.push(currentMessage);
+      continue;
+    }
+    correctedMessages.push(currentMessage);
+  }
+  return deepMerge(state, {
+    messages: correctedMessages,
+    copilotkit: {
+      actions
+    }
+  });
+}
+__name(langGraphDefaultMergeState, "langGraphDefaultMergeState");
+function deepMerge(obj1, obj2) {
+  let result = {
+    ...obj1
+  };
+  for (let key in obj2) {
+    if (typeof obj2[key] === "object" && !Array.isArray(obj2[key])) {
+      if (obj1[key]) {
+        result[key] = deepMerge(obj1[key], obj2[key]);
+      } else {
+        result[key] = {
+          ...obj2[key]
+        };
+      }
+    } else {
+      result[key] = obj2[key];
+    }
+  }
+  return result;
+}
+__name(deepMerge, "deepMerge");
+function formatMessages(messages) {
+  return messages.map((message) => {
+    if (message.isTextMessage() && message.role === "assistant") {
+      return message;
+    }
+    if (message.isTextMessage() && message.role === "system") {
+      return message;
+    }
+    if (message.isTextMessage() && message.role === "user") {
+      return message;
+    }
+    if (message.isActionExecutionMessage()) {
+      const toolCall = {
+        name: message.name,
+        args: message.arguments,
+        id: message.id
+      };
       return {
-        actions: [],
-        agents: []
+        type: message.type,
+        content: "",
+        tool_calls: [
+          toolCall
+        ],
+        role: MessageRole.assistant,
+        id: message.id
       };
     }
-    const json = await response.json();
-    logger2.debug({
-      json
-    }, "Fetched actions from url");
-    return json;
-  } catch (error) {
-    logger2.error({
-      error: error.message ? error.message : error + ""
-    }, "Failed to fetch actions from url");
-    return {
-      actions: [],
-      agents: []
-    };
-  }
+    if (message.isResultMessage()) {
+      return {
+        type: message.type,
+        content: message.result,
+        id: message.id,
+        tool_call_id: message.actionExecutionId,
+        name: message.actionName,
+        role: MessageRole.assistant
+      };
+    }
+    throw new Error(`Unknown message type ${message.type}`);
+  });
 }
-__name(fetchRemoteInfo, "fetchRemoteInfo");
+__name(formatMessages, "formatMessages");
+
+// src/lib/runtime/remote-action-constructors.ts
+function constructLGCRemoteAction({ endpoint, graphqlContext, logger: logger2, messages, agentStates }) {
+  const agents = endpoint.agents.map((agent) => ({
+    name: agent.name,
+    description: agent.description,
+    parameters: [],
+    handler: async (_args) => {
+    },
+    langGraphAgentHandler: async ({ name, actionInputsWithoutAgents, threadId, nodeName }) => {
+      var _a;
+      logger2.debug({
+        actionName: agent.name
+      }, "Executing LangGraph Cloud agent");
+      telemetry_client_default.capture("oss.runtime.remote_action_executed", {});
+      let state = {};
+      if (agentStates) {
+        const jsonState = (_a = agentStates.find((state2) => state2.agentName === name)) == null ? void 0 : _a.state;
+        if (jsonState) {
+          state = JSON.parse(jsonState);
+        }
+      }
+      try {
+        const response = await execute({
+          logger: logger2,
+          deploymentUrl: endpoint.deploymentUrl,
+          langsmithApiKey: endpoint.langsmithApiKey,
+          agent,
+          threadId,
+          nodeName,
+          messages,
+          state,
+          properties: graphqlContext.properties,
+          actions: actionInputsWithoutAgents.map((action) => ({
+            name: action.name,
+            description: action.description,
+            parameters: JSON.parse(action.jsonSchema)
+          }))
+        });
+        const eventSource = new RemoteLangGraphEventSource();
+        streamResponse(response, eventSource.eventStream$);
+        return eventSource.processLangGraphEvents();
+      } catch (error) {
+        logger2.error({
+          url: endpoint.deploymentUrl,
+          status: 500,
+          body: error.message
+        }, "Failed to execute LangGraph Cloud agent");
+        throw new Error("Failed to execute LangGraph Cloud agent");
+      }
+    }
+  }));
+  return [
+    ...agents
+  ];
+}
+__name(constructLGCRemoteAction, "constructLGCRemoteAction");
 function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, logger: logger2, messages, agentStates }) {
   const actions = json["actions"].map((action) => ({
     name: action.name,
@@ -1756,7 +2196,7 @@ function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, lo
         throw new Error("Failed to execute remote agent");
       }
       const eventSource = new RemoteLangGraphEventSource();
-      eventSource.streamResponse(response);
+      streamResponse(response.body, eventSource.eventStream$);
       return eventSource.processLangGraphEvents();
     }
   }));
@@ -1766,34 +2206,165 @@ function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, lo
   ];
 }
 __name(constructRemoteActions, "constructRemoteActions");
-async function setupRemoteActions({ remoteActionDefinitions, graphqlContext, messages, agentStates, frontendUrl }) {
+async function streamResponse(response, eventStream$) {
+  const reader = response.getReader();
+  const decoder = new TextDecoder();
+  let buffer = [];
+  function flushBuffer() {
+    const currentBuffer = buffer.join("");
+    if (currentBuffer.trim().length === 0) {
+      return;
+    }
+    const parts = currentBuffer.split("\n");
+    if (parts.length === 0) {
+      return;
+    }
+    const lastPartIsComplete = currentBuffer.endsWith("\n");
+    buffer = [];
+    if (!lastPartIsComplete) {
+      buffer.push(parts.pop());
+    }
+    parts.map((part) => part.trim()).filter((part) => part != "").forEach((part) => {
+      eventStream$.next(JSON.parse(part));
+    });
+  }
+  __name(flushBuffer, "flushBuffer");
+  try {
+    while (true) {
+      const { done, value } = await reader.read();
+      if (!done) {
+        buffer.push(decoder.decode(value, {
+          stream: true
+        }));
+      }
+      flushBuffer();
+      if (done) {
+        break;
+      }
+    }
+  } catch (error) {
+    console.error("Error in stream", error);
+    eventStream$.error(error);
+    return;
+  }
+  eventStream$.complete();
+}
+__name(streamResponse, "streamResponse");
+function createHeaders(onBeforeRequest, graphqlContext) {
+  const headers = {
+    "Content-Type": "application/json"
+  };
+  if (onBeforeRequest) {
+    const { headers: additionalHeaders } = onBeforeRequest({
+      ctx: graphqlContext
+    });
+    if (additionalHeaders) {
+      Object.assign(headers, additionalHeaders);
+    }
+  }
+  return headers;
+}
+__name(createHeaders, "createHeaders");
+
+// src/lib/runtime/remote-actions.ts
+var EndpointType;
+(function(EndpointType2) {
+  EndpointType2["CopilotKit"] = "copilotKit";
+  EndpointType2["LangGraphCloud"] = "langgraph-cloud";
+})(EndpointType || (EndpointType = {}));
+function isLangGraphAgentAction(action) {
+  if (!action) {
+    return false;
+  }
+  return typeof action.langGraphAgentHandler === "function";
+}
+__name(isLangGraphAgentAction, "isLangGraphAgentAction");
+async function fetchRemoteInfo({ url, onBeforeRequest, graphqlContext, logger: logger2, frontendUrl }) {
+  logger2.debug({
+    url
+  }, "Fetching actions from url");
+  const headers = createHeaders(onBeforeRequest, graphqlContext);
+  try {
+    const response = await fetch(`${url}/info`, {
+      method: "POST",
+      headers,
+      body: JSON.stringify({
+        properties: graphqlContext.properties,
+        frontendUrl
+      })
+    });
+    if (!response.ok) {
+      logger2.error({
+        url,
+        status: response.status,
+        body: await response.text()
+      }, "Failed to fetch actions from url");
+      return {
+        actions: [],
+        agents: []
+      };
+    }
+    const json = await response.json();
+    logger2.debug({
+      json
+    }, "Fetched actions from url");
+    return json;
+  } catch (error) {
+    logger2.error({
+      error: error.message ? error.message : error + ""
+    }, "Failed to fetch actions from url");
+    return {
+      actions: [],
+      agents: []
+    };
+  }
+}
+__name(fetchRemoteInfo, "fetchRemoteInfo");
+async function setupRemoteActions({ remoteEndpointDefinitions, graphqlContext, messages, agentStates, frontendUrl }) {
   const logger2 = graphqlContext.logger.child({
     component: "remote-actions.fetchRemoteActions"
   });
   logger2.debug({
-    remoteActionDefinitions
-  }, "Fetching remote actions");
-  const filtered = remoteActionDefinitions.filter((value, index, self) => index === self.findIndex((t) => t.url === value.url));
-  const result = await Promise.all(filtered.map(async (actionDefinition) => {
+    remoteEndpointDefinitions
+  }, "Fetching from remote endpoints");
+  const filtered = remoteEndpointDefinitions.filter((value, index, self) => {
+    if (value.type === "langgraph-cloud") {
+      return value;
+    }
+    return index === self.findIndex((t) => t.url === value.url);
+  });
+  const result = await Promise.all(filtered.map(async (endpoint) => {
+    if (endpoint.type === "langgraph-cloud") {
+      return constructLGCRemoteAction({
+        endpoint,
+        messages,
+        graphqlContext,
+        logger: logger2.child({
+          component: "remote-actions.constructLGCRemoteAction",
+          endpoint
+        }),
+        agentStates
+      });
+    }
     const json = await fetchRemoteInfo({
-      url: actionDefinition.url,
-      onBeforeRequest: actionDefinition.onBeforeRequest,
+      url: endpoint.url,
+      onBeforeRequest: endpoint.onBeforeRequest,
       graphqlContext,
       logger: logger2.child({
         component: "remote-actions.fetchActionsFromUrl",
-        actionDefinition
+        endpoint
       }),
       frontendUrl
     });
     return constructRemoteActions({
       json,
       messages,
-      url: actionDefinition.url,
-      onBeforeRequest: actionDefinition.onBeforeRequest,
+      url: endpoint.url,
+      onBeforeRequest: endpoint.onBeforeRequest,
       graphqlContext,
       logger: logger2.child({
         component: "remote-actions.constructActions",
-        actionDefinition
+        endpoint
       }),
       agentStates
     });
@@ -1991,7 +2562,7 @@ async function executeAction(eventStream$, guardrailsResult$, action, actionArgu
 __name(executeAction, "executeAction");
 
 // src/graphql/types/base/index.ts
-var import_type_graphql = require("type-graphql");
+var import_type_graphql2 = require("type-graphql");
 function _ts_decorate(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function")
@@ -2014,15 +2585,15 @@ var BaseMessageInput = class {
 };
 __name(BaseMessageInput, "BaseMessageInput");
 _ts_decorate([
-  (0, import_type_graphql.Field)(() => String),
+  (0, import_type_graphql2.Field)(() => String),
   _ts_metadata("design:type", String)
 ], BaseMessageInput.prototype, "id", void 0);
 _ts_decorate([
-  (0, import_type_graphql.Field)(() => Date),
+  (0, import_type_graphql2.Field)(() => Date),
   _ts_metadata("design:type", typeof Date === "undefined" ? Object : Date)
 ], BaseMessageInput.prototype, "createdAt", void 0);
 BaseMessageInput = _ts_decorate([
-  (0, import_type_graphql.InputType)()
+  (0, import_type_graphql2.InputType)()
 ], BaseMessageInput);
 
 // src/graphql/types/converted/index.ts
@@ -2126,7 +2697,7 @@ __name(convertGqlInputToMessages, "convertGqlInputToMessages");
 var import_rxjs3 = require("rxjs");
 var CopilotRuntime = class {
   actions;
-  remoteActionDefinitions;
+  remoteEndpointDefinitions;
   langserve = [];
   onBeforeRequest;
   onAfterRequest;
@@ -2137,7 +2708,7 @@ var CopilotRuntime = class {
       const remoteChain = new RemoteChain(chain);
       this.langserve.push(remoteChain.toAction());
     }
-    this.remoteActionDefinitions = (params == null ? void 0 : params.remoteActions) || [];
+    this.remoteEndpointDefinitions = (params == null ? void 0 : params.remoteEndpoints) || [];
     this.onBeforeRequest = (_a = params == null ? void 0 : params.middleware) == null ? void 0 : _a.onBeforeRequest;
     this.onAfterRequest = (_b = params == null ? void 0 : params.middleware) == null ? void 0 : _b.onAfterRequest;
   }
@@ -2206,7 +2777,7 @@ var CopilotRuntime = class {
   async processAgentRequest(request) {
     var _a;
     const { messages: rawMessages, outputMessagesPromise, graphqlContext, agentSession } = request;
-    const { threadId = (0, import_shared9.randomId)(), agentName, nodeName } = agentSession;
+    const { threadId, agentName, nodeName } = agentSession;
     const serverSideActions = await this.getServerSideActions(request);
     const messages = convertGqlInputToMessages(rawMessages);
     const agent = serverSideActions.find((action) => action.name === agentName && isLangGraphAgentAction(action));
@@ -2278,8 +2849,12 @@ var CopilotRuntime = class {
         console.error("Error loading langserve chain:", error);
       }
     }
+    const remoteEndpointDefinitions = this.remoteEndpointDefinitions.map((endpoint) => ({
+      ...endpoint,
+      type: this.resolveEndpointType(endpoint)
+    }));
     const remoteActions = await setupRemoteActions({
-      remoteActionDefinitions: this.remoteActionDefinitions,
+      remoteEndpointDefinitions,
       graphqlContext,
       messages: inputMessages,
       agentStates,
@@ -2295,6 +2870,12 @@ var CopilotRuntime = class {
       ...remoteActions
     ];
   }
+  resolveEndpointType(endpoint) {
+    if (!endpoint.type && "langsmithApiKey" in endpoint && "deploymentUrl" in endpoint && "agents" in endpoint) {
+      return EndpointType.LangGraphCloud;
+    }
+    return endpoint.type;
+  }
 };
 __name(CopilotRuntime, "CopilotRuntime");
 function flattenToolCallsNoDuplicates(toolsByPriority) {
@@ -2309,6 +2890,20 @@ function flattenToolCallsNoDuplicates(toolsByPriority) {
   return allTools;
 }
 __name(flattenToolCallsNoDuplicates, "flattenToolCallsNoDuplicates");
+function copilotKitEndpoint(config2) {
+  return {
+    ...config2,
+    type: EndpointType.CopilotKit
+  };
+}
+__name(copilotKitEndpoint, "copilotKitEndpoint");
+function langGraphCloudEndpoint(config2) {
+  return {
+    ...config2,
+    type: EndpointType.LangGraphCloud
+  };
+}
+__name(langGraphCloudEndpoint, "langGraphCloudEndpoint");
 
 // src/lib/integrations/shared.ts
 var import_type_graphql16 = require("type-graphql");
@@ -2322,43 +2917,6 @@ var import_type_graphql11 = require("type-graphql");
 
 // src/graphql/inputs/message.input.ts
 var import_type_graphql3 = require("type-graphql");
-
-// src/graphql/types/enums.ts
-var import_type_graphql2 = require("type-graphql");
-var MessageRole;
-(function(MessageRole2) {
-  MessageRole2["user"] = "user";
-  MessageRole2["assistant"] = "assistant";
-  MessageRole2["system"] = "system";
-})(MessageRole || (MessageRole = {}));
-var ActionExecutionScope;
-(function(ActionExecutionScope2) {
-  ActionExecutionScope2["server"] = "server";
-  ActionExecutionScope2["client"] = "client";
-  ActionExecutionScope2["passThrough"] = "passThrough";
-})(ActionExecutionScope || (ActionExecutionScope = {}));
-var CopilotRequestType;
-(function(CopilotRequestType2) {
-  CopilotRequestType2["Chat"] = "Chat";
-  CopilotRequestType2["Task"] = "Task";
-  CopilotRequestType2["TextareaCompletion"] = "TextareaCompletion";
-  CopilotRequestType2["TextareaPopover"] = "TextareaPopover";
-  CopilotRequestType2["Suggestion"] = "Suggestion";
-})(CopilotRequestType || (CopilotRequestType = {}));
-(0, import_type_graphql2.registerEnumType)(MessageRole, {
-  name: "MessageRole",
-  description: "The role of the message"
-});
-(0, import_type_graphql2.registerEnumType)(ActionExecutionScope, {
-  name: "ActionExecutionScope",
-  description: "The scope of the action"
-});
-(0, import_type_graphql2.registerEnumType)(CopilotRequestType, {
-  name: "CopilotRequestType",
-  description: "The type of Copilot request"
-});
-
-// src/graphql/inputs/message.input.ts
 function _ts_decorate2(decorators, target, key, desc) {
   var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
   if (typeof Reflect === "object" && typeof Reflect.decorate === "function")
@@ -4020,6 +4578,7 @@ __name(copilotRuntimeNestEndpoint, "copilotRuntimeNestEndpoint");
   UnknownErrorResponse,
   buildSchema,
   config,
+  copilotKitEndpoint,
   copilotRuntimeNestEndpoint,
   copilotRuntimeNextJSAppRouterEndpoint,
   copilotRuntimeNextJSPagesRouterEndpoint,
@@ -4027,6 +4586,7 @@ __name(copilotRuntimeNestEndpoint, "copilotRuntimeNestEndpoint");
   copilotRuntimeNodeHttpEndpoint,
   createContext,
   flattenToolCallsNoDuplicates,
-  getCommonConfig
+  getCommonConfig,
+  langGraphCloudEndpoint
 });
 //# sourceMappingURL=index.js.map