@raindrop-ai/ai-sdk 0.0.2 → 0.0.4

package/dist/index.js

@@ -86,7 +86,7 @@ async function postJson(url, body, headers, opts) {
 // package.json
 var package_default = {
   name: "@raindrop-ai/ai-sdk",
-  version: "0.0.2"};
+  version: "0.0.4"};
 
 // src/internal/version.ts
 var libraryName = package_default.name;
@@ -133,19 +133,30 @@ var EventShipper = class {
     return this.debug;
   }
   async patch(eventId, patch) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k;
     if (!this.enabled) return;
     if (!eventId) return;
-    const sticky = (_a = this.sticky.get(eventId)) != null ? _a : {};
-    const existing = (_b = this.buffers.get(eventId)) != null ? _b : {};
+    if (this.debug) {
+      console.log("[raindrop-ai/ai-sdk] queue patch", {
+        eventId,
+        userId: patch.userId,
+        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 = (_e = (_d = (_c = patch.isPending) != null ? _c : existing.isPending) != null ? _d : sticky.isPending) != null ? _e : true;
+    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: (_f = merged.userId) != null ? _f : sticky.userId,
-      convoId: (_g = merged.convoId) != null ? _g : sticky.convoId,
-      eventName: (_h = merged.eventName) != null ? _h : sticky.eventName,
-      isPending: (_i = merged.isPending) != null ? _i : sticky.isPending
+      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);
@@ -242,7 +253,7 @@ var EventShipper = class {
     }
   }
   async flushOne(eventId) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i;
+    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) {
@@ -280,6 +291,24 @@ var EventShipper = class {
       is_pending: ((_i = (_h = accumulated.isPending) != null ? _h : sticky.isPending) != null ? _i : true) !== false
     };
     const url = `${this.baseUrl}events/track_partial`;
+    if (this.debug) {
+      console.log("[raindrop-ai/ai-sdk] sending track_partial", {
+        eventId,
+        eventName,
+        userId,
+        isPending: payload.is_pending,
+        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,
@@ -522,6 +551,12 @@ var TraceShipper = class {
       const batch = this.queue.splice(0, this.maxBatchSize);
       const body = buildExportTraceServiceRequest(batch);
       const url = `${this.baseUrl}traces`;
+      if (this.debug) {
+        console.log("[raindrop-ai/ai-sdk] sending traces batch", {
+          spans: batch.length,
+          endpoint: url
+        });
+      }
       const p = postJson(
         url,
         body,
@@ -849,7 +884,13 @@ function isAgentClass(value) {
 function extractModel(result) {
   if (!isRecord(result)) return void 0;
   const model = result["model"];
-  return typeof model === "string" && model.length ? model : void 0;
+  if (typeof model === "string" && model.length) return model;
+  const response = result["response"];
+  if (isRecord(response)) {
+    const modelId = response["modelId"];
+    if (typeof modelId === "string" && modelId.length) return modelId;
+  }
+  return void 0;
 }
 function extractFinishReason(result) {
   if (!isRecord(result)) return void 0;
@@ -859,6 +900,174 @@ function extractFinishReason(result) {
   }
   return void 0;
 }
+function bytesToBase64(bytes) {
+  if (typeof Buffer !== "undefined") return Buffer.from(bytes).toString("base64");
+  let binary = "";
+  for (let i = 0; i < bytes.length; i++) {
+    binary += String.fromCharCode(bytes[i]);
+  }
+  if (typeof btoa === "function") return btoa(binary);
+  return "";
+}
+function asDataUrl(value, mediaType) {
+  if (value instanceof URL) return value.toString();
+  if (typeof value === "string") {
+    if (value.startsWith("data:")) return value;
+    if (value.startsWith("http://") || value.startsWith("https://")) return value;
+    return `data:${mediaType};base64,${value}`;
+  }
+  if (value instanceof Uint8Array) {
+    const base64 = bytesToBase64(value);
+    if (!base64) return void 0;
+    return `data:${mediaType};base64,${base64}`;
+  }
+  if (value instanceof ArrayBuffer) {
+    const base64 = bytesToBase64(new Uint8Array(value));
+    if (!base64) return void 0;
+    return `data:${mediaType};base64,${base64}`;
+  }
+  return void 0;
+}
+function dataPartToAttachmentValue(value) {
+  if (typeof value === "string") return value;
+  if (value instanceof URL) return value.toString();
+  if (value instanceof Uint8Array) return `[binary:${value.byteLength} bytes]`;
+  if (value instanceof ArrayBuffer) return `[binary:${value.byteLength} bytes]`;
+  return void 0;
+}
+function attachmentMediaType(part) {
+  if (typeof part["mediaType"] === "string") return part["mediaType"];
+  if (typeof part["mimeType"] === "string") return part["mimeType"];
+  const file = part["file"];
+  if (isRecord(file)) {
+    if (typeof file["mediaType"] === "string") return file["mediaType"];
+    if (typeof file["mimeType"] === "string") return file["mimeType"];
+  }
+  return void 0;
+}
+function attachmentName(part) {
+  if (typeof part["filename"] === "string") return part["filename"];
+  if (typeof part["name"] === "string") return part["name"];
+  const file = part["file"];
+  if (isRecord(file)) {
+    if (typeof file["filename"] === "string") return file["filename"];
+    if (typeof file["name"] === "string") return file["name"];
+  }
+  return void 0;
+}
+function attachmentData(part) {
+  if ("data" in part) return part["data"];
+  const file = part["file"];
+  if (isRecord(file)) {
+    if ("file_data" in file) return file["file_data"];
+    if ("data" in file) return file["data"];
+  }
+  return void 0;
+}
+function contentPartToAttachment(part, role) {
+  var _a, _b, _c;
+  const partType = part["type"];
+  if (typeof partType !== "string") return void 0;
+  if (partType === "image") {
+    const mediaType = (_a = attachmentMediaType(part)) != null ? _a : "image/png";
+    const value = asDataUrl(part["image"], mediaType);
+    if (!value) return void 0;
+    return { type: "image", role, value };
+  }
+  if (partType === "image_url") {
+    const imageUrlPart = part["image_url"];
+    const imageUrlValue = isRecord(imageUrlPart) ? imageUrlPart["url"] : imageUrlPart;
+    const value = asDataUrl(imageUrlValue, "image/png");
+    if (!value) return void 0;
+    return { type: "image", role, value };
+  }
+  if (partType === "file") {
+    const mediaType = attachmentMediaType(part);
+    const data = attachmentData(part);
+    const isImage = (mediaType == null ? void 0 : mediaType.startsWith("image/")) === true;
+    const value = isImage && mediaType ? asDataUrl(data, mediaType) : dataPartToAttachmentValue(data);
+    if (!value) return void 0;
+    const name = (_c = (_b = attachmentName(part)) != null ? _b : mediaType) != null ? _c : "file";
+    return { type: isImage ? "image" : "text", role, name, value };
+  }
+  return void 0;
+}
+function attachmentsFromContent(content, role) {
+  if (!Array.isArray(content)) return void 0;
+  const attachments = [];
+  for (const part of content) {
+    if (!isRecord(part)) continue;
+    const attachment = contentPartToAttachment(part, role);
+    if (attachment) attachments.push(attachment);
+  }
+  return attachments.length ? attachments : void 0;
+}
+function generatedFileToAttachment(file) {
+  var _a, _b, _c, _d;
+  const mediaType = typeof file["mediaType"] === "string" ? file["mediaType"] : typeof file["mimeType"] === "string" ? file["mimeType"] : void 0;
+  const data = (_d = (_c = (_b = (_a = file["base64Data"]) != null ? _a : file["base64"]) != null ? _b : file["uint8ArrayData"]) != null ? _c : file["uint8Array"]) != null ? _d : file["data"];
+  const isImage = (mediaType == null ? void 0 : mediaType.startsWith("image/")) === true;
+  const value = isImage && mediaType ? asDataUrl(data, mediaType) : dataPartToAttachmentValue(data);
+  if (!value) return void 0;
+  const name = typeof file["filename"] === "string" ? file["filename"] : typeof file["name"] === "string" ? file["name"] : mediaType != null ? mediaType : "file";
+  return {
+    type: isImage ? "image" : "text",
+    role: "output",
+    name,
+    value
+  };
+}
+async function outputAttachmentsFromFiles(files) {
+  let resolvedFiles = files;
+  if (resolvedFiles && (typeof resolvedFiles === "object" || typeof resolvedFiles === "function") && typeof resolvedFiles.then === "function") {
+    try {
+      resolvedFiles = await resolvedFiles;
+    } catch (e) {
+      return void 0;
+    }
+  }
+  if (!Array.isArray(resolvedFiles)) return void 0;
+  const attachments = [];
+  for (const file of resolvedFiles) {
+    if (!isRecord(file)) continue;
+    const attachment = generatedFileToAttachment(file);
+    if (attachment) attachments.push(attachment);
+  }
+  return attachments.length ? attachments : void 0;
+}
+function extractTextFromMessageContent(content) {
+  if (typeof content === "string") return content;
+  if (!Array.isArray(content)) return void 0;
+  let result = "";
+  for (const part of content) {
+    if (!isRecord(part) || part["type"] !== "text" || typeof part["text"] !== "string") continue;
+    result += part["text"];
+  }
+  return result.length ? result : void 0;
+}
+function extractInputAttachmentsFromArgs(args) {
+  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;
+    return attachmentsFromContent(message["content"], "input");
+  }
+  return void 0;
+}
+async function extractOutputAttachmentsFromResult(result) {
+  if (!isRecord(result)) return void 0;
+  const fileAttachments = await outputAttachmentsFromFiles(result["files"]);
+  if (fileAttachments == null ? void 0 : fileAttachments.length) return fileAttachments;
+  const responseMessages = extractResponseMessages(result);
+  for (let i = responseMessages.length - 1; i >= 0; i--) {
+    const message = responseMessages[i];
+    if (!isRecord(message) || message["role"] !== "assistant") continue;
+    return attachmentsFromContent(message["content"], "output");
+  }
+  return attachmentsFromContent(result["content"], "output");
+}
 function lastUserMessageTextFromArgs(args) {
   var _a;
   if (!isRecord(args)) return void 0;
@@ -868,7 +1077,8 @@ function lastUserMessageTextFromArgs(args) {
     const message = messages[i];
     if (!isRecord(message) || message["role"] !== "user") continue;
     const content = message["content"];
-    if (typeof content === "string") return content;
+    const text = extractTextFromMessageContent(content);
+    if (text !== void 0) return text;
     return (_a = safeJsonWithUint8(content)) != null ? _a : String(content);
   }
   return void 0;
@@ -883,7 +1093,8 @@ function extractInputFromArgs(args) {
     const last = messages[messages.length - 1];
     if (isRecord(last)) {
       const content = last["content"];
-      if (typeof content === "string") return content;
+      const text = extractTextFromMessageContent(content);
+      if (text !== void 0) return text;
       const asJson = safeJson(content);
       if (asJson) return asJson;
     }
@@ -917,7 +1128,9 @@ function extractResponseMessages(result) {
   if (!isRecord(result)) return [];
   const response = result["response"];
   if (isRecord(response) && Array.isArray(response["messages"])) {
-    return response["messages"].filter((message) => isRecord(message) && typeof message["role"] === "string" && "content" in message).map((message) => message);
+    return response["messages"].filter(
+      (message) => isRecord(message) && typeof message["role"] === "string" && "content" in message
+    ).map((message) => message);
   }
   const steps = result["steps"];
   if (Array.isArray(steps) && steps.length > 0) {
@@ -925,7 +1138,9 @@ function extractResponseMessages(result) {
     if (isRecord(lastStep) && isRecord(lastStep["response"])) {
       const responseMessages = lastStep["response"]["messages"];
       if (Array.isArray(responseMessages)) {
-        return responseMessages.filter((message) => isRecord(message) && typeof message["role"] === "string" && "content" in message).map((message) => message);
+        return responseMessages.filter(
+          (message) => isRecord(message) && typeof message["role"] === "string" && "content" in message
+        ).map((message) => message);
       }
     }
   }
@@ -975,9 +1190,11 @@ function opName(operationId, functionId) {
 function toOtlpAttr(key, value) {
   if (value === void 0 || value === null) return void 0;
   if (typeof value === "string") return attrString(key, value);
-  if (typeof value === "number") return Number.isInteger(value) ? attrInt(key, value) : attrDouble(key, value);
+  if (typeof value === "number")
+    return Number.isInteger(value) ? attrInt(key, value) : attrDouble(key, value);
   if (typeof value === "boolean") return attrBool(key, value);
-  if (Array.isArray(value) && value.every((v) => typeof v === "string")) return attrStringArray(key, value);
+  if (Array.isArray(value) && value.every((v) => typeof v === "string"))
+    return attrStringArray(key, value);
   const asJson = safeJsonWithUint8(value);
   return asJson ? attrString(key, asJson) : void 0;
 }
@@ -1023,13 +1240,31 @@ function attrsFromSettings(args) {
 function attrsFromGenAiRequest(options) {
   if (!isRecord(options)) return [];
   return [
-    attrDouble("gen_ai.request.frequency_penalty", typeof options["frequencyPenalty"] === "number" ? options["frequencyPenalty"] : void 0),
-    attrInt("gen_ai.request.max_tokens", typeof options["maxOutputTokens"] === "number" ? options["maxOutputTokens"] : void 0),
-    attrDouble("gen_ai.request.presence_penalty", typeof options["presencePenalty"] === "number" ? options["presencePenalty"] : void 0),
+    attrDouble(
+      "gen_ai.request.frequency_penalty",
+      typeof options["frequencyPenalty"] === "number" ? options["frequencyPenalty"] : void 0
+    ),
+    attrInt(
+      "gen_ai.request.max_tokens",
+      typeof options["maxOutputTokens"] === "number" ? options["maxOutputTokens"] : void 0
+    ),
+    attrDouble(
+      "gen_ai.request.presence_penalty",
+      typeof options["presencePenalty"] === "number" ? options["presencePenalty"] : void 0
+    ),
     ...Array.isArray(options["stopSequences"]) && options["stopSequences"].every((x) => typeof x === "string") ? [attrStringArray("gen_ai.request.stop_sequences", options["stopSequences"])] : [],
-    attrDouble("gen_ai.request.temperature", typeof options["temperature"] === "number" ? options["temperature"] : void 0),
-    attrInt("gen_ai.request.top_k", typeof options["topK"] === "number" ? options["topK"] : void 0),
-    attrDouble("gen_ai.request.top_p", typeof options["topP"] === "number" ? options["topP"] : void 0)
+    attrDouble(
+      "gen_ai.request.temperature",
+      typeof options["temperature"] === "number" ? options["temperature"] : void 0
+    ),
+    attrInt(
+      "gen_ai.request.top_k",
+      typeof options["topK"] === "number" ? options["topK"] : void 0
+    ),
+    attrDouble(
+      "gen_ai.request.top_p",
+      typeof options["topP"] === "number" ? options["topP"] : void 0
+    )
   ];
 }
 
@@ -1077,6 +1312,7 @@ function wrapAISDK(aiSDK, deps) {
   const agentClasses = /* @__PURE__ */ new Set(["ToolLoopAgent"]);
   const sendEvents = ((_a = deps.options.send) == null ? void 0 : _a.events) !== false;
   const sendTraces = ((_b = deps.options.send) == null ? void 0 : _b.traces) !== false;
+  const autoAttachmentEnabled = deps.options.autoAttachment !== false;
   const proxyTarget = isModuleNamespace(aiSDK) ? Object.setPrototypeOf({}, aiSDK) : aiSDK;
   return new Proxy(proxyTarget, {
     get(target, prop, receiver) {
@@ -1130,14 +1366,27 @@ function wrapAISDK(aiSDK, deps) {
           ]
         }) : void 0;
         const rootParentForChildren = rootSpan ? { traceIdB64: rootSpan.ids.traceIdB64, spanIdB64: rootSpan.ids.spanIdB64 } : inheritedParent;
-        const wrapCtx = { eventId, telemetry, sendTraces, traceShipper: deps.traceShipper, rootParentForChildren };
+        const wrapCtx = {
+          eventId,
+          telemetry,
+          sendTraces,
+          traceShipper: deps.traceShipper,
+          rootParentForChildren
+        };
         const toolCalls = [];
         const argWithWrappedTools = wrapTools(arg, wrapCtx, toolCalls);
-        const argWithWrappedModel = wrapModel(argWithWrappedTools, aiSDK, outerOperationId, wrapCtx);
+        const argWithWrappedModel = wrapModel(
+          argWithWrappedTools,
+          aiSDK,
+          outerOperationId,
+          wrapCtx
+        );
         const finalize = async (result, error) => {
           var _a3, _b3, _c2;
           const usage = extractUsage(result);
           const model = extractModel(result);
+          const inputAttachments = autoAttachmentEnabled ? extractInputAttachmentsFromArgs(arg) : void 0;
+          const outputAttachments = autoAttachmentEnabled ? await extractOutputAttachmentsFromResult(result) : void 0;
           const baseMessages = coerceMessagesFromArgs(arg);
           const responseMessages = extractResponseMessages(result);
           const allMessages = [...baseMessages, ...responseMessages];
@@ -1150,7 +1399,7 @@ function wrapAISDK(aiSDK, deps) {
             output: defaultOutput,
             model,
             properties: ctx.properties,
-            attachments: ctx.attachments
+            attachments: mergeAttachments(ctx.attachments, inputAttachments, outputAttachments)
           };
           const built = await maybeBuildEvent(deps.options.buildEvent, allMessages);
           const patch = mergeBuildEventPatch(defaultPatch, built);
@@ -1170,7 +1419,10 @@ function wrapAISDK(aiSDK, deps) {
                   attrString("ai.response.finishReason", finishReason),
                   operation === "generateObject" || operation === "streamObject" ? attrString("ai.response.object", output) : attrString("ai.response.text", output),
                   attrString("ai.response.toolCalls", resultToolCalls),
-                  attrString("ai.response.providerMetadata", safeJsonWithUint8(providerMetadata))
+                  attrString(
+                    "ai.response.providerMetadata",
+                    safeJsonWithUint8(providerMetadata)
+                  )
                 ],
                 attrInt("ai.usage.promptTokens", usage == null ? void 0 : usage.inputTokens),
                 attrInt("ai.usage.completionTokens", usage == null ? void 0 : usage.outputTokens),
@@ -1180,7 +1432,12 @@ function wrapAISDK(aiSDK, deps) {
                 attrInt("ai.usage.reasoningTokens", reasoningTokens),
                 attrInt("ai.usage.cachedInputTokens", cachedInputTokens),
                 attrInt("ai.toolCall.count", toolCalls.length),
-                ...error ? [attrString("error.message", error instanceof Error ? error.message : String(error))] : []
+                ...error ? [
+                  attrString(
+                    "error.message",
+                    error instanceof Error ? error.message : String(error)
+                  )
+                ] : []
               ]
             });
           }
@@ -1196,12 +1453,18 @@ function wrapAISDK(aiSDK, deps) {
               attachments: patch.attachments,
               isPending: false
             }).catch((err) => {
-              if (debug) console.warn(`[raindrop-ai/ai-sdk] event patch failed: ${err instanceof Error ? err.message : err}`);
+              if (debug)
+                console.warn(
+                  `[raindrop-ai/ai-sdk] event patch failed: ${err instanceof Error ? err.message : err}`
+                );
             });
           }
         };
         const callOriginal = async (...args) => {
-          return await original.call(aiSDK, ...args);
+          return await original.call(
+            aiSDK,
+            ...args
+          );
         };
         const runWithContext = async (fn) => {
           if (!rootSpan) return await fn();
@@ -1215,7 +1478,10 @@ function wrapAISDK(aiSDK, deps) {
             try {
               await finalize(result);
             } catch (err) {
-              if (debug) console.warn(`[raindrop-ai/ai-sdk] finalize failed: ${err instanceof Error ? err.message : err}`);
+              if (debug)
+                console.warn(
+                  `[raindrop-ai/ai-sdk] finalize failed: ${err instanceof Error ? err.message : err}`
+                );
             }
           });
           try {
@@ -1242,7 +1508,10 @@ function wrapAISDK(aiSDK, deps) {
             try {
               await finalize(void 0, error);
             } catch (err) {
-              if (debug) console.warn(`[raindrop-ai/ai-sdk] finalize failed: ${err instanceof Error ? err.message : err}`);
+              if (debug)
+                console.warn(
+                  `[raindrop-ai/ai-sdk] finalize failed: ${err instanceof Error ? err.message : err}`
+                );
             }
             throw error;
           }
@@ -1256,14 +1525,20 @@ function wrapAISDK(aiSDK, deps) {
           try {
             await finalize(result);
           } catch (err) {
-            if (debug) console.warn(`[raindrop-ai/ai-sdk] finalize failed: ${err instanceof Error ? err.message : err}`);
+            if (debug)
+              console.warn(
+                `[raindrop-ai/ai-sdk] finalize failed: ${err instanceof Error ? err.message : err}`
+              );
           }
           return result;
         } catch (error) {
           try {
             await finalize(void 0, error);
           } catch (err) {
-            if (debug) console.warn(`[raindrop-ai/ai-sdk] finalize failed: ${err instanceof Error ? err.message : err}`);
+            if (debug)
+              console.warn(
+                `[raindrop-ai/ai-sdk] finalize failed: ${err instanceof Error ? err.message : err}`
+              );
           }
           throw error;
         }
@@ -1315,6 +1590,7 @@ function wrapAgentGenerate(generate, instance, agentSettings, className, aiSDK,
   var _a, _b;
   const sendEvents = ((_a = deps.options.send) == null ? void 0 : _a.events) !== false;
   const sendTraces = ((_b = deps.options.send) == null ? void 0 : _b.traces) !== false;
+  const autoAttachmentEnabled = deps.options.autoAttachment !== false;
   return async (...callArgs) => {
     var _a2, _b2, _c, _d;
     const callParams = callArgs[0];
@@ -1358,13 +1634,28 @@ function wrapAgentGenerate(generate, instance, agentSettings, className, aiSDK,
       ]
     }) : void 0;
     const rootParentForChildren = rootSpan ? { traceIdB64: rootSpan.ids.traceIdB64, spanIdB64: rootSpan.ids.spanIdB64 } : inheritedParent;
-    const wrapCtx = { eventId, telemetry, sendTraces, traceShipper: deps.traceShipper, rootParentForChildren };
+    const wrapCtx = {
+      eventId,
+      telemetry,
+      sendTraces,
+      traceShipper: deps.traceShipper,
+      rootParentForChildren
+    };
     const toolCalls = [];
-    const callParamsWithWrappedTools = callParams ? wrapTools(callParams, wrapCtx, toolCalls) : callParams;
+    const mergedArgsWithWrappedTools = wrapTools(mergedArgs, wrapCtx, toolCalls);
+    const mergedArgsWithWrappedModel = wrapModel(
+      mergedArgsWithWrappedTools,
+      aiSDK,
+      outerOperationId,
+      wrapCtx
+    );
+    const callParamsWithWrappedToolsAndModel = mergedArgsWithWrappedModel != null ? mergedArgsWithWrappedModel : {};
     const finalize = async (result, error) => {
       var _a3, _b3, _c2;
       const usage = extractUsage(result);
       const model = extractModel(result);
+      const inputAttachments = autoAttachmentEnabled ? extractInputAttachmentsFromArgs(mergedArgs) : void 0;
+      const outputAttachments = autoAttachmentEnabled ? await extractOutputAttachmentsFromResult(result) : void 0;
       const baseMessages = coerceMessagesFromArgs(mergedArgs);
       const responseMessages = extractResponseMessages(result);
       const allMessages = [...baseMessages, ...responseMessages];
@@ -1375,7 +1666,7 @@ function wrapAgentGenerate(generate, instance, agentSettings, className, aiSDK,
         output: outputText,
         model,
         properties: ctx.properties,
-        attachments: ctx.attachments
+        attachments: mergeAttachments(ctx.attachments, inputAttachments, outputAttachments)
       };
       const built = await maybeBuildEvent(deps.options.buildEvent, allMessages);
       const patch = mergeBuildEventPatch(defaultPatch, built);
@@ -1405,7 +1696,12 @@ function wrapAgentGenerate(generate, instance, agentSettings, className, aiSDK,
             attrInt("ai.usage.reasoningTokens", reasoningTokens),
             attrInt("ai.usage.cachedInputTokens", cachedInputTokens),
             attrInt("ai.toolCall.count", toolCalls.length),
-            ...error ? [attrString("error.message", error instanceof Error ? error.message : String(error))] : []
+            ...error ? [
+              attrString(
+                "error.message",
+                error instanceof Error ? error.message : String(error)
+              )
+            ] : []
           ]
         });
       }
@@ -1429,7 +1725,10 @@ function wrapAgentGenerate(generate, instance, agentSettings, className, aiSDK,
           attachments: patch.attachments,
           isPending: false
         }).catch((err) => {
-          if (debug) console.warn(`[raindrop-ai/ai-sdk] event patch failed: ${err instanceof Error ? err.message : err}`);
+          if (debug)
+            console.warn(
+              `[raindrop-ai/ai-sdk] event patch failed: ${err instanceof Error ? err.message : err}`
+            );
         });
       } else if (debug) {
         console.log(`[raindrop-ai/ai-sdk] Agent ${operation} sendEvents=false, skipping event`);
@@ -1451,7 +1750,7 @@ function wrapAgentGenerate(generate, instance, agentSettings, className, aiSDK,
     }
     try {
       const result = await runWithContext(async () => {
-        return await generate.call(instance, callParamsWithWrappedTools);
+        return await generate.call(instance, callParamsWithWrappedToolsAndModel);
       });
       if (debug) {
         console.log(`[raindrop-ai/ai-sdk] Agent ${operation} completed, finalizing...`);
@@ -1459,14 +1758,20 @@ function wrapAgentGenerate(generate, instance, agentSettings, className, aiSDK,
       try {
         await finalize(result);
       } catch (err) {
-        if (debug) console.warn(`[raindrop-ai/ai-sdk] finalize failed: ${err instanceof Error ? err.message : err}`);
+        if (debug)
+          console.warn(
+            `[raindrop-ai/ai-sdk] finalize failed: ${err instanceof Error ? err.message : err}`
+          );
       }
       return result;
     } catch (error) {
       try {
         await finalize(void 0, error);
       } catch (err) {
-        if (debug) console.warn(`[raindrop-ai/ai-sdk] finalize failed: ${err instanceof Error ? err.message : err}`);
+        if (debug)
+          console.warn(
+            `[raindrop-ai/ai-sdk] finalize failed: ${err instanceof Error ? err.message : err}`
+          );
       }
       throw error;
     }
@@ -1476,6 +1781,7 @@ function wrapAgentStream(stream, instance, agentSettings, className, aiSDK, deps
   var _a, _b;
   const sendEvents = ((_a = deps.options.send) == null ? void 0 : _a.events) !== false;
   const sendTraces = ((_b = deps.options.send) == null ? void 0 : _b.traces) !== false;
+  const autoAttachmentEnabled = deps.options.autoAttachment !== false;
   return async (...callArgs) => {
     var _a2, _b2, _c, _d;
     const callParams = callArgs[0];
@@ -1519,13 +1825,28 @@ function wrapAgentStream(stream, instance, agentSettings, className, aiSDK, deps
       ]
     }) : void 0;
     const rootParentForChildren = rootSpan ? { traceIdB64: rootSpan.ids.traceIdB64, spanIdB64: rootSpan.ids.spanIdB64 } : inheritedParent;
-    const wrapCtx = { eventId, telemetry, sendTraces, traceShipper: deps.traceShipper, rootParentForChildren };
+    const wrapCtx = {
+      eventId,
+      telemetry,
+      sendTraces,
+      traceShipper: deps.traceShipper,
+      rootParentForChildren
+    };
     const toolCalls = [];
-    const callParamsWithWrappedTools = callParams ? wrapTools(callParams, wrapCtx, toolCalls) : callParams;
+    const mergedArgsWithWrappedTools = wrapTools(mergedArgs, wrapCtx, toolCalls);
+    const mergedArgsWithWrappedModel = wrapModel(
+      mergedArgsWithWrappedTools,
+      aiSDK,
+      outerOperationId,
+      wrapCtx
+    );
+    const callParamsWithWrappedToolsAndModel = mergedArgsWithWrappedModel != null ? mergedArgsWithWrappedModel : {};
     const finalize = async (result, error) => {
       var _a3, _b3, _c2;
       const usage = extractUsage(result);
       const model = extractModel(result);
+      const inputAttachments = autoAttachmentEnabled ? extractInputAttachmentsFromArgs(mergedArgs) : void 0;
+      const outputAttachments = autoAttachmentEnabled ? await extractOutputAttachmentsFromResult(result) : void 0;
       const baseMessages = coerceMessagesFromArgs(mergedArgs);
       const responseMessages = extractResponseMessages(result);
       const allMessages = [...baseMessages, ...responseMessages];
@@ -1536,7 +1857,7 @@ function wrapAgentStream(stream, instance, agentSettings, className, aiSDK, deps
         output: outputText,
         model,
         properties: ctx.properties,
-        attachments: ctx.attachments
+        attachments: mergeAttachments(ctx.attachments, inputAttachments, outputAttachments)
       };
       const built = await maybeBuildEvent(deps.options.buildEvent, allMessages);
       const patch = mergeBuildEventPatch(defaultPatch, built);
@@ -1566,7 +1887,12 @@ function wrapAgentStream(stream, instance, agentSettings, className, aiSDK, deps
             attrInt("ai.usage.reasoningTokens", reasoningTokens),
             attrInt("ai.usage.cachedInputTokens", cachedInputTokens),
             attrInt("ai.toolCall.count", toolCalls.length),
-            ...error ? [attrString("error.message", error instanceof Error ? error.message : String(error))] : []
+            ...error ? [
+              attrString(
+                "error.message",
+                error instanceof Error ? error.message : String(error)
+              )
+            ] : []
           ]
         });
       }
@@ -1590,7 +1916,10 @@ function wrapAgentStream(stream, instance, agentSettings, className, aiSDK, deps
           attachments: patch.attachments,
           isPending: false
         }).catch((err) => {
-          if (debug) console.warn(`[raindrop-ai/ai-sdk] event patch failed: ${err instanceof Error ? err.message : err}`);
+          if (debug)
+            console.warn(
+              `[raindrop-ai/ai-sdk] event patch failed: ${err instanceof Error ? err.message : err}`
+            );
         });
       } else if (debug) {
         console.log(`[raindrop-ai/ai-sdk] Agent ${operation} sendEvents=false, skipping event`);
@@ -1610,16 +1939,22 @@ function wrapAgentStream(stream, instance, agentSettings, className, aiSDK, deps
         eventName: ctx.eventName
       });
     }
-    const callParamsWithOnFinish = wrapOnFinish(callParamsWithWrappedTools != null ? callParamsWithWrappedTools : {}, async (result) => {
-      if (debug) {
-        console.log(`[raindrop-ai/ai-sdk] Agent ${operation} onFinish callback, finalizing...`);
-      }
-      try {
-        await finalize(result);
-      } catch (err) {
-        if (debug) console.warn(`[raindrop-ai/ai-sdk] finalize failed: ${err instanceof Error ? err.message : err}`);
+    const callParamsWithOnFinish = wrapOnFinish(
+      callParamsWithWrappedToolsAndModel != null ? callParamsWithWrappedToolsAndModel : {},
+      async (result) => {
+        if (debug) {
+          console.log(`[raindrop-ai/ai-sdk] Agent ${operation} onFinish callback, finalizing...`);
+        }
+        try {
+          await finalize(result);
+        } catch (err) {
+          if (debug)
+            console.warn(
+              `[raindrop-ai/ai-sdk] finalize failed: ${err instanceof Error ? err.message : err}`
+            );
+        }
       }
-    });
+    );
     try {
       const result = await runWithContext(async () => {
         return await stream.call(instance, callParamsWithOnFinish);
@@ -1629,7 +1964,10 @@ function wrapAgentStream(stream, instance, agentSettings, className, aiSDK, deps
       try {
         await finalize(void 0, error);
       } catch (err) {
-        if (debug) console.warn(`[raindrop-ai/ai-sdk] finalize failed: ${err instanceof Error ? err.message : err}`);
+        if (debug)
+          console.warn(
+            `[raindrop-ai/ai-sdk] finalize failed: ${err instanceof Error ? err.message : err}`
+          );
       }
       throw error;
     }
@@ -1672,7 +2010,9 @@ function wrapToolExecute(name, tool, ctx, toolCalls) {
     if (!span) return;
     if (error) {
       ctx.traceShipper.endSpan(span, {
-        attributes: [attrString("error.message", error instanceof Error ? error.message : String(error))]
+        attributes: [
+          attrString("error.message", error instanceof Error ? error.message : String(error))
+        ]
       });
     } else {
       ctx.traceShipper.endSpan(span, {
@@ -1729,7 +2069,11 @@ function wrapToolExecute(name, tool, ctx, toolCalls) {
       };
       if (!toolSpan) return await run();
       return await runWithParentSpanContext(
-        { traceIdB64: toolSpan.ids.traceIdB64, spanIdB64: toolSpan.ids.spanIdB64, eventId: ctx.eventId },
+        {
+          traceIdB64: toolSpan.ids.traceIdB64,
+          spanIdB64: toolSpan.ids.spanIdB64,
+          eventId: ctx.eventId
+        },
         run
       );
     })();
@@ -1768,7 +2112,15 @@ function wrapModel(args, aiSDK, outerOperationId, ctx) {
             if (span) endDoGenerateSpan(span, result, modelInfo, ctx);
             return result;
           } catch (error) {
-            if (span) ctx.traceShipper.endSpan(span, { attributes: [attrString("error.message", error instanceof Error ? error.message : String(error))] });
+            if (span)
+              ctx.traceShipper.endSpan(span, {
+                attributes: [
+                  attrString(
+                    "error.message",
+                    error instanceof Error ? error.message : String(error)
+                  )
+                ]
+              });
             throw error;
           }
         };
@@ -1796,7 +2148,12 @@ function wrapModel(args, aiSDK, outerOperationId, ctx) {
           } catch (error) {
             if (span)
               ctx.traceShipper.endSpan(span, {
-                attributes: [attrString("error.message", error instanceof Error ? error.message : String(error))]
+                attributes: [
+                  attrString(
+                    "error.message",
+                    error instanceof Error ? error.message : String(error)
+                  )
+                ]
               });
             throw error;
           }
@@ -1822,11 +2179,17 @@ function wrapModel(args, aiSDK, outerOperationId, ctx) {
                 ...((_a = ctx.telemetry) == null ? void 0 : _a.recordOutputs) === false ? [] : [
                   attrString("ai.response.finishReason", finishReason),
                   attrString("ai.response.text", activeText.length ? activeText : void 0),
-                  attrString("ai.response.toolCalls", safeJsonWithUint8(toolCallsLocal.length ? toolCallsLocal : void 0)),
+                  attrString(
+                    "ai.response.toolCalls",
+                    safeJsonWithUint8(toolCallsLocal.length ? toolCallsLocal : void 0)
+                  ),
                   attrString("ai.response.id", responseId),
                   attrString("ai.response.model", responseModelId),
                   attrString("ai.response.timestamp", responseTimestampIso),
-                  attrString("ai.response.providerMetadata", safeJsonWithUint8(providerMetadata))
+                  attrString(
+                    "ai.response.providerMetadata",
+                    safeJsonWithUint8(providerMetadata)
+                  )
                 ],
                 attrInt("ai.usage.inputTokens", inputTokens),
                 attrInt("ai.usage.outputTokens", outputTokens),
@@ -1838,7 +2201,12 @@ function wrapModel(args, aiSDK, outerOperationId, ctx) {
                 ...msToFirstChunk !== void 0 ? [attrInt("ai.stream.msToFirstChunk", msToFirstChunk)] : [],
                 ...msToFinish !== void 0 ? [attrInt("ai.stream.msToFinish", msToFinish)] : [],
                 ...avgOutTokensPerSecond !== void 0 ? [attrDouble("ai.stream.avgOutputTokensPerSecond", avgOutTokensPerSecond)] : [],
-                ...error ? [attrString("error.message", error instanceof Error ? error.message : String(error))] : []
+                ...error ? [
+                  attrString(
+                    "error.message",
+                    error instanceof Error ? error.message : String(error)
+                  )
+                ] : []
               ]
             });
           };
@@ -1855,15 +2223,20 @@ function wrapModel(args, aiSDK, outerOperationId, ctx) {
                 if (firstChunkMs === void 0) firstChunkMs = Date.now();
                 if (isRecord(value)) {
                   const type = value["type"];
-                  if (type === "text-delta" && typeof value["textDelta"] === "string") activeText += value["textDelta"];
-                  if (type === "finish" && typeof value["finishReason"] === "string") finishReason = value["finishReason"];
+                  if (type === "text-delta" && typeof value["textDelta"] === "string")
+                    activeText += value["textDelta"];
+                  if (type === "finish" && typeof value["finishReason"] === "string")
+                    finishReason = value["finishReason"];
                   if (type === "tool-call") toolCallsLocal.push(value);
                   if ("response" in value && isRecord(value["response"])) {
                     const response = value["response"];
                     if (typeof response["id"] === "string") responseId = response["id"];
-                    if (typeof response["modelId"] === "string") responseModelId = response["modelId"];
-                    if (response["timestamp"] instanceof Date) responseTimestampIso = response["timestamp"].toISOString();
-                    else if (typeof response["timestamp"] === "string") responseTimestampIso = response["timestamp"];
+                    if (typeof response["modelId"] === "string")
+                      responseModelId = response["modelId"];
+                    if (response["timestamp"] instanceof Date)
+                      responseTimestampIso = response["timestamp"].toISOString();
+                    else if (typeof response["timestamp"] === "string")
+                      responseTimestampIso = response["timestamp"];
                   }
                   if ("usage" in value) usage = value["usage"];
                   if ("providerMetadata" in value) providerMetadata = value["providerMetadata"];
@@ -1905,7 +2278,11 @@ function startDoGenerateSpan(operationId, options, modelInfo, parent, ctx) {
       attrString("ai.telemetry.functionId", (_b = ctx.telemetry) == null ? void 0 : _b.functionId),
       attrString("ai.model.provider", modelInfo.provider),
       attrString("ai.model.id", modelInfo.modelId),
-      ...((_c = ctx.telemetry) == null ? void 0 : _c.recordInputs) === false ? [] : [attrString("ai.prompt.messages", promptJson), attrStringArray("ai.prompt.tools", toolsJson), attrString("ai.prompt.toolChoice", toolChoiceJson)],
+      ...((_c = ctx.telemetry) == null ? void 0 : _c.recordInputs) === false ? [] : [
+        attrString("ai.prompt.messages", promptJson),
+        attrStringArray("ai.prompt.tools", toolsJson),
+        attrString("ai.prompt.toolChoice", toolChoiceJson)
+      ],
       attrString("gen_ai.system", modelInfo.provider),
       attrString("gen_ai.request.model", modelInfo.modelId),
       ...attrsFromGenAiRequest(options)
@@ -1946,7 +2323,10 @@ function endDoGenerateSpan(span, result, modelInfo, ctx) {
       ...((_a = ctx.telemetry) == null ? void 0 : _a.recordOutputs) === false ? [] : [
         attrString("ai.response.finishReason", finishReason),
         attrString("ai.response.text", extractTextFromLmContent(content)),
-        attrString("ai.response.toolCalls", safeJsonWithUint8(extractToolCallsFromLmContent(content))),
+        attrString(
+          "ai.response.toolCalls",
+          safeJsonWithUint8(extractToolCallsFromLmContent(content))
+        ),
         attrString("ai.response.id", responseId),
         attrString("ai.response.model", responseModelId),
         attrString("ai.response.timestamp", responseTimestampIso),
@@ -1980,7 +2360,11 @@ function startDoStreamSpan(operationId, options, modelInfo, parent, ctx) {
       attrString("ai.telemetry.functionId", (_b = ctx.telemetry) == null ? void 0 : _b.functionId),
       attrString("ai.model.provider", modelInfo.provider),
       attrString("ai.model.id", modelInfo.modelId),
-      ...((_c = ctx.telemetry) == null ? void 0 : _c.recordInputs) === false ? [] : [attrString("ai.prompt.messages", promptJson), attrStringArray("ai.prompt.tools", toolsJson), attrString("ai.prompt.toolChoice", toolChoiceJson)],
+      ...((_c = ctx.telemetry) == null ? void 0 : _c.recordInputs) === false ? [] : [
+        attrString("ai.prompt.messages", promptJson),
+        attrStringArray("ai.prompt.tools", toolsJson),
+        attrString("ai.prompt.toolChoice", toolChoiceJson)
+      ],
       attrString("gen_ai.system", modelInfo.provider),
       attrString("gen_ai.request.model", modelInfo.modelId),
       ...attrsFromGenAiRequest(options)
@@ -2032,6 +2416,13 @@ function mergeBuildEventPatch(defaults, override) {
     attachments: override.attachments !== void 0 ? [...(_g = defaults.attachments) != null ? _g : [], ...(_h = override.attachments) != null ? _h : []] : defaults.attachments
   };
 }
+function mergeAttachments(...groups) {
+  const merged = [];
+  for (const group of groups) {
+    if (group == null ? void 0 : group.length) merged.push(...group);
+  }
+  return merged.length ? merged : void 0;
+}
 function extractNestedTokens(usage, key) {
   if (!isRecord(usage)) return void 0;
   const val = usage[key];
@@ -2051,23 +2442,30 @@ function eventMetadata(options) {
   if (options.properties) result["raindrop.properties"] = JSON.stringify(options.properties);
   return result;
 }
+function envDebugEnabled() {
+  var _a;
+  if (typeof process === "undefined") return false;
+  const flag = (_a = process.env) == null ? void 0 : _a.RAINDROP_AI_DEBUG;
+  return flag === "1" || flag === "true";
+}
 function createRaindropAISDK(opts) {
   var _a, _b, _c, _d, _e, _f, _g, _h, _i;
   const eventsEnabled = ((_a = opts.events) == null ? void 0 : _a.enabled) !== false;
   const tracesEnabled = ((_b = opts.traces) == null ? void 0 : _b.enabled) !== false;
+  const envDebug = envDebugEnabled();
   const eventShipper = new EventShipper({
     writeKey: opts.writeKey,
     endpoint: opts.endpoint,
     enabled: eventsEnabled,
-    debug: ((_c = opts.events) == null ? void 0 : _c.debug) === true,
+    debug: ((_c = opts.events) == null ? void 0 : _c.debug) === true || envDebug,
     partialFlushMs: (_d = opts.events) == null ? void 0 : _d.partialFlushMs
   });
   const traceShipper = new TraceShipper({
     writeKey: opts.writeKey,
     endpoint: opts.endpoint,
     enabled: tracesEnabled,
-    debug: ((_e = opts.traces) == null ? void 0 : _e.debug) === true,
-    debugSpans: ((_f = opts.traces) == null ? void 0 : _f.debugSpans) === true,
+    debug: ((_e = opts.traces) == null ? void 0 : _e.debug) === true || envDebug,
+    debugSpans: ((_f = opts.traces) == null ? void 0 : _f.debugSpans) === true || envDebug,
     flushIntervalMs: (_g = opts.traces) == null ? void 0 : _g.flushIntervalMs,
     maxBatchSize: (_h = opts.traces) == null ? void 0 : _h.maxBatchSize,
     maxQueueSize: (_i = opts.traces) == null ? void 0 : _i.maxQueueSize