update otel test

pjirsa · web-flow · commit 016aa2220f61 · 2025-03-02T07:48:33.000-06:00
diff --git a/aiservices/openai/src/test/java/com/microsoft/semantickernel/aiservices/openai/OtelCaptureTest.java b/aiservices/openai/src/test/java/com/microsoft/semantickernel/aiservices/openai/OtelCaptureTest.java
@@ -118,9 +118,9 @@ public void otelChatCaptureTest() {
             spans.get(0).getAttributes().get(AttributeKey.stringKey("gen_ai.system")));
         Assertions.assertEquals(21,
             spans.get(0).getAttributes()
-                .get(AttributeKey.longKey("gen_ai.response.completion_tokens")));
+                .get(AttributeKey.longKey("gen_ai.usage.output_tokens")));
         Assertions.assertEquals(42,
             spans.get(0).getAttributes()
-                .get(AttributeKey.longKey("gen_ai.response.prompt_tokens")));
+                .get(AttributeKey.longKey("gen_ai.usage.input_tokens")));
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -118,9 +118,9 @@ public void otelChatCaptureTest() {`
`118`	`118`	`spans.get(0).getAttributes().get(AttributeKey.stringKey("gen_ai.system")));`
`119`	`119`	`Assertions.assertEquals(21,`
`120`	`120`	`spans.get(0).getAttributes()`
`121`		`- .get(AttributeKey.longKey("gen_ai.response.completion_tokens")));`
	`121`	`+ .get(AttributeKey.longKey("gen_ai.usage.output_tokens")));`
`122`	`122`	`Assertions.assertEquals(42,`
`123`	`123`	`spans.get(0).getAttributes()`
`124`		`- .get(AttributeKey.longKey("gen_ai.response.prompt_tokens")));`
	`124`	`+ .get(AttributeKey.longKey("gen_ai.usage.input_tokens")));`
`125`	`125`	`}`
`126`	`126`	`}`