@copilotkit/runtime 1.8.15-next.0 → 1.9.0-next.2

package/dist/index.js

@@ -44,7 +44,7 @@ var require_package = __commonJS({
       publishConfig: {
         access: "public"
       },
-      version: "1.8.15-next.0",
+      version: "1.9.0-next.2",
       sideEffects: false,
       main: "./dist/index.js",
       module: "./dist/index.mjs",
@@ -85,10 +85,12 @@ 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:*",
         "@graphql-yoga/plugin-defer-stream": "^3.3.1",
+        "@langchain/aws": "^0.1.9",
         "@langchain/community": "^0.3.29",
         "@langchain/core": "^0.3.38",
         "@langchain/google-gauth": "^0.1.0",
@@ -137,7 +139,9 @@ var require_package = __commonJS({
 var src_exports = {};
 __export(src_exports, {
   AnthropicAdapter: () => AnthropicAdapter,
+  BedrockAdapter: () => BedrockAdapter,
   CopilotRuntime: () => CopilotRuntime,
+  CustomEventNames: () => CustomEventNames2,
   EmptyAdapter: () => EmptyAdapter,
   ExperimentalEmptyAdapter: () => ExperimentalEmptyAdapter,
   ExperimentalOllamaAdapter: () => ExperimentalOllamaAdapter,
@@ -145,6 +149,7 @@ __export(src_exports, {
   GroqAdapter: () => GroqAdapter,
   GuardrailsValidationFailureResponse: () => GuardrailsValidationFailureResponse,
   LangChainAdapter: () => LangChainAdapter,
+  LangGraphAgent: () => LangGraphAgent,
   MessageStreamInterruptedResponse: () => MessageStreamInterruptedResponse,
   OpenAIAdapter: () => OpenAIAdapter,
   OpenAIAssistantAdapter: () => OpenAIAssistantAdapter,
@@ -2778,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
@@ -4641,6 +4646,27 @@ var ExperimentalOllamaAdapter = class {
 };
 __name(ExperimentalOllamaAdapter, "ExperimentalOllamaAdapter");
 
+// src/service-adapters/bedrock/bedrock-adapter.ts
+var import_aws = require("@langchain/aws");
+var BedrockAdapter = class extends LangChainAdapter {
+  constructor(options) {
+    super({
+      chainFn: async ({ messages, tools, threadId }) => {
+        const model = new import_aws.ChatBedrockConverse({
+          model: (options == null ? void 0 : options.model) ?? "amazon.nova-lite-v1:0",
+          region: (options == null ? void 0 : options.region) ?? "us-east-1",
+          credentials: (options == null ? void 0 : options.credentials) ? {
+            accessKeyId: options.credentials.accessKeyId,
+            secretAccessKey: options.credentials.secretAccessKey
+          } : void 0
+        }).bindTools(tools);
+        return model.stream(messages);
+      }
+    });
+  }
+};
+__name(BedrockAdapter, "BedrockAdapter");
+
 // src/service-adapters/empty/empty-adapter.ts
 var import_shared18 = require("@copilotkit/shared");
 var EmptyAdapter = class {
@@ -7055,10 +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,
@@ -7066,6 +7223,7 @@ __name(copilotRuntimeNestEndpoint, "copilotRuntimeNestEndpoint");
   GroqAdapter,
   GuardrailsValidationFailureResponse,
   LangChainAdapter,
+  LangGraphAgent,
   MessageStreamInterruptedResponse,
   OpenAIAdapter,
   OpenAIAssistantAdapter,