@arizeai/openinference-instrumentation-beeai 1.5.5 → 1.5.11

package/package.json

@@ -1,56 +1,57 @@
 {
   "name": "@arizeai/openinference-instrumentation-beeai",
-  "version": "1.5.5",
+  "version": "1.5.11",
+  "private": false,
+  "description": "OpenInference Instrumentation for BeeAI framework",
+  "keywords": [
+    "beeai",
+    "opentelemetry",
+    "tracing"
+  ],
   "license": "Apache-2.0",
   "author": "gallas.milan@gmail.com",
-  "description": "OpenInference Instrumentation for BeeAI framework",
-  "private": false,
-  "main": "dist/src/index.js",
-  "module": "dist/esm/index.js",
-  "esnext": "dist/esnext/index.js",
-  "types": "dist/src/index.d.ts",
   "repository": {
     "type": "git",
     "url": "https://github.com/Arize-ai/openinference.git"
   },
+  "files": [
+    "dist",
+    "src"
+  ],
+  "main": "dist/src/index.js",
+  "module": "dist/esm/index.js",
+  "types": "dist/src/index.d.ts",
+  "esnext": "dist/esnext/index.js",
   "exports": {
     ".": {
       "import": "./dist/esm/index.js",
       "require": "./dist/src/index.js"
     }
   },
-  "files": [
-    "dist",
-    "src"
-  ],
-  "keywords": [
-    "beeai",
-    "opentelemetry",
-    "tracing"
-  ],
   "dependencies": {
     "@opentelemetry/api": "^1.9.0",
     "@opentelemetry/instrumentation": "^0.57.1",
     "remeda": "^2.20.2",
     "semver": "^7.7.0",
-    "@arizeai/openinference-core": "2.0.0",
-    "@arizeai/openinference-semantic-conventions": "2.1.2"
-  },
-  "peerDependencies": {
-    "beeai-framework": "^0.1.13"
+    "@arizeai/openinference-core": "2.0.6",
+    "@arizeai/openinference-semantic-conventions": "2.2.0"
   },
   "devDependencies": {
-    "@types/node": "^20.14.11",
-    "@types/semver": "7.7.0",
-    "beeai-framework": "^0.1.13",
-    "import-in-the-middle": "^1.13.0",
     "@opentelemetry/exporter-trace-otlp-proto": "^0.50.0",
     "@opentelemetry/resources": "^1.30.1",
     "@opentelemetry/sdk-node": "^0.57.1",
     "@opentelemetry/sdk-trace-node": "^1.30.1",
     "@opentelemetry/semantic-conventions": "^1.28.0",
-    "vitest": "^4.0.3",
-    "ts-node": "^10.9.2"
+    "@types/node": "^20.14.11",
+    "@types/semver": "7.7.0",
+    "beeai-framework": "^0.1.13",
+    "import-in-the-middle": "^1.13.0",
+    "ts-node": "^10.9.2",
+    "typescript": "^5.8.3",
+    "vitest": "^4.0.3"
+  },
+  "peerDependencies": {
+    "beeai-framework": "^0.1.13"
   },
   "optionalDependencies": {
     "ollama": "^0.5.12"
@@ -60,6 +61,6 @@
     "build": "tsc --build tsconfig.json tsconfig.esm.json tsconfig.esnext.json && tsc-alias -p tsconfig.esm.json",
     "postbuild": "echo '{\"type\": \"module\"}' > ./dist/esm/package.json && rimraf dist/test",
     "type:check": "tsc --noEmit",
-    "test": "vitest --typecheck"
+    "test": "vitest run --typecheck"
   }
 }

package/src/config.ts

@@ -15,18 +15,14 @@
  */
 
 import type { ReActAgentCallbacks } from "beeai-framework/agents/react/types";
-import { ChatModelEvents } from "beeai-framework/backend/chat";
-import { ToolEvents } from "beeai-framework/tools/base";
+import type { ChatModelEvents } from "beeai-framework/backend/chat";
+import type { ToolEvents } from "beeai-framework/tools/base";
 
-export const INSTRUMENTATION_IGNORED_KEYS = process.env
-  .BEE_FRAMEWORK_INSTRUMENTATION_IGNORED_KEYS
-  ? process.env.BEE_FRAMEWORK_INSTRUMENTATION_IGNORED_KEYS.split(",").filter(
-      Boolean,
-    )
+export const INSTRUMENTATION_IGNORED_KEYS = process.env.BEE_FRAMEWORK_INSTRUMENTATION_IGNORED_KEYS
+  ? process.env.BEE_FRAMEWORK_INSTRUMENTATION_IGNORED_KEYS.split(",").filter(Boolean)
   : [];
 
-export const partialUpdateEventName: keyof ReActAgentCallbacks =
-  "partialUpdate";
+export const partialUpdateEventName: keyof ReActAgentCallbacks = "partialUpdate";
 export const updateEventName: keyof ReActAgentCallbacks = "update";
 export const toolErrorEventName: keyof ReActAgentCallbacks = "toolError";
 export const toolStartEventName: keyof ReActAgentCallbacks = "toolStart";

package/src/helpers/buildTraceTree.ts

@@ -1,13 +1,14 @@
-import { OITracer } from "@arizeai/openinference-core";
+import type { TimeInput } from "@opentelemetry/api";
+import { SpanStatusCode } from "@opentelemetry/api";
+
+import type { OITracer } from "@arizeai/openinference-core";
 import {
   OpenInferenceSpanKind,
   SemanticConventions,
 } from "@arizeai/openinference-semantic-conventions";
 
-import { SpanStatusCode, TimeInput } from "@opentelemetry/api";
-
 import { startEventName, successEventName } from "../config";
-import { FrameworkSpan, GeneratedResponse } from "../types";
+import type { FrameworkSpan, GeneratedResponse } from "../types";
 
 interface BuiltTraceTreeProps {
   tracer: OITracer;
@@ -87,12 +88,8 @@ function buildAgentMainSpanData(data: BuiltTraceTreeProps["data"]) {
 }
 
 function buildToolMainSpanData(data: BuiltTraceTreeProps["data"]) {
-  const startBeeaiSpan = data.spans.find(
-    (span) => span.name === startEventName,
-  );
-  const successBeeeaiSpan = data.spans.find(
-    (span) => span.name === successEventName,
-  );
+  const startBeeaiSpan = data.spans.find((span) => span.name === startEventName);
+  const successBeeeaiSpan = data.spans.find((span) => span.name === successEventName);
 
   return {
     ...(startBeeaiSpan && {
@@ -109,12 +106,8 @@ function buildToolMainSpanData(data: BuiltTraceTreeProps["data"]) {
 }
 
 function buildLLMMainSpanData(data: BuiltTraceTreeProps["data"]) {
-  const startBeeaiSpan = data.spans.find(
-    (span) => span.name === startEventName,
-  );
-  const successBeeeaiSpan = data.spans.find(
-    (span) => span.name === successEventName,
-  );
+  const startBeeaiSpan = data.spans.find((span) => span.name === startEventName);
+  const successBeeeaiSpan = data.spans.find((span) => span.name === successEventName);
 
   if (!startBeeaiSpan && !successBeeeaiSpan) return {};
 
@@ -135,9 +128,7 @@ function buildLLMMainSpanData(data: BuiltTraceTreeProps["data"]) {
     }),
     ...(successBeeeaiSpan && {
       [SemanticConventions.OUTPUT_MIME_TYPE]:
-        successBeeeaiSpan.attributes.data?.[
-          SemanticConventions.OUTPUT_MIME_TYPE
-        ],
+        successBeeeaiSpan.attributes.data?.[SemanticConventions.OUTPUT_MIME_TYPE],
       [SemanticConventions.OUTPUT_VALUE]:
         successBeeeaiSpan.attributes.data?.[SemanticConventions.OUTPUT_VALUE],
     }),
@@ -146,11 +137,7 @@ function buildLLMMainSpanData(data: BuiltTraceTreeProps["data"]) {
   };
 }
 
-export function buildTraceTree({
-  tracer,
-  data,
-  mainSpanKind,
-}: BuiltTraceTreeProps) {
+export function buildTraceTree({ tracer, data, mainSpanKind }: BuiltTraceTreeProps) {
   const computedData =
     mainSpanKind === OpenInferenceSpanKind.AGENT
       ? buildAgentMainSpanData(data)

package/src/helpers/create-span.ts

@@ -14,12 +14,12 @@
  * limitations under the License.
  */
 
-import { SpanStatusCode, TimeInput } from "@opentelemetry/api";
-
-import { FrameworkSpan } from "../types";
-
+import type { TimeInput } from "@opentelemetry/api";
+import { SpanStatusCode } from "@opentelemetry/api";
 import { isEmpty } from "remeda";
 
+import type { FrameworkSpan } from "../types";
+
 interface CreateSpanProps {
   id: string;
   name: string;

package/src/helpers/getSerializedObjectSafe.ts

@@ -14,6 +14,18 @@
  * limitations under the License.
  */
 
+import { diag } from "@opentelemetry/api";
+import { BaseAgent } from "beeai-framework/agents/base";
+import type { ReActAgentCallbacks } from "beeai-framework/agents/react/types";
+import type { ChatModelEvents } from "beeai-framework/backend/chat";
+import { ChatModel } from "beeai-framework/backend/chat";
+import type { Message, MessageContentPart } from "beeai-framework/backend/message";
+import type { EventMeta, InferCallbackValue } from "beeai-framework/emitter/types";
+import { getProp } from "beeai-framework/internals/helpers/object";
+import { Serializable } from "beeai-framework/internals/serializable";
+import type { ToolEvents } from "beeai-framework/tools/base";
+import { Tool } from "beeai-framework/tools/base";
+
 import {
   LLMAttributePostfixes,
   MessageAttributePostfixes,
@@ -23,8 +35,6 @@ import {
   SemanticConventions,
 } from "@arizeai/openinference-semantic-conventions";
 
-import { diag } from "@opentelemetry/api";
-
 import {
   errorEventName,
   errorLLMEventName,
@@ -47,18 +57,7 @@ import {
   updateEventName,
 } from "../config";
 
-import { BaseAgent } from "beeai-framework/agents/base";
-import type { ReActAgentCallbacks } from "beeai-framework/agents/react/types";
-import { ChatModel, ChatModelEvents } from "beeai-framework/backend/chat";
-import { Message, MessageContentPart } from "beeai-framework/backend/message";
-import { EventMeta, InferCallbackValue } from "beeai-framework/emitter/types";
-import { getProp } from "beeai-framework/internals/helpers/object";
-import { Serializable } from "beeai-framework/internals/serializable";
-import { Tool, ToolEvents } from "beeai-framework/tools/base";
-
-function parserLLMInputMessages(
-  messages: readonly Message<MessageContentPart, string>[],
-) {
+function parserLLMInputMessages(messages: readonly Message<MessageContentPart, string>[]) {
   return messages.reduce(
     (acc, item, key) => ({
       ...acc,
@@ -74,9 +73,7 @@ function parserLLMInputMessages(
   );
 }
 
-function parseLLMOutputMessages(
-  messages: readonly Message<MessageContentPart, string>[],
-) {
+function parseLLMOutputMessages(messages: readonly Message<MessageContentPart, string>[]) {
   return messages.reduce(
     (acc, item, key) => ({
       ...acc,
@@ -92,31 +89,23 @@ function parseLLMOutputMessages(
   );
 }
 
-export function getSerializedObjectSafe(
-  dataObject: unknown,
-  meta: EventMeta<unknown>,
-) {
+export function getSerializedObjectSafe(dataObject: unknown, meta: EventMeta<unknown>) {
   try {
     // agent events
     if (
-      [
-        startEventName,
-        successEventName,
-        errorEventName,
-        retryEventName,
-      ].includes(meta.name as keyof ReActAgentCallbacks) &&
+      [startEventName, successEventName, errorEventName, retryEventName].includes(
+        meta.name as keyof ReActAgentCallbacks,
+      ) &&
       meta.creator instanceof BaseAgent
     ) {
-      const { meta, tools, memory, error, data } =
-        dataObject as InferCallbackValue<
-          | ReActAgentCallbacks["start"]
-          | ReActAgentCallbacks["error"]
-          | ReActAgentCallbacks["success"]
-          | ReActAgentCallbacks["retry"]
-        >;
+      const { meta, tools, memory, error, data } = dataObject as InferCallbackValue<
+        | ReActAgentCallbacks["start"]
+        | ReActAgentCallbacks["error"]
+        | ReActAgentCallbacks["success"]
+        | ReActAgentCallbacks["retry"]
+      >;
       return {
-        [SemanticConventions.OPENINFERENCE_SPAN_KIND]:
-          OpenInferenceSpanKind.AGENT,
+        [SemanticConventions.OPENINFERENCE_SPAN_KIND]: OpenInferenceSpanKind.AGENT,
         iteration: meta?.iteration,
         ...(tools?.length > 0 && {
           [SemanticConventions.LLM_TOOLS]: tools.map((tool) => ({
@@ -143,9 +132,7 @@ export function getSerializedObjectSafe(
 
     // update events
     if (
-      [updateEventName, partialUpdateEventName].includes(
-        meta.name as keyof ReActAgentCallbacks,
-      )
+      [updateEventName, partialUpdateEventName].includes(meta.name as keyof ReActAgentCallbacks)
     ) {
       const { data } = dataObject as InferCallbackValue<
         ReActAgentCallbacks["partialUpdate"] | ReActAgentCallbacks["update"]
@@ -153,16 +140,13 @@ export function getSerializedObjectSafe(
 
       const output = data?.final_answer || data?.tool_output;
       return {
-        [SemanticConventions.OPENINFERENCE_SPAN_KIND]:
-          OpenInferenceSpanKind.AGENT,
+        [SemanticConventions.OPENINFERENCE_SPAN_KIND]: OpenInferenceSpanKind.AGENT,
         thought: data.thought,
         ...(data?.tool_name && {
           [SemanticConventions.TOOL_NAME]: data.tool_name,
         }),
         ...(data?.tool_input && {
-          [SemanticConventions.TOOL_PARAMETERS]: JSON.stringify(
-            data.tool_input,
-          ),
+          [SemanticConventions.TOOL_PARAMETERS]: JSON.stringify(data.tool_input),
         }),
         ...(output && {
           [SemanticConventions.OUTPUT_MIME_TYPE]: MimeType.JSON,
@@ -186,8 +170,7 @@ export function getSerializedObjectSafe(
       const output: any = data?.result && data.result.createSnapshot();
 
       return {
-        [SemanticConventions.OPENINFERENCE_SPAN_KIND]:
-          OpenInferenceSpanKind.TOOL,
+        [SemanticConventions.OPENINFERENCE_SPAN_KIND]: OpenInferenceSpanKind.TOOL,
         thought: data?.iteration?.thought,
         ...(data?.input
           ? {
@@ -223,23 +206,16 @@ export function getSerializedObjectSafe(
     ) {
       if (!dataObject) {
         return {
-          [SemanticConventions.OPENINFERENCE_SPAN_KIND]:
-            OpenInferenceSpanKind.TOOL,
+          [SemanticConventions.OPENINFERENCE_SPAN_KIND]: OpenInferenceSpanKind.TOOL,
         };
       }
       const { input, output, error } = dataObject as InferCallbackValue<
-        | ToolEvents["start"]
-        | ToolEvents["success"]
-        | ToolEvents["retry"]
-        | ToolEvents["error"]
+        ToolEvents["start"] | ToolEvents["success"] | ToolEvents["retry"] | ToolEvents["error"]
       >;
 
       return {
-        [SemanticConventions.OPENINFERENCE_SPAN_KIND]:
-          OpenInferenceSpanKind.TOOL,
-        ...(input
-          ? { [SemanticConventions.TOOL_PARAMETERS]: JSON.stringify(input) }
-          : {}),
+        [SemanticConventions.OPENINFERENCE_SPAN_KIND]: OpenInferenceSpanKind.TOOL,
+        ...(input ? { [SemanticConventions.TOOL_PARAMETERS]: JSON.stringify(input) } : {}),
         ...(output && {
           [SemanticConventions.OUTPUT_VALUE]: JSON.stringify(output),
         }),
@@ -253,12 +229,9 @@ export function getSerializedObjectSafe(
 
     // llm events
     if (
-      [
-        successLLMEventName,
-        startLLMEventName,
-        errorLLMEventName,
-        newTokenLLMEventName,
-      ].includes(meta.name as keyof ChatModelEvents) &&
+      [successLLMEventName, startLLMEventName, errorLLMEventName, newTokenLLMEventName].includes(
+        meta.name as keyof ChatModelEvents,
+      ) &&
       meta.creator instanceof ChatModel
     ) {
       const { value, input, error } = dataObject as InferCallbackValue<
@@ -270,19 +243,15 @@ export function getSerializedObjectSafe(
 
       const creator = meta.creator.createSnapshot();
       return {
-        [SemanticConventions.OPENINFERENCE_SPAN_KIND]:
-          OpenInferenceSpanKind.LLM,
+        [SemanticConventions.OPENINFERENCE_SPAN_KIND]: OpenInferenceSpanKind.LLM,
         ...(value?.usage?.completionTokens && {
-          [SemanticConventions.LLM_TOKEN_COUNT_COMPLETION]:
-            value?.usage?.completionTokens,
+          [SemanticConventions.LLM_TOKEN_COUNT_COMPLETION]: value?.usage?.completionTokens,
         }),
         ...(value?.usage?.promptTokens && {
-          [SemanticConventions.LLM_TOKEN_COUNT_PROMPT]:
-            value?.usage?.promptTokens,
+          [SemanticConventions.LLM_TOKEN_COUNT_PROMPT]: value?.usage?.promptTokens,
         }),
         ...(value?.usage?.totalTokens && {
-          [SemanticConventions.LLM_TOKEN_COUNT_TOTAL]:
-            value?.usage?.totalTokens,
+          [SemanticConventions.LLM_TOKEN_COUNT_TOTAL]: value?.usage?.totalTokens,
         }),
         ...(input?.messages.length > 0 && {
           [SemanticConventions.INPUT_MIME_TYPE]: MimeType.JSON,
@@ -310,16 +279,13 @@ export function getSerializedObjectSafe(
             creator.modelId,
         }),
         ...(creator?.parameters && {
-          [SemanticConventions.LLM_INVOCATION_PARAMETERS]: JSON.stringify(
-            creator.parameters,
-          ),
+          [SemanticConventions.LLM_INVOCATION_PARAMETERS]: JSON.stringify(creator.parameters),
         }),
       };
     }
     if (meta.name === finishLLMEventName && meta.creator instanceof ChatModel) {
       return {
-        [SemanticConventions.OPENINFERENCE_SPAN_KIND]:
-          OpenInferenceSpanKind.LLM,
+        [SemanticConventions.OPENINFERENCE_SPAN_KIND]: OpenInferenceSpanKind.LLM,
       };
     }
 

package/src/helpers/traceSerializer.ts

@@ -14,19 +14,9 @@
  * limitations under the License.
  */
 
-const DEFAULT_IGNORE_KEYS = [
-  "emitter",
-  "logger",
-  "tokens",
-  "createdBy",
-  "client",
-];
+const DEFAULT_IGNORE_KEYS = ["emitter", "logger", "tokens", "createdBy", "client"];
 
-export function traceSerializer({
-  ignored_keys = [],
-}: {
-  ignored_keys?: string[];
-}) {
+export function traceSerializer({ ignored_keys = [] }: { ignored_keys?: string[] }) {
   const mergedIgnoreKeys = new Set([...DEFAULT_IGNORE_KEYS, ...ignored_keys]);
 
   return (body: object) =>

package/src/instrumentation.ts

@@ -1,19 +1,20 @@
-import { OITracer, TraceConfigOptions } from "@arizeai/openinference-core";
-import { OpenInferenceSpanKind } from "@arizeai/openinference-semantic-conventions";
-
-import { diag, Tracer, TracerProvider } from "@opentelemetry/api";
+import type { Tracer, TracerProvider } from "@opentelemetry/api";
+import { diag } from "@opentelemetry/api";
+import type { InstrumentationConfig } from "@opentelemetry/instrumentation";
 import {
   InstrumentationBase,
-  InstrumentationConfig,
   InstrumentationNodeModuleDefinition,
 } from "@opentelemetry/instrumentation";
-
-import { createTelemetryMiddleware } from "./middleware";
-
 import { Version } from "beeai-framework";
-import * as bee from "beeai-framework";
+import type * as bee from "beeai-framework";
 import { satisfies } from "semver";
 
+import type { TraceConfigOptions } from "@arizeai/openinference-core";
+import { OITracer } from "@arizeai/openinference-core";
+import { OpenInferenceSpanKind } from "@arizeai/openinference-semantic-conventions";
+
+import { createTelemetryMiddleware } from "./middleware";
+
 const MODULE_NAME = "beeai-framework";
 
 const INSTRUMENTATION_NAME = "@arizeai/openinference-instrumentation-beeai";
@@ -61,16 +62,11 @@ export class BeeAIInstrumentation extends InstrumentationBase {
      */
     tracerProvider?: TracerProvider;
   } = {}) {
-    super(
-      INSTRUMENTATION_NAME,
-      Version,
-      Object.assign({}, instrumentationConfig),
-    );
+    super(INSTRUMENTATION_NAME, Version, Object.assign({}, instrumentationConfig));
     this.tracerProvider = tracerProvider;
     this.traceConfig = traceConfig;
     this.oiTracer = new OITracer({
-      tracer:
-        this.tracerProvider?.getTracer(INSTRUMENTATION_NAME) ?? this.tracer,
+      tracer: this.tracerProvider?.getTracer(INSTRUMENTATION_NAME) ?? this.tracer,
       traceConfig,
     });
   }
@@ -91,9 +87,7 @@ export class BeeAIInstrumentation extends InstrumentationBase {
    * @param {openai} module
    */
   manuallyInstrument(module: typeof bee) {
-    if (
-      !INSTRUMENTS.some((instrument) => satisfies(module.Version, instrument))
-    ) {
+    if (!INSTRUMENTS.some((instrument) => satisfies(module.Version, instrument))) {
       const supportedVersionsString = `beeai-framework ["${INSTRUMENTS.join(',"')}"]`;
       diag.warn(
         `DependencyConflict: requested: '${supportedVersionsString}' but found: 'beeai-framework "${module.Version}"'`,
@@ -123,9 +117,7 @@ export class BeeAIInstrumentation extends InstrumentationBase {
 
   private patch(module: typeof bee & { openInferencePatched?: boolean }) {
     if (!module?.BaseAgent) {
-      diag.warn(
-        "[BeeaiInstrumentation] BaseAgent not found, skipping instrumentation.",
-      );
+      diag.warn("[BeeaiInstrumentation] BaseAgent not found, skipping instrumentation.");
       return module;
     }
 
@@ -142,9 +134,7 @@ export class BeeAIInstrumentation extends InstrumentationBase {
       return function wrappedMethod(this: unknown, ...args: unknown[]) {
         const returned = original.apply(this, args);
         if (returned?.middleware) {
-          returned.middleware(
-            createTelemetryMiddleware(instrumentation.oiTracer, mainSpanKind),
-          );
+          returned.middleware(createTelemetryMiddleware(instrumentation.oiTracer, mainSpanKind));
         }
         return returned;
       };
@@ -155,47 +145,27 @@ export class BeeAIInstrumentation extends InstrumentationBase {
     const instrumentation: BeeAIInstrumentation = this;
 
     this._wrap(module.BaseAgent.prototype, "run", (original) =>
-      createWrappedMethod(
-        instrumentation,
-        original,
-        OpenInferenceSpanKind.AGENT,
-      ),
+      createWrappedMethod(instrumentation, original, OpenInferenceSpanKind.AGENT),
     );
 
     // tool instrumentation support
     if (module.Tool) {
       this._wrap(module.Tool.prototype, "run", (original) =>
-        createWrappedMethod(
-          instrumentation,
-          original,
-          OpenInferenceSpanKind.TOOL,
-        ),
+        createWrappedMethod(instrumentation, original, OpenInferenceSpanKind.TOOL),
       );
     }
     // model instrumentation support
     if (module.ChatModel) {
       this._wrap(module.ChatModel.prototype, "create", (original) =>
-        createWrappedMethod(
-          instrumentation,
-          original,
-          OpenInferenceSpanKind.LLM,
-        ),
+        createWrappedMethod(instrumentation, original, OpenInferenceSpanKind.LLM),
       );
       this._wrap(module.ChatModel.prototype, "createStructure", (original) =>
-        createWrappedMethod(
-          instrumentation,
-          original,
-          OpenInferenceSpanKind.LLM,
-        ),
+        createWrappedMethod(instrumentation, original, OpenInferenceSpanKind.LLM),
       );
     }
     if (module.EmbeddingModel) {
       this._wrap(module.EmbeddingModel.prototype, "create", (original) =>
-        createWrappedMethod(
-          instrumentation,
-          original,
-          OpenInferenceSpanKind.EMBEDDING,
-        ),
+        createWrappedMethod(instrumentation, original, OpenInferenceSpanKind.EMBEDDING),
       );
     }