@arizeai/openinference-instrumentation-openai 3.2.3 → 3.2.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@arizeai/openinference-instrumentation-openai",
-  "version": "3.2.3",
+  "version": "3.2.5",
   "description": "OpenInference instrumentation for OpenAI",
   "private": false,
   "main": "dist/src/index.js",
@@ -21,7 +21,7 @@
     "@opentelemetry/api": "^1.9.0",
     "@opentelemetry/core": "^1.25.1",
     "@opentelemetry/instrumentation": "^0.46.0",
-    "@arizeai/openinference-core": "1.0.7",
+    "@arizeai/openinference-core": "2.0.0",
     "@arizeai/openinference-semantic-conventions": "2.1.2"
   },
   "keywords": [],
@@ -39,7 +39,8 @@
     "@opentelemetry/semantic-conventions": "^1.25.1",
     "vitest": "^2.1.8",
     "openai": "^5.7.0",
-    "zod": "^3.24.3"
+    "zod": "^3.24.3",
+    "typescript": "^5.5.4"
   },
   "scripts": {
     "prebuild": "rimraf dist && pnpm run version:update",

package/src/instrumentation.ts

@@ -1,4 +1,28 @@
-import openai, { APIPromise } from "openai";
+import {
+  OITracer,
+  safelyJSONStringify,
+  TraceConfigOptions,
+} from "@arizeai/openinference-core";
+import {
+  LLMProvider,
+  LLMSystem,
+  MimeType,
+  OpenInferenceSpanKind,
+  SemanticConventions,
+} from "@arizeai/openinference-semantic-conventions";
+
+import {
+  Attributes,
+  context,
+  diag,
+  Span,
+  SpanKind,
+  SpanStatusCode,
+  trace,
+  Tracer,
+  TracerProvider,
+} from "@opentelemetry/api";
+import { isTracingSuppressed } from "@opentelemetry/core";
 import {
   InstrumentationBase,
   InstrumentationConfig,
@@ -6,25 +30,22 @@ import {
   InstrumentationNodeModuleDefinition,
   safeExecuteInTheMiddle,
 } from "@opentelemetry/instrumentation";
+
 import {
-  diag,
-  context,
-  trace,
-  SpanKind,
-  Attributes,
-  SpanStatusCode,
-  Span,
-  TracerProvider,
-  Tracer,
-} from "@opentelemetry/api";
+  consumeResponseStreamEvents,
+  getResponsesInputMessagesAttributes,
+  getResponsesOutputMessagesAttributes,
+  getResponsesUsageAttributes,
+} from "./responsesAttributes";
+import { assertUnreachable, isString } from "./typeUtils";
 import { VERSION } from "./version";
+
+import openai, { APIPromise } from "openai";
 import {
-  SemanticConventions,
-  OpenInferenceSpanKind,
-  MimeType,
-  LLMSystem,
-  LLMProvider,
-} from "@arizeai/openinference-semantic-conventions";
+  Completion,
+  CreateEmbeddingResponse,
+  EmbeddingCreateParams,
+} from "openai/resources";
 import {
   ChatCompletion,
   ChatCompletionChunk,
@@ -33,32 +54,12 @@ import {
   ChatCompletionMessageParam,
 } from "openai/resources/chat/completions";
 import { CompletionCreateParamsBase } from "openai/resources/completions";
-import { Stream } from "openai/streaming";
-import {
-  Completion,
-  CreateEmbeddingResponse,
-  EmbeddingCreateParams,
-} from "openai/resources";
-import { assertUnreachable, isString } from "./typeUtils";
-import { isTracingSuppressed } from "@opentelemetry/core";
-
-import {
-  OITracer,
-  safelyJSONStringify,
-  TraceConfigOptions,
-} from "@arizeai/openinference-core";
 import {
+  Response as ResponseType,
   ResponseCreateParamsBase,
   ResponseStreamEvent,
-  Response as ResponseType,
 } from "openai/resources/responses/responses";
-
-import {
-  consumeResponseStreamEvents,
-  getResponsesInputMessagesAttributes,
-  getResponsesOutputMessagesAttributes,
-  getResponsesUsageAttributes,
-} from "./responsesAttributes";
+import { Stream } from "openai/streaming";
 
 const MODULE_NAME = "openai";
 
@@ -746,7 +747,7 @@ function getChatCompletionInputMessageAttributes(
             ] = toolCall.id;
           }
           // Make sure the tool call has a function
-          if (toolCall.function) {
+          if (toolCall.type === "function") {
             attributes[
               `${toolCallIndexPrefix}${SemanticConventions.TOOL_CALL_FUNCTION_NAME}`
             ] = toolCall.function.name;
@@ -875,7 +876,7 @@ function getChatCompletionOutputMessageAttributes(
       }
       // Double check that the tool call has a function
       // NB: OpenAI only supports tool calls with functions right now but this may change
-      if (toolCall.function) {
+      if (toolCall.type === "function") {
         attributes[
           toolCallIndexPrefix + SemanticConventions.TOOL_CALL_FUNCTION_NAME
         ] = toolCall.function.name;

package/src/responsesAttributes.ts

@@ -1,11 +1,13 @@
 import { SemanticConventions } from "@arizeai/openinference-semantic-conventions";
+
 import { Attributes } from "@opentelemetry/api";
+
 import {
+  Response as ResponseType,
   ResponseCreateParamsBase,
   ResponseInputItem,
   ResponseOutputItem,
   ResponseStreamEvent,
-  Response as ResponseType,
 } from "openai/resources/responses/responses";
 import { Stream } from "openai/streaming";
 

package/src/version.ts

@@ -1,2 +1,2 @@
 // this is an auto-generated file. see js/scripts/version-update.js
-export const VERSION = "3.2.3";
+export const VERSION = "3.2.5";