@copilotkit/runtime 1.4.0-pre-1-4-0.11 → 1.4.0-pre-1-4-0-alpha.10

package/dist/lib/integrations/nest/index.js

@@ -12,11 +12,11 @@ var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
 };
-var __copyProps = (to, from2, except, desc) => {
-  if (from2 && typeof from2 === "object" || typeof from2 === "function") {
-    for (let key of __getOwnPropNames(from2))
+var __copyProps = (to, from3, except, desc) => {
+  if (from3 && typeof from3 === "object" || typeof from3 === "function") {
+    for (let key of __getOwnPropNames(from3))
       if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from2[key], enumerable: !(desc = __getOwnPropDesc(from2, key)) || desc.enumerable });
+        __defProp(to, key, { get: () => from3[key], enumerable: !(desc = __getOwnPropDesc(from3, key)) || desc.enumerable });
   }
   return to;
 };
@@ -44,7 +44,7 @@ var require_package = __commonJS({
       publishConfig: {
         access: "public"
       },
-      version: "1.4.0-pre-1-4-0.11",
+      version: "1.4.0-pre-1-4-0-alpha.10",
       sideEffects: false,
       main: "./dist/index.js",
       module: "./dist/index.mjs",
@@ -136,7 +136,7 @@ var import_type_graphql16 = require("type-graphql");
 
 // src/graphql/resolvers/copilot.resolver.ts
 var import_type_graphql15 = require("type-graphql");
-var import_rxjs2 = require("rxjs");
+var import_rxjs3 = require("rxjs");
 
 // src/graphql/inputs/generate-copilot-response.input.ts
 var import_type_graphql11 = require("type-graphql");
@@ -1180,15 +1180,133 @@ CopilotResponse = _ts_decorate13([
 var import_graphql_yoga = require("graphql-yoga");
 
 // src/service-adapters/events.ts
-var import_rxjs = require("rxjs");
+var import_shared3 = require("@copilotkit/shared");
+var import_rxjs2 = require("rxjs");
+
+// src/lib/telemetry-client.ts
+var import_shared2 = require("@copilotkit/shared");
+
+// src/lib/runtime/remote-actions.ts
+var EndpointType;
+(function(EndpointType2) {
+  EndpointType2["CopilotKit"] = "copilotKit";
+  EndpointType2["LangGraphCloud"] = "langgraph-cloud";
+})(EndpointType || (EndpointType = {}));
 
 // src/lib/telemetry-client.ts
+var import_node_crypto = require("crypto");
+
+// src/lib/runtime/copilot-runtime.ts
 var import_shared = require("@copilotkit/shared");
+
+// src/graphql/types/converted/index.ts
+var Message = class extends BaseMessageInput {
+  type;
+  isTextMessage() {
+    return this.type === "TextMessage";
+  }
+  isActionExecutionMessage() {
+    return this.type === "ActionExecutionMessage";
+  }
+  isResultMessage() {
+    return this.type === "ResultMessage";
+  }
+  isAgentStateMessage() {
+    return this.type === "AgentStateMessage";
+  }
+};
+__name(Message, "Message");
+var TextMessage = class extends Message {
+  type = "TextMessage";
+  content;
+  role;
+};
+__name(TextMessage, "TextMessage");
+var ActionExecutionMessage = class extends Message {
+  type = "ActionExecutionMessage";
+  name;
+  arguments;
+  scope;
+};
+__name(ActionExecutionMessage, "ActionExecutionMessage");
+var ResultMessage = class extends Message {
+  type = "ResultMessage";
+  actionExecutionId;
+  actionName;
+  result;
+};
+__name(ResultMessage, "ResultMessage");
+var AgentStateMessage = class extends Message {
+  type = "AgentStateMessage";
+  threadId;
+  agentName;
+  nodeName;
+  runId;
+  active;
+  role;
+  state;
+  running;
+};
+__name(AgentStateMessage, "AgentStateMessage");
+
+// src/lib/runtime/copilot-runtime.ts
+var import_rxjs = require("rxjs");
+function resolveEndpointType(endpoint) {
+  if (!endpoint.type) {
+    if ("langsmithApiKey" in endpoint && "deploymentUrl" in endpoint && "agents" in endpoint) {
+      return EndpointType.LangGraphCloud;
+    } else {
+      return EndpointType.CopilotKit;
+    }
+  }
+  return endpoint.type;
+}
+__name(resolveEndpointType, "resolveEndpointType");
+
+// src/lib/telemetry-client.ts
 var packageJson = require_package();
-var telemetryClient = new import_shared.TelemetryClient({
+var telemetryClient = new import_shared2.TelemetryClient({
   packageName: packageJson.name,
   packageVersion: packageJson.version
 });
+function getRuntimeInstanceTelemetryInfo(runtime) {
+  const endpointsInfo = runtime.remoteEndpointDefinitions.reduce((acc, endpoint) => {
+    let info = {
+      ...acc
+    };
+    const endpointType = resolveEndpointType(endpoint);
+    if (!info.endpointTypes.includes(endpointType)) {
+      info = {
+        ...info,
+        endpointTypes: [
+          ...info.endpointTypes,
+          endpointType
+        ]
+      };
+    }
+    if (endpointType === EndpointType.LangGraphCloud) {
+      const ep = endpoint;
+      info = {
+        ...info,
+        agentsAmount: ep.agents.length,
+        hashedKey: (0, import_node_crypto.createHash)("sha256").update(ep.langsmithApiKey).digest("hex")
+      };
+    }
+    return info;
+  }, {
+    endpointTypes: [],
+    agentsAmount: null,
+    hashedKey: null
+  });
+  return {
+    actionsAmount: runtime.actions.length,
+    endpointsAmount: runtime.remoteEndpointDefinitions.length,
+    endpointTypes: endpointsInfo.endpointTypes,
+    agentsAmount: endpointsInfo.agentsAmount,
+    hashedLgcKey: endpointsInfo.hashedKey
+  };
+}
+__name(getRuntimeInstanceTelemetryInfo, "getRuntimeInstanceTelemetryInfo");
 var telemetry_client_default = telemetryClient;
 
 // src/service-adapters/events.ts
@@ -1242,58 +1360,8 @@ var UnknownErrorResponse = class extends FailedResponseStatus {
 };
 __name(UnknownErrorResponse, "UnknownErrorResponse");
 
-// src/graphql/types/converted/index.ts
-var Message = class extends BaseMessageInput {
-  type;
-  isTextMessage() {
-    return this.type === "TextMessage";
-  }
-  isActionExecutionMessage() {
-    return this.type === "ActionExecutionMessage";
-  }
-  isResultMessage() {
-    return this.type === "ResultMessage";
-  }
-  isAgentStateMessage() {
-    return this.type === "AgentStateMessage";
-  }
-};
-__name(Message, "Message");
-var TextMessage = class extends Message {
-  type = "TextMessage";
-  content;
-  role;
-};
-__name(TextMessage, "TextMessage");
-var ActionExecutionMessage = class extends Message {
-  type = "ActionExecutionMessage";
-  name;
-  arguments;
-  scope;
-};
-__name(ActionExecutionMessage, "ActionExecutionMessage");
-var ResultMessage = class extends Message {
-  type = "ResultMessage";
-  actionExecutionId;
-  actionName;
-  result;
-};
-__name(ResultMessage, "ResultMessage");
-var AgentStateMessage = class extends Message {
-  type = "AgentStateMessage";
-  threadId;
-  agentName;
-  nodeName;
-  runId;
-  active;
-  role;
-  state;
-  running;
-};
-__name(AgentStateMessage, "AgentStateMessage");
-
 // src/graphql/resolvers/copilot.resolver.ts
-var import_shared2 = require("@copilotkit/shared");
+var import_shared4 = require("@copilotkit/shared");
 function _ts_decorate14(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")
@@ -1399,9 +1467,9 @@ var CopilotResolver = class {
       });
     }
     logger2.debug("Setting up subjects");
-    const responseStatus$ = new import_rxjs2.ReplaySubject();
-    const interruptStreaming$ = new import_rxjs2.ReplaySubject();
-    const guardrailsResult$ = new import_rxjs2.ReplaySubject();
+    const responseStatus$ = new import_rxjs3.ReplaySubject();
+    const interruptStreaming$ = new import_rxjs3.ReplaySubject();
+    const guardrailsResult$ = new import_rxjs3.ReplaySubject();
     let outputMessages = [];
     let resolveOutputMessagesPromise;
     let rejectOutputMessagesPromise;
@@ -1410,7 +1478,7 @@ var CopilotResolver = class {
       rejectOutputMessagesPromise = reject;
     });
     logger2.debug("Processing");
-    const { eventSource, threadId = (0, import_shared2.randomId)(), runId, serverSideActions, actionInputsWithoutAgents } = await copilotRuntime.processRuntimeRequest({
+    const { eventSource, threadId = (0, import_shared4.randomId)(), runId, serverSideActions, actionInputsWithoutAgents } = await copilotRuntime.processRuntimeRequest({
       serviceAdapter,
       messages: data.messages,
       actions: data.frontend.actions,
@@ -1428,7 +1496,7 @@ var CopilotResolver = class {
     const response = {
       threadId,
       runId,
-      status: (0, import_rxjs2.firstValueFrom)(responseStatus$),
+      status: (0, import_rxjs3.firstValueFrom)(responseStatus$),
       messages: new import_graphql_yoga.Repeater(async (pushMessage, stopStreamingMessages) => {
         var _a2, _b2;
         logger2.debug("Messages repeater created");
@@ -1455,7 +1523,7 @@ var CopilotResolver = class {
                 });
                 outputMessages = [
                   (0, import_class_transformer.plainToInstance)(TextMessage, {
-                    id: (0, import_shared2.randomId)(),
+                    id: (0, import_shared4.randomId)(),
                     createdAt: /* @__PURE__ */ new Date(),
                     content: result.reason,
                     role: MessageRole.assistant
@@ -1489,8 +1557,8 @@ var CopilotResolver = class {
         }).pipe(
           // shareReplay() ensures that later subscribers will see the whole stream instead of
           // just the events that were emitted after the subscriber was added.
-          (0, import_rxjs2.shareReplay)(),
-          (0, import_rxjs2.finalize)(() => {
+          (0, import_rxjs3.shareReplay)(),
+          (0, import_rxjs3.finalize)(() => {
             logger2.debug("Event stream finalized");
           })
         );
@@ -1501,22 +1569,24 @@ var CopilotResolver = class {
               case RuntimeEventTypes.TextMessageStart:
                 const textMessageContentStream = eventStream.pipe(
                   // skip until this message start event
-                  (0, import_rxjs2.skipWhile)((e) => e !== event),
+                  (0, import_rxjs3.skipWhile)((e) => e !== event),
                   // take until the message end event
-                  (0, import_rxjs2.takeWhile)((e) => e.type != RuntimeEventTypes.TextMessageEnd)
+                  (0, import_rxjs3.takeWhile)((e) => !(e.type === RuntimeEventTypes.TextMessageEnd && e.messageId == event.messageId)),
+                  // filter out any other message events or message ids
+                  (0, import_rxjs3.filter)((e) => e.type == RuntimeEventTypes.TextMessageContent && e.messageId == event.messageId)
                 );
-                const streamingTextStatus = new import_rxjs2.Subject();
+                const streamingTextStatus = new import_rxjs3.Subject();
                 const messageId = event.messageId;
                 pushMessage({
                   id: messageId,
-                  status: (0, import_rxjs2.firstValueFrom)(streamingTextStatus),
+                  status: (0, import_rxjs3.firstValueFrom)(streamingTextStatus),
                   createdAt: /* @__PURE__ */ new Date(),
                   role: MessageRole.assistant,
                   content: new import_graphql_yoga.Repeater(async (pushTextChunk, stopStreamingText) => {
                     logger2.debug("Text message content repeater created");
                     const textChunks = [];
                     let textSubscription;
-                    interruptStreaming$.pipe((0, import_rxjs2.shareReplay)(), (0, import_rxjs2.take)(1), (0, import_rxjs2.tap)(({ reason, messageId: messageId2 }) => {
+                    interruptStreaming$.pipe((0, import_rxjs3.shareReplay)(), (0, import_rxjs3.take)(1), (0, import_rxjs3.tap)(({ reason, messageId: messageId2 }) => {
                       logger2.debug({
                         reason,
                         messageId: messageId2
@@ -1567,11 +1637,17 @@ var CopilotResolver = class {
                 break;
               case RuntimeEventTypes.ActionExecutionStart:
                 logger2.debug("Action execution start event received");
-                const actionExecutionArgumentStream = eventStream.pipe((0, import_rxjs2.skipWhile)((e) => e !== event), (0, import_rxjs2.takeWhile)((e) => e.type != RuntimeEventTypes.ActionExecutionEnd));
-                const streamingArgumentsStatus = new import_rxjs2.Subject();
+                const actionExecutionArgumentStream = eventStream.pipe(
+                  (0, import_rxjs3.skipWhile)((e) => e !== event),
+                  // take until the action execution end event
+                  (0, import_rxjs3.takeWhile)((e) => !(e.type === RuntimeEventTypes.ActionExecutionEnd && e.actionExecutionId == event.actionExecutionId)),
+                  // filter out any other action execution events or action execution ids
+                  (0, import_rxjs3.filter)((e) => e.type == RuntimeEventTypes.ActionExecutionArgs && e.actionExecutionId == event.actionExecutionId)
+                );
+                const streamingArgumentsStatus = new import_rxjs3.Subject();
                 pushMessage({
                   id: event.actionExecutionId,
-                  status: (0, import_rxjs2.firstValueFrom)(streamingArgumentsStatus),
+                  status: (0, import_rxjs3.firstValueFrom)(streamingArgumentsStatus),
                   createdAt: /* @__PURE__ */ new Date(),
                   name: event.actionName,
                   scope: event.scope,
@@ -1618,7 +1694,7 @@ var CopilotResolver = class {
                   result: event.result
                 }, "Action execution result event received");
                 pushMessage({
-                  id: (0, import_shared2.randomId)(),
+                  id: (0, import_shared4.randomId)(),
                   status: new SuccessMessageStatus(),
                   createdAt: /* @__PURE__ */ new Date(),
                   actionExecutionId: event.actionExecutionId,
@@ -1626,7 +1702,7 @@ var CopilotResolver = class {
                   result: event.result
                 });
                 outputMessages.push((0, import_class_transformer.plainToInstance)(ResultMessage, {
-                  id: (0, import_shared2.randomId)(),
+                  id: (0, import_shared4.randomId)(),
                   createdAt: /* @__PURE__ */ new Date(),
                   actionExecutionId: event.actionExecutionId,
                   actionName: event.actionName,
@@ -1638,7 +1714,7 @@ var CopilotResolver = class {
                   event
                 }, "Agent message event received");
                 pushMessage({
-                  id: (0, import_shared2.randomId)(),
+                  id: (0, import_shared4.randomId)(),
                   status: new SuccessMessageStatus(),
                   threadId: event.threadId,
                   agentName: event.agentName,
@@ -1651,7 +1727,7 @@ var CopilotResolver = class {
                   createdAt: /* @__PURE__ */ new Date()
                 });
                 outputMessages.push((0, import_class_transformer.plainToInstance)(AgentStateMessage, {
-                  id: (0, import_shared2.randomId)(),
+                  id: (0, import_shared4.randomId)(),
                   threadId: event.threadId,
                   agentName: event.agentName,
                   nodeName: event.nodeName,
@@ -1681,7 +1757,7 @@ var CopilotResolver = class {
             logger2.debug("Event stream completed");
             if ((_a3 = data.cloud) == null ? void 0 : _a3.guardrails) {
               logger2.debug("Guardrails is enabled, waiting for guardrails result");
-              await (0, import_rxjs2.firstValueFrom)(guardrailsResult$);
+              await (0, import_rxjs3.firstValueFrom)(guardrailsResult$);
             }
             responseStatus$.next(new SuccessResponseStatus());
             eventStreamSubscription == null ? void 0 : eventStreamSubscription.unsubscribe();
@@ -1838,7 +1914,7 @@ function copilotRuntimeNodeHttpEndpoint(options) {
       _copilotkit: options.properties._copilotkit
     });
   }
-  telemetry_client_default.capture("oss.runtime.instance_created", {});
+  telemetry_client_default.capture("oss.runtime.instance_created", getRuntimeInstanceTelemetryInfo(options.runtime));
   const logger2 = commonConfig.logging;
   logger2.debug("Creating Node HTTP endpoint");
   const yoga = (0, import_graphql_yoga2.createYoga)({
@@ -1856,7 +1932,7 @@ function copilotRuntimeNestEndpoint(options) {
       framework: "nest"
     }
   });
-  telemetry_client_default.capture("oss.runtime.instance_created", {});
+  telemetry_client_default.capture("oss.runtime.instance_created", getRuntimeInstanceTelemetryInfo(options.runtime));
   return copilotRuntimeNodeHttpEndpoint(options);
 }
 __name(copilotRuntimeNestEndpoint, "copilotRuntimeNestEndpoint");