@copilotkit/runtime 1.8.15-next.1 → 1.9.0

package/dist/index.d.ts

@@ -4,6 +4,7 @@ export { CopilotRuntimeServerInstance, config, copilotRuntimeNextJSAppRouterEndp
 export { copilotRuntimeNodeHttpEndpoint } from './lib/integrations/node-http/index.js';
 export { copilotRuntimeNodeExpressEndpoint } from './lib/integrations/node-express/index.js';
 export { copilotRuntimeNestEndpoint } from './lib/integrations/nest/index.js';
+export { CustomEventNames, LangGraphAgent, PredictStateTool, State, TextMessageEvents, ToolCallEvents } from './lib/index.js';
 export { GuardrailsValidationFailureResponse, MessageStreamInterruptedResponse, UnknownErrorResponse } from './utils/index.js';
 export { C as CopilotRuntimeChatCompletionRequest, a as CopilotRuntimeChatCompletionResponse, b as CopilotServiceAdapter, c as RemoteChain, R as RemoteChainParameters } from './langserve-4a5c9217.js';
 export { AnthropicAdapter, AnthropicAdapterParams, BedrockAdapter, BedrockAdapterParams, EmptyAdapter, ExperimentalEmptyAdapter, ExperimentalOllamaAdapter } from './service-adapters/index.js';
@@ -21,4 +22,6 @@ import './graphql/types/base/index.js';
 import './lib/cloud/index.js';
 import '@ag-ui/client';
 import 'rxjs';
+import '@ag-ui/langgraph';
+import '@langchain/langgraph-sdk/dist/types.messages';
 import '@anthropic-ai/sdk';

package/dist/index.js

@@ -44,7 +44,7 @@ var require_package = __commonJS({
       publishConfig: {
         access: "public"
       },
-      version: "1.8.15-next.1",
+      version: "1.9.0",
       sideEffects: false,
       main: "./dist/index.js",
       module: "./dist/index.mjs",
@@ -85,6 +85,7 @@ var require_package = __commonJS({
         "@ag-ui/client": "0.0.28",
         "@ag-ui/core": "0.0.28",
         "@ag-ui/encoder": "0.0.28",
+        "@ag-ui/langgraph": "0.0.3",
         "@ag-ui/proto": "0.0.28",
         "@anthropic-ai/sdk": "^0.27.3",
         "@copilotkit/shared": "workspace:*",
@@ -140,6 +141,7 @@ __export(src_exports, {
   AnthropicAdapter: () => AnthropicAdapter,
   BedrockAdapter: () => BedrockAdapter,
   CopilotRuntime: () => CopilotRuntime,
+  CustomEventNames: () => CustomEventNames2,
   EmptyAdapter: () => EmptyAdapter,
   ExperimentalEmptyAdapter: () => ExperimentalEmptyAdapter,
   ExperimentalOllamaAdapter: () => ExperimentalOllamaAdapter,
@@ -147,6 +149,7 @@ __export(src_exports, {
   GroqAdapter: () => GroqAdapter,
   GuardrailsValidationFailureResponse: () => GuardrailsValidationFailureResponse,
   LangChainAdapter: () => LangChainAdapter,
+  LangGraphAgent: () => LangGraphAgent,
   MessageStreamInterruptedResponse: () => MessageStreamInterruptedResponse,
   OpenAIAdapter: () => OpenAIAdapter,
   OpenAIAssistantAdapter: () => OpenAIAssistantAdapter,
@@ -2780,11 +2783,11 @@ var MetaEventNames;
   MetaEventNames2["CopilotKitLangGraphInterruptEvent"] = "CopilotKitLangGraphInterruptEvent";
 })(MetaEventNames || (MetaEventNames = {}));
 var CustomEventNames;
-(function(CustomEventNames2) {
-  CustomEventNames2["CopilotKitManuallyEmitMessage"] = "copilotkit_manually_emit_message";
-  CustomEventNames2["CopilotKitManuallyEmitToolCall"] = "copilotkit_manually_emit_tool_call";
-  CustomEventNames2["CopilotKitManuallyEmitIntermediateState"] = "copilotkit_manually_emit_intermediate_state";
-  CustomEventNames2["CopilotKitExit"] = "copilotkit_exit";
+(function(CustomEventNames3) {
+  CustomEventNames3["CopilotKitManuallyEmitMessage"] = "copilotkit_manually_emit_message";
+  CustomEventNames3["CopilotKitManuallyEmitToolCall"] = "copilotkit_manually_emit_tool_call";
+  CustomEventNames3["CopilotKitManuallyEmitIntermediateState"] = "copilotkit_manually_emit_intermediate_state";
+  CustomEventNames3["CopilotKitExit"] = "copilotkit_exit";
 })(CustomEventNames || (CustomEventNames = {}));
 
 // src/agents/langgraph/event-source.ts
@@ -7078,11 +7081,141 @@ function copilotRuntimeNestEndpoint(options) {
   return copilotRuntimeNodeHttpEndpoint(options);
 }
 __name(copilotRuntimeNestEndpoint, "copilotRuntimeNestEndpoint");
+
+// src/lib/runtime/langgraph/langgraph-agent.ts
+var import_client = require("@ag-ui/client");
+var import_rxjs5 = require("rxjs");
+var import_langgraph = require("@ag-ui/langgraph");
+var CustomEventNames2;
+(function(CustomEventNames3) {
+  CustomEventNames3["CopilotKitManuallyEmitMessage"] = "copilotkit_manually_emit_message";
+  CustomEventNames3["CopilotKitManuallyEmitToolCall"] = "copilotkit_manually_emit_tool_call";
+  CustomEventNames3["CopilotKitManuallyEmitIntermediateState"] = "copilotkit_manually_emit_intermediate_state";
+  CustomEventNames3["CopilotKitExit"] = "copilotkit_exit";
+})(CustomEventNames2 || (CustomEventNames2 = {}));
+var LangGraphAgent = class extends import_langgraph.LangGraphAgent {
+  constructor(config2) {
+    super(config2);
+  }
+  dispatchEvent(event) {
+    if (event.type === import_client.EventType.CUSTOM) {
+      const customEvent = event;
+      if (customEvent.name === "copilotkit_manually_emit_message") {
+        this.subscriber.next({
+          type: import_client.EventType.TEXT_MESSAGE_START,
+          role: "assistant",
+          messageId: customEvent.value.message_id,
+          rawEvent: event
+        });
+        this.subscriber.next({
+          type: import_client.EventType.TEXT_MESSAGE_CONTENT,
+          messageId: customEvent.value.message_id,
+          delta: customEvent.value.message,
+          rawEvent: event
+        });
+        this.subscriber.next({
+          type: import_client.EventType.TEXT_MESSAGE_END,
+          messageId: customEvent.value.message_id,
+          rawEvent: event
+        });
+        return true;
+      }
+      if (customEvent.name === "copilotkit_manually_emit_tool_call") {
+        this.subscriber.next({
+          type: import_client.EventType.TOOL_CALL_START,
+          toolCallId: customEvent.value.id,
+          toolCallName: customEvent.value.name,
+          parentMessageId: customEvent.value.id,
+          rawEvent: event
+        });
+        this.subscriber.next({
+          type: import_client.EventType.TOOL_CALL_ARGS,
+          toolCallId: customEvent.value.id,
+          delta: customEvent.value.args,
+          rawEvent: event
+        });
+        this.subscriber.next({
+          type: import_client.EventType.TOOL_CALL_END,
+          toolCallId: customEvent.value.id,
+          rawEvent: event
+        });
+        return true;
+      }
+      if (customEvent.name === "copilotkit_manually_emit_intermediate_state") {
+        this.activeRun.manuallyEmittedState = customEvent.value;
+        this.dispatchEvent({
+          type: import_client.EventType.STATE_SNAPSHOT,
+          snapshot: this.getStateSnapshot(this.activeRun.manuallyEmittedState),
+          rawEvent: event
+        });
+        return true;
+      }
+      if (customEvent.name === "copilotkit_exit") {
+        this.subscriber.next({
+          type: import_client.EventType.CUSTOM,
+          name: "Exit",
+          value: true
+        });
+        return true;
+      }
+    }
+    const rawEvent = event.rawEvent;
+    if (!rawEvent) {
+      this.subscriber.next(event);
+      return true;
+    }
+    const isMessageEvent = event.type === import_client.EventType.TEXT_MESSAGE_START || event.type === import_client.EventType.TEXT_MESSAGE_CONTENT || event.type === import_client.EventType.TEXT_MESSAGE_END;
+    const isToolEvent = event.type === import_client.EventType.TOOL_CALL_START || event.type === import_client.EventType.TOOL_CALL_ARGS || event.type === import_client.EventType.TOOL_CALL_END;
+    if ("copilotkit:emit-tool-calls" in (rawEvent.metadata || {})) {
+      if (rawEvent.metadata["copilotkit:emit-tool-calls"] === false && isToolEvent) {
+        return false;
+      }
+    }
+    if ("copilotkit:emit-messages" in (rawEvent.metadata || {})) {
+      if (rawEvent.metadata["copilotkit:emit-messages"] === false && isMessageEvent) {
+        return false;
+      }
+    }
+    this.subscriber.next(event);
+    return true;
+  }
+  // @ts-ignore
+  run(input) {
+    return super.run(input).pipe((0, import_rxjs5.map)((processedEvent) => {
+      var _a, _b, _c, _d, _e;
+      if (processedEvent.type === import_client.EventType.RAW) {
+        const event = processedEvent.event ?? processedEvent.rawEvent;
+        const eventType = event.event;
+        const toolCallData = (_c = (_b = (_a = event.data) == null ? void 0 : _a.chunk) == null ? void 0 : _b.tool_call_chunks) == null ? void 0 : _c[0];
+        const toolCallUsedToPredictState = (_e = (_d = event.metadata) == null ? void 0 : _d["copilotkit:emit-intermediate-state"]) == null ? void 0 : _e.some((predictStateTool) => predictStateTool.tool === (toolCallData == null ? void 0 : toolCallData.name));
+        if (eventType === LangGraphEventTypes.OnChatModelStream && toolCallUsedToPredictState) {
+          return {
+            type: import_client.EventType.CUSTOM,
+            name: "PredictState",
+            value: event.metadata["copilotkit:emit-intermediate-state"]
+          };
+        }
+      }
+      return processedEvent;
+    }));
+  }
+  langGraphDefaultMergeState(state, messages, tools) {
+    const { tools: returnedTools, ...rest } = super.langGraphDefaultMergeState(state, messages, tools);
+    return {
+      ...rest,
+      copilotkit: {
+        actions: returnedTools
+      }
+    };
+  }
+};
+__name(LangGraphAgent, "LangGraphAgent");
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   AnthropicAdapter,
   BedrockAdapter,
   CopilotRuntime,
+  CustomEventNames,
   EmptyAdapter,
   ExperimentalEmptyAdapter,
   ExperimentalOllamaAdapter,
@@ -7090,6 +7223,7 @@ __name(copilotRuntimeNestEndpoint, "copilotRuntimeNestEndpoint");
   GroqAdapter,
   GuardrailsValidationFailureResponse,
   LangChainAdapter,
+  LangGraphAgent,
   MessageStreamInterruptedResponse,
   OpenAIAdapter,
   OpenAIAssistantAdapter,