@brizz/sdk 0.1.3-rc.1 → 0.1.3

package/dist/index.js

@@ -1429,7 +1429,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";
@@ -1439,10 +1439,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;
     }
   }
 };
@@ -1457,26 +1458,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[`${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[`${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[`${SpanAttributes.LLM_PROMPTS}.${index}.content`] = lastContent.text;
             }
           } else {
             attributes[`${SpanAttributes.LLM_PROMPTS}.${index}.content`] = JSON.stringify(
-              msg.content
+              message.content
             );
           }
         }
-        attributes[`${SpanAttributes.LLM_PROMPTS}.${index}.role`] = msg.role;
-      });
+        attributes[`${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
+      });
     }
   }
 };
@@ -1502,11 +1507,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[SpanAttributes.LLM_SYSTEM] = "OpenAI";
-    } else {
-      attributes[SpanAttributes.LLM_SYSTEM] = vendor;
-    }
+    attributes[SpanAttributes.LLM_SYSTEM] = vendor && vendor.startsWith("openai") ? "OpenAI" : vendor;
     delete attributes[AI_MODEL_PROVIDER];
   }
 };
@@ -1522,6 +1523,10 @@ var shouldHandleSpan = (span) => {
   return span.name in HANDLED_SPAN_NAMES;
 };
 var transformAiSdkSpan = (span) => {
+  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;
@@ -1532,10 +1537,6 @@ var transformAiSdkSpan = (span) => {
     spanContext: span.spanContext(),
     attributes: span.attributes
   });
-  if (!shouldHandleSpan(span)) {
-    logger.debug("Skipping span transformation", { spanName: span.name });
-    return;
-  }
   transformAiSdkSpanName(span);
   transformAiSdkAttributes(span.attributes);
 };