@raindrop-ai/pi-agent 0.0.2

package/dist/extension.cjs

@@ -0,0 +1,1311 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/extension.ts
+var extension_exports = {};
+__export(extension_exports, {
+  default: () => extension,
+  registerTracing: () => registerTracing
+});
+module.exports = __toCommonJS(extension_exports);
+
+// src/internal/config.ts
+var import_node_fs = require("fs");
+var import_node_os = require("os");
+var import_node_path = require("path");
+function getPiAgentDirectory() {
+  var _a;
+  return (_a = process.env["PI_CODING_AGENT_DIR"]) != null ? _a : (0, import_node_path.join)((0, import_node_os.homedir)(), ".pi", "agent");
+}
+function loadConfig(projectDirectory) {
+  var _a, _b, _c, _d, _e, _f, _g;
+  let merged = {};
+  const configPaths = [
+    (0, import_node_path.join)(getPiAgentDirectory(), "raindrop.json"),
+    (0, import_node_path.join)(projectDirectory, ".pi", "raindrop.json")
+  ];
+  for (const configPath of configPaths) {
+    try {
+      if ((0, import_node_fs.existsSync)(configPath)) {
+        const content = (0, import_node_fs.readFileSync)(configPath, "utf-8");
+        const parsed = JSON.parse(content);
+        merged = { ...merged, ...parsed };
+      }
+    } catch (e) {
+    }
+  }
+  let eventMetadata;
+  const envMeta = process.env["RAINDROP_EVENT_METADATA"];
+  if (envMeta) {
+    try {
+      eventMetadata = JSON.parse(envMeta);
+    } catch (e) {
+    }
+  }
+  return {
+    writeKey: (_b = (_a = process.env["RAINDROP_WRITE_KEY"]) != null ? _a : merged.write_key) != null ? _b : "",
+    endpoint: (_d = (_c = process.env["RAINDROP_API_URL"]) != null ? _c : merged.api_url) != null ? _d : "https://api.raindrop.ai/v1",
+    eventName: (_e = merged.event_name) != null ? _e : "pi_session",
+    debug: process.env["RAINDROP_DEBUG"] === "true" ? true : (_f = merged.debug) != null ? _f : false,
+    captureSystemPrompt: process.env["RAINDROP_CAPTURE_SYSTEM_PROMPT"] !== void 0 ? process.env["RAINDROP_CAPTURE_SYSTEM_PROMPT"] === "true" : (_g = merged.capture_system_prompt) != null ? _g : false,
+    eventMetadata
+  };
+}
+
+// package.json
+var package_default = {
+  name: "@raindrop-ai/pi-agent",
+  version: "0.0.2",
+  description: "Raindrop observability for Pi Agent \u2014 automatic tracing via subscriber or pi-coding-agent extension",
+  type: "module",
+  license: "MIT",
+  repository: {
+    type: "git",
+    url: "git+https://github.com/raindrop-ai/raindrop-js.git",
+    directory: "packages/pi-agent"
+  },
+  homepage: "https://github.com/raindrop-ai/raindrop-js/tree/main/packages/pi-agent#readme",
+  bugs: {
+    url: "https://github.com/raindrop-ai/raindrop-js/issues"
+  },
+  keywords: [
+    "pi-package",
+    "raindrop",
+    "pi-agent",
+    "observability",
+    "tracing"
+  ],
+  exports: {
+    ".": {
+      types: "./dist/index.d.ts",
+      import: "./dist/index.js",
+      require: "./dist/index.cjs"
+    },
+    "./extension": {
+      types: "./dist/extension.d.ts",
+      import: "./dist/extension.js",
+      require: "./dist/extension.cjs"
+    }
+  },
+  pi: {
+    extensions: [
+      "./dist/extension.js"
+    ]
+  },
+  sideEffects: false,
+  files: [
+    "dist/**",
+    "README.md"
+  ],
+  scripts: {
+    build: "tsup",
+    dev: "tsup --watch",
+    clean: "rm -rf dist",
+    test: "vitest run",
+    "test:watch": "vitest"
+  },
+  peerDependencies: {
+    "@mariozechner/pi-agent-core": ">=0.60.0",
+    "@mariozechner/pi-coding-agent": ">=0.65.2"
+  },
+  peerDependenciesMeta: {
+    "@mariozechner/pi-coding-agent": {
+      optional: true
+    }
+  },
+  devDependencies: {
+    "@raindrop-ai/core": "workspace:*",
+    "@mariozechner/pi-agent-core": "^0.66.0",
+    "@mariozechner/pi-coding-agent": "^0.66.0",
+    "@types/node": "^20.11.17",
+    msw: "^2.12.7",
+    tsup: "^8.5.1",
+    typescript: "^5.7.3",
+    vitest: "^2.1.9"
+  },
+  tsup: {
+    entry: [
+      "src/index.ts",
+      "src/extension.ts"
+    ],
+    format: [
+      "cjs",
+      "esm"
+    ],
+    dts: {
+      resolve: true
+    },
+    clean: true,
+    noExternal: [
+      "@raindrop-ai/core"
+    ]
+  },
+  publishConfig: {
+    access: "public"
+  }
+};
+
+// src/version.ts
+var libraryName = package_default.name;
+var libraryVersion = package_default.version;
+
+// ../core/dist/chunk-4UCYIEH4.js
+function getCrypto() {
+  const c = globalThis.crypto;
+  return c;
+}
+function randomBytes(length) {
+  const cryptoObj = getCrypto();
+  const out = new Uint8Array(length);
+  if (cryptoObj && typeof cryptoObj.getRandomValues === "function") {
+    cryptoObj.getRandomValues(out);
+    return out;
+  }
+  for (let i = 0; i < out.length; i++) out[i] = Math.floor(Math.random() * 256);
+  return out;
+}
+function base64Encode(bytes) {
+  const maybeBuffer = globalThis.Buffer;
+  if (maybeBuffer) {
+    return maybeBuffer.from(bytes).toString("base64");
+  }
+  let binary = "";
+  for (let i2 = 0; i2 < bytes.length; i2++) {
+    binary += String.fromCharCode(bytes[i2]);
+  }
+  const btoaFn = globalThis.btoa;
+  if (typeof btoaFn === "function") return btoaFn(binary);
+  const alphabet = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
+  let out = "";
+  let i = 0;
+  while (i < binary.length) {
+    const c1 = binary.charCodeAt(i++) & 255;
+    const c2 = i < binary.length ? binary.charCodeAt(i++) & 255 : NaN;
+    const c3 = i < binary.length ? binary.charCodeAt(i++) & 255 : NaN;
+    const e1 = c1 >> 2;
+    const e2 = (c1 & 3) << 4 | (Number.isNaN(c2) ? 0 : c2 >> 4);
+    const e3 = Number.isNaN(c2) ? 64 : (c2 & 15) << 2 | (Number.isNaN(c3) ? 0 : c3 >> 6);
+    const e4 = Number.isNaN(c3) ? 64 : c3 & 63;
+    out += alphabet.charAt(e1);
+    out += alphabet.charAt(e2);
+    out += e3 === 64 ? "=" : alphabet.charAt(e3);
+    out += e4 === 64 ? "=" : alphabet.charAt(e4);
+  }
+  return out;
+}
+function generateId() {
+  return base64Encode(randomBytes(12)).replace(/[+/=]/g, "").slice(0, 16);
+}
+function wait(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+function formatEndpoint(endpoint) {
+  if (!endpoint) return void 0;
+  return endpoint.endsWith("/") ? endpoint : `${endpoint}/`;
+}
+function parseRetryAfter(headers) {
+  var _a;
+  const value = (_a = headers.get("Retry-After")) != null ? _a : headers.get("retry-after");
+  if (!value) return void 0;
+  const asNumber = Number(value);
+  if (value.trim() !== "" && !Number.isNaN(asNumber)) return asNumber * 1e3;
+  const asDate = new Date(value).getTime();
+  if (!Number.isNaN(asDate)) {
+    const delta = asDate - Date.now();
+    return delta > 0 ? delta : 0;
+  }
+  return void 0;
+}
+function getRetryDelayMs(attemptNumber, previousError) {
+  if (previousError && typeof previousError === "object" && previousError !== null && "retryAfterMs" in previousError) {
+    const v = previousError.retryAfterMs;
+    if (typeof v === "number") return Math.max(0, v);
+  }
+  if (attemptNumber <= 1) return 0;
+  const base = 500;
+  const factor = Math.pow(2, attemptNumber - 2);
+  return base * factor;
+}
+async function withRetry(operation, opName, opts) {
+  const prefix = opts.sdkName ? `[raindrop-ai/${opts.sdkName}]` : "[raindrop-ai/core]";
+  let lastError = void 0;
+  for (let attemptNumber = 1; attemptNumber <= opts.maxAttempts; attemptNumber++) {
+    if (attemptNumber > 1) {
+      const delay = getRetryDelayMs(attemptNumber, lastError);
+      if (opts.debug) {
+        console.warn(
+          `${prefix} ${opName} retry ${attemptNumber}/${opts.maxAttempts} in ${delay}ms`
+        );
+      }
+      if (delay > 0) await wait(delay);
+    } else if (opts.debug) {
+      console.log(`${prefix} ${opName} attempt ${attemptNumber}/${opts.maxAttempts}`);
+    }
+    try {
+      return await operation();
+    } catch (err) {
+      lastError = err;
+      if (opts.debug) {
+        const msg = err instanceof Error ? err.message : String(err);
+        console.warn(
+          `${prefix} ${opName} attempt ${attemptNumber} failed: ${msg}${attemptNumber === opts.maxAttempts ? " (no more retries)" : ""}`
+        );
+      }
+      if (lastError && typeof lastError === "object" && "retryable" in lastError && !lastError.retryable)
+        break;
+      if (attemptNumber === opts.maxAttempts) break;
+    }
+  }
+  throw lastError instanceof Error ? lastError : new Error(String(lastError));
+}
+async function postJson(url, body, headers, opts) {
+  const opName = `POST ${url}`;
+  await withRetry(
+    async () => {
+      const resp = await fetch(url, {
+        method: "POST",
+        headers: {
+          "Content-Type": "application/json",
+          ...headers
+        },
+        body: JSON.stringify(body)
+      });
+      if (!resp.ok) {
+        const text = await resp.text().catch(() => "");
+        const err = new Error(
+          `HTTP ${resp.status} ${resp.statusText}${text ? `: ${text}` : ""}`
+        );
+        const retryAfterMs = parseRetryAfter(resp.headers);
+        if (typeof retryAfterMs === "number") err.retryAfterMs = retryAfterMs;
+        err.retryable = resp.status === 429 || resp.status >= 500;
+        throw err;
+      }
+    },
+    opName,
+    opts
+  );
+}
+var SpanStatusCode = {
+  UNSET: 0,
+  OK: 1,
+  ERROR: 2
+};
+function createSpanIds(parent) {
+  const traceId = parent ? parent.traceIdB64 : base64Encode(randomBytes(16));
+  const spanId = base64Encode(randomBytes(8));
+  return {
+    traceIdB64: traceId,
+    spanIdB64: spanId,
+    parentSpanIdB64: parent ? parent.spanIdB64 : void 0
+  };
+}
+function nowUnixNanoString() {
+  return Date.now().toString() + "000000";
+}
+function attrString(key, value) {
+  if (value === void 0) return void 0;
+  return { key, value: { stringValue: value } };
+}
+function attrInt(key, value) {
+  if (value === void 0) return void 0;
+  if (!Number.isFinite(value)) return void 0;
+  return { key, value: { intValue: String(Math.trunc(value)) } };
+}
+function buildOtlpSpan(args) {
+  const attrs = args.attributes.filter((x) => x !== void 0);
+  const span = {
+    traceId: args.ids.traceIdB64,
+    spanId: args.ids.spanIdB64,
+    name: args.name,
+    startTimeUnixNano: args.startTimeUnixNano,
+    endTimeUnixNano: args.endTimeUnixNano
+  };
+  if (args.ids.parentSpanIdB64) span.parentSpanId = args.ids.parentSpanIdB64;
+  if (attrs.length) span.attributes = attrs;
+  if (args.status) span.status = args.status;
+  return span;
+}
+function buildExportTraceServiceRequest(spans, serviceName = "raindrop.core", serviceVersion = "0.0.0") {
+  return {
+    resourceSpans: [
+      {
+        resource: {
+          attributes: [{ key: "service.name", value: { stringValue: serviceName } }]
+        },
+        scopeSpans: [
+          {
+            scope: { name: serviceName, version: serviceVersion },
+            spans
+          }
+        ]
+      }
+    ]
+  };
+}
+function mergePatches(target, source) {
+  var _a, _b, _c, _d;
+  const out = { ...target, ...source };
+  if (target.properties || source.properties) {
+    out.properties = { ...(_a = target.properties) != null ? _a : {}, ...(_b = source.properties) != null ? _b : {} };
+  }
+  if (target.attachments || source.attachments) {
+    out.attachments = [...(_c = target.attachments) != null ? _c : [], ...(_d = source.attachments) != null ? _d : []];
+  }
+  return out;
+}
+var EventShipper = class {
+  constructor(opts) {
+    this.buffers = /* @__PURE__ */ new Map();
+    this.sticky = /* @__PURE__ */ new Map();
+    this.timers = /* @__PURE__ */ new Map();
+    this.inFlight = /* @__PURE__ */ new Set();
+    var _a, _b, _c, _d, _e, _f, _g;
+    this.writeKey = (_a = opts.writeKey) == null ? void 0 : _a.trim();
+    this.baseUrl = (_b = formatEndpoint(opts.endpoint)) != null ? _b : "https://api.raindrop.ai/v1/";
+    this.enabled = opts.enabled !== false;
+    this.debug = opts.debug;
+    this.partialFlushMs = (_c = opts.partialFlushMs) != null ? _c : 1e3;
+    this.sdkName = (_d = opts.sdkName) != null ? _d : "core";
+    this.prefix = `[raindrop-ai/${this.sdkName}]`;
+    this.defaultEventName = (_e = opts.defaultEventName) != null ? _e : "ai_generation";
+    const isNode = typeof process !== "undefined" && typeof process.version === "string";
+    this.context = {
+      library: {
+        name: (_f = opts.libraryName) != null ? _f : "@raindrop-ai/core",
+        version: (_g = opts.libraryVersion) != null ? _g : "0.0.0"
+      },
+      metadata: {
+        jsRuntime: isNode ? "node" : "web",
+        ...isNode ? { nodeVersion: process.version } : {}
+      }
+    };
+  }
+  isDebugEnabled() {
+    return this.debug;
+  }
+  authHeaders() {
+    return this.writeKey ? { Authorization: `Bearer ${this.writeKey}` } : {};
+  }
+  async patch(eventId, patch) {
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k;
+    if (!this.enabled) return;
+    if (!eventId || !eventId.trim()) return;
+    if (this.debug) {
+      console.log(`${this.prefix} queue patch`, {
+        eventId,
+        userId: patch.userId,
+        convoId: patch.convoId,
+        eventName: patch.eventName,
+        hasInput: typeof patch.input === "string" && patch.input.length > 0,
+        hasOutput: typeof patch.output === "string" && patch.output.length > 0,
+        attachments: (_b = (_a = patch.attachments) == null ? void 0 : _a.length) != null ? _b : 0,
+        isPending: patch.isPending
+      });
+    }
+    const sticky = (_c = this.sticky.get(eventId)) != null ? _c : {};
+    const existing = (_d = this.buffers.get(eventId)) != null ? _d : {};
+    const merged = mergePatches(existing, patch);
+    merged.isPending = (_g = (_f = (_e = patch.isPending) != null ? _e : existing.isPending) != null ? _f : sticky.isPending) != null ? _g : true;
+    this.buffers.set(eventId, merged);
+    this.sticky.set(eventId, {
+      userId: (_h = merged.userId) != null ? _h : sticky.userId,
+      convoId: (_i = merged.convoId) != null ? _i : sticky.convoId,
+      eventName: (_j = merged.eventName) != null ? _j : sticky.eventName,
+      isPending: (_k = merged.isPending) != null ? _k : sticky.isPending
+    });
+    const t = this.timers.get(eventId);
+    if (t) clearTimeout(t);
+    if (merged.isPending === false) {
+      await this.flushOne(eventId);
+      return;
+    }
+    const timeout = setTimeout(() => {
+      void this.flushOne(eventId).catch(() => {
+      });
+    }, this.partialFlushMs);
+    this.timers.set(eventId, timeout);
+  }
+  async finish(eventId, patch) {
+    await this.patch(eventId, { ...patch, isPending: false });
+  }
+  async flush() {
+    if (!this.enabled) return;
+    const ids = [...this.buffers.keys()];
+    await Promise.all(ids.map((id) => this.flushOne(id)));
+    await Promise.all([...this.inFlight].map((p) => p.catch(() => {
+    })));
+  }
+  async shutdown() {
+    for (const t of this.timers.values()) clearTimeout(t);
+    this.timers.clear();
+    await this.flush();
+  }
+  async trackSignal(signal) {
+    var _a, _b;
+    if (!this.enabled) return;
+    const body = [
+      {
+        event_id: signal.eventId,
+        signal_name: signal.name,
+        signal_type: (_a = signal.type) != null ? _a : "default",
+        timestamp: signal.timestamp,
+        sentiment: signal.sentiment,
+        attachment_id: signal.attachmentId,
+        properties: {
+          ...(_b = signal.properties) != null ? _b : {},
+          ...signal.comment ? { comment: signal.comment } : {},
+          ...signal.after ? { after: signal.after } : {}
+        }
+      }
+    ];
+    const url = `${this.baseUrl}signals/track`;
+    try {
+      await postJson(url, body, this.authHeaders(), {
+        maxAttempts: 3,
+        debug: this.debug,
+        sdkName: this.sdkName
+      });
+    } catch (err) {
+      const msg = err instanceof Error ? err.message : String(err);
+      console.warn(`${this.prefix} failed to send signal (dropping): ${msg}`);
+    }
+  }
+  async identify(users) {
+    if (!this.enabled) return;
+    const list = Array.isArray(users) ? users : [users];
+    const body = list.filter((user) => {
+      if (!(user == null ? void 0 : user.userId) || !user.userId.trim()) {
+        if (this.debug) {
+          console.warn(`${this.prefix} skipping identify: missing userId`);
+        }
+        return false;
+      }
+      return true;
+    }).map((user) => {
+      var _a;
+      return {
+        user_id: user.userId,
+        traits: (_a = user.traits) != null ? _a : {}
+      };
+    });
+    if (body.length === 0) return;
+    const url = `${this.baseUrl}users/identify`;
+    try {
+      await postJson(url, body, this.authHeaders(), {
+        maxAttempts: 3,
+        debug: this.debug,
+        sdkName: this.sdkName
+      });
+    } catch (err) {
+      const msg = err instanceof Error ? err.message : String(err);
+      console.warn(`${this.prefix} failed to send identify (dropping): ${msg}`);
+    }
+  }
+  async flushOne(eventId) {
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m;
+    if (!this.enabled) return;
+    const timer = this.timers.get(eventId);
+    if (timer) {
+      clearTimeout(timer);
+      this.timers.delete(eventId);
+    }
+    const accumulated = this.buffers.get(eventId);
+    this.buffers.delete(eventId);
+    if (!accumulated) return;
+    const sticky = (_a = this.sticky.get(eventId)) != null ? _a : {};
+    const eventName = (_c = (_b = accumulated.eventName) != null ? _b : sticky.eventName) != null ? _c : this.defaultEventName;
+    const userId = (_d = accumulated.userId) != null ? _d : sticky.userId;
+    if (!userId) {
+      if (this.debug) {
+        console.warn(`${this.prefix} skipping track_partial for ${eventId}: missing userId`);
+      }
+      this.sticky.delete(eventId);
+      return;
+    }
+    const { wizardSession, ...restProperties } = (_e = accumulated.properties) != null ? _e : {};
+    const convoId = (_f = accumulated.convoId) != null ? _f : sticky.convoId;
+    const isPending = (_h = (_g = accumulated.isPending) != null ? _g : sticky.isPending) != null ? _h : true;
+    const payload = {
+      event_id: eventId,
+      user_id: userId,
+      event: eventName,
+      timestamp: (_i = accumulated.timestamp) != null ? _i : (/* @__PURE__ */ new Date()).toISOString(),
+      ai_data: {
+        input: accumulated.input,
+        output: accumulated.output,
+        model: accumulated.model,
+        convo_id: convoId
+      },
+      properties: {
+        ...restProperties,
+        ...wizardSession ? { "raindrop.wizardSession": wizardSession } : {},
+        $context: this.context
+      },
+      attachments: accumulated.attachments,
+      is_pending: isPending
+    };
+    const url = `${this.baseUrl}events/track_partial`;
+    if (this.debug) {
+      console.log(`${this.prefix} sending track_partial`, {
+        eventId,
+        eventName,
+        userId,
+        convoId,
+        isPending,
+        inputPreview: typeof accumulated.input === "string" ? accumulated.input.slice(0, 120) : void 0,
+        outputPreview: typeof accumulated.output === "string" ? accumulated.output.slice(0, 120) : void 0,
+        attachments: (_k = (_j = accumulated.attachments) == null ? void 0 : _j.length) != null ? _k : 0,
+        attachmentKinds: (_m = (_l = accumulated.attachments) == null ? void 0 : _l.map((a) => ({
+          type: a.type,
+          role: a.role,
+          name: a.name,
+          valuePreview: a.value.slice(0, 60)
+        }))) != null ? _m : [],
+        endpoint: url
+      });
+    }
+    const p = postJson(url, payload, this.authHeaders(), {
+      maxAttempts: 3,
+      debug: this.debug,
+      sdkName: this.sdkName
+    });
+    this.inFlight.add(p);
+    try {
+      try {
+        await p;
+        if (this.debug) {
+          console.log(`${this.prefix} sent track_partial ${eventId} (${eventName})`);
+        }
+      } catch (err) {
+        const msg = err instanceof Error ? err.message : String(err);
+        console.warn(`${this.prefix} failed to send track_partial (dropping): ${msg}`);
+      }
+    } finally {
+      this.inFlight.delete(p);
+    }
+    if (!isPending) {
+      this.sticky.delete(eventId);
+    }
+  }
+};
+var LOCAL_DEBUGGER_ENV_VAR = "RAINDROP_LOCAL_DEBUGGER";
+function resolveLocalDebuggerBaseUrl(baseUrl) {
+  var _a, _b, _c;
+  const resolved = (_b = baseUrl != null ? baseUrl : typeof process !== "undefined" ? (_a = process.env) == null ? void 0 : _a[LOCAL_DEBUGGER_ENV_VAR] : void 0) != null ? _b : null;
+  return resolved ? (_c = formatEndpoint(resolved)) != null ? _c : null : null;
+}
+function mirrorTraceExportToLocalDebugger(body, options = {}) {
+  var _a;
+  const baseUrl = resolveLocalDebuggerBaseUrl(options.baseUrl);
+  if (!baseUrl) return;
+  void postJson(`${baseUrl}traces`, body, {}, {
+    maxAttempts: 1,
+    debug: (_a = options.debug) != null ? _a : false,
+    sdkName: options.sdkName
+  }).catch(() => {
+  });
+}
+var TraceShipper = class {
+  constructor(opts) {
+    this.queue = [];
+    this.inFlight = /* @__PURE__ */ new Set();
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
+    this.writeKey = (_a = opts.writeKey) == null ? void 0 : _a.trim();
+    this.baseUrl = (_b = formatEndpoint(opts.endpoint)) != null ? _b : "https://api.raindrop.ai/v1/";
+    this.enabled = opts.enabled !== false;
+    this.debug = opts.debug;
+    this.debugSpans = opts.debugSpans === true;
+    this.flushIntervalMs = (_c = opts.flushIntervalMs) != null ? _c : 1e3;
+    this.maxBatchSize = (_d = opts.maxBatchSize) != null ? _d : 50;
+    this.maxQueueSize = (_e = opts.maxQueueSize) != null ? _e : 5e3;
+    this.sdkName = (_f = opts.sdkName) != null ? _f : "core";
+    this.prefix = `[raindrop-ai/${this.sdkName}]`;
+    this.serviceName = (_g = opts.serviceName) != null ? _g : "raindrop.core";
+    this.serviceVersion = (_h = opts.serviceVersion) != null ? _h : "0.0.0";
+    const localDebugger = typeof process !== "undefined" ? (_i = process.env) == null ? void 0 : _i.RAINDROP_LOCAL_DEBUGGER : void 0;
+    if (localDebugger) {
+      this.localDebuggerUrl = (_j = resolveLocalDebuggerBaseUrl(localDebugger)) != null ? _j : void 0;
+      if (this.debug) {
+        console.log(`${this.prefix} Local debugger mirroring: ${this.localDebuggerUrl}`);
+      }
+    }
+  }
+  isDebugEnabled() {
+    return this.debug;
+  }
+  authHeaders() {
+    return this.writeKey ? { Authorization: `Bearer ${this.writeKey}` } : {};
+  }
+  startSpan(args) {
+    var _a, _b;
+    const ids = createSpanIds(args.parent);
+    const started = (_a = args.startTimeUnixNano) != null ? _a : nowUnixNanoString();
+    const attrs = [
+      attrString("ai.telemetry.metadata.raindrop.eventId", args.eventId),
+      attrString("ai.operationId", args.operationId)
+    ];
+    if ((_b = args.attributes) == null ? void 0 : _b.length) attrs.push(...args.attributes);
+    const span = { ids, name: args.name, startTimeUnixNano: started, attributes: attrs };
+    if (this.localDebuggerUrl) {
+      const openSpan = buildOtlpSpan({
+        ids: span.ids,
+        name: span.name,
+        startTimeUnixNano: span.startTimeUnixNano,
+        endTimeUnixNano: span.startTimeUnixNano,
+        // placeholder — will be updated on endSpan
+        attributes: span.attributes,
+        status: { code: SpanStatusCode.UNSET }
+      });
+      const body = buildExportTraceServiceRequest([openSpan], this.serviceName, this.serviceVersion);
+      mirrorTraceExportToLocalDebugger(body, {
+        baseUrl: this.localDebuggerUrl,
+        debug: false,
+        sdkName: this.sdkName
+      });
+    }
+    return span;
+  }
+  endSpan(span, extra) {
+    var _a, _b;
+    if (span.endTimeUnixNano) return;
+    span.endTimeUnixNano = (_a = extra == null ? void 0 : extra.endTimeUnixNano) != null ? _a : nowUnixNanoString();
+    if ((_b = extra == null ? void 0 : extra.attributes) == null ? void 0 : _b.length) {
+      span.attributes.push(...extra.attributes);
+    }
+    let status = extra == null ? void 0 : extra.status;
+    if (!status && (extra == null ? void 0 : extra.error) !== void 0) {
+      const message = extra.error instanceof Error ? extra.error.message : String(extra.error);
+      status = { code: SpanStatusCode.ERROR, message };
+    }
+    const otlp = buildOtlpSpan({
+      ids: span.ids,
+      name: span.name,
+      startTimeUnixNano: span.startTimeUnixNano,
+      endTimeUnixNano: span.endTimeUnixNano,
+      attributes: span.attributes,
+      status
+    });
+    this.enqueue(otlp);
+    if (this.localDebuggerUrl) {
+      const body = buildExportTraceServiceRequest([otlp], this.serviceName, this.serviceVersion);
+      mirrorTraceExportToLocalDebugger(body, {
+        baseUrl: this.localDebuggerUrl,
+        debug: false,
+        sdkName: this.sdkName
+      });
+    }
+  }
+  createSpan(args) {
+    var _a;
+    const ids = createSpanIds(args.parent);
+    const attrs = [
+      attrString("ai.telemetry.metadata.raindrop.eventId", args.eventId)
+    ];
+    if ((_a = args.attributes) == null ? void 0 : _a.length) attrs.push(...args.attributes);
+    const otlp = buildOtlpSpan({
+      ids,
+      name: args.name,
+      startTimeUnixNano: args.startTimeUnixNano,
+      endTimeUnixNano: args.endTimeUnixNano,
+      attributes: attrs,
+      status: args.status
+    });
+    this.enqueue(otlp);
+    if (this.localDebuggerUrl) {
+      const body = buildExportTraceServiceRequest([otlp], this.serviceName, this.serviceVersion);
+      mirrorTraceExportToLocalDebugger(body, {
+        baseUrl: this.localDebuggerUrl,
+        debug: false,
+        sdkName: this.sdkName
+      });
+    }
+  }
+  enqueue(span) {
+    if (!this.enabled) return;
+    if (this.debugSpans) {
+      const short = (s) => s ? s.slice(-8) : "none";
+      console.log(
+        `${this.prefix}[span] name=${span.name} trace=${short(span.traceId)} span=${short(span.spanId)} parent=${short(
+          span.parentSpanId
+        )}`
+      );
+    }
+    if (this.queue.length >= this.maxQueueSize) {
+      this.queue.shift();
+    }
+    this.queue.push(span);
+    if (this.queue.length >= this.maxBatchSize) {
+      void this.flush().catch(() => {
+      });
+      return;
+    }
+    if (!this.timer) {
+      this.timer = setTimeout(() => {
+        this.timer = void 0;
+        void this.flush().catch(() => {
+        });
+      }, this.flushIntervalMs);
+    }
+  }
+  async flush() {
+    if (!this.enabled) return;
+    if (this.timer) {
+      clearTimeout(this.timer);
+      this.timer = void 0;
+    }
+    while (this.queue.length > 0) {
+      const batch = this.queue.splice(0, this.maxBatchSize);
+      const body = buildExportTraceServiceRequest(batch, this.serviceName, this.serviceVersion);
+      const url = `${this.baseUrl}traces`;
+      if (this.debug) {
+        console.log(`${this.prefix} sending traces batch`, {
+          spans: batch.length,
+          endpoint: url
+        });
+      }
+      const p = postJson(url, body, this.authHeaders(), {
+        maxAttempts: 3,
+        debug: this.debug,
+        sdkName: this.sdkName
+      });
+      this.inFlight.add(p);
+      try {
+        try {
+          await p;
+          if (this.debug) console.log(`${this.prefix} sent ${batch.length} spans`);
+        } catch (err) {
+          const msg = err instanceof Error ? err.message : String(err);
+          console.warn(`${this.prefix} failed to send ${batch.length} spans: ${msg}`);
+        }
+      } finally {
+        this.inFlight.delete(p);
+      }
+    }
+  }
+  async shutdown() {
+    if (this.timer) {
+      clearTimeout(this.timer);
+      this.timer = void 0;
+    }
+    await this.flush();
+    await Promise.all([...this.inFlight].map((p) => p.catch(() => {
+    })));
+  }
+};
+
+// ../core/dist/index.node.js
+var import_async_hooks = require("async_hooks");
+globalThis.RAINDROP_ASYNC_LOCAL_STORAGE = import_async_hooks.AsyncLocalStorage;
+
+// src/internal/shipper.ts
+var EventShipper2 = class extends EventShipper {
+  constructor(opts) {
+    var _a, _b, _c, _d;
+    super({
+      ...opts,
+      sdkName: (_a = opts.sdkName) != null ? _a : "pi-agent",
+      libraryName: (_b = opts.libraryName) != null ? _b : libraryName,
+      libraryVersion: (_c = opts.libraryVersion) != null ? _c : libraryVersion,
+      defaultEventName: (_d = opts.defaultEventName) != null ? _d : "pi_agent_prompt"
+    });
+  }
+};
+var TraceShipper2 = class extends TraceShipper {
+  constructor(opts) {
+    var _a, _b, _c;
+    super({
+      ...opts,
+      sdkName: (_a = opts.sdkName) != null ? _a : "pi-agent",
+      serviceName: (_b = opts.serviceName) != null ? _b : "raindrop.pi-agent",
+      serviceVersion: (_c = opts.serviceVersion) != null ? _c : libraryVersion
+    });
+  }
+  enqueue(span) {
+    var _a;
+    const attrs = (_a = span.attributes) != null ? _a : [];
+    attrs.unshift(
+      { key: "span.id", value: { stringValue: span.spanId } },
+      ...span.parentSpanId ? [{ key: "span.parent.id", value: { stringValue: span.parentSpanId } }] : []
+    );
+    span.attributes = attrs;
+    super.enqueue(span);
+  }
+};
+
+// src/internal/helpers.ts
+var import_node_os2 = require("os");
+function extractAssistantText(message) {
+  if (!("role" in message) || message.role !== "assistant") return void 0;
+  if (!Array.isArray(message.content)) return void 0;
+  const textParts = [];
+  for (const block of message.content) {
+    if (block.type === "text" && typeof block.text === "string") {
+      textParts.push(block.text);
+    }
+  }
+  return textParts.length > 0 ? textParts.join("\n") : void 0;
+}
+function formatToolSpanName(toolName, args) {
+  if (!args || typeof args !== "object") return toolName;
+  try {
+    const obj = args;
+    const firstValue = Object.values(obj)[0];
+    if (typeof firstValue === "string" && firstValue.length > 0) {
+      const preview = firstValue.length > 40 ? firstValue.slice(0, 37) + "..." : firstValue;
+      return `${toolName}: ${preview}`;
+    }
+  } catch (e) {
+  }
+  return toolName;
+}
+function safeStringify(value) {
+  if (value === void 0 || value === null) return void 0;
+  try {
+    return JSON.stringify(value);
+  } catch (e) {
+    try {
+      return String(value);
+    } catch (e2) {
+      return "[unserializable]";
+    }
+  }
+}
+var MAX_ATTR_LENGTH = 32768;
+function truncate(value) {
+  if (value === void 0) return void 0;
+  if (value.length <= MAX_ATTR_LENGTH) return value;
+  const suffix = "\n...[truncated]";
+  return value.slice(0, MAX_ATTR_LENGTH - suffix.length) + suffix;
+}
+var _hostname;
+function getHostname() {
+  var _a;
+  if (_hostname) return _hostname;
+  try {
+    _hostname = (0, import_node_os2.hostname)();
+  } catch (e) {
+    _hostname = (_a = process.env.HOSTNAME) != null ? _a : "unknown";
+  }
+  return _hostname;
+}
+var _username;
+function getUsername() {
+  var _a, _b;
+  if (_username) return _username;
+  try {
+    _username = (0, import_node_os2.userInfo)().username;
+  } catch (e) {
+    _username = (_b = (_a = process.env.USER) != null ? _a : process.env.USERNAME) != null ? _b : "unknown";
+  }
+  return _username;
+}
+
+// src/internal/extension-tracing.ts
+function safeParsArgs(argsStr) {
+  try {
+    return JSON.parse(argsStr);
+  } catch (e) {
+    return void 0;
+  }
+}
+var MAX_SYSTEM_PROMPT_LENGTH = 32768;
+function createSessionState(sessionId) {
+  return {
+    sessionId,
+    currentInput: "",
+    turnNumber: 0,
+    totalInputTokens: 0,
+    totalOutputTokens: 0,
+    totalCacheReadTokens: 0,
+    toolSpanStarts: /* @__PURE__ */ new Map(),
+    toolCallToLlmParent: /* @__PURE__ */ new Map()
+  };
+}
+function getUserId(state, metadata) {
+  var _a;
+  return (_a = metadata == null ? void 0 : metadata.userId) != null ? _a : state.sessionId;
+}
+function getEventName(config) {
+  var _a, _b;
+  return (_b = (_a = config.eventMetadata) == null ? void 0 : _a.eventName) != null ? _b : config.eventName;
+}
+function truncateSystemPrompt(systemPrompt) {
+  if (systemPrompt.length <= MAX_SYSTEM_PROMPT_LENGTH) return systemPrompt;
+  const suffix = "\n...[truncated]";
+  return systemPrompt.slice(0, MAX_SYSTEM_PROMPT_LENGTH - suffix.length) + suffix;
+}
+function getBaseProperties(config, ctx) {
+  var _a;
+  return {
+    workspace: ctx.cwd,
+    directory: ctx.cwd,
+    hostname: getHostname(),
+    os: process.platform,
+    username: getUsername(),
+    sdk_version: libraryVersion,
+    ...(_a = config.eventMetadata) == null ? void 0 : _a.properties
+  };
+}
+function getSystemPromptAttributes(systemPrompt) {
+  if (!systemPrompt) return [];
+  return [
+    attrString("gen_ai.prompt.0.role", "system"),
+    attrString("gen_ai.prompt.0.content", systemPrompt)
+  ];
+}
+function getAssistantText(message) {
+  var _a;
+  return (_a = extractAssistantText(message)) != null ? _a : "";
+}
+function getAssistantError(message) {
+  var _a;
+  if (message.role !== "assistant") return void 0;
+  if (message.stopReason !== "error" && message.stopReason !== "aborted") return void 0;
+  return (_a = message.errorMessage) != null ? _a : `Assistant ${message.stopReason}`;
+}
+function getState(stateRef, ctx) {
+  const sessionId = ctx.sessionManager.getSessionId();
+  if (!stateRef.current || stateRef.current.sessionId !== sessionId) {
+    stateRef.current = createSessionState(sessionId);
+  }
+  return stateRef.current;
+}
+function registerTracing(pi, config, eventShipper, traceShipper) {
+  const stateRef = {};
+  function logError(hook, err) {
+    console.log(
+      `[raindrop-ai/pi-agent] [error] Error in ${hook}: ${err instanceof Error ? err.message : String(err)}`
+    );
+  }
+  function startTurnSpan(state) {
+    if (!state.currentEventId || !state.currentRootSpan) return;
+    if (state.currentTurnSpan) {
+      traceShipper.endSpan(state.currentTurnSpan);
+    }
+    state.turnNumber += 1;
+    state.currentTurnSpan = traceShipper.startSpan({
+      name: `Turn ${state.turnNumber}`,
+      parent: state.currentRootSpan.ids,
+      eventId: state.currentEventId,
+      attributes: [
+        attrString("ai.operationId", "ai.turn"),
+        attrInt("ai.turn_number", state.turnNumber)
+      ]
+    });
+  }
+  function ensureTurnSpan(state) {
+    if (state.currentTurnSpan) return;
+    startTurnSpan(state);
+  }
+  pi.on("session_start", async (_event, ctx) => {
+    try {
+      stateRef.current = createSessionState(ctx.sessionManager.getSessionId());
+    } catch (err) {
+      logError("session_start", err);
+    }
+  });
+  pi.on("before_agent_start", async (event, ctx) => {
+    var _a, _b, _c;
+    try {
+      const state = getState(stateRef, ctx);
+      if (state.currentTurnSpan) {
+        traceShipper.endSpan(state.currentTurnSpan);
+        state.currentTurnSpan = void 0;
+      }
+      if (state.currentRootSpan) {
+        traceShipper.endSpan(state.currentRootSpan, { error: "Previous run was not finalized" });
+        state.currentRootSpan = void 0;
+      }
+      if (state.currentEventId) {
+        await eventShipper.finish((_a = state.currentEventRequestId) != null ? _a : state.currentEventId, {
+          userId: getUserId(state, config.eventMetadata),
+          properties: { sdk_version: libraryVersion, incomplete: true }
+        });
+        state.currentEventRequestId = void 0;
+        state.currentEventId = void 0;
+      }
+      state.toolSpanStarts.clear();
+      state.toolCallToLlmParent.clear();
+      state.currentInput = event.prompt;
+      state.turnNumber = 0;
+      state.currentSystemPrompt = config.captureSystemPrompt ? truncateSystemPrompt(event.systemPrompt) : void 0;
+      state.currentEventRequestId = generateId();
+      state.totalInputTokens = 0;
+      state.totalOutputTokens = 0;
+      state.totalCacheReadTokens = 0;
+      const attachments = (_c = (_b = event.images) == null ? void 0 : _b.map((image, index) => {
+        var _a2, _b2;
+        return {
+          type: "image",
+          role: "input",
+          name: `image-${index + 1}.${(_b2 = (_a2 = image.mimeType.split("/")[1]) == null ? void 0 : _a2.split("+")[0]) != null ? _b2 : "png"}`,
+          value: `data:${image.mimeType};base64,${image.data}`
+        };
+      })) != null ? _c : [];
+      await eventShipper.patch(state.currentEventRequestId, {
+        isPending: true,
+        userId: getUserId(state, config.eventMetadata),
+        convoId: state.sessionId,
+        eventName: getEventName(config),
+        input: event.prompt,
+        ...attachments.length > 0 ? { attachments } : {},
+        properties: getBaseProperties(config, ctx)
+      });
+      await eventShipper.flush();
+      state.currentEventId = state.currentEventRequestId;
+      state.currentRootSpan = traceShipper.startSpan({
+        name: "ai.event",
+        eventId: state.currentEventId,
+        attributes: [
+          attrString("ai.prompt", truncate(state.currentInput)),
+          attrString("workspace", ctx.cwd),
+          attrString("hostname", getHostname()),
+          attrString("os", process.platform),
+          attrString("username", getUsername())
+        ]
+      });
+    } catch (err) {
+      logError("before_agent_start", err);
+    }
+  });
+  pi.on("turn_start", async (_event, ctx) => {
+    try {
+      const state = getState(stateRef, ctx);
+      startTurnSpan(state);
+    } catch (err) {
+      logError("turn_start", err);
+    }
+  });
+  pi.on("turn_end", async (_event, ctx) => {
+    try {
+      const state = getState(stateRef, ctx);
+      if (!state.currentTurnSpan) return;
+      traceShipper.endSpan(state.currentTurnSpan);
+      state.currentTurnSpan = void 0;
+    } catch (err) {
+      logError("turn_end", err);
+    }
+  });
+  pi.on("message_end", async (event, ctx) => {
+    var _a, _b, _c, _d, _e, _f, _g, _h;
+    try {
+      if (((_a = event.message) == null ? void 0 : _a.role) !== "assistant") return;
+      const state = getState(stateRef, ctx);
+      if (!state.currentEventId || !state.currentRootSpan) return;
+      ensureTurnSpan(state);
+      const message = event.message;
+      const provider = (_b = message.provider) != null ? _b : "";
+      const modelId = (_c = message.model) != null ? _c : "";
+      const modelName = provider && modelId ? `${provider}/${modelId}` : modelId || "llm";
+      const errorForSpan = getAssistantError(message);
+      const outputText = getAssistantText(message);
+      const inputTokens = (_d = message.usage) == null ? void 0 : _d.input;
+      const outputTokens = (_e = message.usage) == null ? void 0 : _e.output;
+      const cacheReadTokens = (_f = message.usage) == null ? void 0 : _f.cacheRead;
+      if (typeof inputTokens === "number") state.totalInputTokens += inputTokens;
+      if (typeof outputTokens === "number") state.totalOutputTokens += outputTokens;
+      if (typeof cacheReadTokens === "number" && cacheReadTokens > 0) state.totalCacheReadTokens += cacheReadTokens;
+      const turnParent = (_g = state.currentTurnSpan) != null ? _g : state.currentRootSpan;
+      const llmAttrs = [
+        attrString("ai.operationId", "generateText"),
+        provider ? attrString("gen_ai.system", provider) : void 0,
+        modelId ? attrString("gen_ai.request.model", modelId) : void 0,
+        modelId ? attrString("gen_ai.response.model", modelId) : void 0,
+        typeof inputTokens === "number" ? attrInt("gen_ai.usage.input_tokens", inputTokens) : void 0,
+        typeof outputTokens === "number" ? attrInt("gen_ai.usage.output_tokens", outputTokens) : void 0,
+        typeof cacheReadTokens === "number" && cacheReadTokens > 0 ? attrInt("gen_ai.usage.cache_read_tokens", cacheReadTokens) : void 0,
+        outputText ? attrString("ai.response.text", truncate(outputText)) : void 0,
+        state.currentInput ? attrString("ai.prompt", truncate(state.currentInput)) : void 0,
+        message.stopReason ? attrString("ai.stop_reason", message.stopReason) : void 0,
+        ...getSystemPromptAttributes(state.currentSystemPrompt)
+      ].filter(Boolean);
+      const llmSpan = traceShipper.startSpan({
+        name: modelName,
+        parent: turnParent.ids,
+        eventId: state.currentEventId,
+        attributes: llmAttrs
+      });
+      traceShipper.endSpan(llmSpan, errorForSpan ? { error: errorForSpan } : void 0);
+      if (Array.isArray(message.content)) {
+        for (const part of message.content) {
+          if (part && typeof part === "object" && part.type === "toolCall" && typeof part.id === "string") {
+            state.toolCallToLlmParent.set(part.id, llmSpan.ids);
+          }
+        }
+      }
+      if (message.stopReason === "toolUse") {
+        await traceShipper.flush();
+        return;
+      }
+      if (state.currentTurnSpan) {
+        traceShipper.endSpan(state.currentTurnSpan);
+        state.currentTurnSpan = void 0;
+      }
+      state.currentRootSpan.name = modelName;
+      traceShipper.endSpan(state.currentRootSpan, {
+        attributes: [
+          ...modelId ? [attrString("gen_ai.response.model", modelId)] : [],
+          ...outputText ? [attrString("ai.response.text", truncate(outputText))] : [],
+          attrInt("ai.total_turns", state.turnNumber)
+        ],
+        ...errorForSpan ? { error: errorForSpan } : {}
+      });
+      state.currentRootSpan = void 0;
+      await eventShipper.finish((_h = state.currentEventRequestId) != null ? _h : state.currentEventId, {
+        userId: getUserId(state, config.eventMetadata),
+        model: modelName,
+        ...outputText.trim() ? { output: outputText } : {},
+        properties: {
+          sdk_version: libraryVersion,
+          stop_reason: message.stopReason,
+          ...state.totalInputTokens > 0 ? { "ai.usage.prompt_tokens": state.totalInputTokens } : {},
+          ...state.totalOutputTokens > 0 ? { "ai.usage.completion_tokens": state.totalOutputTokens } : {},
+          ...state.totalCacheReadTokens > 0 ? { "ai.usage.cache_read_tokens": state.totalCacheReadTokens } : {}
+        }
+      });
+      state.currentEventRequestId = void 0;
+      state.currentEventId = void 0;
+      state.toolSpanStarts.clear();
+      state.toolCallToLlmParent.clear();
+      await traceShipper.flush();
+    } catch (err) {
+      logError("message_end", err);
+    }
+  });
+  pi.on("tool_execution_start", async (event, ctx) => {
+    var _a, _b, _c;
+    try {
+      const state = getState(stateRef, ctx);
+      if (!state.currentEventId || !state.currentRootSpan) return;
+      const llmParent = state.toolCallToLlmParent.get(event.toolCallId);
+      const parent = (_b = llmParent != null ? llmParent : (_a = state.currentTurnSpan) == null ? void 0 : _a.ids) != null ? _b : state.currentRootSpan.ids;
+      state.toolSpanStarts.set(event.toolCallId, {
+        startTimeUnixNano: nowUnixNanoString(),
+        parent,
+        eventId: state.currentEventId,
+        name: event.toolName,
+        args: (_c = safeStringify(event.args)) != null ? _c : "{}"
+      });
+    } catch (err) {
+      logError("tool_execution_start", err);
+    }
+  });
+  pi.on("tool_execution_end", async (event, ctx) => {
+    try {
+      const state = getState(stateRef, ctx);
+      const start = state.toolSpanStarts.get(event.toolCallId);
+      if (!start) return;
+      state.toolSpanStarts.delete(event.toolCallId);
+      state.toolCallToLlmParent.delete(event.toolCallId);
+      const resultStr = safeStringify(event.result);
+      traceShipper.createSpan({
+        name: formatToolSpanName(start.name, safeParsArgs(start.args)),
+        parent: start.parent,
+        eventId: start.eventId,
+        startTimeUnixNano: start.startTimeUnixNano,
+        endTimeUnixNano: nowUnixNanoString(),
+        attributes: [
+          attrString("ai.operationId", "ai.toolCall"),
+          attrString("ai.toolCall.name", start.name),
+          attrString("ai.toolCall.id", event.toolCallId),
+          attrString("ai.toolCall.args", truncate(start.args)),
+          ...resultStr ? [attrString("ai.toolCall.result", truncate(resultStr))] : []
+        ],
+        ...event.isError ? { status: { code: 2, message: `Tool "${start.name}" failed` } } : {}
+      });
+    } catch (err) {
+      logError("tool_execution_end", err);
+    }
+  });
+  pi.on("agent_end", async (_event, ctx) => {
+    var _a;
+    try {
+      const state = getState(stateRef, ctx);
+      if (state.currentTurnSpan) {
+        traceShipper.endSpan(state.currentTurnSpan);
+        state.currentTurnSpan = void 0;
+      }
+      if (state.currentRootSpan) {
+        traceShipper.endSpan(state.currentRootSpan, {
+          error: "Agent ended before a final assistant response was recorded"
+        });
+        state.currentRootSpan = void 0;
+      }
+      if (state.currentEventId) {
+        await eventShipper.finish((_a = state.currentEventRequestId) != null ? _a : state.currentEventId, {
+          userId: getUserId(state, config.eventMetadata),
+          properties: {
+            sdk_version: libraryVersion,
+            incomplete: true
+          }
+        });
+        state.currentEventRequestId = void 0;
+        state.currentEventId = void 0;
+      }
+      state.toolSpanStarts.clear();
+      state.toolCallToLlmParent.clear();
+      await Promise.all([eventShipper.flush(), traceShipper.flush()]);
+    } catch (err) {
+      logError("agent_end", err);
+    }
+  });
+  pi.on("session_shutdown", async (_event, _ctx) => {
+    try {
+      stateRef.current = void 0;
+      await Promise.all([eventShipper.shutdown(), traceShipper.shutdown()]);
+    } catch (err) {
+      logError("session_shutdown", err);
+    }
+  });
+}
+
+// src/extension.ts
+var PLUGIN_NAME = "@raindrop-ai/pi-agent";
+function extension(pi) {
+  const config = loadConfig(process.cwd());
+  function appLog(level, message) {
+    console.log(`[raindrop-ai/pi-agent] [${level}] ${message}`);
+  }
+  appLog("info", `Loading ${PLUGIN_NAME} v${libraryVersion}`);
+  if (!config.writeKey) {
+    appLog(
+      "warn",
+      "RAINDROP_WRITE_KEY not set \u2014 Raindrop tracing disabled. Set it in your environment or in ~/.pi/agent/raindrop.json."
+    );
+    return;
+  }
+  if (config.debug) {
+    appLog("info", `Raindrop tracing enabled (endpoint: ${config.endpoint})`);
+  }
+  const eventShipper = new EventShipper2({
+    writeKey: config.writeKey,
+    endpoint: config.endpoint,
+    debug: config.debug
+  });
+  const traceShipper = new TraceShipper2({
+    writeKey: config.writeKey,
+    endpoint: config.endpoint,
+    debug: config.debug
+  });
+  registerTracing(pi, config, eventShipper, traceShipper);
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  registerTracing
+});