autotel-cloudflare 2.1.0

package/dist/agents.js

@@ -0,0 +1,276 @@
+import { trace, SpanStatusCode, SpanKind } from '@opentelemetry/api';
+import { resourceFromAttributes } from '@opentelemetry/resources';
+import { createInitialiser, WorkerTracerProvider, WorkerTracer } from 'autotel-edge';
+
+// src/agents/otel-observability.ts
+var activeSpans = /* @__PURE__ */ new Map();
+var providerInitialized = false;
+function initProvider(config) {
+  if (providerInitialized) return;
+  const resource = resourceFromAttributes({
+    "service.name": config.service.name,
+    "service.version": config.service.version,
+    "service.namespace": config.service.namespace,
+    "cloud.provider": "cloudflare",
+    "cloud.platform": "cloudflare.workers",
+    "telemetry.sdk.name": "autotel-cloudflare",
+    "telemetry.sdk.language": "js",
+    "agent.framework": "cloudflare-agents"
+  });
+  const provider = new WorkerTracerProvider(config.spanProcessors, resource);
+  provider.register();
+  const tracer = trace.getTracer("autotel-cloudflare/agents");
+  tracer.setHeadSampler(config.sampling.headSampler);
+  providerInitialized = true;
+}
+function getDefaultSpanName(event) {
+  switch (event.type) {
+    case "rpc": {
+      return `agent.rpc ${event.payload.method}`;
+    }
+    case "schedule:create": {
+      return `agent.schedule.create ${event.payload.callback}`;
+    }
+    case "schedule:execute": {
+      return `agent.schedule.execute ${event.payload.callback}`;
+    }
+    case "schedule:cancel": {
+      return `agent.schedule.cancel ${event.payload.callback}`;
+    }
+    case "connect": {
+      return `agent.connect`;
+    }
+    case "destroy": {
+      return `agent.destroy`;
+    }
+    case "state:update": {
+      return `agent.state.update`;
+    }
+    case "message:request": {
+      return `agent.message.request`;
+    }
+    case "message:response": {
+      return `agent.message.response`;
+    }
+    case "message:clear": {
+      return `agent.message.clear`;
+    }
+    case "mcp:client:preconnect": {
+      return `mcp.preconnect ${event.payload.serverId}`;
+    }
+    case "mcp:client:connect": {
+      return `mcp.connect ${event.payload.url}`;
+    }
+    case "mcp:client:authorize": {
+      return `mcp.authorize ${event.payload.serverId}`;
+    }
+    case "mcp:client:discover": {
+      return `mcp.discover`;
+    }
+    default: {
+      return `agent.${event.type}`;
+    }
+  }
+}
+function getDefaultAttributes(event) {
+  const attrs = {
+    "agent.event.type": event.type,
+    "agent.event.id": event.id
+  };
+  switch (event.type) {
+    case "rpc": {
+      attrs["agent.rpc.method"] = event.payload.method;
+      if (event.payload.streaming !== void 0) {
+        attrs["agent.rpc.streaming"] = event.payload.streaming;
+      }
+      break;
+    }
+    case "schedule:create":
+    case "schedule:execute":
+    case "schedule:cancel": {
+      attrs["agent.schedule.callback"] = event.payload.callback;
+      attrs["agent.schedule.id"] = event.payload.id;
+      break;
+    }
+    case "connect": {
+      attrs["agent.connection.id"] = event.payload.connectionId;
+      break;
+    }
+    case "mcp:client:preconnect": {
+      attrs["agent.mcp.server_id"] = event.payload.serverId;
+      break;
+    }
+    case "mcp:client:connect": {
+      attrs["agent.mcp.url"] = event.payload.url;
+      attrs["agent.mcp.transport"] = event.payload.transport;
+      attrs["agent.mcp.state"] = event.payload.state;
+      if (event.payload.error) {
+        attrs["agent.mcp.error"] = event.payload.error;
+      }
+      break;
+    }
+    case "mcp:client:authorize": {
+      attrs["agent.mcp.server_id"] = event.payload.serverId;
+      attrs["agent.mcp.auth_url"] = event.payload.authUrl;
+      if (event.payload.clientId) {
+        attrs["agent.mcp.client_id"] = event.payload.clientId;
+      }
+      break;
+    }
+  }
+  for (const [key, value] of Object.entries(event.payload)) {
+    if (attrs[`agent.${key}`] === void 0 && (typeof value === "string" || typeof value === "number" || typeof value === "boolean")) {
+      attrs[`agent.payload.${key}`] = value;
+    }
+  }
+  return attrs;
+}
+function getSpanKind(event) {
+  switch (event.type) {
+    case "rpc": {
+      return SpanKind.SERVER;
+    }
+    case "connect": {
+      return SpanKind.SERVER;
+    }
+    case "mcp:client:connect":
+    case "mcp:client:preconnect":
+    case "mcp:client:authorize":
+    case "mcp:client:discover": {
+      return SpanKind.CLIENT;
+    }
+    default: {
+      return SpanKind.INTERNAL;
+    }
+  }
+}
+function shouldTraceEvent(event, options) {
+  const defaults = {
+    traceRpc: true,
+    traceSchedule: true,
+    traceMcp: true,
+    traceStateUpdates: false,
+    traceMessages: true,
+    traceLifecycle: true
+  };
+  const opts = { ...defaults, ...options };
+  switch (event.type) {
+    case "rpc": {
+      return opts.traceRpc ?? true;
+    }
+    case "schedule:create":
+    case "schedule:execute":
+    case "schedule:cancel": {
+      return opts.traceSchedule ?? true;
+    }
+    case "mcp:client:preconnect":
+    case "mcp:client:connect":
+    case "mcp:client:authorize":
+    case "mcp:client:discover": {
+      return opts.traceMcp ?? true;
+    }
+    case "state:update": {
+      return opts.traceStateUpdates ?? false;
+    }
+    case "message:request":
+    case "message:response":
+    case "message:clear": {
+      return opts.traceMessages ?? true;
+    }
+    case "connect":
+    case "destroy": {
+      return opts.traceLifecycle ?? true;
+    }
+    default: {
+      return true;
+    }
+  }
+}
+async function exportSpans(traceId, ctx) {
+  const tracer = trace.getTracer("autotel-cloudflare/agents");
+  if (tracer instanceof WorkerTracer) {
+    try {
+      await scheduler.wait(1);
+      await tracer.forceFlush(traceId);
+    } catch (error) {
+      console.error("[autotel-cloudflare/agents] Failed to export spans:", error);
+    }
+  }
+}
+var OtelObservability = class {
+  config;
+  options;
+  initialized = false;
+  constructor(config) {
+    const initialiser = createInitialiser(config);
+    this.config = initialiser({}, void 0);
+    this.options = config.agents ?? {};
+  }
+  /**
+   * Initialize the tracer provider (called lazily on first emit)
+   */
+  initialize() {
+    if (this.initialized) return;
+    initProvider(this.config);
+    this.initialized = true;
+  }
+  /**
+   * Emit an observability event
+   *
+   * Converts the event to an OpenTelemetry span based on the event type.
+   */
+  emit(event, ctx) {
+    this.initialize();
+    if (!shouldTraceEvent(event, this.options)) {
+      return;
+    }
+    const tracer = trace.getTracer("autotel-cloudflare/agents");
+    const spanName = this.options.spanNameFormatter ? this.options.spanNameFormatter(event) : getDefaultSpanName(event);
+    const defaultAttrs = getDefaultAttributes(event);
+    const customAttrs = this.options.attributeExtractor ? this.options.attributeExtractor(event) : {};
+    const attributes = { ...defaultAttrs, ...customAttrs };
+    const kind = getSpanKind(event);
+    const span = tracer.startSpan(spanName, {
+      kind,
+      attributes,
+      startTime: event.timestamp
+    });
+    span.setStatus({ code: SpanStatusCode.OK });
+    span.end(event.timestamp + 1);
+    activeSpans.set(event.id, span);
+    const traceId = span.spanContext().traceId;
+    if (ctx && "waitUntil" in ctx && typeof ctx.waitUntil === "function") {
+      ctx.waitUntil(exportSpans(traceId));
+    } else {
+      void exportSpans(traceId);
+    }
+  }
+};
+function createOtelObservability(config) {
+  return new OtelObservability(config);
+}
+function createOtelObservabilityFromEnv(env, options) {
+  const endpoint = env.OTEL_EXPORTER_OTLP_ENDPOINT || void 0;
+  const serviceName = env.OTEL_SERVICE_NAME || "cloudflare-agent";
+  let headers;
+  const headersStr = env.OTEL_EXPORTER_OTLP_HEADERS;
+  if (headersStr) {
+    headers = {};
+    for (const pair of headersStr.split(",")) {
+      const [key, value] = pair.split("=");
+      if (key && value) {
+        headers[key.trim()] = value.trim();
+      }
+    }
+  }
+  const exporterUrl = endpoint || "http://localhost:4318/v1/traces";
+  return createOtelObservability({
+    service: { name: serviceName },
+    exporter: { url: exporterUrl, headers },
+    agents: options
+  });
+}
+
+export { OtelObservability, createOtelObservability, createOtelObservabilityFromEnv };
+//# sourceMappingURL=agents.js.map
+//# sourceMappingURL=agents.js.map

package/dist/agents.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/agents/otel-observability.ts"],"names":[],"mappings":";;;;;AAkDA,IAAM,WAAA,uBAAkB,GAAA,EAAkB;AAK1C,IAAI,mBAAA,GAAsB,KAAA;AAK1B,SAAS,aAAa,MAAA,EAAkC;AACtD,EAAA,IAAI,mBAAA,EAAqB;AAGzB,EAAA,MAAM,WAAW,sBAAA,CAAuB;AAAA,IACtC,cAAA,EAAgB,OAAO,OAAA,CAAQ,IAAA;AAAA,IAC/B,iBAAA,EAAmB,OAAO,OAAA,CAAQ,OAAA;AAAA,IAClC,mBAAA,EAAqB,OAAO,OAAA,CAAQ,SAAA;AAAA,IACpC,gBAAA,EAAkB,YAAA;AAAA,IAClB,gBAAA,EAAkB,oBAAA;AAAA,IAClB,oBAAA,EAAsB,oBAAA;AAAA,IACtB,wBAAA,EAA0B,IAAA;AAAA,IAC1B,iBAAA,EAAmB;AAAA,GACpB,CAAA;AAGD,EAAA,MAAM,QAAA,GAAW,IAAI,oBAAA,CAAqB,MAAA,CAAO,gBAAgB,QAAQ,CAAA;AACzE,EAAA,QAAA,CAAS,QAAA,EAAS;AAGlB,EAAA,MAAM,MAAA,GAAS,KAAA,CAAM,SAAA,CAAU,2BAA2B,CAAA;AAC1D,EAAA,MAAA,CAAO,cAAA,CAAe,MAAA,CAAO,QAAA,CAAS,WAAW,CAAA;AAEjD,EAAA,mBAAA,GAAsB,IAAA;AACxB;AAKA,SAAS,mBAAmB,KAAA,EAAmC;AAC7D,EAAA,QAAQ,MAAM,IAAA;AAAM,IAClB,KAAK,KAAA,EAAO;AACV,MAAA,OAAO,CAAA,UAAA,EAAa,KAAA,CAAM,OAAA,CAAQ,MAAM,CAAA,CAAA;AAAA,IAC1C;AAAA,IACA,KAAK,iBAAA,EAAmB;AACtB,MAAA,OAAO,CAAA,sBAAA,EAAyB,KAAA,CAAM,OAAA,CAAQ,QAAQ,CAAA,CAAA;AAAA,IACxD;AAAA,IACA,KAAK,kBAAA,EAAoB;AACvB,MAAA,OAAO,CAAA,uBAAA,EAA0B,KAAA,CAAM,OAAA,CAAQ,QAAQ,CAAA,CAAA;AAAA,IACzD;AAAA,IACA,KAAK,iBAAA,EAAmB;AACtB,MAAA,OAAO,CAAA,sBAAA,EAAyB,KAAA,CAAM,OAAA,CAAQ,QAAQ,CAAA,CAAA;AAAA,IACxD;AAAA,IACA,KAAK,SAAA,EAAW;AACd,MAAA,OAAO,CAAA,aAAA,CAAA;AAAA,IACT;AAAA,IACA,KAAK,SAAA,EAAW;AACd,MAAA,OAAO,CAAA,aAAA,CAAA;AAAA,IACT;AAAA,IACA,KAAK,cAAA,EAAgB;AACnB,MAAA,OAAO,CAAA,kBAAA,CAAA;AAAA,IACT;AAAA,IACA,KAAK,iBAAA,EAAmB;AACtB,MAAA,OAAO,CAAA,qBAAA,CAAA;AAAA,IACT;AAAA,IACA,KAAK,kBAAA,EAAoB;AACvB,MAAA,OAAO,CAAA,sBAAA,CAAA;AAAA,IACT;AAAA,IACA,KAAK,eAAA,EAAiB;AACpB,MAAA,OAAO,CAAA,mBAAA,CAAA;AAAA,IACT;AAAA,IACA,KAAK,uBAAA,EAAyB;AAC5B,MAAA,OAAO,CAAA,eAAA,EAAkB,KAAA,CAAM,OAAA,CAAQ,QAAQ,CAAA,CAAA;AAAA,IACjD;AAAA,IACA,KAAK,oBAAA,EAAsB;AACzB,MAAA,OAAO,CAAA,YAAA,EAAe,KAAA,CAAM,OAAA,CAAQ,GAAG,CAAA,CAAA;AAAA,IACzC;AAAA,IACA,KAAK,sBAAA,EAAwB;AAC3B,MAAA,OAAO,CAAA,cAAA,EAAiB,KAAA,CAAM,OAAA,CAAQ,QAAQ,CAAA,CAAA;AAAA,IAChD;AAAA,IACA,KAAK,qBAAA,EAAuB;AAC1B,MAAA,OAAO,CAAA,YAAA,CAAA;AAAA,IACT;AAAA,IACA,SAAS;AACP,MAAA,OAAO,CAAA,MAAA,EAAU,MAA6B,IAAI,CAAA,CAAA;AAAA,IACpD;AAAA;AAEJ;AAKA,SAAS,qBAAqB,KAAA,EAAuC;AACnE,EAAA,MAAM,KAAA,GAAoB;AAAA,IACxB,oBAAoB,KAAA,CAAM,IAAA;AAAA,IAC1B,kBAAkB,KAAA,CAAM;AAAA,GAC1B;AAGA,EAAA,QAAQ,MAAM,IAAA;AAAM,IAClB,KAAK,KAAA,EAAO;AACV,MAAA,KAAA,CAAM,kBAAkB,CAAA,GAAI,KAAA,CAAM,OAAA,CAAQ,MAAA;AAC1C,MAAA,IAAI,KAAA,CAAM,OAAA,CAAQ,SAAA,KAAc,MAAA,EAAW;AACzC,QAAA,KAAA,CAAM,qBAAqB,CAAA,GAAI,KAAA,CAAM,OAAA,CAAQ,SAAA;AAAA,MAC/C;AACA,MAAA;AAAA,IACF;AAAA,IAEA,KAAK,iBAAA;AAAA,IACL,KAAK,kBAAA;AAAA,IACL,KAAK,iBAAA,EAAmB;AACtB,MAAA,KAAA,CAAM,yBAAyB,CAAA,GAAI,KAAA,CAAM,OAAA,CAAQ,QAAA;AACjD,MAAA,KAAA,CAAM,mBAAmB,CAAA,GAAI,KAAA,CAAM,OAAA,CAAQ,EAAA;AAC3C,MAAA;AAAA,IACF;AAAA,IAEA,KAAK,SAAA,EAAW;AACd,MAAA,KAAA,CAAM,qBAAqB,CAAA,GAAI,KAAA,CAAM,OAAA,CAAQ,YAAA;AAC7C,MAAA;AAAA,IACF;AAAA,IAEA,KAAK,uBAAA,EAAyB;AAC5B,MAAA,KAAA,CAAM,qBAAqB,CAAA,GAAI,KAAA,CAAM,OAAA,CAAQ,QAAA;AAC7C,MAAA;AAAA,IACF;AAAA,IAEA,KAAK,oBAAA,EAAsB;AACzB,MAAA,KAAA,CAAM,eAAe,CAAA,GAAI,KAAA,CAAM,OAAA,CAAQ,GAAA;AACvC,MAAA,KAAA,CAAM,qBAAqB,CAAA,GAAI,KAAA,CAAM,OAAA,CAAQ,SAAA;AAC7C,MAAA,KAAA,CAAM,iBAAiB,CAAA,GAAI,KAAA,CAAM,OAAA,CAAQ,KAAA;AACzC,MAAA,IAAI,KAAA,CAAM,QAAQ,KAAA,EAAO;AACvB,QAAA,KAAA,CAAM,iBAAiB,CAAA,GAAI,KAAA,CAAM,OAAA,CAAQ,KAAA;AAAA,MAC3C;AACA,MAAA;AAAA,IACF;AAAA,IAEA,KAAK,sBAAA,EAAwB;AAC3B,MAAA,KAAA,CAAM,qBAAqB,CAAA,GAAI,KAAA,CAAM,OAAA,CAAQ,QAAA;AAC7C,MAAA,KAAA,CAAM,oBAAoB,CAAA,GAAI,KAAA,CAAM,OAAA,CAAQ,OAAA;AAC5C,MAAA,IAAI,KAAA,CAAM,QAAQ,QAAA,EAAU;AAC1B,QAAA,KAAA,CAAM,qBAAqB,CAAA,GAAI,KAAA,CAAM,OAAA,CAAQ,QAAA;AAAA,MAC/C;AACA,MAAA;AAAA,IACF;AAAA;AAIF,EAAA,KAAA,MAAW,CAAC,KAAK,KAAK,CAAA,IAAK,OAAO,OAAA,CAAQ,KAAA,CAAM,OAAO,CAAA,EAAG;AACxD,IAAA,IACE,KAAA,CAAM,CAAA,MAAA,EAAS,GAAG,CAAA,CAAE,MAAM,MAAA,KACzB,OAAO,KAAA,KAAU,QAAA,IAAY,OAAO,KAAA,KAAU,QAAA,IAAY,OAAO,UAAU,SAAA,CAAA,EAC5E;AACA,MAAA,KAAA,CAAM,CAAA,cAAA,EAAiB,GAAG,CAAA,CAAE,CAAA,GAAI,KAAA;AAAA,IAClC;AAAA,EACF;AAEA,EAAA,OAAO,KAAA;AACT;AAKA,SAAS,YAAY,KAAA,EAAqC;AACxD,EAAA,QAAQ,MAAM,IAAA;AAAM,IAClB,KAAK,KAAA,EAAO;AACV,MAAA,OAAO,QAAA,CAAS,MAAA;AAAA,IAClB;AAAA,IACA,KAAK,SAAA,EAAW;AACd,MAAA,OAAO,QAAA,CAAS,MAAA;AAAA,IAClB;AAAA,IACA,KAAK,oBAAA;AAAA,IACL,KAAK,uBAAA;AAAA,IACL,KAAK,sBAAA;AAAA,IACL,KAAK,qBAAA,EAAuB;AAC1B,MAAA,OAAO,QAAA,CAAS,MAAA;AAAA,IAClB;AAAA,IACA,SAAS;AACP,MAAA,OAAO,QAAA,CAAS,QAAA;AAAA,IAClB;AAAA;AAEJ;AAKA,SAAS,gBAAA,CACP,OACA,OAAA,EACS;AACT,EAAA,MAAM,QAAA,GAAwC;AAAA,IAC5C,QAAA,EAAU,IAAA;AAAA,IACV,aAAA,EAAe,IAAA;AAAA,IACf,QAAA,EAAU,IAAA;AAAA,IACV,iBAAA,EAAmB,KAAA;AAAA,IACnB,aAAA,EAAe,IAAA;AAAA,IACf,cAAA,EAAgB;AAAA,GAClB;AAEA,EAAA,MAAM,IAAA,GAAO,EAAE,GAAG,QAAA,EAAU,GAAG,OAAA,EAAQ;AAEvC,EAAA,QAAQ,MAAM,IAAA;AAAM,IAClB,KAAK,KAAA,EAAO;AACV,MAAA,OAAO,KAAK,QAAA,IAAY,IAAA;AAAA,IAC1B;AAAA,IAEA,KAAK,iBAAA;AAAA,IACL,KAAK,kBAAA;AAAA,IACL,KAAK,iBAAA,EAAmB;AACtB,MAAA,OAAO,KAAK,aAAA,IAAiB,IAAA;AAAA,IAC/B;AAAA,IAEA,KAAK,uBAAA;AAAA,IACL,KAAK,oBAAA;AAAA,IACL,KAAK,sBAAA;AAAA,IACL,KAAK,qBAAA,EAAuB;AAC1B,MAAA,OAAO,KAAK,QAAA,IAAY,IAAA;AAAA,IAC1B;AAAA,IAEA,KAAK,cAAA,EAAgB;AACnB,MAAA,OAAO,KAAK,iBAAA,IAAqB,KAAA;AAAA,IACnC;AAAA,IAEA,KAAK,iBAAA;AAAA,IACL,KAAK,kBAAA;AAAA,IACL,KAAK,eAAA,EAAiB;AACpB,MAAA,OAAO,KAAK,aAAA,IAAiB,IAAA;AAAA,IAC/B;AAAA,IAEA,KAAK,SAAA;AAAA,IACL,KAAK,SAAA,EAAW;AACd,MAAA,OAAO,KAAK,cAAA,IAAkB,IAAA;AAAA,IAChC;AAAA,IAEA,SAAS;AACP,MAAA,OAAO,IAAA;AAAA,IACT;AAAA;AAEJ;AAKA,eAAe,WAAA,CAAY,SAAiB,GAAA,EAAyC;AACnF,EAAA,MAAM,MAAA,GAAS,KAAA,CAAM,SAAA,CAAU,2BAA2B,CAAA;AAC1D,EAAA,IAAI,kBAAkB,YAAA,EAAc;AAClC,IAAA,IAAI;AACF,MAAA,MAAM,SAAA,CAAU,KAAK,CAAC,CAAA;AACtB,MAAA,MAAM,MAAA,CAAO,WAAW,OAAO,CAAA;AAAA,IACjC,SAAS,KAAA,EAAO;AACd,MAAA,OAAA,CAAQ,KAAA,CAAM,uDAAuD,KAAK,CAAA;AAAA,IAC5E;AAAA,EACF;AAMF;AAQO,IAAM,oBAAN,MAAiD;AAAA,EAC9C,MAAA;AAAA,EACA,OAAA;AAAA,EACA,WAAA,GAAc,KAAA;AAAA,EAEtB,YAAY,MAAA,EAAiC;AAE3C,IAAA,MAAM,WAAA,GAAc,kBAAkB,MAAM,CAAA;AAC5C,IAAA,IAAA,CAAK,MAAA,GAAS,WAAA,CAAY,EAAC,EAAG,MAAS,CAAA;AACvC,IAAA,IAAA,CAAK,OAAA,GAAU,MAAA,CAAO,MAAA,IAAU,EAAC;AAAA,EACnC;AAAA;AAAA;AAAA;AAAA,EAKQ,UAAA,GAAmB;AACzB,IAAA,IAAI,KAAK,WAAA,EAAa;AACtB,IAAA,YAAA,CAAa,KAAK,MAAM,CAAA;AACxB,IAAA,IAAA,CAAK,WAAA,GAAc,IAAA;AAAA,EACrB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,IAAA,CAAK,OAA2B,GAAA,EAAgC;AAE9D,IAAA,IAAA,CAAK,UAAA,EAAW;AAGhB,IAAA,IAAI,CAAC,gBAAA,CAAiB,KAAA,EAAO,IAAA,CAAK,OAAO,CAAA,EAAG;AAC1C,MAAA;AAAA,IACF;AAEA,IAAA,MAAM,MAAA,GAAS,KAAA,CAAM,SAAA,CAAU,2BAA2B,CAAA;AAG1D,IAAA,MAAM,QAAA,GAAW,IAAA,CAAK,OAAA,CAAQ,iBAAA,GAC1B,IAAA,CAAK,QAAQ,iBAAA,CAAkB,KAAK,CAAA,GACpC,kBAAA,CAAmB,KAAK,CAAA;AAG5B,IAAA,MAAM,YAAA,GAAe,qBAAqB,KAAK,CAAA;AAC/C,IAAA,MAAM,WAAA,GAAc,KAAK,OAAA,CAAQ,kBAAA,GAC7B,KAAK,OAAA,CAAQ,kBAAA,CAAmB,KAAK,CAAA,GACrC,EAAC;AACL,IAAA,MAAM,UAAA,GAAa,EAAE,GAAG,YAAA,EAAc,GAAG,WAAA,EAAY;AAGrD,IAAA,MAAM,IAAA,GAAO,YAAY,KAAK,CAAA;AAG9B,IAAA,MAAM,IAAA,GAAO,MAAA,CAAO,SAAA,CAAU,QAAA,EAAU;AAAA,MACtC,IAAA;AAAA,MACA,UAAA;AAAA,MACA,WAAW,KAAA,CAAM;AAAA,KAClB,CAAA;AAKD,IAAA,IAAA,CAAK,SAAA,CAAU,EAAE,IAAA,EAAM,cAAA,CAAe,IAAI,CAAA;AAC1C,IAAA,IAAA,CAAK,GAAA,CAAI,KAAA,CAAM,SAAA,GAAY,CAAC,CAAA;AAG5B,IAAA,WAAA,CAAY,GAAA,CAAI,KAAA,CAAM,EAAA,EAAI,IAAI,CAAA;AAG9B,IAAA,MAAM,OAAA,GAAU,IAAA,CAAK,WAAA,EAAY,CAAE,OAAA;AACnC,IAAA,IAAI,OAAO,WAAA,IAAe,GAAA,IAAO,OAAQ,GAAA,CAAY,cAAc,UAAA,EAAY;AAC7E,MAAC,GAAA,CAAY,SAAA,CAAU,WAAA,CAAY,OAAY,CAAC,CAAA;AAAA,IAClD,CAAA,MAAO;AAEL,MAAA,KAAK,WAAA,CAAY,OAAY,CAAA;AAAA,IAC/B;AAAA,EACF;AACF;AAkBO,SAAS,wBAAwB,MAAA,EAAoD;AAC1F,EAAA,OAAO,IAAI,kBAAkB,MAAM,CAAA;AACrC;AAsBO,SAAS,8BAAA,CACd,KACA,OAAA,EACmB;AAEnB,EAAA,MAAM,QAAA,GAAY,IAAI,2BAAA,IAA0C,MAAA;AAChE,EAAA,MAAM,WAAA,GAAe,IAAI,iBAAA,IAAgC,kBAAA;AAGzD,EAAA,IAAI,OAAA;AACJ,EAAA,MAAM,aAAa,GAAA,CAAI,0BAAA;AACvB,EAAA,IAAI,UAAA,EAAY;AACd,IAAA,OAAA,GAAU,EAAC;AACX,IAAA,KAAA,MAAW,IAAA,IAAQ,UAAA,CAAW,KAAA,CAAM,GAAG,CAAA,EAAG;AACxC,MAAA,MAAM,CAAC,GAAA,EAAK,KAAK,CAAA,GAAI,IAAA,CAAK,MAAM,GAAG,CAAA;AACnC,MAAA,IAAI,OAAO,KAAA,EAAO;AAChB,QAAA,OAAA,CAAQ,GAAA,CAAI,IAAA,EAAM,CAAA,GAAI,MAAM,IAAA,EAAK;AAAA,MACnC;AAAA,IACF;AAAA,EACF;AAIA,EAAA,MAAM,cAAc,QAAA,IAAY,iCAAA;AAEhC,EAAA,OAAO,uBAAA,CAAwB;AAAA,IAC7B,OAAA,EAAS,EAAE,IAAA,EAAM,WAAA,EAAY;AAAA,IAC7B,QAAA,EAAU,EAAE,GAAA,EAAK,WAAA,EAAa,OAAA,EAAQ;AAAA,IACtC,MAAA,EAAQ;AAAA,GACT,CAAA;AACH","file":"agents.js","sourcesContent":["/**\n * OpenTelemetry-based Observability implementation for Cloudflare Agents SDK\n *\n * Converts Agent events into OpenTelemetry spans for distributed tracing.\n *\n * @example\n * ```typescript\n * import { Agent } from 'agents'\n * import { createOtelObservability } from 'autotel-cloudflare/agents'\n *\n * class MyAgent extends Agent<Env> {\n *   observability = createOtelObservability({\n *     service: { name: 'my-agent' },\n *     exporter: { url: env.OTLP_ENDPOINT }\n *   })\n *\n *   @callable()\n *   async doSomething() {\n *     // This RPC call will be automatically traced\n *     return 'done'\n *   }\n * }\n * ```\n */\n\nimport {\n  trace,\n  SpanStatusCode,\n  SpanKind,\n  type Span,\n  type Attributes,\n} from '@opentelemetry/api';\nimport { resourceFromAttributes } from '@opentelemetry/resources';\nimport {\n  createInitialiser,\n  WorkerTracerProvider,\n  WorkerTracer,\n  type ResolvedEdgeConfig,\n} from 'autotel-edge';\nimport type {\n  Observability,\n  ObservabilityEvent,\n  OtelObservabilityConfig,\n  AgentInstrumentationOptions,\n} from './types';\n\n/**\n * Map of active spans keyed by event ID\n * Used to correlate start/end events\n */\nconst activeSpans = new Map<string, Span>();\n\n/**\n * Whether the provider has been initialized\n */\nlet providerInitialized = false;\n\n/**\n * Initialize the tracer provider for Agents\n */\nfunction initProvider(config: ResolvedEdgeConfig): void {\n  if (providerInitialized) return;\n\n  // Create resource with agent-specific attributes\n  const resource = resourceFromAttributes({\n    'service.name': config.service.name,\n    'service.version': config.service.version,\n    'service.namespace': config.service.namespace,\n    'cloud.provider': 'cloudflare',\n    'cloud.platform': 'cloudflare.workers',\n    'telemetry.sdk.name': 'autotel-cloudflare',\n    'telemetry.sdk.language': 'js',\n    'agent.framework': 'cloudflare-agents',\n  });\n\n  // Create and register provider\n  const provider = new WorkerTracerProvider(config.spanProcessors, resource);\n  provider.register();\n\n  // Set head sampler on tracer\n  const tracer = trace.getTracer('autotel-cloudflare/agents') as WorkerTracer;\n  tracer.setHeadSampler(config.sampling.headSampler);\n\n  providerInitialized = true;\n}\n\n/**\n * Get default span name for an event\n */\nfunction getDefaultSpanName(event: ObservabilityEvent): string {\n  switch (event.type) {\n    case 'rpc': {\n      return `agent.rpc ${event.payload.method}`;\n    }\n    case 'schedule:create': {\n      return `agent.schedule.create ${event.payload.callback}`;\n    }\n    case 'schedule:execute': {\n      return `agent.schedule.execute ${event.payload.callback}`;\n    }\n    case 'schedule:cancel': {\n      return `agent.schedule.cancel ${event.payload.callback}`;\n    }\n    case 'connect': {\n      return `agent.connect`;\n    }\n    case 'destroy': {\n      return `agent.destroy`;\n    }\n    case 'state:update': {\n      return `agent.state.update`;\n    }\n    case 'message:request': {\n      return `agent.message.request`;\n    }\n    case 'message:response': {\n      return `agent.message.response`;\n    }\n    case 'message:clear': {\n      return `agent.message.clear`;\n    }\n    case 'mcp:client:preconnect': {\n      return `mcp.preconnect ${event.payload.serverId}`;\n    }\n    case 'mcp:client:connect': {\n      return `mcp.connect ${event.payload.url}`;\n    }\n    case 'mcp:client:authorize': {\n      return `mcp.authorize ${event.payload.serverId}`;\n    }\n    case 'mcp:client:discover': {\n      return `mcp.discover`;\n    }\n    default: {\n      return `agent.${(event as ObservabilityEvent).type}`;\n    }\n  }\n}\n\n/**\n * Get default attributes for an event\n */\nfunction getDefaultAttributes(event: ObservabilityEvent): Attributes {\n  const attrs: Attributes = {\n    'agent.event.type': event.type,\n    'agent.event.id': event.id,\n  };\n\n  // Add type-specific attributes\n  switch (event.type) {\n    case 'rpc': {\n      attrs['agent.rpc.method'] = event.payload.method;\n      if (event.payload.streaming !== undefined) {\n        attrs['agent.rpc.streaming'] = event.payload.streaming;\n      }\n      break;\n    }\n\n    case 'schedule:create':\n    case 'schedule:execute':\n    case 'schedule:cancel': {\n      attrs['agent.schedule.callback'] = event.payload.callback;\n      attrs['agent.schedule.id'] = event.payload.id;\n      break;\n    }\n\n    case 'connect': {\n      attrs['agent.connection.id'] = event.payload.connectionId;\n      break;\n    }\n\n    case 'mcp:client:preconnect': {\n      attrs['agent.mcp.server_id'] = event.payload.serverId;\n      break;\n    }\n\n    case 'mcp:client:connect': {\n      attrs['agent.mcp.url'] = event.payload.url;\n      attrs['agent.mcp.transport'] = event.payload.transport;\n      attrs['agent.mcp.state'] = event.payload.state;\n      if (event.payload.error) {\n        attrs['agent.mcp.error'] = event.payload.error;\n      }\n      break;\n    }\n\n    case 'mcp:client:authorize': {\n      attrs['agent.mcp.server_id'] = event.payload.serverId;\n      attrs['agent.mcp.auth_url'] = event.payload.authUrl;\n      if (event.payload.clientId) {\n        attrs['agent.mcp.client_id'] = event.payload.clientId;\n      }\n      break;\n    }\n  }\n\n  // Add any additional payload properties as attributes\n  for (const [key, value] of Object.entries(event.payload)) {\n    if (\n      attrs[`agent.${key}`] === undefined &&\n      (typeof value === 'string' || typeof value === 'number' || typeof value === 'boolean')\n    ) {\n      attrs[`agent.payload.${key}`] = value;\n    }\n  }\n\n  return attrs;\n}\n\n/**\n * Determine span kind based on event type\n */\nfunction getSpanKind(event: ObservabilityEvent): SpanKind {\n  switch (event.type) {\n    case 'rpc': {\n      return SpanKind.SERVER;\n    }\n    case 'connect': {\n      return SpanKind.SERVER;\n    }\n    case 'mcp:client:connect':\n    case 'mcp:client:preconnect':\n    case 'mcp:client:authorize':\n    case 'mcp:client:discover': {\n      return SpanKind.CLIENT;\n    }\n    default: {\n      return SpanKind.INTERNAL;\n    }\n  }\n}\n\n/**\n * Check if an event type should be traced based on options\n */\nfunction shouldTraceEvent(\n  event: ObservabilityEvent,\n  options: AgentInstrumentationOptions,\n): boolean {\n  const defaults: AgentInstrumentationOptions = {\n    traceRpc: true,\n    traceSchedule: true,\n    traceMcp: true,\n    traceStateUpdates: false,\n    traceMessages: true,\n    traceLifecycle: true,\n  };\n\n  const opts = { ...defaults, ...options };\n\n  switch (event.type) {\n    case 'rpc': {\n      return opts.traceRpc ?? true;\n    }\n\n    case 'schedule:create':\n    case 'schedule:execute':\n    case 'schedule:cancel': {\n      return opts.traceSchedule ?? true;\n    }\n\n    case 'mcp:client:preconnect':\n    case 'mcp:client:connect':\n    case 'mcp:client:authorize':\n    case 'mcp:client:discover': {\n      return opts.traceMcp ?? true;\n    }\n\n    case 'state:update': {\n      return opts.traceStateUpdates ?? false;\n    }\n\n    case 'message:request':\n    case 'message:response':\n    case 'message:clear': {\n      return opts.traceMessages ?? true;\n    }\n\n    case 'connect':\n    case 'destroy': {\n      return opts.traceLifecycle ?? true;\n    }\n\n    default: {\n      return true;\n    }\n  }\n}\n\n/**\n * Export spans asynchronously\n */\nasync function exportSpans(traceId: string, ctx?: DurableObjectState): Promise<void> {\n  const tracer = trace.getTracer('autotel-cloudflare/agents');\n  if (tracer instanceof WorkerTracer) {\n    try {\n      await scheduler.wait(1);\n      await tracer.forceFlush(traceId);\n    } catch (error) {\n      console.error('[autotel-cloudflare/agents] Failed to export spans:', error);\n    }\n  }\n\n  // If we have a DurableObject context, use waitUntil for export\n  if (ctx && 'waitUntil' in ctx) {\n    // Already exported above, but could defer more work here\n  }\n}\n\n/**\n * OpenTelemetry-based Observability implementation\n *\n * Implements the Agents SDK Observability interface and converts\n * events into OpenTelemetry spans.\n */\nexport class OtelObservability implements Observability {\n  private config: ResolvedEdgeConfig;\n  private options: AgentInstrumentationOptions;\n  private initialized = false;\n\n  constructor(config: OtelObservabilityConfig) {\n    // Use createInitialiser to resolve the config\n    const initialiser = createInitialiser(config);\n    this.config = initialiser({}, undefined);\n    this.options = config.agents ?? {};\n  }\n\n  /**\n   * Initialize the tracer provider (called lazily on first emit)\n   */\n  private initialize(): void {\n    if (this.initialized) return;\n    initProvider(this.config);\n    this.initialized = true;\n  }\n\n  /**\n   * Emit an observability event\n   *\n   * Converts the event to an OpenTelemetry span based on the event type.\n   */\n  emit(event: ObservabilityEvent, ctx?: DurableObjectState): void {\n    // Initialize provider on first emit\n    this.initialize();\n\n    // Check if this event type should be traced\n    if (!shouldTraceEvent(event, this.options)) {\n      return;\n    }\n\n    const tracer = trace.getTracer('autotel-cloudflare/agents');\n\n    // Get span name (custom or default)\n    const spanName = this.options.spanNameFormatter\n      ? this.options.spanNameFormatter(event)\n      : getDefaultSpanName(event);\n\n    // Get attributes (custom + default)\n    const defaultAttrs = getDefaultAttributes(event);\n    const customAttrs = this.options.attributeExtractor\n      ? this.options.attributeExtractor(event)\n      : {};\n    const attributes = { ...defaultAttrs, ...customAttrs };\n\n    // Determine span kind\n    const kind = getSpanKind(event);\n\n    // Create span with event timestamp\n    const span = tracer.startSpan(spanName, {\n      kind,\n      attributes,\n      startTime: event.timestamp,\n    });\n\n    // For short-lived events, end immediately\n    // For events that have duration (like RPC), we would ideally track start/end\n    // But the Agents SDK emits single events, so we create point-in-time spans\n    span.setStatus({ code: SpanStatusCode.OK });\n    span.end(event.timestamp + 1); // End 1ms after start\n\n    // Store span for potential correlation\n    activeSpans.set(event.id, span);\n\n    // Schedule span export\n    const traceId = span.spanContext().traceId;\n    if (ctx && 'waitUntil' in ctx && typeof (ctx as any).waitUntil === 'function') {\n      (ctx as any).waitUntil(exportSpans(traceId, ctx));\n    } else {\n      // In environments without waitUntil, export synchronously-ish\n      void exportSpans(traceId, ctx);\n    }\n  }\n}\n\n/**\n * Create an OtelObservability instance\n *\n * @example\n * ```typescript\n * import { Agent } from 'agents'\n * import { createOtelObservability } from 'autotel-cloudflare/agents'\n *\n * class MyAgent extends Agent<Env> {\n *   observability = createOtelObservability({\n *     service: { name: 'my-agent' },\n *     exporter: { url: env.OTLP_ENDPOINT }\n *   })\n * }\n * ```\n */\nexport function createOtelObservability(config: OtelObservabilityConfig): OtelObservability {\n  return new OtelObservability(config);\n}\n\n/**\n * Create an OtelObservability instance with environment-based config\n *\n * Use this when you need to access environment variables for configuration.\n *\n * @example\n * ```typescript\n * import { Agent } from 'agents'\n * import { createOtelObservabilityFromEnv } from 'autotel-cloudflare/agents'\n *\n * class MyAgent extends Agent<Env> {\n *   observability?: OtelObservability\n *\n *   constructor(state: DurableObjectState, env: Env) {\n *     super(state, env)\n *     this.observability = createOtelObservabilityFromEnv(env)\n *   }\n * }\n * ```\n */\nexport function createOtelObservabilityFromEnv(\n  env: Record<string, unknown>,\n  options?: AgentInstrumentationOptions,\n): OtelObservability {\n  // Extract standard OTLP env vars\n  const endpoint = (env.OTEL_EXPORTER_OTLP_ENDPOINT as string) || undefined;\n  const serviceName = (env.OTEL_SERVICE_NAME as string) || 'cloudflare-agent';\n\n  // Parse headers if present\n  let headers: Record<string, string> | undefined;\n  const headersStr = env.OTEL_EXPORTER_OTLP_HEADERS as string;\n  if (headersStr) {\n    headers = {};\n    for (const pair of headersStr.split(',')) {\n      const [key, value] = pair.split('=');\n      if (key && value) {\n        headers[key.trim()] = value.trim();\n      }\n    }\n  }\n\n  // If no endpoint is configured, use a default localhost endpoint\n  // In production, users should set OTEL_EXPORTER_OTLP_ENDPOINT\n  const exporterUrl = endpoint || 'http://localhost:4318/v1/traces';\n\n  return createOtelObservability({\n    service: { name: serviceName },\n    exporter: { url: exporterUrl, headers },\n    agents: options,\n  });\n}\n"]}

package/dist/bindings.d.ts

@@ -0,0 +1,40 @@
+/**
+ * Auto-instrumentation for Cloudflare Workers bindings
+ *
+ * Note: This file uses Cloudflare Workers types (KVNamespace, R2Bucket, D1Database, Fetcher, etc.)
+ * which are globally available via @cloudflare/workers-types when listed in tsconfig.json.
+ * These types are devDependencies only - they're not runtime dependencies.
+ * At runtime, Cloudflare Workers runtime provides the actual implementations.
+ *
+ * This module provides automatic tracing for Cloudflare bindings:
+ * - KV (key-value operations)
+ * - R2 (object storage operations)
+ * - D1 (database operations)
+ * - Service Bindings
+ * - Events Engine
+ * - Workers AI
+ * - Vectorize
+ * - Hyperdrive
+ */
+/**
+ * Instrument KV namespace
+ */
+declare function instrumentKV<K extends KVNamespace>(kv: K, namespaceName?: string): K;
+/**
+ * Instrument R2 bucket
+ */
+declare function instrumentR2<R extends R2Bucket>(r2: R, bucketName?: string): R;
+/**
+ * Instrument D1 database
+ */
+declare function instrumentD1<D extends D1Database>(d1: D, databaseName?: string): D;
+/**
+ * Instrument service binding (Fetcher)
+ */
+declare function instrumentServiceBinding<F extends Fetcher>(fetcher: F, serviceName?: string): F;
+/**
+ * Auto-instrument all Cloudflare bindings in the environment
+ */
+declare function instrumentBindings(env: Record<string, any>): Record<string, any>;
+
+export { instrumentBindings, instrumentD1, instrumentKV, instrumentR2, instrumentServiceBinding };

package/dist/bindings.js

@@ -0,0 +1,4 @@
+export { instrumentBindings, instrumentD1, instrumentKV, instrumentR2, instrumentServiceBinding } from './chunk-JDPN3HND.js';
+import './chunk-SKKRPS5K.js';
+//# sourceMappingURL=bindings.js.map
+//# sourceMappingURL=bindings.js.map

package/dist/bindings.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"names":[],"mappings":"","file":"bindings.js"}