@copilotkit/runtime 1.9.2-next.3 → 1.9.2-next.4

package/dist/index.mjs

@@ -1,18 +1,18 @@
 import {
   CustomEventNames,
   LangGraphAgent
-} from "./chunk-XDBXF3Q6.mjs";
+} from "./chunk-5FNEYAWH.mjs";
 import {
   config,
   copilotRuntimeNextJSAppRouterEndpoint,
   copilotRuntimeNextJSPagesRouterEndpoint
-} from "./chunk-WFYPJXWX.mjs";
+} from "./chunk-NK2XARWT.mjs";
 import {
   copilotRuntimeNestEndpoint
-} from "./chunk-MIPAKFI5.mjs";
+} from "./chunk-CJ44W4R2.mjs";
 import {
   copilotRuntimeNodeExpressEndpoint
-} from "./chunk-N24X5I3C.mjs";
+} from "./chunk-6W7TKZNY.mjs";
 import {
   CopilotRuntime,
   addCustomHeaderPlugin,
@@ -28,7 +28,7 @@ import {
   getCommonConfig,
   langGraphPlatformEndpoint,
   resolveEndpointType
-} from "./chunk-GSYE3DGY.mjs";
+} from "./chunk-ZUF3A5TW.mjs";
 import {
   AnthropicAdapter,
   BedrockAdapter,

package/dist/lib/index.d.ts

@@ -1,5 +1,5 @@
 export { G as GoogleGenerativeAIAdapter, f as GroqAdapter, e as GroqAdapterParams, L as LangChainAdapter, a as OpenAIAdapter, O as OpenAIAdapterParams, c as OpenAIAssistantAdapter, b as OpenAIAssistantAdapterParams, d as UnifyAdapter, U as UnifyAdapterParams } from '../groq-adapter-25a2bd35.js';
-export { e as CommonConfig, C as CopilotRequestContextProperties, k as CopilotRuntime, j as CopilotRuntimeConstructorParams, f as CopilotRuntimeLogger, i as CopilotRuntimeRequest, b as CreateCopilotRuntimeServerOptions, G as GraphQLContext, L as LogLevel, o as MCPClient, p as MCPEndpointConfig, M as MCPTool, a as addCustomHeaderPlugin, d as buildSchema, s as convertMCPToolsToActions, m as copilotKitEndpoint, c as createContext, h as createLogger, q as extractParametersFromSchema, l as flattenToolCallsNoDuplicates, t as generateMcpToolInstructions, g as getCommonConfig, n as langGraphPlatformEndpoint, r as resolveEndpointType } from '../shared-e272b15a.js';
+export { e as CommonConfig, C as CopilotRequestContextProperties, k as CopilotRuntime, j as CopilotRuntimeConstructorParams, f as CopilotRuntimeLogger, i as CopilotRuntimeRequest, b as CreateCopilotRuntimeServerOptions, G as GraphQLContext, L as LogLevel, o as MCPClient, p as MCPEndpointConfig, M as MCPTool, a as addCustomHeaderPlugin, d as buildSchema, s as convertMCPToolsToActions, m as copilotKitEndpoint, c as createContext, h as createLogger, q as extractParametersFromSchema, l as flattenToolCallsNoDuplicates, t as generateMcpToolInstructions, g as getCommonConfig, n as langGraphPlatformEndpoint, r as resolveEndpointType } from '../shared-ba062831.js';
 export { CopilotRuntimeServerInstance, config, copilotRuntimeNextJSAppRouterEndpoint, copilotRuntimeNextJSPagesRouterEndpoint } from './integrations/index.js';
 export { copilotRuntimeNodeHttpEndpoint } from './integrations/node-http/index.js';
 export { copilotRuntimeNodeExpressEndpoint } from './integrations/node-express/index.js';

package/dist/lib/index.js

@@ -44,7 +44,7 @@ var require_package = __commonJS({
       publishConfig: {
         access: "public"
       },
-      version: "1.9.2-next.3",
+      version: "1.9.2-next.4",
       sideEffects: false,
       main: "./dist/index.js",
       module: "./dist/index.mjs",
@@ -4682,6 +4682,8 @@ var CopilotRuntime = class {
   delegateAgentProcessingToServiceAdapter;
   observability;
   availableAgents;
+  onTrace;
+  hasWarnedAboutTracing = false;
   // +++ MCP Properties +++
   mcpServersConfig;
   mcpActionCache = /* @__PURE__ */ new Map();
@@ -4706,6 +4708,7 @@ var CopilotRuntime = class {
     this.delegateAgentProcessingToServiceAdapter = (params == null ? void 0 : params.delegateAgentProcessingToServiceAdapter) || false;
     this.observability = params == null ? void 0 : params.observability_c;
     this.agents = (params == null ? void 0 : params.agents) ?? {};
+    this.onTrace = params == null ? void 0 : params.onTrace;
     this.mcpServersConfig = params == null ? void 0 : params.mcpServers;
     this.createMCPClientImpl = params == null ? void 0 : params.createMCPClient;
     if (this.mcpServersConfig && this.mcpServersConfig.length > 0 && !this.createMCPClientImpl) {
@@ -4784,6 +4787,27 @@ var CopilotRuntime = class {
     const eventSource = new RuntimeEventSource();
     const requestStartTime = Date.now();
     const streamedChunks = [];
+    await this.trace("request", {
+      threadId,
+      runId,
+      source: "runtime",
+      request: {
+        operation: "processRuntimeRequest",
+        method: "POST",
+        url,
+        startTime: requestStartTime
+      },
+      agent: agentSession ? {
+        name: agentSession.agentName
+      } : void 0,
+      messages: {
+        input: rawMessages,
+        messageCount: rawMessages.length
+      },
+      technical: {
+        environment: process.env.NODE_ENV
+      }
+    }, void 0, publicApiKey);
     try {
       if (Object.keys(this.agents).length && (agentSession == null ? void 0 : agentSession.agentName) && !this.delegateAgentProcessingToServiceAdapter) {
         this.agents = {
@@ -4953,11 +4977,35 @@ please use an LLM adapter instead.`
           console.error("Error logging LLM error:", logError);
         }
       }
+      let structuredError;
       if (error instanceof import_shared19.CopilotKitError) {
-        throw error;
+        structuredError = error;
+      } else {
+        console.error("Error getting response:", error);
+        structuredError = this.convertStreamingErrorToStructured(error);
       }
-      console.error("Error getting response:", error);
-      const structuredError = this.convertStreamingErrorToStructured(error);
+      await this.trace("error", {
+        threadId,
+        runId,
+        source: "runtime",
+        request: {
+          operation: "processRuntimeRequest",
+          method: "POST",
+          url,
+          startTime: requestStartTime
+        },
+        response: {
+          endTime: Date.now(),
+          latency: Date.now() - requestStartTime
+        },
+        agent: agentSession ? {
+          name: agentSession.agentName
+        } : void 0,
+        technical: {
+          environment: process.env.NODE_ENV,
+          stackTrace: error instanceof Error ? error.stack : void 0
+        }
+      }, structuredError, publicApiKey);
       throw structuredError;
     }
   }
@@ -5142,6 +5190,26 @@ please use an LLM adapter instead.`
     const requestStartTime = Date.now();
     const streamedChunks = [];
     const threadId = threadIdFromRequest ?? agentSession.threadId;
+    await this.trace("agent_state", {
+      threadId,
+      source: "agent",
+      request: {
+        operation: "processAgentRequest",
+        method: "POST",
+        startTime: requestStartTime
+      },
+      agent: {
+        name: agentName,
+        nodeName
+      },
+      messages: {
+        input: rawMessages,
+        messageCount: rawMessages.length
+      },
+      technical: {
+        environment: process.env.NODE_ENV
+      }
+    }, void 0, publicApiKey);
     const serverSideActions = await this.getServerSideActions(request);
     const messages = convertGqlInputToMessages(rawMessages);
     const currentAgent = serverSideActions.find((action) => action.name === agentName && isRemoteAgentAction(action));
@@ -5237,7 +5305,7 @@ please use an LLM adapter instead.`
       eventSource.stream(async (eventStream$) => {
         (0, import_rxjs2.from)(stream).subscribe({
           next: (event) => eventStream$.next(event),
-          error: (err) => {
+          error: async (err) => {
             var _a2;
             console.error("Error in stream", err);
             if (((_a2 = this.observability) == null ? void 0 : _a2.enabled) && publicApiKey) {
@@ -5259,6 +5327,27 @@ please use an LLM adapter instead.`
               }
             }
             const structuredError = this.convertStreamingErrorToStructured(err);
+            await this.trace("error", {
+              threadId,
+              source: "agent",
+              request: {
+                operation: "processAgentRequest",
+                method: "POST",
+                startTime: requestStartTime
+              },
+              response: {
+                endTime: Date.now(),
+                latency: Date.now() - requestStartTime
+              },
+              agent: {
+                name: agentName,
+                nodeName
+              },
+              technical: {
+                environment: process.env.NODE_ENV,
+                stackTrace: err instanceof Error ? err.stack : void 0
+              }
+            }, structuredError, publicApiKey);
             eventStream$.error(structuredError);
             eventStream$.complete();
           },
@@ -5326,8 +5415,35 @@ please use an LLM adapter instead.`
           console.error("Error logging agent error:", logError);
         }
       }
+      let structuredError;
+      if (error instanceof import_shared19.CopilotKitError) {
+        structuredError = error;
+      } else {
+        structuredError = this.convertStreamingErrorToStructured(error);
+      }
+      await this.trace("error", {
+        threadId,
+        source: "agent",
+        request: {
+          operation: "processAgentRequest",
+          method: "POST",
+          startTime: requestStartTime
+        },
+        response: {
+          endTime: Date.now(),
+          latency: Date.now() - requestStartTime
+        },
+        agent: {
+          name: agentName,
+          nodeName
+        },
+        technical: {
+          environment: process.env.NODE_ENV,
+          stackTrace: error instanceof Error ? error.stack : void 0
+        }
+      }, structuredError, publicApiKey);
       console.error("Error getting response:", error);
-      throw error;
+      throw structuredError;
     }
   }
   async getServerSideActions(request) {
@@ -5447,6 +5563,62 @@ please use an LLM adapter instead.`
       code: import_shared19.CopilotKitErrorCode.UNKNOWN
     });
   }
+  async trace(type, context, error, publicApiKey) {
+    if (!this.onTrace)
+      return;
+    if (!publicApiKey) {
+      if (!this.hasWarnedAboutTracing) {
+        console.warn("CopilotKit: onTrace handler provided but requires publicApiKey for tracing to work. This is a CopilotKit Cloud feature. See: https://docs.copilotkit.ai/cloud");
+        this.hasWarnedAboutTracing = true;
+      }
+      return;
+    }
+    try {
+      const traceEvent = {
+        type,
+        timestamp: Date.now(),
+        context,
+        ...error && {
+          error
+        }
+      };
+      await this.onTrace(traceEvent);
+    } catch (traceError) {
+      console.error("Error in onTrace handler:", traceError);
+    }
+  }
+  /**
+  * Public method to trace GraphQL validation errors
+  * This allows the GraphQL resolver to send validation errors through the trace system
+  */
+  async traceGraphQLError(error, context) {
+    if (!this.onTrace)
+      return;
+    try {
+      await this.onTrace({
+        type: "error",
+        timestamp: Date.now(),
+        context: {
+          source: "runtime",
+          request: {
+            operation: context.operation,
+            startTime: Date.now()
+          },
+          technical: {
+            environment: process.env.NODE_ENV
+          },
+          metadata: {
+            errorType: "GraphQLValidationError",
+            cloudConfigPresent: context.cloudConfigPresent,
+            guardrailsEnabled: context.guardrailsEnabled
+          }
+        },
+        error
+      });
+    } catch (traceError) {
+      console.error("Error in onTrace handler:", traceError);
+    }
+  }
 };
 __name(CopilotRuntime, "CopilotRuntime");
 function flattenToolCallsNoDuplicates(toolsByPriority) {
@@ -6037,7 +6209,7 @@ var CopilotResolver = class {
     };
   }
   async generateCopilotResponse(ctx, data, properties) {
-    var _a, _b, _c, _d;
+    var _a, _b, _c, _d, _e;
     telemetry_client_default.capture("oss.runtime.copilot_request_created", {
       "cloud.guardrails.enabled": ((_a = data.cloud) == null ? void 0 : _a.guardrails) !== void 0,
       requestType: data.metadata.requestType,
@@ -6068,23 +6240,32 @@ var CopilotResolver = class {
     const serviceAdapter = ctx._copilotkit.serviceAdapter;
     let copilotCloudPublicApiKey = null;
     let copilotCloudBaseUrl;
+    const publicApiKeyFromHeaders = ctx.request.headers.get("x-copilotcloud-public-api-key");
+    if (publicApiKeyFromHeaders) {
+      copilotCloudPublicApiKey = publicApiKeyFromHeaders;
+    }
     if (data.cloud) {
       logger2 = logger2.child({
         cloud: true
       });
       logger2.debug("Cloud configuration provided, checking for public API key in headers");
-      const key = ctx.request.headers.get("x-copilotcloud-public-api-key");
-      if (key) {
-        logger2.debug("Public API key found in headers");
-        copilotCloudPublicApiKey = key;
-      } else {
+      if (!copilotCloudPublicApiKey) {
         logger2.error("Public API key not found in headers");
+        await copilotRuntime.traceGraphQLError({
+          message: "X-CopilotCloud-Public-API-Key header is required",
+          code: "MISSING_PUBLIC_API_KEY",
+          type: "GraphQLError"
+        }, {
+          operation: "generateCopilotResponse",
+          cloudConfigPresent: Boolean(data.cloud),
+          guardrailsEnabled: Boolean((_b = data.cloud) == null ? void 0 : _b.guardrails)
+        });
         throw new import_graphql.GraphQLError("X-CopilotCloud-Public-API-Key header is required");
       }
       if (process.env.COPILOT_CLOUD_BASE_URL) {
         copilotCloudBaseUrl = process.env.COPILOT_CLOUD_BASE_URL;
-      } else if ((_b = ctx._copilotkit.cloud) == null ? void 0 : _b.baseUrl) {
-        copilotCloudBaseUrl = (_c = ctx._copilotkit.cloud) == null ? void 0 : _c.baseUrl;
+      } else if ((_c = ctx._copilotkit.cloud) == null ? void 0 : _c.baseUrl) {
+        copilotCloudBaseUrl = (_d = ctx._copilotkit.cloud) == null ? void 0 : _d.baseUrl;
       } else {
         copilotCloudBaseUrl = "https://api.cloud.copilotkit.ai";
       }
@@ -6126,7 +6307,7 @@ var CopilotResolver = class {
     logger2.debug("Event source created, creating response");
     const eventStream = eventSource.processRuntimeEvents({
       serverSideActions,
-      guardrailsResult$: ((_d = data.cloud) == null ? void 0 : _d.guardrails) ? guardrailsResult$ : null,
+      guardrailsResult$: ((_e = data.cloud) == null ? void 0 : _e.guardrails) ? guardrailsResult$ : null,
       actionInputsWithoutAgents: actionInputsWithoutAgents.filter(
         // TODO-AGENTS: do not exclude ALL server side actions
         (action) => !serverSideActions.find((serverSideAction) => serverSideAction.name == action.name)