@copilotkit/runtime 1.3.12-lgc-alpha-1.0 → 1.3.13

package/dist/index.js

@@ -44,7 +44,7 @@ var require_package = __commonJS({
       publishConfig: {
         access: "public"
       },
-      version: "1.3.12-lgc-alpha-1.0",
+      version: "1.3.13",
       sideEffects: false,
       main: "./dist/index.js",
       module: "./dist/index.mjs",
@@ -77,16 +77,16 @@ var require_package = __commonJS({
         "ts-node": "^10.9.2",
         tsconfig: "workspace:*",
         tsup: "^6.7.0",
-        typescript: "^5.2.3"
+        typescript: "^5.2.3",
+        "zod-to-json-schema": "^3.23.5"
       },
       dependencies: {
         "@anthropic-ai/sdk": "^0.27.3",
         "@copilotkit/shared": "workspace:*",
-        "@langchain/google-gauth": "^0.1.0",
         "@graphql-yoga/plugin-defer-stream": "^3.3.1",
         "@langchain/community": "^0.0.53",
         "@langchain/core": "^0.3.13",
-        "@langchain/langgraph-sdk": "^0.0.16",
+        "@langchain/google-gauth": "^0.1.0",
         "@langchain/openai": "^0.0.28",
         "class-transformer": "^0.5.1",
         express: "^4.19.2",
@@ -96,7 +96,6 @@ 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",
@@ -137,7 +136,6 @@ __export(src_exports, {
   UnknownErrorResponse: () => UnknownErrorResponse,
   buildSchema: () => buildSchema,
   config: () => config,
-  copilotKitEndpoint: () => copilotKitEndpoint,
   copilotRuntimeNestEndpoint: () => copilotRuntimeNestEndpoint,
   copilotRuntimeNextJSAppRouterEndpoint: () => copilotRuntimeNextJSAppRouterEndpoint,
   copilotRuntimeNextJSPagesRouterEndpoint: () => copilotRuntimeNextJSPagesRouterEndpoint,
@@ -145,8 +143,7 @@ __export(src_exports, {
   copilotRuntimeNodeHttpEndpoint: () => copilotRuntimeNodeHttpEndpoint,
   createContext: () => createContext,
   flattenToolCallsNoDuplicates: () => flattenToolCallsNoDuplicates,
-  getCommonConfig: () => getCommonConfig,
-  langGraphCloudEndpoint: () => langGraphCloudEndpoint
+  getCommonConfig: () => getCommonConfig
 });
 module.exports = __toCommonJS(src_exports);
 var import_reflect_metadata = require("reflect-metadata");
@@ -503,24 +500,27 @@ __name(convertMessageToLangChainMessage, "convertMessageToLangChainMessage");
 function convertJsonSchemaToZodSchema(jsonSchema, required) {
   if (jsonSchema.type === "object") {
     const spec = {};
+    if (!jsonSchema.properties || !Object.keys(jsonSchema.properties).length) {
+      return !required ? import_zod.z.object(spec).optional() : import_zod.z.object(spec);
+    }
     for (const [key, value] of Object.entries(jsonSchema.properties)) {
       spec[key] = convertJsonSchemaToZodSchema(value, jsonSchema.required ? jsonSchema.required.includes(key) : false);
     }
     let schema = import_zod.z.object(spec);
-    return !required ? schema.optional() : schema;
+    return required ? schema : schema.optional();
   } else if (jsonSchema.type === "string") {
     let schema = import_zod.z.string().describe(jsonSchema.description);
-    return !required ? schema.optional() : schema;
+    return required ? schema : schema.optional();
   } else if (jsonSchema.type === "number") {
     let schema = import_zod.z.number().describe(jsonSchema.description);
-    return !required ? schema.optional() : schema;
+    return required ? schema : schema.optional();
   } else if (jsonSchema.type === "boolean") {
     let schema = import_zod.z.boolean().describe(jsonSchema.description);
-    return !required ? schema.optional() : schema;
+    return required ? schema : schema.optional();
   } else if (jsonSchema.type === "array") {
     let itemSchema = convertJsonSchemaToZodSchema(jsonSchema.items, true);
     let schema = import_zod.z.array(itemSchema);
-    return !required ? schema.optional() : schema;
+    return required ? schema : schema.optional();
   }
 }
 __name(convertJsonSchemaToZodSchema, "convertJsonSchemaToZodSchema");
@@ -583,20 +583,39 @@ async function streamLangChainResponse({ result, eventStream$, actionExecution }
     maybeSendActionExecutionResultIsMessage(eventStream$, actionExecution);
     let reader = result.getReader();
     let mode = null;
+    const toolCallDetails = {
+      name: null,
+      id: null,
+      index: null,
+      prevIndex: null
+    };
     while (true) {
       try {
         const { done, value } = await reader.read();
         let toolCallName = void 0;
         let toolCallId = void 0;
         let toolCallArgs = void 0;
+        let toolCallIndex = void 0;
+        let toolCallPrevIndex = void 0;
         let hasToolCall = false;
         let content = value == null ? void 0 : value.content;
         if (isAIMessageChunk(value)) {
           let chunk = (_d = value.tool_call_chunks) == null ? void 0 : _d[0];
-          toolCallName = chunk == null ? void 0 : chunk.name;
-          toolCallId = chunk == null ? void 0 : chunk.id;
           toolCallArgs = chunk == null ? void 0 : chunk.args;
           hasToolCall = chunk != void 0;
+          if (chunk == null ? void 0 : chunk.name)
+            toolCallDetails.name = chunk.name;
+          if ((chunk == null ? void 0 : chunk.index) != null) {
+            toolCallDetails.index = chunk.index;
+            if (toolCallDetails.prevIndex == null)
+              toolCallDetails.prevIndex = chunk.index;
+          }
+          if (chunk == null ? void 0 : chunk.id)
+            toolCallDetails.id = chunk.index != null ? `${chunk.id}-idx-${chunk.index}` : chunk.id;
+          toolCallName = toolCallDetails.name;
+          toolCallId = toolCallDetails.id;
+          toolCallIndex = toolCallDetails.index;
+          toolCallPrevIndex = toolCallDetails.prevIndex;
         } else if (isBaseMessageChunk(value)) {
           let chunk = (_f = (_e = value.additional_kwargs) == null ? void 0 : _e.tool_calls) == null ? void 0 : _f[0];
           toolCallName = (_g = chunk == null ? void 0 : chunk.function) == null ? void 0 : _g.name;
@@ -615,7 +634,7 @@ async function streamLangChainResponse({ result, eventStream$, actionExecution }
           break;
         }
         if (mode === null) {
-          if (hasToolCall) {
+          if (hasToolCall && toolCallId && toolCallName) {
             mode = "function";
             eventStream$.sendActionExecutionStart(toolCallId, toolCallName);
           } else if (content) {
@@ -626,6 +645,11 @@ async function streamLangChainResponse({ result, eventStream$, actionExecution }
         if (mode === "message" && content) {
           eventStream$.sendTextMessageContent(Array.isArray(content) ? ((_i = content[0]) == null ? void 0 : _i.text) ?? "" : content);
         } else if (mode === "function" && toolCallArgs) {
+          if (toolCallIndex !== toolCallPrevIndex) {
+            eventStream$.sendActionExecutionEnd();
+            eventStream$.sendActionExecutionStart(toolCallId, toolCallName);
+            toolCallDetails.prevIndex = toolCallDetails.index;
+          }
           eventStream$.sendActionExecutionArgs(toolCallArgs);
         }
       } catch (error) {
@@ -1338,6 +1362,50 @@ var LangGraphEventTypes;
 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;
@@ -1360,25 +1428,20 @@ var RemoteLangGraphEventSource = class {
         } else {
           acc.content = null;
         }
-        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) {
+        if ((_h = (_g = (_f = event.data) == null ? void 0 : _f.chunk) == null ? void 0 : _g.kwargs) == null ? void 0 : _h.tool_call_chunks) {
           acc.prevToolCallMessageId = acc.toolCallMessageId;
-          acc.toolCallMessageId = toolCallMessageId;
-          if ((_p = toolCallChunks[0]) == null ? void 0 : _p.name) {
-            acc.toolCallName = toolCallChunks[0].name;
+          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;
           }
-          if ((_q = toolCallChunks[0]) == null ? void 0 : _q.id) {
-            acc.toolCallId = toolCallChunks[0].id;
+          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;
           }
           acc.prevMessageId = acc.messageId;
-          acc.messageId = toolCallMessageId;
+          acc.messageId = (_n = (_m = (_l = event.data) == null ? void 0 : _l.chunk) == null ? void 0 : _m.kwargs) == null ? void 0 : _n.id;
         } else if (acc.content && acc.content != "") {
           acc.prevMessageId = acc.messageId;
-          acc.messageId = toolCallMessageId;
+          acc.messageId = (_q = (_p = (_o = event.data) == null ? void 0 : _o.chunk) == null ? void 0 : _p.kwargs) == null ? void 0 : _q.id;
         } else {
           acc.prevToolCallMessageId = acc.toolCallMessageId;
           acc.prevMessageId = acc.messageId;
@@ -1403,7 +1466,7 @@ var RemoteLangGraphEventSource = class {
       prevMessageId: null,
       content: null
     }), (0, import_rxjs.mergeMap)((eventWithState) => {
-      var _a, _b, _c, _d, _e, _f, _g, _h, _i;
+      var _a, _b, _c, _d, _e;
       const events = [];
       let shouldEmitMessages = true;
       let shouldEmitToolCalls = false;
@@ -1486,8 +1549,7 @@ 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) ?? // @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 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 content = eventWithState.content;
           if (args) {
             if (this.shouldEmitToolCall(shouldEmitToolCalls, eventWithState.toolCallName)) {
@@ -1538,446 +1600,70 @@ var RemoteLangGraphEventSource = class {
 };
 __name(RemoteLangGraphEventSource, "RemoteLangGraphEventSource");
 
-// 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(execute, "execute");
-async function streamEvents(controller, args) {
-  const { deploymentUrl, langsmithApiKey, threadId: agrsInitialThreadId, agent, nodeName: initialNodeName, state: initialState, messages, actions } = 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 (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);
-  if (mode === "continue") {
-    await client.threads.updateState(threadId, {
-      values: state,
-      asNode: nodeName
-    });
-  }
-  const assistantId = initialAssistantId ?? retrievedAssistant.assistant_id;
-  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"
-    ].includes(chunk.event))
-      continue;
-    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");
-  }
-  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(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) {
-  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));
-  for (const message of messages) {
-    if (!existingMessageIds.has(message.id)) {
-      mergedMessages.push(message);
-    }
+// src/lib/runtime/remote-actions.ts
+function isLangGraphAgentAction(action) {
+  if (!action) {
+    return false;
   }
-  return deepMerge(state, {
-    messages: mergedMessages,
-    copilotkit: {
-      actions
-    }
-  });
+  return typeof action.langGraphAgentHandler === "function";
 }
-__name(langGraphDefaultMergeState, "langGraphDefaultMergeState");
-function deepMerge(obj1, obj2) {
-  let result = {
-    ...obj1
+__name(isLangGraphAgentAction, "isLangGraphAgentAction");
+function createHeaders(onBeforeRequest, graphqlContext) {
+  const headers = {
+    "Content-Type": "application/json"
   };
-  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];
+  if (onBeforeRequest) {
+    const { headers: additionalHeaders } = onBeforeRequest({
+      ctx: graphqlContext
+    });
+    if (additionalHeaders) {
+      Object.assign(headers, additionalHeaders);
     }
   }
-  return result;
+  return headers;
 }
-__name(deepMerge, "deepMerge");
-function formatMessages(messages) {
-  return messages.map((message) => {
-    if ("content" in message) {
-      return message;
-    }
-    if ("arguments" in message) {
-      const toolCall = {
-        name: message["name"],
-        args: message["arguments"],
-        id: message["id"]
-      };
-      return {
-        ...message,
-        content: "",
-        tool_calls: [
-          toolCall
-        ],
-        role: message["role"] ?? MessageRole.assistant
-      };
-    }
-    if ("actionExecutionId" in message) {
+__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");
       return {
-        ...message,
-        content: message["result"],
-        name: message["actionName"],
-        tool_call_id: message["actionExecutionId"],
-        role: message["role"] ?? MessageRole.user
+        actions: [],
+        agents: []
       };
     }
-    return message;
-  });
-}
-__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({
-          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
-  ];
+    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(constructLGCRemoteAction, "constructLGCRemoteAction");
+__name(fetchRemoteInfo, "fetchRemoteInfo");
 function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, logger: logger2, messages, agentStates }) {
   const actions = json["actions"].map((action) => ({
     name: action.name,
@@ -2069,7 +1755,7 @@ function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, lo
         throw new Error("Failed to execute remote agent");
       }
       const eventSource = new RemoteLangGraphEventSource();
-      streamResponse(response.body, eventSource.eventStream$);
+      eventSource.streamResponse(response);
       return eventSource.processLangGraphEvents();
     }
   }));
@@ -2079,165 +1765,34 @@ function constructRemoteActions({ json, url, onBeforeRequest, graphqlContext, lo
   ];
 }
 __name(constructRemoteActions, "constructRemoteActions");
-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 }) {
+async function setupRemoteActions({ remoteActionDefinitions, graphqlContext, messages, agentStates, frontendUrl }) {
   const logger2 = graphqlContext.logger.child({
     component: "remote-actions.fetchRemoteActions"
   });
   logger2.debug({
-    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
-      });
-    }
+    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) => {
     const json = await fetchRemoteInfo({
-      url: endpoint.url,
-      onBeforeRequest: endpoint.onBeforeRequest,
+      url: actionDefinition.url,
+      onBeforeRequest: actionDefinition.onBeforeRequest,
       graphqlContext,
       logger: logger2.child({
         component: "remote-actions.fetchActionsFromUrl",
-        endpoint
+        actionDefinition
       }),
       frontendUrl
     });
     return constructRemoteActions({
       json,
       messages,
-      url: endpoint.url,
-      onBeforeRequest: endpoint.onBeforeRequest,
+      url: actionDefinition.url,
+      onBeforeRequest: actionDefinition.onBeforeRequest,
       graphqlContext,
       logger: logger2.child({
         component: "remote-actions.constructActions",
-        endpoint
+        actionDefinition
       }),
       agentStates
     });
@@ -2401,7 +1956,13 @@ async function executeAction(eventStream$, guardrailsResult$, action, actionArgu
   }
   let args = [];
   if (actionArguments) {
-    args = JSON.parse(actionArguments);
+    try {
+      args = JSON.parse(actionArguments);
+    } catch (e) {
+      console.warn("Action argument unparsable", {
+        actionArguments
+      });
+    }
   }
   if (isLangGraphAgentAction(action)) {
     eventStream$.sendActionExecutionResult(actionExecutionId, action.name, `${action.name} agent started`);
@@ -2429,7 +1990,7 @@ async function executeAction(eventStream$, guardrailsResult$, action, actionArgu
 __name(executeAction, "executeAction");
 
 // src/graphql/types/base/index.ts
-var import_type_graphql2 = require("type-graphql");
+var import_type_graphql = 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")
@@ -2452,15 +2013,15 @@ var BaseMessageInput = class {
 };
 __name(BaseMessageInput, "BaseMessageInput");
 _ts_decorate([
-  (0, import_type_graphql2.Field)(() => String),
+  (0, import_type_graphql.Field)(() => String),
   _ts_metadata("design:type", String)
 ], BaseMessageInput.prototype, "id", void 0);
 _ts_decorate([
-  (0, import_type_graphql2.Field)(() => Date),
+  (0, import_type_graphql.Field)(() => Date),
   _ts_metadata("design:type", typeof Date === "undefined" ? Object : Date)
 ], BaseMessageInput.prototype, "createdAt", void 0);
 BaseMessageInput = _ts_decorate([
-  (0, import_type_graphql2.InputType)()
+  (0, import_type_graphql.InputType)()
 ], BaseMessageInput);
 
 // src/graphql/types/converted/index.ts
@@ -2564,7 +2125,7 @@ __name(convertGqlInputToMessages, "convertGqlInputToMessages");
 var import_rxjs3 = require("rxjs");
 var CopilotRuntime = class {
   actions;
-  remoteEndpointDefinitions;
+  remoteActionDefinitions;
   langserve = [];
   onBeforeRequest;
   onAfterRequest;
@@ -2575,7 +2136,7 @@ var CopilotRuntime = class {
       const remoteChain = new RemoteChain(chain);
       this.langserve.push(remoteChain.toAction());
     }
-    this.remoteEndpointDefinitions = (params == null ? void 0 : params.remoteEndpoints) || [];
+    this.remoteActionDefinitions = (params == null ? void 0 : params.remoteActions) || [];
     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;
   }
@@ -2644,7 +2205,7 @@ var CopilotRuntime = class {
   async processAgentRequest(request) {
     var _a;
     const { messages: rawMessages, outputMessagesPromise, graphqlContext, agentSession } = request;
-    const { threadId, agentName, nodeName } = agentSession;
+    const { threadId = (0, import_shared9.randomId)(), agentName, nodeName } = agentSession;
     const serverSideActions = await this.getServerSideActions(request);
     const messages = convertGqlInputToMessages(rawMessages);
     const agent = serverSideActions.find((action) => action.name === agentName && isLangGraphAgentAction(action));
@@ -2716,12 +2277,8 @@ 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({
-      remoteEndpointDefinitions,
+      remoteActionDefinitions: this.remoteActionDefinitions,
       graphqlContext,
       messages: inputMessages,
       agentStates,
@@ -2737,12 +2294,6 @@ 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) {
@@ -2757,20 +2308,6 @@ 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");
@@ -2784,6 +2321,43 @@ 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")
@@ -4445,7 +4019,6 @@ __name(copilotRuntimeNestEndpoint, "copilotRuntimeNestEndpoint");
   UnknownErrorResponse,
   buildSchema,
   config,
-  copilotKitEndpoint,
   copilotRuntimeNestEndpoint,
   copilotRuntimeNextJSAppRouterEndpoint,
   copilotRuntimeNextJSPagesRouterEndpoint,
@@ -4453,7 +4026,6 @@ __name(copilotRuntimeNestEndpoint, "copilotRuntimeNestEndpoint");
   copilotRuntimeNodeHttpEndpoint,
   createContext,
   flattenToolCallsNoDuplicates,
-  getCommonConfig,
-  langGraphCloudEndpoint
+  getCommonConfig
 });
 //# sourceMappingURL=index.js.map