@brizz/sdk 0.1.3-rc.1 → 0.1.4

package/dist/index.d.cts

@@ -116,7 +116,8 @@ declare function emitEvent(name: string, attributes?: Record<string, string | nu
 declare function getSpanExporter(): SpanExporter;
 declare function getSpanProcessor(): SpanProcessor;
 
-declare function WithSessionId<A extends unknown[], F extends (...args: A) => ReturnType<F>>(sessionId: string, fn: F, thisArg?: ThisParameterType<F>, ...args: A): ReturnType<F>;
+declare function withSessionId<F extends (this: any, ...args: any[]) => any>(sessionId: string, fn: F, thisArg?: ThisParameterType<F>): F;
+declare function callWithSessionId<A extends unknown[], F extends (...args: A) => ReturnType<F>>(sessionId: string, fn: F, thisArg?: ThisParameterType<F>, ...args: A): ReturnType<F>;
 
 declare function getMetricsExporter(): OTLPMetricExporter;
 declare function getMetricsReader(): MetricReader;
@@ -134,4 +135,4 @@ declare namespace init {
   export {  };
 }
 
-export { Brizz, DEFAULT_PII_PATTERNS, type IAttributesMaskingRule, type IBrizzInitializeOptions, type IEventMaskingConfig, type IEventMaskingRule, type IInstrumentModules, type ILogMaskingConfig, type ILogger, type IMaskingConfig, type ISpanMaskingConfig, LogLevel, type MaskingMode, type RuntimeInfo, WithSessionId, detectRuntime, emitEvent, getLogLevel, getMetricsExporter, getMetricsReader, getSpanExporter, getSpanProcessor, init, logger, maskAttributes, maskValue, setLogLevel };
+export { Brizz, DEFAULT_PII_PATTERNS, type IAttributesMaskingRule, type IBrizzInitializeOptions, type IEventMaskingConfig, type IEventMaskingRule, type IInstrumentModules, type ILogMaskingConfig, type ILogger, type IMaskingConfig, type ISpanMaskingConfig, LogLevel, type MaskingMode, type RuntimeInfo, callWithSessionId, detectRuntime, emitEvent, getLogLevel, getMetricsExporter, getMetricsReader, getSpanExporter, getSpanProcessor, init, logger, maskAttributes, maskValue, setLogLevel, withSessionId };

package/dist/index.d.ts

@@ -116,7 +116,8 @@ declare function emitEvent(name: string, attributes?: Record<string, string | nu
 declare function getSpanExporter(): SpanExporter;
 declare function getSpanProcessor(): SpanProcessor;
 
-declare function WithSessionId<A extends unknown[], F extends (...args: A) => ReturnType<F>>(sessionId: string, fn: F, thisArg?: ThisParameterType<F>, ...args: A): ReturnType<F>;
+declare function withSessionId<F extends (this: any, ...args: any[]) => any>(sessionId: string, fn: F, thisArg?: ThisParameterType<F>): F;
+declare function callWithSessionId<A extends unknown[], F extends (...args: A) => ReturnType<F>>(sessionId: string, fn: F, thisArg?: ThisParameterType<F>, ...args: A): ReturnType<F>;
 
 declare function getMetricsExporter(): OTLPMetricExporter;
 declare function getMetricsReader(): MetricReader;
@@ -134,4 +135,4 @@ declare namespace init {
   export {  };
 }
 
-export { Brizz, DEFAULT_PII_PATTERNS, type IAttributesMaskingRule, type IBrizzInitializeOptions, type IEventMaskingConfig, type IEventMaskingRule, type IInstrumentModules, type ILogMaskingConfig, type ILogger, type IMaskingConfig, type ISpanMaskingConfig, LogLevel, type MaskingMode, type RuntimeInfo, WithSessionId, detectRuntime, emitEvent, getLogLevel, getMetricsExporter, getMetricsReader, getSpanExporter, getSpanProcessor, init, logger, maskAttributes, maskValue, setLogLevel };
+export { Brizz, DEFAULT_PII_PATTERNS, type IAttributesMaskingRule, type IBrizzInitializeOptions, type IEventMaskingConfig, type IEventMaskingRule, type IInstrumentModules, type ILogMaskingConfig, type ILogger, type IMaskingConfig, type ISpanMaskingConfig, LogLevel, type MaskingMode, type RuntimeInfo, callWithSessionId, detectRuntime, emitEvent, getLogLevel, getMetricsExporter, getMetricsReader, getSpanExporter, getSpanProcessor, init, logger, maskAttributes, maskValue, setLogLevel, withSessionId };

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);
 };
@@ -1750,7 +1751,15 @@ function withProperties(properties, fn, thisArg, ...args) {
   const newContext = context3.active().setValue(PROPERTIES_CONTEXT_KEY, properties);
   return context3.with(newContext, fn, thisArg, ...args);
 }
-function WithSessionId(sessionId, fn, thisArg, ...args) {
+function withSessionId(sessionId, fn, thisArg) {
+  return function wrapped(...args) {
+    const base = context3.active();
+    const prev = base.getValue(PROPERTIES_CONTEXT_KEY);
+    const next = base.setValue(PROPERTIES_CONTEXT_KEY, prev ? { ...prev, [SESSION_ID]: sessionId } : { [SESSION_ID]: sessionId });
+    return context3.with(next, fn, thisArg ?? this, ...args);
+  };
+}
+function callWithSessionId(sessionId, fn, thisArg, ...args) {
   return withProperties({ [SESSION_ID]: sessionId }, fn, thisArg, ...args);
 }
 
@@ -1993,7 +2002,7 @@ export {
   DEFAULT_PII_PATTERNS,
   LogLevel,
   SeverityNumber2 as SeverityNumber,
-  WithSessionId,
+  callWithSessionId,
   detectRuntime,
   emitEvent,
   getLogLevel,
@@ -2005,6 +2014,7 @@ export {
   logger,
   maskAttributes,
   maskValue,
-  setLogLevel
+  setLogLevel,
+  withSessionId
 };
 //# sourceMappingURL=index.js.map