@copilotkit/runtime 1.3.16-mme-lgc-langgraph-package.11 → 1.3.16-mme-revert-rxjs-changes.10

package/src/agents/langgraph/events.ts

@@ -8,6 +8,8 @@ export enum LangGraphEventTypes {
   OnToolStart = "on_tool_start",
   OnToolEnd = "on_tool_end",
   OnCopilotKitStateSync = "on_copilotkit_state_sync",
+  OnCopilotKitEmitMessage = "on_copilotkit_emit_message",
+  OnCopilotKitEmitToolCall = "on_copilotkit_emit_tool_call",
   OnCustomEvent = "on_custom_event",
 }
 

package/src/lib/integrations/nest/index.ts

@@ -1,6 +1,6 @@
 import { CreateCopilotRuntimeServerOptions } from "../shared";
 import { copilotRuntimeNodeHttpEndpoint } from "../node-http";
-import telemetry from "../../telemetry-client";
+import telemetry, { getRuntimeInstanceTelemetryInfo } from "../../telemetry-client";
 
 export function copilotRuntimeNestEndpoint(options: CreateCopilotRuntimeServerOptions) {
   telemetry.setGlobalProperties({
@@ -9,6 +9,9 @@ export function copilotRuntimeNestEndpoint(options: CreateCopilotRuntimeServerOp
     },
   });
 
-  telemetry.capture("oss.runtime.instance_created", {});
+  telemetry.capture(
+    "oss.runtime.instance_created",
+    getRuntimeInstanceTelemetryInfo(options.runtime),
+  );
   return copilotRuntimeNodeHttpEndpoint(options);
 }

package/src/lib/integrations/nextjs/app-router.ts

@@ -1,6 +1,6 @@
 import { createYoga } from "graphql-yoga";
 import { CreateCopilotRuntimeServerOptions, getCommonConfig } from "../shared";
-import telemetry from "../../telemetry-client";
+import telemetry, { getRuntimeInstanceTelemetryInfo } from "../../telemetry-client";
 
 export function copilotRuntimeNextJSAppRouterEndpoint(options: CreateCopilotRuntimeServerOptions) {
   const commonConfig = getCommonConfig(options);
@@ -17,7 +17,10 @@ export function copilotRuntimeNextJSAppRouterEndpoint(options: CreateCopilotRunt
     });
   }
 
-  telemetry.capture("oss.runtime.instance_created", {});
+  telemetry.capture(
+    "oss.runtime.instance_created",
+    getRuntimeInstanceTelemetryInfo(options.runtime),
+  );
 
   const logger = commonConfig.logging;
   logger.debug("Creating NextJS App Router endpoint");

package/src/lib/integrations/nextjs/pages-router.ts

@@ -1,6 +1,6 @@
 import { YogaServerInstance, createYoga } from "graphql-yoga";
 import { CreateCopilotRuntimeServerOptions, GraphQLContext, getCommonConfig } from "../shared";
-import telemetry from "../../telemetry-client";
+import telemetry, { getRuntimeInstanceTelemetryInfo } from "../../telemetry-client";
 
 export const config = {
   api: {
@@ -32,7 +32,10 @@ export function copilotRuntimeNextJSPagesRouterEndpoint(
     });
   }
 
-  telemetry.capture("oss.runtime.instance_created", {});
+  telemetry.capture(
+    "oss.runtime.instance_created",
+    getRuntimeInstanceTelemetryInfo(options.runtime),
+  );
 
   const logger = commonConfig.logging;
   logger.debug("Creating NextJS Pages Router endpoint");

package/src/lib/integrations/node-express/index.ts

@@ -1,6 +1,6 @@
 import { CreateCopilotRuntimeServerOptions } from "../shared";
 import { copilotRuntimeNodeHttpEndpoint } from "../node-http";
-import telemetry from "../../telemetry-client";
+import telemetry, { getRuntimeInstanceTelemetryInfo } from "../../telemetry-client";
 
 export function copilotRuntimeNodeExpressEndpoint(options: CreateCopilotRuntimeServerOptions) {
   telemetry.setGlobalProperties({
@@ -9,6 +9,9 @@ export function copilotRuntimeNodeExpressEndpoint(options: CreateCopilotRuntimeS
     },
   });
 
-  telemetry.capture("oss.runtime.instance_created", {});
+  telemetry.capture(
+    "oss.runtime.instance_created",
+    getRuntimeInstanceTelemetryInfo(options.runtime),
+  );
   return copilotRuntimeNodeHttpEndpoint(options);
 }

package/src/lib/integrations/node-http/index.ts

@@ -1,6 +1,6 @@
 import { createYoga } from "graphql-yoga";
 import { CreateCopilotRuntimeServerOptions, getCommonConfig } from "../shared";
-import telemetry from "../../telemetry-client";
+import telemetry, { getRuntimeInstanceTelemetryInfo } from "../../telemetry-client";
 
 export function copilotRuntimeNodeHttpEndpoint(options: CreateCopilotRuntimeServerOptions) {
   const commonConfig = getCommonConfig(options);
@@ -17,7 +17,10 @@ export function copilotRuntimeNodeHttpEndpoint(options: CreateCopilotRuntimeServ
     });
   }
 
-  telemetry.capture("oss.runtime.instance_created", {});
+  telemetry.capture(
+    "oss.runtime.instance_created",
+    getRuntimeInstanceTelemetryInfo(options.runtime),
+  );
 
   const logger = commonConfig.logging;
   logger.debug("Creating Node HTTP endpoint");

package/src/lib/runtime/copilot-runtime.ts

@@ -12,7 +12,7 @@
  * ```
  */
 
-import { Action, actionParametersToJsonSchema, Parameter } from "@copilotkit/shared";
+import { Action, actionParametersToJsonSchema, Parameter, randomId } from "@copilotkit/shared";
 import { CopilotServiceAdapter, RemoteChain, RemoteChainParameters } from "../../service-adapters";
 import { MessageInput } from "../../graphql/inputs/message.input";
 import { ActionInput } from "../../graphql/inputs/action.input";
@@ -27,7 +27,7 @@ import {
   setupRemoteActions,
   EndpointDefinition,
   CopilotKitEndpoint,
-  LangGraphCloudEndpoint,
+  LangGraphPlatformEndpoint,
 } from "./remote-actions";
 import { GraphQLContext } from "../integrations/shared";
 import { AgentSessionInput } from "../../graphql/inputs/agent-session.input";
@@ -142,7 +142,7 @@ export interface CopilotRuntimeConstructorParams<T extends Parameter[] | [] = []
 
 export class CopilotRuntime<const T extends Parameter[] | [] = []> {
   public actions: ActionsConfiguration<T>;
-  private remoteEndpointDefinitions: EndpointDefinition[];
+  public remoteEndpointDefinitions: EndpointDefinition[];
   private langserve: Promise<Action<any>>[] = [];
   private onBeforeRequest?: OnBeforeRequestHandler;
   private onAfterRequest?: OnAfterRequestHandler;
@@ -155,7 +155,7 @@ export class CopilotRuntime<const T extends Parameter[] | [] = []> {
       this.langserve.push(remoteChain.toAction());
     }
 
-    this.remoteEndpointDefinitions = params?.remoteEndpoints || [];
+    this.remoteEndpointDefinitions = params?.remoteEndpoints ?? params?.remoteActions ?? [];
 
     this.onBeforeRequest = params?.middleware?.onBeforeRequest;
     this.onAfterRequest = params?.middleware?.onAfterRequest;
@@ -175,36 +175,36 @@ export class CopilotRuntime<const T extends Parameter[] | [] = []> {
       url,
     } = request;
 
-    if (agentSession) {
-      return this.processAgentRequest(request);
-    }
+    const eventSource = new RuntimeEventSource();
 
-    const messages = rawMessages.filter((message) => !message.agentStateMessage);
+    try {
+      if (agentSession) {
+        return await this.processAgentRequest(request);
+      }
 
-    const inputMessages = convertGqlInputToMessages(messages);
-    const serverSideActions = await this.getServerSideActions(request);
+      const messages = rawMessages.filter((message) => !message.agentStateMessage);
 
-    const serverSideActionsInput: ActionInput[] = serverSideActions.map((action) => ({
-      name: action.name,
-      description: action.description,
-      jsonSchema: JSON.stringify(actionParametersToJsonSchema(action.parameters)),
-    }));
+      const inputMessages = convertGqlInputToMessages(messages);
+      const serverSideActions = await this.getServerSideActions(request);
 
-    const actionInputs = flattenToolCallsNoDuplicates([
-      ...serverSideActionsInput,
-      ...clientSideActionsInput,
-    ]);
+      const serverSideActionsInput: ActionInput[] = serverSideActions.map((action) => ({
+        name: action.name,
+        description: action.description,
+        jsonSchema: JSON.stringify(actionParametersToJsonSchema(action.parameters)),
+      }));
 
-    await this.onBeforeRequest?.({
-      threadId,
-      runId,
-      inputMessages,
-      properties: graphqlContext.properties,
-      url,
-    });
+      const actionInputs = flattenToolCallsNoDuplicates([
+        ...serverSideActionsInput,
+        ...clientSideActionsInput,
+      ]);
 
-    try {
-      const eventSource = new RuntimeEventSource();
+      await this.onBeforeRequest?.({
+        threadId,
+        runId,
+        inputMessages,
+        properties: graphqlContext.properties,
+        url,
+      });
 
       const result = await serviceAdapter.process({
         messages: inputMessages,
@@ -244,7 +244,14 @@ export class CopilotRuntime<const T extends Parameter[] | [] = []> {
       };
     } catch (error) {
       console.error("Error getting response:", error);
-      throw error;
+      eventSource.sendErrorMessageToChat();
+      return {
+        threadId: threadId || randomId(),
+        runId: runId || randomId(),
+        eventSource,
+        serverSideActions: [],
+        actionInputsWithoutAgents: [],
+      };
     }
   }
 
@@ -344,7 +351,7 @@ export class CopilotRuntime<const T extends Parameter[] | [] = []> {
       (endpoint) =>
         ({
           ...endpoint,
-          type: this.resolveEndpointType(endpoint),
+          type: resolveEndpointType(endpoint),
         }) as EndpointDefinition,
     );
 
@@ -363,19 +370,6 @@ export class CopilotRuntime<const T extends Parameter[] | [] = []> {
 
     return [...configuredActions, ...langserveFunctions, ...remoteActions];
   }
-
-  private resolveEndpointType(endpoint: EndpointDefinition) {
-    if (
-      !endpoint.type &&
-      "langsmithApiKey" in endpoint &&
-      "deploymentUrl" in endpoint &&
-      "agents" in endpoint
-    ) {
-      return EndpointType.LangGraphCloud;
-    }
-
-    return endpoint.type;
-  }
 }
 
 export function flattenToolCallsNoDuplicates(toolsByPriority: ActionInput[]): ActionInput[] {
@@ -398,11 +392,23 @@ export function copilotKitEndpoint(config: Omit<CopilotKitEndpoint, "type">): Co
   };
 }
 
-export function langGraphCloudEndpoint(
-  config: Omit<LangGraphCloudEndpoint, "type">,
-): LangGraphCloudEndpoint {
+export function langGraphPlatformEndpoint(
+  config: Omit<LangGraphPlatformEndpoint, "type">,
+): LangGraphPlatformEndpoint {
   return {
     ...config,
-    type: EndpointType.LangGraphCloud,
+    type: EndpointType.LangGraphPlatform,
   };
 }
+
+export function resolveEndpointType(endpoint: EndpointDefinition) {
+  if (!endpoint.type) {
+    if ("langsmithApiKey" in endpoint && "deploymentUrl" in endpoint && "agents" in endpoint) {
+      return EndpointType.LangGraphPlatform;
+    } else {
+      return EndpointType.CopilotKit;
+    }
+  }
+
+  return endpoint.type;
+}

package/src/lib/runtime/remote-action-constructors.ts

@@ -1,8 +1,9 @@
+import { createHash } from "node:crypto";
 import {
   CopilotKitEndpoint,
   LangGraphAgentHandlerParams,
   RemoteActionInfoResponse,
-  LangGraphCloudEndpoint,
+  LangGraphPlatformEndpoint,
 } from "./remote-actions";
 import { GraphQLContext } from "../integrations";
 import { Logger } from "pino";
@@ -14,7 +15,7 @@ import telemetry from "../telemetry-client";
 import { RemoteLangGraphEventSource } from "../../agents/langgraph/event-source";
 import { Action } from "@copilotkit/shared";
 import { LangGraphEvent } from "../../agents/langgraph/events";
-import { execute } from "./remote-lg-cloud-action";
+import { execute } from "./remote-lg-action";
 
 export function constructLGCRemoteAction({
   endpoint,
@@ -23,7 +24,7 @@ export function constructLGCRemoteAction({
   messages,
   agentStates,
 }: {
-  endpoint: LangGraphCloudEndpoint;
+  endpoint: LangGraphPlatformEndpoint;
   graphqlContext: GraphQLContext;
   logger: Logger;
   messages: Message[];
@@ -40,9 +41,14 @@ export function constructLGCRemoteAction({
       threadId,
       nodeName,
     }: LangGraphAgentHandlerParams): Promise<Observable<RuntimeEvent>> => {
-      logger.debug({ actionName: agent.name }, "Executing LangGraph Cloud agent");
+      logger.debug({ actionName: agent.name }, "Executing LangGraph Platform agent");
 
-      telemetry.capture("oss.runtime.remote_action_executed", {});
+      telemetry.capture("oss.runtime.remote_action_executed", {
+        agentExecution: true,
+        type: "langgraph-platform",
+        agentsAmount: endpoint.agents.length,
+        hashedLgcKey: createHash("sha256").update(endpoint.langsmithApiKey).digest("hex"),
+      });
 
       let state = {};
       if (agentStates) {
@@ -76,9 +82,9 @@ export function constructLGCRemoteAction({
       } catch (error) {
         logger.error(
           { url: endpoint.deploymentUrl, status: 500, body: error.message },
-          "Failed to execute LangGraph Cloud agent",
+          "Failed to execute LangGraph Platform agent",
         );
-        throw new Error("Failed to execute LangGraph Cloud agent");
+        throw new Error("Failed to execute LangGraph Platform agent");
       }
     },
   }));
@@ -111,7 +117,11 @@ export function constructRemoteActions({
       logger.debug({ actionName: action.name, args }, "Executing remote action");
 
       const headers = createHeaders(onBeforeRequest, graphqlContext);
-      telemetry.capture("oss.runtime.remote_action_executed", {});
+      telemetry.capture("oss.runtime.remote_action_executed", {
+        agentExecution: false,
+        type: "self-hosted",
+        agentsAmount: json["agents"].length,
+      });
 
       try {
         const response = await fetch(`${url}/actions/execute`, {
@@ -162,7 +172,11 @@ export function constructRemoteActions({
       logger.debug({ actionName: agent.name }, "Executing remote agent");
 
       const headers = createHeaders(onBeforeRequest, graphqlContext);
-      telemetry.capture("oss.runtime.remote_action_executed", {});
+      telemetry.capture("oss.runtime.remote_action_executed", {
+        agentExecution: true,
+        type: "self-hosted",
+        agentsAmount: json["agents"].length,
+      });
 
       let state = {};
       if (agentStates) {

package/src/lib/runtime/remote-actions.ts

@@ -12,11 +12,11 @@ import {
   createHeaders,
 } from "./remote-action-constructors";
 
-export type EndpointDefinition = CopilotKitEndpoint | LangGraphCloudEndpoint;
+export type EndpointDefinition = CopilotKitEndpoint | LangGraphPlatformEndpoint;
 
 export enum EndpointType {
   CopilotKit = "copilotKit",
-  LangGraphCloud = "langgraph-cloud",
+  LangGraphPlatform = "langgraph-platform",
 }
 
 export interface BaseEndpointDefinition<TActionType extends EndpointType> {
@@ -30,17 +30,17 @@ export interface CopilotKitEndpoint extends BaseEndpointDefinition<EndpointType.
   };
 }
 
-export interface LangGraphCloudAgent {
+export interface LangGraphPlatformAgent {
   name: string;
   description: string;
   assistantId?: string;
 }
 
-export interface LangGraphCloudEndpoint
-  extends BaseEndpointDefinition<EndpointType.LangGraphCloud> {
+export interface LangGraphPlatformEndpoint
+  extends BaseEndpointDefinition<EndpointType.LangGraphPlatform> {
   deploymentUrl: string;
   langsmithApiKey: string;
-  agents: LangGraphCloudAgent[];
+  agents: LangGraphPlatformAgent[];
 }
 
 export type RemoteActionInfoResponse = {
@@ -127,7 +127,7 @@ export async function setupRemoteActions({
 
   // Remove duplicates of remoteEndpointDefinitions.url
   const filtered = remoteEndpointDefinitions.filter((value, index, self) => {
-    if (value.type === EndpointType.LangGraphCloud) {
+    if (value.type === EndpointType.LangGraphPlatform) {
       return value;
     }
     return index === self.findIndex((t: CopilotKitEndpoint) => t.url === value.url);
@@ -135,8 +135,8 @@ export async function setupRemoteActions({
 
   const result = await Promise.all(
     filtered.map(async (endpoint) => {
-      // Check for properties that can distinguish LG cloud from other actions
-      if (endpoint.type === EndpointType.LangGraphCloud) {
+      // Check for properties that can distinguish LG platform from other actions
+      if (endpoint.type === EndpointType.LangGraphPlatform) {
         return constructLGCRemoteAction({
           endpoint,
           messages,

package/src/lib/runtime/{remote-lg-cloud-action.ts → remote-lg-action.ts}

@@ -1,20 +1,21 @@
 import { Client } from "@langchain/langgraph-sdk";
-import { randomUUID } from "node:crypto";
+import { createHash, randomUUID } from "node:crypto";
 import { parse as parsePartialJson } from "partial-json";
 import { Logger } from "pino";
 import { ActionInput } from "../../graphql/inputs/action.input";
-import { LangGraphCloudAgent, LangGraphCloudEndpoint } from "./remote-actions";
+import { LangGraphPlatformAgent, LangGraphPlatformEndpoint } from "./remote-actions";
 import { CopilotRequestContextProperties } from "../integrations";
 import { Message, MessageType } from "../../graphql/types/converted";
 import { MessageRole } from "../../graphql/types/enums";
 import { CustomEventNames, LangGraphEventTypes } from "../../agents/langgraph/events";
+import telemetry from "../telemetry-client";
 
 type State = Record<string, any>;
 
 type ExecutionAction = Pick<ActionInput, "name" | "description"> & { parameters: string };
 
-interface ExecutionArgs extends Omit<LangGraphCloudEndpoint, "agents"> {
-  agent: LangGraphCloudAgent;
+interface ExecutionArgs extends Omit<LangGraphPlatformEndpoint, "agents"> {
+  agent: LangGraphPlatformAgent;
   threadId: string;
   nodeName: string;
   messages: Message[];
@@ -24,14 +25,14 @@ interface ExecutionArgs extends Omit<LangGraphCloudEndpoint, "agents"> {
   logger: Logger;
 }
 
-// The following types are our own definition to the messages accepted by LangGraph cloud, enhanced with some of our extra data.
+// The following types are our own definition to the messages accepted by LangGraph Platform, enhanced with some of our extra data.
 interface ToolCall {
   id: string;
   name: string;
   args: Record<string, unknown>;
 }
 
-type BaseLangGraphCloudMessage = Omit<
+type BaseLangGraphPlatformMessage = Omit<
   Message,
   | "isResultMessage"
   | "isTextMessage"
@@ -46,19 +47,19 @@ type BaseLangGraphCloudMessage = Omit<
   type: MessageType;
 };
 
-interface LangGraphCloudResultMessage extends BaseLangGraphCloudMessage {
+interface LangGraphPlatformResultMessage extends BaseLangGraphPlatformMessage {
   tool_call_id: string;
   name: string;
 }
 
-interface LangGraphCloudActionExecutionMessage extends BaseLangGraphCloudMessage {
+interface LangGraphPlatformActionExecutionMessage extends BaseLangGraphPlatformMessage {
   tool_calls: ToolCall[];
 }
 
-type LangGraphCloudMessage =
-  | LangGraphCloudActionExecutionMessage
-  | LangGraphCloudResultMessage
-  | BaseLangGraphCloudMessage;
+type LangGraphPlatformMessage =
+  | LangGraphPlatformActionExecutionMessage
+  | LangGraphPlatformResultMessage
+  | BaseLangGraphPlatformMessage;
 
 export async function execute(args: ExecutionArgs): Promise<ReadableStream<Uint8Array>> {
   return new ReadableStream({
@@ -88,7 +89,6 @@ async function streamEvents(controller: ReadableStreamDefaultController, args: E
   let state = initialState;
   const { name, assistantId: initialAssistantId } = agent;
 
-  // TODO: deploymentUrl is not required in local development
   const client = new Client({ apiUrl: deploymentUrl, apiKey: langsmithApiKey });
   let initialThreadId = agrsInitialThreadId;
   const wasInitiatedWithExistingThread = !!initialThreadId;
@@ -128,7 +128,7 @@ async function streamEvents(controller: ReadableStreamDefaultController, args: E
   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.\n
+      Please check your available agents or provide an agent ID in the LangGraph Platform endpoint definition.\n
       
       These are the available agents: [${assistants.map((a) => `${a.name} (ID: ${a.assistant_id})`).join(", ")}]
       `);
@@ -152,8 +152,22 @@ async function streamEvents(controller: ReadableStreamDefaultController, args: E
 
   let latestStateValues = {};
   let updatedState = state;
+  // If a manual emittance happens, it is the ultimate source of truth of state, unless a node has exited.
+  // Therefore, this value should either hold null, or the only edition of state that should be used.
+  let manuallyEmittedState = null;
+  let streamInfo: {
+    provider?: string;
+    langGraphHost?: string;
+    langGraphVersion?: string;
+    hashedLgcKey: string;
+  } = {
+    hashedLgcKey: createHash("sha256").update(langsmithApiKey).digest("hex"),
+  };
 
   try {
+    telemetry.capture("oss.runtime.agent_execution_stream_started", {
+      hashedLgcKey: streamInfo.hashedLgcKey,
+    });
     for await (const chunk of streamResponse) {
       if (!["events", "values", "error"].includes(chunk.event)) continue;
 
@@ -174,6 +188,16 @@ async function streamEvents(controller: ReadableStreamDefaultController, args: E
       externalRunId = runId;
       const metadata = event.metadata;
 
+      if (event.data?.output?.model != null && event.data?.output?.model != "") {
+        streamInfo.provider = event.data?.output?.model;
+      }
+      if (metadata.langgraph_host != null && metadata.langgraph_host != "") {
+        streamInfo.langGraphHost = metadata.langgraph_host;
+      }
+      if (metadata.langgraph_version != null && metadata.langgraph_version != "") {
+        streamInfo.langGraphVersion = metadata.langgraph_version;
+      }
+
       shouldExit =
         shouldExit ||
         (eventType === LangGraphEventTypes.OnCustomEvent &&
@@ -184,33 +208,36 @@ async function streamEvents(controller: ReadableStreamDefaultController, args: E
         eventType === LangGraphEventTypes.OnCustomEvent &&
         event.name === CustomEventNames.CopilotKitManuallyEmitIntermediateState;
 
+      const exitingNode =
+        nodeName === currentNodeName && eventType === LangGraphEventTypes.OnChainEnd;
+
+      // See manuallyEmittedState for explanation
+      if (exitingNode) {
+        manuallyEmittedState = null;
+      }
+
       // we only want to update the node name under certain conditions
       // since we don't need any internal node names to be sent to the frontend
       if (graphInfo["nodes"].some((node) => node.id === currentNodeName)) {
         nodeName = currentNodeName;
-
-        // only update state from values when entering or exiting a known node
-        if (
-          eventType === LangGraphEventTypes.OnChainStart ||
-          eventType === LangGraphEventTypes.OnChainEnd
-        ) {
-          updatedState = latestStateValues;
-        }
       }
 
+      updatedState = manuallyEmittedState ?? latestStateValues;
+
       if (!nodeName) {
         continue;
       }
 
       if (manuallyEmitIntermediateState) {
-        updatedState = event.data;
+        // See manuallyEmittedState for explanation
+        manuallyEmittedState = event.data;
         emit(
           getStateSyncEvent({
             threadId,
             runId,
             agentName: agent.name,
             nodeName,
-            state: updatedState,
+            state: manuallyEmittedState,
             running: true,
             active: true,
           }),
@@ -247,9 +274,6 @@ async function streamEvents(controller: ReadableStreamDefaultController, args: E
         emitIntermediateStateUntilEnd = null;
       }
 
-      const exitingNode =
-        nodeName === currentNodeName && eventType === LangGraphEventTypes.OnChainEnd;
-
       if (
         JSON.stringify(updatedState) !== JSON.stringify(state) ||
         prevNodeName != nodeName ||
@@ -277,6 +301,8 @@ async function streamEvents(controller: ReadableStreamDefaultController, args: E
     const isEndNode = state.next.length === 0;
     nodeName = Object.keys(state.metadata.writes)[0];
 
+    telemetry.capture("oss.runtime.agent_execution_stream_ended", streamInfo);
+
     emit(
       getStateSyncEvent({
         threadId,
@@ -413,7 +439,7 @@ class StreamingStateExtractor {
 // Start of Selection
 function langGraphDefaultMergeState(
   state: State,
-  messages: LangGraphCloudMessage[],
+  messages: LangGraphPlatformMessage[],
   actions: ExecutionAction[],
   agentName: string,
 ): State {
@@ -423,7 +449,7 @@ function langGraphDefaultMergeState(
   }
 
   // merge with existing messages
-  const mergedMessages: LangGraphCloudMessage[] = state.messages || [];
+  const mergedMessages: LangGraphPlatformMessage[] = state.messages || [];
   const existingMessageIds = new Set(mergedMessages.map((message) => message.id));
   const existingToolCallResults = new Set<string>();
 
@@ -489,7 +515,7 @@ function langGraphDefaultMergeState(
   }
 
   // try to auto-correct and log alignment issues
-  const correctedMessages: LangGraphCloudMessage[] = [];
+  const correctedMessages: LangGraphPlatformMessage[] = [];
 
   for (let i = 0; i < mergedMessages.length; i++) {
     const currentMessage = mergedMessages[i];
@@ -560,7 +586,7 @@ function langGraphDefaultMergeState(
   };
 }
 
-function formatMessages(messages: Message[]): LangGraphCloudMessage[] {
+function formatMessages(messages: Message[]): LangGraphPlatformMessage[] {
   return messages.map((message) => {
     if (message.isTextMessage() && message.role === "assistant") {
       return message;
@@ -583,7 +609,7 @@ function formatMessages(messages: Message[]): LangGraphCloudMessage[] {
         tool_calls: [toolCall],
         role: MessageRole.assistant,
         id: message.id,
-      } satisfies LangGraphCloudActionExecutionMessage;
+      } satisfies LangGraphPlatformActionExecutionMessage;
     }
     if (message.isResultMessage()) {
       return {
@@ -593,7 +619,7 @@ function formatMessages(messages: Message[]): LangGraphCloudMessage[] {
         tool_call_id: message.actionExecutionId,
         name: message.actionName,
         role: MessageRole.tool,
-      } satisfies LangGraphCloudResultMessage;
+      } satisfies LangGraphPlatformResultMessage;
     }
 
     throw new Error(`Unknown message type ${message.type}`);