@copilotkit/runtime 1.6.0-next.7 → 1.6.0-next.9

package/CHANGELOG.md

@@ -1,5 +1,23 @@
 # @copilotkit/runtime
 
+## 1.6.0-next.9
+
+### Patch Changes
+
+- d07f49c: - fix(runtime): fix execution of runtime set backend action handlers
+  - @copilotkit/shared@1.6.0-next.9
+
+## 1.6.0-next.8
+
+### Minor Changes
+
+- fea916f: - feat: support input and output schema of langgraph
+  - docs: add input output schema docs
+
+### Patch Changes
+
+- @copilotkit/shared@1.6.0-next.8
+
 ## 1.6.0-next.7
 
 ### Patch Changes

package/dist/{chunk-LVJSZOFH.mjs → chunk-4HQ5OZXN.mjs}

@@ -2,7 +2,7 @@ import {
   copilotRuntimeNodeHttpEndpoint,
   getRuntimeInstanceTelemetryInfo,
   telemetry_client_default
-} from "./chunk-SAHKR6PK.mjs";
+} from "./chunk-L7OTGCAG.mjs";
 import {
   __name
 } from "./chunk-FHD4JECV.mjs";
@@ -22,4 +22,4 @@ __name(copilotRuntimeNodeExpressEndpoint, "copilotRuntimeNodeExpressEndpoint");
 export {
   copilotRuntimeNodeExpressEndpoint
 };
-//# sourceMappingURL=chunk-LVJSZOFH.mjs.map
+//# sourceMappingURL=chunk-4HQ5OZXN.mjs.map

package/dist/{chunk-LTUE24TH.mjs → chunk-5ZBUMQE2.mjs}

@@ -2,7 +2,7 @@ import {
   copilotRuntimeNodeHttpEndpoint,
   getRuntimeInstanceTelemetryInfo,
   telemetry_client_default
-} from "./chunk-SAHKR6PK.mjs";
+} from "./chunk-L7OTGCAG.mjs";
 import {
   __name
 } from "./chunk-FHD4JECV.mjs";
@@ -22,4 +22,4 @@ __name(copilotRuntimeNestEndpoint, "copilotRuntimeNestEndpoint");
 export {
   copilotRuntimeNestEndpoint
 };
-//# sourceMappingURL=chunk-LTUE24TH.mjs.map
+//# sourceMappingURL=chunk-5ZBUMQE2.mjs.map

package/dist/{chunk-DV4VS5ER.mjs → chunk-7BCFHZLY.mjs}

@@ -2,7 +2,7 @@ import {
   getCommonConfig,
   getRuntimeInstanceTelemetryInfo,
   telemetry_client_default
-} from "./chunk-SAHKR6PK.mjs";
+} from "./chunk-L7OTGCAG.mjs";
 import {
   __name
 } from "./chunk-FHD4JECV.mjs";
@@ -77,4 +77,4 @@ export {
   config,
   copilotRuntimeNextJSPagesRouterEndpoint
 };
-//# sourceMappingURL=chunk-DV4VS5ER.mjs.map
+//# sourceMappingURL=chunk-7BCFHZLY.mjs.map

package/dist/{chunk-SAHKR6PK.mjs → chunk-L7OTGCAG.mjs}

@@ -39,7 +39,7 @@ var require_package = __commonJS({
       publishConfig: {
         access: "public"
       },
-      version: "1.6.0-next.7",
+      version: "1.6.0-next.9",
       sideEffects: false,
       main: "./dist/index.js",
       module: "./dist/index.mjs",
@@ -1499,10 +1499,15 @@ var RemoteLangGraphEventSource = class {
         acc.lastMessageId = this.getCurrentMessageId(event) ?? acc.lastMessageId;
         const toolCallChunks = this.getCurrentToolCallChunks(event) ?? [];
         const responseMetadata = this.getResponseMetadata(event);
+        const toolCallCheck = toolCallChunks && toolCallChunks.length > 0;
+        let isToolCallEnd = (responseMetadata == null ? void 0 : responseMetadata.finish_reason) === "tool_calls";
         acc.isToolCallStart = toolCallChunks.some((chunk) => chunk.name && chunk.id);
         acc.isMessageStart = prevMessageId !== acc.lastMessageId && !acc.isToolCallStart;
-        acc.isToolCall = toolCallChunks && toolCallChunks.length > 0;
-        acc.isToolCallEnd = (responseMetadata == null ? void 0 : responseMetadata.finish_reason) === "tool_calls";
+        if (acc.isToolCall && !toolCallCheck) {
+          isToolCallEnd = true;
+        }
+        acc.isToolCall = toolCallCheck;
+        acc.isToolCallEnd = isToolCallEnd;
         acc.isMessageEnd = (responseMetadata == null ? void 0 : responseMetadata.finish_reason) === "stop";
         ({ name: acc.lastToolCallName, id: acc.lastToolCallId } = toolCallChunks.find((chunk) => chunk.name && chunk.id) ?? {
           name: acc.lastToolCallName,
@@ -1550,13 +1555,12 @@ var RemoteLangGraphEventSource = class {
         });
       }
       const responseMetadata = this.getResponseMetadata(acc.event);
-      if ((responseMetadata == null ? void 0 : responseMetadata.finish_reason) === "tool_calls" && this.shouldEmitToolCall(shouldEmitToolCalls, acc.lastToolCallName)) {
+      if (acc.isToolCallEnd && this.shouldEmitToolCall(shouldEmitToolCalls, acc.lastToolCallName)) {
         events.push({
           type: RuntimeEventTypes.ActionExecutionEnd,
           actionExecutionId: acc.lastToolCallId
         });
-      }
-      if ((responseMetadata == null ? void 0 : responseMetadata.finish_reason) === "stop" && shouldEmitMessages) {
+      } else if ((responseMetadata == null ? void 0 : responseMetadata.finish_reason) === "stop" && shouldEmitMessages) {
         events.push({
           type: RuntimeEventTypes.TextMessageEnd,
           messageId: acc.lastMessageId
@@ -1797,6 +1801,11 @@ async function streamEvents(controller, args) {
     });
   }
   const graphInfo = await client.assistants.getGraph(assistantId);
+  const graphSchema = await client.assistants.getSchemas(assistantId);
+  const schemaKeys = getSchemaKeys(graphSchema);
+  if (payload.input && schemaKeys.input) {
+    payload.input = Object.fromEntries(Object.entries(payload.input).filter(([key]) => schemaKeys.input.includes(key)));
+  }
   let streamingStateExtractor = new StreamingStateExtractor([]);
   let prevNodeName = null;
   let emitIntermediateStateUntilEnd = null;
@@ -1888,7 +1897,8 @@ async function streamEvents(controller, args) {
           nodeName,
           state: manuallyEmittedState,
           running: true,
-          active: true
+          active: true,
+          schemaKeys
         }));
         continue;
       }
@@ -1920,7 +1930,8 @@ async function streamEvents(controller, args) {
           nodeName,
           state,
           running: true,
-          active: !exitingNode
+          active: !exitingNode,
+          schemaKeys
         }));
       }
       emit(JSON.stringify(event) + "\n");
@@ -1938,7 +1949,8 @@ async function streamEvents(controller, args) {
       state: state.values,
       running: !shouldExit,
       active: false,
-      includeMessages: true
+      includeMessages: true,
+      schemaKeys
     }));
     return Promise.resolve();
   } catch (e) {
@@ -1951,7 +1963,7 @@ async function streamEvents(controller, args) {
   }
 }
 __name(streamEvents, "streamEvents");
-function getStateSyncEvent({ threadId, runId, agentName, nodeName, state, running, active, includeMessages = false }) {
+function getStateSyncEvent({ threadId, runId, agentName, nodeName, state, running, active, includeMessages = false, schemaKeys }) {
   if (!includeMessages) {
     state = Object.keys(state).reduce((acc, key) => {
       if (key !== "messages") {
@@ -1965,6 +1977,9 @@ function getStateSyncEvent({ threadId, runId, agentName, nodeName, state, runnin
       messages: langchainMessagesToCopilotKit(state.messages || [])
     };
   }
+  if (schemaKeys.output) {
+    state = Object.fromEntries(Object.entries(state).filter(([key]) => schemaKeys.output.includes(key)));
+  }
   return JSON.stringify({
     event: LangGraphEventTypes.OnCopilotKitStateSync,
     thread_id: threadId,
@@ -2199,6 +2214,25 @@ function copilotkitMessagesToLangChain(messages) {
   return result;
 }
 __name(copilotkitMessagesToLangChain, "copilotkitMessagesToLangChain");
+function getSchemaKeys(graphSchema) {
+  const CONSTANT_KEYS = [
+    "messages",
+    "copilotkit"
+  ];
+  const inputSchema = Object.keys(graphSchema.input_schema.properties);
+  const outputSchema = Object.keys(graphSchema.output_schema.properties);
+  return {
+    input: inputSchema && inputSchema.length ? [
+      ...inputSchema,
+      ...CONSTANT_KEYS
+    ] : null,
+    output: outputSchema && outputSchema.length ? [
+      ...outputSchema,
+      ...CONSTANT_KEYS
+    ] : null
+  };
+}
+__name(getSchemaKeys, "getSchemaKeys");
 
 // src/lib/runtime/remote-action-constructors.ts
 import { CopilotKitError, CopilotKitLowLevelError } from "@copilotkit/shared";
@@ -2971,7 +3005,7 @@ please use an LLM adapter instead.`
         threadId,
         runId: void 0,
         eventSource,
-        serverSideActions: [],
+        serverSideActions,
         actionInputsWithoutAgents: allAvailableActions
       };
     } catch (error) {
@@ -4262,4 +4296,4 @@ export {
   getCommonConfig,
   copilotRuntimeNodeHttpEndpoint
 };
-//# sourceMappingURL=chunk-SAHKR6PK.mjs.map
+//# sourceMappingURL=chunk-L7OTGCAG.mjs.map