raindrop-ai 0.0.86 → 0.0.87

package/dist/{chunk-JCKOY5OJ.mjs → chunk-KZ6B3DZM.mjs}

@@ -1,7 +1,7 @@
 // package.json
 var package_default = {
   name: "raindrop-ai",
-  version: "0.0.86",
+  version: "0.0.87",
   main: "dist/index.js",
   module: "dist/index.mjs",
   types: "dist/index.d.ts",
@@ -36,13 +36,13 @@ var package_default = {
     test: "vitest run",
     smoke: "tsx ./scripts/smoke.ts",
     "smoke:ai-sdk-self-diagnostics": "tsx ./scripts/smoke.ai-sdk-self-diagnostics.ts",
-    "smoke:self-diagnostics:v4": "yarn build && cd tests/v4 && yarn install && yarn smoke",
-    "smoke:self-diagnostics:v5": "yarn build && cd tests/v5 && yarn install && yarn smoke",
-    "smoke:self-diagnostics:v6": "yarn build && cd tests/v6 && yarn install && yarn smoke",
-    "smoke:self-diagnostics:ai-sdk:all": "yarn smoke:self-diagnostics:v4 && yarn smoke:self-diagnostics:v5 && yarn smoke:self-diagnostics:v6",
-    "smoke:self-diagnostics:openai-sdk": "yarn build && cd tests/openai && yarn install && yarn smoke",
-    "smoke:self-diagnostics:anthropic-sdk": "yarn build && cd tests/anthropic && yarn install && yarn smoke",
-    "smoke:self-diagnostics:all": "yarn smoke:self-diagnostics:ai-sdk:all && yarn smoke:self-diagnostics:openai-sdk && yarn smoke:self-diagnostics:anthropic-sdk"
+    "smoke:self-diagnostics:v4": "pnpm build && cd tests/v4 && pnpm install && pnpm smoke",
+    "smoke:self-diagnostics:v5": "pnpm build && cd tests/v5 && pnpm install && pnpm smoke",
+    "smoke:self-diagnostics:v6": "pnpm build && cd tests/v6 && pnpm install && pnpm smoke",
+    "smoke:self-diagnostics:ai-sdk:all": "pnpm smoke:self-diagnostics:v4 && pnpm smoke:self-diagnostics:v5 && pnpm smoke:self-diagnostics:v6",
+    "smoke:self-diagnostics:openai-sdk": "pnpm build && cd tests/openai && pnpm install && pnpm smoke",
+    "smoke:self-diagnostics:anthropic-sdk": "pnpm build && cd tests/anthropic && pnpm install && pnpm smoke",
+    "smoke:self-diagnostics:all": "pnpm smoke:self-diagnostics:ai-sdk:all && pnpm smoke:self-diagnostics:openai-sdk && pnpm smoke:self-diagnostics:anthropic-sdk"
   },
   devDependencies: {
     "@types/node": "^20.11.17",
@@ -53,7 +53,7 @@ var package_default = {
     vitest: "^4.0.10"
   },
   dependencies: {
-    "@dawn-analytics/redact-pii": "0.1.2",
+    "@dawn-analytics/redact-pii": "workspace:*",
     "@opentelemetry/api": "^1.9.0",
     "@opentelemetry/resources": "^2.0.1",
     "@traceloop/node-server-sdk": "0.19.0-otel-v1",
@@ -99,7 +99,7 @@ var package_default = {
 };
 
 // src/tracing/tracer-core.ts
-import { context as context2, trace as trace4 } from "@opentelemetry/api";
+import { context as context2, SpanStatusCode as SpanStatusCode3, trace as trace4 } from "@opentelemetry/api";
 import { AnthropicInstrumentation } from "@traceloop/instrumentation-anthropic";
 import { BedrockInstrumentation } from "@traceloop/instrumentation-bedrock";
 import { ChromaDBInstrumentation } from "@traceloop/instrumentation-chromadb";
@@ -937,12 +937,83 @@ var LiveTracer = class {
 };
 
 // src/tracing/tracer-core.ts
+function patchBedrockStreamingBug(inst) {
+  const anyInst = inst;
+  const origEndSpan = anyInst._endSpan.bind(anyInst);
+  anyInst._wrapPromise = function(span, promise) {
+    return promise.then(async (result) => {
+      const rec = result;
+      const body = rec == null ? void 0 : rec.body;
+      const isStreamBody = body != null && typeof body === "object" && Symbol.asyncIterator in body && !(body instanceof Uint8Array);
+      if (!isStreamBody) {
+        await origEndSpan({ span, result });
+        return result;
+      }
+      const origIterator = body[Symbol.asyncIterator]();
+      const chunks = [];
+      let spanEnded = false;
+      rec.body = {
+        [Symbol.asyncIterator]() {
+          return {
+            async next() {
+              try {
+                const iterResult = await origIterator.next();
+                if (iterResult.done) {
+                  async function* replay() {
+                    for (const c of chunks) yield c;
+                  }
+                  await origEndSpan({
+                    span,
+                    result: { ...rec, body: replay() }
+                  });
+                  spanEnded = true;
+                  return { done: true, value: void 0 };
+                }
+                chunks.push(iterResult.value);
+                return { done: false, value: iterResult.value };
+              } catch (error) {
+                if (!spanEnded) {
+                  const msg = error instanceof Error ? error.message : String(error);
+                  span.setStatus({ code: SpanStatusCode3.ERROR, message: msg });
+                  span.recordException(error);
+                  span.end();
+                  spanEnded = true;
+                }
+                throw error;
+              }
+            },
+            async return(value) {
+              var _a;
+              if (!spanEnded) {
+                span.end();
+                spanEnded = true;
+              }
+              await ((_a = origIterator.return) == null ? void 0 : _a.call(origIterator, value));
+              return { done: true, value: void 0 };
+            }
+          };
+        }
+      };
+      return result;
+    }).catch((error) => {
+      span.setStatus({
+        code: SpanStatusCode3.ERROR,
+        message: error.message
+      });
+      span.recordException(error);
+      span.end();
+      throw error;
+    });
+  };
+}
 function createDefaultInstrumentations() {
+  const bedrockInst = new BedrockInstrumentation();
+  patchBedrockStreamingBug(bedrockInst);
   return [
     new OpenAIInstrumentation(),
     new AnthropicInstrumentation(),
     new CohereInstrumentation(),
-    new BedrockInstrumentation(),
+    bedrockInst,
     new VertexAIInstrumentation(),
     new AIPlatformInstrumentation(),
     new PineconeInstrumentation(),
@@ -970,6 +1041,7 @@ function createManualInstrumentations(instrumentModules) {
   }
   if (instrumentModules.bedrock) {
     const inst = new BedrockInstrumentation();
+    patchBedrockStreamingBug(inst);
     inst.manuallyInstrument(instrumentModules.bedrock);
     instrumentations.push(inst);
   }

package/dist/index.js

@@ -152,7 +152,7 @@ var CategorizationRequestSchema = import_zod.z.object({
 // package.json
 var package_default = {
   name: "raindrop-ai",
-  version: "0.0.86",
+  version: "0.0.87",
   main: "dist/index.js",
   module: "dist/index.mjs",
   types: "dist/index.d.ts",
@@ -187,13 +187,13 @@ var package_default = {
     test: "vitest run",
     smoke: "tsx ./scripts/smoke.ts",
     "smoke:ai-sdk-self-diagnostics": "tsx ./scripts/smoke.ai-sdk-self-diagnostics.ts",
-    "smoke:self-diagnostics:v4": "yarn build && cd tests/v4 && yarn install && yarn smoke",
-    "smoke:self-diagnostics:v5": "yarn build && cd tests/v5 && yarn install && yarn smoke",
-    "smoke:self-diagnostics:v6": "yarn build && cd tests/v6 && yarn install && yarn smoke",
-    "smoke:self-diagnostics:ai-sdk:all": "yarn smoke:self-diagnostics:v4 && yarn smoke:self-diagnostics:v5 && yarn smoke:self-diagnostics:v6",
-    "smoke:self-diagnostics:openai-sdk": "yarn build && cd tests/openai && yarn install && yarn smoke",
-    "smoke:self-diagnostics:anthropic-sdk": "yarn build && cd tests/anthropic && yarn install && yarn smoke",
-    "smoke:self-diagnostics:all": "yarn smoke:self-diagnostics:ai-sdk:all && yarn smoke:self-diagnostics:openai-sdk && yarn smoke:self-diagnostics:anthropic-sdk"
+    "smoke:self-diagnostics:v4": "pnpm build && cd tests/v4 && pnpm install && pnpm smoke",
+    "smoke:self-diagnostics:v5": "pnpm build && cd tests/v5 && pnpm install && pnpm smoke",
+    "smoke:self-diagnostics:v6": "pnpm build && cd tests/v6 && pnpm install && pnpm smoke",
+    "smoke:self-diagnostics:ai-sdk:all": "pnpm smoke:self-diagnostics:v4 && pnpm smoke:self-diagnostics:v5 && pnpm smoke:self-diagnostics:v6",
+    "smoke:self-diagnostics:openai-sdk": "pnpm build && cd tests/openai && pnpm install && pnpm smoke",
+    "smoke:self-diagnostics:anthropic-sdk": "pnpm build && cd tests/anthropic && pnpm install && pnpm smoke",
+    "smoke:self-diagnostics:all": "pnpm smoke:self-diagnostics:ai-sdk:all && pnpm smoke:self-diagnostics:openai-sdk && pnpm smoke:self-diagnostics:anthropic-sdk"
   },
   devDependencies: {
     "@types/node": "^20.11.17",
@@ -204,7 +204,7 @@ var package_default = {
     vitest: "^4.0.10"
   },
   dependencies: {
-    "@dawn-analytics/redact-pii": "0.1.2",
+    "@dawn-analytics/redact-pii": "workspace:*",
     "@opentelemetry/api": "^1.9.0",
     "@opentelemetry/resources": "^2.0.1",
     "@traceloop/node-server-sdk": "0.19.0-otel-v1",
@@ -1174,12 +1174,83 @@ var LiveTracer = class {
 };
 
 // src/tracing/tracer-core.ts
+function patchBedrockStreamingBug(inst) {
+  const anyInst = inst;
+  const origEndSpan = anyInst._endSpan.bind(anyInst);
+  anyInst._wrapPromise = function(span, promise) {
+    return promise.then(async (result) => {
+      const rec = result;
+      const body = rec == null ? void 0 : rec.body;
+      const isStreamBody = body != null && typeof body === "object" && Symbol.asyncIterator in body && !(body instanceof Uint8Array);
+      if (!isStreamBody) {
+        await origEndSpan({ span, result });
+        return result;
+      }
+      const origIterator = body[Symbol.asyncIterator]();
+      const chunks = [];
+      let spanEnded = false;
+      rec.body = {
+        [Symbol.asyncIterator]() {
+          return {
+            async next() {
+              try {
+                const iterResult = await origIterator.next();
+                if (iterResult.done) {
+                  async function* replay() {
+                    for (const c of chunks) yield c;
+                  }
+                  await origEndSpan({
+                    span,
+                    result: { ...rec, body: replay() }
+                  });
+                  spanEnded = true;
+                  return { done: true, value: void 0 };
+                }
+                chunks.push(iterResult.value);
+                return { done: false, value: iterResult.value };
+              } catch (error) {
+                if (!spanEnded) {
+                  const msg = error instanceof Error ? error.message : String(error);
+                  span.setStatus({ code: import_api4.SpanStatusCode.ERROR, message: msg });
+                  span.recordException(error);
+                  span.end();
+                  spanEnded = true;
+                }
+                throw error;
+              }
+            },
+            async return(value) {
+              var _a;
+              if (!spanEnded) {
+                span.end();
+                spanEnded = true;
+              }
+              await ((_a = origIterator.return) == null ? void 0 : _a.call(origIterator, value));
+              return { done: true, value: void 0 };
+            }
+          };
+        }
+      };
+      return result;
+    }).catch((error) => {
+      span.setStatus({
+        code: import_api4.SpanStatusCode.ERROR,
+        message: error.message
+      });
+      span.recordException(error);
+      span.end();
+      throw error;
+    });
+  };
+}
 function createDefaultInstrumentations() {
+  const bedrockInst = new import_instrumentation_bedrock.BedrockInstrumentation();
+  patchBedrockStreamingBug(bedrockInst);
   return [
     new import_instrumentation_openai.OpenAIInstrumentation(),
     new import_instrumentation_anthropic.AnthropicInstrumentation(),
     new import_instrumentation_cohere.CohereInstrumentation(),
-    new import_instrumentation_bedrock.BedrockInstrumentation(),
+    bedrockInst,
     new import_instrumentation_vertexai.VertexAIInstrumentation(),
     new import_instrumentation_vertexai.AIPlatformInstrumentation(),
     new import_instrumentation_pinecone.PineconeInstrumentation(),
@@ -1207,6 +1278,7 @@ function createManualInstrumentations(instrumentModules) {
   }
   if (instrumentModules.bedrock) {
     const inst = new import_instrumentation_bedrock.BedrockInstrumentation();
+    patchBedrockStreamingBug(inst);
     inst.manuallyInstrument(instrumentModules.bedrock);
     instrumentations.push(inst);
   }

package/dist/index.mjs

@@ -1,7 +1,7 @@
 import {
   package_default,
   tracing
-} from "./chunk-JCKOY5OJ.mjs";
+} from "./chunk-KZ6B3DZM.mjs";
 
 // ../schemas/src/ingest/index.ts
 import { z } from "zod";

package/dist/tracing/index.js

@@ -184,7 +184,7 @@ function base64Encode(bytes) {
 // package.json
 var package_default = {
   name: "raindrop-ai",
-  version: "0.0.86",
+  version: "0.0.87",
   main: "dist/index.js",
   module: "dist/index.mjs",
   types: "dist/index.d.ts",
@@ -219,13 +219,13 @@ var package_default = {
     test: "vitest run",
     smoke: "tsx ./scripts/smoke.ts",
     "smoke:ai-sdk-self-diagnostics": "tsx ./scripts/smoke.ai-sdk-self-diagnostics.ts",
-    "smoke:self-diagnostics:v4": "yarn build && cd tests/v4 && yarn install && yarn smoke",
-    "smoke:self-diagnostics:v5": "yarn build && cd tests/v5 && yarn install && yarn smoke",
-    "smoke:self-diagnostics:v6": "yarn build && cd tests/v6 && yarn install && yarn smoke",
-    "smoke:self-diagnostics:ai-sdk:all": "yarn smoke:self-diagnostics:v4 && yarn smoke:self-diagnostics:v5 && yarn smoke:self-diagnostics:v6",
-    "smoke:self-diagnostics:openai-sdk": "yarn build && cd tests/openai && yarn install && yarn smoke",
-    "smoke:self-diagnostics:anthropic-sdk": "yarn build && cd tests/anthropic && yarn install && yarn smoke",
-    "smoke:self-diagnostics:all": "yarn smoke:self-diagnostics:ai-sdk:all && yarn smoke:self-diagnostics:openai-sdk && yarn smoke:self-diagnostics:anthropic-sdk"
+    "smoke:self-diagnostics:v4": "pnpm build && cd tests/v4 && pnpm install && pnpm smoke",
+    "smoke:self-diagnostics:v5": "pnpm build && cd tests/v5 && pnpm install && pnpm smoke",
+    "smoke:self-diagnostics:v6": "pnpm build && cd tests/v6 && pnpm install && pnpm smoke",
+    "smoke:self-diagnostics:ai-sdk:all": "pnpm smoke:self-diagnostics:v4 && pnpm smoke:self-diagnostics:v5 && pnpm smoke:self-diagnostics:v6",
+    "smoke:self-diagnostics:openai-sdk": "pnpm build && cd tests/openai && pnpm install && pnpm smoke",
+    "smoke:self-diagnostics:anthropic-sdk": "pnpm build && cd tests/anthropic && pnpm install && pnpm smoke",
+    "smoke:self-diagnostics:all": "pnpm smoke:self-diagnostics:ai-sdk:all && pnpm smoke:self-diagnostics:openai-sdk && pnpm smoke:self-diagnostics:anthropic-sdk"
   },
   devDependencies: {
     "@types/node": "^20.11.17",
@@ -236,7 +236,7 @@ var package_default = {
     vitest: "^4.0.10"
   },
   dependencies: {
-    "@dawn-analytics/redact-pii": "0.1.2",
+    "@dawn-analytics/redact-pii": "workspace:*",
     "@opentelemetry/api": "^1.9.0",
     "@opentelemetry/resources": "^2.0.1",
     "@traceloop/node-server-sdk": "0.19.0-otel-v1",
@@ -971,12 +971,83 @@ var LiveTracer = class {
 };
 
 // src/tracing/tracer-core.ts
+function patchBedrockStreamingBug(inst) {
+  const anyInst = inst;
+  const origEndSpan = anyInst._endSpan.bind(anyInst);
+  anyInst._wrapPromise = function(span, promise) {
+    return promise.then(async (result) => {
+      const rec = result;
+      const body = rec == null ? void 0 : rec.body;
+      const isStreamBody = body != null && typeof body === "object" && Symbol.asyncIterator in body && !(body instanceof Uint8Array);
+      if (!isStreamBody) {
+        await origEndSpan({ span, result });
+        return result;
+      }
+      const origIterator = body[Symbol.asyncIterator]();
+      const chunks = [];
+      let spanEnded = false;
+      rec.body = {
+        [Symbol.asyncIterator]() {
+          return {
+            async next() {
+              try {
+                const iterResult = await origIterator.next();
+                if (iterResult.done) {
+                  async function* replay() {
+                    for (const c of chunks) yield c;
+                  }
+                  await origEndSpan({
+                    span,
+                    result: { ...rec, body: replay() }
+                  });
+                  spanEnded = true;
+                  return { done: true, value: void 0 };
+                }
+                chunks.push(iterResult.value);
+                return { done: false, value: iterResult.value };
+              } catch (error) {
+                if (!spanEnded) {
+                  const msg = error instanceof Error ? error.message : String(error);
+                  span.setStatus({ code: import_api4.SpanStatusCode.ERROR, message: msg });
+                  span.recordException(error);
+                  span.end();
+                  spanEnded = true;
+                }
+                throw error;
+              }
+            },
+            async return(value) {
+              var _a;
+              if (!spanEnded) {
+                span.end();
+                spanEnded = true;
+              }
+              await ((_a = origIterator.return) == null ? void 0 : _a.call(origIterator, value));
+              return { done: true, value: void 0 };
+            }
+          };
+        }
+      };
+      return result;
+    }).catch((error) => {
+      span.setStatus({
+        code: import_api4.SpanStatusCode.ERROR,
+        message: error.message
+      });
+      span.recordException(error);
+      span.end();
+      throw error;
+    });
+  };
+}
 function createDefaultInstrumentations() {
+  const bedrockInst = new import_instrumentation_bedrock.BedrockInstrumentation();
+  patchBedrockStreamingBug(bedrockInst);
   return [
     new import_instrumentation_openai.OpenAIInstrumentation(),
     new import_instrumentation_anthropic.AnthropicInstrumentation(),
     new import_instrumentation_cohere.CohereInstrumentation(),
-    new import_instrumentation_bedrock.BedrockInstrumentation(),
+    bedrockInst,
     new import_instrumentation_vertexai.VertexAIInstrumentation(),
     new import_instrumentation_vertexai.AIPlatformInstrumentation(),
     new import_instrumentation_pinecone.PineconeInstrumentation(),
@@ -1004,6 +1075,7 @@ function createManualInstrumentations(instrumentModules) {
   }
   if (instrumentModules.bedrock) {
     const inst = new import_instrumentation_bedrock.BedrockInstrumentation();
+    patchBedrockStreamingBug(inst);
     inst.manuallyInstrument(instrumentModules.bedrock);
     instrumentations.push(inst);
   }

package/dist/tracing/index.mjs

@@ -1,6 +1,6 @@
 import {
   tracing
-} from "../chunk-JCKOY5OJ.mjs";
+} from "../chunk-KZ6B3DZM.mjs";
 
 // src/tracing/index.ts
 function initTracing() {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "raindrop-ai",
-  "version": "0.0.86",
+  "version": "0.0.87",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
   "types": "dist/index.d.ts",
@@ -27,22 +27,6 @@
   "files": [
     "dist/**"
   ],
-  "scripts": {
-    "build": "tsup",
-    "dev": "tsup --watch",
-    "clean": "rm -rf .turbo && rm -rf dist",
-    "try": "tsx ./src/example/index.ts",
-    "test": "vitest run",
-    "smoke": "tsx ./scripts/smoke.ts",
-    "smoke:ai-sdk-self-diagnostics": "tsx ./scripts/smoke.ai-sdk-self-diagnostics.ts",
-    "smoke:self-diagnostics:v4": "yarn build && cd tests/v4 && yarn install && yarn smoke",
-    "smoke:self-diagnostics:v5": "yarn build && cd tests/v5 && yarn install && yarn smoke",
-    "smoke:self-diagnostics:v6": "yarn build && cd tests/v6 && yarn install && yarn smoke",
-    "smoke:self-diagnostics:ai-sdk:all": "yarn smoke:self-diagnostics:v4 && yarn smoke:self-diagnostics:v5 && yarn smoke:self-diagnostics:v6",
-    "smoke:self-diagnostics:openai-sdk": "yarn build && cd tests/openai && yarn install && yarn smoke",
-    "smoke:self-diagnostics:anthropic-sdk": "yarn build && cd tests/anthropic && yarn install && yarn smoke",
-    "smoke:self-diagnostics:all": "yarn smoke:self-diagnostics:ai-sdk:all && yarn smoke:self-diagnostics:openai-sdk && yarn smoke:self-diagnostics:anthropic-sdk"
-  },
   "devDependencies": {
     "@types/node": "^20.11.17",
     "msw": "^2.12.8",
@@ -52,12 +36,12 @@
     "vitest": "^4.0.10"
   },
   "dependencies": {
-    "@dawn-analytics/redact-pii": "0.1.2",
     "@opentelemetry/api": "^1.9.0",
     "@opentelemetry/resources": "^2.0.1",
     "@traceloop/node-server-sdk": "0.19.0-otel-v1",
     "weakref": "^0.2.1",
-    "zod": "^3.23.8"
+    "zod": "^3.23.8",
+    "@dawn-analytics/redact-pii": "0.1.2"
   },
   "publishConfig": {
     "access": "public",
@@ -94,5 +78,21 @@
       "resolve": true
     },
     "clean": true
+  },
+  "scripts": {
+    "build": "tsup",
+    "dev": "tsup --watch",
+    "clean": "rm -rf .turbo && rm -rf dist",
+    "try": "tsx ./src/example/index.ts",
+    "test": "vitest run",
+    "smoke": "tsx ./scripts/smoke.ts",
+    "smoke:ai-sdk-self-diagnostics": "tsx ./scripts/smoke.ai-sdk-self-diagnostics.ts",
+    "smoke:self-diagnostics:v4": "pnpm build && cd tests/v4 && pnpm install && pnpm smoke",
+    "smoke:self-diagnostics:v5": "pnpm build && cd tests/v5 && pnpm install && pnpm smoke",
+    "smoke:self-diagnostics:v6": "pnpm build && cd tests/v6 && pnpm install && pnpm smoke",
+    "smoke:self-diagnostics:ai-sdk:all": "pnpm smoke:self-diagnostics:v4 && pnpm smoke:self-diagnostics:v5 && pnpm smoke:self-diagnostics:v6",
+    "smoke:self-diagnostics:openai-sdk": "pnpm build && cd tests/openai && pnpm install && pnpm smoke",
+    "smoke:self-diagnostics:anthropic-sdk": "pnpm build && cd tests/anthropic && pnpm install && pnpm smoke",
+    "smoke:self-diagnostics:all": "pnpm smoke:self-diagnostics:ai-sdk:all && pnpm smoke:self-diagnostics:openai-sdk && pnpm smoke:self-diagnostics:anthropic-sdk"
   }
-}
+}