@brizz/sdk 0.1.10 → 0.1.12

package/dist/index.cjs

@@ -62,7 +62,6 @@ var import_instrumentation_anthropic = require("@traceloop/instrumentation-anthr
 var import_instrumentation_bedrock = require("@traceloop/instrumentation-bedrock");
 var import_instrumentation_chromadb = require("@traceloop/instrumentation-chromadb");
 var import_instrumentation_cohere = require("@traceloop/instrumentation-cohere");
-var import_instrumentation_langchain = require("@traceloop/instrumentation-langchain");
 var import_instrumentation_llamaindex = require("@traceloop/instrumentation-llamaindex");
 var import_instrumentation_openai = require("@traceloop/instrumentation-openai");
 var import_instrumentation_pinecone = require("@traceloop/instrumentation-pinecone");
@@ -238,7 +237,6 @@ function loadGenAIInstrumentations() {
     { class: import_instrumentation_vertexai.VertexAIInstrumentation, name: "Vertex AI" },
     { class: import_instrumentation_bedrock.BedrockInstrumentation, name: "Bedrock" },
     { class: import_instrumentation_pinecone.PineconeInstrumentation, name: "Pinecone" },
-    { class: import_instrumentation_langchain.LangChainInstrumentation, name: "LangChain" },
     { class: import_instrumentation_llamaindex.LlamaIndexInstrumentation, name: "LlamaIndex" },
     { class: import_instrumentation_chromadb.ChromaDBInstrumentation, name: "ChromaDB" },
     { class: import_instrumentation_qdrant.QdrantInstrumentation, name: "Qdrant" },
@@ -280,7 +278,7 @@ function autoInitializeInstrumentations() {
 autoInitializeInstrumentations();
 
 // src/internal/sdk.ts
-var import_resources2 = require("@opentelemetry/resources");
+var import_resources3 = require("@opentelemetry/resources");
 var import_sdk_node = require("@opentelemetry/sdk-node");
 
 // src/internal/config.ts
@@ -346,7 +344,7 @@ function resolveConfig(options) {
       });
     } catch (error) {
       logger.error("Failed to parse BRIZZ_HEADERS environment variable", { error });
-      throw new Error("Invalid JSON in BRIZZ_HEADERS environment variable");
+      throw new Error("Invalid JSON in BRIZZ_HEADERS environment variable", { cause: error });
     }
   }
   logger.debug("Configuration resolved with environment variables", {
@@ -370,11 +368,11 @@ var BRIZZ_SDK_LANGUAGE = "brizz.sdk.language";
 var SDK_LANGUAGE = "typescript";
 
 // src/internal/instrumentation/registry.ts
+var import_openinference_instrumentation_langchain = require("@arizeai/openinference-instrumentation-langchain");
 var import_instrumentation_anthropic2 = require("@traceloop/instrumentation-anthropic");
 var import_instrumentation_bedrock2 = require("@traceloop/instrumentation-bedrock");
 var import_instrumentation_chromadb2 = require("@traceloop/instrumentation-chromadb");
 var import_instrumentation_cohere2 = require("@traceloop/instrumentation-cohere");
-var import_instrumentation_langchain2 = require("@traceloop/instrumentation-langchain");
 var import_instrumentation_llamaindex2 = require("@traceloop/instrumentation-llamaindex");
 var import_instrumentation_openai2 = require("@traceloop/instrumentation-openai");
 var import_instrumentation_pinecone2 = require("@traceloop/instrumentation-pinecone");
@@ -448,7 +446,6 @@ var InstrumentationRegistry = class _InstrumentationRegistry {
       },
       { class: import_instrumentation_bedrock2.BedrockInstrumentation, name: "Bedrock", module: this.manualModules?.bedrock },
       { class: import_instrumentation_pinecone2.PineconeInstrumentation, name: "Pinecone", module: this.manualModules?.pinecone },
-      { class: import_instrumentation_langchain2.LangChainInstrumentation, name: "LangChain", module: this.manualModules?.langchain },
       {
         class: import_instrumentation_llamaindex2.LlamaIndexInstrumentation,
         name: "LlamaIndex",
@@ -471,9 +468,65 @@ var InstrumentationRegistry = class _InstrumentationRegistry {
         }
       }
     }
+    if (this.manualModules?.langchain?.callbackManagerModule) {
+      try {
+        const lcInst = new import_openinference_instrumentation_langchain.LangChainInstrumentation();
+        lcInst.manuallyInstrument(this.manualModules.langchain.callbackManagerModule);
+        instrumentations.push(lcInst);
+        logger.debug("Manual instrumentation enabled for LangChain");
+      } catch (error) {
+        logger.error(`Failed to load LangChain instrumentation: ${String(error)}`);
+      }
+    }
   }
 };
 
+// src/internal/instrumentation/langchain-instrumentation.ts
+var import_node_module = require("module");
+var import_node_url = require("url");
+var import_openinference_instrumentation_langchain2 = require("@arizeai/openinference-instrumentation-langchain");
+var initialized = false;
+async function importCallbackManager() {
+  try {
+    return await import("@langchain/core/callbacks/manager");
+  } catch {
+  }
+  const resolvers = [
+    () => (0, import_node_module.createRequire)((0, import_node_url.pathToFileURL)(process.cwd() + "/package.json").href),
+    () => {
+      const main = process.argv[1];
+      if (!main) {
+        throw new Error("no main");
+      }
+      return (0, import_node_module.createRequire)((0, import_node_url.pathToFileURL)(main).href);
+    }
+  ];
+  for (const getRequire of resolvers) {
+    try {
+      const req = getRequire();
+      const cjsPath = req.resolve("@langchain/core/callbacks/manager");
+      const esmPath = cjsPath.replace(/\.cjs$/, ".js");
+      return await import((0, import_node_url.pathToFileURL)(esmPath).href);
+    } catch {
+    }
+  }
+  return null;
+}
+async function setupLangChainInstrumentation() {
+  if (initialized) {
+    return;
+  }
+  const callbackManagerModule = await importCallbackManager();
+  if (!callbackManagerModule) {
+    logger.debug("@langchain/core not found \u2014 skipping LangChain instrumentation");
+    return;
+  }
+  const instrumentation = new import_openinference_instrumentation_langchain2.LangChainInstrumentation();
+  instrumentation.manuallyInstrument(callbackManagerModule);
+  initialized = true;
+  logger.info("LangChain instrumentation enabled (OpenInference)");
+}
+
 // src/internal/log/logging.ts
 var import_api_logs = require("@opentelemetry/api-logs");
 var import_exporter_logs_otlp_http = require("@opentelemetry/exporter-logs-otlp-http");
@@ -482,7 +535,7 @@ var import_sdk_logs2 = require("@opentelemetry/sdk-logs");
 
 // src/internal/version.ts
 function getSDKVersion() {
-  return "0.1.10";
+  return "0.1.12";
 }
 
 // src/internal/log/processors/log-processor.ts
@@ -928,7 +981,9 @@ function getGroupedPattern(patternEntry) {
   try {
     new RegExp(patternEntry.pattern);
   } catch (error) {
-    throw new Error(`Invalid regex pattern '${patternEntry.pattern}': ${String(error)}`);
+    throw new Error(`Invalid regex pattern '${patternEntry.pattern}': ${String(error)}`, {
+      cause: error
+    });
   }
   return `(?<${name}>${patternEntry.pattern})`;
 }
@@ -1480,6 +1535,39 @@ function getMetricsReader() {
 // src/internal/trace/tracing.ts
 var import_exporter_trace_otlp_proto = require("@opentelemetry/exporter-trace-otlp-proto");
 
+// src/internal/trace/exporters/span-exporter.ts
+var import_resources2 = require("@opentelemetry/resources");
+var BrizzSpanExporter = class {
+  _delegate;
+  _brizzResource;
+  constructor(delegate, config) {
+    this._delegate = delegate;
+    const resourceAttrs = {
+      "service.name": config.appName,
+      [BRIZZ_SDK_VERSION]: getSDKVersion(),
+      [BRIZZ_SDK_LANGUAGE]: SDK_LANGUAGE
+    };
+    if (config.environment) {
+      resourceAttrs["deployment.environment"] = config.environment;
+    }
+    this._brizzResource = (0, import_resources2.resourceFromAttributes)(resourceAttrs);
+  }
+  export(spans, resultCallback) {
+    const patchedSpans = spans.map((span) => ({
+      ...span,
+      resource: span.resource.merge(this._brizzResource),
+      spanContext: span.spanContext.bind(span)
+    }));
+    this._delegate.export(patchedSpans, resultCallback);
+  }
+  async shutdown() {
+    return this._delegate.shutdown();
+  }
+  async forceFlush() {
+    return this._delegate.forceFlush?.();
+  }
+};
+
 // src/internal/trace/processors/span-processor.ts
 var import_api4 = require("@opentelemetry/api");
 var import_sdk_trace_base = require("@opentelemetry/sdk-trace-base");
@@ -1605,16 +1693,17 @@ var TracingModule = class _TracingModule {
     }
     if (config.customSpanExporter) {
       logger.debug("Using custom span exporter");
-      this.spanExporter = config.customSpanExporter;
+      this.spanExporter = new BrizzSpanExporter(config.customSpanExporter, config);
       logger.debug("Custom span exporter initialized successfully");
       return;
     }
     const tracesUrl = config.baseUrl.replace(/\/$/, "") + "/v1/traces";
     logger.debug("Initializing default OTLP span exporter", { url: tracesUrl });
-    this.spanExporter = new import_exporter_trace_otlp_proto.OTLPTraceExporter({
+    const otlpExporter = new import_exporter_trace_otlp_proto.OTLPTraceExporter({
       url: tracesUrl,
       headers: config.headers
     });
+    this.spanExporter = new BrizzSpanExporter(otlpExporter, config);
     logger.debug("OTLP span exporter initialized successfully");
   }
   /**
@@ -1824,6 +1913,7 @@ var _Brizz = class __Brizz {
       const resolvedConfig = resolveConfig(options);
       this.initializeModules(resolvedConfig);
       this.setupInstrumentation(options);
+      this.initLangChainInstrumentation();
       this.createAndStartNodeSDK(options, resolvedConfig);
       this._initialized = true;
       logger.info("Brizz SDK initialization completed successfully", {
@@ -1831,9 +1921,18 @@ var _Brizz = class __Brizz {
       });
     } catch (error) {
       logger.error("Failed to initialize Brizz SDK", { error });
-      throw new Error(`Failed to initialize SDK: ${String(error)}`);
+      throw new Error(`Failed to initialize SDK: ${String(error)}`, { cause: error });
     }
   }
+  /**
+   * Async LangChain instrumentation setup (fire-and-forget).
+   * @private
+   */
+  initLangChainInstrumentation() {
+    setupLangChainInstrumentation().catch((error) => {
+      logger.debug(`LangChain instrumentation skipped: ${String(error)}`);
+    });
+  }
   /**
    * Set up instrumentation registry and configure manual modules.
    * @private
@@ -1867,7 +1966,7 @@ var _Brizz = class __Brizz {
     this._sdk = new import_sdk_node.NodeSDK({
       spanProcessors: [getSpanProcessor()],
       metricReader: getMetricsReader(),
-      resource: (0, import_resources2.resourceFromAttributes)(resourceAttributes),
+      resource: (0, import_resources3.resourceFromAttributes)(resourceAttributes),
       instrumentations: manualInstrumentations
     });
     this._sdk.start();