@arizeai/phoenix-otel 0.0.1 → 0.2.0

package/dist/esm/createNoOpProvider.d.ts

@@ -0,0 +1,6 @@
+import { NodeTracerProvider } from "@opentelemetry/sdk-trace-node";
+/**
+ * For dry runs we create a provider that doesn't export traces.
+ */
+export declare function createNoOpProvider(): NodeTracerProvider;
+//# sourceMappingURL=createNoOpProvider.d.ts.map

package/dist/esm/createNoOpProvider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"createNoOpProvider.d.ts","sourceRoot":"","sources":["../../src/createNoOpProvider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,+BAA+B,CAAC;AAEnE;;GAEG;AACH,wBAAgB,kBAAkB,uBAIjC"}

package/dist/esm/createNoOpProvider.js

@@ -0,0 +1,9 @@
+import { NodeTracerProvider } from "@opentelemetry/sdk-trace-node";
+/**
+ * For dry runs we create a provider that doesn't export traces.
+ */
+export function createNoOpProvider() {
+    const provider = new NodeTracerProvider({});
+    return provider;
+}
+//# sourceMappingURL=createNoOpProvider.js.map

package/dist/esm/createNoOpProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"createNoOpProvider.js","sourceRoot":"","sources":["../../src/createNoOpProvider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,+BAA+B,CAAC;AAEnE;;GAEG;AACH,MAAM,UAAU,kBAAkB;IAChC,MAAM,QAAQ,GAAG,IAAI,kBAAkB,CAAC,EAAE,CAAC,CAAC;IAE5C,OAAO,QAAQ,CAAC;AAClB,CAAC"}

package/dist/esm/index.d.ts

@@ -1,3 +1,8 @@
-export { trace } from "@opentelemetry/api";
+export { trace, context, type DiagLogLevel, type Tracer, SpanStatusCode, } from "@opentelemetry/api";
+export { registerInstrumentations } from "@opentelemetry/instrumentation";
+export { type Instrumentation } from "@opentelemetry/instrumentation";
+export { type NodeTracerProvider } from "@opentelemetry/sdk-trace-node";
+export * from "./createNoOpProvider.js";
 export * from "./register.js";
+export * from "./utils.js";
 //# sourceMappingURL=index.d.ts.map

package/dist/esm/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAC3C,cAAc,YAAY,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,EACL,OAAO,EACP,KAAK,YAAY,EACjB,KAAK,MAAM,EACX,cAAc,GACf,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,wBAAwB,EAAE,MAAM,gCAAgC,CAAC;AAC1E,OAAO,EAAE,KAAK,eAAe,EAAE,MAAM,gCAAgC,CAAC;AACtE,OAAO,EAAE,KAAK,kBAAkB,EAAE,MAAM,+BAA+B,CAAC;AAGxE,cAAc,sBAAsB,CAAC;AACrC,cAAc,YAAY,CAAC;AAC3B,cAAc,SAAS,CAAC"}

package/dist/esm/index.js

@@ -1,3 +1,7 @@
-export { trace } from "@opentelemetry/api";
+export { trace, context, SpanStatusCode, } from "@opentelemetry/api";
+export { registerInstrumentations } from "@opentelemetry/instrumentation";
+// Phoenix abstractions
+export * from "./createNoOpProvider.js";
 export * from "./register.js";
+export * from "./utils.js";
 //# sourceMappingURL=index.js.map

package/dist/esm/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAC3C,cAAc,YAAY,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,EACL,OAAO,EAGP,cAAc,GACf,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,wBAAwB,EAAE,MAAM,gCAAgC,CAAC;AAI1E,uBAAuB;AACvB,cAAc,sBAAsB,CAAC;AACrC,cAAc,YAAY,CAAC;AAC3B,cAAc,SAAS,CAAC"}

package/dist/esm/register.d.ts

@@ -1,12 +1,18 @@
 import { NodeTracerProvider } from "@opentelemetry/sdk-trace-node";
-type RegisterParams = {
+import { Instrumentation } from "@opentelemetry/instrumentation";
+import { DiagLogLevel } from "@opentelemetry/api";
+export type Headers = Record<string, string>;
+/**
+ * Configuration parameters for registering Phoenix OpenTelemetry tracing
+ */
+export type RegisterParams = {
     /**
      * The project the spans should be associated to
      * @default "default"
      */
     projectName?: string;
     /**
-     * the URL to the phoenix server. Can be postfixed with tracing path.
+     * The URL to the phoenix server. Can be postfixed with tracing path.
      * If not provided, environment variables are checked.
      * @example "https://app.phoenix.arize.com"
      */
@@ -16,24 +22,64 @@ type RegisterParams = {
      * If not provided, environment variables are checked.
      */
     apiKey?: string;
+    /**
+     * Headers to be used when communicating with the OTLP collector
+     */
+    headers?: Headers;
     /**
      * Whether to use batching for span processing.
      * It is recommended to use batching in production environments
      * @default true
      */
     batch?: boolean;
+    /**
+     * A list of instrumentation to register.
+     * Note: this may only work with commonjs projects. ESM projects will require manually applying instrumentation.
+     */
+    instrumentations?: Instrumentation[];
     /**
      * Whether to set the tracer as a global provider.
      * @default true
      */
-    setGlobalTracerProvider?: boolean;
+    global?: boolean;
+    /**
+     * The diag log level to set for the built in DiagConsoleLogger instance.
+     * Omit to disable built in logging.
+     */
+    diagLogLevel?: DiagLogLevel;
 };
-export declare function register({ url: paramsUrl, apiKey: paramsApiKey, projectName, batch, setGlobalTracerProvider, }: RegisterParams): NodeTracerProvider;
 /**
- * A utility method to normalize the url to be http compatible.
- * Assumes we are using http over gRPC.
- * @param url the url to the phoenix server. May contain a slug
+ * Registers Phoenix OpenTelemetry tracing with the specified configuration
+ *
+ * @param params - Configuration parameters for Phoenix tracing
+ * @param params.url - The URL to the phoenix server. Can be postfixed with tracing path
+ * @param params.apiKey - The API key for the phoenix instance
+ * @param params.projectName - The project the spans should be associated to
+ * @param params.instrumentations - A list of instrumentation to register
+ * @param params.batch - Whether to use batching for span processing
+ * @param params.global - Whether to set the tracer as a global provider
+ * @param params.diagLogLevel - the diagnostics log level
+ * @returns {NodeTracerProvider} The configured NodeTracerProvider instance
+ *
+ * @example
+ * ```typescript
+ * import { register } from '@arizeai/phoenix-otel';
+ *
+ * const provider = register({
+ *   projectName: 'my-app',
+ *   url: 'https://app.phoenix.arize.com',
+ *   apiKey: 'your-api-key',
+ *   batch: true
+ * });
+ * ```
+ */
+export declare function register({ url: paramsUrl, apiKey: paramsApiKey, headers: paramsHeaders, projectName, instrumentations, batch, global, diagLogLevel, }: RegisterParams): NodeTracerProvider;
+/**
+ * A utility method to normalize the URL to be HTTP compatible.
+ * Assumes we are using HTTP over gRPC and ensures the URL ends with the correct traces endpoint.
+ *
+ * @param url - The URL to the phoenix server. May contain a slug
+ * @returns {string} The normalized URL with the '/v1/traces' endpoint
  */
 export declare function ensureCollectorEndpoint(url: string): string;
-export {};
 //# sourceMappingURL=register.d.ts.map

package/dist/esm/register.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"register.d.ts","sourceRoot":"","sources":["../../src/register.ts"],"names":[],"mappings":"AAOA,OAAO,EACL,kBAAkB,EAEnB,MAAM,+BAA+B,CAAC;AAGvC,KAAK,cAAc,GAAG;IACpB;;;OAGG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;;;OAIG;IACH,GAAG,CAAC,EAAE,MAAM,CAAC;IACb;;;OAGG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB;;;;OAIG;IACH,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB;;;OAGG;IACH,uBAAuB,CAAC,EAAE,OAAO,CAAC;CACnC,CAAC;AAEF,wBAAgB,QAAQ,CAAC,EACvB,GAAG,EAAE,SAAS,EACd,MAAM,EAAE,YAAY,EACpB,WAAuB,EACvB,KAAY,EACZ,uBAA8B,GAC/B,EAAE,cAAc,sBA+BhB;AAED;;;;GAIG;AACH,wBAAgB,uBAAuB,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAK3D"}
+{"version":3,"file":"register.d.ts","sourceRoot":"","sources":["../../src/register.ts"],"names":[],"mappings":"AAOA,OAAO,EACL,kBAAkB,EAEnB,MAAM,+BAA+B,CAAC;AAEvC,OAAO,EACL,eAAe,EAEhB,MAAM,gCAAgC,CAAC;AACxC,OAAO,EAA2B,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAE3E,MAAM,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;AAE7C;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B;;;OAGG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB;;;;OAIG;IACH,GAAG,CAAC,EAAE,MAAM,CAAC;IACb;;;OAGG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB;;OAEG;IACH,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB;;;;OAIG;IACH,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB;;;OAGG;IACH,gBAAgB,CAAC,EAAE,eAAe,EAAE,CAAC;IACrC;;;OAGG;IACH,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB;;;OAGG;IACH,YAAY,CAAC,EAAE,YAAY,CAAC;CAC7B,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,wBAAgB,QAAQ,CAAC,EACvB,GAAG,EAAE,SAAS,EACd,MAAM,EAAE,YAAY,EACpB,OAAO,EAAE,aAAkB,EAC3B,WAAuB,EACvB,gBAAgB,EAChB,KAAY,EACZ,MAAa,EACb,YAAY,GACb,EAAE,cAAc,GAAG,kBAAkB,CA0CrC;AAED;;;;;;GAMG;AACH,wBAAgB,uBAAuB,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAK3D"}

package/dist/esm/register.js

@@ -4,10 +4,42 @@ import { OTLPTraceExporter } from "@opentelemetry/exporter-trace-otlp-proto";
 import { resourceFromAttributes } from "@opentelemetry/resources";
 import { NodeTracerProvider, } from "@opentelemetry/sdk-trace-node";
 import { getEnvApiKey, getEnvCollectorURL } from "./config.js";
-export function register({ url: paramsUrl, apiKey: paramsApiKey, projectName = "default", batch = true, setGlobalTracerProvider = true, }) {
+import { registerInstrumentations, } from "@opentelemetry/instrumentation";
+import { diag, DiagConsoleLogger } from "@opentelemetry/api";
+/**
+ * Registers Phoenix OpenTelemetry tracing with the specified configuration
+ *
+ * @param params - Configuration parameters for Phoenix tracing
+ * @param params.url - The URL to the phoenix server. Can be postfixed with tracing path
+ * @param params.apiKey - The API key for the phoenix instance
+ * @param params.projectName - The project the spans should be associated to
+ * @param params.instrumentations - A list of instrumentation to register
+ * @param params.batch - Whether to use batching for span processing
+ * @param params.global - Whether to set the tracer as a global provider
+ * @param params.diagLogLevel - the diagnostics log level
+ * @returns {NodeTracerProvider} The configured NodeTracerProvider instance
+ *
+ * @example
+ * ```typescript
+ * import { register } from '@arizeai/phoenix-otel';
+ *
+ * const provider = register({
+ *   projectName: 'my-app',
+ *   url: 'https://app.phoenix.arize.com',
+ *   apiKey: 'your-api-key',
+ *   batch: true
+ * });
+ * ```
+ */
+export function register({ url: paramsUrl, apiKey: paramsApiKey, headers: paramsHeaders = {}, projectName = "default", instrumentations, batch = true, global = true, diagLogLevel, }) {
+    if (diagLogLevel) {
+        diag.setLogger(new DiagConsoleLogger(), diagLogLevel);
+    }
     const url = ensureCollectorEndpoint(paramsUrl || getEnvCollectorURL() || "http://localhost:6006");
     const apiKey = paramsApiKey || getEnvApiKey();
-    const headers = {};
+    const headers = Array.isArray(paramsHeaders)
+        ? Object.fromEntries(paramsHeaders)
+        : paramsHeaders;
     const configureHeaders = typeof apiKey == "string";
     if (configureHeaders) {
         headers["authorization"] = `Bearer ${apiKey}`;
@@ -29,15 +61,23 @@ export function register({ url: paramsUrl, apiKey: paramsApiKey, projectName = "
         }),
         spanProcessors: [spanProcessor],
     });
-    if (setGlobalTracerProvider) {
+    if (instrumentations) {
+        registerInstrumentations({
+            instrumentations,
+            tracerProvider: provider,
+        });
+    }
+    if (global) {
         provider.register();
     }
     return provider;
 }
 /**
- * A utility method to normalize the url to be http compatible.
- * Assumes we are using http over gRPC.
- * @param url the url to the phoenix server. May contain a slug
+ * A utility method to normalize the URL to be HTTP compatible.
+ * Assumes we are using HTTP over gRPC and ensures the URL ends with the correct traces endpoint.
+ *
+ * @param url - The URL to the phoenix server. May contain a slug
+ * @returns {string} The normalized URL with the '/v1/traces' endpoint
  */
 export function ensureCollectorEndpoint(url) {
     if (!url.includes("/v1/traces")) {

package/dist/esm/register.js.map

@@ -1 +1 @@
-{"version":3,"file":"register.js","sourceRoot":"","sources":["../../src/register.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,wBAAwB,EAAE,MAAM,6CAA6C,CAAC;AACvF,OAAO,EACL,+BAA+B,EAC/B,gCAAgC,GACjC,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAE,iBAAiB,EAAE,MAAM,0CAA0C,CAAC;AAC7E,OAAO,EAAE,sBAAsB,EAAE,MAAM,0BAA0B,CAAC;AAClE,OAAO,EACL,kBAAkB,GAEnB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAE,YAAY,EAAE,kBAAkB,EAAE,MAAM,UAAU,CAAC;AAgC5D,MAAM,UAAU,QAAQ,CAAC,EACvB,GAAG,EAAE,SAAS,EACd,MAAM,EAAE,YAAY,EACpB,WAAW,GAAG,SAAS,EACvB,KAAK,GAAG,IAAI,EACZ,uBAAuB,GAAG,IAAI,GACf;IACf,MAAM,GAAG,GAAG,uBAAuB,CACjC,SAAS,IAAI,kBAAkB,EAAE,IAAI,uBAAuB,CAC7D,CAAC;IACF,MAAM,MAAM,GAAG,YAAY,IAAI,YAAY,EAAE,CAAC;IAC9C,MAAM,OAAO,GAA2B,EAAE,CAAC;IAC3C,MAAM,gBAAgB,GAAG,OAAO,MAAM,IAAI,QAAQ,CAAC;IACnD,IAAI,gBAAgB,EAAE,CAAC;QACrB,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,MAAM,EAAE,CAAC;IAChD,CAAC;IACD,MAAM,QAAQ,GAAG,IAAI,iBAAiB,CAAC;QACrC,GAAG;QACH,OAAO;KACR,CAAC,CAAC;IACH,IAAI,aAA4B,CAAC;IACjC,IAAI,KAAK,EAAE,CAAC;QACV,aAAa,GAAG,IAAI,+BAA+B,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;IACpE,CAAC;SAAM,CAAC;QACN,aAAa,GAAG,IAAI,gCAAgC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;IACrE,CAAC;IACD,MAAM,QAAQ,GAAG,IAAI,kBAAkB,CAAC;QACtC,QAAQ,EAAE,sBAAsB,CAAC;YAC/B,CAAC,wBAAwB,CAAC,EAAE,WAAW;SACxC,CAAC;QACF,cAAc,EAAE,CAAC,aAAa,CAAC;KAChC,CAAC,CAAC;IAEH,IAAI,uBAAuB,EAAE,CAAC;QAC5B,QAAQ,CAAC,QAAQ,EAAE,CAAC;IACtB,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,uBAAuB,CAAC,GAAW;IACjD,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;QAChC,OAAO,IAAI,GAAG,CAAC,YAAY,EAAE,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC/C,CAAC;IACD,OAAO,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC;AACjC,CAAC"}
+{"version":3,"file":"register.js","sourceRoot":"","sources":["../../src/register.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,wBAAwB,EAAE,MAAM,6CAA6C,CAAC;AACvF,OAAO,EACL,+BAA+B,EAC/B,gCAAgC,GACjC,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAE,iBAAiB,EAAE,MAAM,0CAA0C,CAAC;AAC7E,OAAO,EAAE,sBAAsB,EAAE,MAAM,0BAA0B,CAAC;AAClE,OAAO,EACL,kBAAkB,GAEnB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAE,YAAY,EAAE,kBAAkB,EAAE,MAAM,UAAU,CAAC;AAC5D,OAAO,EAEL,wBAAwB,GACzB,MAAM,gCAAgC,CAAC;AACxC,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAgB,MAAM,oBAAoB,CAAC;AAmD3E;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,MAAM,UAAU,QAAQ,CAAC,EACvB,GAAG,EAAE,SAAS,EACd,MAAM,EAAE,YAAY,EACpB,OAAO,EAAE,aAAa,GAAG,EAAE,EAC3B,WAAW,GAAG,SAAS,EACvB,gBAAgB,EAChB,KAAK,GAAG,IAAI,EACZ,MAAM,GAAG,IAAI,EACb,YAAY,GACG;IACf,IAAI,YAAY,EAAE,CAAC;QACjB,IAAI,CAAC,SAAS,CAAC,IAAI,iBAAiB,EAAE,EAAE,YAAY,CAAC,CAAC;IACxD,CAAC;IACD,MAAM,GAAG,GAAG,uBAAuB,CACjC,SAAS,IAAI,kBAAkB,EAAE,IAAI,uBAAuB,CAC7D,CAAC;IACF,MAAM,MAAM,GAAG,YAAY,IAAI,YAAY,EAAE,CAAC;IAC9C,MAAM,OAAO,GAAY,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC;QACnD,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC;QACnC,CAAC,CAAC,aAAa,CAAC;IAClB,MAAM,gBAAgB,GAAG,OAAO,MAAM,IAAI,QAAQ,CAAC;IACnD,IAAI,gBAAgB,EAAE,CAAC;QACrB,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,MAAM,EAAE,CAAC;IAChD,CAAC;IACD,MAAM,QAAQ,GAAG,IAAI,iBAAiB,CAAC;QACrC,GAAG;QACH,OAAO;KACR,CAAC,CAAC;IACH,IAAI,aAA4B,CAAC;IACjC,IAAI,KAAK,EAAE,CAAC;QACV,aAAa,GAAG,IAAI,+BAA+B,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;IACpE,CAAC;SAAM,CAAC;QACN,aAAa,GAAG,IAAI,gCAAgC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;IACrE,CAAC;IACD,MAAM,QAAQ,GAAG,IAAI,kBAAkB,CAAC;QACtC,QAAQ,EAAE,sBAAsB,CAAC;YAC/B,CAAC,wBAAwB,CAAC,EAAE,WAAW;SACxC,CAAC;QACF,cAAc,EAAE,CAAC,aAAa,CAAC;KAChC,CAAC,CAAC;IAEH,IAAI,gBAAgB,EAAE,CAAC;QACrB,wBAAwB,CAAC;YACvB,gBAAgB;YAChB,cAAc,EAAE,QAAQ;SACzB,CAAC,CAAC;IACL,CAAC;IACD,IAAI,MAAM,EAAE,CAAC;QACX,QAAQ,CAAC,QAAQ,EAAE,CAAC;IACtB,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,uBAAuB,CAAC,GAAW;IACjD,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;QAChC,OAAO,IAAI,GAAG,CAAC,YAAY,EAAE,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC/C,CAAC;IACD,OAAO,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC;AACjC,CAAC"}