@raindrop-ai/ai-sdk 0.0.19-beta.4 → 0.0.20

package/README.md

@@ -58,7 +58,20 @@ If `userId` is missing from both `wrap()` context and `eventMetadata()`, the SDK
 
 ## Runtime support
 
-### Node.js (recommended)
+### Browsers and edge runtimes
+
+Use the browser entrypoint:
+
+```ts
+import { createRaindropAISDK } from "@raindrop-ai/ai-sdk/browser";
+```
+
+The SDK works without `async_hooks` shims. When `AsyncLocalStorage` is not available,
+Raindrop falls back to synchronous context scoping: nested work in the same call stack
+still inherits context, but automatic propagation across arbitrary async boundaries is
+not guaranteed.
+
+### Node.js
 
 Use the default import:
 
@@ -66,17 +79,23 @@ Use the default import:
 import { createRaindropAISDK } from "@raindrop-ai/ai-sdk";
 ```
 
+In Node, the default entrypoint wires up `AsyncLocalStorage` automatically.
+
 ### Cloudflare Workers
 
 Cloudflare Workers can provide `AsyncLocalStorage` via `node:async_hooks` when `nodejs_compat` is enabled ([docs](https://developers.cloudflare.com/workers/runtime-apis/nodejs/asynclocalstorage/)).
 
-Use the Workers entrypoint:
+If `nodejs_compat` is enabled, use the Workers entrypoint:
 
 ```ts
 import { createRaindropAISDK } from "@raindrop-ai/ai-sdk/workers";
 ```
 
-If `nodejs_compat` is not enabled, AsyncLocalStorage-based context propagation cannot work.
+This enables real `AsyncLocalStorage` propagation in Workers.
+
+If `nodejs_compat` is not enabled, use the browser entrypoint instead. The SDK still works,
+but it uses the same synchronous fallback described above rather than real
+`AsyncLocalStorage`.
 
 ## Supported AI SDK Versions
 

package/dist/{chunk-MDH47MA5.mjs → chunk-FD5GVIE2.mjs}

@@ -1,4 +1,6 @@
-// ../core/dist/chunk-FOHDGBT5.js
+import { AsyncLocalStorage } from 'async_hooks';
+
+// ../core/dist/chunk-H6VSZSLN.js
 function getCrypto() {
   const c = globalThis.crypto;
   return c;
@@ -616,49 +618,25 @@ var NOOP_SPAN = {
   log() {
   }
 };
-function isPromiseLike(value) {
-  return value !== null && (typeof value === "object" || typeof value === "function") && "then" in value && typeof value.then === "function";
+function getAsyncLocalStorageCtor() {
+  return globalThis.RAINDROP_ASYNC_LOCAL_STORAGE;
 }
-var PendingAsyncLocalStorage = class {
+var SynchronousContextStorage = class {
   constructor() {
-    this._real = null;
     this._stack = [];
   }
-  setReal(real) {
-    const current = this._stack.length ? this._stack[this._stack.length - 1] : void 0;
-    this._real = real;
-    if (current !== void 0 && typeof this._real.enterWith === "function") {
-      this._real.enterWith(current);
-    }
+  isEmpty() {
+    return this._stack.length === 0;
   }
   getStore() {
-    var _a, _b;
-    return (_b = (_a = this._real) == null ? void 0 : _a.getStore()) != null ? _b : this._stack[this._stack.length - 1];
+    return this._stack[this._stack.length - 1];
   }
   run(store, callback) {
-    if (this._real) {
-      return this._real.run(store, callback);
-    }
     this._stack.push(store);
     try {
-      const result = callback();
-      if (isPromiseLike(result)) {
-        return result.then(
-          (value) => {
-            this._stack.pop();
-            return value;
-          },
-          (err) => {
-            this._stack.pop();
-            throw err;
-          }
-        );
-      }
-      this._stack.pop();
-      return result;
-    } catch (err) {
+      return callback();
+    } finally {
       this._stack.pop();
-      throw err;
     }
   }
 };
@@ -666,53 +644,29 @@ var ContextManager = class {
 };
 var RaindropContextManager = class extends ContextManager {
   constructor() {
-    var _a;
     super();
-    this._storage = null;
-    this._pending = null;
-    this._initPromise = null;
-    const isNode = typeof process !== "undefined" && typeof ((_a = process == null ? void 0 : process.versions) == null ? void 0 : _a.node) === "string";
-    const Ctor = globalThis.RAINDROP_ASYNC_LOCAL_STORAGE;
+    this._fallback = null;
+    const Ctor = getAsyncLocalStorageCtor();
     if (Ctor) {
       this._storage = new Ctor();
-      this._pending = null;
-      this._initPromise = null;
       return;
     }
-    if (isNode) {
-      this._pending = new PendingAsyncLocalStorage();
-      this._storage = this._pending;
-      this._initPromise = this._initialize();
-    } else {
-      this._storage = null;
-      this._pending = null;
-      this._initPromise = null;
-    }
+    this._fallback = new SynchronousContextStorage();
+    this._storage = this._fallback;
   }
-  async _initialize() {
-    try {
-      const mod = await import('async_hooks');
-      const real = new mod.AsyncLocalStorage();
-      if (this._pending) {
-        this._pending.setReal(real);
-      }
-      this._storage = real;
-    } catch (e) {
-      this._storage = null;
-    }
-  }
-  async ensureReady() {
-    if (this._initPromise) {
-      await this._initPromise;
-      this._initPromise = null;
-    }
+  maybeAdoptAsyncLocalStorage() {
+    if (!this._fallback || !this._fallback.isEmpty()) return;
+    const Ctor = getAsyncLocalStorageCtor();
+    if (!Ctor) return;
+    this._storage = new Ctor();
+    this._fallback = null;
   }
   isReady() {
-    return this._initPromise === null;
+    return true;
   }
   getParentSpanIds() {
-    var _a;
-    const span = (_a = this._storage) == null ? void 0 : _a.getStore();
+    this.maybeAdoptAsyncLocalStorage();
+    const span = this._storage.getStore();
     if (!span || span === NOOP_SPAN) return void 0;
     return {
       traceIdB64: span.traceIdB64,
@@ -721,12 +675,12 @@ var RaindropContextManager = class extends ContextManager {
     };
   }
   runInContext(span, callback) {
-    if (!this._storage) return callback();
+    this.maybeAdoptAsyncLocalStorage();
     return this._storage.run(span, callback);
   }
   getCurrentSpan() {
-    var _a;
-    return (_a = this._storage) == null ? void 0 : _a.getStore();
+    this.maybeAdoptAsyncLocalStorage();
+    return this._storage.getStore();
   }
 };
 var _contextManager = null;
@@ -744,17 +698,10 @@ function withCurrent(span, callback) {
   return getContextManager().runInContext(span, callback);
 }
 async function getCurrentParentSpanContext() {
-  const cm = getContextManager();
-  if (cm instanceof RaindropContextManager) {
-    await cm.ensureReady();
-  }
-  return cm.getParentSpanIds();
+  return getContextManager().getParentSpanIds();
 }
 async function runWithParentSpanContext(ctx, fn) {
   const cm = getContextManager();
-  if (cm instanceof RaindropContextManager) {
-    await cm.ensureReady();
-  }
   const span = {
     traceIdB64: ctx.traceIdB64,
     spanIdB64: ctx.spanIdB64,
@@ -781,28 +728,7 @@ async function* asyncGeneratorWithCurrent(span, gen) {
     nextValue = yield result.value;
   }
 }
-
-// package.json
-var package_default = {
-  name: "@raindrop-ai/ai-sdk",
-  version: "0.0.19-beta.4"};
-
-// src/internal/version.ts
-var libraryName = package_default.name;
-var libraryVersion = package_default.version;
-
-// src/internal/events.ts
-var EventShipper2 = class extends EventShipper {
-  constructor(opts) {
-    var _a, _b, _c;
-    super({
-      ...opts,
-      sdkName: (_a = opts.sdkName) != null ? _a : "ai-sdk",
-      libraryName: (_b = opts.libraryName) != null ? _b : libraryName,
-      libraryVersion: (_c = opts.libraryVersion) != null ? _c : libraryVersion
-    });
-  }
-};
+globalThis.RAINDROP_ASYNC_LOCAL_STORAGE = AsyncLocalStorage;
 
 // src/internal/wrap/helpers.ts
 function isRecord(value) {
@@ -1057,17 +983,29 @@ function extractTextFromMessageContent(content) {
   }
   return result.length ? result : void 0;
 }
-function extractInputAttachmentsFromArgs(args) {
-  if (!isRecord(args)) return void 0;
+function messagesFromArgs(args) {
   const messages = args["messages"];
-  if (!Array.isArray(messages)) return void 0;
+  if (Array.isArray(messages)) return messages;
+  const prompt = args["prompt"];
+  if (Array.isArray(prompt)) return prompt;
+  return void 0;
+}
+function lastUserMessageFromArgs(args) {
+  const messages = messagesFromArgs(args);
+  if (!messages) return void 0;
   for (let i = messages.length - 1; i >= 0; i--) {
     const message = messages[i];
-    if (!isRecord(message) || message["role"] !== "user") continue;
-    return attachmentsFromContent(message["content"], "input");
+    if (isRecord(message) && message["role"] === "user") {
+      return message;
+    }
   }
   return void 0;
 }
+function extractInputAttachmentsFromArgs(args) {
+  var _a;
+  if (!isRecord(args)) return void 0;
+  return attachmentsFromContent((_a = lastUserMessageFromArgs(args)) == null ? void 0 : _a["content"], "input");
+}
 async function extractOutputAttachmentsFromResult(result) {
   if (!isRecord(result)) return void 0;
   const fileAttachments = await outputAttachmentsFromFiles(result["files"]);
@@ -1081,26 +1019,20 @@ async function extractOutputAttachmentsFromResult(result) {
   return attachmentsFromContent(result["content"], "output");
 }
 function lastUserMessageTextFromArgs(args) {
-  var _a;
+  var _a, _b;
   if (!isRecord(args)) return void 0;
-  const messages = args["messages"];
-  if (!Array.isArray(messages)) return void 0;
-  for (let i = messages.length - 1; i >= 0; i--) {
-    const message = messages[i];
-    if (!isRecord(message) || message["role"] !== "user") continue;
-    const content = message["content"];
-    const text = extractTextFromMessageContent(content);
-    if (text !== void 0) return text;
-    return (_a = safeJsonWithUint8(content)) != null ? _a : String(content);
-  }
-  return void 0;
+  const content = (_a = lastUserMessageFromArgs(args)) == null ? void 0 : _a["content"];
+  if (content === void 0) return void 0;
+  const text = extractTextFromMessageContent(content);
+  if (text !== void 0) return text;
+  return (_b = safeJsonWithUint8(content)) != null ? _b : String(content);
 }
 function extractInputFromArgs(args) {
   var _a;
   if (!isRecord(args)) return void 0;
   const prompt = args["prompt"];
   if (typeof prompt === "string") return prompt;
-  const messages = args["messages"];
+  const messages = messagesFromArgs(args);
   if (Array.isArray(messages) && messages.length > 0) {
     const last = messages[messages.length - 1];
     if (isRecord(last)) {
@@ -1122,7 +1054,7 @@ function coerceMessagesFromArgs(args) {
   if (typeof args["system"] === "string" && args["system"]) {
     result.push({ role: "system", content: args["system"] });
   }
-  const messages = args["messages"];
+  const messages = messagesFromArgs(args);
   if (Array.isArray(messages)) {
     for (const message of messages) {
       if (isRecord(message) && typeof message["role"] === "string") {
@@ -1949,19 +1881,6 @@ var RaindropTelemetryIntegration = class {
   }
 };
 
-// src/internal/traces.ts
-var TraceShipper2 = class extends TraceShipper {
-  constructor(opts) {
-    var _a, _b, _c;
-    super({
-      ...opts,
-      sdkName: (_a = opts.sdkName) != null ? _a : "ai-sdk",
-      serviceName: (_b = opts.serviceName) != null ? _b : "raindrop.ai-sdk",
-      serviceVersion: (_c = opts.serviceVersion) != null ? _c : libraryVersion
-    });
-  }
-};
-
 // src/internal/wrap/wrapAISDK.ts
 var AGENT_REPORTING_TOOL_NAME_DEFAULT = "__raindrop_report";
 var AGENT_REPORTING_SIGNALS_DEFAULT = {
@@ -3867,6 +3786,41 @@ function extractNestedTokens(usage, key) {
   return void 0;
 }
 
+// package.json
+var package_default = {
+  name: "@raindrop-ai/ai-sdk",
+  version: "0.0.20"};
+
+// src/internal/version.ts
+var libraryName = package_default.name;
+var libraryVersion = package_default.version;
+
+// src/internal/events.ts
+var EventShipper2 = class extends EventShipper {
+  constructor(opts) {
+    var _a, _b, _c;
+    super({
+      ...opts,
+      sdkName: (_a = opts.sdkName) != null ? _a : "ai-sdk",
+      libraryName: (_b = opts.libraryName) != null ? _b : libraryName,
+      libraryVersion: (_c = opts.libraryVersion) != null ? _c : libraryVersion
+    });
+  }
+};
+
+// src/internal/traces.ts
+var TraceShipper2 = class extends TraceShipper {
+  constructor(opts) {
+    var _a, _b, _c;
+    super({
+      ...opts,
+      sdkName: (_a = opts.sdkName) != null ? _a : "ai-sdk",
+      serviceName: (_b = opts.serviceName) != null ? _b : "raindrop.ai-sdk",
+      serviceVersion: (_c = opts.serviceVersion) != null ? _c : libraryVersion
+    });
+  }
+};
+
 // src/index.ts
 function eventMetadata(options) {
   const result = {};