@copilotkit/runtime 1.50.0-beta.10 → 1.50.0-beta.11

package/dist/index.mjs

@@ -1,5 +1,5 @@
 import 'reflect-metadata';
-import { TelemetryClient, CopilotKitLowLevelError, CopilotKitErrorCode, randomUUID, randomId, CopilotKitMisuseError, getZodParameters, readBody, CopilotKitError, CopilotKitAgentDiscoveryError, parseJson, convertJsonSchemaToZodSchema } from '@copilotkit/shared';
+import { TelemetryClient, CopilotKitLowLevelError, CopilotKitErrorCode, randomUUID, randomId, isTelemetryDisabled, CopilotKitMisuseError, getZodParameters, readBody, CopilotKitError, CopilotKitAgentDiscoveryError, parseJson, convertJsonSchemaToZodSchema } from '@copilotkit/shared';
 import { HumanMessage, AIMessage, SystemMessage, ToolMessage } from '@langchain/core/messages';
 import { DynamicStructuredTool } from '@langchain/core/tools';
 import { registerEnumType, Field, InputType, ObjectType, createUnionType, InterfaceType, Query, Ctx, Mutation, Arg, Resolver, buildSchemaSync } from 'type-graphql';
@@ -65,7 +65,7 @@ var require_package = __commonJS({
       publishConfig: {
         access: "public"
       },
-      version: "1.50.0-beta.9",
+      version: "1.50.0-beta.10",
       sideEffects: false,
       main: "./dist/index.js",
       module: "./dist/index.mjs",
@@ -3462,14 +3462,16 @@ var CopilotRuntime = class {
   constructor(params) {
     const agents = (params == null ? void 0 : params.agents) ?? {};
     const endpointAgents = this.assignEndpointsToAgents((params == null ? void 0 : params.remoteEndpoints) ?? []);
+    const baseRunner = (params == null ? void 0 : params.runner) ?? new InMemoryAgentRunner();
+    const runner = isTelemetryDisabled() ? baseRunner : new TelemetryAgentRunner({
+      runner: baseRunner
+    });
     this.runtimeArgs = {
       agents: {
         ...endpointAgents,
         ...agents
       },
-      // Use TelemetryAgentRunner by default to track agent execution telemetry
-      // Users can pass their own runner which will be wrapped for telemetry
-      runner: (params == null ? void 0 : params.runner) ?? new TelemetryAgentRunner(),
+      runner,
       // TODO: add support for transcriptionService from CopilotRuntimeOptionsVNext once it is ready
       // transcriptionService: params?.transcriptionService,
       beforeRequestMiddleware: this.createOnBeforeRequestHandler(params).bind(this),
@@ -3562,11 +3564,24 @@ var CopilotRuntime = class {
   }
   createOnBeforeRequestHandler(params) {
     return async (hookParams) => {
-      var _a, _b;
-      (_a = params == null ? void 0 : params.beforeRequestMiddleware) == null ? void 0 : _a.call(params, hookParams);
-      if ((_b = params == null ? void 0 : params.middleware) == null ? void 0 : _b.onBeforeRequest) {
-        const { request, runtime, path } = hookParams;
-        const body = await readBody(request);
+      var _a, _b, _c, _d;
+      const { request } = hookParams;
+      const publicApiKey = request.headers.get("x-copilotcloud-public-api-key");
+      const body = await readBody(request);
+      const forwardedProps = body.forwardedProps;
+      const cloudBaseUrl = process.env.COPILOT_CLOUD_BASE_URL || "https://api.cloud.copilotkit.ai";
+      telemetry_client_default.capture("oss.runtime.copilot_request_created", {
+        "cloud.guardrails.enabled": ((_a = forwardedProps == null ? void 0 : forwardedProps.cloud) == null ? void 0 : _a.guardrails) !== void 0,
+        requestType: ((_b = forwardedProps == null ? void 0 : forwardedProps.metadata) == null ? void 0 : _b.requestType) ?? "unknown",
+        "cloud.api_key_provided": !!publicApiKey,
+        ...publicApiKey ? {
+          "cloud.public_api_key": publicApiKey
+        } : {},
+        "cloud.base_url": cloudBaseUrl
+      });
+      (_c = params == null ? void 0 : params.beforeRequestMiddleware) == null ? void 0 : _c.call(params, hookParams);
+      if ((_d = params == null ? void 0 : params.middleware) == null ? void 0 : _d.onBeforeRequest) {
+        const { request: request2, runtime, path } = hookParams;
         const gqlMessages = aguiToGQL(body.messages).reduce((acc, msg) => {
           if ("role" in msg && msg.role === "user") {
             acc.inputMessages.push(msg);
@@ -3584,7 +3599,7 @@ var CopilotRuntime = class {
           runId: body.runId,
           inputMessages,
           properties: body.forwardedProps,
-          url: request.url
+          url: request2.url
         });
       }
     };