@brizz/sdk 0.1.3 → 0.1.5

package/dist/index.d.cts

@@ -88,6 +88,7 @@ interface IBrizzInitializeOptions {
     apiKey?: string;
     headers?: Record<string, string>;
     disableBatch?: boolean;
+    environment?: string;
     masking?: boolean | IMaskingConfig;
     logLevel?: LogLevel | string;
     instrumentModules?: IInstrumentModules;
@@ -116,7 +117,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 +136,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

@@ -88,6 +88,7 @@ interface IBrizzInitializeOptions {
     apiKey?: string;
     headers?: Record<string, string>;
     disableBatch?: boolean;
+    environment?: string;
     masking?: boolean | IMaskingConfig;
     logLevel?: LogLevel | string;
     instrumentModules?: IInstrumentModules;
@@ -116,7 +117,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 +136,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

@@ -280,6 +280,7 @@ function resolveConfig(options) {
     headers: { ...options.headers },
     apiKey: process.env["BRIZZ_API_KEY"] || options.apiKey,
     disableBatch: process.env["BRIZZ_DISABLE_BATCH"] === "true" || !!options.disableBatch,
+    environment: process.env["BRIZZ_ENVIRONMENT"] || options.environment,
     logLevel: resolvedLogLevel,
     masking: resolvedMasking
   };
@@ -976,7 +977,7 @@ function maskStringByPattern(value, pattern, mode = "full") {
     logger.warn("Regex execution failed, skipping masking", error);
     return value;
   }
-  for (const matchInfo of matches.reverse()) {
+  for (const matchInfo of matches.toReversed()) {
     let patternName = "unknown";
     if (matchInfo.groups) {
       for (const [groupName, groupValue] of Object.entries(matchInfo.groups)) {
@@ -1222,9 +1223,13 @@ var LoggingModule = class _LoggingModule {
     logger.debug("Creating resource with service name", {
       serviceName: config.appName
     });
-    const resource = resourceFromAttributes({
+    const resourceAttributes = {
       "service.name": config.appName
-    });
+    };
+    if (config.environment) {
+      resourceAttributes["deployment.environment"] = config.environment;
+    }
+    const resource = resourceFromAttributes(resourceAttributes);
     logger.debug("Creating logger provider with resource");
     this.loggerProvider = new LoggerProvider({
       resource,
@@ -1418,130 +1423,6 @@ import {
   BatchSpanProcessor,
   SimpleSpanProcessor
 } from "@opentelemetry/sdk-trace-base";
-
-// src/internal/trace/transformations/vercel-ai.ts
-import { SpanAttributes } from "@traceloop/ai-semantic-conventions";
-var AI_GENERATE_TEXT_DO_GENERATE = "ai.generateText.doGenerate";
-var AI_STREAM_TEXT_DO_STREAM = "ai.streamText.doStream";
-var HANDLED_SPAN_NAMES = {
-  [AI_GENERATE_TEXT_DO_GENERATE]: "gen_ai.chat",
-  [AI_STREAM_TEXT_DO_STREAM]: "gen_ai.chat",
-  "ai.streamText": "ai.streamText",
-  "ai.toolCall": (span) => {
-    const toolName = span.attributes["ai.toolCall.name"];
-    return `${String(toolName ?? "unknown")}.tool`;
-  }
-};
-var AI_RESPONSE_TEXT = "ai.response.text";
-var AI_PROMPT_MESSAGES = "ai.prompt.messages";
-var AI_USAGE_PROMPT_TOKENS = "ai.usage.promptTokens";
-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) {
-    const handler = HANDLED_SPAN_NAMES[span.name];
-    if (typeof handler === "function") {
-      span.name = handler(span);
-    } else if (handler) {
-      span.name = handler;
-    }
-  }
-};
-var transformResponseText = (attributes) => {
-  if (AI_RESPONSE_TEXT in attributes) {
-    attributes[`${SpanAttributes.LLM_COMPLETIONS}.0.content`] = attributes[AI_RESPONSE_TEXT];
-    attributes[`${SpanAttributes.LLM_COMPLETIONS}.0.role`] = "assistant";
-    delete attributes[AI_RESPONSE_TEXT];
-  }
-};
-var transformPromptMessages = (attributes) => {
-  if (AI_PROMPT_MESSAGES in attributes) {
-    try {
-      const messages = JSON.parse(attributes[AI_PROMPT_MESSAGES]);
-      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(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(
-              message.content
-            );
-          }
-        }
-        attributes[`${SpanAttributes.LLM_PROMPTS}.${index}.role`] = message.role;
-      }
-      delete attributes[AI_PROMPT_MESSAGES];
-    } catch (error) {
-      logger.debug("Skipping prompt messages transformation because of JSON parsing error", {
-        e: error
-      });
-    }
-  }
-};
-var transformPromptTokens = (attributes) => {
-  if (AI_USAGE_PROMPT_TOKENS in attributes) {
-    attributes[`${SpanAttributes.LLM_USAGE_PROMPT_TOKENS}`] = attributes[AI_USAGE_PROMPT_TOKENS];
-    delete attributes[AI_USAGE_PROMPT_TOKENS];
-  }
-};
-var transformCompletionTokens = (attributes) => {
-  if (AI_USAGE_COMPLETION_TOKENS in attributes) {
-    attributes[`${SpanAttributes.LLM_USAGE_COMPLETION_TOKENS}`] = attributes[AI_USAGE_COMPLETION_TOKENS];
-    delete attributes[AI_USAGE_COMPLETION_TOKENS];
-  }
-};
-var calculateTotalTokens = (attributes) => {
-  const promptTokens = attributes[`${SpanAttributes.LLM_USAGE_PROMPT_TOKENS}`];
-  const completionTokens = attributes[`${SpanAttributes.LLM_USAGE_COMPLETION_TOKENS}`];
-  if (promptTokens && completionTokens) {
-    attributes[`${SpanAttributes.LLM_USAGE_TOTAL_TOKENS}`] = Number(promptTokens) + Number(completionTokens);
-  }
-};
-var transformVendor = (attributes) => {
-  if (AI_MODEL_PROVIDER in attributes) {
-    const vendor = attributes[AI_MODEL_PROVIDER];
-    attributes[SpanAttributes.LLM_SYSTEM] = vendor && vendor.startsWith("openai") ? "OpenAI" : vendor;
-    delete attributes[AI_MODEL_PROVIDER];
-  }
-};
-var transformAiSdkAttributes = (attributes) => {
-  transformResponseText(attributes);
-  transformPromptMessages(attributes);
-  transformPromptTokens(attributes);
-  transformCompletionTokens(attributes);
-  calculateTotalTokens(attributes);
-  transformVendor(attributes);
-};
-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;
-    }
-  }
-  logger.debug("Transforming AI SDK span", {
-    spanName: span.name,
-    spanContext: span.spanContext(),
-    attributes: span.attributes
-  });
-  transformAiSdkSpanName(span);
-  transformAiSdkAttributes(span.attributes);
-};
-
-// src/internal/trace/processors/span-processor.ts
 var DEFAULT_MASKING_RULES = [
   {
     mode: "partial",
@@ -1584,10 +1465,6 @@ var BrizzSimpleSpanProcessor = class extends SimpleSpanProcessor {
     }
     super.onStart(span, parentContext);
   }
-  onEnd(span) {
-    transformAiSdkSpan(span);
-    super.onEnd(span);
-  }
 };
 var BrizzBatchSpanProcessor = class extends BatchSpanProcessor {
   config;
@@ -1608,10 +1485,6 @@ var BrizzBatchSpanProcessor = class extends BatchSpanProcessor {
     }
     super.onStart(span, parentContext);
   }
-  onEnd(span) {
-    transformAiSdkSpan(span);
-    super.onEnd(span);
-  }
 };
 function maskSpan(span, config) {
   if (!span.attributes || Object.keys(span.attributes).length === 0) {
@@ -1751,7 +1624,18 @@ 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);
 }
 
@@ -1825,12 +1709,16 @@ var _Brizz = class __Brizz {
     }
     const registry = InstrumentationRegistry.getInstance();
     const manualInstrumentations = registry.getManualInstrumentations();
+    const resourceAttributes = {
+      "service.name": resolvedConfig.appName
+    };
+    if (resolvedConfig.environment) {
+      resourceAttributes["deployment.environment"] = resolvedConfig.environment;
+    }
     this._sdk = new NodeSDK({
       spanProcessors: [getSpanProcessor()],
       metricReader: getMetricsReader(),
-      resource: resourceFromAttributes2({
-        "service.name": resolvedConfig.appName
-      }),
+      resource: resourceFromAttributes2(resourceAttributes),
       instrumentations: manualInstrumentations
     });
     this._sdk.start();
@@ -1954,7 +1842,10 @@ function detectRuntime() {
   const isESM = noNodeJSGlobals && noModuleSystem;
   const isCJS = hasRequire && typeof module !== "undefined" && typeof exports !== "undefined" && typeof __filename === "string" && typeof __dirname === "string";
   const supportsLoaderAPI = (major ?? 0) >= 21 || (major ?? 0) === 20 && (minor ?? 0) >= 6 || (major ?? 0) === 18 && (minor ?? 0) >= 19;
-  const supportsRegister = !!process.features?.typescript || !!globalThis.module?.register;
+  const supportsRegister = (
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    !!process.features?.typescript || !!globalThis.module?.register
+  );
   logger.debug("Runtime detection results:", {
     nodeVersion: `${major ?? 0}.${minor ?? 0}`,
     isESM,
@@ -1975,6 +1866,7 @@ function detectRuntime() {
       module: typeof module,
       exports: typeof exports,
       "process.features": process.features,
+      // eslint-disable-next-line @typescript-eslint/no-explicit-any
       "globalThis.module": globalThis.module
     }
   });
@@ -1994,7 +1886,7 @@ export {
   DEFAULT_PII_PATTERNS,
   LogLevel,
   SeverityNumber2 as SeverityNumber,
-  WithSessionId,
+  callWithSessionId,
   detectRuntime,
   emitEvent,
   getLogLevel,
@@ -2006,6 +1898,7 @@ export {
   logger,
   maskAttributes,
   maskValue,
-  setLogLevel
+  setLogLevel,
+  withSessionId
 };
 //# sourceMappingURL=index.js.map