@arizeai/openinference-instrumentation-openai 4.4.2 → 4.4.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@arizeai/openinference-instrumentation-openai",
-  "version": "4.4.2",
+  "version": "4.4.3",
   "private": false,
   "description": "OpenInference instrumentation for OpenAI",
   "keywords": [],

package/src/instrumentation.ts

@@ -556,7 +556,11 @@ export class OpenAIInstrumentation extends InstrumentationBase<typeof openai> {
                   ...getResponsesOutputMessagesAttributes(result),
                   ...getResponsesUsageAttributes(result),
                 });
-                span.setStatus({ code: SpanStatusCode.OK });
+                span.setStatus(
+                  result.status === "failed"
+                    ? { code: SpanStatusCode.ERROR, message: result.error?.message }
+                    : { code: SpanStatusCode.OK },
+                );
                 span.end();
               };
               if (isResponseCreateResponse(result)) {

package/src/responsesAttributes.ts

@@ -255,7 +255,11 @@ export async function consumeResponseStreamEvents(
 
   for await (const event of stream) {
     switch (event.type) {
-      case "response.completed": {
+      // A stream can end on any of these; the incomplete and failed ones still carry the
+      // partial output and the usage.
+      case "response.completed":
+      case "response.incomplete":
+      case "response.failed": {
         response = event.response;
         break;
       }

package/src/version.ts

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