@raindrop-ai/langchain 0.0.1

package/dist/index.js

@@ -0,0 +1,1147 @@
+"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/index.ts
+var index_exports = {};
+__export(index_exports, {
+  RaindropCallbackHandler: () => RaindropCallbackHandler,
+  createRaindropLangChain: () => createRaindropLangChain
+});
+module.exports = __toCommonJS(index_exports);
+
+// src/callback-handler.ts
+var import_base = require("@langchain/core/callbacks/base");
+
+// ../core/dist/chunk-FOHDGBT5.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 randomUUID() {
+  const cryptoObj = getCrypto();
+  if (cryptoObj && typeof cryptoObj.randomUUID === "function") {
+    return cryptoObj.randomUUID();
+  }
+  const b = randomBytes(16);
+  b[6] = b[6] & 15 | 64;
+  b[8] = b[8] & 63 | 128;
+  const hex = [...b].map((x) => x.toString(16).padStart(2, "0")).join("");
+  return `${hex.slice(0, 8)}-${hex.slice(8, 12)}-${hex.slice(12, 16)}-${hex.slice(16, 20)}-${hex.slice(20)}`;
+}
+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 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 TraceShipper = class {
+  constructor(opts) {
+    this.queue = [];
+    this.inFlight = /* @__PURE__ */ new Set();
+    var _a, _b, _c, _d, _e, _f, _g, _h;
+    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";
+  }
+  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);
+    return { ids, name: args.name, startTimeUnixNano: started, attributes: attrs };
+  }
+  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);
+  }
+  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);
+  }
+  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(() => {
+    })));
+  }
+};
+
+// src/callback-handler.ts
+var LANGGRAPH_INTERNAL_NAMES = /* @__PURE__ */ new Set([
+  "LangGraph",
+  "__start__",
+  "__end__",
+  "ChannelWrite",
+  "ChannelRead"
+]);
+var RaindropCallbackHandler = class extends import_base.BaseCallbackHandler {
+  constructor(opts) {
+    var _a, _b, _c;
+    super();
+    this.name = "RaindropCallbackHandler";
+    this.spans = /* @__PURE__ */ new Map();
+    this.rootRunIds = /* @__PURE__ */ new Map();
+    this.eventIds = /* @__PURE__ */ new Map();
+    this.eventShipper = opts.eventShipper;
+    this.traceShipper = opts.traceShipper;
+    this.userId = opts.userId;
+    this.convoId = opts.convoId;
+    this.traceChains = (_a = opts.traceChains) != null ? _a : true;
+    this.traceRetrievers = (_b = opts.traceRetrievers) != null ? _b : true;
+    this.filterLangGraphInternals = (_c = opts.filterLangGraphInternals) != null ? _c : true;
+  }
+  getEventId(runId, parentRunId) {
+    var _a;
+    const rootId = parentRunId ? (_a = this.rootRunIds.get(parentRunId)) != null ? _a : parentRunId : runId;
+    this.rootRunIds.set(runId, rootId);
+    if (!this.eventIds.has(rootId)) {
+      this.eventIds.set(rootId, randomUUID());
+    }
+    return this.eventIds.get(rootId);
+  }
+  getParent(parentRunId) {
+    if (!parentRunId) return void 0;
+    const parentSpan = this.spans.get(parentRunId);
+    if (!parentSpan) return void 0;
+    return {
+      traceIdB64: parentSpan.ids.traceIdB64,
+      spanIdB64: parentSpan.ids.spanIdB64
+    };
+  }
+  cleanup(runId) {
+    var _a;
+    this.spans.delete(runId);
+    this.spans.delete(`${runId}:agent_action`);
+    const rootId = (_a = this.rootRunIds.get(runId)) != null ? _a : runId;
+    this.rootRunIds.delete(runId);
+    if (rootId === runId) {
+      this.eventIds.delete(rootId);
+    }
+  }
+  async finalizeEventIfRoot(runId, patch) {
+    var _a;
+    const rootId = (_a = this.rootRunIds.get(runId)) != null ? _a : runId;
+    const isRoot = rootId === runId;
+    const eventId = this.eventIds.get(rootId);
+    if (!eventId) return;
+    if (isRoot) {
+      if (this.convoId) {
+        await this.eventShipper.patch(eventId, { userId: this.userId, convoId: this.convoId, isPending: true });
+      }
+      await this.eventShipper.finish(eventId, { userId: this.userId, ...patch });
+    } else if (patch) {
+      await this.eventShipper.patch(eventId, { userId: this.userId, convoId: this.convoId, ...patch, isPending: true });
+    }
+  }
+  // ---------------------------------------------------------------------------
+  // LLM events
+  // ---------------------------------------------------------------------------
+  /** Called at the start of an LLM run with string prompts. */
+  async handleLLMStart(llm, prompts, runId, parentRunId, _extraParams, tags, metadata, runName) {
+    try {
+      if (this.filterLangGraphInternals && this.spans.has(runId)) return;
+      const eventId = this.getEventId(runId, parentRunId);
+      const modelId = getName(llm, runName);
+      const span = this.traceShipper.startSpan({
+        name: "llm",
+        parent: this.getParent(parentRunId),
+        eventId,
+        attributes: [
+          attrString("ai.model.provider", modelId),
+          attrString("ai.operationId", "llm")
+        ]
+      });
+      this.spans.set(runId, span);
+      await this.eventShipper.patch(eventId, {
+        userId: this.userId,
+        convoId: this.convoId,
+        input: prompts.join("\n"),
+        properties: buildProperties(tags, metadata),
+        isPending: true
+      });
+    } catch (e) {
+    }
+  }
+  /** Called at the start of a chat model run with structured messages. */
+  async handleChatModelStart(llm, messages, runId, parentRunId, _extraParams, tags, metadata, runName) {
+    try {
+      if (this.filterLangGraphInternals && this.spans.has(runId)) return;
+      const eventId = this.getEventId(runId, parentRunId);
+      const modelId = getName(llm, runName);
+      const span = this.traceShipper.startSpan({
+        name: "llm",
+        parent: this.getParent(parentRunId),
+        eventId,
+        attributes: [
+          attrString("ai.model.provider", modelId),
+          attrString("ai.operationId", "chat")
+        ]
+      });
+      this.spans.set(runId, span);
+      const allMessages = messages.flat();
+      const userMessages = allMessages.filter((m) => {
+        var _a, _b;
+        const role = (_b = (_a = m._getType) == null ? void 0 : _a.call(m)) != null ? _b : "unknown";
+        return role === "human";
+      });
+      const toExtract = userMessages.length > 0 ? userMessages : allMessages.slice(-1);
+      const inputText = toExtract.map((m) => typeof m.content === "string" ? m.content : safeStringify(m.content)).join("\n");
+      await this.eventShipper.patch(eventId, {
+        userId: this.userId,
+        convoId: this.convoId,
+        input: inputText,
+        properties: buildProperties(tags, metadata),
+        isPending: true
+      });
+    } catch (e) {
+    }
+  }
+  async handleLLMEnd(output, runId, _parentRunId) {
+    try {
+      const span = this.spans.get(runId);
+      if (!span) {
+        await this.finalizeEventIfRoot(runId);
+        return;
+      }
+      const { model, promptTokens, completionTokens, outputText } = extractLLMMetadata(output);
+      this.traceShipper.endSpan(span, {
+        attributes: [
+          attrString("ai.response.model", model),
+          attrInt("ai.usage.prompt_tokens", promptTokens),
+          attrInt("ai.usage.completion_tokens", completionTokens)
+        ]
+      });
+      const properties = {};
+      if (promptTokens != null) properties["ai.usage.prompt_tokens"] = promptTokens;
+      if (completionTokens != null) properties["ai.usage.completion_tokens"] = completionTokens;
+      await this.finalizeEventIfRoot(runId, {
+        output: outputText,
+        model,
+        properties: Object.keys(properties).length > 0 ? properties : void 0
+      });
+    } catch (e) {
+    } finally {
+      this.cleanup(runId);
+    }
+  }
+  /** Called when an LLM run encounters an error. */
+  async handleLLMError(err, runId) {
+    try {
+      const span = this.spans.get(runId);
+      if (span) {
+        this.traceShipper.endSpan(span, { error: err });
+      }
+      await this.finalizeEventIfRoot(runId);
+    } catch (e) {
+    } finally {
+      this.cleanup(runId);
+    }
+  }
+  // ---------------------------------------------------------------------------
+  // Chain events
+  // ---------------------------------------------------------------------------
+  async handleChainStart(chain, _inputs, runId, parentRunId, _tags, _metadata, _runType, runName) {
+    var _a;
+    try {
+      const chainName = getName(chain, runName);
+      if (this.filterLangGraphInternals && isLangGraphInternal(chainName)) return;
+      this.getEventId(runId, parentRunId);
+      if (!this.traceChains) return;
+      const eventId = this.eventIds.get((_a = this.rootRunIds.get(runId)) != null ? _a : runId);
+      const span = this.traceShipper.startSpan({
+        name: "chain",
+        parent: this.getParent(parentRunId),
+        eventId,
+        attributes: [
+          attrString("chain.name", chainName),
+          attrString("ai.operationId", "chain")
+        ]
+      });
+      this.spans.set(runId, span);
+    } catch (e) {
+    }
+  }
+  async handleChainEnd(_outputs, runId) {
+    try {
+      if (this.traceChains) {
+        const span = this.spans.get(runId);
+        if (span) this.traceShipper.endSpan(span);
+      }
+      await this.finalizeEventIfRoot(runId);
+    } catch (e) {
+    } finally {
+      this.cleanup(runId);
+    }
+  }
+  async handleChainError(err, runId) {
+    try {
+      const span = this.spans.get(runId);
+      if (span) this.traceShipper.endSpan(span, { error: err });
+      await this.finalizeEventIfRoot(runId);
+    } catch (e) {
+    } finally {
+      this.cleanup(runId);
+    }
+  }
+  // ---------------------------------------------------------------------------
+  // Tool events
+  // ---------------------------------------------------------------------------
+  async handleToolStart(tool, input, runId, parentRunId, _tags, _metadata, runName) {
+    try {
+      const eventId = this.getEventId(runId, parentRunId);
+      const toolName = getName(tool, runName);
+      const span = this.traceShipper.startSpan({
+        name: "tool",
+        parent: this.getParent(parentRunId),
+        eventId,
+        attributes: [
+          attrString("tool.name", toolName),
+          attrString("tool.input", truncate(input, 4096))
+        ]
+      });
+      this.spans.set(runId, span);
+    } catch (e) {
+    }
+  }
+  /** Called at the end of a tool run. */
+  async handleToolEnd(output, runId) {
+    try {
+      const span = this.spans.get(runId);
+      if (span) {
+        const outputStr = typeof output === "string" ? output : safeStringify(output);
+        this.traceShipper.endSpan(span, {
+          attributes: [attrString("tool.output", truncate(outputStr, 4096))]
+        });
+      }
+      await this.finalizeEventIfRoot(runId);
+    } catch (e) {
+    } finally {
+      this.cleanup(runId);
+    }
+  }
+  /** Called when a tool run encounters an error. */
+  async handleToolError(err, runId) {
+    try {
+      const span = this.spans.get(runId);
+      if (span) this.traceShipper.endSpan(span, { error: err });
+      await this.finalizeEventIfRoot(runId);
+    } catch (e) {
+    } finally {
+      this.cleanup(runId);
+    }
+  }
+  // ---------------------------------------------------------------------------
+  // Retriever events
+  // ---------------------------------------------------------------------------
+  /** Called at the start of a retriever run. */
+  async handleRetrieverStart(retriever, query, runId, parentRunId, _tags, _metadata, name) {
+    try {
+      const eventId = this.getEventId(runId, parentRunId);
+      if (!this.traceRetrievers) return;
+      const retrieverName = getName(retriever, name);
+      const span = this.traceShipper.startSpan({
+        name: "retriever",
+        parent: this.getParent(parentRunId),
+        eventId,
+        attributes: [
+          attrString("retriever.name", retrieverName),
+          attrString("retriever.query", truncate(query, 2048))
+        ]
+      });
+      this.spans.set(runId, span);
+    } catch (e) {
+    }
+  }
+  /** Called at the end of a retriever run. */
+  async handleRetrieverEnd(documents, runId) {
+    try {
+      if (this.traceRetrievers) {
+        const span = this.spans.get(runId);
+        if (span) {
+          this.traceShipper.endSpan(span, {
+            attributes: [attrInt("retriever.document_count", documents.length)]
+          });
+        }
+      }
+      await this.finalizeEventIfRoot(runId);
+    } catch (e) {
+    } finally {
+      this.cleanup(runId);
+    }
+  }
+  /** Called when a retriever run encounters an error. */
+  async handleRetrieverError(err, runId) {
+    try {
+      const span = this.spans.get(runId);
+      if (span) this.traceShipper.endSpan(span, { error: err });
+      await this.finalizeEventIfRoot(runId);
+    } catch (e) {
+    } finally {
+      this.cleanup(runId);
+    }
+  }
+  // ---------------------------------------------------------------------------
+  // Agent events
+  // ---------------------------------------------------------------------------
+  async handleAgentAction(action, runId, parentRunId) {
+    try {
+      const eventId = this.getEventId(runId, parentRunId);
+      const span = this.traceShipper.startSpan({
+        name: "agent_action",
+        parent: this.getParent(parentRunId),
+        eventId,
+        attributes: [
+          attrString("agent.tool", action.tool),
+          attrString("agent.tool_input", truncate(
+            typeof action.toolInput === "string" ? action.toolInput : safeStringify(action.toolInput),
+            4096
+          ))
+        ]
+      });
+      this.spans.set(`${runId}:agent_action`, span);
+    } catch (e) {
+    }
+  }
+  async handleAgentEnd(_action, runId) {
+    try {
+      const spanKey = `${runId}:agent_action`;
+      const span = this.spans.get(spanKey);
+      if (span) {
+        this.traceShipper.endSpan(span);
+        this.spans.delete(spanKey);
+      }
+    } catch (e) {
+    }
+  }
+};
+function getName(serialized, runName) {
+  var _a;
+  if (runName) return runName;
+  if ((_a = serialized == null ? void 0 : serialized.id) == null ? void 0 : _a.length) return serialized.id[serialized.id.length - 1];
+  if ("name" in serialized && typeof serialized.name === "string") return serialized.name;
+  return "unknown";
+}
+function isLangGraphInternal(name) {
+  if (LANGGRAPH_INTERNAL_NAMES.has(name)) return true;
+  if (name.startsWith("Branch")) return true;
+  if (name.startsWith("ChannelWrite")) return true;
+  if (name.startsWith("__") && name.endsWith("__")) return true;
+  return false;
+}
+function buildProperties(tags, metadata) {
+  if (!(tags == null ? void 0 : tags.length) && !metadata) return void 0;
+  const props = {};
+  if (tags == null ? void 0 : tags.length) props["langchain.tags"] = tags;
+  if (metadata) Object.assign(props, metadata);
+  return props;
+}
+function safeStringify(value) {
+  var _a;
+  try {
+    return (_a = JSON.stringify(value)) != null ? _a : "";
+  } catch (e) {
+    return String(value);
+  }
+}
+function truncate(str, maxLen) {
+  if (str.length <= maxLen) return str;
+  return str.slice(0, maxLen - 3) + "...";
+}
+function extractLLMMetadata(output) {
+  var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m;
+  const gen = (_b = (_a = output.generations) == null ? void 0 : _a[0]) == null ? void 0 : _b[0];
+  const message = gen == null ? void 0 : gen.message;
+  const responseMeta = message == null ? void 0 : message.response_metadata;
+  const usageMeta = message == null ? void 0 : message.usage_metadata;
+  const llmOutput = output.llmOutput;
+  const tokenUsage = (_c = llmOutput == null ? void 0 : llmOutput.token_usage) != null ? _c : llmOutput == null ? void 0 : llmOutput.tokenUsage;
+  const estimatedTokens = llmOutput == null ? void 0 : llmOutput.estimatedTokens;
+  const model = (_f = (_e = (_d = responseMeta == null ? void 0 : responseMeta.model_name) != null ? _d : responseMeta == null ? void 0 : responseMeta.model) != null ? _e : llmOutput == null ? void 0 : llmOutput.model_name) != null ? _f : llmOutput == null ? void 0 : llmOutput.model;
+  const usage = (_g = usageMeta != null ? usageMeta : tokenUsage) != null ? _g : estimatedTokens;
+  const promptTokens = (_i = (_h = usage == null ? void 0 : usage.input_tokens) != null ? _h : usage == null ? void 0 : usage.prompt_tokens) != null ? _i : usage == null ? void 0 : usage.promptTokens;
+  const completionTokens = (_k = (_j = usage == null ? void 0 : usage.output_tokens) != null ? _j : usage == null ? void 0 : usage.completion_tokens) != null ? _k : usage == null ? void 0 : usage.completionTokens;
+  const outputText = (_m = (_l = gen == null ? void 0 : gen.text) != null ? _l : typeof (message == null ? void 0 : message.content) === "string" ? message.content : void 0) != null ? _m : (message == null ? void 0 : message.content) ? JSON.stringify(message.content) : void 0;
+  return { model, promptTokens, completionTokens, outputText };
+}
+
+// package.json
+var package_default = {
+  name: "@raindrop-ai/langchain",
+  version: "0.0.1",
+  description: "Raindrop integration for LangChain",
+  main: "dist/index.js",
+  module: "dist/index.mjs",
+  types: "dist/index.d.ts",
+  exports: {
+    ".": {
+      types: "./dist/index.d.ts",
+      import: "./dist/index.mjs",
+      require: "./dist/index.js"
+    }
+  },
+  sideEffects: false,
+  files: ["dist/**"],
+  scripts: {
+    build: "tsup",
+    dev: "tsup --watch",
+    clean: "rm -rf dist",
+    test: "vitest run"
+  },
+  peerDependencies: {
+    "@langchain/core": ">=0.3.0 <1.0.0"
+  },
+  devDependencies: {
+    "@raindrop-ai/core": "workspace:*",
+    "@langchain/core": ">=0.3.0",
+    "@types/node": "^20.11.17",
+    msw: "^2.12.7",
+    tsup: "^8.4.0",
+    typescript: "^5.3.3",
+    vitest: "^2.1.9"
+  },
+  tsup: {
+    entry: ["src/index.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;
+
+// src/index.ts
+function createRaindropLangChain(opts) {
+  var _a, _b;
+  const writeKey = opts.writeKey;
+  const enabled = !!writeKey;
+  if (!writeKey) {
+    console.warn("[raindrop-ai/langchain] writeKey not provided; telemetry shipping is disabled");
+  }
+  const eventShipper = new EventShipper({
+    writeKey,
+    endpoint: opts.endpoint,
+    enabled,
+    debug: (_a = opts.debug) != null ? _a : false,
+    sdkName: "langchain",
+    libraryName,
+    libraryVersion
+  });
+  const traceShipper = new TraceShipper({
+    writeKey,
+    endpoint: opts.endpoint,
+    enabled,
+    debug: (_b = opts.debug) != null ? _b : false,
+    sdkName: "langchain",
+    serviceName: "raindrop.langchain",
+    serviceVersion: libraryVersion
+  });
+  const handler = new RaindropCallbackHandler({
+    eventShipper,
+    traceShipper,
+    userId: opts.userId,
+    convoId: opts.convoId,
+    traceChains: opts.traceChains,
+    traceRetrievers: opts.traceRetrievers,
+    filterLangGraphInternals: opts.filterLangGraphInternals
+  });
+  return {
+    handler,
+    async flush() {
+      await Promise.all([eventShipper.flush(), traceShipper.flush()]);
+    },
+    async shutdown() {
+      await Promise.all([eventShipper.shutdown(), traceShipper.shutdown()]);
+    },
+    events: {
+      async patch(eventId, patch) {
+        await eventShipper.patch(eventId, patch);
+      },
+      async finish(eventId, patch) {
+        await eventShipper.finish(eventId, patch);
+      },
+      async addAttachments(eventId, attachments) {
+        await eventShipper.patch(eventId, { attachments });
+      },
+      async setProperties(eventId, properties) {
+        await eventShipper.patch(eventId, { properties });
+      }
+    },
+    users: {
+      async identify(users) {
+        await eventShipper.identify(users);
+      }
+    },
+    signals: {
+      async track(signal) {
+        await eventShipper.trackSignal(signal);
+      }
+    }
+  };
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  RaindropCallbackHandler,
+  createRaindropLangChain
+});