@arizeai/openinference-instrumentation-openai 4.0.5 → 4.0.6

package/src/responsesAttributes.ts

@@ -1,16 +1,15 @@
-import { safelyJSONStringify } from "@arizeai/openinference-core";
-import { SemanticConventions } from "@arizeai/openinference-semantic-conventions";
-
-import { Attributes } from "@opentelemetry/api";
-
-import {
+import type { Attributes } from "@opentelemetry/api";
+import type {
   Response as ResponseType,
   ResponseCreateParamsBase,
   ResponseInputItem,
   ResponseOutputItem,
   ResponseStreamEvent,
 } from "openai/resources/responses/responses";
-import { Stream } from "openai/streaming";
+import type { Stream } from "openai/streaming";
+
+import { safelyJSONStringify } from "@arizeai/openinference-core";
+import { SemanticConventions } from "@arizeai/openinference-semantic-conventions";
 
 /**
  * Get attributes for responses api Items that are not typical messages with role
@@ -31,23 +30,17 @@ function getResponseItemAttributes(
     case "function_call": {
       attributes[`${prefix}${SemanticConventions.MESSAGE_ROLE}`] = "assistant";
       // now using tool call prefix to simulate multiple tool calls per message
-      attributes[`${toolCallPrefix}${SemanticConventions.TOOL_CALL_ID}`] =
-        item.call_id;
-      attributes[
-        `${toolCallPrefix}${SemanticConventions.TOOL_CALL_FUNCTION_NAME}`
-      ] = item.name;
-      attributes[
-        `${toolCallPrefix}${SemanticConventions.TOOL_CALL_FUNCTION_ARGUMENTS_JSON}`
-      ] = item.arguments;
+      attributes[`${toolCallPrefix}${SemanticConventions.TOOL_CALL_ID}`] = item.call_id;
+      attributes[`${toolCallPrefix}${SemanticConventions.TOOL_CALL_FUNCTION_NAME}`] = item.name;
+      attributes[`${toolCallPrefix}${SemanticConventions.TOOL_CALL_FUNCTION_ARGUMENTS_JSON}`] =
+        item.arguments;
       break;
     }
     case "function_call_output": {
       attributes[`${prefix}${SemanticConventions.MESSAGE_ROLE}`] = "tool";
-      attributes[`${prefix}${SemanticConventions.MESSAGE_TOOL_CALL_ID}`] =
-        item.call_id;
+      attributes[`${prefix}${SemanticConventions.MESSAGE_TOOL_CALL_ID}`] = item.call_id;
       if (typeof item.output === "string") {
-        attributes[`${prefix}${SemanticConventions.MESSAGE_CONTENT}`] =
-          item.output;
+        attributes[`${prefix}${SemanticConventions.MESSAGE_CONTENT}`] = item.output;
       } else {
         // TODO(2410): figure out how to serialize the list of tools
         attributes[`${prefix}${SemanticConventions.MESSAGE_CONTENT}`] =
@@ -61,12 +54,10 @@ function getResponseItemAttributes(
       item.summary.forEach((summaryItem, index) => {
         const summaryItemPrefix = `${prefix}${SemanticConventions.MESSAGE_CONTENTS}.${index}.`;
         if (summaryItem.type === "summary_text") {
-          attributes[
-            `${summaryItemPrefix}${SemanticConventions.MESSAGE_CONTENT_TYPE}`
-          ] = "summary_text";
-          attributes[
-            `${summaryItemPrefix}${SemanticConventions.MESSAGE_CONTENT_TEXT}`
-          ] = summaryItem.text;
+          attributes[`${summaryItemPrefix}${SemanticConventions.MESSAGE_CONTENT_TYPE}`] =
+            "summary_text";
+          attributes[`${summaryItemPrefix}${SemanticConventions.MESSAGE_CONTENT_TEXT}`] =
+            summaryItem.text;
         }
       });
       break;
@@ -77,56 +68,42 @@ function getResponseItemAttributes(
     case "file_search_call": {
       if (!item.results) {
         // its a tool call
-        attributes[`${prefix}${SemanticConventions.MESSAGE_ROLE}`] =
-          "assistant";
+        attributes[`${prefix}${SemanticConventions.MESSAGE_ROLE}`] = "assistant";
         // now using tool call prefix to simulate multiple tool calls per message
-        attributes[`${toolCallPrefix}${SemanticConventions.TOOL_CALL_ID}`] =
-          item.id;
-        attributes[
-          `${toolCallPrefix}${SemanticConventions.TOOL_CALL_FUNCTION_NAME}`
-        ] = item.type;
-        attributes[
-          `${toolCallPrefix}${SemanticConventions.TOOL_CALL_FUNCTION_ARGUMENTS_JSON}`
-        ] = JSON.stringify(item.queries);
+        attributes[`${toolCallPrefix}${SemanticConventions.TOOL_CALL_ID}`] = item.id;
+        attributes[`${toolCallPrefix}${SemanticConventions.TOOL_CALL_FUNCTION_NAME}`] = item.type;
+        attributes[`${toolCallPrefix}${SemanticConventions.TOOL_CALL_FUNCTION_ARGUMENTS_JSON}`] =
+          JSON.stringify(item.queries);
       } else {
         // its a tool call output
         attributes[`${prefix}${SemanticConventions.MESSAGE_ROLE}`] = "tool";
-        attributes[`${prefix}${SemanticConventions.MESSAGE_TOOL_CALL_ID}`] =
-          item.id;
-        attributes[`${prefix}${SemanticConventions.MESSAGE_CONTENT}`] =
-          JSON.stringify(item.results);
+        attributes[`${prefix}${SemanticConventions.MESSAGE_TOOL_CALL_ID}`] = item.id;
+        attributes[`${prefix}${SemanticConventions.MESSAGE_CONTENT}`] = JSON.stringify(
+          item.results,
+        );
       }
       break;
     }
     case "computer_call": {
       attributes[`${prefix}${SemanticConventions.MESSAGE_ROLE}`] = "assistant";
       // now using tool call prefix to simulate multiple tool calls per message
-      attributes[`${toolCallPrefix}${SemanticConventions.TOOL_CALL_ID}`] =
-        item.id;
-      attributes[
-        `${toolCallPrefix}${SemanticConventions.TOOL_CALL_FUNCTION_NAME}`
-      ] = item.type;
-      attributes[
-        `${toolCallPrefix}${SemanticConventions.TOOL_CALL_FUNCTION_ARGUMENTS_JSON}`
-      ] = JSON.stringify(item.action);
+      attributes[`${toolCallPrefix}${SemanticConventions.TOOL_CALL_ID}`] = item.id;
+      attributes[`${toolCallPrefix}${SemanticConventions.TOOL_CALL_FUNCTION_NAME}`] = item.type;
+      attributes[`${toolCallPrefix}${SemanticConventions.TOOL_CALL_FUNCTION_ARGUMENTS_JSON}`] =
+        JSON.stringify(item.action);
       break;
     }
     case "computer_call_output": {
       attributes[`${prefix}${SemanticConventions.MESSAGE_ROLE}`] = "tool";
-      attributes[`${prefix}${SemanticConventions.MESSAGE_TOOL_CALL_ID}`] =
-        item.call_id;
-      attributes[`${prefix}${SemanticConventions.MESSAGE_CONTENT}`] =
-        JSON.stringify(item.output);
+      attributes[`${prefix}${SemanticConventions.MESSAGE_TOOL_CALL_ID}`] = item.call_id;
+      attributes[`${prefix}${SemanticConventions.MESSAGE_CONTENT}`] = JSON.stringify(item.output);
       break;
     }
     case "web_search_call": {
       attributes[`${prefix}${SemanticConventions.MESSAGE_ROLE}`] = "assistant";
       // now using tool call prefix to simulate multiple tool calls per message
-      attributes[`${toolCallPrefix}${SemanticConventions.TOOL_CALL_ID}`] =
-        item.id;
-      attributes[
-        `${toolCallPrefix}${SemanticConventions.TOOL_CALL_FUNCTION_NAME}`
-      ] = item.type;
+      attributes[`${toolCallPrefix}${SemanticConventions.TOOL_CALL_ID}`] = item.id;
+      attributes[`${toolCallPrefix}${SemanticConventions.TOOL_CALL_FUNCTION_NAME}`] = item.type;
       // web search call does not share its arguments with the caller
       // it will show "undefined" in the arguments when traced
       break;
@@ -159,29 +136,21 @@ function getResponseItemMessageAttributes(
   };
   // add contents from message
   if (typeof message.content === "string") {
-    attributes[`${prefix}${SemanticConventions.MESSAGE_CONTENT}`] =
-      message.content;
+    attributes[`${prefix}${SemanticConventions.MESSAGE_CONTENT}`] = message.content;
   } else if (Array.isArray(message.content)) {
     message.content.forEach((part, index) => {
       const contentsIndexPrefix = `${prefix}${SemanticConventions.MESSAGE_CONTENTS}.${index}.`;
       if (part.type === "input_text") {
-        attributes[
-          `${contentsIndexPrefix}${SemanticConventions.MESSAGE_CONTENT_TYPE}`
-        ] = "input_text";
-        attributes[
-          `${contentsIndexPrefix}${SemanticConventions.MESSAGE_CONTENT_TEXT}`
-        ] = part.text;
+        attributes[`${contentsIndexPrefix}${SemanticConventions.MESSAGE_CONTENT_TYPE}`] =
+          "input_text";
+        attributes[`${contentsIndexPrefix}${SemanticConventions.MESSAGE_CONTENT_TEXT}`] = part.text;
       } else if (part.type === "output_text") {
-        attributes[
-          `${contentsIndexPrefix}${SemanticConventions.MESSAGE_CONTENT_TYPE}`
-        ] = "output_text";
-        attributes[
-          `${contentsIndexPrefix}${SemanticConventions.MESSAGE_CONTENT_TEXT}`
-        ] = part.text;
+        attributes[`${contentsIndexPrefix}${SemanticConventions.MESSAGE_CONTENT_TYPE}`] =
+          "output_text";
+        attributes[`${contentsIndexPrefix}${SemanticConventions.MESSAGE_CONTENT_TEXT}`] = part.text;
       } else if (part.type === "input_image") {
-        attributes[
-          `${contentsIndexPrefix}${SemanticConventions.MESSAGE_CONTENT_TYPE}`
-        ] = "input_image";
+        attributes[`${contentsIndexPrefix}${SemanticConventions.MESSAGE_CONTENT_TYPE}`] =
+          "input_image";
         if (part.image_url) {
           attributes[
             `${contentsIndexPrefix}${SemanticConventions.MESSAGE_CONTENT_IMAGE}.${SemanticConventions.IMAGE_URL}`
@@ -190,12 +159,9 @@ function getResponseItemMessageAttributes(
       } else if (part.type === "input_file") {
         // TODO: Handle input file
       } else if (part.type === "refusal") {
-        attributes[
-          `${contentsIndexPrefix}${SemanticConventions.MESSAGE_CONTENT_TYPE}`
-        ] = "refusal";
-        attributes[
-          `${contentsIndexPrefix}${SemanticConventions.MESSAGE_CONTENT_TEXT}`
-        ] = part.refusal;
+        attributes[`${contentsIndexPrefix}${SemanticConventions.MESSAGE_CONTENT_TYPE}`] = "refusal";
+        attributes[`${contentsIndexPrefix}${SemanticConventions.MESSAGE_CONTENT_TEXT}`] =
+          part.refusal;
       }
     });
   }
@@ -219,9 +185,7 @@ function getResponseItemMessageAttributes(
   return attributes;
 }
 
-export function getResponsesInputMessagesAttributes(
-  body: ResponseCreateParamsBase,
-): Attributes {
+export function getResponsesInputMessagesAttributes(body: ResponseCreateParamsBase): Attributes {
   const attributes: Attributes = {};
   const items: ResponseInputItem[] = [];
   if (body.instructions) {
@@ -243,13 +207,10 @@ export function getResponsesInputMessagesAttributes(
   return attributes;
 }
 
-export function getResponsesUsageAttributes(
-  response: ResponseType,
-): Attributes {
+export function getResponsesUsageAttributes(response: ResponseType): Attributes {
   if (response.usage) {
     return {
-      [SemanticConventions.LLM_TOKEN_COUNT_COMPLETION]:
-        response.usage.output_tokens,
+      [SemanticConventions.LLM_TOKEN_COUNT_COMPLETION]: response.usage.output_tokens,
       [SemanticConventions.LLM_TOKEN_COUNT_PROMPT]: response.usage.input_tokens,
       [SemanticConventions.LLM_TOKEN_COUNT_TOTAL]: response.usage.total_tokens,
       [SemanticConventions.LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_READ]:
@@ -262,9 +223,7 @@ export function getResponsesUsageAttributes(
   return {};
 }
 
-export function getResponsesOutputMessagesAttributes(
-  response: ResponseType,
-): Attributes {
+export function getResponsesOutputMessagesAttributes(response: ResponseType): Attributes {
   const attributes: Attributes = {};
   const items = response.output;
   items.forEach((item, index) => {

package/src/version.ts

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