@brizz/sdk 0.1.3-rc.0 → 0.1.3

package/README.md

@@ -73,6 +73,7 @@ const { openai } = require('@ai-sdk/openai');
 generateText({
   model: openai('gpt-3.5-turbo'),
   prompt: 'Hello, world!',
+  experimental_telemetry: { isEnabled: true },
 }).then(result => console.log(result.text));
 ```
 
@@ -136,6 +137,7 @@ import { openai } from '@ai-sdk/openai';
 const result = await generateText({
   model: openai('gpt-3.5-turbo'),
   prompt: 'Hello, world!',
+  experimental_telemetry: { isEnabled: true },
 });
 ```
 
@@ -174,6 +176,7 @@ The SDK automatically instruments:
 - **OpenAI** - `openai` package
 - **Anthropic** - `@anthropic-ai/sdk` package
 - **Vercel AI SDK** - `ai` package (generateText, streamText, etc.)
+  - Requires `experimental_telemetry: { isEnabled: true }` in function calls
 - **LangChain** - `langchain` and `@langchain/*` packages
 - **LlamaIndex** - `llamaindex` package
 - **AWS Bedrock** - `@aws-sdk/client-bedrock-runtime`
@@ -181,6 +184,35 @@ The SDK automatically instruments:
 - **Vector Databases** - Pinecone, Qdrant, ChromaDB
 - **HTTP/Fetch** - Automatic network request tracing
 
+### Vercel AI SDK Integration
+
+For Vercel AI SDK instrumentation, you need to enable telemetry in your function calls:
+
+```typescript
+import { generateText, streamText } from 'ai';
+import { openai } from '@ai-sdk/openai';
+
+// For generateText
+const result = await generateText({
+  model: openai('gpt-4'),
+  prompt: 'Hello, world!',
+  experimental_telemetry: { isEnabled: true }, // Required for instrumentation
+});
+
+// For streamText
+const stream = streamText({
+  model: openai('gpt-4'),
+  messages: [{ role: 'user', content: 'Hello!' }],
+  experimental_telemetry: { isEnabled: true }, // Required for instrumentation
+});
+```
+
+This enables automatic tracing of:
+- Model calls and responses
+- Token usage and costs
+- Tool calls and executions
+- Streaming data
+
 ## PII Protection & Data Masking
 
 Automatically protects sensitive data in traces and logs:
@@ -226,6 +258,7 @@ async function processUserWorkflow(userId: string) {
   const result = await generateText({
     model: openai('gpt-4'),
     messages: [{ role: 'user', content: 'Hello' }],
+    experimental_telemetry: { isEnabled: true },
   });
 
   emitEvent('workflow.completed', { userId, result: result.text });
@@ -380,6 +413,7 @@ Check out the [examples](./examples/) directory for complete working examples:
 - For ESM, use loader + `Brizz.initialize()`
 - For CommonJS, use preload (with optional `Brizz.initialize()`)
 - Check that `BRIZZ_API_KEY` is set
+- For Vercel AI SDK: Add `experimental_telemetry: { isEnabled: true }` to function calls
 
 **CJS/ESM compatibility issues**
 

package/dist/index.cjs

@@ -1469,7 +1469,7 @@ var HANDLED_SPAN_NAMES = {
   "ai.streamText": "ai.streamText",
   "ai.toolCall": (span) => {
     const toolName = span.attributes["ai.toolCall.name"];
-    return `${toolName}.tool`;
+    return `${String(toolName ?? "unknown")}.tool`;
   }
 };
 var AI_RESPONSE_TEXT = "ai.response.text";
@@ -1479,10 +1479,11 @@ var AI_USAGE_COMPLETION_TOKENS = "ai.usage.completionTokens";
 var AI_MODEL_PROVIDER = "ai.model.provider";
 var transformAiSdkSpanName = (span) => {
   if (span.name in HANDLED_SPAN_NAMES) {
-    if (typeof HANDLED_SPAN_NAMES[span.name] === "function") {
-      span.name = HANDLED_SPAN_NAMES[span.name](span);
-    } else {
-      span.name = HANDLED_SPAN_NAMES[span.name];
+    const handler = HANDLED_SPAN_NAMES[span.name];
+    if (typeof handler === "function") {
+      span.name = handler(span);
+    } else if (handler) {
+      span.name = handler;
     }
   }
 };
@@ -1497,26 +1498,30 @@ var transformPromptMessages = (attributes) => {
   if (AI_PROMPT_MESSAGES in attributes) {
     try {
       const messages = JSON.parse(attributes[AI_PROMPT_MESSAGES]);
-      messages.forEach((msg, index) => {
-        logger.debug("Transforming prompt message", { msg, type: typeof msg.content });
-        if (typeof msg.content === "string") {
-          attributes[`${import_ai_semantic_conventions.SpanAttributes.LLM_PROMPTS}.${index}.content`] = msg.content;
+      for (const [index, msg] of messages.entries()) {
+        const message = msg;
+        logger.debug("Transforming prompt message", { msg: message, type: typeof message.content });
+        if (typeof message.content === "string") {
+          attributes[`${import_ai_semantic_conventions.SpanAttributes.LLM_PROMPTS}.${index}.content`] = message.content;
         } else {
-          if (Array.isArray(msg.content) && msg.content.length > 0) {
-            const lastContent = msg.content[msg.content.length - 1];
-            if (lastContent.text) {
+          if (Array.isArray(message.content) && message.content.length > 0) {
+            const lastContent = message.content.at(-1);
+            if (lastContent?.text) {
               attributes[`${import_ai_semantic_conventions.SpanAttributes.LLM_PROMPTS}.${index}.content`] = lastContent.text;
             }
           } else {
             attributes[`${import_ai_semantic_conventions.SpanAttributes.LLM_PROMPTS}.${index}.content`] = JSON.stringify(
-              msg.content
+              message.content
             );
           }
         }
-        attributes[`${import_ai_semantic_conventions.SpanAttributes.LLM_PROMPTS}.${index}.role`] = msg.role;
-      });
+        attributes[`${import_ai_semantic_conventions.SpanAttributes.LLM_PROMPTS}.${index}.role`] = message.role;
+      }
       delete attributes[AI_PROMPT_MESSAGES];
-    } catch {
+    } catch (error) {
+      logger.debug("Skipping prompt messages transformation because of JSON parsing error", {
+        e: error
+      });
     }
   }
 };
@@ -1542,11 +1547,7 @@ var calculateTotalTokens = (attributes) => {
 var transformVendor = (attributes) => {
   if (AI_MODEL_PROVIDER in attributes) {
     const vendor = attributes[AI_MODEL_PROVIDER];
-    if (vendor && vendor.startsWith("openai")) {
-      attributes[import_ai_semantic_conventions.SpanAttributes.LLM_SYSTEM] = "OpenAI";
-    } else {
-      attributes[import_ai_semantic_conventions.SpanAttributes.LLM_SYSTEM] = vendor;
-    }
+    attributes[import_ai_semantic_conventions.SpanAttributes.LLM_SYSTEM] = vendor && vendor.startsWith("openai") ? "OpenAI" : vendor;
     delete attributes[AI_MODEL_PROVIDER];
   }
 };
@@ -1562,11 +1563,20 @@ var shouldHandleSpan = (span) => {
   return span.name in HANDLED_SPAN_NAMES;
 };
 var transformAiSdkSpan = (span) => {
-  logger.debug("Transforming AI SDK span", { spanName: span.name });
   if (!shouldHandleSpan(span)) {
     logger.debug("Skipping span transformation", { spanName: span.name });
     return;
   }
+  for (const key in span.attributes) {
+    if (Number.isNaN(span.attributes[key])) {
+      span.attributes[key] = 0;
+    }
+  }
+  logger.debug("Transforming AI SDK span", {
+    spanName: span.name,
+    spanContext: span.spanContext(),
+    attributes: span.attributes
+  });
   transformAiSdkSpanName(span);
   transformAiSdkAttributes(span.attributes);
 };