@brizz/sdk 0.1.4 → 0.1.6

package/README.md

@@ -292,6 +292,21 @@ import { callWithSessionId } from '@brizz/sdk';
 await callWithSessionId('session-123', processUserWorkflow, null, 'user-456');
 ```
 
+### Custom Properties
+
+Add custom properties to telemetry context:
+
+```typescript
+import { withProperties, callWithProperties } from '@brizz/sdk';
+
+// Wrapper pattern - properties applied to all calls
+const taggedFn = withProperties({ env: 'prod', feature: 'chat' }, myFunction);
+await taggedFn(args);
+
+// Immediate execution - properties applied once
+await callWithProperties({ userId: 'user-123' }, myFunction, null, args);
+```
+
 ### Handling Method Context
 
 When wrapping methods that use `this`, you have several options:
@@ -334,6 +349,18 @@ const sessionedProcess = withSessionId(
 **Note:** The arrow function approach (Option 1) is recommended as it's more explicit, avoids lint
 warnings, and is less prone to `this` binding issues.
 
+## Deployment Environment
+
+Optionally specify the deployment environment for better filtering and organization:
+
+```typescript
+Brizz.initialize({
+  apiKey: 'your-api-key',
+  appName: 'my-app',
+  environment: 'production', // Optional: 'dev', 'staging', 'production', etc.
+});
+```
+
 ## Custom Events & Logging
 
 Emit custom events and structured logs:

package/dist/index.cjs

@@ -34,6 +34,7 @@ __export(src_exports, {
   DEFAULT_PII_PATTERNS: () => DEFAULT_PII_PATTERNS,
   LogLevel: () => LogLevel,
   SeverityNumber: () => import_api_logs2.SeverityNumber,
+  callWithProperties: () => callWithProperties,
   callWithSessionId: () => callWithSessionId,
   detectRuntime: () => detectRuntime,
   emitEvent: () => emitEvent,
@@ -47,6 +48,7 @@ __export(src_exports, {
   maskAttributes: () => maskAttributes,
   maskValue: () => maskValue,
   setLogLevel: () => setLogLevel,
+  withProperties: () => withProperties,
   withSessionId: () => withSessionId
 });
 module.exports = __toCommonJS(src_exports);
@@ -326,6 +328,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
   };
@@ -1020,7 +1023,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)) {
@@ -1266,9 +1269,13 @@ var LoggingModule = class _LoggingModule {
     logger.debug("Creating resource with service name", {
       serviceName: config.appName
     });
-    const resource = (0, import_resources.resourceFromAttributes)({
+    const resourceAttributes = {
       "service.name": config.appName
-    });
+    };
+    if (config.environment) {
+      resourceAttributes["deployment.environment"] = config.environment;
+    }
+    const resource = (0, import_resources.resourceFromAttributes)(resourceAttributes);
     logger.debug("Creating logger provider with resource");
     this.loggerProvider = new import_sdk_logs2.LoggerProvider({
       resource,
@@ -1459,130 +1466,6 @@ var import_exporter_trace_otlp_http = require("@opentelemetry/exporter-trace-otl
 // src/internal/trace/processors/span-processor.ts
 var import_api4 = require("@opentelemetry/api");
 var import_sdk_trace_base = require("@opentelemetry/sdk-trace-base");
-
-// src/internal/trace/transformations/vercel-ai.ts
-var import_ai_semantic_conventions = require("@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[`${import_ai_semantic_conventions.SpanAttributes.LLM_COMPLETIONS}.0.content`] = attributes[AI_RESPONSE_TEXT];
-    attributes[`${import_ai_semantic_conventions.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[`${import_ai_semantic_conventions.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[`${import_ai_semantic_conventions.SpanAttributes.LLM_PROMPTS}.${index}.content`] = lastContent.text;
-            }
-          } else {
-            attributes[`${import_ai_semantic_conventions.SpanAttributes.LLM_PROMPTS}.${index}.content`] = JSON.stringify(
-              message.content
-            );
-          }
-        }
-        attributes[`${import_ai_semantic_conventions.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[`${import_ai_semantic_conventions.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[`${import_ai_semantic_conventions.SpanAttributes.LLM_USAGE_COMPLETION_TOKENS}`] = attributes[AI_USAGE_COMPLETION_TOKENS];
-    delete attributes[AI_USAGE_COMPLETION_TOKENS];
-  }
-};
-var calculateTotalTokens = (attributes) => {
-  const promptTokens = attributes[`${import_ai_semantic_conventions.SpanAttributes.LLM_USAGE_PROMPT_TOKENS}`];
-  const completionTokens = attributes[`${import_ai_semantic_conventions.SpanAttributes.LLM_USAGE_COMPLETION_TOKENS}`];
-  if (promptTokens && completionTokens) {
-    attributes[`${import_ai_semantic_conventions.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[import_ai_semantic_conventions.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",
@@ -1625,10 +1508,6 @@ var BrizzSimpleSpanProcessor = class extends import_sdk_trace_base.SimpleSpanPro
     }
     super.onStart(span, parentContext);
   }
-  onEnd(span) {
-    transformAiSdkSpan(span);
-    super.onEnd(span);
-  }
 };
 var BrizzBatchSpanProcessor = class extends import_sdk_trace_base.BatchSpanProcessor {
   config;
@@ -1649,10 +1528,6 @@ var BrizzBatchSpanProcessor = class extends import_sdk_trace_base.BatchSpanProce
     }
     super.onStart(span, parentContext);
   }
-  onEnd(span) {
-    transformAiSdkSpan(span);
-    super.onEnd(span);
-  }
 };
 function maskSpan(span, config) {
   if (!span.attributes || Object.keys(span.attributes).length === 0) {
@@ -1785,23 +1660,28 @@ function getSpanProcessor() {
 
 // src/internal/trace/session.ts
 var import_api5 = require("@opentelemetry/api");
-function withProperties(properties, fn, thisArg, ...args) {
-  if (Object.keys(properties).length === 0) {
-    return fn.apply(thisArg, args);
-  }
-  const newContext = import_api5.context.active().setValue(PROPERTIES_CONTEXT_KEY, properties);
-  return import_api5.context.with(newContext, fn, thisArg, ...args);
+function callWithProperties(properties, fn, thisArg, ...args) {
+  const base = import_api5.context.active();
+  const prev = base.getValue(PROPERTIES_CONTEXT_KEY);
+  const merged = prev ? { ...prev, ...properties } : properties;
+  const next = base.setValue(PROPERTIES_CONTEXT_KEY, merged);
+  return import_api5.context.with(next, fn, thisArg, ...args);
 }
-function withSessionId(sessionId, fn, thisArg) {
+function withProperties(properties, fn, thisArg) {
   return function wrapped(...args) {
-    const base = import_api5.context.active();
-    const prev = base.getValue(PROPERTIES_CONTEXT_KEY);
-    const next = base.setValue(PROPERTIES_CONTEXT_KEY, prev ? { ...prev, [SESSION_ID]: sessionId } : { [SESSION_ID]: sessionId });
-    return import_api5.context.with(next, fn, thisArg ?? this, ...args);
+    return callWithProperties(
+      properties,
+      fn,
+      thisArg !== void 0 ? thisArg : this,
+      ...args
+    );
   };
 }
+function withSessionId(sessionId, fn, thisArg) {
+  return withProperties({ [SESSION_ID]: sessionId }, fn, thisArg);
+}
 function callWithSessionId(sessionId, fn, thisArg, ...args) {
-  return withProperties({ [SESSION_ID]: sessionId }, fn, thisArg, ...args);
+  return callWithProperties({ [SESSION_ID]: sessionId }, fn, thisArg, ...args);
 }
 
 // src/internal/sdk.ts
@@ -1874,12 +1754,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 import_sdk_node.NodeSDK({
       spanProcessors: [getSpanProcessor()],
       metricReader: getMetricsReader(),
-      resource: (0, import_resources2.resourceFromAttributes)({
-        "service.name": resolvedConfig.appName
-      }),
+      resource: (0, import_resources2.resourceFromAttributes)(resourceAttributes),
       instrumentations: manualInstrumentations
     });
     this._sdk.start();
@@ -2003,7 +1887,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,
@@ -2024,6 +1911,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
     }
   });
@@ -2044,6 +1932,7 @@ var init_exports = {};
   DEFAULT_PII_PATTERNS,
   LogLevel,
   SeverityNumber,
+  callWithProperties,
   callWithSessionId,
   detectRuntime,
   emitEvent,
@@ -2057,6 +1946,7 @@ var init_exports = {};
   maskAttributes,
   maskValue,
   setLogLevel,
+  withProperties,
   withSessionId
 });
 //# sourceMappingURL=index.cjs.map