@arizeai/phoenix-client 4.2.0 → 5.1.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@arizeai/phoenix-client",
-  "version": "4.2.0",
+  "version": "5.1.0",
   "description": "A client for the Phoenix API",
   "main": "dist/src/index.js",
   "module": "dist/esm/index.js",
@@ -72,25 +72,19 @@
   "dependencies": {
     "@arizeai/openinference-semantic-conventions": "^1.1.0",
     "@arizeai/openinference-vercel": "^2.3.1",
-    "@opentelemetry/api": "^1.9.0",
-    "@opentelemetry/core": "^1.25.1",
-    "@opentelemetry/exporter-trace-otlp-proto": "^0.57.2",
-    "@opentelemetry/instrumentation": "^0.57.2",
-    "@opentelemetry/resources": "^2.0.0",
-    "@opentelemetry/sdk-trace-base": "^1.30.1",
-    "@opentelemetry/sdk-trace-node": "^1.30.1",
     "async": "^3.2.6",
     "openapi-fetch": "^0.12.5",
     "tiny-invariant": "^1.3.3",
     "zod": "^3.24.2",
-    "zod-to-json-schema": "^3.24.3"
+    "zod-to-json-schema": "^3.24.3",
+    "@arizeai/phoenix-otel": "0.2.0"
   },
   "engines": {
     "node": ">=18"
   },
   "optionalDependencies": {
     "@anthropic-ai/sdk": "^0.35.0",
-    "ai": "^5.0.38",
+    "ai": "^5.0.76",
     "openai": "^5.12.1"
   },
   "scripts": {

package/src/__generated__/api/v1.ts

@@ -939,6 +939,11 @@ export interface components {
              * @description ID of the dataset version over which the experiment will be run (if omitted, the latest version will be used)
              */
             version_id?: string | null;
+            /**
+             * Splits
+             * @description List of dataset split identifiers (GlobalIDs or names) to filter by
+             */
+            splits?: string[] | null;
             /**
              * Repetitions
              * @description Number of times the experiment should be repeated for each example
@@ -1397,6 +1402,8 @@ export interface components {
             dataset_id: string;
             /** Version Id */
             version_id: string;
+            /** Filtered Splits */
+            filtered_splits?: string[];
             /** Examples */
             examples: components["schemas"]["DatasetExample"][];
         };
@@ -3437,6 +3444,8 @@ export interface operations {
             query?: {
                 /** @description The ID of the dataset version (if omitted, returns data from the latest version) */
                 version_id?: string | null;
+                /** @description List of dataset split identifiers (GlobalIDs or names) to filter by */
+                split?: string[] | null;
             };
             header?: never;
             path: {

package/src/datasets/getDataset.ts

@@ -5,24 +5,22 @@ import { getDatasetExamples } from "./getDatasetExamples";
 import { getDatasetInfo } from "./getDatasetInfo";
 
 export type GetDatasetParams = ClientFn & {
+  /** Dataset selector (ID or name) */
   dataset: DatasetSelector;
-  versionId?: string;
 };
 
 /**
  * Get dataset info and examples from the dataset
  * @param dataset - Dataset selector (ID or name)
- * @param versionId - Optional specific version ID (if omitted, returns data from the latest version)
  */
 export async function getDataset({
   client: _client,
   dataset,
-  versionId,
 }: GetDatasetParams): Promise<Dataset> {
   const client = _client || createClient();
   const [datasetInfo, datasetExamples] = await Promise.all([
     getDatasetInfo({ client, dataset }),
-    getDatasetExamples({ client, dataset, versionId }),
+    getDatasetExamples({ client, dataset }),
   ]);
   return {
     ...datasetInfo,

package/src/datasets/getDatasetExamples.ts

@@ -5,44 +5,44 @@ import { DatasetSelector, DatasetExamples } from "../types/datasets";
 import { getDatasetInfoByName } from "./getDatasetInfoByName";
 
 export type GetDatasetExamplesParams = ClientFn & {
+  /** Dataset selector (ID, name, or version ID) */
   dataset: DatasetSelector;
-  versionId?: string;
 };
 
 /**
  * Get examples from a dataset
- * @param dataset - Dataset selector (ID, name, or version ID)
- * @param versionId - Optional specific version ID (ignored if dataset selector is datasetVersionId)
+ * @param dataset - Dataset selector (ID, name, version ID, or splits)
+ * @returns Dataset examples
  */
 export async function getDatasetExamples({
   client: _client,
-  dataset,
-  versionId,
+  dataset: datasetSelector,
 }: GetDatasetExamplesParams): Promise<DatasetExamples> {
   const client = _client || createClient();
 
   let datasetId: string;
 
-  if ("datasetName" in dataset) {
+  if ("datasetName" in datasetSelector) {
     const datasetInfo = await getDatasetInfoByName({
       client,
-      datasetName: dataset.datasetName,
+      datasetName: datasetSelector.datasetName,
     });
     datasetId = datasetInfo.id;
   } else {
-    datasetId = dataset.datasetId;
+    datasetId = datasetSelector.datasetId;
   }
 
+  const { versionId, splits } = datasetSelector;
+
   const response = await client.GET("/v1/datasets/{id}/examples", {
     params: {
       path: {
         id: datasetId,
       },
-      query: versionId
-        ? {
-            version_id: versionId,
-          }
-        : undefined,
+      query: {
+        ...(versionId ? { version_id: versionId } : {}),
+        ...(splits ? { split: splits } : {}),
+      },
     },
   });
 

package/src/experiments/getExperimentInfo.ts

@@ -18,16 +18,14 @@ export async function getExperimentInfo({
   experimentId: experiment_id,
 }: GetExperimentParams): Promise<ExperimentInfo> {
   const client = _client || createClient();
-  const { data: { data: experimentData } = {} } = await client.GET(
-    "/v1/experiments/{experiment_id}",
-    {
+  const { data: { data: experimentData } = { data: undefined } } =
+    await client.GET("/v1/experiments/{experiment_id}", {
       params: {
         path: {
           experiment_id,
         },
       },
-    }
-  );
+    });
   invariant(experimentData, "Failed to get experiment");
   return {
     id: experimentData.id,

package/src/experiments/runExperiment.ts

@@ -22,27 +22,29 @@ import { getDataset } from "../datasets/getDataset";
 import { pluralize } from "../utils/pluralize";
 import { promisifyResult } from "../utils/promisifyResult";
 import { AnnotatorKind } from "../types/annotations";
-import { createProvider, createNoOpProvider } from "./instrumentation";
 import {
   type DiagLogLevel,
   SpanStatusCode,
   Tracer,
   trace,
-} from "@opentelemetry/api";
+  NodeTracerProvider,
+  objectAsAttributes,
+  createNoOpProvider,
+  register,
+} from "@arizeai/phoenix-otel";
 import {
   MimeType,
   OpenInferenceSpanKind,
   SemanticConventions,
 } from "@arizeai/openinference-semantic-conventions";
 import { ensureString } from "../utils/ensureString";
-import type { NodeTracerProvider } from "@opentelemetry/sdk-trace-node";
-import { objectAsAttributes } from "../utils/objectAsAttributes";
 import {
   getDatasetUrl,
   getDatasetExperimentsUrl,
   getExperimentUrl,
 } from "../utils/urlUtils";
 import assert from "assert";
+import { toObjectHeaders } from "../utils/toObjectHeaders";
 
 /**
  * Validate that a repetition is valid
@@ -160,7 +162,7 @@ export async function runExperiment({
   experimentDescription,
   experimentMetadata = {},
   client: _client,
-  dataset: DatasetSelector,
+  dataset: datasetSelector,
   task,
   evaluators,
   logger = console,
@@ -180,7 +182,10 @@ export async function runExperiment({
   let provider: NodeTracerProvider | undefined;
   const isDryRun = typeof dryRun === "number" || dryRun === true;
   const client = _client ?? createClient();
-  const dataset = await getDataset({ dataset: DatasetSelector, client });
+  const dataset = await getDataset({
+    dataset: datasetSelector,
+    client,
+  });
   invariant(dataset, `Dataset not found`);
   invariant(dataset.examples.length > 0, `Dataset has no examples`);
   const nExamples =
@@ -197,6 +202,8 @@ export async function runExperiment({
       id: localId(),
       datasetId: dataset.id,
       datasetVersionId: dataset.versionId,
+      // @todo: the dataset should return splits in response body
+      datasetSplits: datasetSelector?.splits ?? [],
       projectName,
       metadata: experimentMetadata,
     };
@@ -215,6 +222,11 @@ export async function runExperiment({
           metadata: experimentMetadata,
           project_name: projectName,
           repetitions,
+          // @todo: the dataset should return splits in response body
+          ...(datasetSelector?.splits
+            ? { splits: datasetSelector.splits }
+            : {}),
+          ...(dataset?.versionId ? { version_id: dataset.versionId } : {}),
         },
       })
       .then((res) => res.data?.data);
@@ -224,6 +236,8 @@ export async function runExperiment({
       id: experimentResponse.id,
       datasetId: experimentResponse.dataset_id,
       datasetVersionId: experimentResponse.dataset_version_id,
+      // @todo: the dataset should return splits in response body
+      datasetSplits: datasetSelector?.splits ?? [],
       projectName,
       metadata: experimentResponse.metadata,
     };
@@ -233,17 +247,18 @@ export async function runExperiment({
       baseUrl,
       "Phoenix base URL not found. Please set PHOENIX_HOST or set baseUrl on the client."
     );
-    provider = createProvider({
+
+    provider = register({
       projectName,
-      baseUrl,
-      headers: client.config.headers ?? {},
-      useBatchSpanProcessor,
+      url: baseUrl,
+      headers: client.config.headers
+        ? toObjectHeaders(client.config.headers)
+        : undefined,
+      batch: useBatchSpanProcessor,
       diagLogLevel,
+      global: setGlobalTracerProvider,
     });
-    // Register the provider
-    if (setGlobalTracerProvider) {
-      provider.register();
-    }
+
     taskTracer = provider.getTracer(projectName);
   }
   if (!record) {
@@ -536,16 +551,16 @@ export async function evaluateExperiment({
   if (paramsTracerProvider) {
     provider = paramsTracerProvider;
   } else if (!isDryRun) {
-    provider = createProvider({
+    provider = register({
       projectName: "evaluators",
-      baseUrl,
-      headers: client.config.headers ?? {},
-      useBatchSpanProcessor,
+      url: baseUrl,
+      headers: client.config.headers
+        ? toObjectHeaders(client.config.headers)
+        : undefined,
+      batch: useBatchSpanProcessor,
       diagLogLevel,
+      global: setGlobalTracerProvider,
     });
-    if (setGlobalTracerProvider) {
-      provider.register();
-    }
   } else {
     provider = createNoOpProvider();
   }
@@ -557,7 +572,11 @@ export async function evaluateExperiment({
       ? Math.min(dryRun, Object.keys(experiment.runs).length)
       : Object.keys(experiment.runs).length;
   const dataset = await getDataset({
-    dataset: { datasetId: experiment.datasetId },
+    dataset: {
+      datasetId: experiment.datasetId,
+      versionId: experiment.datasetVersionId,
+      splits: experiment.datasetSplits,
+    },
     client,
   });
   invariant(dataset, `Dataset "${experiment.datasetId}" not found`);

package/src/types/datasets.ts

@@ -1,17 +1,13 @@
 import { Node } from "./core";
 
-/**
- * A dataset can be identified by its datasetId, datasetName, or datasetVersionId
- */
-export type DatasetSelector = { datasetId: string } | { datasetName: string };
+type DatasetSelectorBase = { versionId?: string; splits?: string[] };
 
 /**
- * Parameters for selecting a specific version of a dataset
+ * A dataset can be identified by its datasetId, datasetName, or datasetVersionId
  */
-export interface DatasetVersionSelector {
-  dataset: DatasetSelector;
-  versionId?: string;
-}
+export type DatasetSelector =
+  | (DatasetSelectorBase & { datasetId: string })
+  | (DatasetSelectorBase & { datasetName: string });
 
 /**
  * Overview information about a dataset

package/src/types/experiments.ts

@@ -8,6 +8,8 @@ import { Example } from "./datasets";
 export interface ExperimentInfo extends Node {
   datasetId: string;
   datasetVersionId: string;
+  // @todo: mark this as required when experiment API returns it
+  datasetSplits?: string[];
   /**
    * The project under which the experiment task traces are recorded
    */

package/src/utils/toObjectHeaders.ts

@@ -0,0 +1,22 @@
+import { HeadersOptions } from "openapi-fetch";
+
+/**
+ * A utility function that simplifies the headers for passing to other clients
+ */
+export function toObjectHeaders(
+  headers: HeadersOptions
+): Record<string, string> {
+  if (Array.isArray(headers)) {
+    return Object.fromEntries(headers);
+  }
+
+  const objectHeaders: Record<string, string> = {};
+  for (const [key, value] of Object.entries(headers)) {
+    if (value !== null && value !== undefined) {
+      objectHeaders[key] = Array.isArray(value)
+        ? value.join(", ")
+        : String(value);
+    }
+  }
+  return objectHeaders;
+}

package/dist/esm/experiments/instrumentation.d.ts

@@ -1,29 +0,0 @@
-import { DiagLogLevel } from "@opentelemetry/api";
-import { NodeTracerProvider } from "@opentelemetry/sdk-trace-node";
-import { HeadersOptions } from "openapi-fetch";
-/**
- * Creates a provider that exports traces to Phoenix.
- */
-export declare function createProvider({ projectName, baseUrl, headers, useBatchSpanProcessor, diagLogLevel, }: {
-    projectName: string;
-    headers: HeadersOptions;
-    /**
-     * Whether to use batching for the span processor.
-     * @default true
-     */
-    useBatchSpanProcessor: boolean;
-    /**
-     * The base URL of the Phoenix. Doesn't include the /v1/traces path.
-     */
-    baseUrl: string;
-    /**
-     * The diag log level to set for the built in DiagConsoleLogger instance.
-     * Omit to disable built in logging.
-     */
-    diagLogLevel?: DiagLogLevel;
-}): NodeTracerProvider;
-/**
- * For dry runs we create a provider that doesn't export traces.
- */
-export declare function createNoOpProvider(): NodeTracerProvider;
-//# sourceMappingURL=instrumentation.d.ts.map

package/dist/esm/experiments/instrumentation.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"instrumentation.d.ts","sourceRoot":"","sources":["../../../src/experiments/instrumentation.ts"],"names":[],"mappings":"AAAA,OAAO,EAA2B,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAG3E,OAAO,EACL,kBAAkB,EAEnB,MAAM,+BAA+B,CAAC;AAEvC,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAM/C;;GAEG;AACH,wBAAgB,cAAc,CAAC,EAC7B,WAAW,EACX,OAAO,EACP,OAAO,EACP,qBAA4B,EAC5B,YAAY,GACb,EAAE;IACD,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,cAAc,CAAC;IACxB;;;OAGG;IACH,qBAAqB,EAAE,OAAO,CAAC;IAC/B;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;;OAGG;IACH,YAAY,CAAC,EAAE,YAAY,CAAC;CAC7B,sBAyBA;AAED;;GAEG;AACH,wBAAgB,kBAAkB,uBAIjC"}

package/dist/esm/experiments/instrumentation.js

@@ -1,40 +0,0 @@
-import { diag, DiagConsoleLogger } from "@opentelemetry/api";
-import { OTLPTraceExporter } from "@opentelemetry/exporter-trace-otlp-proto";
-import { resourceFromAttributes } from "@opentelemetry/resources";
-import { NodeTracerProvider, } from "@opentelemetry/sdk-trace-node";
-import { SEMRESATTRS_PROJECT_NAME } from "@arizeai/openinference-semantic-conventions";
-import { OpenInferenceBatchSpanProcessor, OpenInferenceSimpleSpanProcessor, } from "@arizeai/openinference-vercel";
-/**
- * Creates a provider that exports traces to Phoenix.
- */
-export function createProvider({ projectName, baseUrl, headers, useBatchSpanProcessor = true, diagLogLevel, }) {
-    if (diagLogLevel) {
-        diag.setLogger(new DiagConsoleLogger(), diagLogLevel);
-    }
-    const exporter = new OTLPTraceExporter({
-        url: `${baseUrl}/v1/traces`,
-        headers: Array.isArray(headers) ? Object.fromEntries(headers) : headers,
-    });
-    let spanProcessor;
-    if (useBatchSpanProcessor) {
-        spanProcessor = new OpenInferenceBatchSpanProcessor({ exporter });
-    }
-    else {
-        spanProcessor = new OpenInferenceSimpleSpanProcessor({ exporter });
-    }
-    const provider = new NodeTracerProvider({
-        resource: resourceFromAttributes({
-            [SEMRESATTRS_PROJECT_NAME]: projectName,
-        }),
-        spanProcessors: [spanProcessor],
-    });
-    return provider;
-}
-/**
- * For dry runs we create a provider that doesn't export traces.
- */
-export function createNoOpProvider() {
-    const provider = new NodeTracerProvider({});
-    return provider;
-}
-//# sourceMappingURL=instrumentation.js.map

package/dist/esm/experiments/instrumentation.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"instrumentation.js","sourceRoot":"","sources":["../../../src/experiments/instrumentation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAgB,MAAM,oBAAoB,CAAC;AAC3E,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,wBAAwB,EAAE,MAAM,6CAA6C,CAAC;AAEvF,OAAO,EACL,+BAA+B,EAC/B,gCAAgC,GACjC,MAAM,+BAA+B,CAAC;AAEvC;;GAEG;AACH,MAAM,UAAU,cAAc,CAAC,EAC7B,WAAW,EACX,OAAO,EACP,OAAO,EACP,qBAAqB,GAAG,IAAI,EAC5B,YAAY,GAkBb;IACC,IAAI,YAAY,EAAE,CAAC;QACjB,IAAI,CAAC,SAAS,CAAC,IAAI,iBAAiB,EAAE,EAAE,YAAY,CAAC,CAAC;IACxD,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,iBAAiB,CAAC;QACrC,GAAG,EAAE,GAAG,OAAO,YAAY;QAC3B,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO;KACxE,CAAC,CAAC;IAEH,IAAI,aAA4B,CAAC;IACjC,IAAI,qBAAqB,EAAE,CAAC;QAC1B,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;IAED,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,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB;IAChC,MAAM,QAAQ,GAAG,IAAI,kBAAkB,CAAC,EAAE,CAAC,CAAC;IAE5C,OAAO,QAAQ,CAAC;AAClB,CAAC"}

package/dist/esm/utils/objectAsAttributes.d.ts

@@ -1,3 +0,0 @@
-import { AttributeValue } from "@opentelemetry/api";
-export declare function objectAsAttributes<T extends Record<string, unknown>>(obj: T): Record<string, AttributeValue>;
-//# sourceMappingURL=objectAsAttributes.d.ts.map

package/dist/esm/utils/objectAsAttributes.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"objectAsAttributes.d.ts","sourceRoot":"","sources":["../../../src/utils/objectAsAttributes.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAEpD,wBAAgB,kBAAkB,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAClE,GAAG,EAAE,CAAC,GACL,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAIhC"}

package/dist/esm/utils/objectAsAttributes.js

@@ -1,4 +0,0 @@
-export function objectAsAttributes(obj) {
-    return Object.fromEntries(Object.entries(obj).filter(([_, value]) => value !== null));
-}
-//# sourceMappingURL=objectAsAttributes.js.map

package/dist/esm/utils/objectAsAttributes.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"objectAsAttributes.js","sourceRoot":"","sources":["../../../src/utils/objectAsAttributes.ts"],"names":[],"mappings":"AAEA,MAAM,UAAU,kBAAkB,CAChC,GAAM;IAEN,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC,CACzB,CAAC;AACtC,CAAC"}

package/dist/src/experiments/instrumentation.d.ts

@@ -1,29 +0,0 @@
-import { DiagLogLevel } from "@opentelemetry/api";
-import { NodeTracerProvider } from "@opentelemetry/sdk-trace-node";
-import { HeadersOptions } from "openapi-fetch";
-/**
- * Creates a provider that exports traces to Phoenix.
- */
-export declare function createProvider({ projectName, baseUrl, headers, useBatchSpanProcessor, diagLogLevel, }: {
-    projectName: string;
-    headers: HeadersOptions;
-    /**
-     * Whether to use batching for the span processor.
-     * @default true
-     */
-    useBatchSpanProcessor: boolean;
-    /**
-     * The base URL of the Phoenix. Doesn't include the /v1/traces path.
-     */
-    baseUrl: string;
-    /**
-     * The diag log level to set for the built in DiagConsoleLogger instance.
-     * Omit to disable built in logging.
-     */
-    diagLogLevel?: DiagLogLevel;
-}): NodeTracerProvider;
-/**
- * For dry runs we create a provider that doesn't export traces.
- */
-export declare function createNoOpProvider(): NodeTracerProvider;
-//# sourceMappingURL=instrumentation.d.ts.map

package/dist/src/experiments/instrumentation.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"instrumentation.d.ts","sourceRoot":"","sources":["../../../src/experiments/instrumentation.ts"],"names":[],"mappings":"AAAA,OAAO,EAA2B,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAG3E,OAAO,EACL,kBAAkB,EAEnB,MAAM,+BAA+B,CAAC;AAEvC,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAM/C;;GAEG;AACH,wBAAgB,cAAc,CAAC,EAC7B,WAAW,EACX,OAAO,EACP,OAAO,EACP,qBAA4B,EAC5B,YAAY,GACb,EAAE;IACD,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,cAAc,CAAC;IACxB;;;OAGG;IACH,qBAAqB,EAAE,OAAO,CAAC;IAC/B;;OAEG;IACH,OAAO,EAAE,MAAM,CAAC;IAChB;;;OAGG;IACH,YAAY,CAAC,EAAE,YAAY,CAAC;CAC7B,sBAyBA;AAED;;GAEG;AACH,wBAAgB,kBAAkB,uBAIjC"}

package/dist/src/experiments/instrumentation.js

@@ -1,44 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.createProvider = createProvider;
-exports.createNoOpProvider = createNoOpProvider;
-const api_1 = require("@opentelemetry/api");
-const exporter_trace_otlp_proto_1 = require("@opentelemetry/exporter-trace-otlp-proto");
-const resources_1 = require("@opentelemetry/resources");
-const sdk_trace_node_1 = require("@opentelemetry/sdk-trace-node");
-const openinference_semantic_conventions_1 = require("@arizeai/openinference-semantic-conventions");
-const openinference_vercel_1 = require("@arizeai/openinference-vercel");
-/**
- * Creates a provider that exports traces to Phoenix.
- */
-function createProvider({ projectName, baseUrl, headers, useBatchSpanProcessor = true, diagLogLevel, }) {
-    if (diagLogLevel) {
-        api_1.diag.setLogger(new api_1.DiagConsoleLogger(), diagLogLevel);
-    }
-    const exporter = new exporter_trace_otlp_proto_1.OTLPTraceExporter({
-        url: `${baseUrl}/v1/traces`,
-        headers: Array.isArray(headers) ? Object.fromEntries(headers) : headers,
-    });
-    let spanProcessor;
-    if (useBatchSpanProcessor) {
-        spanProcessor = new openinference_vercel_1.OpenInferenceBatchSpanProcessor({ exporter });
-    }
-    else {
-        spanProcessor = new openinference_vercel_1.OpenInferenceSimpleSpanProcessor({ exporter });
-    }
-    const provider = new sdk_trace_node_1.NodeTracerProvider({
-        resource: (0, resources_1.resourceFromAttributes)({
-            [openinference_semantic_conventions_1.SEMRESATTRS_PROJECT_NAME]: projectName,
-        }),
-        spanProcessors: [spanProcessor],
-    });
-    return provider;
-}
-/**
- * For dry runs we create a provider that doesn't export traces.
- */
-function createNoOpProvider() {
-    const provider = new sdk_trace_node_1.NodeTracerProvider({});
-    return provider;
-}
-//# sourceMappingURL=instrumentation.js.map

package/dist/src/experiments/instrumentation.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"instrumentation.js","sourceRoot":"","sources":["../../../src/experiments/instrumentation.ts"],"names":[],"mappings":";;AAiBA,wCAgDC;AAKD,gDAIC;AA1ED,4CAA2E;AAC3E,wFAA6E;AAC7E,wDAAkE;AAClE,kEAGuC;AACvC,oGAAuF;AAEvF,wEAGuC;AAEvC;;GAEG;AACH,SAAgB,cAAc,CAAC,EAC7B,WAAW,EACX,OAAO,EACP,OAAO,EACP,qBAAqB,GAAG,IAAI,EAC5B,YAAY,GAkBb;IACC,IAAI,YAAY,EAAE,CAAC;QACjB,UAAI,CAAC,SAAS,CAAC,IAAI,uBAAiB,EAAE,EAAE,YAAY,CAAC,CAAC;IACxD,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,6CAAiB,CAAC;QACrC,GAAG,EAAE,GAAG,OAAO,YAAY;QAC3B,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO;KACxE,CAAC,CAAC;IAEH,IAAI,aAA4B,CAAC;IACjC,IAAI,qBAAqB,EAAE,CAAC;QAC1B,aAAa,GAAG,IAAI,sDAA+B,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;IACpE,CAAC;SAAM,CAAC;QACN,aAAa,GAAG,IAAI,uDAAgC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;IACrE,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,mCAAkB,CAAC;QACtC,QAAQ,EAAE,IAAA,kCAAsB,EAAC;YAC/B,CAAC,6DAAwB,CAAC,EAAE,WAAW;SACxC,CAAC;QACF,cAAc,EAAE,CAAC,aAAa,CAAC;KAChC,CAAC,CAAC;IAEH,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED;;GAEG;AACH,SAAgB,kBAAkB;IAChC,MAAM,QAAQ,GAAG,IAAI,mCAAkB,CAAC,EAAE,CAAC,CAAC;IAE5C,OAAO,QAAQ,CAAC;AAClB,CAAC"}

package/dist/src/utils/objectAsAttributes.d.ts

@@ -1,3 +0,0 @@
-import { AttributeValue } from "@opentelemetry/api";
-export declare function objectAsAttributes<T extends Record<string, unknown>>(obj: T): Record<string, AttributeValue>;
-//# sourceMappingURL=objectAsAttributes.d.ts.map

package/dist/src/utils/objectAsAttributes.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"objectAsAttributes.d.ts","sourceRoot":"","sources":["../../../src/utils/objectAsAttributes.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAEpD,wBAAgB,kBAAkB,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAClE,GAAG,EAAE,CAAC,GACL,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAIhC"}

package/dist/src/utils/objectAsAttributes.js

@@ -1,7 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.objectAsAttributes = objectAsAttributes;
-function objectAsAttributes(obj) {
-    return Object.fromEntries(Object.entries(obj).filter(([_, value]) => value !== null));
-}
-//# sourceMappingURL=objectAsAttributes.js.map

package/dist/src/utils/objectAsAttributes.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"objectAsAttributes.js","sourceRoot":"","sources":["../../../src/utils/objectAsAttributes.ts"],"names":[],"mappings":";;AAEA,gDAMC;AAND,SAAgB,kBAAkB,CAChC,GAAM;IAEN,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC,CACzB,CAAC;AACtC,CAAC"}

package/src/experiments/instrumentation.ts

@@ -1,75 +0,0 @@
-import { diag, DiagConsoleLogger, DiagLogLevel } from "@opentelemetry/api";
-import { OTLPTraceExporter } from "@opentelemetry/exporter-trace-otlp-proto";
-import { resourceFromAttributes } from "@opentelemetry/resources";
-import {
-  NodeTracerProvider,
-  SpanProcessor,
-} from "@opentelemetry/sdk-trace-node";
-import { SEMRESATTRS_PROJECT_NAME } from "@arizeai/openinference-semantic-conventions";
-import { HeadersOptions } from "openapi-fetch";
-import {
-  OpenInferenceBatchSpanProcessor,
-  OpenInferenceSimpleSpanProcessor,
-} from "@arizeai/openinference-vercel";
-
-/**
- * Creates a provider that exports traces to Phoenix.
- */
-export function createProvider({
-  projectName,
-  baseUrl,
-  headers,
-  useBatchSpanProcessor = true,
-  diagLogLevel,
-}: {
-  projectName: string;
-  headers: HeadersOptions;
-  /**
-   * Whether to use batching for the span processor.
-   * @default true
-   */
-  useBatchSpanProcessor: boolean;
-  /**
-   * The base URL of the Phoenix. Doesn't include the /v1/traces path.
-   */
-  baseUrl: string;
-  /**
-   * The diag log level to set for the built in DiagConsoleLogger instance.
-   * Omit to disable built in logging.
-   */
-  diagLogLevel?: DiagLogLevel;
-}) {
-  if (diagLogLevel) {
-    diag.setLogger(new DiagConsoleLogger(), diagLogLevel);
-  }
-
-  const exporter = new OTLPTraceExporter({
-    url: `${baseUrl}/v1/traces`,
-    headers: Array.isArray(headers) ? Object.fromEntries(headers) : headers,
-  });
-
-  let spanProcessor: SpanProcessor;
-  if (useBatchSpanProcessor) {
-    spanProcessor = new OpenInferenceBatchSpanProcessor({ exporter });
-  } else {
-    spanProcessor = new OpenInferenceSimpleSpanProcessor({ exporter });
-  }
-
-  const provider = new NodeTracerProvider({
-    resource: resourceFromAttributes({
-      [SEMRESATTRS_PROJECT_NAME]: projectName,
-    }),
-    spanProcessors: [spanProcessor],
-  });
-
-  return provider;
-}
-
-/**
- * For dry runs we create a provider that doesn't export traces.
- */
-export function createNoOpProvider() {
-  const provider = new NodeTracerProvider({});
-
-  return provider;
-}